§
    wio  ã                   óà   — 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dS )é    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                 ód   •— t          ¦   «                              t          t          |¬¦  «         d S )N)ÚproviderÚbase_urlr   ©ÚsuperÚ__init__Ú	_PROVIDERÚ	_BASE_URL)Úselfr   Ú	__class__s     €ú…/var/www/development/aibuddy-work/election-extract/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                   óP   ‡ — 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                 ó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)r9   r:   )Úgenerated_textÚdetails)r
   Úget)r   r4   r5   Úoutputs       r   Úget_responsez'TogetherTextGenerationTask.get_response'   sW   € Ý˜(Ñ#Ô# 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/   #   s{   ø€ € € € € ðAð Að Að Að Að
ð 
 U¨5°$¨;Ô%7ð 
ÈÐRcÔIdð 
Ðpsð 
ð 
ð 
ð 
ð 
ð 
ð 
ð 
r   r/   c            	       óF   ‡ — e Zd Zˆ fd„Zdedededee         fˆ fd„Zˆ xZ	S )ÚTogetherConversationalTaskc                 ób   •— t          ¦   «                              t          t          ¬¦  «         d S r1   r   r2   s    €r   r   z#TogetherConversationalTask.__init__3   r3   r   ÚinputsÚ
parametersÚprovider_mapping_infor!   c                 ó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)rJ   rL   )r   Ú_prepare_payload_as_dictr=   Ú
isinstanceÚdict)r   rE   rF   rG   ÚpayloadrI   Újson_schema_detailsr   s          €r   rN   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   r   rN   r,   r-   s   @r   rC   rC   2   s   ø€ € € € € ðAð Að Að Að AðØðØ'+ðØD\ðà	$Œðð ð ð ð ð ð ð ð ð r   rC   c            	       ót   ‡ — 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                 óJ   •— t          ¦   «                              d¦  «         d S )Nr#   )r   r   r2   s    €r   r   z TogetherTextToImageTask.__init__G   s!   ø€ Ý‰Œ×Ò˜Ñ)Ô)Ð)Ð)Ð)r   rE   rF   rG   r!   c                 ó´   — |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)ÚpromptrI   Úmodel)Úprovider_idr   Úpop)r   rE   rF   rG   r   s        r   rN   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   Nr4   r5   c                 ól   — t          |¦  «        }t          j        |d         d         d         ¦  «        S )NÚdatar   Úb64_json)r
   r[   Ú	b64decode)r   r4   r5   Ú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   r   rN   r   rA   r	   r?   r,   r-   s   @r   rT   rT   F   sÂ   ø€ € € € € ð*ð *ð *ð *ð *ð
dØð
dØ'+ð
dØD\ð
dà	$Œð
dð 
dð 
dð 
dðFð F U¨5°$¨;Ô%7ð FÈÐRcÔIdð FÐpsð Fð Fð Fð Fð Fð Fð Fð Fr   rT   )r[   Ú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   rT   © r   r   ú<module>rk      s  ðØ €€€Ø Ð Ð Ð Ð Ð Ø -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -à ;Ð ;Ð ;Ð ;Ð ;Ð ;Ø 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   