From 5d21deae042ac0879910c2d763712bf7c0bbbbe7 Mon Sep 17 00:00:00 2001
From: yokowu <18836617@qq.com>
Date: Tue, 1 Jul 2025 21:26:58 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E8=BE=93=E5=87=BA=E4=B8=BA=E7=A9=BA?=
 =?UTF-8?q?=E7=9A=84=E4=BB=A3=E7=A0=81=E8=A1=A5=E5=85=A8=E4=B8=8D=E5=86=8D?=
 =?UTF-8?q?=E8=AE=B0=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/internal/proxy/proxy.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/backend/internal/proxy/proxy.go b/backend/internal/proxy/proxy.go
index 74ddfa4..a8b32c6 100644
--- a/backend/internal/proxy/proxy.go
+++ b/backend/internal/proxy/proxy.go
@@ -376,11 +376,15 @@ func (p *LLMProxy) handleCompletionStream(ctx context.Context, w http.ResponseWr
 					if len(t.Choices) > 0 {
 						rc.Completion += t.Choices[0].Text
 					}
-					rc.InputTokens += int64(t.Usage.PromptTokens)
+					rc.InputTokens = int64(t.Usage.PromptTokens)
 					rc.OutputTokens += int64(t.Usage.CompletionTokens)
 				}
 			}
 
+			if rc.OutputTokens == 0 {
+				return
+			}
+
 			p.logger.With("record", rc).DebugContext(ctx, "流式记录")
 			if err := p.usecase.Record(context.Background(), rc); err != nil {
 				p.logger.With("modelID", m.ID, "modelName", m.ModelName, "modelType", consts.ModelTypeLLM).
@@ -469,6 +473,10 @@ func (p *LLMProxy) handleCompletion(ctx context.Context, w http.ResponseWriter,
 }
 
 func (p *LLMProxy) recordCompletion(c *Ctx, modelID string, req domain.CompletionRequest, resp *openai.CompletionResponse) {
+	if resp.Usage.CompletionTokens == 0 {
+		return
+	}
+
 	ctx := context.Background()
 	prompt := req.Prompt.(string)
 	rc := &domain.RecordParam{