Update app.py

app.py

@@ -9,12 +9,12 @@ from langchain.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 from langchain.chat_models import ChatOpenAI
 
-# enable_chat = False  # 初始化为False
+enable_chat = True  # 初始化为False
 
-# def toggle_enable_chat():
-#     global enable_chat
-#     enable_chat = not enable_chat
-#     return f"Enable Chat set to {enable_chat}"
+def toggle_enable_chat():
+    global enable_chat
+    enable_chat = not enable_chat
+    return f"Enable Chat set to {enable_chat}"
 
 
 def initialize_aliyun_qa_bot(vector_store_dir: str="aliyun_qa"):
@@ -36,7 +36,7 @@ def aliyun_chat(message, history):
     print(f"[message]{message}")
     print(f"[history]{history}")
     # TODO: 从命令行参数中获取
-    enable_chat = False
+    enable_chat = True
 
     ans = ALIYUN_BOT({"query": message})
     # 如果检索出结果，或者开了大模型聊天模式
@@ -44,7 +44,7 @@ def aliyun_chat(message, history):
     if ans["source_documents"] or enable_chat:
         print(f"[result]{ans['result']}")
         print(f"[source_documents]{ans['source_documents']}")
-        return ans["result"]
+        return ans["result"]+"source_documents"+ans['source_documents']
     # 否则输出套路话术
     else:
         return "基于现有文档数据，无法回答你的问题，请您联系对应的阿里云企业咨询顾问。"