shen/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-02-23 08:23:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

logs: add baidu reply log

Browse Source
This commit is contained in:
zhayujie
2023-08-25 11:19:00 +08:00
parent cdf8609d24
commit 2dcd026e9f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
bot/baidu/baidu_wenxin.py
View File

@@ -77,6 +77,7 @@ class BaiduWenxinBot(Bot):
payload = {'messages': session.messages}
response = requests.request("POST", url, headers=headers, data=json.dumps(payload))
response_text = json.loads(response.text)
logger.info(f"[BAIDU] response text={response_text}")
res_content = response_text["result"]
total_tokens = response_text["usage"]["total_tokens"]
completion_tokens = response_text["usage"]["completion_tokens"]