#183: simplifed code by introducing a new var for the content

EinStack · Mar 19, 2024 · fb96526 · fb96526
1 parent f03b3d9
commit fb96526
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pkg/providers/anthropic/chat.go b/pkg/providers/anthropic/chat.go
@@ -139,6 +139,7 @@ func (c *Client) doChatRequest(ctx context.Context, payload *ChatRequest) (*sche
 		return nil, err
 	}
 
+	completion := anthropicResponse.Content[0]
 	usage := anthropicResponse.Usage
 
 	// Map response to ChatResponse schema
@@ -151,11 +152,11 @@ func (c *Client) doChatRequest(ctx context.Context, payload *ChatRequest) (*sche
 		ModelResponse: schemas.ModelResponse{
 			SystemID: map[string]string{},
 			Message: schemas.ChatMessage{
-				Role:    anthropicResponse.Content[0].Type,
-				Content: anthropicResponse.Content[0].Text,
+				Role:    completion.Type,
+				Content: completion.Text,
 			},
 			TokenUsage: schemas.TokenUsage{
-				PromptTokens:   usage.InputTokens, // Anthropic doesn't send prompt tokens
+				PromptTokens:   usage.InputTokens,
 				ResponseTokens: usage.OutputTokens,
 				TotalTokens:    usage.InputTokens + usage.OutputTokens,
 			},