File: //usr/local/lib/python3.10/dist-packages/langchain/chains/router/__pycache__/base.cpython-310.pyc
o
���g� � @ s� d Z ddlmZ ddlmZ ddlmZmZmZm Z m
Z
m Z ddl m
Z
mZmZ ddlmZ ddlmZ G dd � d e
�ZG d
d � d ee�ZG d d
� d
e�ZdS )z Base classes for chain routing.� )� annotations)�ABC)�Any�Dict�List�Mapping�
NamedTuple�Optional)� AsyncCallbackManagerForChainRun�CallbackManagerForChainRun� Callbacks)�
ConfigDict)�Chainc @ s e Zd ZU ded< ded< dS )�Routez
Optional[str]� destination�Dict[str, Any]� next_inputsN)�__name__�
__module__� __qualname__�__annotations__� r r �G/usr/local/lib/python3.10/dist-packages/langchain/chains/router/base.pyr s
r c @ s8 e Zd ZdZeddd��Zddd d
�Z dddd�ZdS )� RouterChainzHChain that outputs the name of a destination chain and the inputs to it.�return� List[str]c C s ddgS )Nr r r ��selfr r r � output_keys s zRouterChain.output_keysN�inputsr � callbacksr r c C s | ||d�}t |d |d �S )z�
Route inputs to a destination chain.
Args:
inputs: inputs to the chain
callbacks: callbacks to use for the chain
Returns:
a Route object
�r r r )r �r r r �resultr r r �route s zRouterChain.routec � s( �| j ||d�I d H }t|d |d �S )Nr! r r )�acallr r" r r r �aroute, s �zRouterChain.aroute�r r �N)r r r r r r )r r r �__doc__�propertyr r$ r& r r r r r s �r c @ s� e Zd ZU dZded< ded< ded< dZd ed
< ed d d
�Zed dd��Z ed dd��Z
d d dd�Z d d!d d �Z dS )"�MultiRouteChainzIUse a single chain to route an input to one of multiple candidate chains.r � router_chainzMapping[str, Chain]�destination_chainsr �
default_chainF�bool�
silent_errorsT�forbid)�arbitrary_types_allowed�extrar r c C s | j jS )zWWill be whatever keys the router chain prompt expects.
:meta private:
)r, �
input_keysr r r r r4 E s zMultiRouteChain.input_keysc C s g S )z=Will always return text key.
:meta private:
r r r r r r M s zMultiRouteChain.output_keysNr r � run_manager�$Optional[CallbackManagerForChainRun]c C s� |pt �� }|�� }| jj||d�}|jt|j�d t|j� | j d� |js/| j
|j|d�S |j| j v r@| j |j |j|d�S | j rK| j
|j|d�S t
d|j� d����Nr! z: )�verbosez)Received invalid destination chain name '�')r �get_noop_manager� get_childr, r$ �on_text�strr r r8 r. r- r0 �
ValueError�r r r5 � _run_managerr r$ r r r �_callU s" �
� �zMultiRouteChain._call�)Optional[AsyncCallbackManagerForChainRun]c � s� �|pt �� }|�� }| jj||d�I d H }|jt|j�d t|j� | j d�I d H |js:| j
j |j|d�I d H S |j| j v rO| j |j j |j|d�I d H S | j
r^| j
j |j|d�I d H S td|j� d���r7 )r
r: r; r, r& r< r= r r r8 r. r% r- r0 r> r? r r r �_acalln s, � � � � � �zMultiRouteChain._acallr' r( )r r r5 r6 r r )r r r5 rB r r )
r r r r) r r0 r
� model_configr* r4 r rA rC r r r r r+ 3 s,
�
� �r+ N)r) �
__future__r �abcr �typingr r r r r r �langchain_core.callbacksr
r r �pydanticr
�langchain.chains.baser r r r+ r r r r �<module> s