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

fix: avoid possible dead loop when discarding

Browse Source
This commit is contained in:
lanvent
2023-03-26 18:01:35 +08:00
parent a3be1fcd8f
commit 3d428ee844
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/chatgpt/chat_gpt_bot.py
View File

@@ -232,9 +232,13 @@ class SessionManager(object):
session.pop(1) session.pop(1)
elif len(session) == 2 and session[1]["role"] == "assistant": elif len(session) == 2 and session[1]["role"] == "assistant":
session.pop(1) session.pop(1)
break
elif len(session) == 2 and session[1]["role"] == "user": elif len(session) == 2 and session[1]["role"] == "user":
logger.warn("user message exceed max_tokens. total_tokens={}".format(dec_tokens)) logger.warn("user message exceed max_tokens. total_tokens={}".format(dec_tokens))
break break
else:
logger.debug("max_tokens={}, total_tokens={}, len(sessions)={}".format(max_tokens, dec_tokens, len(session)))
break
try: try:
cur_tokens = num_tokens_from_messages(session, self.model) cur_tokens = num_tokens_from_messages(session, self.model)
dec_tokens = cur_tokens dec_tokens = cur_tokens