
    IThU                         S SK JrJrJrJr  S SKJr  S SKJrJ	r	  S SK
JrJr  S SKJr  SrSr " S S	\5      r " S
 S\5      r " S S\5      rg)    )AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict)TaskProviderHelperfilter_none)get_session	replicatezhttps://api.replicate.comc            	          ^  \ rS rSrS\4U 4S jjrS\S\S\4U 4S jjrS\S\S\4S	 jrS
\	S\S\
S\\   4S jrSS\\\4   S\\   S\	4S jjrSrU =r$ )ReplicateTask   taskc                 4   > [         TU ]  [        [        US9  g )N)providerbase_urlr   )super__init__	_PROVIDER	_BASE_URL)selfr   	__class__s     f/var/www/auris/envauris/lib/python3.13/site-packages/huggingface_hub/inference/_providers/replicate.pyr   ReplicateTask.__init__   s    )idK    headersapi_keyreturnc                 0   > [         TU ]  X5      nSUS'   U$ )NwaitPrefer)r   _prepare_headers)r   r   r   r   s      r   r$   ReplicateTask._prepare_headers   s!    '*7<"r   mapped_modelc                     SU;   a  gSU S3$ )N:z/v1/predictionsz/v1/models/z/predictions )r   r&   r   s      r   _prepare_routeReplicateTask._prepare_route   s    ,$\N,77r   inputs
parametersprovider_mapping_infoc                 ~    UR                   nSSU0[        U5      E0nSU;   a  UR                  SS5      S   nXeS'   U$ )Ninputpromptr(      version)provider_idr   split)r   r,   r-   r.   r&   payloadr3   s          r   _prepare_payload_as_dict&ReplicateTask._prepare_payload_as_dict   sT     -88#*Xv,YZAX,Y"Z,"((a03G!(Ir   responserequest_paramsc                    [        U5      nUR                  S5      c  [        SUR                  S5       S35      e[        US   [        5      (       a  US   OUS   S   n[        5       R                  U5      R                  $ )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   r9   r:   response_dict
output_urls        r   get_responseReplicateTask.get_response%   s     *X&.^_l_p_pqx_y^zZ[ 
 (2-2I3'O'OM(#UbckUlmnUo 	 }  ,444r   r)   )N)__name__
__module____qualname____firstlineno__rA   r   r   r$   r*   r   r   r   r7   r   bytesr   rE   __static_attributes____classcell__r   s   @r   r   r      s    LS L s t 
83 8 8 8
'+D\	$
5U5$;%7 
5RcId 
5ps 
5 
5r   r   c            	       N   ^  \ rS rSrU 4S jrS\S\S\S\\   4U 4S jjr	Sr
U =r$ )	ReplicateTextToImageTask2   c                 $   > [         TU ]  S5        g )Nztext-to-imager   r   r   r   s    r   r   !ReplicateTextToImageTask.__init__3   s    )r   r,   r-   r.   r    c                 l   > [         TU ]  XU5      nUR                  b  SUR                   3US   S'   U$ )Nzhttps://huggingface.co/r0   lora_weights)r   r7   adapter_weights_pathhf_model_idr   r,   r-   r.   r6   r   s        r   r7   1ReplicateTextToImageTask._prepare_payload_as_dict6   sF     8Mbc 55A1HI^IjIjHk/lGG^,r   r)   rG   rH   rI   rJ   r   r   r   r   r   r7   rL   rM   rN   s   @r   rP   rP   2   s6    *'+D\	$ r   rP   c            	       N   ^  \ rS rSrU 4S jrS\S\S\S\\   4U 4S jjr	Sr
U =r$ )	ReplicateTextToSpeechTask?   c                 $   > [         TU ]  S5        g )Nztext-to-speechrS   rT   s    r   r   "ReplicateTextToSpeechTask.__init__@   s    )*r   r,   r-   r.   r    c                 \   > [         TU ]  XU5      nUS   R                  S5      US   S'   U$ )Nr0   r1   text)r   r7   poprZ   s        r   r7   2ReplicateTextToSpeechTask._prepare_payload_as_dictC   s;     8Mbc#*7#3#7#7#A r   r)   r\   rN   s   @r   r^   r^   ?   s6    +'+D\	$ r   r^   N)typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   ,huggingface_hub.inference._providers._commonr
   r   huggingface_hub.utilsr   r   r   r   rP   r^   r)   r   r   <module>rk      sI    - - ; I X - 	'	"5& "5J
} 
	 	r   