Ë
    Z²j  ã                   ót   — 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
 dZdZ G d„ d	e	«      Z G d
„ de«      Zy)é    )ÚAny)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dicté   )ÚBaseConversationalTaskÚBaseTextGenerationTaskÚfilter_noneÚ	deepinfrazhttps://api.deepinfra.comc            	       ó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 )ÚDeepInfraTextGenerationTaskc                 ó8   •— t         ‰| �  t        t        ¬«       y ©N)ÚproviderÚbase_url©ÚsuperÚ__init__Ú	_PROVIDERÚ	_BASE_URL©ÚselfÚ	__class__s    €úm/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/deepinfra.pyr   z$DeepInfraTextGenerationTask.__init__   ó   ø€ Ü‰Ñ¤)´iÐÕ@ó    Úmapped_modelÚapi_keyÚreturnc                  ó   — y)Nz/v1/openai/completions© ©r   r   r   s      r   Ú_prepare_routez*DeepInfraTextGenerationTask._prepare_route   s   € Ø'r   ÚinputsÚ
parametersÚprovider_mapping_infoNc                 ó†   — t        |j                  «       «      }|j                  dd «      |d<   d|i|¥d|j                  i¥S )NÚmax_new_tokensÚ
max_tokensÚpromptÚmodel)r
   ÚcopyÚpopÚprovider_id)r   r$   r%   r&   Úparamss        r   Ú_prepare_payload_as_dictz4DeepInfraTextGenerationTask._prepare_payload_as_dict   sI   € ô ˜ZŸ_™_Ó.Ó/ˆØ%Ÿz™zÐ*:¸DÓAˆˆ|Ñà˜&ÐW FÐW¨GÐ5J×5VÑ5VÑWÐWr   ÚresponseÚrequest_paramsc                 óv   — t        |«      d   d   }|d   |j                  d«      |j                  d«      dœdœS )NÚchoicesr   ÚtextÚfinish_reasonÚseed)r6   r7   )Úgenerated_textÚdetails)r   Úget)r   r1   r2   Úoutputs       r   Úget_responsez(DeepInfraTextGenerationTask.get_response   sD   € Ü˜(Ó# IÑ.¨qÑ1ˆà$ V™nà!'§¡¨OÓ!<ØŸ
™
 6Ó*ññ
ð 	
r   )N)Ú__name__Ú
__module__Ú__qualname__r   Ústrr#   r   Údictr   r0   Úbytesr   r<   Ú__classcell__©r   s   @r   r   r      sw   ø„ ôAð(¨3ð (¸ð (Àó (ðXØðXØ'+ðXØD\ðXà	�‰óXñ
 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 )ÚDeepInfraConversationalTaskc                 ó8   •— t         ‰| �  t        t        ¬«       y r   r   r   s    €r   r   z$DeepInfraConversationalTask.__init__(   r   r   r   r   r   c                  ó   — y)Nz/v1/openai/chat/completionsr!   r"   s      r   r#   z*DeepInfraConversationalTask._prepare_route+   s   € Ø,r   )r=   r>   r?   r   r@   r#   rC   rD   s   @r   rF   rF   '   s#   ø„ ôAð-¨3ð -¸ð -À÷ -r   rF   N)Útypingr   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r   Ú_commonr   r	   r
   r   r   r   rF   r!   r   r   ú<module>rM      s:   ðÝ å ;ß Iç PÑ Pð €	Ø'€	ô
Ð"8ô 
ô4-Ð"8õ -r   