o
    RTh                     @   sN   d dl Z d dlZd dlmZ d dlmZmZ d dlmZ G dd deeZ	dS )    N)util)ListableAPIResourceUpdateableAPIResource)TryAgainc                   @   s,   e Zd ZdZd	ddZd	ddZdd ZdS )
EngineenginesNc              
   K   s~   t   }	 z| jd|  d ||dddW S  ty= } z|d ur,t   || kr, tjd|d W Y d }~nd }~ww qNTpostz	/generatestream)r
   plain_old_datazWaiting for model to warm up)error)timerequestinstance_urlgetr   r   log_infoselftimeoutparamsstarte r   ^/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/openai/api_resources/engine.pygenerate   s"   
zEngine.generatec              
      s   t   }	 z| jd|  d ||dddI d H W S  tyA } z|d ur0t   || kr0 tjd|d W Y d }~nd }~ww qr   )r   arequestr   r   r   r   r   r   r   r   r   	agenerate   s$   
zEngine.ageneratec                 K   s"   t dt | d|  d |S )Nz5Engine.embeddings is deprecated, use Embedding.creater	   z/embeddings)warningswarnDeprecationWarningr   r   )r   r   r   r   r   
embeddings.   s   zEngine.embeddings)N)__name__
__module____qualname__OBJECT_NAMEr   r   r    r   r   r   r   r   	   s
    

r   )
r   r   openair   openai.api_resources.abstractr   r   openai.errorr   r   r   r   r   r   <module>   s    