Update app.py

增加 chatGPT 的能力

app.py

@@ -33,7 +33,8 @@ embedding_model_dict = {
 llm_model_dict = {
     "ChatGLM-6B-int8": "THUDM/chatglm-6b-int8",
     "ChatGLM-6B-int4": "THUDM/chatglm-6b-int4",
-    "ChatGLM-6b-int4-qe": "THUDM/chatglm-6b-int4-qe"
+    "ChatGLM-6b-int4-qe": "THUDM/chatglm-6b-int4-qe",
+    "ChatGPT": "ChatGPT"
 }
 
 DEVICE = "cuda" if torch.cuda.is_available(
@@ -109,9 +110,9 @@ def get_knowledge_based_answer(query,
     prompt = PromptTemplate(template=prompt_template,
                             input_variables=["context", "question"])
     chatLLM = ChatLLM()
-    chatLLM.history = chat_history[-history_len:] if history_len > 0 else []
+    chatLLM.history = chat_history[-history_len:] if history_len > 0 else chat_history
     if large_language_model == "ChatGPT":
-        chatLLM.model = OpenAI()
+        chatLLM.model = "ChatGPT"
     else:
         chatLLM.load_model(
             model_name_or_path=llm_model_dict[large_language_model])