i?&UdZddlmZddlZddlmZiddddd dd dd dd d dd ddddddddddddddddddddd dd!Zd"ed#<ehd$Z d%ed&<edhZ d%ed'<ehd(Z d%ed)<ed d*hZ d%ed+<ehd,Z d%ed-<edhZd%ed.<ehd/Zd%ed0<ehd1Zd%ed2<ejd3ZdCd7ZdCd8ZdCd9ZdDd;ZdEd=ZdFd?ZdCd@ZdGdBZdS)HuPer-provider model name normalization. Different LLM providers expect model identifiers in different formats: - **Aggregators** (OpenRouter, Nous, AI Gateway, Kilo Code) need ``vendor/model`` slugs like ``anthropic/claude-sonnet-4.6``. - **Anthropic** native API expects bare names with dots replaced by hyphens: ``claude-sonnet-4-6``. - **Copilot** expects bare names *with* dots preserved: ``claude-sonnet-4.6``. - **OpenCode Zen** preserves dots for GPT/GLM/Gemini/Kimi/MiniMax-style model IDs, but Claude still uses hyphenated native names like ``claude-sonnet-4-6``. - **OpenCode Go** preserves dots in model names: ``minimax-m2.7``. - **DeepSeek** accepts ``deepseek-chat`` (V3), ``deepseek-reasoner`` (R1-family), and the first-class V-series IDs (``deepseek-v4-pro``, ``deepseek-v4-flash``, and any future ``deepseek-v-*``). Older Hermes revisions folded every non-reasoner input into ``deepseek-chat``, which on aggregators routes to V3 — so a user picking V4 Pro was silently downgraded. - **Custom** and remaining providers pass the name through as-is. This module centralises that translation so callers can simply write:: api_model = normalize_model_for_provider(user_input, provider) Inspired by Clawdbot's ``normalizeAnthropicModelId`` pattern. ) annotationsN)Optionalclaude anthropicgptopenaio1o3o4geminigooglegemmadeepseekglmzz-aikimi moonshotaiminimaxgrokzx-aiqwenmimoxiaomitrinityzarcee-ainemotronnvidiallamaz meta-llamastepfun)steprzdict[str, str]_VENDOR_PREFIXES> ai-gatewaynouskilocode openrouterzfrozenset[str]_AGGREGATOR_PROVIDERS_DOT_TO_HYPHEN_PROVIDERS> openai-codexcopilot copilot-acp_STRIP_VENDOR_ONLY_PROVIDERS huggingface_AUTHORITATIVE_NATIVE_PROVIDERS> minimax-cn qwen-oauth kimi-coding ollama-cloud minimax-oauthkimi-coding-cnzaiarceecustomralibabar _MATCHING_PREFIX_STRIP_PROVIDERS_LOWERCASE_MODEL_PROVIDERS>r1cotthinkreasoner reasoning_DEEPSEEK_REASONER_KEYWORDS>deepseek-v4-prodeepseek-v4-flash deepseek-chatdeepseek-reasoner_DEEPSEEK_CANONICAL_MODELSz^deepseek-v\d+([-.].+)?$ model_namestrreturnct|}|tvr|St|r|St D] }||vrdS dS)aMap a model input to a DeepSeek-accepted identifier. Rules: - Already a known canonical (``deepseek-chat``/``deepseek-reasoner``/ ``deepseek-v4-pro``/``deepseek-v4-flash``) -> pass through. - Matches the V-series pattern ``deepseek-v...`` -> pass through (covers future ``deepseek-v5-*`` and dated variants without a release). - Contains a reasoner keyword (r1, think, reasoning, cot, reasoner) -> ``deepseek-reasoner``. - Everything else -> ``deepseek-chat``. Args: model_name: The bare model name (vendor prefix already stripped). Returns: A DeepSeek-accepted model identifier. r@r?)_strip_vendor_prefixlowerrA_DEEPSEEK_V_SERIES_REmatchr<)rBbarekeywords ?/home/ubuntu/.hermes/hermes-agent/hermes_cli/model_normalize.py_normalize_for_deepseekrMsz$ + + 1 1 3 3D ))) ""4(( /'' d??&&&  ?cFd|vr|dddS|S)a@Remove a ``vendor/`` prefix if present. Examples:: >>> _strip_vendor_prefix("anthropic/claude-sonnet-4.6") 'claude-sonnet-4.6' >>> _strip_vendor_prefix("claude-sonnet-4.6") 'claude-sonnet-4.6' >>> _strip_vendor_prefix("meta-llama/llama-4-scout") 'llama-4-scout' /)splitrBs rLrFrFs0 jQ''** rNc.|ddS)zReplace dots with hyphens in a model name. Anthropic's native API uses hyphens where marketing names use dots: ``claude-sonnet-4.6`` -> ``claude-sonnet-4-6``. .-)replacerSs rL_dots_to_hyphensrXs   c3 ' ''rN provider_namec|pd}|s|S ddlm}||S#t$r|cYSwxYw)z2Resolve provider aliases to Hermes' canonical ids.r)normalize_provider)striprGhermes_cli.modelsr\ Exception)rYrawr\s rL_normalize_provider_aliasras}  B % % ' ' - - / /C  888888!!#&&&  s? A Atarget_providerc d|vr|S|dd\}}|r|s|St|}t|}|r||kr|S|S)aStrip ``provider/`` only when the prefix matches the target provider. This prevents arbitrary slash-bearing model IDs from being mangled on native providers while still repairing manual config values like ``zai/glm-5.1`` for the ``zai`` provider. rPrQ)rRr]ra)rBrbprefix remaindernormalized_prefixnormalized_targets rL_strip_matching_provider_prefixrhs *"((a00FI <<>>!2!21&991/BB!.2CCC   rN Optional[str]c|}|sdSd|vr0|dddpdS|}|dd}|tvr t|StD]\}}||r|cSdS)a*Detect the vendor slug from a bare model name. Uses the first hyphen-delimited token of the model name to look up the corresponding vendor in ``_VENDOR_PREFIXES``. Also handles case-insensitive matching and special patterns. Args: model_name: A model name, optionally already including a ``vendor/`` prefix. If a prefix is present it is used directly. Returns: The vendor slug (e.g. ``"anthropic"``, ``"openai"``) or ``None`` if no vendor can be confidently detected. Examples:: >>> detect_vendor("claude-sonnet-4.6") 'anthropic' >>> detect_vendor("gpt-5.4-mini") 'openai' >>> detect_vendor("anthropic/claude-sonnet-4.6") 'anthropic' >>> detect_vendor("my-custom-model") NrPrQrrV)r]rRrGritems startswith)rBname name_lower first_tokenrdvendors rL detect_vendorrqs4     D t d{{zz#q!!!$**,,44J""3''*K&&& ,,+0022   ( ( MMM  4rNcBd|vr|St|}|r|d|S|S)aNPrepend the detected ``vendor/`` prefix if missing. Used for aggregator providers that require ``vendor/model`` format. If the name already contains a ``/``, it is returned as-is. If no vendor can be detected, the name is returned unchanged (aggregators may still accept it or return an error). Examples:: >>> _prepend_vendor("claude-sonnet-4.6") 'anthropic/claude-sonnet-4.6' >>> _prepend_vendor("anthropic/claude-sonnet-4.6") 'anthropic/claude-sonnet-4.6' >>> _prepend_vendor("my-custom-thing") 'my-custom-thing' rP)rq)rBrps rL_prepend_vendorrs)sD" j : & &F ('':''' rN model_inputcb|pd}|s|St|}|tvrt|S|dkrNt ||}d|vr|S|drt|S|S|tvr%t ||}d|vr|St|S|dvr' ddl m }||}|r|Sn#t$rYnwxYw|tvrIt ||}||kr1|dr| dd d S|S|d kr%t ||}d|vr|St|S|tvr/t ||}|t vr|}|S|t"vr|S|S) a.Translate a model name into the format the target provider's API expects. This is the primary entry point for model name normalisation. It accepts any user-facing model identifier and transforms it for the specific provider that will receive the API call. Args: model_input: The model name as provided by the user or config. Can be bare (``"claude-sonnet-4.6"``), vendor-prefixed (``"anthropic/claude-sonnet-4.6"``), or already in native format (``"claude-sonnet-4-6"``). target_provider: The canonical Hermes provider id, e.g. ``"openrouter"``, ``"anthropic"``, ``"copilot"``, ``"deepseek"``, ``"custom"``. Should already be normalised via ``hermes_cli.models.normalize_provider()``. Returns: The model identifier string that the target provider's API expects. Raises: No exceptions -- always returns a best-effort string. Examples:: >>> normalize_model_for_provider("claude-sonnet-4.6", "openrouter") 'anthropic/claude-sonnet-4.6' >>> normalize_model_for_provider("anthropic/claude-sonnet-4.6", "anthropic") 'claude-sonnet-4-6' >>> normalize_model_for_provider("anthropic/claude-sonnet-4.6", "copilot") 'claude-sonnet-4.6' >>> normalize_model_for_provider("openai/gpt-5.4", "copilot") 'gpt-5.4' >>> normalize_model_for_provider("claude-sonnet-4.6", "opencode-zen") 'claude-sonnet-4-6' >>> normalize_model_for_provider("minimax-m2.5-free", "opencode-zen") 'minimax-m2.5-free' >>> normalize_model_for_provider("deepseek-v3", "deepseek") 'deepseek-chat' >>> normalize_model_for_provider("deepseek-r1", "deepseek") 'deepseek-reasoner' >>> normalize_model_for_provider("my-model", "custom") 'my-model' >>> normalize_model_for_provider("claude-sonnet-4.6", "zai") 'claude-sonnet-4.6' >>> normalize_model_for_provider("MiMo-V2.5-Pro", "xiaomi") 'mimo-v2.5-pro' r[z opencode-zenrPzclaude->r&r'r)normalize_copilot_model_idzopenai/rQr)r]rar#rsrhrGrlrXr$r^rvr_r(rRrMr5r6r*) rtrbrmproviderrJrv normalizedstrippedresults rLnormalize_model_for_providerr{Gs5v  2 $ $ & &D  (99H(((t$$$>!!.tX>> $;;K ::<< " "9 - - *#D)) ) +++.tX>> $;;K%%%---  D D D D D D33D99J "!! "    D ///24BB t   : : ::c1%%a( (:.tX>> $;;K&t,,,3330x@@ 1 1 1\\^^F 222  Ks C C,+C,)rBrCrDrC)rYrCrDrC)rBrCrbrCrDrC)rBrCrDri)rtrCrbrCrDrC)__doc__ __future__rretypingrr__annotations__ frozensetr#r$r(r*r5r6r<rAcompilerHrMrFrXrarhrqrsr{rNrLrs:#""""" $ k$ 8$ ($ ( $  ( $  h $ X$ $ 6$ L$y$ F$ F$ H$z$ !$" \#$$ '$$$.)2 333)),596,, 09y:::003<) =334=9 > > > 4 4    (.7Y 8../8i999//.7Y888..# #>??    N"((((    *////d