diff --git a/langchain/chains/combine_documents/reduce.py b/langchain/chains/combine_documents/reduce.py index 7ecde047cbb..a5b38695b10 100644 --- a/langchain/chains/combine_documents/reduce.py +++ b/langchain/chains/combine_documents/reduce.py @@ -193,7 +193,7 @@ class ReduceDocumentsChain(BaseCombineDocumentsChain): element returned is a dictionary of other keys to return. """ result_docs, extra_return_dict = self._collapse( - docs, token_max, callbacks=callbacks, **kwargs + docs, token_max=token_max, callbacks=callbacks, **kwargs ) return self.combine_documents_chain.combine_docs( docs=result_docs, callbacks=callbacks, **kwargs @@ -222,7 +222,7 @@ class ReduceDocumentsChain(BaseCombineDocumentsChain): element returned is a dictionary of other keys to return. """ result_docs, extra_return_dict = await self._acollapse( - docs, callbacks=callbacks, **kwargs + docs, token_max=token_max, callbacks=callbacks, **kwargs ) return await self.combine_documents_chain.acombine_docs( docs=result_docs, callbacks=callbacks, **kwargs