captain-awesome commited on
Commit
eef9355
·
1 Parent(s): d944073

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from langchain.chains import ConversationalRetrievalChain
2
  from langchain.chains.question_answering import load_qa_chain
3
- from lanchain.chains import RetrievalQA
4
  from langchain.memory import ConversationBufferMemory
5
  from langchain.memory import ConversationTokenBufferMemory
6
  from langchain.llms import HuggingFacePipeline
@@ -322,7 +322,8 @@ def retrieve_bot_answer(query, loaded_documents):
322
  # return bot_response
323
  # else:
324
  # raise KeyError("Expected 'answer' key in bot_response, but it was not found.")
325
- result = bot_response['answer']
 
326
  sources = []
327
  for source in bot_response["source_documents"]:
328
  sources.append(source.metadata['source'])
 
1
  from langchain.chains import ConversationalRetrievalChain
2
  from langchain.chains.question_answering import load_qa_chain
3
+ from langchain.chains import RetrievalQA
4
  from langchain.memory import ConversationBufferMemory
5
  from langchain.memory import ConversationTokenBufferMemory
6
  from langchain.llms import HuggingFacePipeline
 
322
  # return bot_response
323
  # else:
324
  # raise KeyError("Expected 'answer' key in bot_response, but it was not found.")
325
+ # result = bot_response['answer']
326
+ result = bot_response['result']
327
  sources = []
328
  for source in bot_response["source_documents"]:
329
  sources.append(source.metadata['source'])