@warmdrift/kgauto-compiler 2.0.0-alpha.10 → 2.0.0-alpha.12

package/dist/index.js

@@ -38,6 +38,7 @@ __export(index_exports, {
   countTokens: () => countTokens,
   execute: () => execute,
   getAllStarterChains: () => getAllStarterChains,
+  getArchetypePerfScore: () => getArchetypePerfScore,
   getDefaultFallbackChain: () => getDefaultFallbackChain,
   getProfile: () => getProfile,
   getReachabilityDiagnostic: () => getReachabilityDiagnostic,
@@ -47,9 +48,16 @@ __export(index_exports, {
   isModelReachable: () => isModelReachable,
   isProviderReachable: () => isProviderReachable,
   learningKey: () => learningKey,
+  loadAliasesFromBrain: () => loadAliasesFromBrain,
+  loadArchetypePerfFromBrain: () => loadArchetypePerfFromBrain,
+  loadChainsFromBrain: () => loadChainsFromBrain,
+  loadModelsFromBrain: () => loadModelsFromBrain,
+  loadPricingFromBrain: () => loadPricingFromBrain,
+  profileToRow: () => profileToRow,
   profilesByProvider: () => profilesByProvider,
   record: () => record,
   resetTokenizer: () => resetTokenizer,
+  resolvePricingAt: () => resolvePricingAt,
   resolveProviderKey: () => resolveProviderKey,
   runAdvisor: () => runAdvisor,
   setTokenizer: () => setTokenizer,
@@ -1439,14 +1447,20 @@ var ALIASES = {
   // Legacy kgauto typo — actual API alias is dash-form (alpha.1 had dot).
   "claude-haiku-4.5": "claude-haiku-4-5"
 };
+var brainHook = {};
+function _setProfileBrainHook(hook) {
+  brainHook = hook;
+}
 function canonicalId(id) {
-  return ALIASES[id] ?? id;
+  return brainHook.resolveAlias?.(id) ?? ALIASES[id] ?? id;
 }
 var PROFILE_INDEX = new Map(
   PROFILES_RAW.map((p) => [p.id, p])
 );
 function getProfile(id) {
   const canonical = canonicalId(id);
+  const fromBrain = brainHook.getProfile?.(canonical);
+  if (fromBrain) return fromBrain;
   const p = PROFILE_INDEX.get(canonical);
   if (!p) {
     const known = [...PROFILE_INDEX.keys(), ...Object.keys(ALIASES)].join(", ");
@@ -1455,11 +1469,15 @@ function getProfile(id) {
   return p;
 }
 function tryGetProfile(id) {
-  return PROFILE_INDEX.get(canonicalId(id));
+  const canonical = canonicalId(id);
+  return brainHook.getProfile?.(canonical) ?? PROFILE_INDEX.get(canonical);
 }
 function allProfiles() {
   return PROFILES_RAW;
 }
+function allProfilesRaw() {
+  return PROFILES_RAW;
+}
 function profilesByProvider(provider) {
   return PROFILES_RAW.filter((p) => p.provider === provider);
 }
@@ -1683,14 +1701,170 @@ function validateFinalFit(ir, profile, tokens) {
   }
 }
 
+// src/brain-query.ts
+var FRESH_SNAPSHOT = {
+  data: null,
+  expiresAt: 0,
+  refreshing: false,
+  warned: false
+};
+var snapshot = { ...FRESH_SNAPSHOT };
+var runtime;
+function configureBrainQuery(rt) {
+  runtime = rt;
+  snapshot = { ...FRESH_SNAPSHOT };
+}
+function createBrainQueryCache(opts) {
+  return () => {
+    const rt = runtime;
+    if (!rt || !rt.enabledTables.has(opts.table)) {
+      return opts.bundledFallback();
+    }
+    const now = Date.now();
+    const stale = snapshot.expiresAt <= now;
+    if (stale && !snapshot.refreshing) {
+      snapshot.refreshing = true;
+      void asyncRefresh(rt);
+    }
+    if (snapshot.data) {
+      const rows = snapshot.data[opts.table];
+      if (Array.isArray(rows) && rows.length > 0) {
+        try {
+          return opts.mapRows(rows);
+        } catch {
+          return opts.bundledFallback();
+        }
+      }
+    }
+    return opts.bundledFallback();
+  };
+}
+var pendingRefresh;
+async function asyncRefresh(rt) {
+  const promise = doRefresh(rt);
+  pendingRefresh = promise;
+  try {
+    await promise;
+  } finally {
+    if (pendingRefresh === promise) pendingRefresh = void 0;
+  }
+}
+var DEFAULT_CONFIG_URL = "https://kgauto-dashboard.vercel.app/api/kgauto-v2/config";
+async function doRefresh(rt) {
+  const url = rt.configEndpoint ?? DEFAULT_CONFIG_URL;
+  try {
+    const res = await rt.fetchImpl(url, { method: "GET" });
+    if (!res.ok) {
+      throw new Error(`brain-query ${res.status}: ${res.statusText}`);
+    }
+    const body = await res.json();
+    if (runtime !== rt) return;
+    snapshot = {
+      data: body,
+      expiresAt: Date.now() + rt.ttlMs,
+      refreshing: false,
+      warned: snapshot.warned
+    };
+  } catch (err) {
+    if (runtime !== rt) return;
+    snapshot.refreshing = false;
+    snapshot.expiresAt = Date.now() + rt.ttlMs;
+    if (!snapshot.warned) {
+      snapshot.warned = true;
+      (rt.onError ?? defaultOnError)(err);
+    }
+  }
+}
+function defaultOnError(err) {
+  console.warn("[kgauto] brain-query failed (using bundled fallback):", err);
+}
+
+// src/pricing-brain.ts
+function isPricingRow(x) {
+  if (!x || typeof x !== "object") return false;
+  const r = x;
+  return typeof r.model_id === "string" && typeof r.cost_input_per_1m === "number" && typeof r.cost_output_per_1m === "number" && typeof r.valid_from === "string";
+}
+function mapRowsToPricing(rows) {
+  const out = [];
+  for (const row of rows) {
+    if (!isPricingRow(row)) continue;
+    out.push({
+      modelId: row.model_id,
+      costInputPer1m: row.cost_input_per_1m,
+      costOutputPer1m: row.cost_output_per_1m,
+      cacheInputPer1m: row.cache_input_per_1m ?? void 0,
+      cacheCreationPer1m: row.cache_creation_per_1m ?? void 0,
+      validFrom: Date.parse(row.valid_from),
+      validUntil: row.valid_until == null ? void 0 : Date.parse(row.valid_until),
+      source: row.source ?? void 0
+    });
+  }
+  return out;
+}
+function bundledPricing() {
+  const out = [];
+  for (const profile of allProfiles()) {
+    out.push({
+      modelId: profile.id,
+      costInputPer1m: profile.costInputPer1m,
+      costOutputPer1m: profile.costOutputPer1m,
+      cacheInputPer1m: profile.lowering.cache.discount !== void 0 && profile.lowering.cache.discount > 0 ? profile.costInputPer1m * profile.lowering.cache.discount : void 0,
+      validFrom: 0,
+      validUntil: void 0,
+      source: "profile_seed"
+    });
+  }
+  return out;
+}
+var loadPricingFromBrain = createBrainQueryCache({
+  table: "kgauto_pricing",
+  mapRows: mapRowsToPricing,
+  bundledFallback: bundledPricing
+});
+function resolvePricingAt(modelId, at = /* @__PURE__ */ new Date()) {
+  const ts = at.getTime();
+  const all = loadPricingFromBrain();
+  let best;
+  for (const row of all) {
+    if (row.modelId !== modelId) continue;
+    if (row.validFrom > ts) continue;
+    if (row.validUntil !== void 0 && row.validUntil <= ts) continue;
+    if (!best || row.validFrom > best.validFrom) best = row;
+  }
+  return best;
+}
+
 // src/brain.ts
 var activeConfig;
 function configureBrain(config) {
   const endpoint = config.endpoint.replace(/\/outcomes\/?$/, "");
   activeConfig = { ...config, endpoint };
+  const bq = config.brainQuery ?? {};
+  const enabledTables = /* @__PURE__ */ new Set();
+  if (bq.chains !== false) enabledTables.add("kgauto_chains");
+  if (bq.perf !== false) enabledTables.add("kgauto_archetype_perf");
+  if (bq.pricing !== false) enabledTables.add("kgauto_pricing");
+  if (bq.models !== false) {
+    enabledTables.add("kgauto_models");
+    enabledTables.add("kgauto_aliases");
+  }
+  if (enabledTables.size === 0) {
+    configureBrainQuery(void 0);
+    return;
+  }
+  configureBrainQuery({
+    endpoint,
+    configEndpoint: bq.configEndpoint,
+    ttlMs: bq.cacheTtlMs ?? 3e5,
+    fetchImpl: config.fetchImpl ?? fetch,
+    enabledTables,
+    onError: config.onError
+  });
 }
 function clearBrain() {
   activeConfig = void 0;
+  configureBrainQuery(void 0);
 }
 var compileRegistry = /* @__PURE__ */ new Map();
 var REGISTRY_MAX_ENTRIES = 1e4;
@@ -1755,7 +1929,7 @@ async function record(input) {
         throw new Error(`brain ${res.status}: ${text}`);
       }
     } catch (err) {
-      (config.onError ?? defaultOnError)(err);
+      (config.onError ?? defaultOnError2)(err);
     }
   };
   if (config.sync) {
@@ -1764,7 +1938,7 @@ async function record(input) {
     void send();
   }
 }
-function defaultOnError(err) {
+function defaultOnError2(err) {
   console.warn("[kgauto] brain record failed:", err);
 }
 function buildPayload(input, reg) {
@@ -1808,6 +1982,12 @@ function buildPayload(input, reg) {
 }
 function computeCostUsd(modelId, tokensIn, tokensOut) {
   if (tokensIn === 0 && tokensOut === 0) return void 0;
+  const brainRow = resolvePricingAt(modelId);
+  if (brainRow && (brainRow.costInputPer1m > 0 || brainRow.costOutputPer1m > 0)) {
+    const inUsd2 = tokensIn / 1e6 * brainRow.costInputPer1m;
+    const outUsd2 = tokensOut / 1e6 * brainRow.costOutputPer1m;
+    return Math.round((inUsd2 + outUsd2) * 1e6) / 1e6;
+  }
   const profile = tryGetProfile(modelId);
   if (!profile) return void 0;
   const inUsd = tokensIn / 1e6 * profile.costInputPer1m;
@@ -2390,6 +2570,37 @@ function clamp(n) {
   return Math.max(0, Math.min(1, n));
 }
 
+// src/chains-brain.ts
+function isChainsRow(x) {
+  if (!x || typeof x !== "object") return false;
+  const r = x;
+  return typeof r.archetype === "string" && typeof r.tier === "number" && typeof r.model_id === "string";
+}
+function mapRowsToChains(rows) {
+  const grouped = /* @__PURE__ */ new Map();
+  for (const row of rows) {
+    if (!isChainsRow(row)) continue;
+    const list = grouped.get(row.archetype) ?? [];
+    list.push(row);
+    grouped.set(row.archetype, list);
+  }
+  const out = {};
+  for (const [archetype, group] of grouped.entries()) {
+    group.sort((a, b) => a.tier - b.tier);
+    out[archetype] = group.map((r) => r.model_id);
+  }
+  const bundled = getAllStarterChains();
+  for (const archetype of Object.keys(bundled)) {
+    if (!out[archetype]) out[archetype] = bundled[archetype];
+  }
+  return out;
+}
+var loadChainsFromBrain = createBrainQueryCache({
+  table: "kgauto_chains",
+  mapRows: mapRowsToChains,
+  bundledFallback: getAllStarterChains
+});
+
 // src/fallback.ts
 var STARTER_CHAINS = {
   // Reasoning floor — never degrade. Walk UP on 429 to Opus → cross-provider.
@@ -2467,10 +2678,11 @@ function getDefaultFallbackChain(opts) {
       `getDefaultFallbackChain: maxDepth must be >= 1, got ${maxDepth}`
     );
   }
-  const starter = STARTER_CHAINS[archetype];
+  const allChains = loadChainsFromBrain();
+  const starter = allChains[archetype];
   if (!starter) {
     throw new Error(
-      `getDefaultFallbackChain: unknown archetype "${archetype}". Known: ${Object.keys(STARTER_CHAINS).join(", ")}`
+      `getDefaultFallbackChain: unknown archetype "${archetype}". Known: ${Object.keys(allChains).join(", ")}`
     );
   }
   let chain;
@@ -2514,6 +2726,157 @@ function getAllStarterChains() {
   return out;
 }
 
+// src/archetype-perf-brain.ts
+function isPerfRow(x) {
+  if (!x || typeof x !== "object") return false;
+  const r = x;
+  return typeof r.model_id === "string" && typeof r.archetype === "string" && typeof r.perf_score === "number";
+}
+function mapRowsToPerfMap(rows) {
+  const out = /* @__PURE__ */ new Map();
+  for (const row of rows) {
+    if (!isPerfRow(row)) continue;
+    const existing = out.get(row.model_id) ?? {};
+    existing[row.archetype] = row.perf_score;
+    out.set(row.model_id, existing);
+  }
+  return out;
+}
+function bundledArchetypePerf() {
+  const out = /* @__PURE__ */ new Map();
+  for (const profile of allProfiles()) {
+    if (profile.archetypePerf) out.set(profile.id, profile.archetypePerf);
+  }
+  return out;
+}
+var loadArchetypePerfFromBrain = createBrainQueryCache({
+  table: "kgauto_archetype_perf",
+  mapRows: mapRowsToPerfMap,
+  bundledFallback: bundledArchetypePerf
+});
+function getArchetypePerfScore(modelId, archetype) {
+  return loadArchetypePerfFromBrain().get(modelId)?.[archetype] ?? 5;
+}
+
+// src/models-brain.ts
+function isModelRow(x) {
+  if (!x || typeof x !== "object") return false;
+  const r = x;
+  return typeof r.model_id === "string" && typeof r.provider === "string";
+}
+function isAliasRow(x) {
+  if (!x || typeof x !== "object") return false;
+  const r = x;
+  return typeof r.alias_id === "string" && typeof r.canonical_id === "string";
+}
+function rowToProfile(row) {
+  try {
+    if (row.cliffs !== void 0 && row.cliffs !== null && !Array.isArray(row.cliffs)) {
+      return null;
+    }
+    if (row.recovery !== void 0 && row.recovery !== null && !Array.isArray(row.recovery)) {
+      return null;
+    }
+    if (row.lowering !== void 0 && row.lowering !== null && (typeof row.lowering !== "object" || Array.isArray(row.lowering))) {
+      return null;
+    }
+    return {
+      id: row.model_id,
+      provider: row.provider,
+      status: row.status ?? "current",
+      maxContextTokens: row.max_context_tokens ?? 0,
+      maxOutputTokens: row.max_output_tokens ?? 0,
+      maxTools: row.max_tools ?? 0,
+      parallelToolCalls: row.parallel_tool_calls ?? false,
+      structuredOutput: row.structured_output ?? "none",
+      systemPromptMode: row.system_prompt_mode ?? "inline",
+      streaming: row.streaming ?? true,
+      cliffs: row.cliffs ?? [],
+      costInputPer1m: row.cost_input_per_1m ?? 0,
+      costOutputPer1m: row.cost_output_per_1m ?? 0,
+      lowering: row.lowering ?? { system: { mode: "inline" }, cache: { strategy: "unsupported" } },
+      recovery: row.recovery ?? [],
+      strengths: row.strengths ?? [],
+      weaknesses: row.weaknesses ?? [],
+      notes: row.notes ?? void 0,
+      verifiedAgainstDocs: row.verified_against_docs ?? void 0,
+      archetypePerf: row.archetype_perf ?? void 0
+    };
+  } catch {
+    return null;
+  }
+}
+function profileToRow(profile, opts = {}) {
+  const row = {
+    model_id: profile.id,
+    provider: profile.provider,
+    status: profile.status,
+    max_context_tokens: profile.maxContextTokens,
+    max_output_tokens: profile.maxOutputTokens,
+    max_tools: profile.maxTools,
+    parallel_tool_calls: profile.parallelToolCalls,
+    structured_output: profile.structuredOutput,
+    system_prompt_mode: profile.systemPromptMode,
+    streaming: profile.streaming,
+    cliffs: profile.cliffs,
+    cost_input_per_1m: profile.costInputPer1m,
+    cost_output_per_1m: profile.costOutputPer1m,
+    lowering: profile.lowering,
+    recovery: profile.recovery,
+    strengths: profile.strengths,
+    weaknesses: profile.weaknesses,
+    notes: profile.notes ?? null,
+    archetype_perf: profile.archetypePerf ?? null,
+    active: opts.active ?? true
+  };
+  if (opts.verifiedAgainstDocs !== void 0) {
+    row.verified_against_docs = opts.verifiedAgainstDocs;
+  } else if (profile.verifiedAgainstDocs !== void 0) {
+    const v = profile.verifiedAgainstDocs;
+    row.verified_against_docs = /^\d{4}-\d{2}-\d{2}/.test(v) ? v : null;
+  }
+  if (opts.versionAdded !== void 0) row.version_added = opts.versionAdded;
+  if (opts.versionRemoved !== void 0) row.version_removed = opts.versionRemoved;
+  return row;
+}
+function mapRowsToModels(rows) {
+  const out = /* @__PURE__ */ new Map();
+  for (const row of rows) {
+    if (!isModelRow(row)) continue;
+    const profile = rowToProfile(row);
+    if (profile) out.set(profile.id, profile);
+  }
+  return out;
+}
+function mapRowsToAliases(rows) {
+  const out = {};
+  for (const row of rows) {
+    if (!isAliasRow(row)) continue;
+    out[row.alias_id] = row.canonical_id;
+  }
+  return out;
+}
+function bundledModels() {
+  return new Map(allProfilesRaw().map((p) => [p.id, p]));
+}
+function bundledAliases() {
+  return { ...ALIASES };
+}
+var loadModelsFromBrain = createBrainQueryCache({
+  table: "kgauto_models",
+  mapRows: mapRowsToModels,
+  bundledFallback: bundledModels
+});
+var loadAliasesFromBrain = createBrainQueryCache({
+  table: "kgauto_aliases",
+  mapRows: mapRowsToAliases,
+  bundledFallback: bundledAliases
+});
+_setProfileBrainHook({
+  getProfile: (canonical) => loadModelsFromBrain().get(canonical),
+  resolveAlias: (id) => loadAliasesFromBrain()[id]
+});
+
 // src/index.ts
 function compile2(ir, opts) {
   const result = compile(ir, opts);
@@ -2540,6 +2903,7 @@ function compile2(ir, opts) {
   countTokens,
   execute,
   getAllStarterChains,
+  getArchetypePerfScore,
   getDefaultFallbackChain,
   getProfile,
   getReachabilityDiagnostic,
@@ -2549,9 +2913,16 @@ function compile2(ir, opts) {
   isModelReachable,
   isProviderReachable,
   learningKey,
+  loadAliasesFromBrain,
+  loadArchetypePerfFromBrain,
+  loadChainsFromBrain,
+  loadModelsFromBrain,
+  loadPricingFromBrain,
+  profileToRow,
   profilesByProvider,
   record,
   resetTokenizer,
+  resolvePricingAt,
   resolveProviderKey,
   runAdvisor,
   setTokenizer,