uPj^UddlmZddlmZddlmZmZddlmZdZ ee d<Gdd eZ y ) )Final)get_secret_str)AllMessageValuesChatCompletionToolParam)OpenAIGPTConfigzhttps://api.z.ai/api/paas/v4 ZAI_API_BASEc eZdZededzfdZdedzdedzdeedzedzffdZ d dedee d ee dzdeee ee dzffd Z dedefd Z y) ZAIChatConfigreturnNcy)Nzai)selfs i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/zai/chat/transformation.pycustom_llm_providerz!ZAIChatConfig.custom_llm_provider sapi_baseapi_keycV|xstdxst}|xs td}||fS)Nr ZAI_API_KEY)rr )rrrdynamic_api_keys r$_get_openai_compatible_provider_infoz2ZAIChatConfig._get_openai_compatible_provider_infos2M~n=M!(!IN=,I((rmodelmessagestoolsc ||fS)zv Override to preserve cache_control for GLM/ZAI. GLM supports cache_control - don't strip it. r)rrrrs r1remove_cache_control_flag_from_messages_and_toolsz?ZAIChatConfig.remove_cache_control_flag_from_messages_and_toolssrcgd}ddl} |j||jr|jd|S#t$rY|SwxYw)N) max_tokensstreamstream_options temperaturetop_pstopr tool_choicer)rrthinking)litellmsupports_reasoningrappend Exception)rr base_paramsr(s rget_supported_openai_paramsz)ZAIChatConfig.get_supported_openai_params$s\    ))4KcKc)d"":.   s.: AA)N) __name__ __module__ __qualname__propertystrrtuplerlistrrrr-rrrr r s S4Z)d )-04Z) sTz3:% &)7;  '( +,t3  t$%t,C'Dt'KK L rr N) typingrlitellm.secret_managers.mainrlitellm.types.llms.openairropenai.chat.gpt_transformationrr __annotations__r rrrr:s'7O=4 e4-O-r