
    g
                     z    d Z ddlZddlmZmZmZmZmZmZm	Z	m
Z
 ddlmZ ddlmZmZmZ ddlmZ  G d d	e
      Zy)
zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.    N   )SDKInitHookBeforeRequestContextBeforeRequestHookAfterSuccessContextAfterSuccessHookAfterErrorContextAfterErrorHookHooks)
init_hooks)ListOptionalTuple)
HttpClientc                   L   e Zd ZddZdeddfdZdeddfdZdeddfdZ	de
ddfdZd	ed
edeeef   fdZdedej$                  dej$                  fdZdedej*                  dej*                  fdZdedeej*                     dee   deeej*                     ee   f   fdZy)SDKHooksreturnNc                 R    g | _         g | _        g | _        g | _        t	        |        y N)sdk_init_hooksbefore_request_hooksafter_success_hooksafter_error_hooksr   )selfs    R/var/www/openai/venv/lib/python3.12/site-packages/mistralai_gcp/_hooks/sdkhooks.py__init__zSDKHooks.__init__   s*    13=?!;= 794    hookc                 :    | j                   j                  |       y r   )r   appendr   r   s     r   register_sdk_init_hookzSDKHooks.register_sdk_init_hook   s    ""4(r   c                 :    | j                   j                  |       y r   )r   r    r!   s     r   register_before_request_hookz%SDKHooks.register_before_request_hook   s    !!((.r   c                 :    | j                   j                  |       y r   )r   r    r!   s     r   register_after_success_hookz$SDKHooks.register_after_success_hook!   s      ''-r   c                 :    | j                   j                  |       y r   )r   r    r!   s     r   register_after_error_hookz"SDKHooks.register_after_error_hook$   s    %%d+r   base_urlclientc                 V    | j                   D ]  }|j                  ||      \  }} ||fS r   )r   sdk_init)r   r)   r*   r   s       r   r,   zSDKHooks.sdk_init'   s2    ''D#}}Xv>Hf (r   hook_ctxrequestc                 t    | j                   D ](  }|j                  ||      }t        |t              r||}* |S r   )r   before_request
isinstance	Exception)r   r-   r.   r   outs        r   r0   zSDKHooks.before_request,   sA     --D%%h8C#y)	G	 . r   responsec                 t    | j                   D ](  }|j                  ||      }t        |t              r||}* |S r   )r   after_successr1   r2   )r   r-   r4   r   r3   s        r   r6   zSDKHooks.after_success7   sA     ,,D$$Xx8C#y)	H	 -
 r   errorc                     | j                   D ],  }|j                  |||      }t        |t              r||\  }}. ||fS r   )r   after_errorr1   r2   )r   r-   r4   r7   r   results         r   r9   zSDKHooks.after_errorA   sL     **D%%h%@F&),$OHe	 +
 r   )r   N)__name__
__module____qualname__r   r   r"   r   r$   r   r&   r
   r(   strr   r   r,   r   httpxRequestr0   r   Responser6   r	   r   r2   r9    r   r   r   r      s   ); )4 )/1B /t /.0@ .T .,n , ,   j  U3
?=S  
	,	7<}}			+7<~~	# 5>>* 	"	
 
x'))<<	=r   r   )__doc__r?   typesr   r   r   r   r   r	   r
   r   registrationr   typingr   r   r   mistralai_gcp.httpclientr   r   rB   r   r   <module>rH      s4    G 	 	 	 % ( ( /9u 9r   