Ë
    Q^(hÒ	  ã                   ó¦   — 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 d dlmZ dZdZ G d„ d	e«      Z G d
„ de«      Z G d„ de«      Zy)é    )ÚAnyÚDictÚOptionalÚUnion)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dict)ÚBaseConversationalTaskÚBaseTextGenerationTaskÚTaskProviderHelperÚfilter_none)Úget_sessionÚnovitazhttps://api.novita.aic                   óX   ‡ — e Zd Zˆ fd„Zdededefd„Zd	deeef   de	e
   defd„Zˆ xZS )
ÚNovitaTextGenerationTaskc                 ó8   •— t         ‰| �  t        t        ¬«       y ©N)ÚproviderÚbase_url©ÚsuperÚ__init__Ú	_PROVIDERÚ	_BASE_URL©ÚselfÚ	__class__s    €úi/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/novita.pyr   z!NovitaTextGenerationTask.__init__   ó   ø€ Ü‰Ñ¤)´iÐÕ@ó    Úmapped_modelÚapi_keyÚreturnc                  ó   — y)Nz/v3/openai/completions© ©r   r!   r"   s      r   Ú_prepare_routez'NovitaTextGenerationTask._prepare_route   s   € à'r    ÚresponseÚrequest_paramsc                 ó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%NovitaTextGenerationTask.get_response   sD   € Ü˜(Ó# IÑ.¨qÑ1ˆà$ V™nà!'§¡¨OÓ!<ØŸ
™
 6Ó*ññ
ð 	
r    ©N)Ú__name__Ú
__module__Ú__qualname__r   Ústrr'   r   Úbytesr   r   r   r   r3   Ú__classcell__©r   s   @r   r   r      sM   ø„ ôAð(¨3ð (¸ð (Àó (ñ
 U¨5°$¨;Ñ%7ð 
ÈÐRcÑIdð 
Ðps÷ 
r    r   c                   ó2   ‡ — e Zd Zˆ fd„Zdededefd„Zˆ xZS )ÚNovitaConversationalTaskc                 ó8   •— t         ‰| �  t        t        ¬«       y r   r   r   s    €r   r   z!NovitaConversationalTask.__init__&   r   r    r!   r"   r#   c                  ó   — y)Nz/v3/openai/chat/completionsr%   r&   s      r   r'   z'NovitaConversationalTask._prepare_route)   s   € à,r    )r5   r6   r7   r   r8   r'   r:   r;   s   @r   r=   r=   %   s#   ø„ ôAð-¨3ð -¸ð -À÷ -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 )ÚNovitaTextToVideoTaskc                 ó:   •— t         ‰| �  t        t        d¬«       y )Nztext-to-video)r   r   Útaskr   r   s    €r   r   zNovitaTextToVideoTask.__init__/   s   ø€ Ü‰Ñ¤)´iÀoÐÕVr    r!   r"   r#   c                 ó   — d|› �S )Nz/v3/hf/r%   r&   s      r   r'   z$NovitaTextToVideoTask._prepare_route2   s   € Ø˜˜Ð'Ð'r    ÚinputsÚ
parametersÚprovider_mapping_infoc                 ó    — d|it        |«      ¥S )NÚprompt)r   )r   rE   rF   rG   s       r   Ú_prepare_payload_as_dictz.NovitaTextToVideoTask._prepare_payload_as_dict5   s   € ð ˜&Ð<¤K°
Ó$;Ð<Ð<r    r(   r)   c                 óà   — t        |«      }t        |t        «      rd|v rt        |d   t        «      rd|d   v st        d«      ‚|d   d   }t	        «       j                  |«      j                  S )NÚvideoÚ	video_urlz>Expected response format: { 'video': { 'video_url': string } })r	   Ú
isinstanceÚdictÚ
ValueErrorr   r1   Úcontent)r   r(   r)   Úresponse_dictrM   s        r   r3   z"NovitaTextToVideoTask.get_response:   sn   € Ü  Ó*ˆä�}¤dÔ+Ø˜=Ñ(Ü˜=¨Ñ1´4Ô8Ø˜}¨WÑ5Ñ5äÐ]Ó^Ð^à! 'Ñ*¨;Ñ7ˆ	Ü‹}× Ñ  Ó+×3Ñ3Ð3r    r4   )r5   r6   r7   r   r8   r'   r   r   r   r   rJ   r   r9   r   r3   r:   r;   s   @r   rA   rA   .   sv   ø„ ôWð(¨3ð (¸ð (Àó (ð=Øð=Ø'+ð=ØD\ð=à	�$‰ó=ñ
4 U¨5°$¨;Ñ%7ð 4ÈÐRcÑIdð 4Ðps÷ 4r    rA   N)Útypingr   r   r   r   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r	   Ú,huggingface_hub.inference._providers._commonr
   r   r   r   Úhuggingface_hub.utilsr   r   r   r   r=   rA   r%   r    r   ú<module>rX      sS   ðß -Ó -å ;ß I÷ó õ .ð €	Ø#€	ô
Ð5ô 
ô&-Ð5ô -ô4Ð.õ 4r    