@theokit/sdk 4.12.2 → 4.13.1

package/dist/index.js

@@ -573,10 +573,10 @@ function buildToolPrompt(prompt) {
 Respond by calling the \`output\` tool with the structured answer that matches the schema.`;
 }
 function setupStructuredOutput(schema, maxRetries) {
-  const z12 = requireZod();
+  const z13 = requireZod();
   const jsonSchema = toJsonSchema(schema, { unrepresentable: "any" });
   return {
-    z: z12,
+    z: z13,
     jsonSchema,
     maxRetries: maxRetries ?? 1,
     initialUsage: { inputTokens: 0, outputTokens: 0 }
@@ -13083,6 +13083,192 @@ function applyToolResultGuard(parts, opts) {
     )
   );
 }
+var MODALITIES = ["text", "audio", "image", "video", "pdf"];
+var costSchema = z.object({
+  /** USD per 1M tokens (models.dev convention). */
+  input: z.number().nonnegative(),
+  output: z.number().nonnegative(),
+  cache_read: z.number().nonnegative().optional(),
+  cache_write: z.number().nonnegative().optional()
+}).loose();
+var limitSchema = z.object({
+  context: z.number().positive(),
+  input: z.number().positive().optional(),
+  output: z.number().positive().optional()
+}).loose();
+var modalitiesSchema = z.object({
+  input: z.array(z.enum(MODALITIES)).optional(),
+  output: z.array(z.enum(MODALITIES)).optional()
+}).loose();
+var catalogModelSchema = z.object({
+  name: z.string().optional(),
+  release_date: z.string().optional(),
+  attachment: z.boolean().optional(),
+  reasoning: z.boolean().optional(),
+  temperature: z.boolean().optional(),
+  tool_call: z.boolean().optional(),
+  /** theokit extension — maps to ModelCapabilities.supportsStructuredOutput. */
+  structured_output: z.boolean().optional(),
+  /** theokit extension — maps to ModelCapabilities.supportsCacheControl. */
+  cache_control: z.boolean().optional(),
+  cost: costSchema.optional(),
+  limit: limitSchema.optional(),
+  modalities: modalitiesSchema.optional(),
+  status: z.enum(["alpha", "beta", "deprecated"]).optional()
+}).loose();
+
+// src/internal/providers/registry.ts
+function globalSingleton(key2, create) {
+  const g = globalThis;
+  const sym = Symbol.for(key2);
+  if (g[sym] === void 0) g[sym] = create();
+  return g[sym];
+}
+var REGISTRY = globalSingleton(
+  "theokit-sdk.providers.registry",
+  () => /* @__PURE__ */ new Map()
+);
+var ALIASES = globalSingleton("theokit-sdk.providers.aliases", () => /* @__PURE__ */ new Map());
+function registerProvider(profile) {
+  if (REGISTRY.has(profile.name)) {
+    process.stderr.write(`[theokit-sdk] Provider "${profile.name}" overridden by user plugin.
+`);
+  }
+  REGISTRY.set(profile.name, profile);
+  for (const alias of profile.aliases ?? []) {
+    const previous = ALIASES.get(alias);
+    if (previous !== void 0 && previous !== profile.name) {
+      process.stderr.write(
+        `[theokit-sdk] Alias "${alias}" collision: was "${previous}", now "${profile.name}".
+`
+      );
+    }
+    ALIASES.set(alias, profile.name);
+  }
+}
+function getProviderProfile(name) {
+  const canonical = ALIASES.get(name) ?? name;
+  return REGISTRY.get(canonical);
+}
+function listProviders() {
+  return Array.from(REGISTRY.values());
+}
+
+// src/internal/providers/catalog-loader.ts
+var __dirname_resolved = dirname(fileURLToPath(import.meta.url));
+function globalSingleton2(key2, create) {
+  const g = globalThis;
+  const sym = Symbol.for(key2);
+  if (g[sym] === void 0) g[sym] = create();
+  return g[sym];
+}
+var modelInfoIndex = globalSingleton2(
+  "theokit-sdk.providers.model-info-index",
+  () => /* @__PURE__ */ new Map()
+);
+var patchedModelKeys = globalSingleton2(
+  "theokit-sdk.providers.model-info-patched",
+  () => /* @__PURE__ */ new Set()
+);
+var indexState = globalSingleton2("theokit-sdk.providers.model-info-loaded", () => ({
+  loaded: false
+}));
+function getCatalogModelInfo(key2) {
+  ensureModelIndexLoaded();
+  return modelInfoIndex.get(key2);
+}
+function isPatchedModelKey(key2) {
+  return patchedModelKeys.has(key2);
+}
+function ensureModelIndexLoaded() {
+  if (indexState.loaded) return;
+  indexState.loaded = true;
+  try {
+    const catalog = loadProviderCatalog();
+    for (const entry of Object.values(catalog)) {
+      indexEntryModels(entry);
+    }
+  } catch (err) {
+    process.stderr.write(
+      `[theokit-sdk] WARN: provider catalog unavailable (${err.message}) \u2014 per-model data disabled
+`
+    );
+  }
+}
+function indexEntryModels(entry) {
+  if (entry.models === void 0 || typeof entry.models !== "object") return;
+  for (const [modelId, raw] of Object.entries(entry.models)) {
+    const parsed = catalogModelSchema.safeParse(raw);
+    if (!parsed.success) {
+      process.stderr.write(
+        `[theokit-sdk] WARN: Skipping malformed catalog model "${entry.id}/${modelId}": ${parsed.error.issues[0]?.message ?? "invalid"}
+`
+      );
+      continue;
+    }
+    modelInfoIndex.set(`${entry.id}/${modelId}`, parsed.data);
+    for (const alias of entry.aliases ?? []) {
+      const key2 = `${alias}/${modelId}`;
+      if (!modelInfoIndex.has(key2)) modelInfoIndex.set(key2, parsed.data);
+    }
+  }
+}
+function validateEntry(raw) {
+  if (typeof raw.id !== "string" || typeof raw.displayName !== "string" || typeof raw.apiMode !== "string" || typeof raw.authType !== "string" || typeof raw.baseUrl !== "string" || !Array.isArray(raw.envVars) || !Array.isArray(raw.fallbackModels) || raw.capabilities == null || typeof raw.capabilities !== "object") {
+    return null;
+  }
+  return raw;
+}
+function loadProviderCatalog(opts) {
+  const catalogPath = join(__dirname_resolved, "provider-catalog.json");
+  const rawText = readFileSync(catalogPath, "utf-8");
+  let entries = JSON.parse(rawText);
+  const result = {};
+  for (const raw of entries) {
+    const validated = validateEntry(raw);
+    if (validated === null) {
+      process.stderr.write(
+        `[theokit-sdk] WARN: Skipping malformed catalog entry: ${JSON.stringify(raw).slice(0, 100)}
+`
+      );
+      continue;
+    }
+    result[validated.id] = validated;
+  }
+  return result;
+}
+var _capabilitiesCache = null;
+function getCatalogCapabilities(providerId) {
+  if (_capabilitiesCache === null) {
+    const catalog = loadProviderCatalog();
+    _capabilitiesCache = {};
+    for (const entry of Object.values(catalog)) {
+      _capabilitiesCache[entry.id] = entry.capabilities;
+    }
+  }
+  return _capabilitiesCache[providerId];
+}
+function registerCatalogProviders(opts) {
+  const catalog = loadProviderCatalog();
+  for (const entry of Object.values(catalog)) {
+    if (getProviderProfile(entry.id) !== void 0) continue;
+    if (entry.aliases?.some((a) => getProviderProfile(a) !== void 0)) continue;
+    const profile = {
+      name: entry.id,
+      apiMode: entry.apiMode,
+      authType: entry.authType,
+      baseUrl: entry.baseUrl,
+      envVars: entry.envVars,
+      fallbackModels: entry.fallbackModels,
+      displayName: entry.displayName,
+      aliases: entry.aliases,
+      modelsUrl: entry.modelsUrl,
+      hostname: entry.hostname,
+      extraHeaders: entry.extraHeaders
+    };
+    registerProvider(profile);
+  }
+}
 
 // src/internal/budget/pricing-data.json
 var pricing_data_default = {
@@ -13175,9 +13361,9 @@ var pricing_data_default = {
       cacheRead: 0.025
     },
     "openai/o3": {
-      input: 10,
-      output: 40,
-      cacheRead: 2.5
+      input: 2,
+      output: 8,
+      cacheRead: 0.5
     },
     "openai/o3-mini": {
       input: 1.1,
@@ -13265,6 +13451,30 @@ function getPricingEntry(opts) {
       if (found2 !== void 0) return buildEntry(stripped, found2);
     }
   }
+  const catalogEntry = catalogCostFallback(opts.provider, cleaned);
+  if (catalogEntry !== void 0) return catalogEntry;
+  return void 0;
+}
+function catalogCostFallback(provider, cleanedModel) {
+  const stripped = stripDateSuffix(cleanedModel);
+  const candidates = [`${provider}/${cleanedModel}`, cleanedModel];
+  if (stripped !== cleanedModel) candidates.push(`${provider}/${stripped}`, stripped);
+  for (const key2 of candidates) {
+    const info = getCatalogModelInfo(key2);
+    const cost = info?.cost;
+    if (cost === void 0) continue;
+    const [prov, ...modelParts] = key2.split("/");
+    return {
+      provider: prov ?? provider,
+      model: modelParts.join("/") || cleanedModel,
+      inputCostPerMillion: cost.input,
+      outputCostPerMillion: cost.output,
+      ...cost.cache_read !== void 0 ? { cacheReadCostPerMillion: cost.cache_read } : {},
+      ...cost.cache_write !== void 0 ? { cacheWriteCostPerMillion: cost.cache_write } : {},
+      // M44 M5 fix — honest provenance: a key patched by the LIVE models-dev source is not "vendored".
+      pricingVersion: isPatchedModelKey(key2) ? "catalog-models-dev" : "catalog-vendored"
+    };
+  }
   return void 0;
 }
 
