
    7jp>                        d dl mZ d dlmZmZmZ d dlmZmZ d dl	Z	ddl
mZmZmZmZmZ ddlmZmZ dd	lmZmZmZmZmZmZ dd
lmZ ddlmZ ddlmZ  G d de      Z  G d de      Z!y)    )annotations)AnyMappingSequence)SelfoverrideN   )	NOT_GIVENOmitHeadersTimeoutNotGiven)	AnthropicAsyncAnthropic   )resolve_regionresolve_api_keyresolve_base_urlresolve_auth_moderesolve_workspace_idvalidate_credentials)AnthropicError)MiddlewareInput)DEFAULT_MAX_RETRIESc                      e Zd ZU ded<   ded<   ded<   ded<   ded<   ded<   ded	<   ded
<   dddddddddeedddddddd	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d fdZeed fd              Z	eed fd              Z
ed fd       Zedd       Zedddddddddddededdddei d	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d fd       ZeZ xZS )AnthropicAWS
str | Noneaws_access_keyaws_secret_key
aws_regionaws_profileaws_session_tokenworkspace_idbool
_use_sigv4
_skip_authNFapi_keyr   r   r    r!   r"   r#   	skip_authbase_urltimeoutmax_retriesdefault_headersdefault_queryhttp_client
middleware_strict_response_validation
auth_tokenwebhook_keyc                  || _         t        ||       |r
d| _        d }n+t        ||||      | _        t	        || j                        }t        |      }| j                  r|t        d      || _        || _        || _	        || _
        || _        |r|| _        nt        |      }|t        d      || _        |s)t        |	t        |	      nd |      }|t        d      |}	t         | E  ||||	|
||||||	       y 
N)r   r   F)r(   r   r   r!   )r(   	use_sigv4zxNo AWS region was provided. Set the `aws_region` argument or the `AWS_REGION`/`AWS_DEFAULT_REGION` environment variable.zpNo workspace ID found. Set the `workspace_id` argument or the `ANTHROPIC_AWS_WORKSPACE_ID` environment variable.)regionzNo AWS region was provided and no base_url was given. Set the `aws_region` argument, the `AWS_REGION`/`AWS_DEFAULT_REGION` environment variable, or provide a `base_url` directly.)r(   r2   r3   r*   r+   r,   r-   r.   r/   r0   r1   r&   r   r%   r   r   r   r   r   r   r    r!   r"   r#   r   r   strsuper__init__selfr(   r   r   r    r!   r"   r#   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   resolved_api_keyresolved_regionresolved_workspace_idresolved_base_url	__class__s                          O/var/www/html/qr/venv/lib/python3.12/site-packages/anthropic/lib/aws/_client.pyr;   zAnthropicAWS.__init__!   O   . $N>Z#DO#/--'	DO  /w$//Z(4??6  K  -,)&!2 ,D$8$F!$,$ G  !6D 0!)!5H4&! !($8 
 )H$!##+'#!(C 	 	
    c                V    i t         |   }| j                  | j                  |d<   |S Nzanthropic-workspace-idr:   r-   r#   r=   headersrB   s     rC   r-   zAnthropicAWS.default_headersz   6     .UW,-(040A0AG,-rE   c                L    | j                   s| j                  ri S t        |   S Nr%   r&   r:   _api_key_authr=   rB   s    rC   rO   zAnthropicAWS._api_key_auth   "     ??dooIw$$rE   c                X    | j                   s| j                  ry t        |   ||       y rM   r%   r&   r:   _validate_headersr=   rJ   custom_headersrB   s      rC   rT   zAnthropicAWS._validate_headers   "    ??doo!'>:rE   c                r   | j                   sy ddlm} |j                         j	                         } ||j
                  t        |j                        |j                  | j                  | j                  | j                  | j                  | j                  |d
      }|j                  j                  |       y Nr   )get_auth_headerszaws-external-anthropic)
methodurlrJ   r   r   r"   r7   profiledataservice_namer%   _authrZ   readdecoder[   r9   r\   rJ   r   r   r"   r    r!   updater=   requestrZ   r^   rJ   s        rC   _prepare_requestzAnthropicAWS._prepare_request   s    +||~$$&">>GKK OO...."44??$$1
 	w'rE   r(   r   r   r    r!   r"   r#   r)   r2   r3   r*   r+   r/   r,   r-   set_default_headersr.   set_default_queryr0   _extra_kwargsc               X   |xs |rd n| j                   }t        | 	  |xs | j                  |	|
||||||||||xs | j                  |xs | j
                  |xs | j                  |xs | j                  |xs | j                  |xs | j                  ||n| j                  d|      S N)r   r   r    r!   r"   r#   r)   )r(   r2   r3   r*   r+   r/   r,   r-   ri   r.   rj   r0   rk   r*   r:   copyr(   r   r   r    r!   r"   r#   r&   r=   r(   r   r   r    r!   r"   r#   r)   r2   r3   r*   r+   r/   r,   r-   ri   r.   rj   r0   rk   rA   rB   s                         rC   ro   zAnthropicAWS.copy       4 %Ow|+t||!#&##+ 3'/!"0"GD4G4G"0"GD4G4G(;DOO*>d.>.>%6%P$:P:P , A0A0A*3*?YT__	  	  
 	
