܎j|ddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZddlZddlmZmZmZmZddlmZmZdd lmZm Z dd l!m"Z"dd l#m$Z$m%Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0ee0dZ1ee/dZ2erddl3m4Z5dZ6dZ7e ge8fZ9e gdfZ:e deej;ej<fZ=e dee$ee%efZ>GddZ?Gdde,e=e>fZ@d5d(ZAd6d.ZBd7d0ZCGd1d2e@ej;e$efeZDGd3d4e@ej<e%efe ZEdS)8) annotationsN) TYPE_CHECKINGAnyUnionMappingTypeVarCallableSequence Awaitablecast)partialcached_property)Selfoverride)HeadersTimeoutNotGiven not_given)asyncifyis_given) AnthropicAsyncAnthropic)FinalRequestOptions)Stream AsyncStream)AnthropicError)MiddlewareInput)DEFAULT_MAX_RETRIES BaseClient merge_headers)refresh_credentialsload_default_credentials google_cloud)extra) Credentialsznhttps://claude.googleapis.com/v1alpha/projects/{project}/locations/{location}/workspaces/{workspace_id}/invokeglobalzstr | Awaitable[str] _HttpxClientT)bound_DefaultStreamTc"eZdZdZd dZd dZd S) _GoogleCredentialsStateuHolder for the Google credentials object, its refresh lock, and the project ADC resolved. Shared between a client and its ``copy()``/``with_options()`` clones (when the credential configuration is inherited) so a lazily-loaded ADC credential is minted once — not once per clone — and concurrent loads/refreshes are serialized: google-auth credential objects are not safe to ``refresh()`` concurrently. credentialsGoogleCredentials | NonereturnNonecRtj|_||_d|_dSN) threadingLock_lockr. adc_project)selfr.s PC:\Projects\DSCons\.venv\Lib\site-packages\anthropic\lib\google_cloud\_client.py__init__z _GoogleCredentialsState.__init__:s'^%% 5@'+strcN|j5|jt\|_|_n,|jjs |jjst |j|jj}|stdt|tsJ|cdddS#1swxYwYdS)zJReturn a valid access token, loading ADC / refreshing as needed. Blocking.NzKCould not resolve a GCP access token from the configured Google credentials) r6r._load_adc_credentialsr7expiredtoken_refresh_credentialsr isinstancer<)r8r@s r9r@z_GoogleCredentialsState.token?s Z  '5J5L5L2 $"2"2!) 71A1G 7$T%5666$*E t$%rssseS)) ) ))                  sBBB!BN)r.r/r0r1r0r<)__name__ __module__ __qualname____doc__r:r@r;r9r-r-/sF,,,,      r;r-ceZdZUdZded<ded<ded<ded<d ed <d ed <eeddZejddZeddZ ddZ dS)BaseGoogleCloudClientzMarker base so ``_is_base_client()`` keeps these clients off the first-party credential-discovery chain (it matches only the exact ``Anthropic`` / ``AsyncAnthropic`` classes). Auth is handled entirely by this helper. str | None workspace_id_projectr< _locationr- _creds_statebool_base_url_deferred_base_url_overriddenr0 httpx.URLc|jSr3) _base_urlr8s r9base_urlzBaseGoogleCloudClient.base_urlZs ~r;urlhttpx.URL | strr1cd|_|t|tjr|ntj||_dS)NF)rQ_enforce_trailing_slashrBhttpxURLrU)r8rXs r9rWzBaseGoogleCloudClient.base_url_sG#(55ZUY=W=W6kcc]b]fgj]k]kllr;r/c|jjS)zThe ``google.auth`` credentials in use (explicit or lazily-loaded ADC), if any. Distinct from ``.credentials``, which is the base client's first-party credentials provider and is always ``None`` on this client. )rOr.rVs r9google_credentialsz(BaseGoogleCloudClient.google_credentialsfs ,,r;c|jsdS|jtd|jJt|j|j|j|_d|_dS)zDerive and set the real base URL once the project is known. Called on every request (after the project back-fill from ADC, if any); a no-op once the base URL has been derived. NzNo `project` was provided and one could not be resolved from Google credentials. Pass the `project` argument, set the `ANTHROPIC_GOOGLE_CLOUD_PROJECT` environment variable, or provide `base_url` directly.projectlocationrLF)rQrMrrLDEFAULT_URL_TEMPLATEformatrNrWrVs r9_resolve_deferred_base_urlz0BaseGoogleCloudClient._resolve_deferred_base_urlos &  F =  H   ,,,,33MDNIZ4   #(r;N)r0rS)rXrYr0r1)r0r/r0r1) rDrErFrG__annotations__propertyrrWsetterr_rfrHr;r9rJrJNsMMNNN))))  XX_mmm_m ---X-((((((r;rJrWstr | httpx.URL | NonerbrKrcr<rLallow_deferred_projectrPr0cv||S|J||rdStdt|||S)a base_url (arg or ``ANTHROPIC_GOOGLE_CLOUD_BASE_URL``, resolved by the caller) > derived template. Returns ``None`` when derivation must wait for the project to be back-filled from Google credentials on the first request (``allow_deferred_project``). NzNo `project` was provided. Pass `project`, set the `ANTHROPIC_GOOGLE_CLOUD_PROJECT` environment variable, or provide `base_url` directly.ra) ValueErrorrdrerWrbrcrLrls r9_resolve_base_urlrpse  # # # ! 4 D    & &wXd & e eer; skip_authtoken_provider object | Noner.r1c4|r||tddSdS)Nzx`skip_auth` is mutually exclusive with `token_provider` and `credentials`; `skip_auth` disables authentication entirely.)rnrqrrr.s r9_reject_skip_auth_conflictrvs=  n0K4K <     4K4Kr;GoogleCredentialschdD].}t||d}t|tr|r|cS/dS)uBest-effort project from an explicit credentials object — service-account / impersonated credentials usually know theirs.) project_idquota_project_idN)getattrrBr<)r.attrvalues r9_project_from_credentialsr~sL3 T400 eS ! ! e LLL 4r;c*eZdZUdZded<ded<dddddddeeddddddd>fd!Zee d?d"Z e e d@d$Z e dAd(Z dBd*Ze dCd-Ze dDd0Zddeeeddedeeddddid1dEd=ZeZxZS)FAnthropicGoogleCloudagSynchronous client for the first-party Anthropic API served through Google's gateway (Claude Platform on Google Cloud). The whole first-party surface is proxied verbatim (no URL or body rewriting), so this subclasses the full ``Anthropic`` client. Authentication is a GCP bearer token; the deprecated Completions endpoint is not exposed. rKrLrP _skip_authNFrbrcrLrrr.rqrWtimeout max_retriesdefault_headers default_query http_client middleware_strict_response_validationrbrcrrTokenProvider | Noner.r/rqrWrkr!float | Timeout | None | NotGivenrintrMapping[str, str] | NonerMapping[str, object] | Nonerhttpx.Client | Noner Sequence[MiddlewareInput] | Nonerr0r1c (t|||||_||_t||_|t jd}|t}||_ |t jd}|t jd}||t|}||_ |t jd}|du|_ |t jd}||r|td||_t||j |j |j| }|du|_t#||nd || | | | | | d|_d|_dS) uConstruct a new synchronous Claude Platform on Google Cloud client. Auth precedence (first match wins, unless ``skip_auth=True``): 1. ``token_provider`` — a callable returning a GCP access token, invoked per request. 2. ``credentials`` — a ``google.auth`` Credentials object, refreshed as needed. 3. Application Default Credentials (``google.auth.default``). Args: project: GCP consumer project id (or ``ANTHROPIC_GOOGLE_CLOUD_PROJECT``, else ``GOOGLE_CLOUD_PROJECT``). Only needed when the base URL must be derived; if omitted there, it is taken from an explicit ``credentials`` object when it exposes one, or back-filled from ADC on the first request. location: GCP location (or ``ANTHROPIC_GOOGLE_CLOUD_LOCATION``). Optional — defaults to ``global``, the region the gateway should normally be addressed through. workspace_id: The Anthropic workspace ID (or ``ANTHROPIC_GOOGLE_CLOUD_WORKSPACE_ID``). Required unless ``skip_auth`` is set with an explicit ``base_url``. skip_auth: For pre-authenticated proxies — skips token attachment. A workspace ID is still needed to derive the base URL; pass ``base_url`` to construct without one. Mutually exclusive with the credential arguments. ruNANTHROPIC_GOOGLE_CLOUD_LOCATIONANTHROPIC_GOOGLE_CLOUD_PROJECTGOOGLE_CLOUD_PROJECTANTHROPIC_GOOGLE_CLOUD_BASE_URL#ANTHROPIC_GOOGLE_CLOUD_WORKSPACE_IDyNo workspace ID found. Set the `workspace_id` argument or the `ANTHROPIC_GOOGLE_CLOUD_WORKSPACE_ID` environment variable.rorWrrrrrrrrvr_token_providerr-rOosenvirongetDEFAULT_LOCATIONrNr~rMrRrnrLrprQsuperr:api_key auth_tokenr8rbrcrLrrr.rqrWrrrrrrrresolved_base_url __class__s r9r:zAnthropicGoogleCloud.__init__sP #Y~cnoooo#-3K@@  z~~&GHHH  'H! ?jnn%EFFG ?jnn%;<>*OPPL   x7KN )-M^*'0=    #4t"; +<*G&&R#+'#!(C  " r;cdSzQCompletions endpoint is deprecated and not supported for the Google Cloud client.NrHrVs r9 completionsz AnthropicGoogleCloud.completions2 tr;dict[str, str]ciSr3rHrVs r9 auth_headersz!AnthropicGoogleCloud.auth_headers8s  r;headersrcustom_headerscdSr3rHr8rrs r9_validate_headersz&AnthropicGoogleCloud._validate_headers?s  r;r<c|j}|V|}tj|r6tt|t d|S|jS)Nzp`token_provider` returned an awaitable. Async token providers are only supported on `AsyncAnthropicGoogleCloud`.) rinspect isawaitabler rcloserrOr@r8providerr@s r9 _get_tokenzAnthropicGoogleCloud._get_tokenEs~'  HJJE"5)) S%  &&((($@L &&(((r;optionsrc|jsS|jrL|j1|j*|j|jj|_||Sr3)rrQrMrrOr@r7rfr8rs r9_prepare_optionsz%AnthropicGoogleCloud._prepare_optionsSsa .4#: .}$)=)E!''))) $ 1 =  + + - - -r;request httpx.Requestc|jrdS|jddSd||jd<dSN AuthorizationzBearer rrrrr8rs r9_prepare_requestz%AnthropicGoogleCloud._prepare_request`sR ?  F ?   / / ; F+HT__5F5F+H+H(((r;rbrcrLrrr.rqrWrrrrrset_default_headersrset_default_query _extra_kwargsstr | None | NotGivenTokenProvider | None | NotGiven#GoogleCredentials | None | NotGiven bool | None+Sequence[MiddlewareInput] | None | NotGivenint | NotGivenrrrMapping[str, Any]rc| | td||td|j}| t|| }n| | }|j}|i||}n||}||n|j}t |pt |}d}d}|r't |r|nd}t |r|nd}n|s|j}|jj}| |j sd}n ||n|j }|j d||n|j ||n|j t |r|n|j||||t|t r|jn|| p|jt | r| n|j||t| t r|jn| |jd|}|s|s |j|_|S)uCreate a new client re-using the current options, with optional overrides. Passing either of ``token_provider`` / ``credentials`` replaces the inherited credential configuration wholesale — the source not passed is cleared, so an explicit lower-precedence credential takes effect. ``workspace_id=None`` clears the workspace ID; ``project`` / ``location`` overrides re-derive a template-derived base URL. NPThe `default_headers` and `set_default_headers` arguments are mutually exclusiveLThe `default_query` and `set_default_query` arguments are mutually exclusiverbrcrLrrr.rqrWrrrrrrrrHrn_custom_headersr! _custom_queryrrrrOr.rRrWrrMrNrLrBrr_clientr _middlewarerr8rbrcrLrrr.rqrWrrrrrrrrrrparamsresolved_skip_authcredential_overriddennew_token_providernew_credentials new_base_urlclients r9copyzAnthropicGoogleCloud.copyms*8  &+>+Jopp p  $):)Fkll l&  &#G_==GG ,)G#  $00-0FF  *&F*3*?YYT_ ( 8 8 QH[)1,)?)?VTEV-'(!$.w$A$ANDLLw#3t|'/ '<'<R $BR# +5j(+K+K[t''Q[(,(H    "% 4-? 4#'"3F  r;)rbrKrcrKrLrKrrrr.r/rqrPrWrkrrrrrrrrrrrrrrPr0r1rgr0rrrrrr0r1rCrrr0rrrr0r1)"rbrKrcrKrLrrrrr.rrqrrWrkrrrrrrrrrrrrrrrrrrr0rrDrErFrGrhrrr:rrrrirrrrrr with_options __classcell__rs@r9rrs ###'/304+/5>.4859+/7;,1!ggggggggR X_ XX X ) ) ) )   X  I I IX I##.7:C;D!%+/5>+/BK&/488<599=+-%UUUUUUnLLLLLr;rc*eZdZUdZded<ded<dddddddeeddddddd>fd!Zee d?d"Z e e d@d$Z e dAd(Z dBd*Ze dCd-Ze dDd0Zddeeeddedeeddddid1dEd=ZeZxZS)FAsyncAnthropicGoogleCloudzAsynchronous client for the first-party Anthropic API served through Google's gateway (Claude Platform on Google Cloud). See ``AnthropicGoogleCloud``. rKrLrPrNFrrbrcrrAsyncTokenProvider | Noner.r/rqrWrkrrrrrrrrrhttpx.AsyncClient | Nonerrrr0r1c (t|||||_||_t||_|t jd}|t}||_ |t jd}|t jd}||t|}||_ |t jd}|du|_ |t jd}||r|td||_t||j |j |j| }|du|_t#||nd || | | | | | d|_d|_dS) aConstruct a new asynchronous Claude Platform on Google Cloud client. ``token_provider`` may be sync or async; sync providers are run off the event loop. See ``AnthropicGoogleCloud`` for the full argument and auth-precedence docs. ruNrrrrrrrorrrrs r9r:z"AsyncAnthropicGoogleCloud.__init__s0 #Y~cnoooo#-3K@@  z~~&GHHH  'H! ?jnn%EFFG ?jnn%;<>*OPPL   x7KN )-M^*'0=    #4t"; +<*G&&R#+'#!(C   r;cdSrrHrVs r9rz%AsyncAnthropicGoogleCloud.completions"rr;rciSr3rHrVs r9rz&AsyncAnthropicGoogleCloud.auth_headers(s  r;rrrcdSr3rHrs r9rz+AsyncAnthropicGoogleCloud._validate_headers-sr;r<cK|j}|tj|r)|d{V}t|tsJ|St |d{V}tj|r|d{V}t|tsJ|St |jjd{VSr3) rriscoroutinefunctionrBr<rrrOr@rs r9rz$AsyncAnthropicGoogleCloud._get_token1s'  *844 &hjj((((((!%----- -(8,,........E"5)) $# eS)) ) ))L6Xd/566888888888r;rrcK|jsa|jrZ|j?|j8t |jjd{V|jj|_||Sr3) rrQrMrrrOr@r7rfrs r9rz*AsyncAnthropicGoogleCloud._prepare_optionsBs~ .4#: .}$)=)E8ht0677999999999 $ 1 =  + + - - -r;rrcK|jrdS|jddSd|d{V|jd<dSrrrs r9rz*AsyncAnthropicGoogleCloud._prepare_requestOsh ?  F ?   / / ; F+N4??;L;L5L5L5L5L5L5L+N+N(((r;rr$AsyncTokenProvider | None | NotGivenrrrrrrrrrc| | td||td|j}| t|| }n| | }|j}|i||}n||}||n|j}t |pt |}d}d}|r't |r|nd}t |r|nd}n|s|j}|jj}| |j sd}n ||n|j }|j d||n|j ||n|j t |r|n|j||||t|t r|jn|| p|jt | r| n|j||t| t r|jn| |jd|}|s|s |j|_|S)zCreate a new client re-using the current options, with optional overrides. See ``AnthropicGoogleCloud.copy`` for the override semantics. NrrrrHrrs r9rzAsyncAnthropicGoogleCloud.copy\s*0  &+>+Jopp p  $):)Fkll l&  &#G_==GG ,)G#  $00-0FF  *&F*3*?YYT_ ( 8 8 QH[)1,)?)?VTEV-'(!$.w$A$ANDLLw#3t|'/ '<'<R $BR# +5j(+K+K[t''Q[(,(H    "% 4-? 4#'"3F  r;)rbrKrcrKrLrKrrrr.r/rqrPrWrkrrrrrrrrrrrrrrPr0r1rgrrrCrr)"rbrKrcrKrLrrrrr.rrqrrWrkrrrrrrrrrrrrrrrrrrr0rrrs@r9rrs ###'4804+/5>.4859047;,1!QQQQQQQQf X_ XXX9999"   X  O O OX O##.7?H;D!%+/5>04BK&/488<599=+-%QQQQQQfLLLLLr;r) rWrkrbrKrcr<rLrKrlrPr0rk)rqrPrrrsr.rsr0r1)r.rwr0rK)F __future__rrrr4typingrrrrrr r r r functoolsr rtyping_extensionsrrr\_typesrrrr_utilsrrrrr_modelsr _streamingrr _exceptionsrrr _base_clientrr r!_extras._google_authr#r$r>rAgoogle.auth.credentialsr'rwrdrr< TokenProviderAsyncTokenProviderClient AsyncClientr)r+r-rJrprvr~rrrHr;r9rs"""""" cccccccccccccccccccccc........,,,,,,,, ;;;;;;;;;;;;((((((((00000000******--------))))))******JJJJJJJJJJPPPPPPPP  8OOOw2.IIIIHHHHHH u S! b"889uU\5CT5T/UVVV '+5kRUFV9V3WXXX>4(4(4(4(4(J}o'EF4(4(4(nffff<     GGGGG0vc{1JKYGGGThhhhh 5e6GUXIY6Y Z\jhhhhhr;