mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-27 17:58:07 +00:00
37e648128b
Follow-up to the salvaged #67409 search aliases: with kimi-k3 now in the curated kimi-coding list (#68108), a Coding Plan key rendered TWO rows for one model — curated 'kimi-k3' plus live-discovered bare 'k3' (merge dedup was exact-string). Add model_alias_canonical() derived from the same alias table and use it as the merge dedup key, so the curated public slug wins and live-only models still surface.
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
"""Picker-only search aliases for model ids.
|
|
|
|
Wire IDs stay unchanged. Some providers report short or brand-less ids
|
|
(Kimi Coding's flagship is literally ``k3``) that users still search for by
|
|
the familiar ``kimi-…`` naming of sibling models.
|
|
|
|
Keep in sync with ``ui-tui/src/lib/model-search-text.ts`` and
|
|
``web/src/lib/model-search-text.ts``.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
# Lowercased wire id → extra tokens appended to the search haystack only.
|
|
_MODEL_SEARCH_ALIASES: dict[str, tuple[str, ...]] = {
|
|
"k3": ("kimi-k3", "kimi"),
|
|
}
|
|
|
|
# Lowercased wire id → canonical public slug it aliases. Used by picker
|
|
# dedup so a live bare id and its curated public slug (``k3`` / ``kimi-k3``)
|
|
# don't render as two rows for the same model. Derived from the FIRST alias
|
|
# entry, which by convention is the full public slug.
|
|
_MODEL_ALIAS_CANONICAL: dict[str, str] = {
|
|
wire_id: aliases[0].lower()
|
|
for wire_id, aliases in _MODEL_SEARCH_ALIASES.items()
|
|
if aliases
|
|
}
|
|
|
|
|
|
def model_alias_canonical(model: str) -> str:
|
|
"""Return the canonical public slug for a bare wire-id alias.
|
|
|
|
Identity for ids with no alias entry. Lowercases the input so callers
|
|
can use the result directly as a dedup key.
|
|
"""
|
|
key = (model or "").strip().lower()
|
|
return _MODEL_ALIAS_CANONICAL.get(key, key)
|
|
|
|
|
|
def model_search_text(model: str) -> str:
|
|
"""Return the haystack used for fuzzy/substring model search.
|
|
|
|
Never changes the wire id passed to the provider.
|
|
"""
|
|
mid = (model or "").strip()
|
|
if not mid:
|
|
return model or ""
|
|
aliases = _MODEL_SEARCH_ALIASES.get(mid.lower())
|
|
if not aliases:
|
|
return mid
|
|
return f"{mid} {' '.join(aliases)}"
|