File: //usr/local/lib/python3.10/dist-packages/langchain/chains/__pycache__/moderation.cpython-310.pyc
o
���g� � @ sl d Z ddlmZmZmZmZ ddlmZmZ ddl m
Z
m Z ddl m
Z
mZ ddlmZ G dd� de�Zd S )
z)Pass input through a moderation endpoint.� )�Any�Dict�List�Optional)� AsyncCallbackManagerForChainRun�CallbackManagerForChainRun)�check_package_version�get_from_dict_or_env)�Field�model_validator)�Chainc sZ e Zd ZU dZdZeed< dZeed< dZe e
ed< dZ e ed< dZ
e
ed < d
Ze
ed < dZe e
ed < dZe e
ed
< edd�Ze ed< edd�ededefdd���Zedee
fdd��Zedee
fdd��Zde
dede
fd d �Z d$d ee
ef d e e dee
ef fd d!�Z d$d ee
ef d e e dee
ef f� fd"d#�
Z � Z S )%�OpenAIModerationChaina� Pass input through a moderation endpoint.
To use, you should have the ``openai`` python package installed, and the
environment variable ``OPENAI_API_KEY`` set with your API key.
Any parameters that are valid to be passed to the openai.create call can be passed
in, even if not explicitly saved on this class.
Example:
.. code-block:: python
from langchain.chains import OpenAIModerationChain
moderation = OpenAIModerationChain()
N�client� async_client�
model_nameF�error�input� input_key�output�
output_key�openai_api_key�openai_organization��default�openai_pre_1_0�before)�mode�values�returnc C s� t |dd�}t |dddd�}zEddl}||_|r||_d |d
< ztd d d
� W n
ty4 d|d
< Y nw |d
rA|j|d< W |S |j|d�|d< |j|d�|d< W |S t y^ t d��w )z?Validate that api key and python package exists in environment.r �OPENAI_API_KEYr �OPENAI_ORGANIZATION� r r NFr �openaiz1.0)� gte_versionTr )�api_keyr zTCould not import openai python package. Please install it with `pip install openai`.)
r r" r$ � organizationr �
ValueError�
Moderation�OpenAI� AsyncOpenAI� ImportError)�clsr r r r" � r, �F/usr/local/lib/python3.10/dist-packages/langchain/chains/moderation.py�validate_environment+ s> �� �
� ���z*OpenAIModerationChain.validate_environmentc C � | j gS )z2Expect input key.
:meta private:
)r ��selfr, r, r- �
input_keysP � z OpenAIModerationChain.input_keysc C r/ )z3Return output key.
:meta private:
)r r0 r, r, r- � output_keysX r3 z!OpenAIModerationChain.output_keys�text�resultsc C s4 | j r|d }n|j}|rd}| jrt|��|S |S )N�flaggedz5Text was found that violates OpenAI's content policy.)r r7 r r& )r1 r5 r6 � condition� error_strr, r, r- � _moderate` s
z OpenAIModerationChain._moderate�inputs� run_managerc C s^ || j }| jr| j�|�}| �||d d �}n| jjj|d�}| �||jd �}| j|iS )Nr6 r �r )r r r �creater: � moderationsr6 r �r1 r; r< r5 r6 r r, r, r- �_callm s
zOpenAIModerationChain._callc � sZ �| j rt� j||d�I d H S || j }| jjj|d�I d H }| �||jd �}| j |iS )N)r< r= r )
r �super�_acallr r r? r> r: r6 r r@ �� __class__r, r- rC { s �
z OpenAIModerationChain._acall)N)!�__name__�
__module__� __qualname__�__doc__r r �__annotations__r r r �strr �boolr r r r r
r r � classmethodr r. �propertyr r2 r4 r: r rA r rC �
__classcell__r, r, rD r- r
sJ
#�
��
��
��
�r
N)rI �typingr r r r �langchain_core.callbacksr r �langchain_core.utilsr r �pydanticr
r �langchain.chains.baser r
r, r, r, r- �<module> s