
    ^hf                         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rSr " S S	\5      r " S
 S\5      rg)    )AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict   )BaseConversationalTaskBaseTextGenerationTaskfilter_nonezfeatherless-aizhttps://api.featherless.aic            	       r   ^  \ rS rSrU 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$ )FeatherlessTextGenerationTask   c                 2   > [         TU ]  [        [        S9  g N)providerbase_urlsuper__init__	_PROVIDER	_BASE_URLself	__class__s    m/home/james-whalen/.local/lib/python3.13/site-packages/huggingface_hub/inference/_providers/featherless_ai.pyr   &FeatherlessTextGenerationTask.__init__       )i@    inputs
parametersprovider_mapping_inforeturnc                     [        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)r5   r6   )generated_textdetails)r	   get)r   r0   r1   outputs       r   get_response*FeatherlessTextGenerationTask.get_response   sD    (#I.q1$Vn!'O!<

6*
 	
r     )N)__name__
__module____qualname____firstlineno__r   r   r   r   r   r.   r   bytesr   r;   __static_attributes____classcell__r   s   @r   r   r      sf    AXX'+XD\X	$X
U5$;%7 
RcId 
ps 
 
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@   rA   r   rC   rD   rE   s   @r   rG   rG   $   s    A Ar    rG   N)typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   _commonr   r   r   r   r   r   rG   r=   r    r   <module>rO      s?    - - ; I P P 	(	
$: 
.A$: Ar    