Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

hermes_cli/model_search.py

@@ -0,0 +1,53 @@
+"""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"),
+    # OpenCode Zen serves the "Ox Alpha" stealth model under an opaque
+    # preview slug; let users find it by its public codename.
+    "x-preview-f-free": ("ox-alpha", "ox"),
+}
+
+# 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)}"