
    RX9i
                         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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    k/var/www/html/land-doc-ocr/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\
\   4S
 jrSS\\\4   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_infoc                    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   r3   r4   r5   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   rF   rG   response_dicts       r   get_response"NscaleTextToImageTask.get_response*   s,     *f 5a 8 DEEr   r   )N)r    r!   r"   r#   r   strr1   r   r   r   r   rD   r   bytesr   rM   r$   r%   r&   s   @r   r(   r(      s    o(3 ( ( ('+D\	$(FU5$;%7 FRcId Fps F Fr   r(   )rJ   typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   _commonr   r   r   r   r(   r   r   r   <module>rU      s8     - - ; I L LY5 Y
F. Fr   