o ưi @s<dZddlmZddlmZddlmZGdddeZdS)z[ Cerebras Chat Completions API this is OpenAI compatible - no translation needed / occurs )Optional)OpenAIGPTConfig)supports_reasoningcsdeZdZUdZdZeeed<dZee ed<dZ eeed<dZ ee ed<dZ eeed<dZeeed<dZeeed <dZeeed <dZeeed <           ddeedee deed eedee d eedeedeed eed eed eeddfddZefddZdedefddZde de dede de f ddZZS)CerebrasConfigzu Reference: https://inference-docs.cerebras.ai/api-reference/chat-completions Below are the parameters: N max_tokensresponse_formatseedstreamtop_p tool_choicetoolsuserreasoning_effortstop temperaturereturnc Cs>t} | D]\} }| dkr|durt|j| |q dS)Nself)localscopyitemssetattr __class__)rrrrrr rr r r r rZlocals_keyvaluerQ/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/cerebras/chat.py__init__s zCerebrasConfig.__init__cs tS)N)super get_config)clsrrrr1s zCerebrasConfig.get_configmodelcCs"gd}t|ddr|d|S)zF Get the supported OpenAI params for the given model ) rmax_completion_tokensrrrr rr r r r Zcerebras)r!Zcustom_llm_providerr)rappend)rr!Zsupported_paramsrrrget_supported_openai_params5s  z*CerebrasConfig.get_supported_openai_paramsnon_default_paramsoptional_params drop_paramscCsD|j|d}|D]\}}|dkr||d<q ||vr|||<q |S)N)r!r"r)r$r)rr%r&r!r'Zsupported_openai_paramsparamrrrrmap_openai_paramsOs  z CerebrasConfig.map_openai_params) NNNNNNNNNNN)__name__ __module__ __qualname____doc__rrint__annotations__rdictrr boolr r strr listr rfloatr classmethodrr$r) __classcell__rrr rr sz      rN)r-typingrZ+litellm.llms.openai.chat.gpt_transformationrZ litellm.utilsrrrrrrs