From f7cdc727df2669068a53eb51d0415647b3af2ab5 Mon Sep 17 00:00:00 2001
From: zuiho <2324465096@qq.com>
Date: Tue, 7 Apr 2026 17:41:08 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20Claude=20=E6=B5=81=E5=BC=8F=E6=96=AD?=
 =?UTF-8?q?=E6=B5=81=E6=97=B6=E4=B8=8D=E5=86=8D=E6=95=B4=E4=BB=BD=E8=A6=86?=
 =?UTF-8?q?=E7=9B=96=20usage=EF=BC=8C=E4=BF=9D=E7=95=99=20cache=20?=
 =?UTF-8?q?=E8=AE=A1=E8=B4=B9=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

HandleStreamFinalResponse 在 !Done 时调用 ResponseText2Usage 整份覆盖
claudeInfo.Usage，导致 message_start 已获取的 CacheReadInputTokens、
CacheCreationInputTokens 等字段丢失，prompt 退化为占位值 1。

修复：
- 只补缺失的 CompletionTokens/PromptTokens，保留已有 cache 数据
- PromptTokens 兜底改用 info.GetEstimatePromptTokens()（与其他渠道对齐）

Fixes #4127
---
 relay/channel/claude/relay-claude.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/relay/channel/claude/relay-claude.go b/relay/channel/claude/relay-claude.go
index dceff5e7..21427b1f 100644
--- a/relay/channel/claude/relay-claude.go
+++ b/relay/channel/claude/relay-claude.go
@@ -809,7 +809,15 @@ func HandleStreamFinalResponse(c *gin.Context, info *relaycommon.RelayInfo, clau
 		if common.DebugEnabled {
 			common.SysLog("claude response usage is not complete, maybe upstream error")
 		}
-		claudeInfo.Usage = service.ResponseText2Usage(c, claudeInfo.ResponseText.String(), info.UpstreamModelName, claudeInfo.Usage.PromptTokens)
+		// 只补缺失字段，不整份覆盖——保留 message_start 已拿到的 cache 字段
+		fallback := service.ResponseText2Usage(c, claudeInfo.ResponseText.String(), info.UpstreamModelName, info.GetEstimatePromptTokens())
+		if claudeInfo.Usage.CompletionTokens == 0 {
+			claudeInfo.Usage.CompletionTokens = fallback.CompletionTokens
+		}
+		if claudeInfo.Usage.PromptTokens == 0 {
+			claudeInfo.Usage.PromptTokens = fallback.PromptTokens
+		}
+		claudeInfo.Usage.TotalTokens = claudeInfo.Usage.PromptTokens + claudeInfo.Usage.CompletionTokens
 	}
 	if claudeInfo.Usage != nil {
 		claudeInfo.Usage.UsageSemantic = "anthropic"