
    ITh]                         S SK r 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Jr   " S S\5      r " S S	\5      r " S
 S\5      rg)    N)AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict)BaseConversationalTaskBaseTextGenerationTaskTaskProviderHelperfilter_nonec                   R   ^  \ rS rSrU 4S jrSS\\\4   S\\	   S\
4S jjrSrU =r$ )	NebiusTextGenerationTask   c                 "   > [         TU ]  SSS9  g Nnebiushttps://api.studio.nebius.ai)providerbase_urlsuper__init__self	__class__s    c/var/www/auris/envauris/lib/python3.13/site-packages/huggingface_hub/inference/_providers/nebius.pyr   !NebiusTextGenerationTask.__init__       (5ST    responserequest_paramsreturnc                 v    [        U5      S   S   nUS   UR                  S5      UR                  S5      S.S.$ )Nchoicesr   textfinish_reasonseed)r'   r(   )generated_textdetails)r	   get)r   r!   r"   outputs       r   get_response%NebiusTextGenerationTask.get_response   sD    (#I.q1$Vn!'O!<

6*
 	
r     N)__name__
__module____qualname____firstlineno__r   r   bytesr   r   r   r   r-   __static_attributes____classcell__r   s   @r   r   r      s8    U
U5$;%7 
RcId 
ps 
 
r    r   c                   (   ^  \ rS rSrU 4S jrSrU =r$ )NebiusConversationalTask   c                 "   > [         TU ]  SSS9  g r   r   r   s    r   r   !NebiusConversationalTask.__init__   r   r    r/   )r1   r2   r3   r4   r   r6   r7   r8   s   @r   r:   r:      s    U U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$ )NebiusTextToImageTask"   c                 $   > [         TU ]  SSSS9  g )Nztext-to-imager   r   )taskr   r   r   r   s    r   r   NebiusTextToImageTask.__init__#   s    oKijr    mapped_modelapi_keyr#   c                     g)Nz/v1/images/generationsr/   )r   rD   rE   s      r   _prepare_route$NebiusTextToImageTask._prepare_route&   s    'r    inputs
parametersprovider_mapping_infoc                     UR                   n[        U5      nSU;   a  UR                  S5        UR                  S5      S;  a  SUS'   SU0UESU0E$ )Nguidance_scaleresponse_format)b64_jsonurlrO   promptmodel)provider_idr   popr+   )r   rI   rJ   rK   rD   s        r   _prepare_payload_as_dict.NebiusTextToImageTask._prepare_payload_as_dict)   sc     -88 ,
z)NN+,>>+,4GG,6J()&FJFFFr    r!   r"   c                 V    [        U5      n[        R                  " US   S   S   5      $ )Ndatar   rO   )r	   base64	b64decode)r   r!   r"   response_dicts       r   r-   "NebiusTextToImageTask.get_response5   s,     *f 5a 8 DEEr    r/   r0   )r1   r2   r3   r4   r   strrG   r   r   r   r   rU   r   r5   r   r-   r6   r7   r8   s   @r   r?   r?   "   s    k(3 ( ( (
G
G'+
GD\
G	$
GFU5$;%7 FRcId Fps F Fr    r?   )rY   typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   ,huggingface_hub.inference._providers._commonr
   r   r   r   r   r:   r?   r/   r    r   <module>rb      sI     - - ; I 
5 
U5 U
F. Fr    