Spaces:
Runtime error
Runtime error
Commit
·
eef9355
1
Parent(s):
d944073
Update app.py
Browse files
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
|
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'])
|