rE   )&r(   r   r   r   r   r   r    r   r!   r   r"   r   r#   r   r)   r$   r*   str | httpx.URL | Noner+   'float | httpx.Timeout | None | NotGivenr,   intr-   Mapping[str, str] | Noner.   Mapping[str, object] | Noner/   httpx.Client | Noner0    Sequence[MiddlewareInput] | Noner1   r$   r2   r   r3   r   returnNonery   zdict[str, str | Omit]ry   zdict[str, str]rJ   r   rV   r   ry   rz   rf   zhttpx.Requestry   rz   )*r(   r   r   r   r   r   r    r   r!   r   r"   r   r#   r   r)   bool | Noner2   r   r3   r   r*   rr   r+   !float | Timeout | None | NotGivenr/   rw   r,   int | NotGivenr-   ru   ri   ru   r.   rv   rj   rv   r0   +Sequence[MiddlewareInput] | None | NotGivenrk   Mapping[str, Any]ry   r   __name__
__module____qualname____annotations__r
   r   r;   propertyr   r-   rO   rT   rg   ro   with_options__classcell__rB   s   @rC   r   r      s   !!
 #%)%)!%"&(,#'+/;D.4859+/7;,1!%"&+W
 W
 #	W

 #W
 W
  W
 &W
 !W
 W
 )W
 9W
 W
 2W
 3W
  )!W
" 5#W
$ &*%W
( )W
*  +W
, 
-W
r    %  %
 ; ;
 ( (,  #%)%)!%"&(,#'!%!%"&+/5>+/&/488<599=BK+--2
 2
 #	2

 #2
 2
  2
 &2
 !2
 2
 2
  2
 )2
 32
 )2
  $!2
" 2#2
$ 6%2
& 3'2
( 7)2
* @+2
, )-2
. 
/2
 2
h LrE   r   c                      e Zd ZU ded<   ded<   ded<   ded<   ded<   ded<   ded	<   ded
<   dddddddddeedddddddd	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d fdZeed fd              Z	eed fd              Z
ed fd       Zedd       Zedddddddddddededdddei d	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d fd       ZeZ xZS )AsyncAnthropicAWSr   r   r   r    r!   r"   r#   r$   r%   r&   NFr'   c                  || _         t        ||       |r
d| _        d }n+t        ||||      | _        t	        || j                        }t        |      }| j                  r|t        d      || _        || _        || _	        || _
        || _        |r|| _        nt        |      }|t        d      || _        |s)t        |	t        |	      nd |      }|t        d      |}	t         | E  ||||	|
||||||	       y r5   r8   r<   s                          rC   r;   zAsyncAnthropicAWS.__init__   rD   rE   c                V    i t         |   }| j                  | j                  |d<   |S rG   rH   rI   s     rC   r-   z!AsyncAnthropicAWS.default_headersA  rK   rE   c                L    | j                   s| j                  ri S t        |   S rM   rN   rP   s    rC   rO   zAsyncAnthropicAWS._api_key_authI  rQ   rE   c                X    | j                   s| j                  ry t        |   ||       y rM   rS   rU   s      rC   rT   z#AsyncAnthropicAWS._validate_headersP  rW   rE   c                z  K   | j                   sy ddlm} |j                         j	                         } ||j
                  t        |j                        |j                  | j                  | j                  | j                  | j                  | j                  |d
      }|j                  j                  |       y wrY   r`   re   s        rC   rg   z"AsyncAnthropicAWS._prepare_requestV  s     +||~$$&">>GKK OO...."44??$$1
 	w's   B9B;rh   c               X   |xs |rd n| j                   }t        | 	  |xs | j                  |	|
||||||||||xs | j                  |xs | j
                  |xs | j                  |xs | j                  |xs | j                  |xs | j                  ||n| j                  d|      S rm   rn   rp   s                         rC   ro   zAsyncAnthropicAWS.copym  rq   rE   )&r(   r   r   r   r   r   r    r   r!   r   r"   r   r#   r   r)   r$   r*   rr   r+   rs   r,   rt   r-   ru   r.   rv   r/   httpx.AsyncClient | Noner0   rx   r1   r$   r2   r   r3   r   ry   rz   r{   r|   r}   r~   )*r(   r   r   r   r   r   r    r   r!   r   r"   r   r#   r   r)   r   r2   r   r3   r   r*   rr   r+   r   r/   r   r,   r   r-   ru   ri   ru   r.   rv   rj   rv   r0   r   rk   r   ry   r   r   r   s   @rC   r   r      s   !!
 #%)%)!%"&(,#'+/;D.4859047;,1!%"&+W
 W
 #	W

 #W
 W
  W
 &W
 !W
 W
 )W
 9W
 W
 2W
 3W
  .!W
" 5#W
$ &*%W
( )W
*  +W
, 
-W
r    %  %
 ; ;
 ( (,  #%)%)!%"&(,#'!%!%"&+/5>04&/488<599=BK+--2
 2
 #	2

 #2
 2
  2
 &2
 !2
 2
 2
  2
 )2
 32
 .2
  $!2
" 2#2
$ 6%2
& 3'2
( 7)2
* @+2
, )-2
. 
/2
 2
h LrE   r   )"
__future__r   typingr   r   r   typing_extensionsr   r   httpx_typesr
   r   r   r   r   _clientr   r   _credentialsr   r   r   r   r   r   _exceptionsr   _middlewarer   _base_clientr   r   r    rE   rC   <module>r      sO    " ) ) ,  A A 0  * * /D9 DND DrE   