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

fix: documents chat not working with chatglm2-6b (#275)

Browse Source 
Close #274
This commit is contained in:
magic.chen 2023-06-26 19:59:02 +08:00 committed by GitHub
commit 68fa925083
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -54,6 +54,11 @@ def chatglm_generate_stream(
try:
query = messages[-2].split("human:")[1]
except IndexError:
# fix doc qa: https://github.com/csunny/DB-GPT/issues/274
doc_qa_message = messages[-2]
if "system:" in doc_qa_message:
query = doc_qa_message.split("system:")[1]
else:
query = messages[-3].split("human:")[1]
print("Query Message: ", query)
# output = ""