
    ITh                         S SK r 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Jr   " S S\5      r " S S	\5      rg)
    N)AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict)BaseConversationalTaskTaskProviderHelperfilter_nonec            	          ^  \ rS rSrU 4S 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$ )HyperbolicTextToImageTask	   c                 $   > [         TU ]  SSSS9  g )N
hyperbolichttps://api.hyperbolic.xyzztext-to-image)providerbase_urltask)super__init__)self	__class__s    g/var/www/auris/envauris/lib/python3.13/site-packages/huggingface_hub/inference/_providers/hyperbolic.pyr   "HyperbolicTextToImageTask.__init__
   s    ,9U\kl    mapped_modelapi_keyreturnc                     g)Nz/v1/images/generations )r   r   r   s      r   _prepare_route(HyperbolicTextToImageTask._prepare_route   s    'r   inputs
parametersprovider_mapping_infoc                     UR                   n[        U5      nSU;   a  UR                  S5      US'   SU;   a  UR                  S5      US'   SU;  a  SUS'   SU;  a  SUS'   XS.UE$ )	Nnum_inference_stepsstepsguidance_scale	cfg_scalewidthi   height)prompt
model_name)provider_idr   pop)r   r$   r%   r&   r   s        r   _prepare_payload_as_dict2HyperbolicTextToImageTask._prepare_payload_as_dict   s     -88 ,
 J.",..1F"GJwz)&0nn5E&FJ{#*$"%Jw:%#&Jx  K
KKr   responserequest_paramsc                 V    [        U5      n[        R                  " US   S   S   5      $ )Nimagesr   image)r	   base64	b64decode)r   r4   r5   response_dicts       r   get_response&HyperbolicTextToImageTask.get_response    s,     *h 7 :7 CDDr   r!   )N)__name__
__module____qualname____firstlineno__r   strr"   r   r   r   r   r2   r   bytesr   r<   __static_attributes____classcell__r   s   @r   r   r   	   s    m(3 ( ( (LL'+LD\L	$L EU5$;%7 ERcId Eps E Er   r   c                   4   ^  \ rS rSrSrS\4U 4S jjrSrU =r$ )HyperbolicTextGenerationTask%   z^
Special case for Hyperbolic, where text-generation task is handled as a conversational task.
r   c                 .   > [         TU ]  SSS9  Xl        g )Nr   r   )r   r   )r   r   r   )r   r   r   s     r   r   %HyperbolicTextGenerationTask.__init__*   s"    !1 	 	
 	r   )r   )	r>   r?   r@   rA   __doc__rB   r   rD   rE   rF   s   @r   rH   rH   %   s    S  r   rH   )r9   typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   ,huggingface_hub.inference._providers._commonr
   r   r   r   rH   r!   r   r   <module>rQ      s6     - - ; I p pE 2 E8
#9 
r   