libpanto fixes

- ensuring stream.next() sequences are interruptible
- Agent.run() takes any user message, not just chat text

1 files changed, 7 insertions, 7 deletions

diff --git a/examples/simple-agent-go/main.go b/examples/simple-agent-go/main.go
index c733cd7..b8e11d6 100644
--- a/examples/simple-agent-go/main.go
+++ b/examples/simple-agent-go/main.go
@@ -101,20 +101,20 @@ func main() {
 		},
 	})
 
-	stream, err := agent.Run(prompt)
+	stream, err := agent.Run(panto.UserMessage(panto.TextBlock(prompt)))
 	if err != nil {
 		panic(err)
 	}
 	defer stream.Close()
 
 	for ev := range stream.Iter() {
-		switch ev.Tag {
-		case panto.EventContentDelta:
-			fmt.Print(ev.ContentDelta.Delta)
+		switch ev := ev.(type) {
+		case panto.ContentDeltaEvent:
+			fmt.Print(ev.Delta)
 			os.Stdout.Sync()
-		case panto.EventToolDispatchStart:
-			fmt.Fprintf(os.Stderr, "\n[bash: running %d tool call(s)]\n", ev.ToolDispatchStart.Count)
-		case panto.EventToolDispatchComplete:
+		case panto.ToolDispatchStartEvent:
+			fmt.Fprintf(os.Stderr, "\n[bash: running %d tool call(s)]\n", ev.Count)
+		case panto.ToolDispatchCompleteEvent:
 			fmt.Fprint(os.Stderr, "[bash: done]\n")
 		}
 	}