1bf8bd2c7d
The OpenCode Zen wire slug for the Ox Alpha stealth model is opaque (x-preview-f-free); users searching the picker for 'ox' or 'ox-alpha' found nothing. Adds the search alias across all four synced alias tables (CLI, desktop, web, TUI) plus tests. Wire id is unchanged and still what renders and gets sent to the provider, matching the k3 → kimi-k3 precedent. No canonical-dedup collision with opencode-go's keyed ox-alpha-free slug.
54 lines
1.8 KiB
Python
54 lines
1.8 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"),
|
|
# 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)}"
|