injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-25 11:29:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

修改对话拼接逻辑

Browse Source
This commit is contained in:
LINSTCL
2023-03-02 23:25:42 +08:00
parent dd3d403de8
commit 2234e9db0e
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/openai/modelmgr.py
View File

@@ -94,10 +94,10 @@ class CompletionModel(ModelRequest):
def msg_handle(self, msgs):
prompt = ''
for msg in msgs:
if msg['role'] == '':
if msg['role'] == 'assistant':
prompt = prompt + "{}\n".format(msg['content'])
else:
prompt = prompt + "{}:{}\n".format(msg['role'] if msg['role']!='system' else '你的回答要遵守此规则', msg['content'])
prompt = prompt + "{}:{}\n".format(msg['role'] , msg['content'])
return prompt
def get_text(self):

5
pkg/openai/session.py
View File

@@ -210,10 +210,7 @@ class Session:
del (res_ans_spt[0])
res_ans = '\n\n'.join(res_ans_spt)
if config.completion_api_params['model'] in pkg.openai.modelmgr.CHAT_COMPLETION_MODELS:
self.prompt.append({'role':'assistant', 'content':res_ans})
elif config.completion_api_params['model'] in pkg.openai.modelmgr.COMPLETION_MODELS:
self.prompt.append({'role':'', 'content':res_ans})
self.prompt.append({'role':'assistant', 'content':res_ans})
if self.just_switched_to_exist_session:
self.just_switched_to_exist_session = False