U
    ”Ê¹jA  ã                   @   s8   d dl mZ d dlmZ d dlmZ G dd„ deƒZdS )é    )ÚOptional)ÚInferenceProviderMapping)ÚBaseConversationalTaskc                       sD   e Zd Z‡ fdd„Zee edœdd„Zee edœdd„Z‡  Z	S )	ÚOpenAIConversationalTaskc                    s   t ƒ jddd� d S )NÚopenaizhttps://api.openai.com)ÚproviderÚbase_url)ÚsuperÚ__init__)Úself©Ú	__class__© új/var/www/html/bwave/python/venv/lib/python3.8/site-packages/huggingface_hub/inference/_providers/openai.pyr
      s    z!OpenAIConversationalTask.__init__)Úapi_keyÚreturnc                 C   s&   |d krt dƒ‚| d¡r"t dƒ‚|S )Nz4You must provide an api_key to work with OpenAI API.Zhf_zbOpenAI provider is not available through Hugging Face routing, please use your own OpenAI API key.)Ú
ValueErrorÚ
startswith)r   r   r   r   r   Ú_prepare_api_key   s    
ÿz)OpenAIConversationalTask._prepare_api_key)Úmodelr   c                 C   s"   |d krt dƒ‚td|dd|d�S )Nz9Please provide an OpenAI model ID, e.g. `gpt-4o` or `o1`.r   ZconversationalÚlive)r   Z
providerIdÚtaskÚstatusZhf_model_id)r   r   )r   r   r   r   r   Ú_prepare_mapping_info   s        ÿz.OpenAIConversationalTask._prepare_mapping_info)
Ú__name__Ú
__module__Ú__qualname__r
   r   Ústrr   r   r   Ú__classcell__r   r   r   r   r      s   	r   N)Útypingr   Zhuggingface_hub.hf_apir   Z,huggingface_hub.inference._providers._commonr   r   r   r   r   r   Ú<module>   s   