
    _i                     J    S SK JrJr  S SKJr  SSKJr  SrSr " S S\5      r	g	)
    )Any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    g/var/www/html/land-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   huggingface_hub.hf_apir   _commonr   r   r   r
   r   r   r   <module>r9      s'      ; + 	$	5 r   