
    ʩh                     p    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y	)
    )AnyDictOptionalUnion)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    k/var/www/html/eduruby.in/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            	       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 )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_infoc                 :    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   r4   r5   
embeddingsr8   s        r   get_responsez*ScalewayFeatureExtractionTask.get_response   s*    h'/
8BC9	+&CCCs   ")N)r   r   r   r   strr+   r   r   r   r   r3   r   bytesr   r:   r    r!   s   @r   r#   r#      s    m 3      [['+[D\[	$[DU5$;%7 DRcId Dps Dr   r#   N)typingr   r   r   r   !huggingface_hub.inference._commonr   r   _commonr
   r   r   r   r   r#   r*   r   r   <module>r@      s2    - - I f fR!7 R
D$6 Dr   