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

2026-02-10 09:42:31 +08:00
parent a308630156
commit 306f49f9ac
2 changed files with 7 additions and 0 deletions
--- a/proxy/handler.go
+++ b/proxy/handler.go
@@ -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))