File: //usr/local/lib/python3.10/dist-packages/langchain/chains/__pycache__/mapreduce.cpython-310.pyc
o
���g � @ s� d Z ddlmZ ddlmZmZmZmZmZ ddl m
Z
ddl m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
l m Z ddl m Z ddl m!Z! ddl"m#Z# e
dddd�G dd� de��Z$dS )z�Map-reduce chain.
Splits up a document, sends the smaller parts to the LLM with one prompt,
then combines the results with another one.
� )� annotations)�Any�Dict�List�Mapping�Optional)�
deprecated)�CallbackManagerForChainRun� Callbacks��Document)�BaseLanguageModel)�BasePromptTemplate)� TextSplitter)�
ConfigDict)�ReduceDocumentsChain)�Chain)�BaseCombineDocumentsChain)�MapReduceDocumentsChain)�StuffDocumentsChain)�LLMChainz0.2.13z1.0a Refer to migration guide here for a recommended implementation using LangGraph: https://python.langchain.com/docs/versions/migrating_chains/map_reduce_chain/. See also LangGraph guides for map-reduce: https://langchain-ai.github.io/langgraph/how-tos/map-reduce/.)�since�removal�messagec @ s� e Zd ZU dZded< ded< dZded< d Zded
< e d(d)dd��Ze ddd �Z
e d*d d ��Z e d*d d!��Z
d+d,d&d'�Zd S )-�MapReduceChainzMap-reduce chain.r �combine_documents_chainr �
text_splitter�
input_text�str� input_key� output_text�
output_keyN�llmr
�promptr � callbacksr
�combine_chain_kwargs�Optional[Mapping[str, Any]]�reduce_chain_kwargs�kwargsr �returnc K sh t |||d�}td||d�|r|ni ��} t| d�}
td||
|d�|r%|ni ��} | d| ||d�|��S )zDConstruct a map-reduce chain that uses the chain for map and reduce.)r" r# r$ )� llm_chainr$ )r )r* �reduce_documents_chainr$ )r r r$ N� )r r r r ) �clsr" r# r r$ r% r' r( r* � stuff_chainr+ r r, r, �E/usr/local/lib/python3.10/dist-packages/langchain/chains/mapreduce.py� from_params/ s0 �
���
���zMapReduceChain.from_paramsT�forbid)�arbitrary_types_allowed�extra� List[str]c C � | j gS )z2Expect input key.
:meta private:
)r ��selfr, r, r/ �
input_keysV � zMapReduceChain.input_keysc C r5 )z3Return output key.
:meta private:
)r! r6 r, r, r/ � output_keys^ r9 zMapReduceChain.output_keys�inputs�Dict[str, str]� run_manager�$Optional[CallbackManagerForChainRun]c C sd |pt �� }|�| j�}| j�|�}dd� |D �}i |�| jj|i�}| jj||�� d�}| j |iS )Nc S s g | ]}t |d ��qS ))� page_contentr )�.0�textr, r, r/ �
<listcomp>o s z(MapReduceChain._call.<locals>.<listcomp>)r$ )
r �get_noop_manager�popr r �
split_textr �run� get_childr! ) r7 r; r= � _run_manager�doc_text�texts�docs�_inputs�outputsr, r, r/ �_callf s ���
zMapReduceChain._call)NNN)r" r
r# r r r r$ r
r% r& r' r& r( r r) r )r) r4 )N)r; r< r= r> r) r< )�__name__�
__module__� __qualname__�__doc__�__annotations__r r! � classmethodr0 r � model_config�propertyr8 r: rN r, r, r, r/ r s.
�!�
�r N)%rR �
__future__r �typingr r r r r �langchain_core._apir �langchain_core.callbacksr r
�langchain_core.documentsr � langchain_core.language_modelsr
�langchain_core.promptsr �langchain_text_splittersr �pydanticr �langchain.chainsr �langchain.chains.baser �'langchain.chains.combine_documents.baser �-langchain.chains.combine_documents.map_reducer �(langchain.chains.combine_documents.stuffr �langchain.chains.llmr r r, r, r, r/ �<module> s, �