`gj5 dZddlmZddlZddlmZddlmZddlm Z ejdZ ddZ dd Z dd ZGd d e Zed dddddddd Zeey)uZAI / GLM provider profile. Z.AI's GLM-4.5-and-later chat models default to thinking-mode ON when the request omits ``thinking``. Hermes' ``reasoning_config = {"enabled": False}`` was previously a silent no-op on this route — the base profile emits nothing, so users who turned thinking off (desktop toggle, ``/reasoning none``, ``reasoning_effort: none``/``false`` in config.yaml) kept burning thinking tokens on every turn. :meth:`ZaiProfile.build_api_kwargs_extras` translates the Hermes reasoning config into the wire shape Z.AI's OpenAI-compat endpoint expects: {"extra_body": {"thinking": {"type": "enabled" | "disabled"}}} When no reasoning preference is set (``reasoning_config is None``) the field is omitted so the server default applies, matching prior behavior. GLM models before 4.5 (e.g. ``glm-4-9b``) don't accept ``thinking`` and are left untouched. GLM-5.2 additionally exposes a native ``reasoning_effort`` knob with exactly two enabled levels — ``high`` and ``max`` — on the OpenAI-compatible endpoint (per Z.AI / BigModel docs). Hermes' richer effort scale is collapsed onto those two so the user's effort preference actually reaches the model instead of being silently dropped. ) annotationsN)Any)register_provider)ProviderProfilez^glm-(\d+)(?:\.(\d+))?c|xsdjj}tj|}|syt |j d}t |j dxsd}||fdk\S)uHGLM thinking-capable model families: glm-4.5 and later (4.5, 4.6, 5…).Fr))striplower_GLM_VERSION_REmatchintgroup)modelmrmajorminors W/root/.hermes/venv/lib/python3.12/site-packages/plugins/model-providers/zai/__init__.py_model_supports_thinkingr&sk "##%A  ! !! $E   A E  A#! $E 5>V ##cv|xsdjjsytfddDS)a Detect GLM-5.2 across the alias spellings providers use. Covers the canonical ``glm-5.2`` plus the ``glm-5-2`` / ``glm-5p2`` variants seen on relays (Fireworks ``glm-5p2``, etc.) and any vendor-prefixed form (``z-ai/glm-5.2``, ``zai-org-glm-5-2``). rFc3&K|]}|v yw)N).0tokenrs r z_is_glm_5_2..;sIeuzIs)glm-5.2zglm-5-2zglm-5p2)r rany)rrs @r _is_glm_5_2r"1s8 "##%A  I'HI IIrct|tsy|jddury|jdxsdjj }|r|dk(ry|dvryy ) a~Map Hermes reasoning effort onto GLM-5.2's native ``high``/``max``. GLM-5.2 only supports two enabled effort levels. ``xhigh``/``max``/``ultra`` request the top tier; everything else that is enabled requests ``high`` (its minimum thinking level). When reasoning is explicitly disabled, or no effort preference is supplied, the server default is left untouched. NenabledFeffortrnone>maxultraxhighr'high) isinstancedictgetr r)reasoning_configr%s r_glm_5_2_reasoning_effortr/>sk & -I&%/""8,2 9 9 ; A A CF Vv% ** rc*eZdZdZddd ddZy) ZaiProfileuEZ.AI / GLM — extra_body.thinking on/off + GLM-5.2 reasoning_effort.N)r.rc i}i}t|st|s||fSt|tr|j ddu}d|rdndi|d<t|rt |}|||d<||fS)Nr$Ftypedisabledthinkingreasoning_effort)rr"r+r,r-r/)selfr.rcontext extra_body top_levelr$r%s rbuild_api_kwargs_extrasz"ZaiProfile.build_api_kwargs_extrasXs&( $& '.{57Iy( ( & -&**95UBG&,7i %SJz " u ./?@F!06 ,-9$$r)r. dict | Noner str | Nonereturnz%tuple[dict[str, Any], dict[str, Any]])__name__ __module__ __qualname____doc__r;rrrr1r1Us)O264%#.%>H% .%rr1zai)glmzz-aizz.aizhipu) GLM_API_KEY ZAI_API_KEY Z_AI_API_KEYz Z.AI (GLM)uZ.AI / GLM — Zhipu AI modelsz https://z.ai/)r zglm-5zglm-4-9bzhttps://api.z.ai/api/paas/v4z glm-4.5-flash) namealiasesenv_vars display_name description signup_urlfallback_modelsbase_urldefault_aux_model)rr=r>bool)r.r<r>r=)rB __future__rretypingr providersrproviders.basercompilerrr"r/r1rCrrrrYs}4# '*"**67$ J.%%4  , ;0 ,% #r