fix: always read the first respond choice

namwoam · Jul 11, 2024 · a56d392 · a56d392
1 parent 911d052
commit a56d392
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai/mistralai/v0/tasks.go b/ai/mistralai/v0/tasks.go
@@ -119,7 +119,7 @@ func (e *execution) taskTextGeneration(in *structpb.Struct) (*structpb.Struct, e
 
 	outputStruct := textGenerationOutput{}
 
-	outputStruct.Text = resp.Choices[len(resp.Choices)-1].Message.Content
+	outputStruct.Text = resp.Choices[0].Message.Content
 	outputStruct.Usage = chatUsage{
 		InputTokens:  resp.Usage.PromptTokens,
 		OutputTokens: resp.Usage.CompletionTokens,