o
    i%                 	   @  st  U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlZd dlmZ d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dl m!Z!m"Z"m#Z# d dl$Z$d dl%Z%d dl&Z&d dl'Z'd dl%m(Z( d dl'm)Z) d	d
l*m+Z+ d	dl,m-Z- d	dl.m/Z/m0Z0 d	dl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZB d	dlCmDZDmEZEmFZFmGZGmHZHmIZImJZJ d	dlKmLZLmMZMmNZN d	dlOmPZPmQZQmRZRmSZS d	dlTmUZUmVZVmWZWmXZX d	dlYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z` d	dlambZbmcZcmdZdmeZe d	dl+mfZfmgZgmhZhmiZi d	dljmkZk d	dllmmZm e	neoZpdeqd< epreD  edddZsedddZtedZuedd d!Zved"ebe dZwed#ede dZxerld d$lymzZz ezZ{nzd d$lymzZ{ W n e|y   e6d%Z{Y nw G d&d' d'Z}G d(d) d)ePeeu Z~G d*d+ d+e~eu eeu ZG d,d- d-eeuetf ZG d.d/ d/e~eu eeu Zed0ee%je%jf dZed1eebe ede f dZG d2d3 d3eeef ZG d4d5 d5e%jZere%jZneZG d6d7 d7eZG d8d9 d9ee%jebe f ZG d:d; d;e%jZzd dlZW n e|y,   G d<d= d=e%jZY n
w G d>d= d=ejZerAe%jZ	 e%jZneZeZG d?d@ d@eZG dAdB dBee%jede f ZdddddeBeBddCdwdUdVZG dWdX dXeedf ZG dYdZ dZZeee!d[ f Zdxd]d^ZeIdd_dydedfZG dgdh dhZeee!di f ZdzdjdkZdzdldmZd{dodpZd|dudvZdS )}    )annotationsN)TracebackType)random)TYPE_CHECKINGAnyDictTypeUnionGenericMappingTypeVarIterableIteratorOptional	GeneratorAsyncIteratorcastoverload)Literaloverride
get_origin)URL)PrivateAttr   )_exceptionsQuerystring)to_httpx_filesasync_to_httpx_files)BodyOmitQueryHeadersTimeoutNotGiven	ResponseT
AnyMapping
PostParserBinaryTypesRequestFilesHttpxSendArgsRequestOptionsAsyncBinaryTypesHttpxRequestFilesModelBuilderProtocol	not_given)SensitiveHeadersFilteris_dictis_listasyncifyis_given	lru_cache
is_mapping)PYDANTIC_V1
model_copy
model_dump)GenericModelFinalRequestOptionsvalidate_typeconstruct_type)APIResponseBaseAPIResponseAsyncAPIResponseextract_response_type)DEFAULT_TIMEOUTMAX_RETRY_DELAYDEFAULT_MAX_RETRIESINITIAL_RETRY_DELAYRAW_RESPONSE_HEADEROVERRIDE_CAST_TO_HEADERDEFAULT_CONNECTION_LIMITS)Stream
SSEDecoderAsyncStreamSSEBytesDecoder)APIStatusErrorAPITimeoutErrorAPIConnectionErrorAPIResponseValidationError)openapi_dumps)LegacyAPIResponsezlogging.Loggerlog	SyncPageTzBaseSyncPage[Any])bound
AsyncPageTzBaseAsyncPage[Any]_T_T_coT)	covariant_StreamT_AsyncStreamT)DEFAULT_TIMEOUT_CONFIGg      @c                   @  sv   e Zd ZU dZded< ded< ded< edddZedddZedddZeeeddddZedddZ	dS )PageInfozStores the necessary information to build the request to retrieve the next page.

    Either `url` or `params` must be set.
    URL | NotGivenurlQuery | NotGivenparamsBody | NotGivenjsonr   returnNonec                C     d S N selfr_   rh   rh   M/home/kim/smarthome/.venv/lib/python3.10/site-packages/openai/_base_client.py__init__      zPageInfo.__init__r!   c                C  rf   rg   rh   )rj   ra   rh   rh   rk   rl      rm   r   c                C  rf   rg   rh   )rj   rc   rh   rh   rk   rl      rm   r_   rc   ra   c                C  s   || _ || _|| _d S rg   rn   )rj   r_   rc   ra   rh   rh   rk   rl      s   
strc                 C  sN   | j r| jj d| j  dS | jr| jj d| j dS | jj d| j dS )Nz(url=)z(json=z(params=)r_   	__class____name__rc   ra   rj   rh   rh   rk   __repr__   s
   zPageInfo.__repr__N)r_   r   rd   re   )ra   r!   rd   re   )rc   r   rd   re   )r_   r^   rc   rb   ra   r`   rd   re   rd   ro   )
rr   
__module____qualname____doc____annotations__r   rl   r/   r   rt   rh   rh   rh   rk   r]   x   s"   
 	r]   c                   @  s`   e Zd ZU dZe Zded< e Zded< ddd	ZdddZ	dddZ
dddZdddZdS )BasePagea<  
    Defines the core interface for pagination.

    Type Args:
        ModelT: The pydantic model that represents an item in the response.

    Methods:
        has_next_page(): Check if there is another page available
        next_page_info(): Get the necessary information to make a request for the next page
    r;   _optionsType[_T]_modelrd   boolc                 C  s   |   }|sdS |  d uS )NF)_get_page_itemsnext_page_info)rj   itemsrh   rh   rk   has_next_page   s   zBasePage.has_next_pageOptional[PageInfo]c                 C  rf   rg   rh   rs   rh   rh   rk   r      s    zBasePage.next_page_infoIterable[_T]c                 C  rf   rg   rh   rs   rh   rh   rk   r         zBasePage._get_page_itemsr_   r   httpx.QueryParamsc                 C  s   t tt| jj|jS rg   )httpxZQueryParamsr   r   r{   ra   mergeri   rh   rh   rk   _params_from_url   s   zBasePage._params_from_urlinfor]   c                 C  s   t | j}|  t|jtsi |j|j|_|S t|jts:| |j}|jj|d}t	|j|_t
||_|S t|jtsht|jsItd|jsTi |j|_|S t|js]tdi |j|j|_|S td)N)ra   z*Pagination is only supported with mappingszUnexpected PageInfo state)r8   r{   Z_strip_raw_response_header
isinstancera   r$   r_   r   	copy_withdictro   rc   r6   	TypeError	json_data
ValueError)rj   r   optionsra   r_   rh   rh   rk   _info_to_options   s,   



zBasePage._info_to_optionsNrd   r~   )rd   r   )rd   r   )r_   r   rd   r   )r   r]   rd   r;   )rr   rv   rw   rx   r   r{   ry   r}   r   r   r   r   r   rh   rh   rh   rk   rz      s   
 



rz   c                   @  F   e Zd ZU e Zded< dd
dZdddZdddZ	dddZ
dS )BaseSyncPageSyncAPIClient_clientclientmodelr|   r   r;   rd   re   c                 C  s0   t st| dd d u ri | _|| _|| _|| _d S N__pydantic_private__r7   getattrr   r}   r   r{   )rj   r   r   r   rh   rh   rk   _set_private_attributes   
   
z$BaseSyncPage._set_private_attributesIterator[_T]c                 c  s(    |   D ]}| D ]}|V  qqd S rg   
iter_pagesr   rj   pageitemrh   rh   rk   __iter__   s   zBaseSyncPage.__iter__rj   rT   Iterator[SyncPageT]c                 c  s&    | }	 |V  |  r| }nd S qrg   r   get_next_pagerj   r   rh   rh   rk   r     s   
zBaseSyncPage.iter_pagesc                 C  s4   |   }|s
td| |}| jj| j| j|dS NzYNo next page expected; please check `.has_next_page()` before calling `.get_next_page()`.)r   r   r   RuntimeErrorr   r   _request_api_listr}   rq   rj   r   r   rh   rh   rk   r     s   
zBaseSyncPage.get_next_pageN)r   r   r   r|   r   r;   rd   re   )rd   r   )rj   rT   rd   r   )rj   rT   rd   rT   )rr   rv   rw   pydanticr   r   ry   r   r   r   r   rh   rh   rh   rk   r      s   
 


	r   c                   @  s4   e Zd ZdddZdddZdddZdddZdS )AsyncPaginatorr   AsyncAPIClientr   r;   page_clsType[AsyncPageT]r   r|   rd   re   c                 C  s   || _ || _|| _|| _d S rg   )r}   r   r{   	_page_cls)rj   r   r   r   r   rh   rh   rk   rl     s   
zAsyncPaginator.__init__ Generator[Any, None, AsyncPageT]c                 C  s   |    S rg   )	_get_page	__await__rs   rh   rh   rk   r   &  s   zAsyncPaginator.__await__rV   c                   s0   d fdd}| j _ j j j I d H S )NresprV   rd   c                   s   | j  j j jd | S )N)r   r   r   )r   r}   r{   r   r   rs   rh   rk   _parser*  s   z)AsyncPaginator._get_page.<locals>._parser)r   rV   rd   rV   )r{   post_parserr   requestr   )rj   r   rh   rs   rk   r   )  s   zAsyncPaginator._get_pageAsyncIterator[_T]c                 C s0   t t| I d H }|2 z	3 d H W }|V  q6 d S rg   )r   rV   r   rh   rh   rk   	__aiter__6  s   zAsyncPaginator.__aiter__N)
r   r   r   r;   r   r   r   r|   rd   re   )rd   r   )rd   rV   rd   r   )rr   rv   rw   rl   r   r   r   rh   rh   rh   rk   r     s
    


r   c                   @  r   )BaseAsyncPager   r   r   r|   r   r   r;   rd   re   c                 C  s0   t st| dd d u ri | _|| _|| _|| _d S r   r   )rj   r   r   r   rh   rh   rk   r   C  r   z%BaseAsyncPage._set_private_attributesr   c                 C s2   |   2 z3 d H W }| D ]}|V  qq6 d S rg   r   r   rh   rh   rk   r   P  s   zBaseAsyncPage.__aiter__rj   rV   AsyncIterator[AsyncPageT]c                 C s,   | }	 |V  |  r| I d H }nd S qrg   r   r   rh   rh   rk   r   U  s   zBaseAsyncPage.iter_pagesc                   s<   |   }|std| |}| jj| j| j|dI d H S r   r   r   rh   rh   rk   r   ^  s   
zBaseAsyncPage.get_next_pageN)r   r|   r   r   r   r;   rd   re   r   )rj   rV   rd   r   )rj   rV   rd   rV   )rr   rv   rw   r   r   r   ry   r   r   r   r   rh   rh   rh   rk   r   @  s   
 


	r   _HttpxClientT_DefaultStreamTc                   @  s  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Zded< eeddddtddZdud d!Zdvd%d&Z	dwd+d,Z
d-d.dxd3d4Zdyd5d6Zdzd8d9Zd-d.d{d;d<Zd|d@dAZd}dDdEZd~dGdHZddJdKZeddMdNZeddPdQZeddSdTZeddVdWZeddXdYZdd\d]Zedd^d_Zedd`daZejddcdaZddedfZdddjdkZ	dddndoZddpdqZ ddrdsZ!dS )
BaseClientr   r   ro   _versionr   	_base_urlintmax_retrieszUnion[float, Timeout, None]timeoutr~   _strict_response_validation
str | None_idempotency_headerNztype[_DefaultStreamT] | None_default_stream_cls)r   r   custom_headerscustom_queryversionbase_url	str | URLfloat | Timeout | Noner   Mapping[str, str] | Noner   Mapping[str, object] | Nonerd   re   c                C  s\   || _ | t|| _|| _|| _|pi | _|pi | _|| _d | _	d | _
|d u r,tdd S )Nzmax_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `openai.DEFAULT_MAX_RETRIES`)r   _enforce_trailing_slashr   r   r   r   _custom_headers_custom_queryr   r   	_platformr   )rj   r   r   r   r   r   r   r   rh   rh   rk   rl   w  s   

zBaseClient.__init__r_   c                 C  s"   |j dr|S |j|j d dS )N   /raw_path)r   endswithr   ri   rh   rh   rk   r     s   z"BaseClient._enforce_trailing_slashresponsehttpx.ResponserM   c                 C  s   |j r|jsd }d|j }n)|j }|}zt|}d|j d| }W n ty7   |p4d|j }Y nw | j|||dS )NzError code: z - )bodyr   )		is_closedZis_stream_consumedstatus_codetextstriprc   loads	Exception_make_status_error)rj   r   r   err_msgZerr_textrh   rh   rk    _make_status_error_from_response  s   

z+BaseClient._make_status_error_from_responser   r   object_exceptions.APIStatusErrorc                C  s   t  rg   )NotImplementedError)rj   r   r   r   rh   rh   rk   r     s   zBaseClient._make_status_errorr   retries_takenr   r;   r   httpx.Headersc          	      C  s   |j pi }t| j|}| || t|}| j}|r'|jr'||vr'|j||< dd |D }d|vr8t||d< d|vrZt	|j
trE| j
n|j
}t	|trP|j}|d urZt||d< |S )Nc                 S  s   g | ]}|  qS rh   )lower).0headerrh   rh   rk   
<listcomp>  s    z-BaseClient._build_headers.<locals>.<listcomp>zx-stainless-retry-countzx-stainless-read-timeout)headers_merge_mappingsdefault_headers_validate_headersr   r"   r   idempotency_keyro   r   r   r$   r#   read)	rj   r   r   r   Zheaders_dictr   Zidempotency_headerZlower_custom_headersr   rh   rh   rk   _build_headers  s"   



zBaseClient._build_headersc                 C  s4   t |}|jr| jj|jd }| jj|dS |S )z
        Merge a URL argument together with any 'base_url' on the client,
        to create the URL used for the outgoing request.
        r   r   )r   Zis_relative_urlr   r   lstripr   )rj   r_   Z	merge_urlZmerge_raw_pathrh   rh   rk   _prepare_url  s
   zBaseClient._prepare_urlSSEDecoder | SSEBytesDecoderc                 C     t  S rg   )rJ   rs   rh   rh   rk   _make_sse_decoder  s   zBaseClient._make_sse_decoderhttpx.Requestc                C  sF  t tjrt dt|dtrdhni d i }|j}|jd ur@|d u r+t	t
|j}nt|r6t||j}n
tdt| d| j||d}t| j|j}|d}|j}|d ur|d	rd
|vrh|d |rt|sxtdt| d| ||d< |st	tt }| |j}	d|	jv rd|	jddi|d< |j  dk}
|
r|j!d ur|d urtd|j!d ur|d urtd|j!d ur|j!|d< nt"|t#r||d< n|st$|r|d urt%|nd |d< ||d< n|dd  |dd  | j&j'd|t"|j(t)r| j(n|j(|j|	|r| j*+t	t,t-t.f |nd d|S )NzRequest options: %sTcontent)Zexclude_unsetexcludezUnexpected JSON data type, z , cannot merge with `extra_body`r   Content-Typezmultipart/form-databoundaryzGExpected query input to be a dictionary for multipart requests but got z	 instead.data_Zsni_hostname-
extensionsgetz7Passing both `content` and `json_data` is not supportedz3Passing both `content` and `files` is not supportedfiles)r   r   methodr_   ra   rh   )/rS   isEnabledForloggingDEBUGdebugr9   r7   r   
extra_jsonr   r   r6   r   r   typer   default_queryra   r
  r  
startswithpopr1   r   _serialize_multipartformr-   ForceMultipartDictr   r_   hostreplacer  r   r  r   bytesr4   rQ   r   Zbuild_requestr   r$   qs	stringifyr   ro   r   )rj   r   r   kwargsr   r   ra   content_typer  Zprepared_urlZis_body_allowedrh   rh   rk   _build_request  s~   






 
"
zBaseClient._build_requestr  Mapping[object, object]dict[str, object]c                 C  s^   | j j|dd}i }|D ] \}}||}|s|||< qt|r&|| q||g||< q|S )NZbrackets)Zarray_format)r  Zstringify_itemsr
  r2   append)rj   r  r   Z
serializedkeyvalueexistingrh   rh   rk   r  G  s   

z#BaseClient._serialize_multipartformcast_totype[ResponseT]c                 C  sD   t |js|S t|j}|tt}t |r ||_ttt |S |S rg   )	r4   r   r   r  rG   r/   r   r   r%   )rj   r&  r   r   Zoverride_cast_torh   rh   rk   _maybe_override_cast_toe  s   

z"BaseClient._maybe_override_cast_tor   c                 C  s   |j tdkS )Nstream)r   r
  rF   rj   r   rh   rh   rk   _should_stream_response_bodyv  s   z'BaseClient._should_stream_response_bodyr%   c             
   C  s   |d u r	t td S |tu rt t|S z,t|r(t|tr(t t|j||dW S | jr5t tt	||dW S t tt
||dW S  tjyR } zt||d|d }~ww )N)r   r  )type_r$  )r   r   )r   r%   r   inspectisclass
issubclassr.   buildr   r<   r=   r   ZValidationErrorrP   )rj   r  r&  r   errrh   rh   rk   _process_response_datay  s   

