datasense commited on
Commit
0005038
·
1 Parent(s): 86b7891

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = True # 初始化为False
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 = True
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 "基于现有文档数据,无法回答你的问题,请您联系对应的阿里云企业咨询顾问。"