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

fix: chatglm stream output

Browse Source
This commit is contained in:
csunny 2023-06-08 21:29:02 +08:00
parent eae38931fa
commit bea3f7c8d2
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/model/llm_out/chatglm_llm.py
View File

@ -51,7 +51,10 @@ def chatglm_generate_stream(
# else: # else:
# once_conversation.append(f"""###system:{message} """) # once_conversation.append(f"""###system:{message} """)
try:
query = messages[-2].split("human:")[1] query = messages[-2].split("human:")[1]
except IndexError:
query = messages[-3].split("human:")[1]
print("Query Message: ", query) print("Query Message: ", query)
# output = "" # output = ""
# i = 0 # i = 0