@layer-ai/core 2.0.46 → 2.0.48

package/dist/index.d.ts

@@ -20,6 +20,7 @@ export type { EncryptedData } from '@layer-ai/sdk';
 export declare const createSessionKey: (userId: string) => Promise<string>;
 export declare const deleteSessionKeysForUser: (userId: string) => Promise<void>;
 export * from './services/task-analysis.js';
+export { initializeRegistry, getRegistry, getModel, hasModel, getModelEntries } from './lib/registry.js';
 export { PROVIDER, PROVIDERS, type Provider, callAdapter, normalizeModelId, getProviderForModel } from './lib/provider-factory.js';
 export { spendingTracker } from './lib/spending-tracker.js';
 export { spendingJobs } from './lib/spending-jobs.js';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AACnF,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGpE,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGpE,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,OAAO,IAAI,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AACxE,OAAO,EAAE,OAAO,IAAI,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,OAAO,IAAI,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAG1D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAG3C,OAAO,EAAE,EAAE,EAAE,MAAM,sBAAsB,CAAC;AAC1C,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAGrD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAGnD,eAAO,MAAM,gBAAgB,GAAU,QAAQ,MAAM,KAAG,OAAO,CAAC,MAAM,CAGrE,CAAC;AAEF,eAAO,MAAM,wBAAwB,GAAU,QAAQ,MAAM,KAAG,OAAO,CAAC,IAAI,CAG3E,CAAC;AAGF,cAAc,6BAA6B,CAAC;AAG5C,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,WAAW,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAGnI,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AACnF,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGpE,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGpE,OAAO,EAAE,OAAO,IAAI,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,OAAO,IAAI,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AACxE,OAAO,EAAE,OAAO,IAAI,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,OAAO,IAAI,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAG1D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAG3C,OAAO,EAAE,EAAE,EAAE,MAAM,sBAAsB,CAAC;AAC1C,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAGrD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC9E,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAGnD,eAAO,MAAM,gBAAgB,GAAU,QAAQ,MAAM,KAAG,OAAO,CAAC,MAAM,CAGrE,CAAC;AAEF,eAAO,MAAM,wBAAwB,GAAU,QAAQ,MAAM,KAAG,OAAO,CAAC,IAAI,CAG3E,CAAC;AAGF,cAAc,6BAA6B,CAAC;AAG5C,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGzG,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,WAAW,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAGnI,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC"}

package/dist/index.js

@@ -33,6 +33,8 @@ export const deleteSessionKeysForUser = async (userId) => {
 };
 // Services
 export * from './services/task-analysis.js';
+// Model Registry
+export { initializeRegistry, getRegistry, getModel, hasModel, getModelEntries } from './lib/registry.js';
 // Provider Factory
 export { PROVIDER, PROVIDERS, callAdapter, normalizeModelId, getProviderForModel } from './lib/provider-factory.js';
 // Spending Management

package/dist/lib/db/migrations/009_rename_document_to_ocr.sql

@@ -0,0 +1,2 @@
+-- Rename 'document' task type to 'ocr' in gates table
+UPDATE gates SET task_type = 'ocr' WHERE task_type = 'document';

