߲j2UdZddlmZddlZddlmZmZejdZdZ de d<d Z de d <d Z de d <d Z de d<dZde d<eddhZde d<dZde d<eddhZde d<dZde d<dZde d<dZde d<d Zde d!<dZde d"<d#d$d%Zd&e d'<d Zde d(<d)d$iZd&e d*<dZde d+<dZde d,<dZde d-<d.Zde d/<d)d$iZd&e d0<d1Z de d2<d)d$iZ!d&e d3<d1Z"de d4<d)d$iZ#d&e d5<dZ$de d6<d)d$iZ%d&e d7<d8Z&de d9<dGd:Z'dHd;Z(dHd<Z) dI dJd=Z*dKd>Z+dLdMd?Z,dNd@Z-e fdOdAZ. dIdBdC dPdDZ/dQdEZ0e Z1de dF<y)RuCanonical reasoning-effort vocabulary and wire clamping. Hermes' internal effort ladder (``VALID_REASONING_EFFORTS`` plus ``none``) is wider than any single provider wire accepts; hand-rolled per-transport maps leaked new levels (``ultra``) to wires that 400 and inverted the ladder (unknown → weak default). Single source of truth: :data:`EFFORT_LADDER` (low→high), :func:`clamp_effort` (verbatim if supported, else the nearest WEAKER level; only when nothing weaker exists the weakest supported), and named wire-vocabulary constants so call sites declare data. Rules: wire shape stays local, only the vocabulary math lives here; unset stays unset (never invent an effort); when a provider rejects a level fix its declared set, never a predicate. ) annotationsN)OptionalSequencez (?:^|[^a-z0-9])k3(?:[^a-z0-9]|$))noneminimallowmediumhighxhighmaxultratuple[str, ...] EFFORT_LADDER)rrrr r r r OPENAI_COMPAT_WIRE_EFFORTS)rrr r r r CODEX_GPT56_EFFORTS)rrr r r CODEX_LEGACY_EFFORTS)rr r r r CODEX_ASTRA_EFFORTSz gpt-6-astrazgpt-6-astra-900kzfrozenset[str]ASTRA_MODEL_IDS)z gpt-6-solz gpt-6-terraz gpt-6-lunaGPT6_TIER_PREFIXESzgpt-daybreak-blue-latestzgpt-daybreak-blue-latest-900kDAYBREAK_MODEL_IDS)rr r r XAI_GROK46_EFFORTS)rr r XAI_LEGACY_EFFORTS)rrr r r ACTUAL_RELAY_EFFORTS)rr r KIMI_K3_EFFORTSKIMI_K2_EFFORTSr r )r r zdict[str, str]KIMI_K3_OVERRIDESOX_ALPHA_EFFORTSr OX_ALPHA_OVERRIDESTOKENHUB_EFFORTSNEBIUS_EFFORTS SOLAR_EFFORTS)r r GLM52_EFFORTSGLM52_OVERRIDES)rr r r GLM53_EFFORTSGLM53_OVERRIDESDEEPSEEK_V4_EFFORTSDEEPSEEK_V4_OVERRIDESOLLAMA_CLOUD_EFFORTSOLLAMA_CLOUD_OVERRIDES)rrr r r META_AI_EFFORTScx|xsdjjjdddtvS)z``gpt-6-astra`` or its Hermes-side ``-900k`` picker alias, with or without a ``vendor/`` prefix. The single home for the slug set: picker gating, effort vocabulary and the request sanitizer all key off it, so a new Astra alias is one edit./)striplowerrsplitr)models ;/home/ubuntu/.hermes/hermes-agent/agent/reasoning_effort.pyis_astra_modelr5Zs7 KR   & & ( / /Q 7 ; NNct|rtS|xsdjjj ddd}d|vs|j t s|tvrtStS)z9Supported effort set for an OpenAI/Codex Responses model.r,r-r.r/zgpt-5.6) r5rr0r1r2 startswithrrrr)r3bares r4codex_supported_effortsr:asoe"" KR   & & ( / /Q 7 ;D  0B CtOaGa "r6c|xsdjjjdd}tj |rt St S)uSupported effort set for a Moonshot/Kimi slug (bare ``k3``, ``k3-256k``, ``kimi-k3*`` → K3). K3 is served as the bare slug ``k3``, plan variants like ``k3-256k``, and the ``kimi-k3*`` aliases; its documented set is low/high/max. Everything earlier speaks low/medium/high. Boundary-matched so K2-era names (``kimi-k2.6``) never match (detection regex from #76427 by @ruizanthony). r,r-r/)r0r1split_KIMI_K3_SLUG_REsearchrr)r3ms r4kimi_supported_effortsr@msG "##%++C04A.55a8?MoMr6c4t|xsdjj}|r|s|Sd|DDcgc] }|tvs |}}|r||vr|S|r|j ||vr||S|tvr|S|Dcgc] }|dk7s | }}|s|Stj |}|Dcgc]}tj ||ks|} }| rt | tj St|tj Scc}wcc}wcc}w)uClamp a requested reasoning effort onto a wire's supported levels. ``overrides`` (a declared vendor mapping, e.g. Kimi K3 ``medium → high``) is consulted first. Otherwise the request passes through unchanged when it is supported, when the supported set is unknown/empty, or when it isn't a recognized ladder level (custom providers may use bespoke names). Else the **nearest weaker** supported level is returned so a clamp never escalates cost; when nothing weaker exists, the weakest supported level (the provider's floor is the closest honest match). Monotonic: a stronger request never resolves weaker than a weaker request would. r,c3jK|]+}t|jj-ywN)strr0r1).0ss r4 zclamp_effort..s#%Pc!flln&:&:&<%Ps13r)key)rDr0r1rgetindexr min) effort supported overrides requestedlvlsupported_normlevel candidates requested_idxbelows r4 clamp_effortrVxsFLb!'')//1I I %Pi%PicTW[hThciNi Y.8 Y]]9-?## % &4GEu%GJG  !'' 2M * Yum.A.A%.H=.XU YE Y273u--- .eSQ^QdQd=eejH Zs# D  D  D D/D Dcn|xsdjjdk(r t|StS)zLevels the (provider, model) route's ENTRY clamp accepts: the Codex/OpenAI Responses set per model generation, else the widest OpenAI-compatible vocabulary (narrower providers clamp again downstream, never upward).r,z openai-codex)r0r1r:r)providerr3s r4route_supported_effortsrYs3 B%%'>9&u-- %%r6ct|xsdjj}t|t ||}|r||k(r|S|d|dS)a Picker / ``/reasoning`` status label for a ladder level: the level itself when the route sends it verbatim, else ``" (sends on this route)"`` so a Hermes-internal step such as ``ultra`` (#61634) is never presented as a distinct wire level the route does not have.r,z (sends z on this route))rDr0r1rVrY)rLrXr3rOclampeds r4effort_display_labelr\s]FLb!'')//1I9&=h&NOG%I)=9qi[PXY`XaapCqqr6ct|tr|jdduryt|jdxsdj j xsdS)uThe user's explicit effort, or None (absent/malformed config, no effort, or reasoning disabled) — callers then omit the wire field.enabledFNrLr,) isinstancedictrIrDr0r1)reasoning_configs r4requested_effortrbsY & -1A1E1Ei1PTY1Y ##H-3 4 : : < B B D LLr6ct|ts|St|jdxsdj j }|r t ||n|}||k7ri|d|iS|S)aReturn ``reasoning_config`` with its ``effort`` clamped onto ``supported`` (non-dicts and configs without an effort pass through untouched). The entry clamp for an OpenAI-compatible chat-completions request builder: Hermes-internal ``ultra`` never reaches a wire (#89503 main transport, #112010 aux/MoA), while provider profiles with narrower vocabularies clamp again downstream. Unset stays unset. rLr,)r_r`rDrIr0r1rV)rarMrLr[s r4clamp_reasoning_configrdss & - !%%h/52 6 < < > D D FF17l69-VG6=6G 2 2' 2]M]]r6F)always_emit_togglect|tr|jddurdddiiifSt|}t |dk(rdn|||}||vr|rdddiinid|ifSdddiiifS) a+Translate a reasoning config onto the Moonshot/DeepSeek chat_completions wire: ``extra_body.thinking`` toggle and top-level ``reasoning_effort``. Moonshot 400s when both are sent, so by default the effort (when it lands in ``efforts``) replaces the toggle. DeepSeek instead requires the toggle on every request (an omitted toggle defaults thinking on and then demands ``reasoning_content`` echoes), hence ``always_emit_toggle``. A requested effort of ``none`` is not a level on these wires; it falls back to the plain toggle. r^FthinkingtypedisabledrNreasoning_effort)r_r`rIrbrV)raeffortsrNrerLr[s r4thinking_toggle_extrasrls "D).>.B.B9.MQV.VVZ01255 . /F6V#34)TG'5Gfi01RSegnRooo + ,b 00r6c|xsdjjdddjdk7riifSt|}t |dk(rdn|t t }|t vrid|ifSiifS) zOx Alpha (``x-preview-f-free``) ``reasoning_effort`` translation for the opencode-zen profile (low/high/max only; anything else 400s).r,r-r.r/zx-preview-f-freerNrj)r0r2r1rbrVrr)rar3rLr[s r4ox_alpha_reasoning_extrasrns  ##C+B/557;MM2v . /F6V#34AQSefG29=M2MB#W- .[TVXZS[[r6CODEX_RESPONSES_EFFORTS)r3 Optional[str]returnbool)r3rprqrrC)rLrprMzOptional[Sequence[str]]rNOptional[dict[str, str]]rqrp)rXrpr3rprqr)NN)rLrprXrpr3rprqrD)raOptional[dict]rqrp)rartrM Sequence[str]rqrt) rartrkrurNrsrerrrqtuple[dict, dict])rartr3rprqrv)2__doc__ __future__rretypingrrcompiler=r__annotations__rrrr frozensetrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r5r:r@rVrYr\rbrdrlrnror6r4rsg # %2::AB "g f/kOj(Y_X(RoR(Q_P"+]F%;/:&-u%5N5%>/=";;!: :"1 0#*E"22"B A#*E"22(H_G)0%(8~8)QoP*15)99$QPO Nfjf f&=fJbffB&rMYs ^$+/1 % 1$1 1(1  1  12\,?>r6