feat: merge chat related sessions

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

View File

@@ -5,6 +5,7 @@ from config import conf
class Session(object):
def __init__(self, session_id, system_prompt=None):
self.session_id = session_id
self.messages = []
if system_prompt is None:
self.system_prompt = conf().get("character_desc", "")
else:
@@ -12,17 +13,20 @@ class Session(object):
# 重置会话
def reset(self):
raise NotImplementedError
system_item = {'role': 'system', 'content': self.system_prompt}
self.messages = [system_item]
def set_system_prompt(self, system_prompt):
self.system_prompt = system_prompt
self.reset()
def add_query(self, query):
raise NotImplementedError
user_item = {'role': 'user', 'content': query}
self.messages.append(user_item)
def add_reply(self, reply):
raise NotImplementedError
assistant_item = {'role': 'assistant', 'content': reply}
self.messages.append(assistant_item)
def discard_exceeding(self, max_tokens=None, cur_tokens=None):
raise NotImplementedError