History

fix(translator): include token usage in message_delta for Claude responses

Luis Pater committed 9 months agov6.7.35

Commit 550da0c1 parent 7ff39361 file · +1 −1Browse files

Unified · Split

internal/translator/openai/claude/openai_claude_response.go+1−1
@@ -347,7 +347,7 @@ func convertOpenAIDoneToAnthropic(param *ConvertOpenAIResponseToAnthropicParams) · 346 unmodified lines
347 347
348 // If we haven't sent message_delta yet (no usage info was received), send it now348 // If we haven't sent message_delta yet (no usage info was received), send it now
349 if param.FinishReason != "" && !param.MessageDeltaSent {349 if param.FinishReason != "" && !param.MessageDeltaSent {
350 messageDeltaJSON := `{"type":"message_delta","delta":{"stop_reason":"","stop_sequence":null}}`350 messageDeltaJSON := `{"type":"message_delta","delta":{"stop_reason":"","stop_sequence":null},"usage":{"input_tokens":0,"output_tokens":0}}`
351 messageDeltaJSON, _ = sjson.Set(messageDeltaJSON, "delta.stop_reason", mapOpenAIFinishReasonToAnthropic(param.FinishReason))351 messageDeltaJSON, _ = sjson.Set(messageDeltaJSON, "delta.stop_reason", mapOpenAIFinishReasonToAnthropic(param.FinishReason))
352 results = append(results, "event: message_delta\ndata: "+messageDeltaJSON+"\n\n")352 results = append(results, "event: message_delta\ndata: "+messageDeltaJSON+"\n\n")
353 param.MessageDeltaSent = true353 param.MessageDeltaSent = true