a
    f	gJ                     @   sP   d dl Z d dlmZ d dlmZmZ d dlmZ d dlm	Z	 G dd deZ
dS )    N)util)DeletableAPIResourceListableAPIResource)EngineAPIResource)TryAgainc                       s4   e Zd ZdZe fddZe fddZ  ZS )
CompletionZcompletionsc              
      s|   t   }|dd}zt j|i |W S  tyt } z4|durRt   || krR tjd|d W Y d}~qd}~0 0 qdS z
        Creates a new completion for the provided prompt and parameters.

        See https://platform.openai.com/docs/api-reference/completions/create for a list
        of valid parameters.
        timeoutNzWaiting for model to warm up)error)timepopsupercreater   r   log_infoclsargskwargsstartr	   e	__class__ b/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/openai/api_resources/completion.pyr      s    zCompletion.createc              
      s   t   }|dd}zt j|i |I dH W S  tyz } z4|durXt   || krX tjd|d W Y d}~qd}~0 0 qdS r   )r   r   r   acreater   r   r   r   r   r   r   r       s    zCompletion.acreate)__name__
__module____qualname__ZOBJECT_NAMEclassmethodr   r   __classcell__r   r   r   r   r   	   s
   r   )r   Zopenair   Zopenai.api_resources.abstractr   r   Z1openai.api_resources.abstract.engine_api_resourcer   Zopenai.errorr   r   r   r   r   r   <module>   s
   