github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-11 13:12:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

bugfix(ChatData): code merge bug fix (#810)

Browse Source 
1.code merge bug fix
This commit is contained in:
明天 2023-11-21 10:57:07 +08:00 committed by GitHub
commit dfe137090b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/model/model_adapter.py
View File

@ -166,10 +166,6 @@ class LLMModelAdaper:
can_use_systems = system_messages[:-1] can_use_systems = system_messages[:-1]
else: else:
can_use_systems = system_messages can_use_systems = system_messages
for i in range(len(user_messages)):
user_messages[-1] = system_messages[-1]
if len(system_messages) > 1:
can_use_system = system_messages[0]
for i in range(len(user_messages)): for i in range(len(user_messages)):
conv.append_message(conv.roles[0], user_messages[i]) conv.append_message(conv.roles[0], user_messages[i])
@ -181,8 +177,6 @@ class LLMModelAdaper:
else: else:
conv.update_system_message("".join(can_use_systems)) conv.update_system_message("".join(can_use_systems))
conv.update_system_message(can_use_system)
# Add a blank message for the assistant. # Add a blank message for the assistant.
conv.append_message(conv.roles[1], None) conv.append_message(conv.roles[1], None)
new_prompt = conv.get_prompt() new_prompt = conv.get_prompt()