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

fix: reduce session max words

Browse Source
This commit is contained in:
zhayujie
2023-02-05 21:45:18 +08:00
parent 7425d9075a
commit 152724f05d
4 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
bot/openai/open_ai_bot.py
View File

@@ -117,22 +117,24 @@ class Session(object):
:param user_id: from user id
:return: query content with conversaction
'''
new_query = ""
prompt = conf().get("character_desc", "")
if prompt:
prompt += "\n\n"
session = user_session.get(user_id, None)
if session:
for conversation in session:
new_query += "Q: " + conversation["question"] + "\n\n\nA: " + conversation["answer"] + "<|im_end|>\n"
new_query += "Q: " + query + "\nA: "
return new_query
prompt += "Q: " + conversation["question"] + "\n\n\nA: " + conversation["answer"] + "<|im_end|>\n"
prompt += "Q: " + query + "\nA: "
return prompt
else:
return "Q: " + query + "\nA: "
return prompt + "Q: " + query + "\nA: "
@staticmethod
def save_session(query, answer, user_id):
max_tokens = conf().get("conversation_max_tokens")
if not max_tokens:
# default 3000
max_tokens = 3000
max_tokens = 1000
conversation = dict()
conversation["question"] = query
conversation["answer"] = answer