mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-04-18 01:53:47 +08:00
fix: reduce session max words
This commit is contained in:
@@ -88,7 +88,7 @@ cp config-template.json config.json
|
||||
"group_chat_prefix": ["@bot"], # 群聊时包含该前缀则会触发机器人回复
|
||||
"group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"], # 开启自动回复的群名称列表
|
||||
"image_create_prefix": ["画", "看", "找"], # 开启图片回复的前缀
|
||||
"conversation_max_tokens": 3000 # 支持上下文记忆的最多字符数
|
||||
"conversation_max_tokens": 1000 # 支持上下文记忆的最多字符数
|
||||
}
|
||||
```
|
||||
**配置说明:**
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -5,5 +5,5 @@
|
||||
"group_chat_prefix": ["@bot"],
|
||||
"group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"],
|
||||
"image_create_prefix": ["画", "看", "找"],
|
||||
"conversation_max_tokens": 3000
|
||||
"conversation_max_tokens": 1000
|
||||
}
|
||||
|
||||
2
requirement.txt
Normal file
2
requirement.txt
Normal file
@@ -0,0 +1,2 @@
|
||||
itchat-uos==1.5.0.dev0
|
||||
openai
|
||||
Reference in New Issue
Block a user