
    ^h                     0    S r SSKrSSKJr   " S S5      rg)z
Nvidia NIM embeddings endpoint: https://docs.api.nvidia.com/nim/reference/nvidia-nv-embedqa-e5-v5-infer

This is OpenAI compatible 

This file only contains param mapping logic

API calling is done using the OpenAI SDK with an api_base
    N)Optionalc                       \ rS rSr% SrSr\\   \S'   Sr	\\   \S'   Sr
\\   \S'   Sr\\   \S'       SS\\   S\\   S\\   S\\   SS4
S	 jjr\S
 5       rS r SS\S\S\\   4S jjrSrg)NvidiaNimEmbeddingConfig   zT
Reference: https://docs.api.nvidia.com/nim/reference/nvidia-nv-embedqa-e5-v5-infer
Nencoding_formatuser
input_typetruncatereturnc                     [        5       nUR                  5        H(  u  pgUS:w  d  M  Uc  M  [        U R                  Xg5        M*     g )Nself)localsitemssetattr	__class__)r   r   r   r	   r
   locals_keyvalues           W/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/nvidia_nim/embed.py__init__!NvidiaNimEmbeddingConfig.__init__   s7     (!--/JCf}!23 *    c                    U R                   R                  5        VVs0 s H_  u  pUR                  S5      (       a  M  [        U[        R
                  [        R                  [        [        45      (       a  MX  Uc  M]  X_Ma     snn$ s  snnf )N__)	__dict__r   
startswith
isinstancetypesFunctionTypeBuiltinFunctionTypeclassmethodstaticmethod)clskvs      r   
get_config#NvidiaNimEmbeddingConfig.get_config(   s     **,
,<<%  &&-- 	   AD,
 	
 
s   B7B9B>Bc                 
    SS/$ )Nr   r    )r   s    r   get_supported_openai_params4NvidiaNimEmbeddingConfig.get_supported_openai_params:   s     "6**r   non_default_paramsoptional_paramskwargsc                     SU;  a  0 US'   UR                  5        HA  u  pEUS:X  a  US   R                  SU05        M#  US:X  d  M+  US   R                  SU05        MC     Ub  US   R                  U5        U$ )N
extra_bodyr	   r
   )r   update)r   r,   r-   r.   r$   r%   s         r   map_openai_params*NvidiaNimEmbeddingConfig.map_openai_params?   s     .,.OL)&,,.DAL -44lA5FGj-44j!_E	 / L)008r   r)   )NNNN)N)__name__
__module____qualname____firstlineno____doc__r   r   str__annotations__r   r	   r
   r   r!   r&   r*   dictr2   __static_attributes__r)   r   r   r   r      s    
 &*OXc])D(3- !%J$"Hhsm" *."$("&
4!#
4 sm
4 SM	
4
 3-
4 

4 
 
"+ "&	   	 r   r   )r8   r   typingr   r   r)   r   r   <module>r>      s     A Ar   