Ë
    ÷¾™hÿ  ã                   ó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                  «       }|j                  «       D ]%  \  }}|dk7  sŒ|€Œt        | j                  ||«       Œ' y )NÚself)ÚlocalsÚcopyÚitemsÚsetattrÚ	__class__)r   r   r   r   r	   Úlocals_ÚkeyÚvalues           úY/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/nvidia_nim/embed.pyÚ__init__z!NvidiaNimEmbeddingConfig.__init__   sA   € ô “(—-‘-“/ˆØ!Ÿ-™-ž/‰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(   st   € ð Ÿ™×*Ñ*Ó,÷
á1Ø—<‘< Ô%ÜØä×&Ñ&Ü×-Ñ-ÜÜ ð	ôð ð ˆq‰Dó
ð 	
ùó 
s   žAA6c                 ó
   — g d¢S )N)r   r   Ú
dimensions© )r   s    r   Úget_supported_openai_paramsz4NvidiaNimEmbeddingConfig.get_supported_openai_params:   s
   € ò 9Ð8r   Únon_default_paramsÚoptional_paramsÚkwargsc                 óî   — d|vri |d<   |j                  «       D ]B  \  }}|dk(  r|d   j                  d|i«       Œ"|dk(  r|d   j                  d|i«       Œ>|||<   ŒD ||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)   Údictr0   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ð ñ
ó ð
ò"9ð "&ñ	à ðð ðð ˜‘ô	r   r   )r4   r   Útypingr   r   r(   r   r   ú<module>r9      s   ðñó Ý ÷Cò Cr   