Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from langchain.vectorstores import FAISS
|
|
| 9 |
from langchain.chains import RetrievalQA
|
| 10 |
from langchain.chat_models import ChatOpenAI
|
| 11 |
|
| 12 |
-
enable_chat =
|
| 13 |
|
| 14 |
def toggle_enable_chat():
|
| 15 |
global enable_chat
|
|
@@ -36,7 +36,7 @@ def aliyun_chat(message, history):
|
|
| 36 |
print(f"[message]{message}")
|
| 37 |
print(f"[history]{history}")
|
| 38 |
# TODO: 从命令行参数中获取
|
| 39 |
-
enable_chat =
|
| 40 |
|
| 41 |
ans = ALIYUN_BOT({"query": message})
|
| 42 |
# 如果检索出结果,或者开了大模型聊天模式
|
|
@@ -44,7 +44,7 @@ def aliyun_chat(message, history):
|
|
| 44 |
if ans["source_documents"] or enable_chat:
|
| 45 |
print(f"[result]{ans['result']}")
|
| 46 |
print(f"[source_documents]{ans['source_documents']}")
|
| 47 |
-
return ans["result"]+"source_documents"+ans['source_documents']
|
| 48 |
# 否则输出套路话术
|
| 49 |
else:
|
| 50 |
return "基于现有文档数据,无法回答你的问题,请您联系对应的阿里云企业咨询顾问。"
|
|
|
|
| 9 |
from langchain.chains import RetrievalQA
|
| 10 |
from langchain.chat_models import ChatOpenAI
|
| 11 |
|
| 12 |
+
enable_chat = False # 初始化为False
|
| 13 |
|
| 14 |
def toggle_enable_chat():
|
| 15 |
global enable_chat
|
|
|
|
| 36 |
print(f"[message]{message}")
|
| 37 |
print(f"[history]{history}")
|
| 38 |
# TODO: 从命令行参数中获取
|
| 39 |
+
enable_chat = False
|
| 40 |
|
| 41 |
ans = ALIYUN_BOT({"query": message})
|
| 42 |
# 如果检索出结果,或者开了大模型聊天模式
|
|
|
|
| 44 |
if ans["source_documents"] or enable_chat:
|
| 45 |
print(f"[result]{ans['result']}")
|
| 46 |
print(f"[source_documents]{ans['source_documents']}")
|
| 47 |
+
return ans["result"]+"source_documents:"+ans['source_documents']
|
| 48 |
# 否则输出套路话术
|
| 49 |
else:
|
| 50 |
return "基于现有文档数据,无法回答你的问题,请您联系对应的阿里云企业咨询顾问。"
|