File: //home/arjun/.local/lib/python3.10/site-packages/openai/lib/__pycache__/azure.cpython-310.pyc
o
���g \ � @ s� d dl mZ d dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z d dl
mZmZ d dlZddlmZmZmZmZmZ ddlmZmZ ddlmZm Z dd l m Z dd
l m Z dd l!m"Z"m#Z# dd l$m%Z% dd
l&m'Z'm(Z( e)g d��Z*e g e+f Z,e g df Z-edeej.ej/f d�Z0edee"e e#e f d�Z1d�2g d��Z3G dd� de%�Z4G dd� de(e0e1f �Z5G dd� de5ej.e"e f e�Z6G dd � d e5ej/e#e f e �Z7dS ) � )� annotationsN)�Any�Union�Mapping�TypeVar�Callable� Awaitable�cast�overload)�Self�override� )� NOT_GIVEN�Omit�Query�Timeout�NotGiven)�is_given�
is_mapping)�OpenAI� AsyncOpenAI)�
model_copy)�FinalRequestOptions)�Stream� AsyncStream)� OpenAIError)�DEFAULT_MAX_RETRIES�
BaseClient)z /completionsz/chat/completionsz /embeddingsz/audio/transcriptionsz/audio/translationsz
/audio/speechz/images/generationszstr | Awaitable[str]�
_HttpxClientT)�bound�_DefaultStreamT� )�<zmissing API key�>c s e Zd Zd� fdd� Z� ZS )�MutuallyExclusiveAuthError�return�Nonec s t � �d� d S )Nz�The `api_key`, `azure_ad_token` and `azure_ad_token_provider` arguments are mutually exclusive; Only one can be passed at a time)�super�__init__)�self�� __class__� �C/home/arjun/.local/lib/python3.10/site-packages/openai/lib/azure.pyr( - s �z#MutuallyExclusiveAuthError.__init__)r% r& )�__name__�
__module__� __qualname__r( �
__classcell__r, r, r* r- r$ , s r$ c s( e Zd Zedd�d � fd d
��Z� ZS ) �BaseAzureClientr ��
retries_taken�optionsr r4 �intr% �
httpx.Requestc sX |j tv r$t|j�r$|j�d�}|d ur$dt| j�vr$d|� |j � �|_ t� j||d�S )N�modelz /deploymentsz
/deployments/r3 ) �url�_deployments_endpointsr � json_data�get�str�base_urlr' �_build_request)r) r5 r4 r8 r* r, r- r? 4 s
z BaseAzureClient._build_request)r5 r r4 r6 r% r7 )r. r/ r0 r r? r1 r, r, r* r- r2 3 s �r2 c s e Zd Zedddddddeeddddd�
d<d d ��Zedddddddeeddddd�
d=d d ��Zeddddddeeddddd!� d>d#d ��Zddddddddddeeddddd$�d?� fd&d �Zeddddddddededdddi d'�d@� fd.d/��ZeZ dAd0d1�Z
edBd4d5��Z dCd:d;�Z � Z
S )D� AzureOpenAINF)
�azure_deployment� api_version�api_key�azure_ad_token�azure_ad_token_provider� organization�websocket_base_url�timeout� max_retries�default_headers�
default_query� http_client�_strict_response_validation�azure_endpointr= rA �
str | NonerB rC rD rE �AzureADTokenProvider | NonerF rG �str | httpx.URL | NonerH �!float | Timeout | None | NotGivenrI r6 rJ �Mapping[str, str] | NonerK �Mapping[str, object] | NonerL �httpx.Client | NonerM �boolr% r& c C � d S �Nr, )r) rN rA rB rC rD rE rF rG rH rI rJ rK rL rM r, r, r- r( D � zAzureOpenAI.__init__c
C rW rX r, )r) rA rB rC rD rE rF rG rH rI rJ rK rL rM r, r, r- r( X � ) rB rC rD rE rF rG rH rI rJ rK rL rM r> c
C rW rX r, )r) r> rB rC rD rE rF rG rH rI rJ rK rL rM r, r, r- r( k rZ )rB rN rA rC rD rE rF �projectrG r> rH rI rJ rK rL rM r[ c
s> |du r
t j�d�}|du rt j�d�}|du r$|du r$|du r$td��|du r.t j�d�}|du r6td��|du r?d|i}ni |�d|i�}|
du ru|du rUt j�d�}|du r]td ��|durl|�d
�� d |� �}
n|�d
�� d �}
n|dur}td
��|du r�t}t� j||||
| | |
||| |d� || _ || _
|| _ dS )a� Construct a new synchronous azure openai client instance.
This automatically infers the following arguments from their corresponding environment variables if they are not provided:
- `api_key` from `AZURE_OPENAI_API_KEY`
- `organization` from `OPENAI_ORG_ID`
- `project` from `OPENAI_PROJECT_ID`
- `azure_ad_token` from `AZURE_OPENAI_AD_TOKEN`
- `api_version` from `OPENAI_API_VERSION`
- `azure_endpoint` from `AZURE_OPENAI_ENDPOINT`
Args:
azure_endpoint: Your Azure endpoint, including the resource, e.g. `https://example-resource.azure.openai.com/`
azure_ad_token: Your Azure Active Directory token, https://www.microsoft.com/en-us/security/business/identity-access/microsoft-entra-id
azure_ad_token_provider: A function that returns an Azure Active Directory token, will be invoked on every request.
azure_deployment: A model deployment, if given sets the base client URL to include `/deployments/{azure_deployment}`.
Note: this means you won't be able to use non-deployment endpoints. Not supported with Assistants APIs.
N�AZURE_OPENAI_API_KEY�AZURE_OPENAI_AD_TOKEN��Missing credentials. Please pass one of `api_key`, `azure_ad_token`, `azure_ad_token_provider`, or the `AZURE_OPENAI_API_KEY` or `AZURE_OPENAI_AD_TOKEN` environment variables.�OPENAI_API_VERSION�_Must provide either the `api_version` argument or the `OPENAI_API_VERSION` environment variable� api-version�AZURE_OPENAI_ENDPOINT�uMust provide one of the `base_url` or `azure_endpoint` arguments, or the `AZURE_OPENAI_ENDPOINT` environment variable�/�/openai/deployments/�/openai�2base_url and azure_endpoint are mutually exclusive� rC rF r[ r> rH rI rJ rK rL rG rM � �os�environr< r �
ValueError�rstrip�API_KEY_SENTINELr' r( � _api_version�_azure_ad_token�_azure_ad_token_provider)r) rB rN rA rC rD rE rF r[ rG r> rH rI rJ rK rL rM r* r, r- r( ~ �^ ( � �
��
�rC rF r[ rG rB rD rE r> rH rL rI rJ �set_default_headersrK �set_default_query�
_extra_kwargs�int | NotGivenrt ru rv �Mapping[str, Any]r c �D t � j|||||| |
| | |
|||p| j|p| j|p| jd�|�d�
S �z~
Create a new client instance re-using the same options given to the current client with optional overriding.
)rB rD rE )
rC rF r[ rG r> rH rL rI rJ rt rK ru rv �r' �copyro rp rq �r) rC rF r[ rG rB rD rE r> rH rL rI rJ rt rK ru rv r* r, r- r| � �( ���zAzureOpenAI.copyc C sH | j d ur| j S | j}|d ur"|� }|rt|t�s td|� ���|S d S �NzOExpected `azure_ad_token_provider` argument to return a string but it returned )rp rq �
isinstancer= rl �r) �provider�tokenr, r, r- �_get_azure_ad_token s
�z AzureOpenAI._get_azure_ad_tokenr5 r c C s� t |j�r
i |j�ni }t|�}||_| �� }|d ur+|�d�d u r)d|� �|d< |S | jtur>|�d�d u r<| j|d< |S td���N�
Authorization�Bearer �api-keyzUnable to handle auth�r �headersr r� r< rC rn rl �r) r5 r� rD r, r, r- �_prepare_options" s
�
�z AzureOpenAI._prepare_optionsr8 � extra_queryr � tuple[Query, dict[str, str]]c C sV i }i |�| j |d��}| jdkrd| ji}||fS | �� }|r'dd|� �i}||fS �N)ra �
deploymentz<missing API key>r� r� r� �ro rC r� �r) r8 r� � auth_headers�queryr� r, r, r- �_configure_realtime6 s ��
�z AzureOpenAI._configure_realtime) rN r= rA rO rB rO rC rO rD rO rE rP rF rO rG rQ rH rR rI r6 rJ rS rK rT rL rU rM rV r% r& ) rA rO rB rO rC rO rD rO rE rP rF rO rG rQ rH rR rI r6 rJ rS rK rT rL rU rM rV r% r& ) r> r= rB rO rC rO rD rO rE rP rF rO rG rQ rH rR rI r6 rJ rS rK rT rL rU rM rV r% r& )"rB rO rN rO rA rO rC rO rD rO rE rP rF rO r[ rO rG rQ r> rO rH rR rI r6 rJ rS rK rT rL rU rM rV r% r& )"rC rO rF rO r[ rO rG rQ rB rO rD rO rE rP r> rQ rH rR rL rU rI rw rJ rS rt rS rK rT ru rT rv rx r% r �r% rO �r5 r r% r �r8 r= r� r r% r� �r. r/ r0 r
r r r( r r| � with_optionsr� r� r� r1 r, r, r* r- r@ C s� ����f�,
r@ c s& e Zd Zeddddddddeeddddd�d<d d ��Zeddddddddeeddddd�d=d!d ��Zedddddddeeddddd"�
d>d$d ��Zddddddddddeeddddd%�d?� fd&d �Zeddddddddededdddi d'�d@� fd.d/��ZeZ dAd0d1�Z
edBd4d5��Z dCd:d;�Z � Z
S )D�AsyncAzureOpenAINF)rA rB rC rD rE rF r[ rG rH rI rJ rK rL rM rN r= rA rO rB rC rD rE � AsyncAzureADTokenProvider | NonerF r[ rG rQ rH rR rI r6 rJ rS rK rT rL �httpx.AsyncClient | NonerM rV r% r& c C rW rX r, )r) rN rA rB rC rD rE rF r[ rG rH rI rJ rK rL rM r, r, r- r( G s zAsyncAzureOpenAI.__init__c C rW rX r, )r) rA rB rC rD rE rF r[ rG rH rI rJ rK rL rM r, r, r- r( \ rY )
rB rC rD rE rF r[ rG rH rI rJ rK rL rM r> c C rW rX r, )r) r> rB rC rD rE rF r[ rG rH rI rJ rK rL rM r, r, r- r( p rY )rN rA rB rC rD rE rF r[ r> rG rH rI rJ rK rL rM c
s> |du r
t j�d�}|du rt j�d�}|du r$|du r$|du r$td��|du r.t j�d�}|du r6td��|du r?d|i}ni |�d|i�}| du ru|du rUt j�d�}|du r]td ��|durl|�d
�� d |� �} n|�d
�� d �} n|dur}td
��|du r�t}t� j|||| | | |
|||
|d� || _ || _
|| _ dS )a� Construct a new asynchronous azure openai client instance.
This automatically infers the following arguments from their corresponding environment variables if they are not provided:
- `api_key` from `AZURE_OPENAI_API_KEY`
- `organization` from `OPENAI_ORG_ID`
- `project` from `OPENAI_PROJECT_ID`
- `azure_ad_token` from `AZURE_OPENAI_AD_TOKEN`
- `api_version` from `OPENAI_API_VERSION`
- `azure_endpoint` from `AZURE_OPENAI_ENDPOINT`
Args:
azure_endpoint: Your Azure endpoint, including the resource, e.g. `https://example-resource.azure.openai.com/`
azure_ad_token: Your Azure Active Directory token, https://www.microsoft.com/en-us/security/business/identity-access/microsoft-entra-id
azure_ad_token_provider: A function that returns an Azure Active Directory token, will be invoked on every request.
azure_deployment: A model deployment, if given sets the base client URL to include `/deployments/{azure_deployment}`.
Note: this means you won't be able to use non-deployment endpoints. Not supported with Assistants APIs.
Nr\ r] r^ r_ r` ra rb rc rd re rf rg rh ri )r) rN rA rB rC rD rE rF r[ r> rG rH rI rJ rK rL rM r* r, r- r( � rr rs rw rt ru rv rx r c ry rz r{ r} r* r, r- r| � r~ zAsyncAzureOpenAI.copyc � sh �| j d ur | j S | j}|d ur2|� }t�|�r |I d H }|r'ttt|�t�s.td|� ���t|�S d S r ) rp rq �inspect� isawaitabler� r r r= rl r� r, r, r- r� s �
�z$AsyncAzureOpenAI._get_azure_ad_tokenr5 r c � s� �t |j�r i |j�ni }t|�}||_| �� I d H }|d ur/|�d�d u r-d|� �|d< |S | jturB|�d�d u r@| j|d< |S td��r� r� r� r, r, r- r� * s �
�
�z!AsyncAzureOpenAI._prepare_optionsr8 r� r r� c � s^ �i }i |�| j |d��}| jdkrd| ji}||fS | �� I d H }|r+dd|� �i}||fS r� r� r� r, r, r- r� > s ���
�z$AsyncAzureOpenAI._configure_realtime) rN r= rA rO rB rO rC rO rD rO rE r� rF rO r[ rO rG rQ rH rR rI r6 rJ rS rK rT rL r� rM rV r% r& ) rA rO rB rO rC rO rD rO rE r� rF rO r[ rO rG rQ rH rR rI r6 rJ rS rK rT rL r� rM rV r% r& ) r> r= rB rO rC rO rD rO rE r� rF rO r[ rO rG rQ rH rR rI r6 rJ rS rK rT rL r� rM rV r% r& )"rN rO rA rO rB rO rC rO rD rO rE r� rF rO r[ rO r> rO rG rQ rH rR rI r6 rJ rS rK rT rL r� rM rV r% r& )"rC rO rF rO r[ rO rG rQ rB rO rD rO rE r� r> rQ rH rR rL r� rI rw rJ rS rt rS rK rT ru rT rv rx r% r r� r� r� r� r, r, r* r- r� F s� ����f�,
r� )8�
__future__r rj r� �typingr r r r r r r r
�typing_extensionsr r �httpx�_typesr r r r r �_utilsr r �_clientr r �_compatr �_modelsr �
_streamingr r � _exceptionsr � _base_clientr r �setr: r= �AzureADTokenProvider�AsyncAzureADTokenProvider�Client� AsyncClientr r �joinrn r$ r2 r@ r� r, r, r, r- �<module> s8 ( �
$