@theokit/sdk 4.12.2 → 4.13.0

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,158 @@ 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
+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));
+var modelInfoIndex = /* @__PURE__ */ new Map();
+function getCatalogModelInfo(key2) {
+  ensureModelIndexLoaded();
+  return modelInfoIndex.get(key2);
+}
+var _modelIndexLoaded = false;
+function ensureModelIndexLoaded() {
+  if (_modelIndexLoaded) return;
+  _modelIndexLoaded = true;
+  const catalog = loadProviderCatalog();
+  for (const entry of Object.values(catalog)) {
+    indexEntryModels(entry);
+  }
+}
+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 +13327,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 +13417,27 @@ 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 keys = [`${provider}/${cleanedModel}`, cleanedModel];
+  for (const key2 of keys) {
+    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 } : {},
+      pricingVersion: "catalog-vendored"
+    };
+  }
   return void 0;
 }
 
@@ -13781,93 +13954,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",