github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 23:00:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix incorrect value of outputKeys on AnalyzeDocumentsChain (#3010)

Browse Source
This commit is contained in:
Nuno Campos 2023-04-18 04:32:46 +01:00 committed by GitHub
parent 19febc77d6
commit e25528c4f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/chains/combine_documents/base.py
View File

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