
    j                     r    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d	S )
    )Any)RequestParameters_as_dict   )BaseConversationalTaskInferenceProviderMappingTaskProviderHelperfilter_nonec                        e Zd Z fdZ xZS )ScalewayConversationalTaskc                 N    t                                          dd           d S )Nscalewayhttps://api.scaleway.ai)providerbase_urlsuper__init__self	__class__s    p/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/huggingface_hub/inference/_providers/scaleway.pyr   z#ScalewayConversationalTask.__init__	   s'    *7PQQQQQ    )__name__
__module____qualname__r   __classcell__r   s   @r   r   r      sA        R R R R R R R 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                 P    t                                          ddd           d S )Nr   r   zfeature-extraction)r   r   taskr   r   s    r   r   z&ScalewayFeatureExtractionTask.__init__   s*    *7PWklllllr   mapped_modelapi_keyreturnc                     dS )Nz/v1/embeddings )r   r#   r$   s      r   _prepare_routez,ScalewayFeatureExtractionTask._prepare_route   s    r   inputs
parametersprovider_mapping_infoNc                 8    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ZZzZZr   responserequest_paramsc                 D    t          |          d         }d |D             S )Ndatac                     g | ]
}|d          S )	embeddingr'   ).0r6   s     r   
<listcomp>z>ScalewayFeatureExtractionTask.get_response.<locals>.<listcomp>   s    CCC9	+&CCCr   )r   )r   r1   r2   
embeddingss       r   get_responsez*ScalewayFeatureExtractionTask.get_response   s(    h''/
CC
CCCCr   )N)r   r   r   r   strr(   r   dictr   r0   bytesr   r:   r   r   s   @r   r    r       s        m m m m m 3            [['+[D\[	[ [ [ [D DUT\ DCTW[C[ Dgj D D D D D D D Dr   r    N)typingr   !huggingface_hub.inference._commonr   r   _commonr   r   r	   r
   r   r    r'   r   r   <module>rA      s          I I I I I I I I f f f f f f f f f f f fR R R R R!7 R R R
D D D D D$6 D D D D Dr   