Ë
    uPêjE  ã                   ó6   — d Z ddlmZ ddlmZ  G d„ de«      Zy)zâ
Nvidia NIM endpoint: https://docs.api.nvidia.com/nim/reference/databricks-dbrx-instruct-infer

This is OpenAI compatible

This file only contains param mapping logic

API calling is done using the OpenAI SDK with an api_base
é    )ÚFinal)ÚOpenAIGPTConfigc            
       ó<   — e Zd ZdZdedefd„Zdededededef
d„Z	y	)
ÚNvidiaNimConfigzä
    Reference: https://docs.api.nvidia.com/nim/reference/databricks-dbrx-instruct-infer

    The class `NvidiaNimConfig` provides configuration for the Nvidia NIM's Chat Completions API interface. Below are the parameters:
    ÚmodelÚreturnc                 óL   — |dv rg d¢S |dk(  rg d¢S |dk(  rdgS |dv rg d¢S g d	¢S )
zž
        Get the supported OpenAI params for the given model


        Updated on July 5th, 2024 - based on https://docs.api.nvidia.com/nim/reference
        )zgoogle/recurrentgemma-2bzgoogle/gemma-2-27b-itzgoogle/gemma-2-9b-itzgemma-2-9b-it)ÚstreamÚtemperatureÚtop_pÚ
max_tokensÚstopÚseedznvidia/nemotron-4-340b-instruct)r
   r   r   r   Úmax_completion_tokensznvidia/nemotron-4-340b-rewardr
   )zgoogle/codegemma-1.1-7b)r
   r   r   Úfrequency_penaltyÚpresence_penaltyr   r   r   )r
   r   r   r   r   r   r   r   r   ÚtoolsÚtool_choiceÚparallel_tool_callsÚresponse_format© )Úselfr   s     úp/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/nvidia_nim/chat/transformation.pyÚget_supported_openai_paramsz+NvidiaNimConfig.get_supported_openai_params   se   € ð ð 
ñ 
ò TÐSØÐ7Ò7òð ð Ð5Ò5àðð ð Ð1Ñ1ò	ð 	òBð ó    Únon_default_paramsÚoptional_paramsÚdrop_paramsc                 ó„   — | j                  |¬«      }|j                  «       D ]  \  }}|dk(  r||d<   Œ||v sŒ|||<   Œ |S )N)r   r   r   )r   Úitems)r   r   r   r   r   Úsupported_openai_paramsÚparamÚvalues           r   Úmap_openai_paramsz!NvidiaNimConfig.map_openai_paramsd   sc   € ð *.×)IÑ)IÐPUÐ)IÓ)VÐØ.×4Ñ4Ó6ò 	/‰LˆE�5ØÐ/Ò/Ø05� Ò-ØÐ1Ò1Ø).� Ò&ð		/ð
 Ðr   N)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstrÚlistr   ÚdictÚboolr$   r   r   r   r   r      sS   „ ñðK°ð K¸ó KðZà ðð ðð ð	ð
 ðð 
ôr   r   N)r(   Útypingr   Ú+litellm.llms.openai.chat.gpt_transformationr   r   r   r   r   ú<module>r/      s    ðñõ å Gôa�oõ ar   