,:jD<ddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z ddlZddlmZmZmZmZddlmZmZddlmZmZdd lmZdd lmZmZdd lm Z dd l!m"Z"m#Z#e$gd Z%e ge&fZ'e gdfZ(edeejRejTfZ+edeeeeefZ,dj[gdZ.Gdde Z/Gdde#e+e,fZ0Gdde0ejReefeZ1Gdde0ejTeefeZ2y)) annotationsN)AnyUnionMappingTypeVarCallable Awaitableoverload)override) NOT_GIVENOmitTimeoutNotGiven)is_given is_mapping)OpenAI AsyncOpenAI)FinalRequestOptions)Stream AsyncStream) OpenAIError)DEFAULT_MAX_RETRIES BaseClient)z /completionsz/chat/completionsz /embeddingsz/audio/transcriptionsz/audio/translationsz/images/generationszstr | Awaitable[str] _HttpxClientT)bound_DefaultStreamT)c eZdZdfd ZxZS)MutuallyExclusiveAuthErrorc$t|dy)NzThe `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__s R/var/www/html/tokenscope/api/venv/lib/python3.12/site-packages/openai/lib/azure.pyr%z#MutuallyExclusiveAuthError.__init__+s  O )returnNone)__name__ __module__ __qualname__r% __classcell__r's@r(r"r"*s   r)r"c2eZdZe dfd ZxZS)BaseAzureClientc|jtvr_t|jrJ|jj d}|-dt |j vrd||j|_t|!|S)Nmodelz /deploymentsz /deployments/) url_deployments_endpointsr json_datagetstrbase_urlr$_build_request)r&optionsr4r's r(r;zBaseAzureClient._build_request2sq ;;0 0Z@Q@Q5R%%))'2E 3t}};M)M -eWW[[MB w%g..r))r<rr*z httpx.Request)r,r-r.r r;r/r0s@r(r2r21s'  /$ /  / /r)r2ceZdZeddddddeeddddd d dZeddddddeeddddd d dZedddddeeddddd ddZddddddddeeddddd dfd Zdd Zedfd Z xZ S) AzureOpenAINF azure_deployment api_versionapi_keyazure_ad_tokenazure_ad_token_provider organizationtimeout max_retriesdefault_headers default_query http_client_strict_response_validationc yNr&azure_endpointr@rArBrCrDrErFrGrHrIrJrKs r(r%zAzureOpenAI.__init__@$ r)c yrMrN r&r@rArBrCrDrErFrGrHrIrJrKs r(r%zAzureOpenAI.__init__T" r) rArBrCrDrErFrGrHrIrJrKc yrMrN r&r:rArBrCrDrErFrGrHrIrJrKs r(r%zAzureOpenAI.__init__grTr))rArPr@rBrCrDrEr:rFrGrHrIrJrKc |tjjd}|tjjd}|| | td|tjjd}| t d| d|i} nd|i| } |>|tjjd}| t d ||d |}n|d }n | t d |t }t ||||| | | | | | ||_||_ y)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` - `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. NAZURE_OPENAI_API_KEYAZURE_OPENAI_AD_TOKENMissing 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-versionAZURE_OPENAI_ENDPOINTuMust provide one of the `base_url` or `azure_endpoint` arguments, or the `AZURE_OPENAI_ENDPOINT` environment variable/openai/deployments//openai2base_url and azure_endpoint are mutually exclusive rBrEr:rFrGrHrIrJrK osenvironr8r ValueErrorAPI_KEY_SENTINELr$r%_azure_ad_token_azure_ad_token_provider)r&rArPr@rBrCrDrEr:rFrGrHrIrJrKr's r(r%zAzureOpenAI.__init__zhJ ?jjnn%;K  q   *K8M*KI=IM  %!#0G!H% L +,--ABRAST,-W5) !UVV ?&G %#+'#(C   .(?%r)c|j |jS|j}|)|}|rt|tst d||SyNzOExpected `azure_ad_token_provider` argument to return a string but it returned )rjrk isinstancer9rhr&providertokens r(_get_azure_ad_tokenzAzureOpenAI._get_azure_ad_tokensa    +'' '00  JE 5# 6 efkelmLr)cNt|jri|jni}||_|j}||jdGd||d<n>|jt ur!|jd|j|d<n t dt|!|SN AuthorizationzBearer zapi-keyzUnable to handle auth rheadersrsr8rBrirhr$_prepare_optionsr&r<rxrCr's r(ryzAzureOpenAI._prepare_optionss@H@Y)r>?s= (,"&"%)?C#'5>.4859+/,1  %      # "= ! 3  2 3 ) &* !  &(,"&"%)?C#'5>.4859+/,1 %     #  "= ! 3  2 3 ) &*    $ #'"%)?C#'5>.4859+/,1      #  "= ! 3  2 3 ) &*    *#'%)'+"%)?C#'#5>.4859+/,1!^@ ^@# ^@ % ^@  ^@#^@"=^@!^@^@3^@^@2^@3^@)^@ &*!^@" #^@@ 11r)r>ceZdZeddddddeeddddd d dZeddddddeeddddd d dZedddddeeddddd ddZddddddddeeddddd dfd Zdd Zedfd Z xZ S)AsyncAzureOpenAINFr?c yrMrNrOs r(r%zAsyncAzureOpenAI.__init__rQr)c yrMrNrSs r(r%zAsyncAzureOpenAI.__init__rTr)rUc yrMrNrWs r(r%zAsyncAzureOpenAI.__init__$rTr))rPr@rArBrCrDrEr:rFrGrHrIrJrKc |tjjd}|tjjd}|| | td|tjjd}| t d| d|i} nd|i| } |>|tjjd}| t d ||d |}n|d }n | t d |t }t ||||| | | | | | ||_||_ y)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` - `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. NrYrZr[r\r]r^r_r`rarbrcrdre)r&rPr@rArBrCrDrEr:rFrGrHrIrJrKr's r(r%zAsyncAzureOpenAI.__init__7rlr)cK|j |jS|j}|H|}tj|r |d{}|rt |t st d||Sy7'wrn)rjrkinspect isawaitableror9rhrps r(rsz$AsyncAzureOpenAI._get_azure_ad_tokens    +'' '00  JE""5)#  5# 6 efkelmL$sAA4 A2 (A4c~Kt|jri|jni}||_|jd{}||jdGd||d<n>|jt ur!|jd|j|d<n t dt|!|d{S7u7wrurwrzs r(ryz!AsyncAzureOpenAI._prepare_optionss@H@Y).485904,1  %      # "B ! 3  2 3 . &* !  &(,"&"%)DH#'5>.485904,1 %     #  "B ! 3  2 3 . &*    $ #'"%)DH#'5>.485904,1      #  "B ! 3  2 3 . &*    *&*'+"&"%)DH#'#5>.485904,1!^@#^@% ^@  ^@  ^@#^@"B^@!^@^@3^@^@2^@3^@.^@ &*!^@" #^@@"77r)r)3 __future__rrfrtypingrrrrrr r typing_extensionsr httpx_typesr rrr_utilsrr_clientrr_modelsr _streamingrr _exceptionsr _base_clientrrsetr6r9AzureADTokenProviderAsyncAzureADTokenProviderClient AsyncClientrrjoinrir"r2r>rrNr)r(rs$" NNN& 77))),%:  C($R)?%?@uU\\5CTCT5T/UV +5kRUFV9V3WX 7789   /j!?@ /z1/%,,s ";