Ë
    uPêjç  ã                   ó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)ÚFinalc                   óÔ   — e Zd ZU dZdZedz  ed<   dZedz  ed<   dZedz  ed<   dZ	edz  ed<   	 	 	 	 ddedz  dedz  dedz  dedz  ddf
d„Z
ed	„ «       Zd
„ Z	 ddedededz  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           úb/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/nvidia_nim/embed.pyÚ__init__z!NvidiaNimEmbeddingConfig.__init__   sF   € ô  ›Ÿ™›ˆØ!Ÿ-™-›/ò 	4‰JˆC�Ø�f‹} Ñ!2Ü˜Ÿ™¨¨UÕ3ñ	4ó    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   ÚstrÚ__annotations__r   r   r	   r   r    r%   r)   Údictr0   r(   r   r   r   r      sÙ   … ñð
 #'€O�S˜4‘ZÓ&Ø€Dˆ#�‰*Óð "€J��d‘
Ó!Ø€Hˆc�D‰jÓð '+ØØ!%Ø#ñ
4à˜t™ð
4ð �D‰jð
4ð ˜$‘Jð	
4ð
 ˜‘*ð
4ð 
ó
4ð ñ
ó ð
ò"9ð #ñ	à ðð ðð �t‘ô	r   r   )r4   r   Útypingr   r   r(   r   r   ú<module>r9      s   ðñó Ý ÷Cò Cr   