
    Y jC                     t    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rSr " S S	\	5      r " S
 S\5      rg)    )Any)InferenceProviderMapping)RequestParameters_as_dict   )BaseConversationalTaskBaseTextGenerationTaskfilter_nonezfeatherless-aizhttps://api.featherless.aic            	       n   ^  \ rS rSrU 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$ )FeatherlessTextGenerationTask   c                 2   > [         TU ]  [        [        S9  g N)providerbase_urlsuper__init__	_PROVIDER	_BASE_URLself	__class__s    ڄ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/huggingface_hub/inference/_providers/featherless_ai.pyr   &FeatherlessTextGenerationTask.__init__       )i@    inputs
parametersprovider_mapping_inforeturnNc                     [        UR                  5       5      nUR                  SS 5      US'   SU0UESUR                  0E$ )Nmax_new_tokens
max_tokenspromptmodel)r
   copypopprovider_id)r   r   r   r    paramss        r   _prepare_payload_as_dict6FeatherlessTextGenerationTask._prepare_payload_as_dict   sI     Z__./%zz*:DA|&WFWG5J5V5VWWr   responserequest_paramsc                 v    [        U5      S   S   nUS   UR                  S5      UR                  S5      S.S.$ )Nchoicesr   textfinish_reasonseed)r2   r3   )generated_textdetails)r   get)r   r-   r.   outputs       r   get_response*FeatherlessTextGenerationTask.get_response   sD    (#I.q1$Vn!'O!<

6*
 	
r    )N)__name__
__module____qualname____firstlineno__r   r   dictr   r+   bytesr   r8   __static_attributes____classcell__r   s   @r   r   r      sb    AXX'+XD\X	X
UT\ 
CTW[C[ 
gj 
 
r   r   c                   (   ^  \ rS rSrU 4S jrSrU =r$ )FeatherlessConversationalTask$   c                 2   > [         TU ]  [        [        S9  g r   r   r   s    r   r   &FeatherlessConversationalTask.__init__%   r   r   r:   )r;   r<   r=   r>   r   rA   rB   rC   s   @r   rE   rE   $   s    A Ar   rE   N)typingr   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r   _commonr   r	   r
   r   r   r   rE   r:   r   r   <module>rM      s<     ; I P P 	(	
$: 
.A$: Ar   