
    ^hy                     :    S r SSKJrJr  SSKJr   " S S\5      rg)zW
AI21 Chat Completions API

this is OpenAI compatible - no translation needed / occurs
    )OptionalUnion   )OpenAILikeChatConfigc                     ^  \ rS rSr% SrSr\\   \S'   Sr	\\
   \S'   Sr\\   \S'   Sr\\   \S'   Sr\\\\4      \S'   Sr\\   \S	'   Sr\\   \S
'   Sr\\   \S'   Sr\\   \S'   Sr\\   \S'              SS\\   S\\
   S\\   S\\   S\\   S\\\\4      S	\\   S
\\   S\\   S\\   S\\   SS4S jjr\U 4S j5       rS\S\4S jrSrU =r$ )AI21ChatConfig   zk
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                     [        5       R                  5       nUR                  5        H(  u  pUS:w  d  M  Uc  M  [        U R                  X5        M*     g )Nself)localscopyitemssetattr	__class__)r   r
   r   r   r   r   r   r   r   r   r   r   locals_keyvalues                  _/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/ai21/chat/transformation.py__init__AI21ChatConfig.__init__   s=     (--/!--/JCf}!23 *    c                     > [         TU ]  5       $ )N)super
get_config)clsr   s    r!   r'   AI21ChatConfig.get_config1   s    w!##r$   modelc                 
    / SQ$ )z6
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_params*AI21ChatConfig.get_supported_openai_params5   s    
 	
r$   r-   )NNNNNNNNNNN)__name__
__module____qualname____firstlineno____doc__r
   r   list__annotations__r   dictr   r   intr   r   strr   r   boolr   r   r   floatr"   classmethodr'   r.   __static_attributes____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)r4   typingr   r   openai_like.chat.transformationr   r   r-   r$   r!   <module>rA      s    # C:
) :
r$   