kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-22 06:37:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: knowledge base miss suffix bug

Browse Source
This commit is contained in:
zhayujie
2023-10-13 19:12:23 +08:00
parent c2477b26c0
commit 792e940279
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/linkai/link_ai_bot.py
View File

@@ -94,11 +94,11 @@ class LinkAIBot(Bot, OpenAIImage):
response = res.json() response = res.json()
reply_content = response["choices"][0]["message"]["content"] reply_content = response["choices"][0]["message"]["content"]
total_tokens = response["usage"]["total_tokens"] 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)
suffix = self._fecth_knowledge_search_suffix(response) suffix = self._fecth_knowledge_search_suffix(response)
if suffix: if suffix:
reply_content += suffix reply_content += 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) return Reply(ReplyType.TEXT, reply_content)
else: else: