
    g.                         d Z ddlmZ ddlZddlmZmZ ddlmZm	Z	m
Z
 ddlmZmZmZ ddlmZmZmZmZmZ dd	lmZmZ  G d
 d      Zy)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.   )SDKConfiguration    N)modelsutils)AfterErrorContextAfterSuccessContextBeforeRequestContext)RetryConfigSerializedRequestBodyget_body_content)CallableListMappingOptionalTuple)parse_qsurlparsec                      e Zd ZU eed<   deddfdZd Z	 	 	 	 	 	 ddee   dee	g ee
   f      d	ee   d
eeeef      dej                  f
dZ	 	 	 	 	 	 ddee   dee	g ee
   f      d	ee   d
eeeef      dej                  f
dZ	 	 	 	 	 	 ddee   dee	g ee
   f      d	ee   d
eeeef      dej                  f
dZ	 	 ddeeeee   f      dej*                  fdZ	 	 ddeeeee   f      dej*                  fdZy)BaseSDKsdk_configuration
sdk_configreturnNc                     || _         y N)r   )selfr   s     L/var/www/openai/venv/lib/python3.12/site-packages/mistralai_azure/basesdk.py__init__zBaseSDK.__init__   s
    !+    c                 x    | j                   j                         \  }}||}||}t        j                  ||      S r   )r   get_server_detailsr   template_url)r   base_urlurl_variablessdk_urlsdk_variabless        r   _get_urlzBaseSDK._get_url   sD    !%!7!7!J!J!LH )M!!(M::r   
timeout_msget_serialized_bodyurl_overridehttp_headersc                 p    | j                   j                  }| j                  ||||||||||	|
||||||      S r   )r   async_client_build_request_with_clientr   methodpathr"   r#   requestrequest_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_value_globalssecurityr'   r(   r)   r*   clients                     r   _build_request_asynczBaseSDK._build_request_async!   s[    * ''44..!#$#
 	
r   c                 p    | j                   j                  }| j                  ||||||||||	|
||||||      S r   )r   r9   r-   r.   s                     r   _build_requestzBaseSDK._build_requestK   s[    * ''....!#$#
 	
r   c           
      :   i }|}|Ot        j                  | j                  ||      ||r|nd |r|nd       }t        j                  |	r|nd |	r|nd       }n+t	        t        |            }t        |j                  d      }t        j                  ||      }||d<   | j                  j                  ||
<   |t        |      r |       }|$t        j                  |      \  }}i ||}i ||}t               }| |       }|r|t        d      ||}|j                  |j                  dvr|j                  |d<   ||j!                         D ]
  \  }}|||<    ||dz  nd }|j#                  ||||j$                  |j&                  |j(                  ||      S )	NT)keep_blank_valuesAcceptzrequest body is required)zmultipart/form-datazmultipart/mixedzcontent-typei  )paramscontentdatafilesheaderstimeout)r   generate_urlr&   get_query_paramsr   strr   queryget_headersr   
user_agentcallableget_securityr   
ValueError
media_typeitemsbuild_requestrA   rB   rC   )r   r9   r/   r0   r"   r#   r1   r2   r3   r4   r5   r6   r7   r8   r'   r(   r)   r*   query_paramsurlparsed_overriderD   security_headerssecurity_query_paramsserialized_request_bodyrbheadervaluerE   s                                r   r-   z"BaseSDK._build_request_with_clientu   s   , ;$$h623	C !1134$L 's<'89O#O$9$9TRL##GX6/%)%;%;%F%F!"!#:6;6H6H6R3355$45GDlD.CDL"7"9*$&B$ !;<<~*,' $..:'22 '>&H&HGN##!-!3!3!5"' "6 (2'=*t#4##+33(--)// $ 	
 		
r   retry_configc                    	  j                   j                   j                   j                  		 fd}|1t        j                  |t        j
                  |d   |d               }n |       }t        j                  |j                        s3 j                   j                         j                  t              |      }|S )Nc            	         d } 	 j                   j                         j                  t              
      }	j	                  d|j
                  |j                  |j                  t        |             j                  |      } | &	j	                  d       t        j                  d      	j	                  d| j                  | j                  | j                  rd	n| j                          t#        j$                  | j                        rzj                   j                         j                  t              | d       \  }}|	j	                  dd       |||} | S 	j	                  d
       t        j                  d      | S # t        $ rY}j                   j                         j                  t              d |      \  }}|	j	                  dd       |Y d }~bd }~ww xY wNz0Request:
