§
    ³Æj8  ã                   óÔ   — d dl Z d dlmZ 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ZdZ G d	„ d
ee¦  «        Z G d„ de¦  «        Z G d„ de¦  «        Z G d„ de¦  «        ZdS )é    N)ÚABC)ÚAny)Ú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                 ód   •— t          ¦   «                              t          t          |¬¦  «         d S )N)ÚproviderÚbase_urlr   ©ÚsuperÚ__init__Ú	_PROVIDERÚ	_BASE_URL)Úselfr   Ú	__class__s     €úp/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/huggingface_hub/inference/_providers/together.pyr   zTogetherTask.__init__   s(   ø€ Ý‰Œ×Ò¥)µiÀdÐÑKÔKÐKÐKÐKó    Úmapped_modelÚapi_keyÚreturnc                 ó€   — | j         dk    rdS | j         dk    rdS | j         dk    rdS 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   sY   € ØŒ9˜Ò'Ð'Ø+Ð+ØŒYÐ*Ò*Ð*Ø)Ð)ØŒYÐ+Ò+Ð+Ø$Ð$ÝÐL¨d¬iÐLÐLÐLÑMÔMÐMr   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ústrr   r#   Ú__classcell__©r   s   @r   r   r      s…   ø€ € € € € Ø,Ð,ðL˜Sð Lð Lð Lð Lð Lð LðN¨3ð N¸ð NÀð Nð Nð Nð Nð Nð Nð Nð Nr   r   c                   ó@   ‡ — e Zd Zˆ fd„Zddeez  dedz  defd„Zˆ xZ	S )ÚTogetherTextGenerationTaskc                 ób   •— t          ¦   «                              t          t          ¬¦  «         d S ©N)r   r   r   ©r   r   s    €r   r   z#TogetherTextGenerationTask.__init__$   ó&   ø€ Ý‰Œ×Ò¥)µiÐÑ@Ô@Ð@Ð@Ð@r   NÚresponseÚrequest_paramsr   c                 ó    — t          |¦  «        d         d         }|d         |                     d¦  «        |                     d¦  «        dœdœS )NÚchoicesr   ÚtextÚfinish_reasonÚseed)r6   r7   )Úgenerated_textÚdetails)r   Úget)r   r1   r2   Úoutputs       r   Úget_responsez'TogetherTextGenerationTask.get_response'   sW   € Ý˜(Ñ#Ô# IÔ.¨qÔ1ˆà$ Vœnà!'§¢¨OÑ!<Ô!<ØŸ
š
 6Ñ*Ô*ðð ð
ð 
ð 	
r   ©N)
r$   r%   r&   r   ÚbytesÚdictr   r   r<   r)   r*   s   @r   r,   r,   #   sw   ø€ € € € € ðAð Að Að Að Að
ð 
 U¨T¡\ð 
