_load_map_reduce_documents_chain() — langchain Function Reference

Source Code

libs/langchain/langchain_classic/chains/loading.py lines 148–182

def _load_map_reduce_documents_chain(
    config: dict,
    **kwargs: Any,
) -> MapReduceDocumentsChain:
    if "llm_chain" in config:
        llm_chain_config = config.pop("llm_chain")
        llm_chain = load_chain_from_config(llm_chain_config, **kwargs)
    elif "llm_chain_path" in config:
        llm_chain = load_chain(config.pop("llm_chain_path"), **kwargs)
    else:
        msg = "One of `llm_chain` or `llm_chain_path` must be present."
        raise ValueError(msg)

    if not isinstance(llm_chain, LLMChain):
        msg = f"Expected LLMChain, got {llm_chain}"
        raise ValueError(msg)  # noqa: TRY004

    if "reduce_documents_chain" in config:
        reduce_documents_chain = load_chain_from_config(
            config.pop("reduce_documents_chain"),
            **kwargs,
        )
    elif "reduce_documents_chain_path" in config:
        reduce_documents_chain = load_chain(
            config.pop("reduce_documents_chain_path"),
            **kwargs,
        )
    else:
        reduce_documents_chain = _load_reduce_documents_chain(config, **kwargs)

    return MapReduceDocumentsChain(
        llm_chain=llm_chain,
        reduce_documents_chain=reduce_documents_chain,
        **config,
    )