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

plugin: add Role plugin

Browse Source
This commit is contained in:
lanvent
2023-03-20 20:43:02 +08:00
parent 5a46e09358
commit 77046000e8
5 changed files with 315 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bot/chatgpt/chat_gpt_bot.py
View File

@@ -145,6 +145,16 @@ class SessionManager(object):
sessions = dict()
self.sessions = sessions
def build_session(self, session_id, system_prompt=None):
session = self.sessions.get(session_id, [])
if len(session) == 0:
if system_prompt is None:
system_prompt = conf().get("character_desc", "")
system_item = {'role': 'system', 'content': system_prompt}
session.append(system_item)
self.sessions[session_id] = session
return session
def build_session_query(self, query, session_id):
'''
build query with conversation history
@@ -158,12 +168,7 @@ class SessionManager(object):
:param session_id: session id
:return: query content with conversaction
'''
session = self.sessions.get(session_id, [])
if len(session) == 0:
system_prompt = conf().get("character_desc", "")
system_item = {'role': 'system', 'content': system_prompt}
session.append(system_item)
self.sessions[session_id] = session
session = self.build_session(session_id)
user_item = {'role': 'user', 'content': query}
session.append(user_item)
return session