From e25528c4f05aeecead923d11a99af4346bdc0e8e Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 18 Apr 2023 04:32:46 +0100 Subject: [PATCH] Fix incorrect value of outputKeys on AnalyzeDocumentsChain (#3010) --- langchain/chains/combine_documents/base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/langchain/chains/combine_documents/base.py b/langchain/chains/combine_documents/base.py index ad6c0b66..731a5528 100644 --- a/langchain/chains/combine_documents/base.py +++ b/langchain/chains/combine_documents/base.py @@ -89,7 +89,6 @@ class AnalyzeDocumentChain(Chain): """Chain that splits documents, then analyzes it in pieces.""" input_key: str = "input_document" #: :meta private: - output_key: str = "output_text" #: :meta private: text_splitter: TextSplitter = Field(default_factory=RecursiveCharacterTextSplitter) combine_docs_chain: BaseCombineDocumentsChain @@ -107,7 +106,7 @@ class AnalyzeDocumentChain(Chain): :meta private: """ - return [self.output_key] + return self.combine_docs_chain.output_keys def _call(self, inputs: Dict[str, Any]) -> Dict[str, str]: document = inputs[self.input_key]