`iD6ddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZmZddlZddlmZmZmZmZddlmZdd lmZmZdd lmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m-Z0m.Z1e ge2fZ3e gdfZ4edeej5ej6fZ7edee"ee#efZ8Gdde%Z9Gdde(e7e8fZ:Gdde-Z;Gdde0Z<Gdde*Z=Gdd e.Z>Gd!d"e1Z?Gd#d$e+Z@Gd%d&e:ej5e"efeZAGd'd(e:ej6e#efeZBdS))) annotationsN)AnyUnionMappingTypeVarCallable Awaitablecastoverload)cached_property)Selfoverride) NOT_GIVENOmitTimeoutNotGiven)is_given) AnthropicAsyncAnthropic) model_copy)FinalRequestOptions)Stream AsyncStream)AnthropicError)DEFAULT_MAX_RETRIES BaseClient)Beta AsyncBeta)Messages AsyncMessageszstr | Awaitable[str] _HttpxClientT)bound_DefaultStreamTc eZdZdfd ZxZS)MutuallyExclusiveAuthErrorreturnNonecJtddS)NznThe `api_key` and `azure_ad_token_provider` arguments are mutually exclusive; Only one can be passed at a time)super__init__)self __class__s i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/anthropic/lib/foundry.pyr+z#MutuallyExclusiveAuthError.__init__s,  |     r'r()__name__ __module__ __qualname__r+ __classcell__r-s@r.r&r&s=          r/r&ceZdZdS)BaseFoundryClientN)r1r2r3r/r.r7r7$sr/r7c6eZdZeeddZdS)MessagesFoundryr'r(cdSz?Batches endpoint is not supported for Anthropic Foundry client.Nr8r,s r.batcheszMessagesFoundry.batches( tr/Nr0r1r2r3r rr>r8r/r.r:r:'< X_r/r:c6eZdZeeddZdS)BetaFoundryMessagesr'r(cdSr<r8r=s r.r>zBetaFoundryMessages.batches0r?r/Nr0r@r8r/r.rCrC/rAr/rCc6eZdZeeddZdS) BetaFoundryr' BetaMessagesc*t|jSzKReturn beta messages resource instance with excluded unsupported endpoints.)rC_clientr=s r.messageszBetaFoundry.messages8s#4<000r/N)r'rGr1r2r3r rrKr8r/r.rFrF7s< 111X_111r/rFc6eZdZeeddZdS)AsyncMessagesFoundryr'r(cdSr<r8r=s r.r>zAsyncMessagesFoundry.batches@r?r/Nr0r@r8r/r.rNrN?rAr/rNc6eZdZeeddZdS)AsyncBetaFoundryMessagesr'r(cdSr<r8r=s r.r>z AsyncBetaFoundryMessages.batchesHr?r/Nr0r@r8r/r.rQrQGrAr/rQc6eZdZeeddZdS)AsyncBetaFoundryr'AsyncBetaMessagesc*t|jSrI)rQrJr=s r.rKzAsyncBetaFoundry.messagesPs( 555r/N)r'rUrLr8r/r.rTrTOs< 666X_666r/rTcfeZdZedddeeddddd d1dZeddeedddddd2dZddddeeddddd d3fdZeed4dZ eed5d Z eed6d"Z eddddededdddid# d7fd,Z e Z d8d-Zed9d0ZxZS):AnthropicFoundryNF resourceapi_keyazure_ad_token_providertimeout max_retriesdefault_headers default_query http_client_strict_response_validationrZ str | Noner[r\AzureADTokenProvider | Noner]!float | Timeout | None | NotGivenr^intr_Mapping[str, str] | Noner`Mapping[str, object] | Nonerahttpx.Client | Nonerbboolr'r(c dSNr8 r,rZr[r\r]r^r_r`rarbs r.r+zAnthropicFoundry.__init__[ sr/r[r\r]r^r_r`rarbbase_urlstrc dSrlr8 r,rpr[r\r]r^r_r`rarbs r.r+zAnthropicFoundry.__init__jrnr/ rZr[r\rpr]r^r_r`rarbc ||ntjd}||ntjd}||ntjd}||td||t dd|d}n|t d t ||||||| | ||_dS) aConstruct a new synchronous Anthropic Foundry client instance. This automatically infers the following arguments from their corresponding environment variables if they are not provided: - `api_key` from `ANTHROPIC_FOUNDRY_API_KEY` - `resource` from `ANTHROPIC_FOUNDRY_RESOURCE` - `base_url` from `ANTHROPIC_FOUNDRY_BASE_URL` Args: resource: Your Foundry resource name, e.g. `example-resource` for `https://example-resource.services.ai.azure.com/anthropic/` azure_ad_token_provider: A function that returns an Azure Active Directory token, will be invoked on every request. NANTHROPIC_FOUNDRY_API_KEYANTHROPIC_FOUNDRY_RESOURCEANTHROPIC_FOUNDRY_BASE_URLMissing credentials. Please pass one of `api_key`, `azure_ad_token_provider`, or the `ANTHROPIC_FOUNDRY_API_KEY` environment variable.tMust provide one of the `base_url` or `resource` arguments, or the `ANTHROPIC_FOUNDRY_RESOURCE` environment variablehttps://!.services.ai.azure.com/anthropic/,base_url and resource are mutually exclusiver[rpr]r^r_r`rarbosenvirongetr ValueErrorr*r+_azure_ad_token_provider r,rZr[r\rpr]r^r_r`rarbr-s r.r+zAnthropicFoundry.__init__y2%0''bjnnE`6a6a'388Hd9e9e'388Hd9e9e ?6> Y    KN(MMMHH  !KLL L #+'#(C  )@%%%r/cdS)z>Models endpoint is not supported for Anthropic Foundry client.Nr8r=s r.modelszAnthropicFoundry.modelsr?r/r:c"t|SzFReturn messages resource instance with excluded unsupported endpoints.client)r:r=s r.rKzAnthropicFoundry.messagessd++++r/rc t|S)BReturn beta resource instance with excluded unsupported endpoints.)rFr=s r.betazAnthropicFoundry.betas4   r/ r[r\ auth_tokenrpr]rar^r_set_default_headersr`set_default_query _extra_kwargsrstr | httpx.URL | Noneint | NotGivenrrrMapping[str, Any]r c rt|||||||| | | d|p|ji|  Sz~ Create a new client instance re-using the same options given to the current client with optional overriding. r\) r[rrpr]rar^r_rr`rrr*copyrr,r[r\rrpr]rar^r_rr`rrr-s r.rzAnthropicFoundry.copy](ww||!##+ 3'/)+B+cdFc   r/c|j}|5|}|rt|tstd||SdSNzOExpected `azure_ad_token_provider` argument to return a string but it returned )r isinstancerqrr,providertokens r._get_azure_ad_tokenz$AnthropicFoundry._get_azure_ad_tokens^0  HJJE  5# 6 6  mfkmmLtr/optionsrcVt|jr i|jni}t|}||_|}||dd||d<n?|j)|d|jJ|j|d<nt d|SN AuthorizationzBearer zapi-keyzUnable to handle authrheadersrrrr[rr,rrazure_ad_tokens r._prepare_optionsz!AnthropicFoundry._prepare_optionss@H@Y@Y)a).4859+/,1     X  #?C5>.4859+/,1     X " $"?C#5>.4859+/,15@5@5@5@5@5@5@5@n X_ ,,,X_, !!!X_!#?C!%+/5>+/&/488<599=+-" " " " " " " X" HL    Xr/rXcfeZdZedddeeddddd d1dZeddeedddddd2dZddddeeddddd d3fdZeed4dZ eed5d Z eed6d"Z eddddededdddid# d7fd,Z e Z d8d-Zed9d0ZxZS):AsyncAnthropicFoundryNFrYrZrcr[r\ AsyncAzureADTokenProvider | Noner]rer^rfr_rgr`rhrahttpx.AsyncClient | Nonerbrjr'r(c dSrlr8rms r.r+zAsyncAnthropicFoundry.__init__ rnr/rorprqc dSrlr8rss r.r+zAsyncAnthropicFoundry.__init__rnr/rtc ||ntjd}||ntjd}||ntjd}||td||t dd|d}n|t d t ||||||| | ||_dS) aConstruct a new asynchronous Anthropic Foundry client instance. This automatically infers the following arguments from their corresponding environment variables if they are not provided: - `api_key` from `ANTHROPIC_FOUNDRY_API_KEY` - `resource` from `ANTHROPIC_FOUNDRY_RESOURCE` - `base_url` from `ANTHROPIC_FOUNDRY_BASE_URL` Args: resource: Your Foundry resource name, e.g. `example-resource` for `https://example-resource.services.ai.azure.com/anthropic/` azure_ad_token_provider: A function that returns an Azure Active Directory token, will be invoked on every request. Nrvrwrxryrzr{r|r}r~rrs r.r+zAsyncAnthropicFoundry.__init__*rr/cdS)z.485904,1     X  #DH5>.485904,1     X " $"DH#5>.485904,15@5@5@5@5@5@5@5@n X_ 111X_1 ---X_-#DH!%+/5>04&/488<599=+-" " " " " " " X" HL    Xr/r)C __future__rrrtypingrrrrrr r r functoolsr typing_extensionsr rhttpx_typesrrrr_utilsrrJrr_compatr_modelsr _streamingrr _exceptionsr _base_clientrrresources.betarrresources.messagesr r!resources.beta.messagesrGrUrqAzureADTokenProviderAsyncAzureADTokenProviderClient AsyncClientr"r$r&r7r:rCrFrNrQrTrXrr8r/r.rs"""""" TTTTTTTTTTTTTTTTTTTT%%%%%%,,,,,,,, 777777777777//////// )))))),,,,,,,,((((((::::::::,,,,,,,,88888888bbbbbbbbC($R)?%?@uU\5CT5T/UVVV '+5kRUFV9V3WXXX        IHHHH =/#ABHHHh,11111$111=066666y666nnnnn(vc{)BCYnnnbppppp-e.?SAQ.QRTbpppppr/