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

refactor: decouple openai session

Browse Source
This commit is contained in:
lanvent
2023-03-26 23:09:05 +08:00
parent 38c901a1c5
commit ea95ab9062
5 changed files with 122 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bot/session_manager.py
View File

@@ -50,7 +50,6 @@ class SessionManager(object):
def session_query(self, query, session_id):
session = self.build_session(session_id)
session.add_query(query)
print(session.messages)
try:
max_tokens = conf().get("conversation_max_tokens", 1000)
total_tokens = session.discard_exceeding(max_tokens, None)
@@ -67,7 +66,7 @@ class SessionManager(object):
tokens_cnt = session.discard_exceeding(max_tokens, total_tokens)
logger.debug("raw total_tokens={}, savesession tokens={}".format(total_tokens, tokens_cnt))
except Exception as e:
logger.debug("Exception when counting tokens precisely for prompt: {}".format(str(e)))
logger.debug("Exception when counting tokens precisely for session: {}".format(str(e)))
return session
def clear_session(self, session_id):