
    Y j                     F    S SK Jr  S SKJr  SSKJr  SrSr " S S\5      rg	)
    )Any)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\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    |/root/GenerationalWealth/GenerationalWealth/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_infoNc                   > [         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'   __static_attributes____classcell__)r   s   @r   r	   r	      sQ    A43 4 4 4'+D\	 r   r	   N)	typingr   huggingface_hub.hf_apir   _commonr   r   r   r	   r   r   r   <module>r8      s'     ; + 	$	5 r   