@krtclcdy/tersio 2.24.0 → 2.25.0

package/extensions/shared/host.js

@@ -1,68 +1,107 @@
-// extensions/shared/host.ts — one extension source, two agent hosts.
-//
-// OMP is a fork of pi, and six surfaces differ: tool schemas (OMP injects
-// `pi.zod`, pi takes any object it forwards as JSON Schema), `setLabel`
-// (pi's takes an entryId, so an extension name must never reach it),
-// `session_branch` (pi reports that switch as `session_tree`), select options
-// (pi takes plain strings), the status-bar theme (pi has no ctx.ui.theme), and
-// prompt injection (pi appends to its sections rather than replacing them).
-//
-// The host is read from the API shape, never a name or an env marker: OMP is
-// the fork that injects zod, and pi does not, including through the SDK.
+// One source for both hosts; each host-specific behaviour lives in one adapter row.
 import { asPromptArray } from './session-state.js';
+const piAdapter = {
+    label() { },
+    on(pi, event, handler) {
+        if (event === 'session_branch')
+            return;
+        pi.on(event, handler);
+    },
+    async select(ui, title, options, dialogOptions) {
+        if (!ui?.select)
+            return undefined;
+        const shown = options.map((o) => (o.description ? `${o.label} — ${o.description}` : o.label));
+        const picked = await ui.select(title, shown, dialogOptions);
+        if (picked === undefined)
+            return undefined;
+        return options[shown.indexOf(picked)]?.label ?? picked;
+    },
+    inject(event, text, staleMarker) {
+        const marker = staleMarker?.replace(/^[^\w]+/, '').trim();
+        const dropStale = (parts) => (marker ? parts.filter((part) => !part.includes(marker)) : parts);
+        const options = event.systemPromptOptions;
+        if (options && typeof options.appendSystemPrompt === 'string') {
+            options.appendSystemPrompt = dropStale([options.appendSystemPrompt]).concat(text).filter(Boolean).join('\n\n');
+            return undefined;
+        }
+        return { systemPrompt: `${dropStale(asPromptArray(event.systemPrompt)).join('\n\n')}\n\n${text}` };
+    },
+    stringArray(_pi, name, spec) {
+        const array = { type: 'array', items: { type: 'string' } };
+        if (spec.minItems)
+            array.minItems = spec.minItems;
+        if (spec.description)
+            array.description = spec.description;
+        return { type: 'object', properties: { [name]: array }, required: [name] };
+    },
+};
+const ompAdapter = {
+    label(pi, text) { pi.setLabel?.(text); },
+    on(pi, event, handler) { pi.on(event, handler); },
+    async select(ui, title, options, dialogOptions) {
+        if (!ui?.select)
+            return undefined;
+        return ui.select(title, options, dialogOptions);
+    },
+    inject(event, text, staleMarker) {
+        const marker = staleMarker?.replace(/^[^\w]+/, '').trim();
+        const dropStale = (parts) => (marker ? parts.filter((part) => !part.includes(marker)) : parts);
+        return { systemPrompt: [...dropStale(asPromptArray(event.systemPrompt)), text] };
+    },
+    stringArray(pi, name, spec) {
+        const { z } = pi.zod;
+        const array = z.array(z.string());
+        const constrained = spec.minItems ? array.min(spec.minItems) : array;
+        return z.object({ [name]: spec.description ? constrained.describe(spec.description) : constrained });
+    },
+};
+const opencodeAdapter = {
+    label() { },
+    on(pi, event, handler) { pi.on(event, handler); },
+    async select(ui, title, options, dialogOptions) {
+        if (!ui?.select)
+            return undefined;
+        return ui.select(title, options, dialogOptions);
+    },
+    inject(event, text, staleMarker) {
+        const marker = staleMarker?.replace(/^[^\w]+/, '').trim();
+        const dropStale = (parts) => (marker ? parts.filter((part) => !part.includes(marker)) : parts);
+        return { systemPrompt: [...dropStale(asPromptArray(event.systemPrompt)), text] };
+    },
+    stringArray(_pi, name, spec) {
+        const array = { type: 'array', items: { type: 'string' } };
+        if (spec.minItems)
+            array.minItems = spec.minItems;
+        if (spec.description)
+            array.description = spec.description;
+        return { type: 'object', properties: { [name]: array }, required: [name] };
+    },
+};
+// Explicit hostId wins; the legacy zod sniff keeps old callers working.
+function adapterFor(pi) {
+    if (pi.hostId === 'opencode')
+        return opencodeAdapter;
+    if (pi.hostId === 'omp')
+        return ompAdapter;
+    if (pi.hostId === 'pi')
+        return piAdapter;
+    return pi.zod ? ompAdapter : piAdapter;
+}
 export function isPiHost(pi) {
     return !pi.zod;
 }
