File: //usr/local/lib/python3.10/dist-packages/langchain/chains/__pycache__/transform.cpython-310.pyc
o
���g2 � @ sz d Z ddlZddlZddlmZmZmZmZmZm Z ddl
m Z m Z ddl
mZ ddlmZ e�e�ZG dd� de�ZdS ) z-Chain that runs an arbitrary python function.� N)�Any� Awaitable�Callable�Dict�List�Optional)� AsyncCallbackManagerForChainRun�CallbackManagerForChainRun)�Field)�Chainc @ s8 e Zd ZU dZee ed< ee ed< edd�Ze e
eef ge
eef f ed< eddd�Z e e e
ee
f gee
ee
f f ed < eejd
ed dfd d
���Zed ee fdd��Zed ee fdd��Z dde
eef de e d e
eef fdd�Z dde
ee
f de e d e
ee
f fdd�ZdS )�TransformChaina Chain that transforms the chain output.
Example:
.. code-block:: python
from langchain.chains import TransformChain
transform_chain = TransformChain(input_variables=["text"],
output_variables["entities"], transform=func())
�input_variables�output_variables� transform)�alias� transform_cbN�
atransform�
atransform_cb�msg�returnc C s t �| � dS )z4Log a message once.
:meta private:
N)�logger�warning)r � r �E/usr/local/lib/python3.10/dist-packages/langchain/chains/transform.py� _log_once( s zTransformChain._log_oncec C � | j S )z3Expect input keys.
:meta private:
)r
��selfr r r �
input_keys1 � zTransformChain.input_keysc C r )z4Return output keys.
:meta private:
)r r r r r � output_keys9 r zTransformChain.output_keys�inputs� run_managerc C s
| � |�S �N)r �r r! r" r r r �_callA s
zTransformChain._callc � s0 �| j d ur| � |�I d H S | �d� | �|�S )NzRTransformChain's atransform is not provided, falling back to synchronous transform)r r r r$ r r r �_acallH s �
�
zTransformChain._acallr# )�__name__�
__module__� __qualname__�__doc__r �str�__annotations__r
r r r r r r r � staticmethod� functools� lru_cacher �propertyr r r r% r r& r r r r r sD
,
,�
�
��
�
�
��
�r )r* r. �logging�typingr r r r r r �langchain_core.callbacksr r �pydanticr
�langchain.chains.baser � getLoggerr' r r r r r r �<module> s