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

feat: merge chat related sessions

Browse Source
This commit is contained in:
lanvent
2023-03-29 12:25:31 +08:00
parent 432b39a9c4
commit d62b7d1a99
3 changed files with 21 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bot/chatgpt/chat_gpt_session.py
View File

@@ -11,22 +11,9 @@ from common.log import logger
class ChatGPTSession(Session):
def __init__(self, session_id, system_prompt=None, model= "gpt-3.5-turbo"):
super().__init__(session_id, system_prompt)
self.messages = []
self.model = model
self.reset()
def reset(self):
system_item = {'role': 'system', 'content': self.system_prompt}
self.messages = [system_item]
def add_query(self, query):
user_item = {'role': 'user', 'content': query}
self.messages.append(user_item)
def add_reply(self, reply):
assistant_item = {'role': 'assistant', 'content': reply}
self.messages.append(assistant_item)
def discard_exceeding(self, max_tokens, cur_tokens= None):
precise = True
try: