o
    ưiE                     @   s$   d Z ddlmZ G dd deZdS )zi
Wandb Chat Completions API - Transformation

This is OpenAI compatible - no translation needed / occurs
    )OpenAIGPTConfigc                
   @   s*   e Zd Zdededededef
ddZdS )	WandbConfignon_default_paramsoptional_paramsmodeldrop_paramsreturnc                 C   sD   | j |d}| D ]\}}|dkr||d< q
||v r|||< q
|S )z?
        map max_completion_tokens param to max_tokens
        )r   Zmax_completion_tokensZ
max_tokens)Zget_supported_openai_paramsitems)selfr   r   r   r   Zsupported_openai_paramsparamvalue r   ]/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/wandb/chat/transformation.pymap_openai_params   s   

zWandbConfig.map_openai_paramsN)__name__
__module____qualname__dictstrboolr   r   r   r   r   r   
   s    r   N)__doc__Z+litellm.llms.openai.chat.gpt_transformationr   r   r   r   r   r   <module>   s    