Ë
    Ýªg±  ã                   ó0   — d Z ddlZddlmZ  G d„ d«      Zy)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                   óÔ   — e Zd ZU dZdZee   ed<   dZee   ed<   dZ	ee   ed<   dZ
ee   ed<   	 	 	 	 ddee   dee   dee   dee   ddf
d„Zed	„ «       Zd
„ Z	 ddededee   fd„Zy)ÚNvidiaNimEmbeddingConfigz\
    Reference: https://docs.api.nvidia.com/nim/reference/nvidia-nv-embedqa-e5-v5-infer
    NÚencoding_formatÚuserÚ
input_typeÚtruncateÚreturnc                 óˆ   — t        «       }|j                  «       D ]%  \  }}|dk7  sŒ|€Œt        | j                  ||«       Œ' y )NÚself)ÚlocalsÚitemsÚsetattrÚ	__class__)r   r   r   r   r	   Úlocals_ÚkeyÚvalues           úR/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/nvidia_nim/embed.pyÚ__init__z!NvidiaNimEmbeddingConfig.__init__   s;   € ô “(ˆØ!Ÿ-™-ž/‰JˆCØf‹} Ñ!2Ü˜Ÿ™¨¨UÕ3ñ *ó    c                 óø   — | j                   j                  «       D ci c]O  \  }}|j                  d«      s9t        |t        j
                  t        j                  t        t        f«      s|||“ŒQ c}}S c c}}w )NÚ__)	Ú__dict__r   Ú
startswithÚ
isinstanceÚtypesÚFunctionTypeÚBuiltinFunctionTypeÚclassmethodÚstaticmethod)ÚclsÚkÚvs      r   Ú
get_configz#NvidiaNimEmbeddingConfig.get_config(   sy   € ð Ÿ™×*Ñ*Ô,ô
á,‘1Ø—<‘< Ô%ÜØä×&Ñ&Ü×-Ñ-ÜÜ ð	ôð ð ˆq‰DØ,ò
ð 	
ùó 
s   žAA6c                 ó
   — ddgS )Nr   r   © )r   s    r   Úget_supported_openai_paramsz4NvidiaNimEmbeddingConfig.get_supported_openai_params:   s   € ð " 6Ð*Ð*r   Únon_default_paramsÚoptional_paramsÚkwargsc                 óä   — d|vri |d<   |j                  «       D ]=  \  }}|dk(  r|d   j                  d|i«       Œ"|dk(  sŒ(|d   j                  d|i«       Œ? ||d   j                  |«       |S )NÚ
extra_bodyr   r	   )r   Úupdate)r   r(   r)   r*   r"   r#   s         r   Úmap_openai_paramsz*NvidiaNimEmbeddingConfig.map_openai_params?   sŒ   € ð ˜Ñ.Ø,.ˆO˜LÑ)Ø&×,Ñ,Ö.‰DˆAˆqØLÒ Ø Ñ-×4Ñ4°lÀAÐ5FÕGØj“Ø Ñ-×4Ñ4°jÀ!°_ÕEð	 /ð Ðà˜LÑ)×0Ñ0°Ô8ØÐr   )NNNN)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚstrÚ__annotations__r   r   r	   r   r   r$   r'   Údictr.   r&   r   r   r   r      sÙ   … ñð
 &*€OX˜c‘]Ó)Ø€Dˆ(3‰-Óð !%€J˜‘Ó$Ø"€Hˆhs‰mÓ"ð *.Ø"Ø$(Ø"&ñ
4à! #™ð
4ð s‰mð
4ð ˜S‘Mð	
4ð
 ˜3‘-ð
4ð 
ó
4ð ñ
ó ð
ò"+ð "&ñ	à ðð ðð ˜‘ô	r   r   )r2   r   Útypingr   r   r&   r   r   Ú<module>r7      s   ðñó Ý ÷Aò Ar   