From c66636a0c737d093a7e813bd4ad8b3f9bc0add60 Mon Sep 17 00:00:00 2001
From: zuiho <2324465096@qq.com>
Date: Tue, 7 Apr 2026 17:52:11 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=87=87=E7=BA=B3=20CodeRabbit=20?=
 =?UTF-8?q?=E5=BB=BA=E8=AE=AE=EF=BC=8C!Done=20=E6=97=B6=E4=B9=9F=E7=94=A8?=
 =?UTF-8?q?=20fallback=20=E8=A6=86=E7=9B=96=E5=8D=A0=E4=BD=8D=20Completion?=
 =?UTF-8?q?Tokens?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

message_start 阶段可能给 CompletionTokens 非零占位值，
只检查 == 0 不够，加上 !Done && fallback > current 条件。
---
 relay/channel/claude/relay-claude.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/relay/channel/claude/relay-claude.go b/relay/channel/claude/relay-claude.go
index 21427b1f..2982945c 100644
--- a/relay/channel/claude/relay-claude.go
+++ b/relay/channel/claude/relay-claude.go
@@ -811,7 +811,8 @@ func HandleStreamFinalResponse(c *gin.Context, info *relaycommon.RelayInfo, clau
 		}
 		// 只补缺失字段，不整份覆盖——保留 message_start 已拿到的 cache 字段
 		fallback := service.ResponseText2Usage(c, claudeInfo.ResponseText.String(), info.UpstreamModelName, info.GetEstimatePromptTokens())
-		if claudeInfo.Usage.CompletionTokens == 0 {
+		if claudeInfo.Usage.CompletionTokens == 0 ||
+			(!claudeInfo.Done && fallback.CompletionTokens > claudeInfo.Usage.CompletionTokens) {
 			claudeInfo.Usage.CompletionTokens = fallback.CompletionTokens
 		}
 		if claudeInfo.Usage.PromptTokens == 0 {