Ë
    uPêj®
  ã                   óB   — d Z ddlmZ ddlmZ ddlmZ  G d„ de«      Zy)z[
Cerebras Chat Completions API

this is OpenAI compatible - no translation needed / occurs
é    )ÚFinal)ÚOpenAIGPTConfig)Úsupports_reasoningc                   ó   ‡ — 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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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dededededef
d„Zˆ xZS )ÚCerebrasConfigzu
    Reference: https://inference-docs.cerebras.ai/api-reference/chat-completions

    Below are the parameters:
    NÚ
max_tokensÚresponse_formatÚseedÚstreamÚtop_pÚtool_choiceÚtoolsÚuserÚreasoning_effortÚstopÚtemperatureÚ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                  ú_/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/cerebras/chat.pyÚ__init__zCerebrasConfig.__init__   sF   € ô  ›Ÿ™›ˆØ!Ÿ-™-›/ò 	4‰JˆC�Ø�f‹} Ñ!2Ü˜Ÿ™¨¨UÕ3ñ	4ó    c                 ó    •— t         ‰| �  «       S )N)ÚsuperÚ
get_config)Úclsr   s    €r   r#   zCerebrasConfig.get_config1   s   ø€ ä‰wÑ!Ó#Ð#r    Úmodelc                 óJ   — g d¢}t        |d¬«      r|j                  d«       |S )zF
        Get the supported OpenAI params for the given model

        )r   Úmax_completion_tokensr	   r
   r   r   r   r   r   r   r   Úmax_retriesÚextra_headersÚcerebras)r%   Úcustom_llm_providerr   )r   Úappend)r   r%   Úsupported_paramss      r   Úget_supported_openai_paramsz*CerebrasConfig.get_supported_openai_params5   s+   € ò#
Ðô"  E¸zÕJØ×#Ñ#Ð$6Ô7àÐr    Únon_default_paramsÚoptional_paramsÚdrop_paramsc                 ó„   — | j                  |¬«      }|j                  «       D ]  \  }}|dk(  r||d<   Œ||v sŒ|||<   Œ |S )N)r%   r'   r   )r.   r   )r   r/   r0   r%   r1   Úsupported_openai_paramsÚparamr   s           r   Úmap_openai_paramsz CerebrasConfig.map_openai_paramsQ   sc   € ð *.×)IÑ)IÐPUÐ)IÓ)VÐØ.×4Ñ4Ó6ò 	/‰LˆE�5ØÐ/Ò/Ø05� Ò-ØÐ1Ò1Ø).� Ò&ð		/ð
 Ðr    )NNNNNNNNNNN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚintÚ__annotations__r	   Údictr
   r   Úboolr   r   Ústrr   Úlistr   r   Úfloatr   Úclassmethodr#   r.   r5   Ú__classcell__)r   s   @r   r   r      s¯  ø… ñð "€J��d‘
Ó!Ø#'€O�T˜D‘[Ó'Ø€Dˆ#�‰*ÓØ€FˆD�4‰KÓØ€Eˆ3�‰:ÓØ"€K��t‘Ó"Ø€Eˆ4�$‰;ÓØ€Dˆ#�‰*ÓØ#'Ð�c˜D‘jÓ'ð "&Ø'+ØØØ"Ø$(Ø Ø"&Ø!ØØ'+ñ4à˜$‘Jð4ð  ™ð4ð �D‰jð	4ð
 �D‰jð4ð �t‘ð4ð ˜T‘\ð4ð �T‰zð4ð ˜4‘Zð4ð �d‰{ð4ð �D‰jð4ð  ™*ð4ð 
ó4ð& ó$ó ð$ð °ð  ¸ó  ð8à ðð ðð ð	ð
 ðð 
÷r    r   N)r9   Útypingr   Ú+litellm.llms.openai.chat.gpt_transformationr   Úlitellm.utilsr   r   © r    r   ú<module>rG      s#   ðñõ å GÝ ,ôQ�_õ Qr    