From aa14e223eeafd93a5d4e893a44254550f4f9a483 Mon Sep 17 00:00:00 2001 From: Zach Schillaci <40636930+zachschillaci27@users.noreply.github.com> Date: Wed, 24 May 2023 17:28:37 +0200 Subject: [PATCH] Reuse `length_func` in `MapReduceDocumentsChain` (#5181) # Reuse `length_func` in `MapReduceDocumentsChain` Pretty straightforward refactor in `MapReduceDocumentsChain`. Reusing the local variable `length_func`, instead of the longer alternative `self.combine_document_chain.prompt_length`. @hwchase17 --- langchain/chains/combine_documents/map_reduce.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/langchain/chains/combine_documents/map_reduce.py b/langchain/chains/combine_documents/map_reduce.py index 8b2925de..06e87e03 100644 --- a/langchain/chains/combine_documents/map_reduce.py +++ b/langchain/chains/combine_documents/map_reduce.py @@ -195,9 +195,7 @@ class MapReduceDocumentsChain(BaseCombineDocumentsChain): for docs in new_result_doc_list: new_doc = _collapse_docs(docs, _collapse_docs_func, **kwargs) result_docs.append(new_doc) - num_tokens = self.combine_document_chain.prompt_length( - result_docs, **kwargs - ) + num_tokens = length_func(result_docs, **kwargs) if self.return_intermediate_steps: _results = [r[self.llm_chain.output_key] for r in results] extra_return_dict = {"intermediate_steps": _results}