kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 13:47:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' into feat-agent

Browse Source
This commit is contained in:
zhayujie
2023-10-27 12:32:35 +08:00
committed by GitHub
parent 34e06fcbf8 5e00704152
commit 6331350239
10 changed files with 64 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bot/linkai/link_ai_bot.py
View File

@@ -94,6 +94,9 @@ class LinkAIBot(Bot, OpenAIImage):
response = res.json()
reply_content = response["choices"][0]["message"]["content"]
total_tokens = response["usage"]["total_tokens"]
logger.info(f"[LINKAI] reply={reply_content}, total_tokens={total_tokens}")
self.sessions.session_reply(reply_content, session_id, total_tokens)
agent_suffix = self._fetch_agent_suffix(response)
if agent_suffix:
reply_content += agent_suffix
@@ -101,8 +104,6 @@ class LinkAIBot(Bot, OpenAIImage):
knowledge_suffix = self._fetch_knowledge_search_suffix(response)
if knowledge_suffix:
reply_content += knowledge_suffix
logger.info(f"[LINKAI] reply={reply_content}, total_tokens={total_tokens}")
self.sessions.session_reply(reply_content, session_id, total_tokens)
return Reply(ReplyType.TEXT, reply_content)
else: