Ë
    Q^(he
  ã                   óÆ   — d dl Z d dlm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 dZdZ G d	„ d
ee«      Z G d„ de«      Z G d„ de«      Z G d„ de«      Zy)é    N)ÚABC)ÚAnyÚDictÚOptionalÚUnion)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dict)ÚBaseConversationalTaskÚBaseTextGenerationTaskÚTaskProviderHelperÚfilter_noneÚtogetherzhttps://api.together.xyzc                   ó<   ‡ — e Zd ZdZdefˆ fd„Zdededefd„Zˆ xZS )ÚTogetherTaskz"Base class for Together API tasks.Útaskc                 ó:   •— t         ‰| �  t        t        |¬«       y )N)ÚproviderÚbase_urlr   ©ÚsuperÚ__init__Ú	_PROVIDERÚ	_BASE_URL)Úselfr   Ú	__class__s     €úk/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/together.pyr   zTogetherTask.__init__   s   ø€ Ü‰Ñ¤)´iÀdÐÕKó    Úmapped_modelÚapi_keyÚreturnc                 ó”   — | j                   dk(  ry| j                   dk(  ry| j                   dk(  ryt        d| j                   › d�«      ‚)	Nútext-to-imagez/v1/images/generationsÚconversationalz/v1/chat/completionsztext-generationz/v1/completionszUnsupported task 'z' for Together API.)r   Ú
ValueError)r   r   r    s      r   Ú_prepare_routezTogetherTask._prepare_route   sK   € Ø�9‰9˜Ò'Ø+Ø�Y‰YÐ*Ò*Ø)Ø�Y‰YÐ+Ò+Ø$ÜÐ-¨d¯i©i¨[Ð8KÐLÓMÐMr   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ústrr   r&   Ú__classcell__©r   s   @r   r   r      s2   ø„ Ù,ðL˜Sõ LðN¨3ð N¸ð NÀ÷ Nr   r   c                   ó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 )ÚTogetherTextGenerationTaskc                 ó8   •— t         ‰| �  t        t        ¬«       y ©N)r   r   r   ©r   r   s    €r   r   z#TogetherTextGenerationTask.__init__$   ó   ø€ Ü‰Ñ¤)´iÐÕ@r   ÚresponseÚrequest_paramsr!   c                 óv   — t        |«      d   d   }|d   |j                  d«      |j                  d«      dœdœS )NÚchoicesr   ÚtextÚfinish_reasonÚseed)r9   r:   )Úgenerated_textÚdetails)r
   Úget)r   r4   r5   Úoutputs       r   Úget_responsez'TogetherTextGenerationTask.get_response'   sD   € Ü˜(Ó# IÑ.¨qÑ1ˆà$ V™nà!'§¡¨OÓ!<ØŸ
™
 6Ó*ññ
ð 	
r   ©N)r'   r(   r)   r   r   Úbytesr   r   r	   r   r?   r,   r-   s   @r   r/   r/   #   s3   ø„ ôAñ
 U¨5°$¨;Ñ%7ð 
ÈÐRcÑIdð 
Ðps÷ 
r   r/   c                   ó   ‡ — e Zd Zˆ fd„Zˆ xZS )ÚTogetherConversationalTaskc                 ó8   •— t         ‰| �  t        t        ¬«       y r1   r   r2   s    €r   r   z#TogetherConversationalTask.__init__3   r3   r   )r'   r(   r)   r   r,   r-   s   @r   rC   rC   2   s   ø„ ÷Að Ar   rC   c            	       ób   ‡ — e Zd Zˆ 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 )ÚTogetherTextToImageTaskc                 ó$   •— t         ‰| �  d«       y )Nr#   )r   r   r2   s    €r   r   z TogetherTextToImageTask.__init__8   s   ø€ Ü‰Ñ˜Õ)r   ÚinputsÚ
parametersÚprovider_mapping_infor!   c                 ó¦   — |j                   }t        |«      }d|v r|j                  d«      |d<   d|v r|j                  d«      |d<   |ddœ|¥d|i¥S )NÚnum_inference_stepsÚstepsÚguidance_scaleÚguidanceÚbase64)ÚpromptÚresponse_formatÚmodel)Úprovider_idr   Úpop)r   rH   rI   rJ   r   s        r   Ú_prepare_payload_as_dictz0TogetherTextToImageTask._prepare_payload_as_dict;   si   € ð -×8Ñ8ˆÜ  Ó,ˆ
Ø  JÑ.Ø",§.¡.Ð1FÓ"GˆJ�wÑØ˜zÑ)Ø%/§^¡^Ð4DÓ%EˆJ�zÑ"à °XÑcÀÐcÈWÐVbÑcÐcr   r4   r5   c                 óT   — t        |«      }t        j                  |d   d   d   «      S )NÚdatar   Úb64_json)r
   rP   Ú	b64decode)r   r4   r5   Úresponse_dicts       r   r?   z$TogetherTextToImageTask.get_responseG   s,   € Ü  Ó*ˆÜ×Ñ ¨fÑ 5°aÑ 8¸Ñ DÓEÐEr   r@   )r'   r(   r)   r   r   r   r   r   rV   r   rA   r	   r?   r,   r-   s   @r   rF   rF   7   sd   ø„ ô*ð
dØð
dØ'+ð
dØD\ð
dà	�$‰ó
dñF U¨5°$¨;Ñ%7ð FÈÐRcÑIdð FÐps÷ Fr   rF   )rP   Úabcr   Ú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/   rC   rF   © r   r   ú<module>rb      sh   ðÛ Ý ß -Ó -å ;ß I÷ó ð €	Ø&€	ôNÐ% sô Nô 
Ð!7ô 
ôAÐ!7ô Aô
F˜lõ Fr   