File: //usr/local/lib/python3.10/dist-packages/langchain_openai/llms/__pycache__/azure.cpython-310.pyc
o
;��g� � @ s� d dl mZ d dlZd dlmZmZmZmZmZm Z m
Z
m Z d dl Z d dl
mZ d dlmZmZ d dlmZmZmZ d dlmZmZ d dlmZ e�e �Z G d d
� d
e�Z dS ) � )� annotationsN)�Any� Awaitable�Callable�Dict�List�Mapping�Optional�Union)�LangSmithParams)�from_env�secret_from_env)�Field� SecretStr�model_validator)�Self�cast)�
BaseOpenAIc sx e Zd ZU dZeeddd�d�Zded< eddd �Zd
ed < ed ed
dd�d�Z ded< ede
ddgdd�d�Z ded< ee
ddd�d�Z ded< dZ
ded< dZded< eedd d�d�Zded < d Zd ed < edAd#d$��ZedBd&d'��ZedCd(d)��Zed*d+�dDd-d.��ZedE� fd0d1� �ZedF� fd3d4� �Z dGdH� fd:d;�
ZedId=d>��ZedFd?d@��Z � Z S )J� AzureOpenAIa� Azure-specific OpenAI large language models.
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_openai import AzureOpenAI
openai = AzureOpenAI(model_name="gpt-3.5-turbo-instruct")
�AZURE_OPENAI_ENDPOINTN)�default)�default_factoryz
Optional[str]�azure_endpoint�azure_deployment)r �aliaszUnion[str, None]�deployment_name� api_version�OPENAI_API_VERSION)r r �openai_api_version�api_key�AZURE_OPENAI_API_KEY�OPENAI_API_KEYzOptional[SecretStr]�openai_api_key�AZURE_OPENAI_AD_TOKEN�azure_ad_tokenz Union[Callable[[], str], None]�azure_ad_token_providerz)Union[Callable[[], Awaitable[str]], None]� azure_ad_async_token_provider�OPENAI_API_TYPE�azure�openai_api_typeT�bool�validate_base_url�return� List[str]c C s g d�S )z*Get the namespace of the langchain object.)� langchain�llms�openai� ��clsr1 r1 �F/usr/local/lib/python3.10/dist-packages/langchain_openai/llms/azure.py�get_lc_namespace] s z AzureOpenAI.get_lc_namespace�Dict[str, str]c C s
ddd�S )Nr r# )r" r$ r1 ��selfr1 r1 r4 �
lc_secretsb s �zAzureOpenAI.lc_secretsc C � dS )z9Return whether this model can be serialized by Langchain.Tr1 r2 r1 r1 r4 �is_lc_serializablei � z AzureOpenAI.is_lc_serializable�after)�moder c
C sN | j dk r td��| jr| j dkrtd��| jr!| jdkr!td��| j}|rD| jrDd|vr=tt| j��d�d | _td��| j rDtd��| j
| j | j | j rR| j �
� nd | jr[| j�
� nd | j| j| j| j| ji | jpki �d
d i�| jd � }| js�d
| ji}tjdi |�|��j| _| js�d
| ji}| j r�| j |d< tj di |�|��j| _| S )z?Validate that api key and python package exists in environment.� zn must be at least 1.z!Cannot stream results when n > 1.z'Cannot stream results when best_of > 1.z/openai�/z�As of openai>=1.0.0, Azure endpoints should be specified via the `azure_endpoint` param not `openai_api_base` (or alias `base_url`).z�As of openai>=1.0.0, if `deployment_name` (or alias `azure_deployment`) is specified then `openai_api_base` (or alias `base_url`) should not be. Instead use `deployment_name` (or alias `azure_deployment`) and `azure_endpoint`.Nz
User-Agentz%langchain-partner-python-azure-openai) r r r r r$ r% � organization�base_url�timeout� max_retries�default_headers�
default_query� http_clientr% r1 ) �n�
ValueError� streaming�best_of�openai_api_baser+ r �str�rstripr r r r" �get_secret_valuer$ r% �openai_organization�request_timeoutrD rE rF �clientrG r0 r � completions� async_client�http_async_clientr& �AsyncAzureOpenAI)r8 rL �
client_params�
sync_specific�async_specificr1 r1 r4 �validate_environmentn sz
��� � ����
���
����z AzureOpenAI.validate_environment�Mapping[str, Any]c s i d| j i�t� j�S )Nr )r �super�_identifying_paramsr7 �� __class__r1 r4 r] � s
��z AzureOpenAI._identifying_params�Dict[str, Any]c s d| j i}i |�t� j�S )N�model)r r\ �_invocation_params)r8 �
openai_paramsr^ r1 r4 rb � s
z AzureOpenAI._invocation_params�stop�Optional[List[str]]�kwargsr r c s> t � jdd|i|��}| j}d|d< |�d� }r ||d< |S )z Get standard params for tracing.rd r( � ls_providerra �
ls_model_nameNr1 )r\ �_get_ls_paramsrb �get)r8 rd rf �params�invocation_params�
model_namer^ r1 r4 ri � s zAzureOpenAI._get_ls_paramsrM c C r: )zReturn type of llm.r( r1 r7 r1 r1 r4 � _llm_type� r< zAzureOpenAI._llm_typec C s | j | jd�S )N�r) r ro r7 r1 r1 r4 �
lc_attributes� s �zAzureOpenAI.lc_attributes)r, r- )r, r6 )r, r* )r, r )r, r[ )r, r` )N)rd re rf r r, r )r, rM ) �__name__�
__module__� __qualname__�__doc__r r r �__annotations__r r r
r" r$ r% r& r) r+ � classmethodr5 �propertyr9 r; r rZ r] rb ri rn rp �
__classcell__r1 r1 r^ r4 r sf
�
���
�
� H� r ) �
__future__r �logging�typingr r r r r r r r
r0 � langchain_core.language_modelsr �langchain_core.utilsr r
�pydanticr r r �typing_extensionsr r �langchain_openai.llms.baser � getLoggerrq �loggerr r1 r1 r1 r4 �<module> s (