ÐCTÐW[ÑC[ð 
Ðgjð 
ð 
ð 
ð 
ð 
ð 
ð 
ð 
r   r,   c            	       ó@   ‡ — e Zd Zˆ fd„Zdededededz  fˆ fd„Zˆ xZS )ÚTogetherConversationalTaskc                 ób   •— t          ¦   «                              t          t          ¬¦  «         d S r.   r   r/   s    €r   r   z#TogetherConversationalTask.__init__3   r0   r   ÚinputsÚ
parametersÚprovider_mapping_infor   Nc                 óL  •— t          ¦   «                              |||¦  «        }|                     d¦  «        }t          |t          ¦  «        rU|                     d¦  «        dk    r<|                     d¦  «        }t          |t          ¦  «        rd|v rd|d         dœ|d<   |S )NÚresponse_formatÚtypeÚjson_schemaÚschemaÚjson_object)rH   rJ   )r   Ú_prepare_payload_as_dictr:   Ú
isinstancer?   )r   rC   rD   rE   ÚpayloadrG   Újson_schema_detailsr   s          €r   rL   z3TogetherConversationalTask._prepare_payload_as_dict6   s´   ø€ õ ‘'”'×2Ò2°6¸:ÐG\Ñ]Ô]ˆØ$Ÿ.š.Ð):Ñ;Ô;ˆÝo¥tÑ,Ô,ð 	°×1DÒ1DÀVÑ1LÔ1LÐP]Ò1]Ð1]Ø"1×"5Ò"5°mÑ"DÔ"DÐÝÐ-­tÑ4Ô4ð ¸ÐEXÐ9XÐ9Xà)Ø1°(Ô;ð.ð .Ð)Ñ*ð
 ˆr   )	r$   r%   r&   r   r   r?   r   rL   r)   r*   s   @r   rA   rA   2   s   ø€ € € € € ðAð Að Að Að AðØðØ'+ðØD\ðà	‰ðð ð ð ð ð ð ð ð ð r   rA   c            	       ó^   ‡ — e Zd Zˆ 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 )ÚTogetherTextToImageTaskc                 óJ   •— t          ¦   «                              d¦  «         d S )Nr    )r   r   r/   s    €r   r   z TogetherTextToImageTask.__init__G   s!   ø€ Ý‰Œ×Ò˜Ñ)Ô)Ð)Ð)Ð)r   rC   rD   rE   r   Nc                 ó´   — |j         }t          |¦  «        }d|v r|                     d¦  «        |d<   d|v r|                     d¦  «        |d<   |ddœ|¥d|i¥S )NÚnum_inference_stepsÚstepsÚguidance_scaleÚguidanceÚbase64)ÚpromptrG   Úmodel)Úprovider_idr   Úpop)r   rC   rD   rE   r   s        r   rL   z0TogetherTextToImageTask._prepare_payload_as_dictJ   s{   € ð -Ô8ˆÝ  Ñ,Ô,ˆ
Ø  JÐ.Ð.Ø",§.¢.Ð1FÑ"GÔ"GˆJwÑØ˜zÐ)Ð)Ø%/§^¢^Ð4DÑ%EÔ%EˆJzÑ"à °XÐcÐcÀÐcÈWÐVbÐcÐcÐcr   r1   r2   c                 ól   — t          |¦  «        }t          j        |d         d         d         ¦  «        S )NÚdatar   Úb64_json)r   rX   Ú	b64decode)r   r1   r2   Úresponse_dicts       r   r<   z$TogetherTextToImageTask.get_responseV   s/   € Ý  Ñ*Ô*ˆÝÔ ¨fÔ 5°aÔ 8¸Ô DÑEÔEÐEr   r=   )r$   r%   r&   r   r   r?   r   rL   r>   r   r<   r)   r*   s   @r   rQ   rQ   F   s¾   ø€ € € € € ð*ð *ð *ð *ð *ð
dØð
dØ'+ð
dØD\ð
dà	‰ð
dð 
dð 
dð 
dðFð F U¨T¡\ð FÐCTÐW[ÑC[ð FÐgjð Fð Fð Fð Fð Fð Fð Fð Fr   rQ   )rX   Úabcr   Útypingr   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r   Ú,huggingface_hub.inference._providers._commonr   r	   r
   r   r   r   r   r,   rA   rQ   © r   r   ú<module>rh      sm  ðØ €€€Ø Ð Ð Ð Ð Ð Ø Ð Ð Ð Ð Ð à ;Ð ;Ð ;Ð ;Ð ;Ð ;Ø IÐ IÐ IÐ IÐ IÐ IÐ IÐ Iðð ð ð ð ð ð ð ð ð ð ð ð €	Ø&€	ðNð Nð Nð Nð NÐ% sñ Nô Nð Nð 
ð 
ð 
ð 
ð 
Ð!7ñ 
ô 
ð 
ðð ð ð ð Ð!7ñ ô ð ð(Fð Fð Fð Fð F˜lñ Fô Fð Fð Fð Fr   