o
    h                     @   sj   d dl Z ddlmZmZ ddlmZmZ e jegef Ze jege j	ddef f Z
G dd deeZdS )    N   )RequestResponse   )AsyncBaseTransportBaseTransportc                   @   sL   e Zd Zdejeef ddfddZdede	fddZ
dede	fd	d
ZdS )MockTransporthandlerreturnNc                 C   s
   || _ d S N)r	   )selfr	    r   L/var/www/html/ai/venv/lib/python3.10/site-packages/httpx/_transports/mock.py__init__   s   
zMockTransport.__init__requestc                 C   s(   |   | |}t|tstd|S )Nz,Cannot use an async handler in a sync Client)readr	   
isinstancer   	TypeErrorr   r   responser   r   r   handle_request   s
   

zMockTransport.handle_requestc                    s2   |  I d H  | |}t|ts|I d H }|S r   )areadr	   r   r   r   r   r   r   handle_async_request   s   


z"MockTransport.handle_async_request)__name__
__module____qualname__typingUnionSyncHandlerAsyncHandlerr   r   r   r   r   r   r   r   r   r   
   s    

r   )r   _modelsr   r   baser   r   Callabler   	Coroutiner   r   r   r   r   r   <module>   s    