Method: %s
URL: %s
Headers: %s
Body: %s)streamzRequest ExceptionT)exc_infozRaising no response SDK errorzNo response receivedz6Response:
Status Code: %s
URL: %s
Headers: %s
Body: %sz<streaming response>zRaising unexpected SDK errorzUnexpected error occurredr   	get_hooksbefore_requestr	   debugr/   rS   rD   r   send	Exceptionafter_errorr   r   SDKErrorstatus_codetextr   match_status_codeshttp_resreqe_resulterrr9   error_status_codeshook_ctxloggerr1   r   r_   s         r   dozBaseSDK.do_request.<locals>.do   s   H,,668GG(2G JJJGGKK$S) ";;s6;: <=oo&<==LLL$$  *0&hmm ''(:H<P<PQ"44>>@LL%h/4 ?LL!4tLDI%%H
 O LL!?@ //*EFFOC  --779EE%h/q1 =LL!4tLDG !	s   BF 	G.AG))G.r   r   )r   r9   debug_loggerr   retryRetriesrk   ri   rb   after_successr   
r   rt   r1   rs   r_   r[   rv   rm   r9   ru   s
   `````   @@r   
do_requestzBaseSDK.do_request   s     ''..''44/	 /	b #{{2u}}\!_lSTo'VWHtH''(:H<P<PQ--779GG#H-xH r   c                    	K    j                   j                   j                   j                  		 fd}|9t        j                  |t        j
                  |d   |d                d {   }n |        d {   }t        j                  |j                        s3 j                   j                         j                  t              |      }|S 7 i7 [w)Nc            	         K   d } 	 j                   j                         j                  t              
      }	j	                  d|j
                  |j                  |j                  t        |             j                  |       d {   } | &	j	                  d       t        j                  d      	j	                  d| j                  | j                  | j                  rd	n| j                          t#        j$                  | j                        rzj                   j                         j                  t              | d       \  }}|	j	                  dd       |||} | S 	j	                  d
       t        j                  d      | S 7 
# t        $ rY}j                   j                         j                  t              d |      \  }}|	j	                  dd       |Y d }~ed }~ww xY wwr^   ra   rl   s         r   rv   z$BaseSDK.do_request_async.<locals>.do%  s    H,,668GG(2G JJJGGKK$S) "(S!@@ <=oo&<==LLL$$  *0&hmm ''(:H<P<PQ"44>>@LL%h/4 ?LL!4tLDI%%H
 O LL!?@ //*EFFOE A --779EE%h/q1 =LL!4tLDG !	sC   G>BF FF DG>F 	G;"AG60G>6G;;G>r   r   )r   r,   rw   r   retry_asyncry   rk   ri   rb   rz   r   r{   s
   `````   @@r   do_request_asynczBaseSDK.do_request_async  s      ''44''44/	 /	b #"..EMM,q/<?C H  TzH''(:H<P<PQ--779GG#H-xH  "s%   A-C$6C 7C$C"AC$"C$)NNNNNN)FN)__name__
__module____qualname__r   __annotations__r   r&   r   intr   r   rH   r   httpxRequestr:   r<   r-   r   r
   r   Responser|   r    r   r   r   r      s   '',#3 , ,	;. $( &*48'(
 SM(
 &R"7889
(
$ sm%(
& wsCx01'(
( 
)(
l $( &*48'(
 SM(
 &R"7889
(
$ sm%(
& wsCx01'(
( 
)(
n $( &*48)[
 SM[
  &R"7889
![
& sm'[
( wsCx01)[
* 
+[
D @DF u[$s)%;<=F 
FZ @DH u[$s)%;<=H 
Hr   r   )__doc__sdkconfigurationr   r   mistralai_azurer   r   mistralai_azure._hooksr   r   r	   mistralai_azure.utilsr
   r   r   typingr   r   r   r   r   urllib.parser   r   r   r   r   r   <module>r      s9    G .  ) 
 W V ; ; +R Rr   