a
    f	g                     @   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 )EngineZenginesNc              
   K   s   t   }z$| jd|  d ||dddW S  tyx } z4|d urVt   || krV tjd|d W Y d }~qd }~0 0 qd S Npostz	/generatestreamT)r	   Zplain_old_datazWaiting for model to warm up)error)timerequestinstance_urlgetr   r   log_infoselftimeoutparamsstarte r   ^/var/www/html/llm_bihealth/app/venv/lib/python3.9/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y~ } z4|d ur\t   || kr\ tjd|d W Y d }~qd }~0 0 qd S r   )r   Z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)N)__name__
__module____qualname__ZOBJECT_NAMEr   r   r   r   r   r   r   r   	   s   

r   )
r   r   Zopenair   Zopenai.api_resources.abstractr   r   Zopenai.errorr   r   r   r   r   r   <module>   s
   