
    RX9i                     x    S SK JrJrJrJr  S SKJr  S SKJrJ	r	  S SK
JrJrJr   " S S\5      r " S S\5      rg	)
    )AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict)BaseConversationalTaskTaskProviderHelperfilter_nonec            	       N   ^  \ rS rSrU 4S jrS\S\S\S\\   4U 4S jjr	Sr
U =r$ )	SambanovaConversationalTask   c                 "   > [         TU ]  SSS9  g )N	sambanovahttps://api.sambanova.ai)providerbase_urlsuper__init__self	__class__s    n/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/huggingface_hub/inference/_providers/sambanova.pyr   $SambanovaConversationalTask.__init__	   s    +8RS    inputs
parametersprovider_mapping_inforeturnc                 (  > UR                  S5      n[        U[        5      (       aZ  UR                  S5      S:X  aE  UR                  S0 5      nUR                  S5      n[        U[        5      (       a  USL d  Uc  SUS'   [        TU ]  XU5      nU$ )Nresponse_formattypejson_schemastrictTF)get
isinstancedictr   _prepare_payload_as_dict)	r   r   r   r    response_format_configjson_schema_configr&   payloadr   s	           r   r*   4SambanovaConversationalTask._prepare_payload_as_dict   s     ",0A!B,d33%))&1]B%;%?%?r%R"+//90$77Vt^v~38&x0'26G\]r    )__name__
__module____qualname____firstlineno__r   r   r   r   r   r*   __static_attributes____classcell__r   s   @r   r   r      s7    T'+D\	$ r   r   c            	          ^  \ rS rSrU 4S jrS\S\S\4S jrS\S\S	\	S\
\   4S
 jrSS\\\4   S\
\   S\4S jjrSrU =r$ )SambanovaFeatureExtractionTask   c                 $   > [         TU ]  SSSS9  g )Nr   r   zfeature-extraction)r   r   taskr   r   s    r   r   'SambanovaFeatureExtractionTask.__init__   s    +8RYmnr   mapped_modelapi_keyr!   c                     g)Nz/v1/embeddingsr/   )r   r=   r>   s      r   _prepare_route-SambanovaFeatureExtractionTask._prepare_route   s    r   r   r   r    c                 8    [        U5      nXR                  S.UE$ )N)inputmodel)r   provider_id)r   r   r   r    s       r   r*   7SambanovaFeatureExtractionTask._prepare_payload_as_dict"   s"     !,
*K*KZzZZr   responserequest_paramsc                 R    [        U5      S   nU Vs/ s H  oDS   PM	     sn$ s  snf )Ndata	embedding)r	   )r   rG   rH   
embeddingsrK   s        r   get_response+SambanovaFeatureExtractionTask.get_response(   s,    h'/
8BC
9+&
CCCs   $r/   )N)r0   r1   r2   r3   r   strr@   r   r   r   r   r*   r   bytesr   rM   r4   r5   r6   s   @r   r8   r8      s    o 3      [['+[D\[	$[DU5$;%7 DRcId Dps D Dr   r8   N)typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   ,huggingface_hub.inference._providers._commonr
   r   r   r   r8   r/   r   r   <module>rU      s3    - - ; I p p"8 &D%7 Dr   