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

Remove combine_documents_chain from RetrievalQAWithSourcesChainComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 15:04:20 -03:00
commit 1f9a559483
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
View file

@ -1,7 +1,6 @@
from typing import Optional from typing import Optional
from langchain.chains import RetrievalQAWithSourcesChain from langchain.chains import RetrievalQAWithSourcesChain
from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
from langchain_core.documents import Document from langchain_core.documents import Document
from langflow import CustomComponent from langflow import CustomComponent
@ -18,6 +17,7 @@ class RetrievalQAWithSourcesChainComponent(CustomComponent):
"chain_type": { "chain_type": {
"display_name": "Chain Type", "display_name": "Chain Type",
"options": ["stuff", "map_reduce", "map_rerank", "refine"], "options": ["stuff", "map_reduce", "map_rerank", "refine"],
"info": "The type of chain to use to combined Documents.",
}, },
"memory": {"display_name": "Memory"}, "memory": {"display_name": "Memory"},
"return_source_documents": {"display_name": "Return Source Documents"}, "return_source_documents": {"display_name": "Return Source Documents"},
@ -27,7 +27,6 @@ class RetrievalQAWithSourcesChainComponent(CustomComponent):
self, self,
retriever: BaseRetriever, retriever: BaseRetriever,
llm: BaseLanguageModel, llm: BaseLanguageModel,
combine_documents_chain: BaseCombineDocumentsChain,
chain_type: str, chain_type: str,
memory: Optional[BaseMemory] = None, memory: Optional[BaseMemory] = None,
return_source_documents: Optional[bool] = True, return_source_documents: Optional[bool] = True,
@ -35,7 +34,6 @@ class RetrievalQAWithSourcesChainComponent(CustomComponent):
runnable = RetrievalQAWithSourcesChain.from_chain_type( runnable = RetrievalQAWithSourcesChain.from_chain_type(
llm=llm, llm=llm,
chain_type=chain_type, chain_type=chain_type,
combine_documents_chain=combine_documents_chain,
memory=memory, memory=memory,
return_source_documents=return_source_documents, return_source_documents=return_source_documents,
retriever=retriever, retriever=retriever,