jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor RetrievalQAComponent to include source documents in the returned result

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 09:51:41 -03:00
commit 0387de2dd2
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/components/chains/RetrievalQA.py
View file

@ -3,6 +3,7 @@ from typing import Callable, Optional, Union
from langchain.chains.combine_documents.base import BaseCombineDocumentsChain from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
from langchain.chains.retrieval_qa.base import BaseRetrievalQA, RetrievalQA from langchain.chains.retrieval_qa.base import BaseRetrievalQA, RetrievalQA
from langchain_core.documents import Document from langchain_core.documents import Document
from langflow import CustomComponent from langflow import CustomComponent
from langflow.field_typing import BaseMemory, BaseRetriever, Text from langflow.field_typing import BaseMemory, BaseRetriever, Text
@ -47,4 +48,11 @@ class RetrievalQAComponent(CustomComponent):
result = result.content if hasattr(result, "content") else result result = result.content if hasattr(result, "content") else result
# Result is a dict with keys "query", "result" and "source_documents" # Result is a dict with keys "query", "result" and "source_documents"
# for now we just return the result # for now we just return the result
return result.get("result") records = self.to_records(result.get("source_documents"))
references_str = ""
if return_source_documents:
references_str = self.create_references_from_records(records)
result_str = result.get("result")
final_result = "\n".join([result_str, references_str])
self.status = final_result
return final_result