
    g                     B    d dl mZ d dlZddlmZmZ dZ G d de      Zy)    )UnionN   )BeforeRequestContextBeforeRequestHookzmistral-client-python/c                   R    e Zd Zdedej
                  deej
                  ef   fdZy)CustomUserAgentHookhook_ctxrequestreturnc                     |j                   d   }|j                  t              r|S t        |j                  d      d   z   |j                   d<   |S )Nz
user-agent r   )headers
startswithPREFIXsplit)selfr	   r
   currents       ]/var/www/openai/venv/lib/python3.12/site-packages/mistralai_azure/_hooks/custom_user_agent.pybefore_requestz"CustomUserAgentHook.before_request   sR     //,/f%N W]]3'** 	%     N)	__name__
__module____qualname__r   httpxRequestr   	Exceptionr    r   r   r   r   
   s1    ,7<}}	u}}i'	(r   r   )typingr   r   typesr   r   r   r   r   r   r   <module>r       s       :	!+ r   