z!BaseClient._process_response_datar   c                 C  r   rg   r   rs   rh   rh   rk   r       zBaseClient.qshttpx.Auth | Nonec                 C  rf   rg   rh   rs   rh   rh   rk   custom_auth     zBaseClient.custom_authdict[str, str]c                 C  s   i S rg   rh   rs   rh   rh   rk   auth_headers  r6  zBaseClient.auth_headersdict[str, str | Omit]c                 C  s"   dd| j d|  | j| jS )Nzapplication/json)Acceptr  z
User-Agent)
user_agentplatform_headersr8  r   rs   rh   rh   rk   r     s   zBaseClient.default_headersc                 C  s
   i | j S rg   )r   rs   rh   rh   rk   r    s   zBaseClient.default_queryr   r"   c                 C     dS )zaValidate the given default headers and custom headers.

        Does nothing by default.
        Nrh   )rj   r   r   rh   rh   rk   r        	zBaseClient._validate_headersc                 C  s   | j j d| j S )Nz/Python )rq   rr   r   rs   rh   rh   rk   r;    s   zBaseClient.user_agentc                 C  s   | j S rg   )r   rs   rh   rh   rk   r     r3  zBaseClient.base_url	URL | strc                 C  s"   |  t|tr	|nt|| _d S rg   )r   r   r   r   ri   rh   rh   rk   r     s   "Dict[str, str]c                 C  s   t | j| jdS )N)platform)r<  r   r   rs   rh   rh   rk   r<    s   zBaseClient.platform_headersresponse_headersOptional[httpx.Headers]float | Nonec              	   C  s   |du rdS z| dd}t|d W S  ttfy   Y nw | d}zt|W S  ttfy4   Y nw tj|}|du rAdS tj|}t|t  S )a@  Returns a float of the number of seconds (not milliseconds) to wait after retrying, or None if unspecified.

        About the Retry-After header: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Retry-After
        See also  https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Retry-After#syntax
        Nzretry-after-ms  zretry-after)	r
  floatr   r   emailutilsparsedate_tz	mktime_tztime)rj   rB  Zretry_ms_headerZretry_headerretry_date_tuple
retry_daterh   rh   rk   _parse_retry_after_header  s&   

z$BaseClient._parse_retry_after_headerremaining_retriesrF  c           
      C  s~   | | j}| |}|d urd|  k rdkr|S  t|| d}tttd| t}ddt   }|| }	|	dkr=|	S dS )Nr   <   rE  g       @r   g      ?)get_max_retriesr   rN  minrE   powrC   r   )
rj   rO  r   rB  r   retry_afterZ
nb_retriesZsleep_secondsZjitterr   rh   rh   rk   _calculate_retry_timeout  s   
z#BaseClient._calculate_retry_timeoutc                 C  s   |j d}|dkrtd dS |dkrtd dS |jdkr*td	|j dS |jd
kr8td	|j dS |jdkrFtd	|j dS |jdkrTtd	|j dS td dS )Nzx-should-retrytruez4Retrying as header `x-should-retry` is set to `true`Tfalsez9Not retrying as header `x-should-retry` is set to `false`Fi  zRetrying due to status code %ii  i  i  zNot retrying)r   r
  rS   r  r   )rj   r   Zshould_retry_headerrh   rh   rk   _should_retry  s*   






zBaseClient._should_retryc                 C  s   dt   S )Nzstainless-python-retry-)uuiduuid4rs   rh   rh   rk   _idempotency_key(  s   zBaseClient._idempotency_key)r   ro   r   r   r   r~   r   r   r   r   r   r   r   r   rd   re   )r_   r   rd   r   )r   r   rd   rM   )r   ro   r   r   r   r   rd   r   )r   r;   r   r   rd   r   )r_   ro   rd   r   )rd   r   )r   r;   r   r   rd   r  )r  r   rd   r!  )r&  r'  r   r;   rd   r'  )r   r  rd   r~   )r  r   r&  r'  r   r   rd   r%   )rd   r   )rd   r4  )rd   r7  )rd   r9  )rd   r!  )r   r"   r   r"   rd   re   ru   )rd   r   )r_   r?  rd   re   )rd   r@  rg   )rB  rC  rd   rD  )rO  r   r   r;   rB  rC  rd   rF  )r   r   rd   r~   )"rr   rv   rw   ry   r   rD   rB   rl   r   r   r   r   r   r   r  r  r(  r+  r2  propertyr  r5  r8  r   r  r   r;  r   setterr<  rN  rU  rX  r[  rh   rh   rh   rk   r   m  sb   
 


	


h





&
#r   c                         e Zd Zd fddZ  ZS )_DefaultHttpxClientr  r   rd   re   c                   :   | dt | dt | dd t jdi | d S Nr   Zlimitsfollow_redirectsTrh   
setdefaultrB   rH   superrl   rj   r  rq   rh   rk   rl   -     z_DefaultHttpxClient.__init__r  r   rd   re   rr   rv   rw   rl   __classcell__rh   rh   rg  rk   r_  ,      r_  c                   @     e Zd ZdddZdS )SyncHttpxClientWrapperrd   re   c                 C  s.   | j rd S z|   W d S  ty   Y d S w rg   )r   closer   rs   rh   rh   rk   __del__A  s   zSyncHttpxClientWrapper.__del__Nrd   re   rr   rv   rw   rp  rh   rh   rh   rk   rn  @      rn  c                      s   e Zd ZU ded< dZded< eeddddd fddZdddZdddZ	dd!d"Z
dd)d*Zdd-d.Zdd1d2Zedd:d;Zed<d=dd@d;Zed<ddAddDd;Zd<ddAddFd;ZddJdKZdLdMddPdQZddWdXZei d<dYdd\d]Zei d^dd`d]Zei ddaddbd]Zi d<ddcdddd]Zeddi dd<deddldmZeddi ddnddodmZeddi dddpddqdmZddi dd<ddrddsdmZdddi dtddudvZdddi dtddwdxZddi dyddzd{Zdi d|d}dddZ  ZS )r   zhttpx.Clientr   Nztype[Stream[Any]] | Noner   r   r   http_clientr   r   r   ro   r   r   r   r   r   !float | Timeout | None | NotGivenru  httpx.Client | Noner   r   r   r   r   r~   rd   re   c          	   	     s   t |s|r|jtkr|j}nt}|d ur$t|tjs$tdt| t	 j
|tt||||||d |p>t|tt|d| _d S )NzOInvalid `http_client` argument; Expected an instance of `httpx.Client` but got )r   r   r   r   r   r   r   r   r   )r4   r   HTTPX_DEFAULT_TIMEOUTrB   r   r   Clientr   r  re  rl   r   r#   rn  r   )	rj   r   r   r   r   ru  r   r   r   rg  rh   rk   rl   O  s*   
zSyncAPIClient.__init__c                 C     | j jS rg   r   r   rs   rh   rh   rk   r   }     zSyncAPIClient.is_closedc                 C  s   t | dr| j  dS dS )`Close the underlying HTTPX client.

        The client will *not* be usable after this.
        r   N)hasattrr   ro  rs   rh   rh   rk   ro    s   
