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

增加会话超长问题

Browse Source
This commit is contained in:
zwssunny
2023-03-05 09:43:59 +08:00
parent efbc9de9d1
commit 3d4d1c734a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bot/chatgpt/chat_gpt_bot.py
View File

@@ -6,6 +6,7 @@ from common.log import logger
from common.expired_dict import ExpiredDict from common.expired_dict import ExpiredDict
import openai import openai
import time import time
import json
if conf().get('expires_in_seconds'): if conf().get('expires_in_seconds'):
user_session = ExpiredDict(conf().get('expires_in_seconds')) user_session = ExpiredDict(conf().get('expires_in_seconds'))
@@ -156,7 +157,7 @@ class Session(object):
for i in range(len(session)-1, -1, -1): for i in range(len(session)-1, -1, -1):
# count tokens of conversation list # count tokens of conversation list
history_conv = session[i] history_conv = session[i]
tokens=history_conv.split() tokens=json.dumps(history_conv).split()
count += len(tokens) count += len(tokens)
count_list.append(count) count_list.append(count)