
    Y j                     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_none	deepinfrazhttps://api.deepinfra.comc            	          ^  \ 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$ )DeepInfraTextGenerationTask   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/deepinfra.pyr   $DeepInfraTextGenerationTask.__init__       )i@    mapped_modelapi_keyreturnc                     g)Nz/v1/openai/completions r   r   r    s      r   _prepare_route*DeepInfraTextGenerationTask._prepare_route   s    'r   inputs
parametersprovider_mapping_infoNc                     [        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_dict4DeepInfraTextGenerationTask._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)r:   r;   )generated_textdetails)r   get)r   r5   r6   outputs       r   get_response(DeepInfraTextGenerationTask.get_response   sD    (#I.q1$Vn!'O!<

6*
 	
r   r#   )N)__name__
__module____qualname____firstlineno__r   strr%   r   dictr   r3   bytesr   r@   __static_attributes____classcell__r   s   @r   r   r      s|    A(3 ( ( (XX'+XD\X	X
UT\ 
CTW[C[ 
gj 
 
r   r   c                   >   ^  \ rS rSrU 4S jrS\S\S\4S jrSrU =r$ )DeepInfraConversationalTask'   c                 2   > [         TU ]  [        [        S9  g r   r   r   s    r   r   $DeepInfraConversationalTask.__init__(   r   r   r   r    r!   c                     g)Nz/v1/openai/chat/completionsr#   r$   s      r   r%   *DeepInfraConversationalTask._prepare_route+   s    ,r   r#   )	rB   rC   rD   rE   r   rF   r%   rI   rJ   rK   s   @r   rM   rM   '   s(    A-3 - - - -r   rM   N)typingr   huggingface_hub.hf_apir   !huggingface_hub.inference._commonr   r   _commonr   r	   r
   r   r   r   rM   r#   r   r   <module>rW      s:     ; I P P 	'	
"8 
4-"8 -r   