-export function setExtensionLabel(pi, label) {
-    if (isPiHost(pi))
-        return;
-    pi.setLabel?.(label);
+export function setExtensionLabel(pi, text) {
+    adapterFor(pi).label(pi, text);
 }
 export function onHostEvent(pi, event, handler) {
-    if (event === 'session_branch' && isPiHost(pi))
-        return;
-    pi.on(event, handler);
+    adapterFor(pi).on(pi, event, handler);
 }
-// Returns the option's label on both hosts: pi renders plain strings, so the
-// description rides along inside the label and is stripped back out.
 export async function hostSelect(pi, ui, title, options, dialogOptions) {
-    if (!ui?.select)
-        return undefined;
-    if (!isPiHost(pi))
-        return ui.select(title, options, dialogOptions);
-    const shown = options.map((option) => (option.description ? `${option.label} — ${option.description}` : option.label));
-    const picked = await ui.select(title, shown, dialogOptions);
-    if (picked === undefined)
-        return undefined;
-    return options[shown.indexOf(picked)]?.label ?? picked;
+    return adapterFor(pi).select(ui, title, options, dialogOptions);
 }
-// What a `before_agent_start` handler hands back; undefined once pi was
-// updated in place.
-export function injectPromptText(pi, event, text) {
-    if (!isPiHost(pi))
-        return { systemPrompt: [...asPromptArray(event.systemPrompt), text] };
-    // Appending keeps pi's sections and the transcript diff; replacing the
-    // prompt the OMP way would drop every section.
-    const options = event.systemPromptOptions;
-    if (options && typeof options.appendSystemPrompt === 'string') {
-        options.appendSystemPrompt = [options.appendSystemPrompt, text].filter(Boolean).join('\n\n');
-        return undefined;
-    }
-    return { systemPrompt: `${asPromptArray(event.systemPrompt).join('\n\n')}\n\n${text}` };
+export function injectPromptText(pi, event, text, staleMarker) {
+    return adapterFor(pi).inject(event, text, staleMarker);
 }
-// One string-array argument in the active host's dialect. pi only requires an
-// object, so a JSON Schema literal suffices and no `typebox` import is needed.
 export function stringArrayToolParams(pi, name, spec = {}) {
-    if (!isPiHost(pi)) {
-        const { z } = pi.zod;
-        const array = z.array(z.string());
-        const constrained = spec.minItems ? array.min(spec.minItems) : array;
-        return z.object({ [name]: spec.description ? constrained.describe(spec.description) : constrained });
-    }
-    const array = { type: 'array', items: { type: 'string' } };
-    if (spec.minItems)
-        array.minItems = spec.minItems;
-    if (spec.description)
-        array.description = spec.description;
-    return { type: 'object', properties: { [name]: array }, required: [name] };
+    return adapterFor(pi).stringArray(pi, name, spec);
 }

package/extensions/shared/host.ts

@@ -1,19 +1,8 @@
-// extensions/shared/host.ts — one extension source, two agent hosts.
-//
-// OMP is a fork of pi, and six surfaces differ: tool schemas (OMP injects
-// `pi.zod`, pi takes any object it forwards as JSON Schema), `setLabel`
-// (pi's takes an entryId, so an extension name must never reach it),
-// `session_branch` (pi reports that switch as `session_tree`), select options
-// (pi takes plain strings), the status-bar theme (pi has no ctx.ui.theme), and
-// prompt injection (pi appends to its sections rather than replacing them).
-//
-// The host is read from the API shape, never a name or an env marker: OMP is
-// the fork that injects zod, and pi does not, including through the SDK.
+// One source for both hosts; each host-specific behaviour lives in one adapter row.
 
 import { asPromptArray } from './session-state.ts';
 import type { ExtensionApi, ExtensionCtx, PromptInjection, SystemPromptEvent, ToolParams, UiApi } from './types.ts';
 
-/** The zod subset the one tool schema needs. */
 export interface ZodChain {
   min: (n: number) => ZodChain;
   describe: (text: string) => ZodChain;
@@ -27,33 +16,121 @@ export interface ZodFactory {
   };
 }
 
-/** The host fields these adapters read. */
 export interface HostHandle {
   zod?: unknown;
   setLabel?: (...args: string[]) => void;
+  hostId?: 'pi' | 'omp' | 'opencode';
+}
+
+export interface SelectOption {
+  label: string;
+  description?: string;
+}
+
+export interface StringArraySpec {
+  minItems?: number;
+  description?: string;
+}
+
+interface HostAdapter {
+  label(pi: HostHandle, text: string): void;
+  on<E>(pi: ExtensionApi, event: string, handler: (event: E, ctx: ExtensionCtx) => unknown): void;
+  select(ui: UiApi | undefined, title: string, options: SelectOption[], dialogOptions?: Record<string, unknown>): Promise<string | undefined>;
+  inject(event: SystemPromptEvent, text: string, staleMarker?: string): PromptInjection | undefined;
+  stringArray(pi: ExtensionApi, name: string, spec: StringArraySpec): ToolParams;
+}
+
+const piAdapter: HostAdapter = {
+  label() { /* no label surface */ },
+  on(pi, event, handler) {
+    if (event === 'session_branch') return;
+    pi.on(event, handler);
+  },
+  async select(ui, title, options, dialogOptions) {
+    if (!ui?.select) return undefined;
+    const shown = options.map((o) => (o.description ? `${o.label} — ${o.description}` : o.label));
+    const picked = await ui.select(title, shown, dialogOptions);
+    if (picked === undefined) return undefined;
+    return options[shown.indexOf(picked)]?.label ?? picked;
+  },
+  inject(event, text, staleMarker) {
+    const marker = staleMarker?.replace(/^[^\w]+/, '').trim();
+    const dropStale = (parts: string[]): string[] => (marker ? parts.filter((part) => !part.includes(marker)) : parts);
+    const options = event.systemPromptOptions;
+    if (options && typeof options.appendSystemPrompt === 'string') {
+      options.appendSystemPrompt = dropStale([options.appendSystemPrompt]).concat(text).filter(Boolean).join('\n\n');
+      return undefined;
+    }
+    return { systemPrompt: `${dropStale(asPromptArray(event.systemPrompt)).join('\n\n')}\n\n${text}` };
+  },
+  stringArray(_pi, name, spec) {
+    const array: Record<string, unknown> = { type: 'array', items: { type: 'string' } };
+    if (spec.minItems) array.minItems = spec.minItems;
+    if (spec.description) array.description = spec.description;
+    return { type: 'object', properties: { [name]: array }, required: [name] };
+  },
+};
+
+const ompAdapter: HostAdapter = {
+  label(pi, text) { pi.setLabel?.(text); },
+  on(pi, event, handler) { pi.on(event, handler); },
+  async select(ui, title, options, dialogOptions) {
+    if (!ui?.select) return undefined;
+    return ui.select(title, options, dialogOptions);
+  },
+  inject(event, text, staleMarker) {
+    const marker = staleMarker?.replace(/^[^\w]+/, '').trim();
+    const dropStale = (parts: string[]): string[] => (marker ? parts.filter((part) => !part.includes(marker)) : parts);
+    return { systemPrompt: [...dropStale(asPromptArray(event.systemPrompt)), text] };
+  },
+  stringArray(pi, name, spec) {
+    const { z } = pi.zod as { z: ZodFactory['z'] };
+    const array = z.array(z.string());
+    const constrained = spec.minItems ? array.min(spec.minItems) : array;
+    return z.object({ [name]: spec.description ? constrained.describe(spec.description) : constrained });
+  },
+};
+
+const opencodeAdapter: HostAdapter = {
+  label() { /* no label surface */ },
+  on(pi, event, handler) { pi.on(event, handler); },
+  async select(ui, title, options, dialogOptions) {
+    if (!ui?.select) return undefined;
+    return ui.select(title, options, dialogOptions);
+  },
+  inject(event, text, staleMarker) {
+    const marker = staleMarker?.replace(/^[^\w]+/, '').trim();
+    const dropStale = (parts: string[]): string[] => (marker ? parts.filter((part) => !part.includes(marker)) : parts);
+    return { systemPrompt: [...dropStale(asPromptArray(event.systemPrompt)), text] };
+  },
+  stringArray(_pi, name, spec) {
+    const array: Record<string, unknown> = { type: 'array', items: { type: 'string' } };
+    if (spec.minItems) array.minItems = spec.minItems;
+    if (spec.description) array.description = spec.description;
+    return { type: 'object', properties: { [name]: array }, required: [name] };
+  },
+};
+
+// Explicit hostId wins; the legacy zod sniff keeps old callers working.
+function adapterFor(pi: HostHandle): HostAdapter {
+  if (pi.hostId === 'opencode') return opencodeAdapter;
+  if (pi.hostId === 'omp') return ompAdapter;
+  if (pi.hostId === 'pi') return piAdapter;
+  return pi.zod ? ompAdapter : piAdapter;
 }
 
 export function isPiHost(pi: HostHandle): boolean {
   return !pi.zod;
 }
 
-export function setExtensionLabel(pi: HostHandle, label: string): void {
-  if (isPiHost(pi)) return;
-  pi.setLabel?.(label);
+export function setExtensionLabel(pi: HostHandle, text: string): void {
+  adapterFor(pi).label(pi, text);
 }
 
 export function onHostEvent<E>(pi: ExtensionApi, event: string, handler: (event: E, ctx: ExtensionCtx) => unknown): void {
-  if (event === 'session_branch' && isPiHost(pi)) return;
-  pi.on<E>(event, handler);
+  adapterFor(pi).on<E>(pi, event, handler);
 }
 
-export interface SelectOption {
-  label: string;
-  description?: string;
-}
-
-// Returns the option's label on both hosts: pi renders plain strings, so the
-// description rides along inside the label and is stripped back out.
 export async function hostSelect(
   pi: HostHandle,
   ui: UiApi | undefined,
@@ -61,44 +138,13 @@ export async function hostSelect(
   options: SelectOption[],
   dialogOptions?: Record<string, unknown>,
 ): Promise<string | undefined> {
-  if (!ui?.select) return undefined;
-  if (!isPiHost(pi)) return ui.select(title, options, dialogOptions);
-  const shown = options.map((option) => (option.description ? `${option.label} — ${option.description}` : option.label));
-  const picked = await ui.select(title, shown, dialogOptions);
-  if (picked === undefined) return undefined;
-  return options[shown.indexOf(picked)]?.label ?? picked;
+  return adapterFor(pi).select(ui, title, options, dialogOptions);
 }
 
-// What a `before_agent_start` handler hands back; undefined once pi was
-// updated in place.
-export function injectPromptText(pi: HostHandle, event: SystemPromptEvent, text: string): PromptInjection | undefined {
-  if (!isPiHost(pi)) return { systemPrompt: [...asPromptArray(event.systemPrompt), text] };
-  // Appending keeps pi's sections and the transcript diff; replacing the
-  // prompt the OMP way would drop every section.
-  const options = event.systemPromptOptions;
-  if (options && typeof options.appendSystemPrompt === 'string') {
-    options.appendSystemPrompt = [options.appendSystemPrompt, text].filter(Boolean).join('\n\n');
-    return undefined;
-  }
-  return { systemPrompt: `${asPromptArray(event.systemPrompt).join('\n\n')}\n\n${text}` };
+export function injectPromptText(pi: HostHandle, event: SystemPromptEvent, text: string, staleMarker?: string): PromptInjection | undefined {
+  return adapterFor(pi).inject(event, text, staleMarker);
 }
 
-export interface StringArraySpec {
-  minItems?: number;
-  description?: string;
-}
-
-// One string-array argument in the active host's dialect. pi only requires an
-// object, so a JSON Schema literal suffices and no `typebox` import is needed.
 export function stringArrayToolParams(pi: ExtensionApi, name: string, spec: StringArraySpec = {}): ToolParams {
-  if (!isPiHost(pi)) {
-    const { z } = pi.zod as { z: ZodFactory['z'] };
-    const array = z.array(z.string());
-    const constrained = spec.minItems ? array.min(spec.minItems) : array;
-    return z.object({ [name]: spec.description ? constrained.describe(spec.description) : constrained });
-  }
-  const array: Record<string, unknown> = { type: 'array', items: { type: 'string' } };
-  if (spec.minItems) array.minItems = spec.minItems;
-  if (spec.description) array.description = spec.description;
-  return { type: 'object', properties: { [name]: array }, required: [name] };
+  return adapterFor(pi).stringArray(pi, name, spec);
 }

package/extensions/shared/plugin-settings.js

@@ -1,29 +1,17 @@
-// Settings reader for the Tersio extensions.
-//
-// One file, under Tersio's own home (~/.tersio/settings.json), serves every
-// host: OMP has no reason to keep its own copy and pi never had a store. The
-// profile CLI writes the same file, so the extensions, `tersio settings`, and
-// both hosts always agree.
-//
-// Tolerant: any parse failure yields {} so every caller falls back to its own
-// default.
+// One settings file for every host; parse failure yields {} with per-caller defaults.
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
-import os from 'node:os';
 import path from 'node:path';
-import { tersioHome } from '../lib/utils.js';
+import { homeDir, tersioHome } from '../lib/utils.js';
 export const PLUGIN_NAME = '@krtclcdy/tersio';
-// Session-start defaults, shared by every host: one file under Tersio's own
-// home, so OMP and pi read the same values. This module owns the path.
+// The one store both hosts read; this module owns its path.
 export function tersioSettingsFile() {
     return path.join(tersioHome(), 'settings.json');
 }
-// Older installs stored the same values as OMP plugin settings, and an install
-// that predates ~/.tersio still has them there. Read once and keep them, so
-// switching the store does not reset a user's defaults.
+// Older installs kept these as OMP plugin settings; still read so switching the store does not reset a user's defaults.
 function legacyOmpSettings() {
-    const configHome = process.env.XDG_CONFIG_HOME || path.join(os.homedir(), '.config');
+    const configHome = process.env.XDG_CONFIG_HOME || path.join(homeDir(), '.config');
     for (const p of [
-        path.join(os.homedir(), '.omp', 'plugins', 'omp-plugins.lock.json'),
+        path.join(homeDir(), '.omp', 'plugins', 'omp-plugins.lock.json'),
         path.join(configHome, 'omp', 'plugins', 'omp-plugins.lock.json'),
     ]) {
         if (!existsSync(p))

package/extensions/shared/plugin-settings.ts

@@ -1,33 +1,21 @@
-// Settings reader for the Tersio extensions.
-//
-// One file, under Tersio's own home (~/.tersio/settings.json), serves every
-// host: OMP has no reason to keep its own copy and pi never had a store. The
-// profile CLI writes the same file, so the extensions, `tersio settings`, and
-// both hosts always agree.
-//
-// Tolerant: any parse failure yields {} so every caller falls back to its own
-// default.
+// One settings file for every host; parse failure yields {} with per-caller defaults.
 
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
-import os from 'node:os';
 import path from 'node:path';
-import { tersioHome } from '../lib/utils.ts';
+import { homeDir, tersioHome } from '../lib/utils.ts';
 
 export const PLUGIN_NAME = '@krtclcdy/tersio';
 
-// Session-start defaults, shared by every host: one file under Tersio's own
-// home, so OMP and pi read the same values. This module owns the path.
+// The one store both hosts read; this module owns its path.
 export function tersioSettingsFile(): string {
   return path.join(tersioHome(), 'settings.json');
 }
 
-// Older installs stored the same values as OMP plugin settings, and an install
-// that predates ~/.tersio still has them there. Read once and keep them, so
-// switching the store does not reset a user's defaults.
+// Older installs kept these as OMP plugin settings; still read so switching the store does not reset a user's defaults.
 function legacyOmpSettings(): Record<string, unknown> | null {
-  const configHome = process.env.XDG_CONFIG_HOME || path.join(os.homedir(), '.config');
+  const configHome = process.env.XDG_CONFIG_HOME || path.join(homeDir(), '.config');
   for (const p of [
-    path.join(os.homedir(), '.omp', 'plugins', 'omp-plugins.lock.json'),
+    path.join(homeDir(), '.omp', 'plugins', 'omp-plugins.lock.json'),
     path.join(configHome, 'omp', 'plugins', 'omp-plugins.lock.json'),
   ]) {
     if (!existsSync(p)) continue;

package/extensions/shared/pricing.js

@@ -1,14 +1,11 @@
-// extensions/shared/pricing.ts — dynamic model pricing from LiteLLM.
-// Exact id → provider-prefix strip → default, which reports with known:false so
-// the Dashboard labels an unpriced model honestly. No static table: prices rot.
+// extensions/shared/pricing.ts — dynamic model pricing from LiteLLM. Exact id → provider-prefix strip → default, reported with known:false when unpriced.
 import fs from 'node:fs';
 import path from 'node:path';
 import { tersioDataPath } from '../lib/utils.js';
 // Sonnet-class default for unpriced models; always paired with known:false.
 export const DEFAULT_PRICE = { input: 2, output: 10, cacheRead: 0.2, cacheWrite: 2.5 };
 const LITELLM_URL = 'https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json';
-// Readers use the file even when stale, and only one refresh runs per process;
-// `tersio update` still forces a fresh fetch.
+// Readers use the file even when stale, and only one refresh runs per process; `tersio update` still forces a fresh fetch.
 const CACHE_TTL_MS = 7 * 24 * 3600 * 1000;
 let refreshInflight = null;
 export function pricesCachePath() {
@@ -23,8 +20,7 @@ export function pricesUrl() {
 function toPrice(t) {
     return { input: t[0], output: t[1], cacheRead: t[2], cacheWrite: t[3] };
 }
-// Tolerate both cache shapes (compact tuples and ModelPrice objects) so a
-// version bump never orphans the cache.
+// Tolerate both cache shapes (compact tuples and ModelPrice objects) so a version bump never orphans the cache.
 function asPrice(v) {
     if (Array.isArray(v) && v.length === 4 && v.every((n) => typeof n === 'number' && Number.isFinite(n) && n >= 0)) {
         return toPrice(v);
@@ -32,14 +28,30 @@ function asPrice(v) {
     if (v && typeof v === 'object') {
         const o = v;
         if (['input', 'output', 'cacheRead', 'cacheWrite'].every((k) => typeof o[k] === 'number' && Number.isFinite(o[k]) && o[k] >= 0)) {
-            // SAFETY: the line above verified all four keys are finite numbers >= 0,
-            // which is the whole of ModelPrice.
+            // SAFETY: the line above verified all four keys are finite numbers >= 0, which is the whole of ModelPrice.
             return o;
         }
     }
     return null;
 }
+let liveCache = null;
 export function loadLivePrices() {
+    // One stat per call beats one full parse: the recent-rows map prices thousands of rows.
+    try {
+        const file = pricesCachePath();
+        const mtime = fs.statSync(file).mtimeMs;
+        if (liveCache && liveCache.path === file && liveCache.mtime === mtime)
+            return liveCache.prices;
+        const prices = readLivePrices();
+        if (prices)
+            liveCache = { path: file, mtime, prices };
+        return prices;
+    }
+    catch {
+        return null;
+    }
+}
+function readLivePrices() {
     let raw;
     try {
         raw = fs.readFileSync(pricesCachePath(), 'utf8');
@@ -70,37 +82,78 @@ export function loadLivePrices() {
 function num(v) {
     return typeof v === 'number' && Number.isFinite(v) && v >= 0 ? v : null;
 }
-// One model, two spellings: transcripts say `space-bunny`, the feed keys it
-// `openrouter/stealth/space-bunny-alpha` at zero. Aliases fold to the first, so
-// a free stealth model never reports the default price.
+// Transcripts and the feed spell one model differently; aliases stop a free model reporting the default price.
 export const MODEL_ALIASES = [
     { id: 'space-bunny', feed: 'stealth/space-bunny-alpha' },
 ];
 export function canonicalPriceId(model) {
     const key = model.toLowerCase();
-    return MODEL_ALIASES.find((a) => a.id === key || a.feed === key)?.id ?? key;
+    // Suffix match: gateways prefix the id (opencode/kilo/stealth/space-bunny-alpha).
+    return MODEL_ALIASES.find((a) => a.id === key || a.feed === key || key.endsWith(`/${a.id}`) || key.endsWith(`/${a.feed}`))?.id ?? key;
 }
 export function priceFor(model, live) {
     const table = live === undefined ? loadLivePrices() : live;
-    // An alias goes through the same exact-then-suffix chain as any other id, so
-    // it still resolves when the feed keys it under a provider prefix.
+    if (live === undefined) {
+        const memo = priceMemo(table);
+        const hit = memo.get(model);
+        if (hit)
+            return hit;
+        const resolved = resolvePrice(model, table);
+        memo.set(model, resolved);
+        return resolved;
+    }
+    return resolvePrice(model, table);
+}
+// Recent rows repeat a handful of models; the memo turns thousands of lookups into one per model. Cleared whenever the table identity changes.
+let memoTable;
+let memo = new Map();
+function priceMemo(table) {
+    if (memoTable !== table) {
+        memoTable = table;
+        memo = new Map();
+    }
+    return memo;
+}
+// Validated once per table instead of once per lookup.
+let compiledTable = null;
+let compiledEntries = [];
+function priceEntries(table) {
+    if (compiledTable === table)
+        return compiledEntries;
+    const entries = [];
+    for (const [k, v] of Object.entries(table.exact)) {
+        const price = asPrice(v);
+        if (price)
+            entries.push([k, price]);
+    }
+    compiledTable = table;
+    compiledEntries = entries;
+    return entries;
+}
+function resolvePrice(model, table) {
+    // An alias goes through the same exact-then-suffix chain as any other id, so it still resolves when the feed keys it under a provider prefix.
     const id = MODEL_ALIASES.find((a) => a.id === model.toLowerCase())?.feed ?? model;
     if (table) {
-        const hit = asPrice(table.exact[id] ?? table.exact[id.toLowerCase()]);
+        const entries = priceEntries(table);
+        const lower = id.toLowerCase();
+        const hit = entries.find(([k]) => k === id) ?? entries.find(([k]) => k === lower);
         if (hit)
-            return { price: hit, known: true, live: true };
+            return { price: hit[1], known: true, live: true };
         // Provider-prefixed ids ("azure/gpt-4o", "dashscope/qwen-max"): match the
-        // first cached id that ends with the bare model name.
-        const name = id.toLowerCase();
-        const key = Object.keys(table.exact).find((k) => k.toLowerCase() === name || k.toLowerCase().endsWith(`/${name}`));
-        const prefixed = key ? asPrice(table.exact[key]) : null;
-        if (prefixed)
-            return { price: prefixed, known: true, live: true };
+        // first cached key that ends with the full id, then with the bare tail
+        // segment, so "codex/openai/gpt-6-luna" still finds "gpt-6-luna".
+        const name = lower;
+        const tail = name.split('/').pop() ?? name;
+        const key = entries.find(([k]) => {
+            const lk = k.toLowerCase();
+            return lk === name || lk.endsWith(`/${name}`) || lk === tail || lk.endsWith(`/${tail}`);
+        });
+        if (key)
+            return { price: key[1], known: true, live: true };
     }
     return { price: DEFAULT_PRICE, known: false, live: false };
 }
-// Fetch LiteLLM pricing and write the cache. False on any failure — callers
-// fall back silently.
+// Fetch LiteLLM pricing and write the cache. False on any failure — callers fall back silently.
 export async function refreshPrices() {
     if (refreshInflight)
         return refreshInflight;
@@ -164,8 +217,7 @@ export async function refreshPrices() {
         refreshInflight = null;
     }
 }
-// Fire-and-forget refresh when the cache is stale; readers keep the stale
-// file, so pricing never blocks.
+// Fire-and-forget refresh when the cache is stale; readers keep the stale file, so pricing never blocks.
 export function refreshPricesIfStale() {
     const live = loadLivePrices();
     if (live && Date.now() - live.fetchedAt <= CACHE_TTL_MS)