
    wif                         d dl mZmZmZmZ d dlmZ d dlmZm	Z	 ddl
mZmZmZ dZdZ G d d	e          Z G d
 de          ZdS )    )AnyDictOptionalUnion)InferenceProviderMapping)RequestParameters_as_dict   )BaseConversationalTaskBaseTextGenerationTaskfilter_nonezfeatherless-aizhttps://api.featherless.aic            	       t     e Zd Z fdZdedededee         fdZdde	e
ef         d	ee         defd
Z xZS )FeatherlessTextGenerationTaskc                 b    t                                          t          t                     d S N)providerbase_urlsuper__init__	_PROVIDER	_BASE_URLself	__class__s    /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/huggingface_hub/inference/_providers/featherless_ai.pyr   z&FeatherlessTextGenerationTask.__init__   &    )i@@@@@    inputs
parametersprovider_mapping_inforeturnc                     t          |                                          }|                    dd           |d<   d|i|d|j        iS )Nmax_new_tokens
max_tokenspromptmodel)r   copypopprovider_id)r   r   r    r!   paramss        r   _prepare_payload_as_dictz6FeatherlessTextGenerationTask._prepare_payload_as_dict   sR     Z__..//%zz*:DAA|&WFWG5J5VWWWr   Nresponserequest_paramsc                     t          |          d         d         }|d         |                    d          |                    d          ddS )Nchoicesr   textfinish_reasonseed)r2   r3   )generated_textdetails)r	   get)r   r-   r.   outputs       r   get_responsez*FeatherlessTextGenerationTask.get_response   sW    (##I.q1$Vn!'O!<!<

6** 
 
 	
r   )N)__name__
__module____qualname__r   r   r   r   r   r,   r   bytesr   r8   __classcell__r   s   @r   r   r      s        A A A A AXX'+XD\X	$X X X X
 
U5$;%7 
RcId 
ps 
 
 
 
 
 
 
 
r   r   c                        e Zd Z fdZ xZS )FeatherlessConversationalTaskc                 b    t                                          t          t                     d S r   r   r   s    r   r   z&FeatherlessConversationalTask.__init__%   r   r   )r9   r:   r;   r   r=   r>   s   @r   r@   r@   $   sA        A A A A A A A A Ar   r@   N)typingr   r   r   r   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r	   _commonr   r   r   r   r   r   r@    r   r   <module>rG      s    - - - - - - - - - - - - ; ; ; ; ; ; I I I I I I I I P P P P P P P P P P 	(	
 
 
 
 
$: 
 
 
.A A A A A$: A A A A Ar   