Upload app.py

app.py

@@ -63,8 +63,8 @@ class Model_center:
         self,
         question: str,
         chat_history: list = [],
-        model: str = "zhipuai",
-        embedding: str = "zhipuai",
+        model: str = "glm-4",
+        embedding: str = "embedding-2",
         temperature: float = 0.0,
         top_k: int = 4,
         history_len: int = 3,
@@ -98,8 +98,8 @@ class Model_center:
         self,
         question: str,
         chat_history: list = [],
-        model: str = "zhipuai",
-        embedding="zhipuai",
+        model: str = "glm-4",
+        embedding="embedding-2",
         temperature: float = 0.0,
         top_k: int = 4,
         file_path: str = DEFAULT_DB_PATH,
@@ -218,8 +218,8 @@ with block as demo:
 
             with gr.Row():
                 # 创建提交按钮。
-                # db_with_his_btn = gr.Button("Chat db with history")
-                # db_wo_his_btn = gr.Button("Chat db without history")
+                db_with_his_btn = gr.Button("Chat db with history")
+                db_wo_his_btn = gr.Button("Chat db without history")
                 llm_btn = gr.Button("Chat with llm")
             with gr.Row():
                 # 创建一个清除按钮，用于清除聊天机器人组件的内容。
@@ -275,13 +275,18 @@ with block as demo:
         # 设置初始化向量数据库按钮的点击事件。当点击时，调用 create_db_info 函数，并传入用户的文件和希望使用的 Embedding 模型。
         init_db.click(create_db_info, inputs=[file, embeddings], outputs=[msg])
 
-        # # 设置按钮的点击事件。当点击时，调用上面定义的 chat_qa_chain_self_answer 函数，并传入用户的消息和聊天历史记录，然后更新文本框和聊天机器人组件。
-        # db_with_his_btn.click(model_center.chat_qa_chain_self_answer, inputs=[
-        #                       msg, chatbot,  llm, embeddings, temperature, top_k, history_len],
-        #                       outputs=[msg, chatbot])
-        # # 设置按钮的点击事件。当点击时，调用上面定义的 qa_chain_self_answer 函数，并传入用户的消息和聊天历史记录，然后更新文本框和聊天机器人组件。
-        # db_wo_his_btn.click(model_center.qa_chain_self_answer, inputs=[
-        #                     msg, chatbot, llm, embeddings, temperature, top_k], outputs=[msg, chatbot])
+        # 设置按钮的点击事件。当点击时，调用上面定义的 chat_qa_chain_self_answer 函数，并传入用户的消息和聊天历史记录，然后更新文本框和聊天机器人组件。
+        db_with_his_btn.click(
+            model_center.chat_qa_chain_self_answer,
+            inputs=[msg, chatbot, llm, embeddings, temperature, top_k, history_len],
+            outputs=[msg, chatbot],
+        )
+        # 设置按钮的点击事件。当点击时，调用上面定义的 qa_chain_self_answer 函数，并传入用户的消息和聊天历史记录，然后更新文本框和聊天机器人组件。
+        db_wo_his_btn.click(
+            model_center.qa_chain_self_answer,
+            inputs=[msg, chatbot, llm, embeddings, temperature, top_k],
+            outputs=[msg, chatbot],
+        )
         # 设置按钮的点击事件。当点击时，调用上面定义的 respond 函数，并传入用户的消息和聊天历史记录，然后更新文本框和聊天机器人组件。
         llm_btn.click(
             respond,