fix: add usage field to OpenAI streaming response final chunk (#10)

This commit is contained in:
Quorinex
2026-02-10 09:42:31 +08:00
parent a308630156
commit 306f49f9ac
2 changed files with 7 additions and 0 deletions

View File

@@ -1204,6 +1204,11 @@ func (h *Handler) handleOpenAIStream(w http.ResponseWriter, account *config.Acco
"delta": map[string]interface{}{},
"finish_reason": finishReason,
}},
"usage": map[string]int{
"prompt_tokens": inputTokens,
"completion_tokens": outputTokens,
"total_tokens": inputTokens + outputTokens,
},
}
data, _ := json.Marshal(chunk)
fmt.Fprintf(w, "data: %s\n\n", string(data))

View File

@@ -16,6 +16,8 @@ var modelMap = map[string]string{
"claude-sonnet-4.5": "claude-sonnet-4.5",
"claude-haiku-4-5": "claude-haiku-4.5",
"claude-haiku-4.5": "claude-haiku-4.5",
"claude-opus-4-6": "claude-opus-4.6",
"claude-opus-4.6": "claude-opus-4.6",
"claude-opus-4-5": "claude-opus-4.5",
"claude-opus-4.5": "claude-opus-4.5",
"claude-sonnet-4": "claude-sonnet-4",