
    ;3h                     N    d dl mZmZmZ d dlmZ ddlmZ dZdZ	 G d de      Z
y	)
    )AnyDictOptional)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	e   f fd	Z
 xZS )
CohereConversationalTaskc                 8    t         |   t        t               y )N)providerbase_url)super__init__	_PROVIDER	_BASE_URL)self	__class__s    k/var/www/html/audio-gradio/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.pyr   z!CohereConversationalTask.__init__   s    )i@    mapped_modelapi_keyreturnc                      y)Nz"/compatibility/v1/chat/completions )r   r   r   s      r   _prepare_routez'CohereConversationalTask._prepare_route   s    3r   inputs
parametersprovider_mapping_infoc                     t         |   |||      }|j                  d      }t        |t              rD|j                  d      dk(  r0|j                  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     '26:G\]$..):;ot,1D1DV1LP]1]"1"5"5m"D-t4EX9X)1(;.)*
 r   )__name__
__module____qualname__r   strr   r   r   r   r   r&   __classcell__)r   s   @r   r   r      sQ    A43 4 4 4'+D\	$ r   r   N)typingr   r   r   huggingface_hub.hf_apir   _commonr   r   r   r   r   r   r   <module>r4      s*    & & ; + 	$	5 r   