Skip to content

Commit 306f49f

Browse files
committed
fix: add usage field to OpenAI streaming response final chunk (#10)
1 parent a308630 commit 306f49f

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

proxy/handler.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1204,6 +1204,11 @@ func (h *Handler) handleOpenAIStream(w http.ResponseWriter, account *config.Acco
12041204
"delta": map[string]interface{}{},
12051205
"finish_reason": finishReason,
12061206
}},
1207+
"usage": map[string]int{
1208+
"prompt_tokens": inputTokens,
1209+
"completion_tokens": outputTokens,
1210+
"total_tokens": inputTokens + outputTokens,
1211+
},
12071212
}
12081213
data, _ := json.Marshal(chunk)
12091214
fmt.Fprintf(w, "data: %s\n\n", string(data))

proxy/translator.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@ var modelMap = map[string]string{
1616
"claude-sonnet-4.5": "claude-sonnet-4.5",
1717
"claude-haiku-4-5": "claude-haiku-4.5",
1818
"claude-haiku-4.5": "claude-haiku-4.5",
19+
"claude-opus-4-6": "claude-opus-4.6",
20+
"claude-opus-4.6": "claude-opus-4.6",
1921
"claude-opus-4-5": "claude-opus-4.5",
2022
"claude-opus-4.5": "claude-opus-4.5",
2123
"claude-sonnet-4": "claude-sonnet-4",

0 commit comments

Comments
 (0)