@genee/omp-opsx-addon 0.1.0

package/lib/tiers-data.ts

@@ -0,0 +1,136 @@
+/**
+ * Data-driven model tier data layer.
+ *
+ * Loads tier assignments from `~/.omp/cache/tiers-data.json`.
+ * Falls back silently to heuristic mode when the file is missing or corrupt.
+ */
+
+import { readFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import type { TierName } from './model-tiers.js';
+
+// ── types ────────────────────────────────────────────────────────────
+
+export interface TierDataEntry {
+    /** Model selector: exact provider/id, bare id, or glob pattern. */
+    selector: string;
+    tier: TierName;
+    reputation: number;
+}
+
+interface TierDataFile {
+    version: number;
+    updated: string;
+    entries: TierDataEntry[];
+}
+
+// ── state ────────────────────────────────────────────────────────────
+
+const CACHE_PATH = join(homedir(), '.omp', 'cache', 'tiers-data.json');
+
+let _data: TierDataFile | null | undefined; // undefined = not yet loaded, null = failed, object = ok
+let _loadError: string | null = null;
+
+// ── public API ───────────────────────────────────────────────────────
+
+/** Path to the tiers data cache file. */
+export const TIERS_DATA_PATH = CACHE_PATH;
+
+/** Age of the loaded data (ms since 1970), or 0 if not loaded. */
+export function tiersDataAge(): number {
+    if (_data?.updated) {
+        const ms = Date.parse(_data.updated);
+        return Number.isNaN(ms) ? 0 : ms;
+    }
+    return 0;
+}
+
+/** Whether the data was loaded successfully. Null if not yet loaded. */
+export function isTiersDataLoaded(): boolean | null {
+    if (_data === undefined) return null;
+    return _data !== null;
+}
+
+/** Last load error message, or null. */
+export function tiersDataLoadError(): string | null {
+    return _loadError;
+}
+
+/**
+ * Match a model against the tier data.
+ * Returns `{ tier, reputation }` on match, or null when no entry matches.
+ */
+export function matchDataEntry(
+    modelId: string,
+    provider?: string,
+): { tier: TierName; reputation: number; reason: string } | null {
+    const data = ensureLoaded();
+    if (!data) return null;
+
+    // 1. Provider-scoped exact match
+    if (provider) {
+        const scoped = `${provider}/${modelId}`;
+        const hit = data.entries.find((e) => e.selector === scoped);
+        if (hit) return { tier: hit.tier, reputation: hit.reputation, reason: `data: ${hit.selector}` };
+    }
+
+    // 2. Bare id exact match
+    const bare = data.entries.find((e) => e.selector === modelId);
+    if (bare) return { tier: bare.tier, reputation: bare.reputation, reason: `data: ${bare.selector}` };
+
+    // 3. Glob patterns (in order)
+    const globs = data.entries.filter((e) => hasGlob(e.selector));
+    for (const g of globs) {
+        if (globMatch(g.selector, modelId)) {
+            return { tier: g.tier, reputation: g.reputation, reason: `data: ${g.selector}` };
+        }
+    }
+
+    return null;
+}
+
+/** Force-reload data (used after pick-model update). */
+export function reloadTierData(): void {
+    _data = undefined;
+    _loadError = null;
+    ensureLoaded();
+}
+
+/** Reset internal state (for tests only). */
+export function _resetTierDataForTest(): void {
+    _data = undefined;
+    _loadError = null;
+}
+
+// ── internal ─────────────────────────────────────────────────────────
+
+function ensureLoaded(): TierDataFile | null {
+    if (_data !== undefined) return _data;
+    try {
+        const raw = readFileSync(CACHE_PATH, 'utf-8');
+        const parsed = JSON.parse(raw) as TierDataFile;
+        if (!Array.isArray(parsed.entries)) throw new Error('missing entries array');
+        _data = parsed;
+        _loadError = null;
+    } catch (err) {
+        _data = null;
+        _loadError = err instanceof Error ? err.message : String(err);
+    }
+    return _data;
+}
+
+function hasGlob(selector: string): boolean {
+    return selector.includes('*') || selector.includes('?');
+}
+
+function globMatch(pattern: string, target: string): boolean {
+    return globToRegex(pattern).test(target);
+}
+
+function globToRegex(pattern: string): RegExp {
+    // Escape regex specials, then replace glob wildcards
+    const escaped = pattern.replace(/[.+^${}()|[\]\\]/g, '\\$&');
+    const withStars = escaped.replace(/\*/g, '.*').replace(/\?/g, '.');
+    return new RegExp(`^${withStars}$`, 'i');
+}

package/lib/tiers-updater.ts

@@ -0,0 +1,146 @@
+/**
+ * Tier data updater — seed data + cache file management.
+ *
+ * Called by `/pick-model update` and auto-refresh on startup
+ * when the cache is missing or stale (> 7 days).
+ */
+
+import { writeFileSync, mkdirSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join, dirname } from 'node:path';
+import type { TierDataEntry } from './tiers-data.js';
+import { reloadTierData, TIERS_DATA_PATH, tiersDataAge, isTiersDataLoaded } from './tiers-data.js';
+
+// ── constants ────────────────────────────────────────────────────────
+
+/** Auto-refresh when cache is older than this (ms). */
+const STALE_THRESHOLD_MS = 7 * 24 * 60 * 60 * 1000; // 7 days
+
+// ── seed data ────────────────────────────────────────────────────────
+
+/**
+ * Curated tier data for known models.
+ * Provider-scoped entries take precedence over bare-id entries.
+ * This list should be updated when new flagship models are released.
+ */
+const SEED_ENTRIES: TierDataEntry[] = [
+    // ── anthropic ──────────────────────────────────
+    { selector: 'claude-fable-5',                tier: 'top',  reputation: 85 },
+    { selector: 'claude-mythos-5',               tier: 'top',  reputation: 90 },
+    { selector: 'claude-opus-4-8',               tier: 'top',  reputation: 80 },
+    { selector: 'claude-opus-*',                 tier: 'top',  reputation: 78 },
+    { selector: 'claude-sonnet-5',               tier: 'high', reputation: 65 },
+    { selector: 'claude-sonnet-4-*',             tier: 'high', reputation: 60 },
+    { selector: 'claude-haiku-4-*',              tier: 'low',  reputation: 35 },
+
+    // ── deepseek ───────────────────────────────────
+    { selector: 'deepseek/deepseek-v4-pro',       tier: 'top',  reputation: 70 },
+    { selector: 'deepseek/deepseek-v4-flash',     tier: 'mid',  reputation: 45 },
+    { selector: 'deepseek/deepseek-v4-pro',       tier: 'top',  reputation: 70 },
+
+    // ── minimax ────────────────────────────────────
+    { selector: 'MiniMax-M3',                    tier: 'top',  reputation: 60 },
+    { selector: 'minimax-code-cn/MiniMax-M3',    tier: 'top',  reputation: 60 },
+
+    // ── kimi / moonshot ────────────────────────────
+    { selector: 'k3',                            tier: 'top',  reputation: 75 },
+    { selector: 'kimi-k2.7',                     tier: 'top',  reputation: 72 },
+    { selector: 'kimi-k2.5',                     tier: 'top',  reputation: 65 },
+    { selector: 'kimi-k2.*',                     tier: 'top',  reputation: 65 },
+    { selector: 'kimi-k3',                       tier: 'top',  reputation: 75 },
+    { selector: 'kimi-k*',                       tier: 'top',  reputation: 60 },
+
+    // ── zhipu / glm ────────────────────────────────
+    { selector: 'glm-5.2',                       tier: 'top',  reputation: 70 },
+    { selector: 'glm-5.*',                       tier: 'top',  reputation: 68 },
+    { selector: 'glm-4.7',                       tier: 'mid',  reputation: 50 },
+
+    // ── openai ─────────────────────────────────────
+    { selector: 'gpt-5.6-luna-pro',              tier: 'top',  reputation: 82 },
+    { selector: 'gpt-5.6-luna',                  tier: 'high', reputation: 70 },
+    { selector: 'gpt-5.6',                       tier: 'high', reputation: 65 },
+    { selector: 'gpt-5.*-pro',                   tier: 'top',  reputation: 80 },
+    { selector: 'gpt-5.*',                       tier: 'high', reputation: 65 },
+    { selector: 'gpt-5.4-mini',                  tier: 'tiny', reputation: 35 },
+
+    // ── google ─────────────────────────────────────
+    { selector: 'gemini-3.1-pro*',               tier: 'top',  reputation: 75 },
+    { selector: 'gemini-3.5-flash',              tier: 'mid',  reputation: 40 },
+    { selector: 'gemini-3.1-flash-lite',         tier: 'tiny', reputation: 30 },
+
+    // ── xai ────────────────────────────────────────
+    { selector: 'grok-4.20*',                    tier: 'mid',  reputation: 50 },
+    { selector: 'grok-3-mini',                   tier: 'tiny', reputation: 30 },
+
+    // ── opencode-go ────────────────────────────────
+    { selector: 'opencode-go/glm-5.2',           tier: 'top',  reputation: 70 },
+    { selector: 'opencode-go/grok-4.5',          tier: 'mid',  reputation: 50 },
+
+    // ── opencode ───────────────────────────────────
+    { selector: 'opencode/gpt-5.4-pro',          tier: 'top',  reputation: 70 },
+    { selector: 'opencode/gpt-5.4',              tier: 'high', reputation: 60 },
+
+    // ── openai-codex ───────────────────────────────
+    { selector: 'openai-codex/gpt-5.6-luna',     tier: 'high', reputation: 70 },
+    { selector: 'openai-codex/gpt-5.4-mini',     tier: 'tiny', reputation: 35 },
+];
+
+// ── public API ───────────────────────────────────────────────────────
+
+/** Check whether cached tier data is stale (> 7 days). */
+export function isTiersDataStale(): boolean {
+    const loaded = isTiersDataLoaded();
+    if (loaded === null) return true;  // not yet loaded → treat as stale
+    if (!loaded) return true;          // failed to load → stale
+    const age = tiersDataAge();
+    if (age === 0) return true;
+    return Date.now() - age > STALE_THRESHOLD_MS;
+}
+
+/**
+ * Auto-refresh tier data if needed.
+ * Returns true if a refresh was performed.
+ */
+export function autoRefreshTiersIfStale(): boolean {
+    if (isTiersDataStale()) {
+        updateTiers({ silent: true });
+        return true;
+    }
+    return false;
+}
+
+/** Result of an update operation. */
+export interface UpdateResult {
+    ok: boolean;
+    path: string;
+    count: number;
+    error?: string;
+}
+
+/**
+ * Write seed tier data to the cache file.
+ * Creates parent directories as needed.
+ */
+export function updateTiers(opts?: { silent?: boolean }): UpdateResult {
+    const path = TIERS_DATA_PATH;
+    try {
+        const dir = dirname(path);
+        if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
+
+        const data = {
+            version: 1,
+            updated: new Date().toISOString(),
+            entries: SEED_ENTRIES,
+        };
+
+        writeFileSync(path, JSON.stringify(data, null, 2) + '\n', 'utf-8');
+
+        // Force runtime to reload on next access
+        reloadTierData();
+
+        return { ok: true, path, count: SEED_ENTRIES.length };
+    } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        return { ok: false, path, count: 0, error: msg };
+    }
+}