o +iD@s(ddlmZddlZddlZddlmZmZmZmZm 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 gdfZ4edeej5ej6fdZ7edee"ee#efdZ8Gddde%Z9Gddde(e7e8fZ:Gddde-Z;Gddde0ZGd!d"d"e1Z?Gd#d$d$e+Z@Gd%d&d&e:ej5e"efeZAGd'd(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_DefaultStreamTcseZdZdfdd ZZS)MutuallyExclusiveAuthErrorreturnNonecstddS)NznThe `api_key` and `azure_ad_token_provider` arguments are mutually exclusive; Only one can be passed at a time)super__init__self __class__R/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/anthropic/lib/foundry.pyr(sz#MutuallyExclusiveAuthError.__init__r%r&)__name__ __module__ __qualname__r( __classcell__r-r-r+r.r$sr$c@s eZdZdS)BaseFoundryClientN)r0r1r2r-r-r-r.r4$s r4c@eZdZeedddZdS)MessagesFoundryr%r&cCdSz?Batches endpoint is not supported for Anthropic Foundry client.Nr-r)r-r-r.batches(zMessagesFoundry.batchesNr/r0r1r2r r r9r-r-r-r.r6'r6c@r5)BetaFoundryMessagesr%r&cCr7r8r-r)r-r-r.r90r:zBetaFoundryMessages.batchesNr/r;r-r-r-r.r=/r<r=c@r5) BetaFoundryr% BetaMessagescC t|jSzKReturn beta messages resource instance with excluded unsupported endpoints.)r=_clientr)r-r-r.messages8 zBetaFoundry.messagesN)r%r?r0r1r2r r rCr-r-r-r.r>7r<r>c@r5)AsyncMessagesFoundryr%r&cCr7r8r-r)r-r-r.r9@r:zAsyncMessagesFoundry.batchesNr/r;r-r-r-r.rF?r<rFc@r5)AsyncBetaFoundryMessagesr%r&cCr7r8r-r)r-r-r.r9Hr:z AsyncBetaFoundryMessages.batchesNr/r;r-r-r-r.rGGr<rGc@r5)AsyncBetaFoundryr%AsyncBetaMessagescCr@rA)rGrBr)r-r-r.rCPrDzAsyncBetaFoundry.messagesN)r%rIrEr-r-r-r.rHOr<rHceZdZedddeeddddd d8ddZeddeedddddd9ddZddddeeddddd d:fddZeed;dd Z eedfd0d1Z e Z d?d2d3Zed@d6d7ZZS)AAnthropicFoundryNF resourceapi_keyazure_ad_token_providertimeout max_retriesdefault_headers default_query http_client_strict_response_validationrM str | NonerNrOAzureADTokenProvider | NonerP!float | Timeout | None | NotGivenrQintrRMapping[str, str] | NonerSMapping[str, object] | NonerThttpx.Client | NonerUboolr%r&c CdSNr- r*rMrNrOrPrQrRrSrTrUr-r-r.r([ zAnthropicFoundry.__init__rNrOrPrQrRrSrTrUbase_urlstrc Cr^r_r- r*rcrNrOrPrQrRrSrTrUr-r-r.r(jra rMrNrOrcrPrQrRrSrTrUc |dur|ntjd}|dur|ntjd}|dur|ntjd}|dur0|dur0td|durC|durModels endpoint is not supported for Anthropic Foundry client.Nr-r)r-r-r.modelsr:zAnthropicFoundry.modelsr6cC t|dSzFReturn messages resource instance with excluded unsupported endpoints.client)r6r)r-r-r.rCrDzAnthropicFoundry.messagesrcCst|S)BReturn beta resource instance with excluded unsupported endpoints.)r>r)r-r-r.betaszAnthropicFoundry.beta rNrO auth_tokenrcrPrTrQrRset_default_headersrSset_default_query _extra_kwargsrstr | httpx.URL | Noneint | NotGivenrrrMapping[str, Any]r c 0tj|||||||| | | d|p|ji| d Sz~ Create a new client instance re-using the same options given to the current client with optional overriding. rO) rNrrcrPrTrQrRrrSrrr'copyrv r*rNrOrrcrPrTrQrRrrSrrr+r-r.r  zAnthropicFoundry.copycCs8|j}|dur|}|rt|tstd||SdSNzOExpected `azure_ad_token_provider` argument to return a string but it returned )rv isinstancerdrur*providertokenr-r-r._get_azure_ad_tokensz$AnthropicFoundry._get_azure_ad_tokenoptionsrcCst|jr i|jni}t|}||_|}|dur+|ddur)d||d<|S|jdurE|ddurC|jdus>J|j|d<|StdN AuthorizationzBearer zapi-keyzUnable to handle authrheadersrrrtrNrur*rrazure_ad_tokenr-r-r._prepare_optionss  z!AnthropicFoundry._prepare_options)rMrVrNrVrOrWrPrXrQrYrRrZrSr[rTr\rUr]r%r&)rcrdrNrVrOrWrPrXrQrYrRrZrSr[rTr\rUr]r%r&)rMrVrNrVrOrWrcrVrPrXrQrYrRrZrSr[rTr\rUr]r%r&r/)r%r6)r%r)rNrVrOrWrrVrcrrPrXrTr\rQrrRrZrrZrSr[rr[rrr%r r%rVrrr%rr0r1r2r rrr(r r ryrCrr with_optionsrrr3r-r-r+r.rKZsx7$  rKcrJ)AAsyncAnthropicFoundryNFrLrMrVrNrO AsyncAzureADTokenProvider | NonerPrXrQrYrRrZrSr[rThttpx.AsyncClient | NonerUr]r%r&c Cr^r_r-r`r-r-r.r( razAsyncAnthropicFoundry.__init__rbrcrdc Cr^r_r-rer-r-r.r(rarfc rg) 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. Nrhrirjrkrlrmrnrorprqrwr+r-r.r(*rxcCr7)zrFrGrHrKrr-r-r-r.sB (        $2