Merge pull request #2012 from 6vision/fix_issue_1959_

Fix issue 1959 wenxin模型返回报错
zhayujie · May 22, 2024 · a46320e · a46320e
2 parents eee96f2 + 071709d
commit a46320e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bot/baidu/baidu_wenxin.py b/bot/baidu/baidu_wenxin.py
@@ -94,7 +94,7 @@ def reply_text(self, session: BaiduWenxinSession, retry_count=0):
             logger.warn("[BAIDU] Exception: {}".format(e))
             need_retry = False
             self.sessions.clear_session(session.session_id)
-            result = {"completion_tokens": 0, "content": "出错了: {}".format(e)}
+            result = {"total_tokens": 0, "completion_tokens": 0, "content": "出错了: {}".format(e)}
             return result
 
     def get_access_token(self):