xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-12 08:46:02 +00: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):