zSyncAPIClient.closerj   rW   c                 C  s   | S rg   rh   rs   rh   rh   rk   	__enter__  r   zSyncAPIClient.__enter__exc_typetype[BaseException] | NoneexcBaseException | Noneexc_tbTracebackType | Nonec                 C  s   |    d S rg   ro  rj   r  r  r  rh   rh   rk   __exit__  s   zSyncAPIClient.__exit__r   r;   c                 C  s   |S z#Hook for mutating the given optionsrh   rj   r   rh   rh   rk   _prepare_options  rm   zSyncAPIClient._prepare_optionsr   r  c                 C  r=  a  This method is used as a callback for mutating the `Request` object
        after it has been constructed.
        This is useful for cases where you want to add certain headers based off of
        the request properties, e.g. `url`, `method` etc.
        Nrh   r*  rh   rh   rk   _prepare_request  r>  zSyncAPIClient._prepare_requestr&  Type[ResponseT]r)  Literal[True]
stream_clsType[_StreamT]rZ   c                C  rf   rg   rh   rj   r&  r   r)  r  rh   rh   rk   r        zSyncAPIClient.requestFr)  Literal[False]r%   c                C  rf   rg   rh   rj   r&  r   r)  rh   rh   rk   r     s   r)  r  Type[_StreamT] | NoneResponseT | _StreamTc                C  rf   rg   rh   r  rh   rh   rk   r     r  type[_StreamT] | Nonec                C  s  |  ||}t|}|jd u r|j dkr|  |_d }|| j}d}t|d D ]}t|}| 	|}|| }	| j
||d}
| |
 i }| jd urS| j|d< |jd ur]|j|d< td|
j|
j d }z| jj|
fd|pv| j|
d	i|}W nb tjy } z%tjd
dd |	dkr| j|||d d W Y d }~q+td t|
d	|d }~w ty } z%tjddd |	dkr| j|||d d W Y d }~q+td t|
d	|d }~ww td|
j|
j|j|j|j td|jd z|  W nK tjyI } z=tjddd |	dkr/|  |j!r/|j!"  | j||||d W Y d }~q+|j!j#s9|j!$  td | %|j!d d }~ww  |d usTJ d| j&||||||dS Nr
  r   r   r   authrb  zSending HTTP Request: %s %sr)  )r   z"Encountered httpx.TimeoutExceptionT)exc_info)r   r   r   r   zRaising timeout errorzEncountered ExceptionzRaising connection errorzHTTP Response: %s %s "%i %s" %szrequest_id: %szx-request-idz!Encountered httpx.HTTPStatusErrorzRe-raising status errorz0could not resolve response (should never happen))r&  r   r   r)  r  r   )'r(  r8   r   r  r   r[  rQ  r   ranger  r  r  r5  rb  rS   r  r_   r   sendr+  r   TimeoutException_sleep_for_retryrN   r   rO   r   reason_phraser   r
  raise_for_statusHTTPStatusErrorrX  r   ro  r   r   r   _process_responserj   r&  r   r)  r  Zinput_optionsr   r   r   rO  r   r  r1  rh   rh   rk   r     s   













r   r   httpx.Response | Nonec                C  s^   || }|dkrt d nt d| | |||r|jnd }t d|j| t| d S Nr   z1 retry leftz%i retries leftz$Retrying request to %s in %f seconds)rS   r  rU  r   r   r_   rK  sleeprj   r   r   r   r   rO  r   rh   rh   rk   r  <  s   zSyncAPIClient._sleep_for_retryr   r   r   1type[Stream[Any]] | type[AsyncStream[Any]] | Nonec          
      C  s   |j jtdkrttt|| |||||dS t|p|}t	|rVt