package/dist/lib/provider-factory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"provider-factory.d.ts","sourceRoot":"","sources":["../../src/lib/provider-factory.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAMH,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEjF,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAG7E,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,CAAC;AAa9C;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,cAAc,CAmBhE;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,cAAc,GAAG,QAAQ,CAMnE;AAED;;;;;GAKG;AACH,wBAAsB,WAAW,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAWhG;AAED;;;;;GAKG;AACH,wBAAuB,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,aAAa,CAAC,CAgB7G"}
+{"version":3,"file":"provider-factory.d.ts","sourceRoot":"","sources":["../../src/lib/provider-factory.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAMH,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACjF,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAI7E,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,CAAC;AAa9C;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,cAAc,CAmBhE;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,cAAc,GAAG,QAAQ,CAMnE;AAED;;;;;GAKG;AACH,wBAAsB,WAAW,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAWhG;AAED;;;;;GAKG;AACH,wBAAuB,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,aAAa,CAAC,CAgB7G"}

package/dist/lib/provider-factory.js

@@ -9,8 +9,8 @@ import { OpenAIAdapter } from '../services/providers/openai-adapter.js';
 import { AnthropicAdapter } from '../services/providers/anthropic-adapter.js';
 import { GoogleAdapter } from '../services/providers/google-adapter.js';
 import { MistralAdapter } from '../services/providers/mistral-adapter.js';
-import { MODEL_REGISTRY } from '@layer-ai/sdk';
 import { PROVIDER, PROVIDERS } from './provider-constants.js';
+import { getModel, hasModel } from './registry.js';
 // Re-export for convenience
 export { PROVIDER, PROVIDERS };
 /**
@@ -28,7 +28,7 @@ const PROVIDER_ADAPTERS = {
  * Supports both full IDs (e.g., "openai/gpt-4") and short IDs (e.g., "gpt-4").
  */
 export function normalizeModelId(modelId) {
-    if (MODEL_REGISTRY[modelId]) {
+    if (hasModel(modelId)) {
         return modelId;
     }
     const providers = [
@@ -39,7 +39,7 @@ export function normalizeModelId(modelId) {
     ];
     for (const provider of providers) {
         const fullId = `${provider}/${modelId}`;
-        if (MODEL_REGISTRY[fullId]) {
+        if (hasModel(fullId)) {
             return fullId;
         }
     }
@@ -49,7 +49,7 @@ export function normalizeModelId(modelId) {
  * Gets the provider type for a given model
  */
 export function getProviderForModel(model) {
-    const modelInfo = MODEL_REGISTRY[model];
+    const modelInfo = getModel(model);
     if (!modelInfo) {
         throw new Error(`Model "${model}" not found in registry`);
     }

package/dist/lib/registry.d.ts

@@ -0,0 +1,17 @@
+/**
+ * Runtime Model Registry
+ *
+ * Starts with static SDK data, fetches fresh data from the API
+ * at startup. Falls back to SDK data if the API is unreachable or
+ * API_URL is not set (e.g., local dev).
+ */
+interface InitOptions {
+    refreshIntervalMs?: number;
+}
+export declare function initializeRegistry(options?: InitOptions): Promise<void>;
+export declare function getRegistry(): Record<string, any>;
+export declare function getModel(modelId: string): any | undefined;
+export declare function hasModel(modelId: string): boolean;
+export declare function getModelEntries(): [string, any][];
+export {};
+//# sourceMappingURL=registry.d.ts.map

package/dist/lib/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../src/lib/registry.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAOH,UAAU,WAAW;IACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,wBAAsB,kBAAkB,CAAC,OAAO,GAAE,WAAgB,iBAcjE;AAgBD,wBAAgB,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAEjD;AAED,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,GAAG,GAAG,SAAS,CAEzD;AAED,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAEjD;AAED,wBAAgB,eAAe,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAEjD"}

package/dist/lib/registry.js

@@ -0,0 +1,50 @@
+/**
+ * Runtime Model Registry
+ *
+ * Starts with static SDK data, fetches fresh data from the API
+ * at startup. Falls back to SDK data if the API is unreachable or
+ * API_URL is not set (e.g., local dev).
+ */
+import { MODEL_REGISTRY as STATIC_REGISTRY } from '@layer-ai/sdk';
+let registry = { ...STATIC_REGISTRY };
+let refreshTimer = null;
+export async function initializeRegistry(options = {}) {
+    const apiUrl = process.env.API_URL;
+    if (!apiUrl) {
+        console.log('[Registry] No API_URL set, using static SDK data');
+        return;
+    }
+    await fetchRegistry(apiUrl);
+    if (options.refreshIntervalMs) {
+        refreshTimer = setInterval(() => fetchRegistry(apiUrl), options.refreshIntervalMs);
+        // Allow the process to exit even if the timer is active
+        refreshTimer.unref();
+    }
+}
+async function fetchRegistry(apiUrl) {
+    try {
+        const res = await fetch(`${apiUrl}/v1/registry`, {
+            signal: AbortSignal.timeout(10000),
+        });
+        if (!res.ok)
+            throw new Error(`HTTP ${res.status}`);
+        const data = await res.json();
+        registry = data.models;
+        console.log(`[Registry] Loaded ${Object.keys(registry).length} models from API`);
+    }
+    catch (err) {
+        console.warn('[Registry] Fetch failed, using existing data:', err);
+    }
+}
+export function getRegistry() {
+    return registry;
+}
+export function getModel(modelId) {
+    return registry[modelId];
+}
+export function hasModel(modelId) {
+    return modelId in registry;
+}
+export function getModelEntries() {
+    return Object.entries(registry);
+}

package/dist/routes/tests/test-chat-streaming.js

@@ -291,7 +291,7 @@ async function testMultiProviderFallback() {
     };
     const modelsToTry = [
         'invalid-openai-model',
-        'claude-3-7-sonnet-20250219',
+        'claude-sonnet-4-20250514',
     ];
     let chunkCount = 0;
     let fullContent = '';

package/dist/routes/tests/test-chat-structured-output.js

@@ -138,7 +138,7 @@ async function testAnthropicBetaMode() {
     console.log('-'.repeat(80));
     const gateConfig = {
         ...baseGateConfig,
-        model: 'claude-3-7-sonnet-20250219',
+        model: 'claude-sonnet-4-20250514',
         responseFormatEnabled: true,
         responseFormatType: 'json_object',
     };

package/dist/routes/v1/gates.js

@@ -2,7 +2,7 @@ import { Router } from 'express';
 import { db } from '../../lib/db/postgres.js';
 import { cache } from '../../lib/db/redis.js';
 import { callAdapter } from '../../lib/provider-factory.js';
-import { MODEL_REGISTRY } from '@layer-ai/sdk';
+import { hasModel } from '../../lib/registry.js';
 import { detectSignificantChanges } from '../../lib/gate-utils.js';
 const router = Router();
 // POST / - Create a new gate
@@ -17,7 +17,7 @@ router.post('/', async (req, res) => {
             res.status(400).json({ error: 'bad_request', message: 'Missing required fields: name and model' });
             return;
         }
-        if (!MODEL_REGISTRY[model]) {
+        if (!hasModel(model)) {
             res.status(400).json({ error: 'bad_request', message: `Unsupported model: ${model}` });
             return;
         }
@@ -259,7 +259,7 @@ router.patch('/:id', async (req, res) => {
             res.status(404).json({ error: 'not_found', message: 'Gate not found' });
             return;
         }
-        if (model && !MODEL_REGISTRY[model]) {
+        if (model && !hasModel(model)) {
             res.status(400).json({ error: 'bad_request', message: `Unsupported model: ${model}` });
             return;
         }

package/dist/routes/v3/ocr.js

@@ -118,7 +118,7 @@ router.post('/', async (req, res) => {
             return;
         }
         // Warn if gate is configured for a different task type
-        if (gateConfig.taskType && gateConfig.taskType !== 'document') {
+        if (gateConfig.taskType && gateConfig.taskType !== 'ocr') {
             console.warn(`[Type Mismatch] Gate "${gateConfig.name}" (${gateConfig.id}) configured for taskType="${gateConfig.taskType}" ` +
                 `but received request to /v3/ocr endpoint. Processing as OCR request.`);
         }

package/dist/services/providers/base-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/base-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,WAAW,EACX,SAAS,EACT,YAAY,EACZ,UAAU,EACV,SAAS,EACT,WAAW,EACX,aAAa,EACb,aAAa,EACb,YAAY,EACZ,UAAU,EACV,cAAc,EACd,eAAe,EAGhB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAEhE,OAAO,EAAE,eAAe,EAAE,CAAC;AAE3B,8BAAsB,mBAAmB;IACvC,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IACtC,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAE1B,SAAS,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9C,SAAS,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC5D,SAAS,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;IAC/D,SAAS,CAAC,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC9D,SAAS,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACxD,SAAS,CAAC,oBAAoB,CAAC,EAAE,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAC9D,SAAS,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1D,SAAS,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACxD,SAAS,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC5D,SAAS,CAAC,qBAAqB,CAAC,EAAE,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAChE,SAAS,CAAC,qBAAqB,CAAC,EAAE,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAChE,SAAS,CAAC,sBAAsB,CAAC,EAAE,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IAElE,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAG7E,UAAU,CAAC,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,aAAa,CAAC;IAEjF,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,GAAG,MAAM;IAcrC,SAAS,CAAC,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,GAAG,SAAS;IAQjE,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS;IAQ3D,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,GAAG,SAAS;IAQpE,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,UAAU,GAAG,MAAM,GAAG,SAAS;IAQ9D,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS;IAQ3D,SAAS,CAAC,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,GAAG,SAAS;IAQjE,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS;IAQvE,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS;IAQvE,SAAS,CAAC,iBAAiB,CAAC,MAAM,EAAE,cAAc,GAAG,MAAM,GAAG,SAAS;IAQvE,SAAS,CAAC,eAAe,CAAC,oBAAoB,EAAE,MAAM,GAAG,YAAY;IAQrE,SAAS,CAAC,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS;IAYxE,SAAS,CAAC,aAAa,CACrB,KAAK,EAAE,MAAM,EACb,YAAY,EAAE,MAAM,EACpB,gBAAgB,EAAE,MAAM,GACvB,MAAM;IAWT,SAAS,CAAC,kBAAkB,CAC1B,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,MAAM,EACb,KAAK,GAAE,MAAU,GAChB,MAAM;IA2BT,SAAS,CAAC,kBAAkB,CAC1B,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,KAAK,GAAE,MAAU,GAChB,MAAM;CAiBV"}
+{"version":3,"file":"base-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/base-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,WAAW,EACX,SAAS,EACT,YAAY,EACZ,UAAU,EACV,SAAS,EACT,WAAW,EACX,aAAa,EACb,aAAa,EACb,YAAY,EACZ,UAAU,EACV,cAAc,EACd,eAAe,EAChB,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAGhE,OAAO,EAAE,eAAe,EAAE,CAAC;AAE3B,8BAAsB,mBAAmB;IACvC,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IACtC,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAE1B,SAAS,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9C,SAAS,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC5D,SAAS,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;IAC/D,SAAS,CAAC,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC9D,SAAS,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACxD,SAAS,CAAC,oBAAoB,CAAC,EAAE,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAC9D,SAAS,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1D,SAAS,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACxD,SAAS,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC5D,SAAS,CAAC,qBAAqB,CAAC,EAAE,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAChE,SAAS,CAAC,qBAAqB,CAAC,EAAE,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAChE,SAAS,CAAC,sBAAsB,CAAC,EAAE,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IAElE,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAG7E,UAAU,CAAC,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,aAAa,CAAC;IAEjF,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,GAAG,MAAM;IAcrC,SAAS,CAAC,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,GAAG,SAAS;IAQjE,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS;IAQ3D,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,GAAG,SAAS;IAQpE,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,UAAU,GAAG,MAAM,GAAG,SAAS;IAQ9D,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS;IAQ3D,SAAS,CAAC,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,GAAG,SAAS;IAQjE,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS;IAQvE,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS;IAQvE,SAAS,CAAC,iBAAiB,CAAC,MAAM,EAAE,cAAc,GAAG,MAAM,GAAG,SAAS;IAQvE,SAAS,CAAC,eAAe,CAAC,oBAAoB,EAAE,MAAM,GAAG,YAAY;IAQrE,SAAS,CAAC,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS;IAYxE,SAAS,CAAC,aAAa,CACrB,KAAK,EAAE,MAAM,EACb,YAAY,EAAE,MAAM,EACpB,gBAAgB,EAAE,MAAM,GACvB,MAAM;IAWT,SAAS,CAAC,kBAAkB,CAC1B,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,MAAM,EACb,KAAK,GAAE,MAAU,GAChB,MAAM;IA2BT,SAAS,CAAC,kBAAkB,CAC1B,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,KAAK,GAAE,MAAU,GAChB,MAAM;CA8BV"}

package/dist/services/providers/base-adapter.js

@@ -1,4 +1,5 @@
-import { ADAPTER_HANDLED, MODEL_REGISTRY, } from '@layer-ai/sdk';
+import { ADAPTER_HANDLED, } from '@layer-ai/sdk';
+import { getModel } from '../../lib/registry.js';
 export { ADAPTER_HANDLED };
 export class BaseProviderAdapter {
     mapRole(role) {
@@ -81,7 +82,7 @@ export class BaseProviderAdapter {
         return this.toolChoiceMappings[choice];
     }
     calculateCost(model, promptTokens, completionTokens) {
-        const modelInfo = MODEL_REGISTRY[model];
+        const modelInfo = getModel(model);
         if (!modelInfo || !('pricing' in modelInfo) || !modelInfo.pricing?.input) {
             return 0;
         }
@@ -91,17 +92,17 @@ export class BaseProviderAdapter {
         return inputCost + outputCost;
     }
     calculateImageCost(model, quality, size, count = 1) {
-        const modelInfo = MODEL_REGISTRY[model];
-        if (!modelInfo || !('imagePricing' in modelInfo) || !modelInfo.imagePricing) {
+        const modelInfo = getModel(model);
+        if (!modelInfo || !('unitPricing' in modelInfo) || !modelInfo.unitPricing) {
             return 0;
         }
-        const imagePricing = modelInfo.imagePricing;
-        // Flat-rate pricing (e.g. Google Imagen models)
-        if (typeof imagePricing === 'number') {
-            return imagePricing * count;
+        const unitPricing = modelInfo.unitPricing;
+        // Flat-rate pricing (e.g. Google Imagen: unitPricing = 0.02)
+        if (typeof unitPricing === 'number') {
+            return unitPricing * count;
         }
-        // Build pricing key from quality and size (e.g., 'hd-1024x1024' or 'standard-1024x1024')
-        const pricingTable = imagePricing;
+        // Object pricing (e.g. DALL-E: unitPricing = { "hd-1024x1024": 0.08, ... })
+        const pricingTable = unitPricing;
         const pricingKey = quality && size ? `${quality}-${size}` : size || 'standard-1024x1024';
         const pricePerImage = pricingTable[pricingKey];
         if (!pricePerImage) {
@@ -112,16 +113,26 @@ export class BaseProviderAdapter {
         return pricePerImage * count;
     }
     calculateVideoCost(model, duration, count = 1) {
-        const modelInfo = MODEL_REGISTRY[model];
-        if (!modelInfo || !('videoPricing' in modelInfo) || !modelInfo.videoPricing) {
+        const modelInfo = getModel(model);
+        if (!modelInfo || !('unitPricing' in modelInfo) || !modelInfo.unitPricing) {
             return 0;
         }
-        const videoPricing = modelInfo.videoPricing;
-        // Video pricing might be per-second or per-video
-        const pricePerUnit = videoPricing.perVideo || videoPricing.perSecond || 0;
-        if (videoPricing.perSecond && duration) {
-            return pricePerUnit * duration * count;
+        const unitPricing = modelInfo.unitPricing;
+        // Flat-rate pricing (e.g. unitPricing = 0.50 per video)
+        if (typeof unitPricing === 'number') {
+            return unitPricing * count;
         }
-        return pricePerUnit * count;
+        // Object pricing with per_second or per_video keys
+        const pricing = unitPricing;
+        if (pricing.per_second && duration) {
+            return pricing.per_second * duration * count;
+        }
+        if (pricing.per_video) {
+            return pricing.per_video * count;
+        }
+        if (pricing.per_minute && duration) {
+            return pricing.per_minute * (duration / 60) * count;
+        }
+        return 0;
     }
 }

package/dist/services/providers/tests/test-google-adapter.js

@@ -53,7 +53,7 @@ async function testImageGeneration() {
     console.log('Testing image generation...');
     const request = {
         gateId: 'test-gate',
-        model: 'imagen-4.0-generate-001',
+        model: 'imagen-4.0-fast-generate-001',
         type: 'image',
         data: {
             prompt: 'A cute cat playing with a ball of yarn',
@@ -64,6 +64,10 @@ async function testImageGeneration() {
     console.log('Generated images:', response.images?.length);
     console.log('Image base64 length:', response.images?.[0]?.base64?.length);
     console.log('Latency:', response.latencyMs + 'ms');
+    console.log('Cost:', response.cost);
+    if (!response.cost || response.cost <= 0) {
+        throw new Error(`Expected image generation cost > 0, got ${response.cost}`);
+    }
     console.log('✅ Image generation test passed\n');
 }
 async function testEmbeddings() {
@@ -196,6 +200,14 @@ async function testVideoGeneration() {
     console.log('Video URL:', response.videos?.[0]?.url);
     console.log('Video duration:', response.videos?.[0]?.duration);
     console.log('Latency:', response.latencyMs + 'ms');
+    console.log('Cost:', response.cost);
+    // Note: video cost may be 0 if unitPricing is not set for this model in the registry
+    if (response.cost && response.cost > 0) {
+        console.log('✅ Video cost calculation working');
+    }
+    else {
+        console.warn('⚠️  Video cost is 0 (unitPricing may not be set for this model)');
+    }
     console.log('✅ Video generation test passed\n');
 }
 async function runTests() {
@@ -203,9 +215,6 @@ async function runTests() {
         console.log('Chat completion tests...');
         await testChatCompletion();
         await testChatWithVision();
-        console.log('Embeddings...');
-        await testEmbeddings();
-        await testEmbeddingsMultiple();
         console.log('Image generation...');
         await testImageGeneration();
         console.log('Text-to-speech...');
@@ -214,6 +223,9 @@ async function runTests() {
         await testVideoGeneration();
         console.log('Tool calling...');
         await testToolCalling();
+        console.log('Embeddings...');
+        await testEmbeddings();
+        await testEmbeddingsMultiple();
         console.log('✅ All tests passed!');
     }
     catch (error) {

package/dist/services/providers/tests/test-json-response-format-openai.js

@@ -14,7 +14,7 @@ async function testJsonObjectMode() {
     console.log('-'.repeat(80));
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [
@@ -118,7 +118,7 @@ async function testTextMode() {
     console.log('-'.repeat(80));
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [

package/dist/services/providers/tests/test-mistral-adapter.js

@@ -8,7 +8,7 @@ async function testChatCompletion() {
     const request = {
         gateId: 'test-gate',
         type: 'chat',
-        model: 'mistral-small-latest',
+        model: 'mistral-small-2501',
         data: {
             messages: [
                 {
@@ -33,7 +33,7 @@ async function testChatWithSystemPrompt() {
     const request = {
         gateId: 'test-gate',
         type: 'chat',
-        model: 'mistral-small-latest',
+        model: 'mistral-small-2501',
         data: {
             systemPrompt: 'You are a helpful assistant that responds in JSON format only.',
             messages: [
@@ -56,7 +56,7 @@ async function testChatWithTools() {
     const request = {
         gateId: 'test-gate',
         type: 'chat',
-        model: 'mistral-small-latest',
+        model: 'mistral-small-2501',
         data: {
             messages: [
                 {
@@ -104,7 +104,7 @@ async function testToolResponse() {
     const initialRequest = {
         gateId: 'test-gate',
         type: 'chat',
-        model: 'mistral-small-latest',
+        model: 'mistral-small-2501',
         data: {
             messages: [
                 {
@@ -145,7 +145,7 @@ async function testToolResponse() {
         const followUpRequest = {
             gateId: 'test-gate',
             type: 'chat',
-            model: 'mistral-small-latest',
+            model: 'mistral-small-2501',
             data: {
                 messages: [
                     {
@@ -226,7 +226,7 @@ async function testResponseFormat() {
     const request = {
         gateId: 'test-gate',
         type: 'chat',
-        model: 'mistral-small-latest',
+        model: 'mistral-small-2501',
         data: {
             messages: [
                 {
@@ -247,7 +247,7 @@ async function testMultiTurn() {
     const request = {
         gateId: 'test-gate',
         type: 'chat',
-        model: 'mistral-small-latest',
+        model: 'mistral-small-2501',
         data: {
             messages: [
                 {
@@ -297,7 +297,7 @@ async function testUnsupportedModality() {
     const request = {
         gateId: 'test-gate',
         type: 'image',
-        model: 'mistral-large-latest',
+        model: 'mistral-large-2512',
         data: {
             prompt: 'A sunset over the ocean',
         },

package/dist/services/providers/tests/test-openai-adapter.js

@@ -4,7 +4,7 @@ async function testChatCompletion() {
     console.log('Testing chat completion...');
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [
@@ -26,7 +26,7 @@ async function testChatWithVision() {
     console.log('Testing chat with vision...');
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [
@@ -64,6 +64,10 @@ async function testImageGeneration() {
     console.log('Generated images:', response.images?.length);
     console.log('Image URL:', response.images?.[0]?.url);
     console.log('Revised prompt:', response.images?.[0]?.revisedPrompt);
+    console.log('Cost:', response.cost);
+    if (!response.cost || response.cost <= 0) {
+        throw new Error(`Expected image generation cost > 0, got ${response.cost}`);
+    }
     console.log('✅ Image generation test passed\n');
 }
 async function testEmbeddings() {
@@ -105,7 +109,7 @@ async function testToolCalling() {
     // Step 1: Initial request with tool available
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [
@@ -146,7 +150,7 @@ async function testToolCalling() {
     // Step 2: Send tool response back
     const toolResponseRequest = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [
@@ -174,7 +178,7 @@ async function testContentAndToolCalls() {
     // This tests the fix we made - assistant messages can have BOTH content and toolCalls
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [

package/dist/services/providers/tests/test-openai-streaming.js

@@ -4,7 +4,7 @@ async function testChatStreamingBasic() {
     console.log('Testing basic chat streaming...');
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [
@@ -48,7 +48,7 @@ async function testChatStreamingWithToolCalls() {
     console.log('Testing chat streaming with tool calls...');
     const request = {
         gateId: 'test-gate',
-        model: 'gpt-4o-mini',
+        model: 'gpt-4o',
         type: 'chat',
         data: {
             messages: [

package/dist/services/task-analysis.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"task-analysis.d.ts","sourceRoot":"","sources":["../../src/services/task-analysis.ts"],"names":[],"mappings":"AACA,OAAO,EAAkB,YAAY,EAAmC,MAAM,eAAe,CAAC;AA2D9F,wBAAsB,WAAW,CAC/B,WAAW,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,GACA,OAAO,CAAC,YAAY,CAAC,CA6HvB"}
+{"version":3,"file":"task-analysis.d.ts","sourceRoot":"","sources":["../../src/services/task-analysis.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAmC,MAAM,eAAe,CAAC;AA6D9E,wBAAsB,WAAW,CAC/B,WAAW,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,GACA,OAAO,CAAC,YAAY,CAAC,CA6HvB"}

package/dist/services/task-analysis.js

@@ -1,5 +1,5 @@
 import Anthropic from '@anthropic-ai/sdk';
-import { MODEL_REGISTRY } from '@layer-ai/sdk';
+import { getModelEntries } from '../lib/registry.js';
 async function detectTaskType(description, anthropic) {
     const prompt = `Analyze this task description and determine what TYPE of AI task it is.
 
@@ -14,7 +14,8 @@ AVAILABLE TASK TYPES:
 - tts: Text-to-speech, voice synthesis
 - stt: Speech-to-text, audio transcription
 - embeddings: Text embeddings, semantic search
-- document: Document processing, OCR
+- ocr: Document processing, OCR
+- moderation: Content moderation
 - responses: Complex reasoning tasks (o3-pro style models)
 - language-completion: Legacy text completion
 
@@ -39,7 +40,7 @@ Return ONLY the task type as a single word, nothing else.`;
             throw new Error('Unexpected response type from Claude');
         }
         const detectedType = responseContent.text.trim().toLowerCase();
-        const validTypes = ['chat', 'image', 'video', 'audio', 'tts', 'stt', 'embeddings', 'document', 'responses', 'language-completion'];
+        const validTypes = ['chat', 'image', 'video', 'audio', 'tts', 'stt', 'embeddings', 'ocr', 'moderation', 'responses', 'language-completion'];
         if (validTypes.includes(detectedType)) {
             return detectedType;
         }
@@ -66,7 +67,7 @@ export async function analyzeTask(description, userPreferences) {
         console.error('Failed to detect task type, defaulting to chat:', error);
     }
     const filteredRegistry = {};
-    for (const [key, model] of Object.entries(MODEL_REGISTRY)) {
+    for (const [key, model] of getModelEntries()) {
         // Filter by task type
         if (model.type !== taskType) {
             continue;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@layer-ai/core",
-  "version": "2.0.46",
+  "version": "2.0.48",
   "description": "Core API routes and services for Layer AI",
   "type": "module",
   "main": "./dist/index.js",
@@ -36,7 +36,7 @@
     "nanoid": "^5.0.4",
     "openai": "^4.24.0",
     "pg": "^8.11.3",
-    "@layer-ai/sdk": "^2.5.11"
+    "@layer-ai/sdk": "^2.5.12"
   },
   "devDependencies": {
     "@types/bcryptjs": "^2.4.6",