jG6ddlmZddlZddlZddlmZmZmZmZm 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 `/DATA/AppData/hermes/projects/honcho/.venv/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/rTcneZdZeddddeeddddd d2dZedddeeddddd d3dZdddddeeddddd d4fdZeed5dZ eed6d!Z eed7d#Z edddddededdddid$ d8fd-Z e Z d9d.Zed:d1ZxZS);AnthropicFoundryNF resourceapi_keyazure_ad_token_provider webhook_keytimeout max_retriesdefault_headers default_query http_client_strict_response_validationrZ str | Noner[r\AzureADTokenProvider | Noner]r^!float | Timeout | None | NotGivenr_intr`Mapping[str, str] | NoneraMapping[str, object] | Nonerbhttpx.Client | Nonercboolr'r(c dSNr8 r,rZr[r\r]r^r_r`rarbrcs r.r+zAnthropicFoundry.__init__[ sr/ r[r\r]r^r_r`rarbrcbase_urlstrc dSrmr8 r,rqr[r\r]r^r_r`rarbrcs r.r+zAnthropicFoundry.__init__kror/ rZr[r\r]rqr^r_r`rarbrcc ||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[r]rqr^r_r`rarbrcosenvirongetr ValueErrorr*r+_azure_ad_token_provider r,rZr[r\r]rqr^r_r`rarbrcr-s r.r+zAnthropicFoundry.__init__{4%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_tokenr]rqr^rbr_r`set_default_headersraset_default_query _extra_kwargsrstr | httpx.URL | Noneint | NotGivenrrrMapping[str, Any]r c tt|||||||| | | | d|p|ji|  Sz~ Create a new client instance re-using the same options given to the current client with optional overriding. r\) r[rr]rqr^rbr_r`rrarrr*copyrr,r[r\rr]rqr^rbr_r`rrarrr-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 isinstancerrrr,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  #?C"&5>.4859+/,1     X $ $"?C"&#5>.4859+/,17@7@7@7@7@7@7@7@r X_ ,,,X_, !!!X_!#?C!%"&+/5>+/&/488<599=+-$ $ $ $ $ $ $ X$ LL    Xr/rXcneZdZeddddeeddddd d2dZedddeeddddd d3dZdddddeeddddd d4fdZeed5dZ eed6d!Z eed7d#Z edddddededdddid$ d8fd-Z e Z d9d.Zed:d1ZxZS);AsyncAnthropicFoundryNFrYrZrdr[r\ AsyncAzureADTokenProvider | Noner]r^rfr_rgr`rhrarirbhttpx.AsyncClient | Nonercrkr'r(c dSrmr8rns r.r+zAsyncAnthropicFoundry.__init__ror/rprqrrc dSrmr8rts r.r+zAsyncAnthropicFoundry.__init__"ror/ruc ||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. Nrwrxryrzr{r|r}r~rrrs r.r+zAsyncAnthropicFoundry.__init__2rr/cdS)z.485904,1     X  #DH"&5>.485904,1     X $ $"DH"&#5>.485904,17@7@7@7@7@7@7@7@r X_ 111X_1 ---X_-#DH!%"&+/5>04&/488<599=+-$ $ $ $ $ $ $ X$ LL    Xr/r)C __future__rrrtypingrrrrrr r r functoolsr typing_extensionsr rhttpx_typesrrrr_utilsrrJrr_compatr_modelsr _streamingrr _exceptionsr _base_clientrrresources.betarrresources.messagesr r!resources.beta.messagesrGrUrrAzureADTokenProviderAsyncAzureADTokenProviderClient AsyncClientr"r$r&r7r:rCrFrNrQrTrXrr8r/r.rs"""""" TTTTTTTTTTTTTTTTTTTT%%%%%%,,,,,,,, 777777777777//////// )))))),,,,,,,,((((((::::::::,,,,,,,,88888888bbbbbbbbC($R)?%?@uU\5CT5T/UVVV '+5kRUFV9V3WXXX        IHHHH =/#ABHHHh,11111$111=066666y666ttttt(vc{)BCYtttnvvvvv-e.?SAQ.QRTbvvvvvr/