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.""" """Chain that splits documents, then analyzes it in pieces."""
input_key: str = "input_document" #: :meta private: input_key: str = "input_document" #: :meta private:
output_key: str = "output_text" #: :meta private:
text_splitter: TextSplitter = Field(default_factory=RecursiveCharacterTextSplitter) text_splitter: TextSplitter = Field(default_factory=RecursiveCharacterTextSplitter)
combine_docs_chain: BaseCombineDocumentsChain combine_docs_chain: BaseCombineDocumentsChain
@ -107,7 +106,7 @@ class AnalyzeDocumentChain(Chain):
:meta private: :meta private:
""" """
return [self.output_key] return self.combine_docs_chain.output_keys
def _call(self, inputs: Dict[str, Any]) -> Dict[str, str]: def _call(self, inputs: Dict[str, Any]) -> Dict[str, str]:
document = inputs[self.input_key] document = inputs[self.input_key]

Loading…
Cancel
Save