
    wi                     ~    d dl mZmZmZ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DictOptionalUnion)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    /var/www/development/aibuddy-work/election-extract/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            	            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 )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_infoc                 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   Nresponserequest_paramsc                 D    t          |          d         }d |D             S )Ndatac                     g | ]
}|d          S )	embeddingr*   ).0r9   s     r   
<listcomp>z>ScalewayFeatureExtractionTask.get_response.<locals>.<listcomp>   s    CCC9	+&CCCr   )r   )r   r4   r5   
embeddingss       r   get_responsez*ScalewayFeatureExtractionTask.get_response   s(    h''/
CC
CCCCr   )N)r   r   r   r   strr+   r   r   r   r   r3   r   bytesr   r=   r    r!   s   @r   r#   r#      s        m m m m m 3            [['+[D\[	$[ [ [ [D DU5$;%7 DRcId Dps D D D D D D D Dr   r#   N)typingr   r   r   r   !huggingface_hub.inference._commonr   r   _commonr
   r   r   r   r   r#   r*   r   r   <module>rC      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   