fix: 修复一些极端情况造成的Index溢出的问题

This commit is contained in:
Soulter
2022-12-14 17:38:16 +08:00
parent 2a8a49fdf6
commit f9dde30a24
+2 -1
View File
@@ -247,7 +247,6 @@ async def oper_msg(message, at=False):
chatgpt_res, current_usage_tokens = await get_chatGPT_response(cache_prompt)
except (PromptExceededError) as e:
print("出现token超限, 清空对应缓存")
# 超过4097tokens错误,清空缓存
session_dict[session_id] = []
cache_data_list = []
@@ -265,6 +264,8 @@ async def oper_msg(message, at=False):
cache_list = session_dict[session_id]
index = 0
while t > max_tokens:
if index >= len(cache_list):
break
t -= int(cache_list[index]['single_tokens'])
index += 1
session_dict[session_id] = cache_list[index:]