
    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rU =r$ )NscaleConversationalTask
   c                 "   > [         TU ]  SSS9  g )Nnscale https://inference.api.nscale.com)providerbase_urlsuper__init__self	__class__s    |/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/huggingface_hub/inference/_providers/nscale.pyr   !NscaleConversationalTask.__init__   s    (5WX     )__name__
__module____qualname____firstlineno__r   __static_attributes____classcell__r   s   @r   r   r   
   s    Y Yr   r   c            	          ^  \ 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$ )NscaleTextToImageTask   c                 $   > [         TU ]  SSSS9  g )Nr   r   ztext-to-image)r   r   taskr   r   s    r   r   NscaleTextToImageTask.__init__   s    (5W^mnr   mapped_modelapi_keyreturnc                     g)Nz/v1/images/generationsr   )r   r*   r+   s      r   _prepare_route$NscaleTextToImageTask._prepare_route   s    'r   inputs
parametersprovider_mapping_infoNc                    UR                   n[        U5      nSU;   a.  SU;   a(  UR                  S5       SUR                  S5       3US'   SU;   a  UR                  S5        SU;   a  UR                  S5        SUUS.UEnU$ )	Nwidthheightxsizenum_inference_steps	cfg_scaleb64_json)response_formatpromptmodel)provider_idr
   pop)r   r0   r1   r2   r*   payloads         r   _prepare_payload_as_dict.NscaleTextToImageTask._prepare_payload_as_dict   s     -88 ,
j X%;$.NN7$;#<AjnnX>V=W!XJv J.NN01*$NN;')!
 	
 r   responserequest_paramsc                 V    [        U5      n[        R                  " US   S   S   5      $ )Ndatar   r:   )r   base64	b64decode)r   rC   rD   response_dicts       r   get_response"NscaleTextToImageTask.get_response*   s,     *f 5a 8 DEEr   r   )N)r   r   r   r    r   strr.   r   dictr   rA   bytesr   rJ   r!   r"   r#   s   @r   r%   r%      s|    o(3 ( ( ('+D\	(FUT\ FCTW[C[ Fgj F Fr   r%   )rG   typingr   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r   _commonr   r	   r
   r   r%   r   r   r   <module>rS      s5      ; I L LY5 Y
F. Fr   