From 16b387eaaf5c9253ff053970e8cb1eef227d9024 Mon Sep 17 00:00:00 2001
From: glzjin <glzjin@zhaojin97.cn>
Date: Fri, 18 Aug 2023 23:13:51 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=99=BE=E5=BA=A6=E8=AE=A1?=
 =?UTF-8?q?=E8=B4=B9=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/relay-baidu.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/controller/relay-baidu.go b/controller/relay-baidu.go
index ad20d6d6..39f31a9a 100644
--- a/controller/relay-baidu.go
+++ b/controller/relay-baidu.go
@@ -215,9 +215,11 @@ func baiduStreamHandler(c *gin.Context, resp *http.Response) (*OpenAIErrorWithSt
 				common.SysError("error unmarshalling stream response: " + err.Error())
 				return true
 			}
-			usage.PromptTokens += baiduResponse.Usage.PromptTokens
-			usage.CompletionTokens += baiduResponse.Usage.CompletionTokens
-			usage.TotalTokens += baiduResponse.Usage.TotalTokens
+			if baiduResponse.Usage.TotalTokens != 0 {
+				usage.TotalTokens = baiduResponse.Usage.TotalTokens
+				usage.PromptTokens = baiduResponse.Usage.PromptTokens
+				usage.CompletionTokens = baiduResponse.Usage.TotalTokens - baiduResponse.Usage.PromptTokens
+			}
 			response := streamResponseBaidu2OpenAI(&baiduResponse)
 			jsonResponse, err := json.Marshal(response)
 			if err != nil {