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

model server fix message model

Browse Source
This commit is contained in:
yhjun1026 2023-06-01 18:08:27 +08:00
parent 084656245e
commit 98a7bf93f6
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -35,7 +35,7 @@ def chatglm_generate_stream(
# # Add history conversation # # Add history conversation
hist = [] hist = []
once_conversation = [] once_conversation = []
for message in messages: for message in messages[:-2]:
if len(message) <= 0: if len(message) <= 0:
continue continue
@ -51,7 +51,7 @@ def chatglm_generate_stream(
# else: # else:
# once_conversation.append(f"""###system:{message} """) # once_conversation.append(f"""###system:{message} """)
query = messages[-1].split("human:")[1] query = messages[-2].split("human:")[1]
print("Query Message: ", query) print("Query Message: ", query)
# output = "" # output = ""
# i = 0 # i = 0

BIN
plugins/DB-GPT-Plugin-ByteBase.zip
View File

Binary file not shown.