github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 15:21:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Format code

Browse Source
This commit is contained in:
FangYin Cheng 2023-11-06 19:55:57 +08:00
parent b380761571
commit d62048ef1e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pilot/model/proxy/llms/tongyi.py
View File

@ -36,7 +36,10 @@ def tongyi_generate_stream(
if message.role == ModelMessageRoleType.HUMAN:
history.append({"role": "user", "content": message.content})
for message in messages:
if message.role == ModelMessageRoleType.SYSTEM or message.role == ModelMessageRoleType.HUMAN:
if (
message.role == ModelMessageRoleType.SYSTEM
or message.role == ModelMessageRoleType.HUMAN
):
history.append({"role": "user", "content": message.content})
# elif message.role == ModelMessageRoleType.HUMAN:
# history.append({"role": "user", "content": message.content})