
    j                     N    d dl mZ d dlmZ ddlmZ dZdZ G d de          Zd	S )
    )Any)InferenceProviderMapping   )BaseConversationalTaskcoherezhttps://api.cohere.comc            	       T     e Zd Z fdZdededefdZdedededed	z  f fd
Z	 xZ
S )CohereConversationalTaskc                 b    t                                          t          t                     d S )N)providerbase_url)super__init__	_PROVIDER	_BASE_URL)self	__class__s    n/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/huggingface_hub/inference/_providers/cohere.pyr   z!CohereConversationalTask.__init__   s&    )i@@@@@    mapped_modelapi_keyreturnc                     dS )Nz"/compatibility/v1/chat/completions )r   r   r   s      r   _prepare_routez'CohereConversationalTask._prepare_route   s    33r   inputs
parametersprovider_mapping_infoNc                 L   t                                          |||          }|                    d          }t          |t                    rU|                    d          dk    r<|                    d          }t          |t                    rd|v rd|d         d|d<   |S )Nresponse_formattypejson_schemaschemajson_object)r    r"   )r   _prepare_payload_as_dictget
isinstancedict)r   r   r   r   payloadr   json_schema_detailsr   s          r   r$   z1CohereConversationalTask._prepare_payload_as_dict   s     ''226:G\]]$..):;;ot,, 	1D1DV1L1LP]1]1]"1"5"5m"D"D-t44 EX9X9X)1(;. .)*
 r   )__name__
__module____qualname__r   strr   r   r'   r   r$   __classcell__)r   s   @r   r	   r	      s        A A A A A43 4 4 4 4 4 4'+D\	         r   r	   N)	typingr   huggingface_hub.hf_apir   _commonr   r   r   r	   r   r   r   <module>r2      s|          ; ; ; ; ; ; + + + + + + 	$	    5     r   