§
    wiì  ã                   óÒ   — d dl mZmZmZmZ d dlmZ d dlmZm	Z	m
Z
 d dl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 G d„ de¦  «        ZdS )é    )ÚAnyÚDictÚOptionalÚUnion)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dictÚ_as_url)ÚTaskProviderHelperÚfilter_none)Úget_sessionÚ	replicatezhttps://api.replicate.comc            	       ó¶   ‡ — e Zd Zdefˆ fd„Zdededeeef         fˆ fd„Zdedede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 )ÚReplicateTaskÚ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/replicate.pyr   zReplicateTask.__init__   s(   ø€ Ý‰Œ×Ò¥)µiÀdÐÑKÔKÐKÐKÐKó    ÚheadersÚapi_keyÚreturnc                 óV   •— t          ¦   «                              ||¦  «        }d|d<   |S )NÚwaitÚPrefer)r   Ú_prepare_headers)r   r   r   r   s      €r   r#   zReplicateTask._prepare_headers   s+   ø€ Ý‘'”'×*Ò*¨7°GÑ<Ô<ˆØ"ˆÑØˆr   Úmapped_modelc                 ó   — d|v rdS d|› dS )Nú:z/v1/predictionsz/v1/models/z/predictions© )r   r$   r   s      r   Ú_prepare_routezReplicateTask._prepare_route   s$   € Ø,ÐÐØ$Ð$Ø7˜\Ð7Ð7Ð7Ð7r   ÚinputsÚ
