
    $	&h                     x    d dl mZmZmZmZ d dlmZ d dlmZm	Z	 d dl
mZmZmZ  G d de      Z G d de      Zy	)
    )AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict)BaseConversationalTaskTaskProviderHelperfilter_nonec                        e Zd Z fdZ xZS )SambanovaConversationalTaskc                 (    t         |   dd       y )N	sambanovahttps://api.sambanova.ai)providerbase_urlsuper__init__self	__class__s    /var/www/pru.catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/huggingface_hub/inference/_providers/sambanova.pyr   z$SambanovaConversationalTask.__init__	   s    +8RS    )__name__
__module____qualname__r   __classcell__r   s   @r   r   r      s    T Tr   r   c            	       v     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d	Z
dd
eeef   de	e   defdZ xZS )SambanovaFeatureExtractionTaskc                 *    t         |   ddd       y )Nr   r   zfeature-extraction)r   r   taskr   r   s    r   r   z'SambanovaFeatureExtractionTask.__init__   s    +8RYmnr   mapped_modelapi_keyreturnc                      y)Nz/v1/embeddings )r   r%   r&   s      r   _prepare_routez-SambanovaFeatureExtractionTask._prepare_route   s    r   inputs
parametersprovider_mapping_infoc                 :    t        |      }||j                  d|S )N)inputmodel)r   provider_id)r   r+   r,   r-   s       r   _prepare_payload_as_dictz7SambanovaFeatureExtractionTask._prepare_payload_as_dict   s%     !,
*?*K*KZzZZr   responserequest_paramsc                 N    t        |      d   }|D cg c]  }|d   	 c}S c c}w )Ndata	embedding)r	   )r   r3   r4   
embeddingsr7   s        r   get_responsez+SambanovaFeatureExtractionTask.get_response   s*    h'/
8BC9	+&CCCs   ")N)r   r   r   r   strr*   r   r   r   r   r2   r   bytesr   r9   r   r    s   @r   r"   r"      s    o 3      [['+[D\[	$[DU5$;%7 DRcId Dps 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>r@      s5    - - ; I p pT"8 T
D%7 Dr   