alejandro-angulo/go-ai-agent
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Switched to streaming responses

Browse source
This commit is contained in:
alejandro-angulo 2025-06-06 23:45:45 -07:00
parent 464b9fc979
commit 1c538067af
Signed by: alejandro-angulo
GPG key ID: 75579581C74554B6
1 changed files with 23 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
main.go
View file

@ -75,7 +75,7 @@ func (a *Agent) Run(ctx context.Context) error {
for _, content := range message.Content { for _, content := range message.Content {
switch content.Type { switch content.Type {
case "text": case "text":
fmt.Printf("\u001b[93mClaude\u001b[0m: %s\n", content.Text) print("\n")
case "tool_use": case "tool_use":
result := a.executeTool(content.ID, content.Name, content.Input) result := a.executeTool(content.ID, content.Name, content.Input)
toolResults = append(toolResults, result) toolResults = append(toolResults, result)
@ -106,7 +106,7 @@ func (a *Agent) executeTool(id, name string, input json.RawMessage) anthropic.Co
return anthropic.NewToolResultBlock(id, "tool not found", true) return anthropic.NewToolResultBlock(id, "tool not found", true)
} }
fmt.Printf("\u001b[92mtool\u001b[0m: %s(%s)\n", name, input) fmt.Printf("\n\u001b[92mtool\u001b[0m: %s(%s)\n", name, input)
response, err := toolDef.Function(input) response, err := toolDef.Function(input)
if err != nil { if err != nil {
return anthropic.NewToolResultBlock(id, err.Error(), true) return anthropic.NewToolResultBlock(id, err.Error(), true)
@ -126,13 +126,32 @@ func (a *Agent) runInference(ctx context.Context, conversation []anthropic.Messa
}) })
} }
message, err := a.client.Messages.New(ctx, anthropic.MessageNewParams{ stream := a.client.Messages.NewStreaming(ctx, anthropic.MessageNewParams{
Model: anthropic.ModelClaude3_7SonnetLatest, Model: anthropic.ModelClaude3_7SonnetLatest,
MaxTokens: int64(1024), MaxTokens: int64(1024),
Messages: conversation, Messages: conversation,
Tools: anthropicTools, Tools: anthropicTools,
}) })
return message, err
print("\u001b[93mClaude\u001b[0m: ")
message := anthropic.Message{}
for stream.Next() {
event := stream.Current()
err := message.Accumulate(event)
if err != nil {
return nil, err
}
switch eventVariant := event.AsAny().(type) {
case anthropic.ContentBlockDeltaEvent:
switch deltaVariant := eventVariant.Delta.AsAny().(type) {
case anthropic.TextDelta:
print(deltaVariant.Text)
}
}
}
return &message, stream.Err()
} }
type ToolDefinition struct { type ToolDefinition struct {