parametersÚprovider_mapping_infoc                 óˆ   — |j         }dd|it          |¦  «        ¥i}d|v r!|                     dd¦  «        d         }||d<   |S )NÚinputÚpromptr&   é   Úversion)Úprovider_idr   Úsplit)r   r)   r*   r+   r$   Úpayloadr0   s          r   Ú_prepare_payload_as_dictz&ReplicateTask._prepare_payload_as_dict   s\   € ð -Ô8ˆØ#*¨X°vÐ,YÅÈZÑAXÔAXÐ,YÐ"ZˆØ,ÐÐØ"×(Ò(¨¨aÑ0Ô0°Ô3ˆGØ!(ˆGIÑØˆr   NÚresponseÚrequest_paramsc                 óD  — t          |¦  «        }|                     d¦  «        €&t          d|                     d¦  «        › d¦  «        ‚t          |d         t          ¦  «        r|d         n|d         d         }t          ¦   «                              |¦  «        j        S )NÚoutputzLInference request timed out after 60 seconds. No output generated for model ÚmodelzHThe model might be in cold state or starting up. Please try again later.r   )r	   ÚgetÚTimeoutErrorÚ
isinstanceÚstrr   Úcontent)r   r5   r6   Úresponse_dictÚ
output_urls        r   Úget_responsezReplicateTask.get_response%   s±   € Ý  Ñ*Ô*ˆØ×Ò˜XÑ&Ô&Ð.Ýð[Ð_l×_pÒ_pÐqxÑ_yÔ_yð [ð [ð [ñô ð õ
 (2°-ÀÔ2IÍ3Ñ'OÔ'OÐoˆM˜(Ô#Ð#ÐUbÐckÔUlÐmnÔUoð 	õ ‰}Œ}× Ò  Ñ,Ô,Ô4Ð4r   )N)Ú__name__Ú
__module__Ú__qualname__r=   r   r   r   r#   r(   r   r   r4   r   Úbytesr   rA   Ú__classcell__©r   s   @r   r   r      s%  ø€ € € € € ðL˜Sð Lð Lð Lð Lð Lð Lð¨ð °sð ¸tÀCÈÀH¼~ð ð ð ð ð ð ð
8¨3ð 8¸ð 8Àð 8ð 8ð 8ð 8ð
ØðØ'+ðØD\ðà	$Œðð ð ð ð
5ð 
5 U¨5°$¨;Ô%7ð 
5ÈÐRcÔIdð 
5Ðpsð 
5ð 
5ð 
5ð 
5ð 
5ð 
5ð 
5ð 
5r   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 )ÚReplicateTextToImageTaskc                 óJ   •— t          ¦   «                              d¦  «         d S )Nztext-to-image©r   r   ©r   r   s    €r   r   z!ReplicateTextToImageTask.__init__3   s!   ø€ Ý‰Œ×Ò˜Ñ)Ô)Ð)Ð)Ð)r   r)   r*   r+   r   c                 ó‚   •— t          ¦   «                              |||¦  «        }|j        d|j        › |d         d<   |S )Nzhttps://huggingface.co/r-   Úlora_weights)r   r4   Úadapter_weights_pathÚhf_model_id©r   r)   r*   r+   r3   r   s        €r   r4   z1ReplicateTextToImageTask._prepare_payload_as_dict6   sJ   ø€ õ ™œ×8Ò8¸ÀÐMbÑcÔcˆØ Ô5ÐAØ/lÐI^ÔIjÐ/lÐ/lˆGGÔ˜^Ñ,Øˆr   ©
rB   rC   rD   r   r   r   r   r   r4   rF   rG   s   @r   rI   rI   2   sz   ø€ € € € € ð*ð *ð *ð *ð *ðØðØ'+ðØD\ðà	$Œðð ð ð ð ð ð ð ð ð r   rI   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 )ÚReplicateTextToSpeechTaskc                 óJ   •— t          ¦   «                              d¦  «         d S )Nztext-to-speechrK   rL   s    €r   r   z"ReplicateTextToSpeechTask.__init__@   ó"   ø€ Ý‰Œ×ÒÐ)Ñ*Ô*Ð*Ð*Ð*r   r)   r*   r+   r   c                 ó–   •— t          ¦   «                              |||¦  «        }|d                              d¦  «        |d         d<   |S )Nr-   r.   Útext)r   r4   ÚpoprQ   s        €r   r4   z2ReplicateTextToSpeechTask._prepare_payload_as_dictC   sH   ø€ õ ™œ×8Ò8¸ÀÐMbÑcÔcˆØ#*¨7Ô#3×#7Ò#7¸Ñ#AÔ#AˆÔ˜Ñ Øˆr   rR   rG   s   @r   rT   rT   ?   sz   ø€ € € € € ð+ð +ð +ð +ð +ðØðØ'+ðØD\ðà	$Œðð ð ð ð ð ð ð ð ð r   rT   c            	       óB   ‡ — e Zd Zˆ fd„Zdedededee         fd„Zˆ xZ	S )ÚReplicateImageToImageTaskc                 óJ   •— t          ¦   «                              d¦  «         d S )Nzimage-to-imagerK   rL   s    €r   r   z"ReplicateImageToImageTask.__init__L   rV   r   r)   r*   r+   r   c                 óª   — t          |d¬¦  «        }dd|it          |¦  «        ¥i}|j        }d|v r!|                     dd¦  «        d         }||d<   |S )Nz
image/jpeg)Údefault_mime_typer-   Úinput_imager&   r/   r0   )r
   r   r1   r2   )r   r)   r*   r+   Ú	image_urlr3   r$   r0   s           r   r4   z2ReplicateImageToImageTask._prepare_payload_as_dictO   so   € õ ˜F°lÐCÑCÔCˆ	à#*¨]¸IÐ,aÍÐU_ÑI`ÔI`Ð,aÐ"bˆà,Ô8ˆØ,ÐÐØ"×(Ò(¨¨aÑ0Ô0°Ô3ˆGØ!(ˆGIÑØˆr   rR   rG   s   @r   r[   r[   K   sp   ø€ € € € € ð+ð +ð +ð +ð +ðØðØ'+ðØD\ðà	$Œðð ð ð ð ð ð ð r   r[   N)Útypingr   r   r   r   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r	   r
   Ú,huggingface_hub.inference._providers._commonr   r   Úhuggingface_hub.utilsr   r   r   r   rI   rT   r[   r'   r   r   ú<module>rf      sC  ðØ -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -Ð -à ;Ð ;Ð ;Ð ;Ð ;Ð ;Ø RÐ RÐ RÐ RÐ RÐ RÐ RÐ RÐ RÐ RØ XÐ XÐ XÐ XÐ XÐ XÐ XÐ XØ -Ð -Ð -Ð -Ð -Ð -ð €	Ø'€	ð"5ð "5ð "5ð "5ð "5Ð&ñ "5ô "5ð "5ðJ
ð 
ð 
ð 
ð 
˜}ñ 
ô 
ð 
ð	ð 	ð 	ð 	ð 	 ñ 	ô 	ð 	ðð ð ð ð  ñ ô ð ð ð r   