
    Y j                     t    S SK r S SK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)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\S
-  4S jr
SS\\-  S\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    ڀ/root/GenerationalWealth/GenerationalWealth/venv/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_infoNc                     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   r1   r2   response_dicts       r   get_response&HyperbolicTextToImageTask.get_response    s,     *h 7 :7 CDDr   r   )N)__name__
__module____qualname____firstlineno__r   strr   r   dictr   r/   bytesr   r9   __static_attributes____classcell__r   s   @r   r   r   	   s    m(3 ( ( (LL'+LD\L	L EUT\ ECTW[C[ Egj 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=   r>   __doc__r?   r   rB   rC   rD   s   @r   rF   rF   %   s    S  r   rF   )r6   typingr   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r   ,huggingface_hub.inference._providers._commonr   r   r	   r   rF   r   r   r   <module>rO      s3      ; I p pE 2 E8
#9 
r   