Ë
    uPêj+  ã                   ó6   — d Z ddl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
é    )ÚFinalé   )ÚOpenAILikeChatConfigc                   ó¤  ‡ — 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Zeez  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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edz  deez  dz  dedz  d	edz  d
edz  dedz  dedz  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                  új/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/ai21/chat/transformation.pyÚ__init__zAI21ChatConfig.__init__   sF   € ô  ›Ÿ™›ˆØ!Ÿ-™-›/ò 	4‰JˆC�Ø�f‹} Ñ!2Ü˜Ÿ™¨¨UÕ3ñ	4ó    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   ÚlistÚ__annotations__r	   Údictr
   r   Úintr   Ústrr   r   Úboolr   r   r   Úfloatr    Úclassmethodr$   r*   Ú__classcell__)r   s   @r   r   r      s‘  ø… ñð €Eˆ4�$‰;ÓØ#'€O�T˜D‘[Ó'Ø!€Iˆt�d‰{Ó!Ø!€J��d‘
Ó!Ø"€Dˆ#�‰*�tÑ
Ó"Ø€A€sˆT�zÓØ€FˆD�4‰KÓØ€Dˆ#�‰*ÓØ"€K��t‘Ó"Ø€Dˆ#�‰*Óð "Ø'+Ø!%Ø$(Ø"Ø"&ØØ"ØØ"&Øñ4à�d‰{ð4ð  ™ð4ð ˜$‘Jð	4ð
 ˜T‘\ð4ð �t‰|ð4ð �D‰j˜4Ñð4ð �‰:ð4ð �t‘ð4ð �D‰jð4ð ˜4‘Zð4ð �D‰jð4ð 
ó4ð& ó$ó ð$ð
°ð 
¸÷ 
r!   r   N)r.   Útypingr   Úopenai_like.chat.transformationr   r   r)   r!   r   ú<module>r:      s   ðñõ å Cô:
Ð)õ :
r!   