Fix incorrect value of outputKeys on AnalyzeDocumentsChain (#3010)

fix_agent_callbacks
Nuno Campos 1 year ago committed by GitHub
parent 19febc77d6
commit e25528c4f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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]

Loading…
Cancel
Save