
    gy                     :    d Z ddlmZmZ ddlmZ  G d de      Zy)zW
AI21 Chat Completions API

this is OpenAI compatible - no translation needed / occurs
    )OptionalUnion   )OpenAILikeChatConfigc                       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Zeeeef      ed<   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Zee   ed<   	 	 	 	 	 	 	 	 	 	 	 ddee   dee	   dee   dee   dee   deeeef      dee   d	ee   d
ee   dee   dee   ddfdZe fd       ZdedefdZ xZS )AI21ChatConfigzw
    Reference: https://docs.ai21.com/reference/jamba-15-api-ref#request-parameters

    Below are the parameters:
    Ntoolsresponse_format	documents
max_tokensstopnstreamseedtool_choiceusertemperaturetop_preturnc                     t               j                         }|j                         D ]%  \  }}|dk7  s|t        | j                  ||       ' y )Nself)localscopyitemssetattr	__class__)r   r	   r
   r   r   r   r   r   r   r   r   r   locals_keyvalues                  Z/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/ai21/chat/transformation.py__init__zAI21ChatConfig.__init__   sA     (--/!--/JCf}!2U3 *    c                      t         |          S )N)super
get_config)clsr   s    r    r%   zAI21ChatConfig.get_config1   s    w!##r"   modelc                 
    g dS )zF
        Get the supported OpenAI params for the given model

        )
r	   r
   r   max_completion_tokensr   r   r   r   r   r    )r   r'   s     r    get_supported_openai_paramsz*AI21ChatConfig.get_supported_openai_params5   s    
 	
r"   )NNNNNNNNNNN)__name__
__module____qualname____doc__r	   r   list__annotations__r
   dictr   r   intr   r   strr   r   boolr   r   r   floatr!   classmethodr%   r+   __classcell__)r   s   @r    r   r      s    !E8D> &*OXd^* $Ix~$ $J$'+D(5d#
$+Ax}!FHTN!D(3-!%K#%D(3- !%*.$('+!%+/!%"%)"4~4 "$4 SM	4
 e_4 4 uS$Y'(4 C=4 4 sm4 c]4 sm4 
4& $ $
 
 
r"   r   N)r/   typingr   r   openai_like.chat.transformationr   r   r*   r"   r    <module>r;      s    # C:
) :
r"   