github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-03 10:12:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

unify return types across map-reduce and refine (#442)

Browse Source
This commit is contained in:
Harrison Chase 2022-12-27 19:53:45 -05:00 committed by GitHub
parent 0c5d3fd894
commit ffe35c396c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -147,7 +147,8 @@ class MapReduceDocumentsChain(BaseCombineDocumentsChain, BaseModel):
result_docs, **kwargs result_docs, **kwargs
) )
if self.return_map_steps: if self.return_map_steps:
extra_return_dict = {"map_steps": results} _results = [r[self.llm_chain.output_key] for r in results]
extra_return_dict = {"map_steps": _results}
else: else:
extra_return_dict = {} extra_return_dict = {}
output, _ = self.combine_document_chain.combine_docs(result_docs, **kwargs) output, _ = self.combine_document_chain.combine_docs(result_docs, **kwargs)