AXjXddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZddlZddlmZmZmZmZmZddlmZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2m3Z3ddl4m2Z5m3Z6e ge7fZ8e gdfZ9edeejtejvfZ<edee$ee%efZ=Gdde'Z>Gdde,er>, r1r>c(eZdZeeddZy)BetaFoundryMessagescyr@r<rAs r0rBzBetaFoundryMessages.batches5rCr1Nr2rDr<r1r0rGrG4rEr1rGc(eZdZeeddZy) BetaFoundryc,t|jSzKReturn beta messages resource instance with excluded unsupported endpoints.)rG_clientrAs r0messageszBetaFoundry.messages=s#4<<00r1N)r3 BetaMessagesr5r6r7r rrNr<r1r0rJrJ<s 11r1rJc(eZdZeeddZy)AsyncMessagesFoundrycyr@r<rAs r0rBzAsyncMessagesFoundry.batchesErCr1Nr2rDr<r1r0rRrRDrEr1rRc(eZdZeeddZy)AsyncBetaFoundryMessagescyr@r<rAs r0rBz AsyncBetaFoundryMessages.batchesMrCr1Nr2rDr<r1r0rUrULrEr1rUc(eZdZeeddZy)AsyncBetaFoundryc,t|jSrL)rUrMrAs r0rNzAsyncBetaFoundry.messagesUs( 55r1N)r3AsyncBetaMessagesrPr<r1r0rXrXTs 66r1rXc&eZdZeddddeedddddd ddZedddeedddddd ddZdddddeedddddd dfdZeedd Z eedd Z eedd Z eddddededdddeid dd Z e Z ddZeddZeeddZeddZxZS)AnthropicFoundryNF resourceapi_keyazure_ad_token_provider webhook_keytimeout max_retriesdefault_headers default_query http_client middleware_strict_response_validationc yNr< r.r^r_r`rarbrcrdrerfrgrhs r0r-zAnthropicFoundry.__init__`r1 r_r`rarbrcrdrerfrgrhc yrjr< r.base_urlr_r`rarbrcrdrerfrgrhs r0r-zAnthropicFoundry.__init__qrlr1 r^r_r`rarprbrcrdrerfrgrhc |||ntjjd}||ntjjd}||ntjjd}| | td|| t dd|d}n | t d t |||||||| | | | ||_y) 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 exclusive r_rarprbrcrdrerfrgrhosenvirongetr ValueErrorr,r-_azure_ad_token_providerr.r^r_r`rarprbrcrdrerfrgrhr/s r0r-zAnthropicFoundry.__init__6%0'bjjnnE`6a'38Hd9e'38Hd9e ?6> Y    K"(+LMH  !KL L ##+'#!(C  )@%r1cy)z>Models endpoint is not supported for Anthropic Foundry client.Nr<rAs r0modelszAnthropicFoundry.modelsrCr1ct|SzFReturn messages resource instance with excluded unsupported endpoints.client)r>rAs r0rNzAnthropicFoundry.messagessd++r1ct|S)BReturn beta resource instance with excluded unsupported endpoints.)rJrAs r0betazAnthropicFoundry.betas4  r1 r_r`rarprbrfrcrdset_default_headersreset_default_queryrg _extra_kwargsc $| | td| | td|j}| t||}n| | }|j}| i|| }n| | }|jd|xs |j |xs |j |xs |jt|xs |jt|tr |jn||xs |jt|r|n |j||t| tr |j n| d | Sz~ Create a new client instance re-using the same options given to the current client with optional overriding. zPThe `default_headers` and `set_default_headers` arguments are mutually exclusivezLThe `default_query` and `set_default_query` arguments are mutually exclusive) r_r`rarprbrfrcrdrergr<r_custom_headersr! _custom_queryr/r_rrastrrp isinstancerrbrMrrc _middlewarer.r_r`rarprbrfrcrdrrerrgrheadersparamss r0copyzAnthropicFoundry.copy-*  &+>+Jop p  $):)Fkl l&&  &#G_=G ,)G##  $00-0F  *&Ft~~  +t||$;$\t?\?\#7t'7'72T]]3$.w$ADLLw#3t||'/ '< $BRBR# +5j(+Kt''Q[    r1cr|j}|)|}|rt|tstd||SyNzOExpected `azure_ad_token_provider` argument to return a string but it returned )rrrrr.providertokens r0_get_azure_ad_tokenz$AnthropicFoundry._get_azure_ad_tokensI00  JE 5# 6 efkelmLr1c|t|jri|jni}t|}||_|j}||j dd||d<|S|j B|j d|j |d<|j d|j |d<|St dN AuthorizationzBearer z x-api-keyzapi-keyzUnable to handle authrrrrrr_rr.optionsrazure_ad_tokens r0_prepare_optionsz!AnthropicFoundry._prepare_optionss@H@Y))r3r")r_rr`rrarrpstr | httpx.URL | Nonerbrrfrrcint | NotGivenrdrrrrerrrrg+Sequence[MiddlewareInput] | None | NotGivenrMapping[str, Any]r3rr3rrrr3rr3zdict[str, str]rrrrr3r4r5r6r7r rrr-r rrrNrr with_optionsrrpropertyrrr8r9s@r0r\r\_s  $"?C"&5>.4859+/7;,1  "=   323)5&*   #?C"&5>.4859+/7;,1  "=   323)5&* & $"?C"&#5>.4859+/7;,19@9@ 9@ "= 9@  9@9@39@9@29@39@)9@59@&*9@ 9@v  ,, !!#?C"&+/5>+/&/488<599=BK+-2 2 "= 2  2 ) 2 32 )2 $2 22 62 32 72 @2 )2 !2 2 hL 4 r1r\c&eZdZeddddeedddddd ddZedddeedddddd ddZdddddeedddddd dfdZeedd Z eedd Z eedd Z eddddededdddeid dd Z e Z ddZeddZeeddZeddZxZS)AsyncAnthropicFoundryNFr]c yrjr<rks r0r-zAsyncAnthropicFoundry.__init__Arlr1rmc yrjr<ros r0r-zAsyncAnthropicFoundry.__init__Rrlr1rqc |||ntjjd}||ntjjd}||ntjjd}| | td|| t dd|d}n | t d t |||||||| | | | ||_y) 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. Nrsrtrurvrwrxryrzr{r|rs r0r-zAsyncAnthropicFoundry.__init__crr1cy)zwr)rinspect isawaitablerr rrrrs r0rz)AsyncAnthropicFoundry._get_azure_ad_tokenss00  JE""5)#  4U+;S A efkelmu: $s0A3A1?A3cKt|jri|jni}t|}||_|jd{}||j dd||d<|S|j B|j d|j |d<|j d|j |d<|St d7zwrrrs r0rz&AsyncAnthropicFoundry._prepare_optionss@H@Y).4859047;,1  "B   323.5&*   #DH"&5>.4859047;,1  "B   323.5&* & $"DH"&#5>.4859047;,19@9@ 9@ "B 9@  9@9@39@9@29@39@.9@59@&*9@ 9@v  11 --#DH"&+/5>04&/488<599=BK+-2 2 "B 2  2 ) 2 32 .2 $2 22 62 32 72 @2 )2 !2 2 hL .  r1r)H __future__rr}rtypingrrrrrr r r r functoolsr typing_extensionsrrhttpx_typesrrrrr_utilsrrMrr_compatr_modelsr _streamingrr _exceptionsrrr _base_clientrr r!resources.betar"r#resources.messagesr$r%resources.beta.messagesrOrZrAzureADTokenProviderAsyncAzureADTokenProviderClient AsyncClientr&r(r*r;r>rGrJrRrUrXr\rr<r1r0rsf" ^^^%, @@/ ),() -8bC($R)?%?@uU\\5CTCT5T/UV +5kRUFV9V3WX  I =/#ABHh,1$1=06y6^(vc{)BCY^BW-e.?.?SAQ.QRTbWr1