
    ITh                     x    S SK JrJrJrJr  S SKJr  S SKJrJ	r	  S SK
JrJrJr   " S S\5      r " S S\5      rg	)
    )AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict)BaseConversationalTaskTaskProviderHelperfilter_nonec                   (   ^  \ rS rSrU 4S jrSrU =r$ )SambanovaConversationalTask   c                 "   > [         TU ]  SSS9  g )N	sambanovahttps://api.sambanova.ai)providerbase_urlsuper__init__self	__class__s    f/var/www/auris/envauris/lib/python3.13/site-packages/huggingface_hub/inference/_providers/sambanova.pyr   $SambanovaConversationalTask.__init__	   s    +8RS     )__name__
__module____qualname____firstlineno__r   __static_attributes____classcell__r   s   @r   r   r      s    T Tr   r   c            	          ^  \ rS rSrU 4S jrS\S\S\4S jrS\S\S	\	S\
\   4S
 jrSS\\\4   S\
\   S\4S jjrSrU =r$ )SambanovaFeatureExtractionTask   c                 $   > [         TU ]  SSSS9  g )Nr   r   zfeature-extraction)r   r   taskr   r   s    r   r   'SambanovaFeatureExtractionTask.__init__   s    +8RYmnr   mapped_modelapi_keyreturnc                     g)Nz/v1/embeddingsr   )r   r,   r-   s      r   _prepare_route-SambanovaFeatureExtractionTask._prepare_route   s    r   inputs
parametersprovider_mapping_infoc                 8    [        U5      nXR                  S.UE$ )N)inputmodel)r   provider_id)r   r2   r3   r4   s       r   _prepare_payload_as_dict7SambanovaFeatureExtractionTask._prepare_payload_as_dict   s"     !,
*K*KZzZZr   responserequest_paramsc                 R    [        U5      S   nU Vs/ s H  oDS   PM	     sn$ s  snf )Ndata	embedding)r	   )r   r;   r<   
embeddingsr?   s        r   get_response+SambanovaFeatureExtractionTask.get_response   s,    h'/
8BC
9+&
CCCs   $r   )N)r   r    r!   r"   r   strr0   r   r   r   r   r9   r   bytesr   rA   r#   r$   r%   s   @r   r'   r'      s    o 3      [['+[D\[	$[DU5$;%7 DRcId Dps D Dr   r'   N)typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   ,huggingface_hub.inference._providers._commonr
   r   r   r   r'   r   r   r   <module>rI      s5    - - ; I p pT"8 T
D%7 Dr   