o
    0j                     @   s@   d dl mZ d dlmZ ddlmZ dZdZG dd deZd	S )
    )Any)InferenceProviderMapping   )BaseConversationalTaskZcoherezhttps://api.cohere.comc                	       sT   e Zd Z fddZdededefddZded	ed
ededB f fddZ	  Z
S )CohereConversationalTaskc                    s   t  jttd d S )N)providerbase_url)super__init__	_PROVIDER	_BASE_URL)self	__class__ l/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/huggingface_hub/inference/_providers/cohere.pyr
      s   z!CohereConversationalTask.__init__mapped_modelapi_keyreturnc                 C   s   dS )Nz"/compatibility/v1/chat/completionsr   )r   r   r   r   r   r   _prepare_route   s   z'CohereConversationalTask._prepare_routeinputs
parametersprovider_mapping_infoNc                    sd   t  |||}|d}t|tr0|ddkr0|d}t|tr0d|v r0d|d d|d< |S )Nresponse_formattypeZjson_schemaschemaZjson_object)r   r   )r	   _prepare_payload_as_dictget
isinstancedict)r   r   r   r   payloadr   Zjson_schema_detailsr   r   r   r      s   


z1CohereConversationalTask._prepare_payload_as_dict)__name__
__module____qualname__r
   strr   r   r   r   r   __classcell__r   r   r   r   r      s    r   N)	typingr   Zhuggingface_hub.hf_apir   _commonr   r   r   r   r   r   r   r   <module>   s    