|trV|r2t|j jtrVt
|tsAtdt d| td|}tt||| t|||||dS |tjkr`tt|S t|| td|||||d}	t|j jtr|tt|	S |	 S )NrV  rawr   r&  r)  r  r   r   !API Response types must subclass ; Received type[BaseAPIResponse[Any]]r'  )r   r   r
  rF   r   r%   rR   r   r-  r.  r/  r?   r~   r>   r   rA   r   Responseparse
rj   r&  r   r   r)  r  r   originZresponse_clsZapi_responserh   rh   rk   r  J  sf   




	
zSyncAPIClient._process_responser   Type[object]r   Type[SyncPageT]rT   c                   s(   d fdd}|_ j|ddS )	Nr   rT   rd   c                   s   | j  d | S )N)r   r   r   )r   r   r   r   rj   rh   rk   r     s   z0SyncAPIClient._request_api_list.<locals>._parserFr  )r   rT   rd   rT   )r   r   )rj   r   r   r   r   rh   r  rk   r     s   zSyncAPIClient._request_api_listr   r)  pathr+   c                C  rf   rg   rh   rj   r  r&  r   r)  rh   rh   rk   r
    r  zSyncAPIClient.getr   type[_StreamT]c                C  rf   rg   rh   rj   r  r&  r   r)  r  rh   rh   rk   r
    r>  r   r  c                C  rf   rg   rh   r  rh   rh   rk   r
    r>  r   r)  r  c                C  s.   t jdd|d|}tt| j||||dS Nr
  )r  r_   r  rh   )r;   	constructr   r%   r   rj   r  r&  r   r)  r  optsrh   rh   rk   r
    s   	)r   r  r   r  r)  r   Body | Noner  BinaryTypes | Noner  RequestFiles | Nonec                C  rf   rg   rh   )rj   r  r&  r   r  r   r  r)  rh   rh   rk   post  s   zSyncAPIClient.post)r   r  r   r  c          	      C  rf   rg   rh   	rj   r  r&  r   r  r   r  r)  r  rh   rh   rk   r       )r   r  r   r  r  c          	      C  rf   rg   rh   r  rh   rh   rk   r    r  )r   r  r   r  r)  r  c          
      C  s   |d ur|d urt d|d ur|d urt dt|tr%tjdtdd tjd	d|||t|d|}	t	t
| j||	||dS 
N2Passing both `body` and `content` is not supported3Passing both `files` and `content` is not supportedPassing raw bytes as `body` is deprecated and will be removed in a future version. Please pass raw bytes via the `content` parameter instead.   
stacklevelr  r  r_   r   r  r  r  rh   )r   r   r  warningswarnDeprecationWarningr;   r  r   r   r%   r   )
rj   r  r&  r   r  r   r  r)  r  r  rh   rh   rk   r    s    
r   r  r  r   c                C  v   |d ur|d urt d|d ur|d urt dt|tr%tjdtdd tjdd|||t|d|}| 	||S 	Nr  r  r  r  r  patchr  rh   
r   r   r  r  r  r  r;   r  r   r   rj   r  r&  r   r  r  r   r  rh   rh   rk   r        

zSyncAPIClient.patchc                C  r  	Nr  r  r  r  r  putr  rh   r  r  rh   rh   rk   r  -  r  zSyncAPIClient.putr   r  r   c                C  sX   |d ur|d urt dt|trtjdtdd tjdd|||d|}| ||S Nr  r  r  r  delete)r  r_   r   r  rh   	r   r   r  r  r  r  r;   r  r   rj   r  r&  r   r  r   r  rh   rh   rk   r  G  s   	
zSyncAPIClient.deleter
  r   r   r  r  c                C  &   t jd|||d|}| |||S N)r  r_   r   rh   r;   r  r   rj   r  r   r   r   r   r  r  rh   rh   rk   get_api_list\     
zSyncAPIClient.get_api_list)r   ro   r   r   r   r   r   rv  ru  rw  r   r   r   r   r   r~   rd   re   r   rq  rj   rW   rd   rW   r  r  r  r  r  r  rd   re   r   r;   rd   r;   r   r  rd   re   )
r&  r  r   r;   r)  r  r  r  rd   rZ   r&  r  r   r;   r)  r  rd   r%   )
r&  r  r   r;   r)  r~   r  r  rd   r  )
r&  r  r   r;   r)  r~   r  r  rd   r  
r   r   r   r   r   r;   r   r  rd   re   r&  r  r   r;   r   r   r)  r~   r  r  r   r   rd   r%   )r   r  r   r  r   r;   rd   rT   
r  ro   r&  r  r   r+   r)  r  rd   r%   )r  ro   r&  r  r   r+   r)  r  r  r  rd   rZ   )r  ro   r&  r  r   r+   r)  r~   r  r  rd   r  )r  ro   r&  r  r   r  r  r  r   r+   r  r  r)  r  rd   r%   )r  ro   r&  r  r   r  r  r  r   r+   r  r  r)  r  r  r  rd   rZ   )r  ro   r&  r  r   r  r  r  r   r+   r  r  r)  r~   r  r  rd   r  )r  ro   r&  r  r   r  r  r  r  r  r   r+   rd   r%   )r  ro   r&  r  r   r  r  r  r   r+   rd   r%   )r  ro   r   r  r   r  r   r  r   r+   r  ro   rd   rT   )rr   rv   rw   ry   r   rD   r/   rl   r   ro  r  r  r  r  r   r   r  r  r   r
  r  r  r  r  r  rk  rh   rh   rg  rk   r   K  s   
 
.





	
x
D	
!r   c                      r^  )_DefaultAsyncHttpxClientr  r   rd   re   c                   r`  ra  rc  rf  rg  rh   rk   rl   k  rh  z!_DefaultAsyncHttpxClient.__init__ri  rj  rh   rh   rg  rk   r  j  rl  r  c                   @  s   e Zd ZdddZdS )	_DefaultAioHttpClient_kwargsr   rd   re   c                 K  s   t d)NzVTo use the aiohttp client you must have installed the package with the `aiohttp` extra)r   )rj   r  rh   rh   rk   rl   w  r}  _DefaultAioHttpClient.__init__N)r  r   rd   re   )rr   rv   rw   rl   rh   rh   rh   rk   r  v  rs  r  c                      r^  )r  r  r   rd   re   c                   r`  ra  rc  rf  rg  rh   rk   rl   |  s   r  ri  rj  rh   rh   rg  rk   r  {  rl  c                   @  rm  )AsyncHttpxClientWrapperrd   re   c                 C  s8   | j rd S zt |   W d S  ty   Y d S w rg   )r   asyncioZget_running_loopZcreate_taskacloser   rs   rh   rh   rk   rp    s   zAsyncHttpxClientWrapper.__del__Nrq  rr  rh   rh   rh   rk   r    rs  r  c                      s  e Zd ZU ded< dZded< eeddddd fddZdddZdddZ	dd!d"Z
dd)d*Zdd-d.Zdd1d2Zed3d4dd:d;Zedd@d;ZeddAddDd;Zd3ddEddFd;ZddJdKZdLdMddPdQZddWdXZei d3dYdd\d]Zei d^dd_d]Zei dd`ddad]Zi d3ddbddcd]Zedddi d3ddddkdlZedddi dmddndlZedddi ddoddpdlZdddi d3ddqddrdlZdddi dmddsdtZdddi dmddudvZddi dwddxdyZdi dzd{dd}d~Z  ZS )r   zhttpx.AsyncClientr   Nztype[AsyncStream[Any]] | Noner   rt  r   ro   r   r   r   r~   r   r   r   rv  ru  httpx.AsyncClient | Noner   r   r   r   rd   re   c          	   	     s   t |s|r|jtkr|j}nt}|d ur$t|tjs$tdt| t	 j
||tt|||||d |p>t|tt|d| _d S )NzTInvalid `http_client` argument; Expected an instance of `httpx.AsyncClient` but got )r   r   r   r   r   r   r   rx  )r4   r   ry  rB   r   r   AsyncClientr   r  re  rl   r   r#   r  r   )	rj   r   r   r   r   r   ru  r   r   rg  rh   rk   rl     s*   
zAsyncAPIClient.__init__c                 C  r{  rg   r|  rs   rh   rh   rk   r     r}  zAsyncAPIClient.is_closedc                   s   | j  I dH  dS )r~  N)r   r  rs   rh   rh   rk   ro    s   zAsyncAPIClient.closerj   rW   c                   s   | S rg   rh   rs   rh   rh   rk   
__aenter__  s   zAsyncAPIClient.__aenter__r  r  r  r  r  r  c                   s   |   I d H  d S rg   r  r  rh   rh   rk   	__aexit__  s   zAsyncAPIClient.__aexit__r   r;   c                   s   |S r  rh   r  rh   rh   rk   r    s   zAsyncAPIClient._prepare_optionsr   r  c                   s   dS r  rh   r*  rh   rh   rk   r       	zAsyncAPIClient._prepare_requestFr  r&  r  r)  r  r%   c                     d S rg   rh   r  rh   rh   rk   r     s   zAsyncAPIClient.requestr  r  type[_AsyncStreamT]r[   c                  r  rg   rh   r  rh   rh   rk   r        )r  type[_AsyncStreamT] | NoneResponseT | _AsyncStreamTc                  r  rg   rh   r  rh   rh   rk   r     r  r  c                  s  | j d u rtt I d H | _ | ||}t|}|jd u r*|j dkr*|  |_d }|	| j
}d}t|d D ]6}t|}| |I d H }|| }	| j||d}
| |
I d H  i }| jd urh| j|d< |jd urr|j|d< td|
j|
j d }z| jj|
fd|p| j|
d	i|I d H }W nh tjy } z(tjd
dd |	dkr| j|||d dI d H  W Y d }~q:td t|
d	|d }~w ty } z(tjddd |	dkr| j|||d dI d H  W Y d }~q:td t|
d	|d }~ww td|
j|
j|j|j|j td|j d z|!  W nT tj"yp } zFtjddd |	dkrS| #|j$rS|j$% I d H  | j||||dI d H  W Y d }~q:|j$j&s`|j$' I d H  td | (|j$d d }~ww  |d us{J d| j)||||||dI d H S r  )*r   r3   get_platformr(  r8   r   r  r   r[  rQ  r   r  r  r  r  r5  rb  rS   r  r_   r   r  r+  r   r  r  rN   r   rO   r   r  r   r
  r  r  rX  r   r  r   Zareadr   r  r  rh   rh   rk   r     s   









