fix: fix reasoning metadata

fix: fix reasoning content

Author: kujtimiihoxha

ai/agent.go

@@ -1068,6 +1068,11 @@ func (a *agent) processStepStream(ctx context.Context, stream StreamResponse, op
 
 	activeToolCalls := make(map[string]*ToolCallContent)
 	activeTextContent := make(map[string]string)
+	type reasoningContent struct {
+		content string
+		options ProviderMetadata
+	}
+	activeReasoningContent := make(map[string]reasoningContent)
 
 	// Process stream parts
 	for part := range stream {
@@ -1125,7 +1130,7 @@ func (a *agent) processStepStream(ctx context.Context, stream StreamResponse, op
 			}
 
 		case StreamPartTypeReasoningStart:
-			activeTextContent[part.ID] = ""
+			activeReasoningContent[part.ID] = reasoningContent{content: ""}
 			if opts.OnReasoningStart != nil {
 				err := opts.OnReasoningStart(part.ID)
 				if err != nil {
@@ -1134,8 +1139,10 @@ func (a *agent) processStepStream(ctx context.Context, stream StreamResponse, op
 			}
 
 		case StreamPartTypeReasoningDelta:
-			if _, exists := activeTextContent[part.ID]; exists {
-				activeTextContent[part.ID] += part.Delta
+			if active, exists := activeReasoningContent[part.ID]; exists {
+				active.content += part.Delta
+				active.options = part.ProviderMetadata
+				activeReasoningContent[part.ID] = active
 			}
 			if opts.OnReasoningDelta != nil {
 				err := opts.OnReasoningDelta(part.ID, part.Delta)
@@ -1145,21 +1152,19 @@ func (a *agent) processStepStream(ctx context.Context, stream StreamResponse, op
 			}
 
 		case StreamPartTypeReasoningEnd:
-			if text, exists := activeTextContent[part.ID]; exists {
-				stepContent = append(stepContent, ReasoningContent{
-					Text:             text,
-					ProviderMetadata: part.ProviderMetadata,
-				})
+			if active, exists := activeReasoningContent[part.ID]; exists {
+				content := ReasoningContent{
+					Text:             active.content,
+					ProviderMetadata: active.options,
+				}
+				stepContent = append(stepContent, content)
 				if opts.OnReasoningEnd != nil {
-					err := opts.OnReasoningEnd(part.ID, ReasoningContent{
-						Text:             text,
-						ProviderMetadata: part.ProviderMetadata,
-					})
+					err := opts.OnReasoningEnd(part.ID, content)
 					if err != nil {
 						return StepResult{}, false, err
 					}
 				}
-				delete(activeTextContent, part.ID)
+				delete(activeReasoningContent, part.ID)
 			}
 
 		case StreamPartTypeToolInputStart:

anthropic/provider_options.go

@@ -3,15 +3,15 @@ package anthropic
 type ProviderOptions struct {
 	SendReasoning          *bool                   `mapstructure:"send_reasoning,omitempty"`
 	Thinking               *ThinkingProviderOption `mapstructure:"thinking,omitempty"`
-	DisableParallelToolUse *bool                   `mapstructure"disable_parallel_tool_use,omitempty"`
+	DisableParallelToolUse *bool                   `mapstructure:"disable_parallel_tool_use,omitempty"`
 }
 
 type ThinkingProviderOption struct {
 	BudgetTokens int64 `mapstructure:"budget_tokens"`
 }
 
 type ReasoningMetadata struct {
-	Signature    string `mapstructure"signature"`
+	Signature    string `mapstructure:"signature"`
 	RedactedData string `mapstructure:"redacted_data"`
 }
 

cspell.json

@@ -0,0 +1,9 @@
+{
+  "language": "en",
+  "version": "0.2",
+  "flagWords": [],
+  "words": [
+    "mapstructure",
+    "mapstructure"
+  ]
+}