diff --git a/pilot/model/llm_out/chatglm_llm.py b/pilot/model/llm_out/chatglm_llm.py index 59032f9e8..dcc4a88bc 100644 --- a/pilot/model/llm_out/chatglm_llm.py +++ b/pilot/model/llm_out/chatglm_llm.py @@ -35,7 +35,7 @@ def chatglm_generate_stream( # # Add history conversation hist = [] once_conversation = [] - for message in messages: + for message in messages[:-2]: if len(message) <= 0: continue @@ -51,7 +51,7 @@ def chatglm_generate_stream( # else: # once_conversation.append(f"""###system:{message} """) - query = messages[-1].split("human:")[1] + query = messages[-2].split("human:")[1] print("Query Message: ", query) # output = "" # i = 0 diff --git a/plugins/DB-GPT-Plugin-ByteBase.zip b/plugins/DB-GPT-Plugin-ByteBase.zip index 60ce2816d..c341b044b 100644 Binary files a/plugins/DB-GPT-Plugin-ByteBase.zip and b/plugins/DB-GPT-Plugin-ByteBase.zip differ