diff options
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/simple-agent-go/main.go | 14 |
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") } } |