From 3cbc6aa77cedaffda26bef4aa3c41158e2990bca Mon Sep 17 00:00:00 2001
From: littlepanda0716 <littlepanda0716@gmail>
Date: Fri, 7 Apr 2023 10:11:26 +0800
Subject: [PATCH] update knowledge_based_chatglm.py

---
 knowledge_based_chatglm.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/knowledge_based_chatglm.py b/knowledge_based_chatglm.py
index 84fe32e..8f36403 100644
--- a/knowledge_based_chatglm.py
+++ b/knowledge_based_chatglm.py
@@ -10,6 +10,8 @@ from langchain.vectorstores import FAISS
 from langchain.document_loaders import UnstructuredFileLoader
 from chatglm_llm import ChatGLM
 
+chatglm = ChatGLM()
+
 
 def init_knowledge_vector_store(filepath):
     embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese", )
@@ -42,7 +44,6 @@ def get_knowledge_based_answer(query, vector_store, chat_history=[]):
     ----------------
     改写后的独立、完整的问题："""
     new_question_prompt = PromptTemplate.from_template(condese_propmt_template)
-    chatglm = ChatGLM()
     chatglm.history = chat_history
     knowledge_chain = ChatVectorDBChain.from_llm(
         llm=chatglm,