Ë
    Z²j¦	  ã                   óš   — d dl 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)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dict)ÚBaseConversationalTaskÚBaseTextGenerationTaskÚTaskProviderHelperÚfilter_none)Úget_sessionÚnovitazhttps://api.novita.aic                   óT   ‡ — e Zd Zˆ fd„Zdededefd„Zd
deez  dedz  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    €új/var/www/html/ia-enhanced/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   NÚ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$   ÚbytesÚdictr   r   r0   Ú__classcell__©r   s   @r   r   r      sI   ø„ ôAð(¨3ð (¸ð (Àó (ñ
 U¨T¡\ð 
ÐCTÐW[ÑC[ð 
Ðgj÷ 
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   )r2   r3   r4   r   r5   r$   r8   r9   s   @r   r;   r;   %   s#   ø„ ôAð-¨3ð -¸ð -À÷ -r   r;   c            	       ór   ‡ — e Zd Zˆ fd„Zdededefd„Zdedededed	z  fd
„Z	dde
ez  ded	z  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_infoNc                 ó    — d|it        |«      ¥S )NÚprompt)r
   )r   rC   rD   rE   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   Ú
isinstancer7   Ú
ValueErrorr   r.   Úcontent)r   r%   r&   Úresponse_dictrK   s        r   r0   z"NovitaTextToVideoTask.get_response:   sn   € Ü  Ó*ˆä�}¤dÔ+Ø˜=Ñ(Ü˜=¨Ñ1´4Ô8Ø˜}¨WÑ5Ñ5äÐ]Ó^Ð^à! 'Ñ*¨;Ñ7ˆ	Ü‹}× Ñ  Ó+×3Ñ3Ð3r   r1   )r2   r3   r4   r   r5   r$   r   r7   r   rH   r6   r   r0   r8   r9   s   @r   r?   r?   .   sr   ø„ ôWð(¨3ð (¸ð (Àó (ð=Øð=Ø'+ð=ØD\ð=à	�‰ó=ñ
4 U¨T¡\ð 4ÐCTÐW[ÑC[ð 4Ðgj÷ 4r   r?   N)Útypingr   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r   Ú,huggingface_hub.inference._providers._commonr   r   r	   r
   Úhuggingface_hub.utilsr   r   r   r   r;   r?   r"   r   r   ú<module>rU      sP   ðÝ å ;ß I÷ó õ .ð €	Ø#€	ô
Ð5ô 
ô&-Ð5ô -ô4Ð.õ 4r   