
    i                     d    d dl mZ d dlmZmZ ddlmZmZmZm	Z	  G d de      Z
 G d de      Zy	)
    )Any)RequestParameters_as_dict   )BaseConversationalTaskInferenceProviderMappingTaskProviderHelperfilter_nonec                        e Zd Z fdZ xZS )ScalewayConversationalTaskc                 (    t         |   dd       y )Nscalewayhttps://api.scaleway.ai)providerbase_urlsuper__init__self	__class__s    ~/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/scaleway.pyr   z#ScalewayConversationalTask.__init__	   s    *7PQ    )__name__
__module____qualname__r   __classcell__r   s   @r   r   r      s    R Rr   r   c            	       r     e Zd Z fdZdededefdZdedededed	z  fd
Z	dde
ez  ded	z  defdZ xZS )ScalewayFeatureExtractionTaskc                 *    t         |   ddd       y )Nr   r   zfeature-extraction)r   r   taskr   r   s    r   r   z&ScalewayFeatureExtractionTask.__init__   s    *7PWklr   mapped_modelapi_keyreturnc                      y)Nz/v1/embeddings )r   r#   r$   s      r   _prepare_routez,ScalewayFeatureExtractionTask._prepare_route   s    r   inputs
parametersprovider_mapping_infoNc                 :    t        |      }||j                  d|S )N)inputmodel)r
   provider_id)r   r)   r*   r+   s       r   _prepare_payload_as_dictz6ScalewayFeatureExtractionTask._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   r1   r2   
embeddingsr5   s        r   get_responsez*ScalewayFeatureExtractionTask.get_response   s*    h'/
8BC9	+&CCCs   ")N)r   r   r   r   strr(   r   dictr   r0   bytesr   r7   r   r   s   @r   r    r       s{    m 3      [['+[D\[	[DUT\ DCTW[C[ Dgj Dr   r    N)typingr   !huggingface_hub.inference._commonr   r   _commonr   r   r	   r
   r   r    r'   r   r   <module>r>      s/     I f fR!7 R
D$6 Dr   