r   r   r  c                  sf   || }|dkrt d nt d| | |||r|jnd }t d|j| t|I d H  d S r  )rS   r  rU  r   r   r_   anyior  r  rh   rh   rk   r    s   zAsyncAPIClient._sleep_for_retryr   r   r   r  c          
        s  |j jtdkrttt|| |||||dS t|p|}t	|rWt
|trW|r3t|j jtrWt
|tsBtdt d| td|}td||| t|||||dS |tjkratt|S t|| td|||||d}	t|j jtr}tt|	S |	 I d H S )NrV  r  r  r  r  r%   r'  )r   r   r
  rF   r   r%   rR   r   r-  r.  r/  r?   r~   r@   r   rA   r   r  r  r  rh   rh   rk   r    sh   




	
z AsyncAPIClient._process_responser   r|   r   r   AsyncPaginator[_T, AsyncPageT]c                 C  s   t | |||dS )N)r   r   r   r   )r   )rj   r   r   r   rh   rh   rk   r     s   z AsyncAPIClient._request_api_listr  r  r+   c                  r  rg   rh   r  rh   rh   rk   r
    r  zAsyncAPIClient.getr  c                  r  rg   rh   r  rh   rh   rk   r
    r  r  c                  r  rg   rh   r  rh   rh   rk   r
    r  r  c                  s0   t jdd|d|}| j||||dI d H S r  )r;   r  r   r  rh   rh   rk   r
    s   	)r   r  r  r   r)  r   r  r  AsyncBinaryTypes | Noner  r  c                  r  rg   rh   )rj   r  r&  r   r  r  r   r)  rh   rh   rk   r    s   zAsyncAPIClient.postr  c          	        r  rg   rh   	rj   r  r&  r   r  r  r   r)  r  rh   rh   rk   r  &     )r   r  r  r   r  c          	        r  rg   rh   r  rh   rh   rk   r  4  r	  )r   r  r  r   r)  r  c          
        s   |d ur|d urt d|d ur|d urt dt|tr&tjdtdd tjd	d|||t|I d H d|}	| j	||	||dI d H S r  
r   r   r  r  r  r  r;   r  r   r   )
rj   r  r&  r   r  r  r   r)  r  r  rh   rh   rk   r  B  s"   
c                     |d ur|d urt d|d ur|d urt dt|tr&tjdtdd tjdd|||t|I d H d|}| 	||I d H S r  r
  r  rh   rh   rk   r  ^  s*   

zAsyncAPIClient.patchc                  r  r  r
  r  rh   rh   rk   r  }  s"   