@@ -13781,93 +13991,6 @@ function abortError(signal) {
 // src/internal/llm/router.ts
 init_errors();
 
-// src/internal/providers/registry.ts
-var REGISTRY = /* @__PURE__ */ new Map();
-var ALIASES = /* @__PURE__ */ new Map();
-function registerProvider(profile) {
-  if (REGISTRY.has(profile.name)) {
-    process.stderr.write(`[theokit-sdk] Provider "${profile.name}" overridden by user plugin.
-`);
-  }
-  REGISTRY.set(profile.name, profile);
-  for (const alias of profile.aliases ?? []) {
-    const previous = ALIASES.get(alias);
-    if (previous !== void 0 && previous !== profile.name) {
-      process.stderr.write(
-        `[theokit-sdk] Alias "${alias}" collision: was "${previous}", now "${profile.name}".
-`
-      );
-    }
-    ALIASES.set(alias, profile.name);
-  }
-}
-function getProviderProfile(name) {
-  const canonical = ALIASES.get(name) ?? name;
-  return REGISTRY.get(canonical);
-}
-function listProviders() {
-  return Array.from(REGISTRY.values());
-}
-
-// src/internal/providers/catalog-loader.ts
-var __dirname_resolved = dirname(fileURLToPath(import.meta.url));
-function validateEntry(raw) {
-  if (typeof raw.id !== "string" || typeof raw.displayName !== "string" || typeof raw.apiMode !== "string" || typeof raw.authType !== "string" || typeof raw.baseUrl !== "string" || !Array.isArray(raw.envVars) || !Array.isArray(raw.fallbackModels) || raw.capabilities == null || typeof raw.capabilities !== "object") {
-    return null;
-  }
-  return raw;
-}
-function loadProviderCatalog(opts) {
-  const catalogPath = join(__dirname_resolved, "provider-catalog.json");
-  const rawText = readFileSync(catalogPath, "utf-8");
-  let entries = JSON.parse(rawText);
-  const result = {};
-  for (const raw of entries) {
-    const validated = validateEntry(raw);
-    if (validated === null) {
-      process.stderr.write(
-        `[theokit-sdk] WARN: Skipping malformed catalog entry: ${JSON.stringify(raw).slice(0, 100)}
-`
-      );
-      continue;
-    }
-    result[validated.id] = validated;
-  }
-  return result;
-}
-var _capabilitiesCache = null;
-function getCatalogCapabilities(providerId) {
-  if (_capabilitiesCache === null) {
-    const catalog = loadProviderCatalog();
-    _capabilitiesCache = {};
-    for (const entry of Object.values(catalog)) {
-      _capabilitiesCache[entry.id] = entry.capabilities;
-    }
-  }
-  return _capabilitiesCache[providerId];
-}
-function registerCatalogProviders(opts) {
-  const catalog = loadProviderCatalog();
-  for (const entry of Object.values(catalog)) {
-    if (getProviderProfile(entry.id) !== void 0) continue;
-    if (entry.aliases?.some((a) => getProviderProfile(a) !== void 0)) continue;
-    const profile = {
-      name: entry.id,
-      apiMode: entry.apiMode,
-      authType: entry.authType,
-      baseUrl: entry.baseUrl,
-      envVars: entry.envVars,
-      fallbackModels: entry.fallbackModels,
-      displayName: entry.displayName,
-      aliases: entry.aliases,
-      modelsUrl: entry.modelsUrl,
-      hostname: entry.hostname,
-      extraHeaders: entry.extraHeaders
-    };
-    registerProvider(profile);
-  }
-}
-
 // src/internal/providers/builtin/anthropic.ts
 var ANTHROPIC = {
   name: "anthropic",