Ë
    Q^(h]  ã                   óš   — d dl Z d dlmZmZmZmZ 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 G d
„ de«      Zy)é    N)ÚAnyÚDictÚOptionalÚUnion)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dict)ÚBaseConversationalTaskÚBaseTextGenerationTaskÚTaskProviderHelperÚfilter_nonec                   óD   ‡ — e Zd Zˆ fd„Zddeeef   dee   de	fd„Z
ˆ xZS )ÚNebiusTextGenerationTaskc                 ó(   •— t         ‰| �  dd¬«       y ©NÚnebiusúhttps://api.studio.nebius.ai)ÚproviderÚbase_url©ÚsuperÚ__init__©ÚselfÚ	__class__s    €úi/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/nebius.pyr   z!NebiusTextGenerationTask.__init__   ó   ø€ Ü‰Ñ (Ð5SÐÕTó    ÚresponseÚrequest_paramsÚreturnc                 óv   — t        |«      d   d   }|d   |j                  d«      |j                  d«      dœdœS )NÚchoicesr   ÚtextÚfinish_reasonÚseed)r%   r&   )Úgenerated_textÚdetails)r	   Úget)r   r   r    Úoutputs       r   Úget_responsez%NebiusTextGenerationTask.get_response   sD   € Ü˜(Ó# IÑ.¨qÑ1ˆà$ V™nà!'§¡¨OÓ!<ØŸ
™
 6Ó*ññ
ð 	
r   ©N)Ú__name__Ú
__module__Ú__qualname__r   r   Úbytesr   r   r   r   r+   Ú__classcell__©r   s   @r   r   r      s3   ø„ ôUñ
 U¨5°$¨;Ñ%7ð 
ÈÐRcÑIdð 
Ðps÷ 
r   r   c                   ó   ‡ — e Zd Zˆ fd„Zˆ xZS )ÚNebiusConversationalTaskc                 ó(   •— t         ‰| �  dd¬«       y r   r   r   s    €r   r   z!NebiusConversationalTask.__init__   r   r   )r-   r.   r/   r   r1   r2   s   @r   r4   r4      s   ø„ ÷Uð Ur   r4   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 )ÚNebiusTextToImageTaskc                 ó*   •— t         ‰| �  ddd¬«       y )Nztext-to-imager   r   )Útaskr   r   r   r   s    €r   r   zNebiusTextToImageTask.__init__#   s   ø€ Ü‰Ñ˜o¸ÐKiÐÕjr   Úmapped_modelÚapi_keyr!   c                  ó   — y)Nz/v1/images/generations© )r   r:   r;   s      r   Ú_prepare_routez$NebiusTextToImageTask._prepare_route&   s   € Ø'r   ÚinputsÚ
parametersÚprovider_mapping_infoc                 óž   — |j                   }t        |«      }d|v r|j                  d«       |j                  d«      dvrd|d<   d|i|¥d|i¥S )NÚguidance_scaleÚresponse_format)Úb64_jsonÚurlrE   ÚpromptÚmodel)Úprovider_idr   Úpopr)   )r   r?   r@   rA   r:   s        r   Ú_prepare_payload_as_dictz.NebiusTextToImageTask._prepare_payload_as_dict)   sc   € ð -×8Ñ8ˆÜ  Ó,ˆ
Ø˜zÑ)Ø�N‰NÐ+Ô,Ø�>‰>Ð+Ó,Ð4GÑGØ,6ˆJÐ(Ñ)à˜&ÐF JÐF°¸ÑFÐFr   r   r    c                 óT   — t        |«      }t        j                  |d   d   d   «      S )NÚdatar   rE   )r	   Úbase64Ú	b64decode)r   r   r    Úresponse_dicts       r   r+   z"NebiusTextToImageTask.get_response5   s,   € Ü  Ó*ˆÜ×Ñ ¨fÑ 5°aÑ 8¸Ñ DÓEÐEr   r,   )r-   r.   r/   r   Ústrr>   r   r   r   r   rK   r   r0   r   r+   r1   r2   s   @r   r7   r7   "   s   ø„ ôkð(¨3ð (¸ð (Àó (ð
GØð
GØ'+ð
GØD\ð
Gà	�$‰ó
GñF U¨5°$¨;Ñ%7ð FÈÐRcÑIdð FÐps÷ Fr   r7   )rN   Útypingr   r   r   r   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r	   Ú,huggingface_hub.inference._providers._commonr
   r   r   r   r   r4   r7   r=   r   r   ú<module>rV      sI   ðÛ ß -Ó -å ;ß I÷ó ô
Ð5ô 
ôUÐ5ô Uô
FÐ.õ Fr   