
    RX9i                     N    S SK JrJrJr  S SKJr  SSKJr  SrSr	 " S S\5      r
g	)
    )AnyDictOptional)InferenceProviderMapping   )BaseConversationalTaskcoherezhttps://api.cohere.comc            	       d   ^  \ rS rSrU 4S jrS\S\S\4S jrS\S\S	\	S\
\   4U 4S
 jjrSrU =r$ )CohereConversationalTask   c                 2   > [         TU ]  [        [        S9  g )N)providerbase_url)super__init__	_PROVIDER	_BASE_URL)self	__class__s    k/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/huggingface_hub/inference/_providers/cohere.pyr   !CohereConversationalTask.__init__   s    )i@    mapped_modelapi_keyreturnc                     g)Nz"/compatibility/v1/chat/completions )r   r   r   s      r   _prepare_route'CohereConversationalTask._prepare_route   s    3r   inputs
parametersprovider_mapping_infoc                   > [         TU ]  XU5      nUR                  S5      n[        U[        5      (       aL  UR                  S5      S:X  a7  UR                  S5      n[        U[        5      (       a  SU;   a  SUS   S.US'   U$ )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)   1CohereConversationalTask._prepare_payload_as_dict   s     '26G\]$..):;ot,,1D1DV1LP]1]"1"5"5m"D-t44EX9X)1(;.)*
 r   r   )__name__
__module____qualname____firstlineno__r   strr   r   r   r   r   r)   __static_attributes____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>r:      s*    & & ; + 	$	5 r   