zAsyncAPIClient.putr  c                  s`   |d ur|d urt dt|trtjdtdd tjdd|||d|}| ||I d H S r  r  r  rh   rh   rk   r    s   	
zAsyncAPIClient.deleter
  r  r  c                C  r  r  r  r  rh   rh   rk   r    r  zAsyncAPIClient.get_api_list)r   ro   r   r   r   r~   r   r   r   rv  ru  r  r   r   r   r   rd   re   r   rq  r  r  r  r  r  )
r&  r  r   r;   r)  r  r  r   rd   r[   )
r&  r  r   r;   r)  r~   r  r  rd   r  r  r  )r   r|   r   r   r   r;   rd   r  r  )r  ro   r&  r  r   r+   r)  r  r  r   rd   r[   )r  ro   r&  r  r   r+   r)  r~   r  r  rd   r  )r  ro   r&  r  r   r  r  r  r  r  r   r+   r)  r  rd   r%   )r  ro   r&  r  r   r  r  r  r  r  r   r+   r)  r  r  r   rd   r[   )r  ro   r&  r  r   r  r  r  r  r  r   r+   r)  r~   r  r  rd   r  )r  ro   r&  r  r   r  r  r  r  r  r   r+   rd   r%   )r  ro   r&  r  r   r  r  r  r   r+   rd   r%   )r  ro   r   r|   r   r   r   r  r   r+   r  ro   rd   r  )rr   rv   rw   ry   r   rD   r/   rl   r   ro  r  r  r  r  r   r   r  r  r   r
  r  r  r  r  r  rk  rh   rh   rg  rk   r     s   
 
.




	
}
D	
!$r   )queryextra_headersextra_query
extra_bodyr   r   r   synthesize_event_and_datar  Query | Noner  Headers | Noner  r  r  r   r   r   'float | httpx.Timeout | None | NotGivenr   PostParser | NotGivenr  bool | Nonerd   r+   c           	      C  s   i }|dur
||d< |durt t||d< | dur| |d< |dur-i |di ||d< t|ts6||d< |dur>||d< t|rF||d< |durN||d< |S )	zECreate a dict of type RequestOptions without keys of NotGiven values.Nr   r  ra   r   r   r   r  )r   r&   r
  r   r$   r4   )	r  r  r  r  r   r   r   r  r   rh   rh   rk   make_request_options  s$   
r  c                   @  rm  )r  rd   r~   c                 C  r=  )NTrh   rs   rh   rh   rk   __bool__  r   zForceMultipartDict.__bool__Nr   )rr   rv   rw   r  rh   rh   rh   rk   r    rs  r  c                   @  $   e Zd Zd
ddZedddZd	S )OtherPlatformnamero   rd   re   c                 C  
   || _ d S rg   r  rj   r  rh   rh   rk   rl        
zOtherPlatform.__init__c                 C     d| j  S )NzOther:r  rs   rh   rh   rk   __str__     zOtherPlatform.__str__Nr  ro   rd   re   ru   rr   rv   rw   rl   r   r   rh   rh   rh   rk   r        
r  )MacOSLinuxWindowsFreeBSDOpenBSDiOSAndroidUnknownPlatformc                  C  s   zt   } t    }W n
 ty   Y dS w d|v s!d|v r#dS | dkr)dS | dkr/dS d	|v r5d
S | dkrKt }|dkrCdS |dkrIdS dS |rQt|S dS )Nr,  iphoneZipadr*  darwinr%  windowsr'  androidr+  linuxfreebsdr(  openbsdr)  r&  )rA  systemr   r   distroidr  )r5  Zplatform_name	distro_idrh   rh   rk   r     s0   r  )maxsizer   ro   rA  Platform | Noner@  c                C  s&   d| t |pt t t t t dS )Npython)zX-Stainless-LangzX-Stainless-Package-VersionzX-Stainless-OSzX-Stainless-ArchzX-Stainless-RuntimezX-Stainless-Runtime-Version)ro   r  get_architectureget_python_runtimeget_python_version)r   rA  rh   rh   rk   r<  *  s   r<  c                   @  r  )	OtherArchr  ro   rd   re   c                 C  r  rg   r  r  rh   rh   rk   rl   7  r  zOtherArch.__init__c                 C  r  )Nzother:r  rs   rh   rh   rk   r   :  r!  zOtherArch.__str__Nr"  ru   r#  rh   rh   rh   rk   r?  6  r$  r?  )x32x64armarm64unknownc                   C      zt  W S  ty   Y dS w NrD  )rA  python_implementationr   rh   rh   rh   rk   r=  B  
   
r=  c                   C  rE  rF  )rA  python_versionr   rh   rh   rh   rk   r>  I  rH  r>  Archc                  C  sh   zt   } W n
 ty   Y dS w | dv rdS | dkrdS | dkr%dS tjdkr,dS | r2t| S dS )	NrD  )rC  aarch64rC  rB  x86_64rA  l        r@  )rA  machiner   r   sysr9  r?  )rM  rh   rh   rk   r<  P  s    
r<  obj1Mapping[_T_co, Union[_T, Omit]]obj2Dict[_T_co, _T]c                 C  s   i | |}dd |  D S )zMerge two mappings of the same type, removing any values that are instances of `Omit`.

    In cases with duplicate keys the second mapping takes precedence.
    c                 S  s    i | ]\}}t |ts||qS rh   )r   r    )r   r#  r$  rh   rh   rk   
<dictcomp>s  s     z#_merge_mappings.<locals>.<dictcomp>)r   )rO  rQ  Zmergedrh   rh   rk   r   j  s   r   )r  r  r  r  r  r  r  r  r   r   r   r  r   r  r  r  rd   r+   )rd   r-  )r   ro   rA  r:  rd   r@  ru   )rd   rJ  )rO  rP  rQ  rP  rd   rR  )
__future__r   rN  rc   rK  rY  rG  r  r-  r  rA  r  email.utilstypesr   r   typingr   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   Ztyping_extensionsr   r   r   r  r   r6  r   r   r    r   Z_qsr   _filesr   r   _typesr   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   Z_utilsr0   r1   r2   r3   r4   r5   r6   Z_compatr7   r8   r9   Z_modelsr:   r;   r<   r=   	_responser>   r?   r@   rA   
_constantsrB   rC   rD   rE   rF   rG   rH   Z
_streamingrI   rJ   rK   rL   rM   rN   rO   rP   Z_utils._jsonrQ   Z_legacy_responserR   	getLoggerrr   rS   ry   	addFilterrT   rV   rW   rX   rZ   r[   Zhttpx._configr\   ry  ImportErrorr]   rz   r   r   r   rz  r  r   r   r   r_  ZDefaultHttpxClientrn  r   r  Zhttpx_aiohttpr  ZHttpxAiohttpClientZDefaultAsyncHttpxClientZDefaultAioHttpClientr  r   r  ro   r  r  r-  r  r<  r?  rJ  r=  r>  r<  r   rh   rh   rh   rk   <module>   s    DL$$	3=1')   B    #	     )	
*	


