܎jZNddlmZddlZddlZddlmZmZmZmZm 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 gdfZ9edeej:ej;fZ<edee$ee%efZ=Gdde'Z>Gdde,er>,< X_r3r>c6eZdZeeddZdS)BetaFoundryMessagesr+r,cdSr@r<rAs r2rBzBetaFoundryMessages.batches5rCr3Nr4rDr<r3r2rGrG4rEr3rGc6eZdZeeddZdS) BetaFoundryr+ BetaMessagesc*t|jSzKReturn beta messages resource instance with excluded unsupported endpoints.)rG_clientrAs r2messageszBetaFoundry.messages=s#4<000r3N)r+rKr5r6r7r rrOr<r3r2rJrJ<s< 111X_111r3rJc6eZdZeeddZdS)AsyncMessagesFoundryr+r,cdSr@r<rAs r2rBzAsyncMessagesFoundry.batchesErCr3Nr4rDr<r3r2rRrRDrEr3rRc6eZdZeeddZdS)AsyncBetaFoundryMessagesr+r,cdSr@r<rAs r2rBz AsyncBetaFoundryMessages.batchesMrCr3Nr4rDr<r3r2rUrULrEr3rUc6eZdZeeddZdS)AsyncBetaFoundryr+AsyncBetaMessagesc*t|jSrM)rUrNrAs r2rOzAsyncBetaFoundry.messagesUs( 555r3N)r+rYrPr<r3r2rXrXTs< 666X_666r3rXceZdZeddddeedddddd d:dZedddeedddddd d;dZdddddeedddddd d<fd Zeed=d!Z eed>d#Z eed?d%Z eddddededdddeid& d@d/Z e Z dAd0ZedBd3ZeedCd5ZedDd9ZxZS)EAnthropicFoundryNF resourceapi_keyazure_ad_token_provider webhook_keytimeout max_retriesdefault_headers default_query http_client middleware_strict_response_validationr^ str | Noner_r`AzureADTokenProvider | Nonerarb!float | Timeout | None | NotGivenrcintrdMapping[str, str] | NonereMapping[str, object] | Nonerfhttpx.Client | Nonerg Sequence[MiddlewareInput] | Nonerhboolr+r,c dSNr< r0r^r_r`rarbrcrdrerfrgrhs r2r/zAnthropicFoundry.__init__` sr3 r_r`rarbrcrdrerfrgrhbase_urlstrc dSrsr< r0rwr_r`rarbrcrdrerfrgrhs r2r/zAnthropicFoundry.__init__qrur3 r^r_r`rarwrbrcrdrerfrgrhc ||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 exclusive r_rarwrbrcrdrerfrgrhosenvirongetr ValueErrorr.r/_azure_ad_token_providerr0r^r_r`rarwrbrcrdrerfrgrhr1s r2r/zAnthropicFoundry.__init__6%0''bjnnE`6a6a'388Hd9e9e'388Hd9e9e ?6> Y    KN(MMMHH  !KLL L ##+'#!(C  )@%%%r3cdSz>Models endpoint is not supported for Anthropic Foundry client.Nr<rAs r2modelszAnthropicFoundry.modelsrCr3r>c"t|SzFReturn messages resource instance with excluded unsupported endpoints.client)r>rAs r2rOzAnthropicFoundry.messagessd++++r3r"c t|S)BReturn beta resource instance with excluded unsupported endpoints.)rJrAs r2betazAnthropicFoundry.betas4   r3 r_r`rarwrbrfrcrdset_default_headersreset_default_queryrg _extra_kwargsstr | httpx.URL | Noneint | NotGivenrr+Sequence[MiddlewareInput] | None | NotGivenrMapping[str, Any]rc || td| | td|j}|t||}n| | }|j}| i|| }n| | }|jd|p|j|p|j|p|jt|p|j t|tr|j n||p|j t|r|n|j||t| tr|jn| d | Sz~ Create a new client instance re-using the same options given to the current client with optional overriding. NzPThe `default_headers` and `set_default_headers` arguments are mutually exclusivezLThe `default_query` and `set_default_query` arguments are mutually exclusive) r_r`rarwrbrfrcrdrergr<r_custom_headersr! _custom_queryr1r_rrarxrw isinstancerrbrNrrc _middlewarer0r_r`rarwrbrfrcrdrrerrgrheadersparamss r2copyzAnthropicFoundry.copyD*  &+>+Jopp p  $):)Fkll l&  &#G_==GG ,)G#  $00-0FF  *&Ft~  +t|$;$\t?\#7t'72T]33$.w$A$ANDLLw#3t|'/ '<'<R $BR# +5j(+K+K[t''Q[      r3c|j}|5|}|rt|tstd||SdSNzOExpected `azure_ad_token_provider` argument to return a string but it returned )rrrxrr0providertokens r2_get_azure_ad_tokenz$AnthropicFoundry._get_azure_ad_tokens^0  HJJE  5# 6 6  mfkmmLtr3optionsrct|jr i|jni}t|}||_|}||dd||d<nU|j?|d |j|d<|d |j|d<nt d|SN AuthorizationzBearer z x-api-keyzapi-keyzUnable to handle authrrrrrr_rr0rrazure_ad_tokens r2_prepare_optionsz!AnthropicFoundry._prepare_optionss@H@Y@Y)a))r+r")r_rir`rjrarirwrrbrkrfrorcrrdrmrrmrernrrnrgrrrr+rr+rirrr+rr+rrrrrr+r,r5r6r7r rrr/r rrrOrr with_optionsrrpropertyrrr8r9s@r2r\r\_sb  $"?C"&5>.4859+/7;,1X  #?C"&5>.4859+/7;,1X& $"?C"&#5>.4859+/7;,19@9@9@9@9@9@9@9@v X_ ,,,X_, !!!X_!#?C"&+/5>+/&/488<599=BK+-2 2 2 2 2 X2 hL    X4 XXXr3r\ceZdZeddddeedddddd d:dZedddeedddddd d;dZdddddeedddddd d<fd Zeed=d!Z eed>d#Z eed?d%Z eddddededdddeid& d@d/Z e Z dAd0ZedBd3ZeedCd5ZedDd9ZxZS)EAsyncAnthropicFoundryNFr]r^rir_r` AsyncAzureADTokenProvider | Nonerarbrkrcrlrdrmrernrfhttpx.AsyncClient | Nonergrprhrqr+r,c dSrsr<rts r2r/zAsyncAnthropicFoundry.__init__Arur3rvrwrxc dSrsr<rzs r2r/zAsyncAnthropicFoundry.__init__Rrur3r{c ||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. Nr}r~rrrrrrrrrs r2r/zAsyncAnthropicFoundry.__init__crr3cdSrr<rAs r2rzAsyncAnthropicFoundry.modelsrCr3rRc"t|Sr)rRrAs r2rOzAsyncAnthropicFoundry.messagess$40000r3rXc"t|S)rr)rXrAs r2rzAsyncAnthropicFoundry.betas t,,,,r3rrrrrrrrrc || td| | td|j}|t||}n| | }|j}| i|| }n| | }|jd|p|j|p|j|p|jt|p|j t|tr|j n||p|j t|r|n|j||t| tr|jn| d | Srrrs r2rzAsyncAnthropicFoundry.copyrr3cK|j}|q|}tj|r|d{V}|r(tt t |t std|t |SdSr)rinspect isawaitablerr rrxrrs r2rz)AsyncAnthropicFoundry._get_azure_ad_tokens0  HJJE"5)) $#   4U+;+;S A A  mfkmmu:: tr3rrcKt|jr i|jni}t|}||_|d{V}||dd||d<nU|j?|d |j|d<|d |j|d<nt d|Srrrs r2rz&AsyncAnthropicFoundry._prepare_optionss@H@Y@Y)a).4859047;,1X  #DH"&5>.4859047;,1X& $"DH"&#5>.4859047;,19@9@9@9@9@9@9@9@v X_ 111X_1 ---X_-#DH"&+/5>04&/488<599=BK+-2 2 2 2 2 X2 hL    X. XX Xr3r)H __future__rrrtypingrrrrrr r r r functoolsr typing_extensionsrrhttpx_typesrrrrr_utilsrrNrr_compatr_modelsr _streamingrr _exceptionsrrr _base_clientrr r!resources.betar"r#resources.messagesr$r%resources.beta.messagesrKrYrxAzureADTokenProviderAsyncAzureADTokenProviderClient AsyncClientr&r(r*r;r>rGrJrRrUrXr\rr<r3r2rs""""""" ^^^^^^^^^^^^^^^^^^^^^^%%%%%%,,,,,,,, @@@@@@@@@@@@@@//////// )))))),,,,,,,,(((((()))))) -,,,,,,,88888888bbbbbbbbC($R)?%?@uU\5CT5T/UVVV '+5kRUFV9V3WXXX        IHHHH =/#ABHHHh,11111$111=066666y666^^^^^(vc{)BCY^^^BWWWWW-e.?SAQ.QRTbWWWWWr3