@layer-ai/core 0.9.1 → 1.0.0

package/dist/lib/key-resolver.d.ts

@@ -5,12 +5,16 @@
  * Priority: User's BYOK key → Platform key → Error
  */
 import type { Provider } from './provider-constants.js';
+export interface ResolvedKey {
+    key: string;
+    usedPlatformKey: boolean;
+}
 /**
  * Resolves the API key to use for a provider
  * @param provider - The provider name
  * @param userId - Optional user ID for BYOK lookup
  * @param platformKey - The platform's API key (fallback)
- * @returns The API key to use
+ * @returns The API key to use and whether platform key was used
  */
-export declare function resolveApiKey(provider: Provider, userId: string | undefined, platformKey: string | undefined): Promise<string>;
+export declare function resolveApiKey(provider: Provider, userId: string | undefined, platformKey: string | undefined): Promise<ResolvedKey>;
 //# sourceMappingURL=key-resolver.d.ts.map

package/dist/lib/key-resolver.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"key-resolver.d.ts","sourceRoot":"","sources":["../../src/lib/key-resolver.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAGxD;;;;;;GAMG;AACH,wBAAsB,aAAa,CACjC,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,GAAG,SAAS,EAC1B,WAAW,EAAE,MAAM,GAAG,SAAS,GAC9B,OAAO,CAAC,MAAM,CAAC,CAoBjB"}
+{"version":3,"file":"key-resolver.d.ts","sourceRoot":"","sources":["../../src/lib/key-resolver.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAGxD,MAAM,WAAW,WAAW;IAC1B,GAAG,EAAE,MAAM,CAAC;IACZ,eAAe,EAAE,OAAO,CAAC;CAC1B;AAED;;;;;;GAMG;AACH,wBAAsB,aAAa,CACjC,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,GAAG,SAAS,EAC1B,WAAW,EAAE,MAAM,GAAG,SAAS,GAC9B,OAAO,CAAC,WAAW,CAAC,CAoBtB"}

package/dist/lib/key-resolver.js

@@ -10,7 +10,7 @@ import { decrypt } from './encryption.js';
  * @param provider - The provider name
  * @param userId - Optional user ID for BYOK lookup
  * @param platformKey - The platform's API key (fallback)
- * @returns The API key to use
+ * @returns The API key to use and whether platform key was used
  */
 export async function resolveApiKey(provider, userId, platformKey) {
     // If userId is provided, check for BYOK key
@@ -18,7 +18,7 @@ export async function resolveApiKey(provider, userId, platformKey) {
         try {
             const byokKey = await getUserProviderKey(userId, provider);
             if (byokKey) {
-                return byokKey;
+                return { key: byokKey, usedPlatformKey: false };
             }
         }
         catch (error) {
@@ -28,7 +28,7 @@ export async function resolveApiKey(provider, userId, platformKey) {
     }
     // Fallback to platform key
     if (platformKey) {
-        return platformKey;
+        return { key: platformKey, usedPlatformKey: true };
     }
     throw new Error(`No API key available for provider: ${provider}`);
 }

package/dist/routes/v1/gates.js

@@ -327,7 +327,7 @@ router.post('/test', async (req, res) => {
         try {
             const request = {
                 type: 'chat',
-                gate: finalGate.name || 'test-gate',
+                gateId: finalGate.id,
                 model: finalGate.model,
                 data: {
                     messages,
@@ -363,7 +363,7 @@ router.post('/test', async (req, res) => {
                 try {
                     const request = {
                         type: 'chat',
-                        gate: finalGate.name || 'test-gate',
+                        gateId: finalGate.id,
                         model: fallbackModel,
                         data: {
                             messages,

package/dist/routes/v2/complete.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"complete.d.ts","sourceRoot":"","sources":["../../../src/routes/v2/complete.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,SAAS,CAAC;AASpD,QAAA,MAAM,MAAM,EAAE,UAAqB,CAAC;AAqQpC,eAAe,MAAM,CAAC"}
+{"version":3,"file":"complete.d.ts","sourceRoot":"","sources":["../../../src/routes/v2/complete.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,SAAS,CAAC;AASpD,QAAA,MAAM,MAAM,EAAE,UAAqB,CAAC;AA+OpC,eAAe,MAAM,CAAC"}

package/dist/routes/v2/complete.js

@@ -1,6 +1,5 @@
 import { Router } from 'express';
 import { db } from '../../lib/db/postgres.js';
-import { cache } from '../../lib/db/redis.js';
 import { authenticate } from '../../middleware/auth.js';
 import { callAdapter, normalizeModelId } from '../../lib/provider-factory.js';
 import { OverrideField } from '@layer-ai/sdk';
@@ -13,29 +12,6 @@ function isOverrideAllowed(allowOverrides, field) {
         return false;
     return allowOverrides[field] ?? false;
 }
-async function getGateConfig(userId, gateIdentifier) {
-    const isUUID = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(gateIdentifier);
-    let gateConfig = null;
-    if (isUUID) {
-        gateConfig = await cache.getGateById(userId, gateIdentifier);
-        if (!gateConfig) {
-            gateConfig = await db.getGateByUserAndId(userId, gateIdentifier);
-            if (gateConfig) {
-                await cache.setGate(userId, gateConfig.name, gateConfig);
-            }
-        }
-    }
-    else {
-        gateConfig = await cache.getGate(userId, gateIdentifier);
-        if (!gateConfig) {
-            gateConfig = await db.getGateByUserAndName(userId, gateIdentifier);
-            if (gateConfig) {
-                await cache.setGate(userId, gateIdentifier, gateConfig);
-            }
-        }
-    }
-    return gateConfig;
-}
 function resolveFinalRequest(gateConfig, request) {
     const finalRequest = { ...request };
     let finalModel = gateConfig.model;
@@ -141,18 +117,23 @@ router.post('/', authenticate, async (req, res) => {
     let request = null;
     try {
         const rawRequest = req.body;
-        if (!rawRequest.gate) {
-            res.status(400).json({ error: 'bad_request', message: 'Missing required field: gate' });
+        if (!rawRequest.gateId) {
+            res.status(400).json({ error: 'bad_request', message: 'Missing required field: gateId' });
+            return;
+        }
+        const isUUID = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(rawRequest.gateId);
+        if (!isUUID) {
+            res.status(400).json({ error: 'bad_request', message: 'gateId must be a valid UUID. Gate names are no longer supported.' });
             return;
         }
-        gateConfig = await getGateConfig(userId, rawRequest.gate);
+        gateConfig = await db.getGateByUserAndId(userId, rawRequest.gateId);
         if (!gateConfig) {
-            res.status(404).json({ error: 'not_found', message: `Gate "${rawRequest.gate}" not found` });
+            res.status(404).json({ error: 'not_found', message: `Gate with ID "${rawRequest.gateId}" not found` });
             return;
         }
         const requestType = rawRequest.type || gateConfig.taskType || 'chat';
         request = {
-            gate: rawRequest.gate,
+            gateId: rawRequest.gateId,
             type: requestType,
             data: rawRequest.data,
             model: rawRequest.model,
@@ -171,7 +152,7 @@ router.post('/', authenticate, async (req, res) => {
         db.logRequest({
             userId,
             gateId: gateConfig.id,
-            gateName: request.gate,
+            gateName: gateConfig.name,
             modelRequested: request.model || gateConfig.model,
             modelUsed: modelUsed,
             promptTokens: result.usage?.promptTokens || 0,

package/dist/routes/v2/tests/test-byok-completion.js

@@ -12,7 +12,7 @@ import { GoogleAdapter } from '../../../services/providers/google-adapter.js';
 // Test user ID from the database
 const TEST_USER_ID = 'ebd64998-465d-4211-ad67-87b4e01ad0da';
 const SIMPLE_REQUEST = {
-    gate: 'byok-test',
+    gateId: 'byok-test',
     model: 'gpt-4o-mini',
     type: 'chat',
     data: {

package/dist/routes/v2/tests/test-complete-anthropic.js

@@ -3,7 +3,7 @@
 async function testBasicChat() {
     console.log('Test 1: Basic chat completion with Anthropic\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -27,7 +27,7 @@ async function testBasicChat() {
 async function testVision() {
     console.log('\n\nTest 2: Vision with Anthropic (not supported in v1)\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -49,7 +49,7 @@ async function testVision() {
 async function testToolCalls() {
     console.log('\n\nTest 3: Tool calls with Anthropic (not supported in v1)\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -86,7 +86,7 @@ async function testToolCalls() {
 async function testSystemPrompt() {
     console.log('\n\nTest 4: System prompt and advanced params\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {

package/dist/routes/v2/tests/test-complete-openai.js

@@ -3,7 +3,7 @@
 async function testBasicChat() {
     console.log('Test 1: Basic chat completion with OpenAI\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -26,7 +26,7 @@ async function testBasicChat() {
 async function testVision() {
     console.log('\n\nTest 2: Vision with GPT-4o\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o',
         type: 'chat',
         data: {
@@ -49,7 +49,7 @@ async function testVision() {
 async function testToolCalls() {
     console.log('\n\nTest 3: Tool calls with GPT-4\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -85,7 +85,7 @@ async function testToolCalls() {
 async function testImageGeneration() {
     console.log('\n\nTest 4: Image generation with DALL-E\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'dall-e-3',
         type: 'image',
         data: {
@@ -102,7 +102,7 @@ async function testImageGeneration() {
 async function testEmbeddings() {
     console.log('\n\nTest 5: Text embeddings\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'text-embedding-3-small',
         type: 'embeddings',
         data: {
@@ -117,7 +117,7 @@ async function testEmbeddings() {
 async function testTextToSpeech() {
     console.log('\n\nTest 6: Text-to-speech\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'tts-1',
         type: 'tts',
         data: {
@@ -133,7 +133,7 @@ async function testTextToSpeech() {
 async function testResponseFormat() {
     console.log('\n\nTest 7: JSON response format\n');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {

package/dist/routes/v2/tests/test-complete-routing.js

@@ -3,7 +3,7 @@
 async function testFallbackRouting() {
     console.log('Test 1: Fallback routing (Anthropic -> OpenAI)\n');
     const request = {
-        gate: 'test-gate-with-fallback',
+        gateId: 'test-gate-with-fallback',
         model: 'claude-sonnet-4-5-20250929', // Primary model
         type: 'chat',
         data: {
@@ -27,7 +27,7 @@ async function testFallbackRouting() {
 async function testRoundRobinRouting() {
     console.log('\n\nTest 2: Round-robin routing across providers\n');
     const request = {
-        gate: 'test-gate-with-round-robin',
+        gateId: 'test-gate-with-round-robin',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -50,7 +50,7 @@ async function testRoundRobinRouting() {
 async function testCrossProviderFallback() {
     console.log('\n\nTest 3: Cross-provider fallback with vision\n');
     const request = {
-        gate: 'test-gate-vision-fallback',
+        gateId: 'test-gate-vision-fallback',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -78,7 +78,7 @@ async function testCrossProviderFallback() {
 async function testToolCallsFallback() {
     console.log('\n\nTest 4: Fallback routing with tool calls\n');
     const request = {
-        gate: 'test-gate-tools-fallback',
+        gateId: 'test-gate-tools-fallback',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -117,7 +117,7 @@ async function testToolCallsFallback() {
 async function testCostOptimization() {
     console.log('\n\nTest 5: Cost optimization with round-robin\n');
     const request = {
-        gate: 'test-gate-cost-optimization',
+        gateId: 'test-gate-cost-optimization',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {

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

@@ -1 +1 @@
-{"version":3,"file":"anthropic-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/anthropic-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAmB,MAAM,mBAAmB,CAAC;AACzE,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EACZ,UAAU,EACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAoB1E,qBAAa,gBAAiB,SAAQ,mBAAmB;IACvD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAsB;IAElD,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAI3D;IAEF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAM1D;IAEF,SAAS,CAAC,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS;IAalE,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAoB5D,UAAU;CA4JzB"}
+{"version":3,"file":"anthropic-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/anthropic-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAmB,MAAM,mBAAmB,CAAC;AACzE,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EACZ,UAAU,EACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAoB1E,qBAAa,gBAAiB,SAAQ,mBAAmB;IACvD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAsB;IAElD,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,CAI3D;IAEF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAM1D;IAEF,SAAS,CAAC,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS;IAalE,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAoB5D,UAAU;CA6JzB"}

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

@@ -55,10 +55,10 @@ export class AnthropicAdapter extends BaseProviderAdapter {
     }
     async call(request, userId) {
         // Resolve API key (BYOK → Platform key)
-        const apiKey = await resolveApiKey(this.provider, userId, process.env.ANTHROPIC_API_KEY);
+        const resolved = await resolveApiKey(this.provider, userId, process.env.ANTHROPIC_API_KEY);
         switch (request.type) {
             case 'chat':
-                return this.handleChat(request, apiKey);
+                return this.handleChat(request, resolved.key, resolved.usedPlatformKey);
             case 'image':
                 throw new Error('Image generation not yet supported by Anthropic');
             case 'embeddings':
@@ -71,7 +71,7 @@ export class AnthropicAdapter extends BaseProviderAdapter {
                 throw new Error(`Unknown modality: ${request.type}`);
         }
     }
-    async handleChat(request, apiKey) {
+    async handleChat(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getAnthropicClient(apiKey);
         const { data: chat, model } = request;
@@ -211,6 +211,7 @@ export class AnthropicAdapter extends BaseProviderAdapter {
             },
             cost,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }

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

@@ -1 +1 @@
-{"version":3,"file":"google-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/google-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAKL,yBAAyB,EAI1B,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EACZ,SAAS,EACV,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAExD,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAkB1E,qBAAa,aAAc,SAAQ,mBAAmB;IACpD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAmB;IAE/C,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAGF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAO1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAIrE;IAGF,SAAS,CAAC,eAAe,EAAE,MAAM,CAC/B,SAAS,EACT;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAC5C,CAKC;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAoB5D,UAAU;YAsLV,qBAAqB;YAmCrB,gBAAgB;YAoChB,qBAAqB;YAoHrB,kBAAkB;IAyChC,OAAO,CAAC,KAAK;CAGd"}
+{"version":3,"file":"google-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/google-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAKL,yBAAyB,EAI1B,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EACZ,SAAS,EACV,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAExD,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAkB1E,qBAAa,aAAc,SAAQ,mBAAmB;IACpD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAmB;IAE/C,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAGF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAO1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAIrE;IAGF,SAAS,CAAC,eAAe,EAAE,MAAM,CAC/B,SAAS,EACT;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAC5C,CAKC;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAoB5D,UAAU;YAwLV,qBAAqB;YAqCrB,gBAAgB;YAsChB,qBAAqB;YAsHrB,kBAAkB;IA2ChC,OAAO,CAAC,KAAK;CAGd"}

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

@@ -52,23 +52,23 @@ export class GoogleAdapter extends BaseProviderAdapter {
     }
     async call(request, userId) {
         // Resolve API key (BYOK → Platform key)
-        const apiKey = await resolveApiKey(this.provider, userId, process.env.GOOGLE_API_KEY);
+        const resolved = await resolveApiKey(this.provider, userId, process.env.GOOGLE_API_KEY);
         switch (request.type) {
             case 'chat':
-                return this.handleChat(request, apiKey);
+                return this.handleChat(request, resolved.key, resolved.usedPlatformKey);
             case 'image':
-                return this.handleImageGeneration(request, apiKey);
+                return this.handleImageGeneration(request, resolved.key, resolved.usedPlatformKey);
             case 'embeddings':
-                return this.handleEmbeddings(request, apiKey);
+                return this.handleEmbeddings(request, resolved.key, resolved.usedPlatformKey);
             case 'tts':
-                return this.handleTextToSpeech(request, apiKey);
+                return this.handleTextToSpeech(request, resolved.key, resolved.usedPlatformKey);
             case 'video':
-                return this.handleVideoGeneration(request, apiKey);
+                return this.handleVideoGeneration(request, resolved.key, resolved.usedPlatformKey);
             default:
                 throw new Error(`Unknown modality: ${request.type}`);
         }
     }
-    async handleChat(request, apiKey) {
+    async handleChat(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getGoogleClient(apiKey);
         const { data: chat, model } = request;
@@ -220,10 +220,11 @@ export class GoogleAdapter extends BaseProviderAdapter {
             },
             cost,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleImageGeneration(request, apiKey) {
+    async handleImageGeneration(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getGoogleClient(apiKey);
         const { data: image, model } = request;
@@ -246,10 +247,11 @@ export class GoogleAdapter extends BaseProviderAdapter {
             images,
             model: model,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleEmbeddings(request, apiKey) {
+    async handleEmbeddings(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getGoogleClient(apiKey);
         const { data: embedding, model } = request;
@@ -273,10 +275,11 @@ export class GoogleAdapter extends BaseProviderAdapter {
                 totalTokens: 0,
             },
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleVideoGeneration(request, apiKey) {
+    async handleVideoGeneration(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getGoogleClient(apiKey);
         const { data: video, model } = request;
@@ -376,10 +379,11 @@ export class GoogleAdapter extends BaseProviderAdapter {
             videos: videos.filter((v) => v.url),
             model: model,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: operation.response,
         };
     }
-    async handleTextToSpeech(request, apiKey) {
+    async handleTextToSpeech(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getGoogleClient(apiKey);
         const { data: tts, model } = request;
@@ -408,6 +412,7 @@ export class GoogleAdapter extends BaseProviderAdapter {
             },
             model: model,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }

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

@@ -1 +1 @@
-{"version":3,"file":"mistral-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/mistral-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EAEb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAoB1E,qBAAa,cAAe,SAAQ,mBAAmB;IACrD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAoB;IAEhD,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAGF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAM1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAIlD;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAsB5D,UAAU;YA+LV,gBAAgB;YAwChB,SAAS;CA4ExB"}
+{"version":3,"file":"mistral-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/mistral-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EAEb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAoB1E,qBAAa,cAAe,SAAQ,mBAAmB;IACrD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAoB;IAEhD,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAGF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAM1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAIlD;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAsB5D,UAAU;YAiMV,gBAAgB;YA0ChB,SAAS;CA8ExB"}

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

@@ -46,14 +46,14 @@ export class MistralAdapter extends BaseProviderAdapter {
     }
     async call(request, userId) {
         // Resolve API key (BYOK → Platform key)
-        const apiKey = await resolveApiKey(this.provider, userId, process.env.MISTRAL_API_KEY);
+        const resolved = await resolveApiKey(this.provider, userId, process.env.MISTRAL_API_KEY);
         switch (request.type) {
             case 'chat':
-                return this.handleChat(request, apiKey);
+                return this.handleChat(request, resolved.key, resolved.usedPlatformKey);
             case 'embeddings':
-                return this.handleEmbeddings(request, apiKey);
+                return this.handleEmbeddings(request, resolved.key, resolved.usedPlatformKey);
             case 'ocr':
-                return this.handleOCR(request, apiKey);
+                return this.handleOCR(request, resolved.key, resolved.usedPlatformKey);
             case 'image':
                 throw new Error('Image generation not supported by Mistral');
             case 'tts':
@@ -64,7 +64,7 @@ export class MistralAdapter extends BaseProviderAdapter {
                 throw new Error(`Unknown modality: ${request.type}`);
         }
     }
-    async handleChat(request, apiKey) {
+    async handleChat(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const mistral = getMistralClient(apiKey);
         const { data: chat, model } = request;
@@ -214,10 +214,11 @@ export class MistralAdapter extends BaseProviderAdapter {
             },
             cost,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleEmbeddings(request, apiKey) {
+    async handleEmbeddings(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const mistral = getMistralClient(apiKey);
         const { data: embedding, model } = request;
@@ -244,10 +245,11 @@ export class MistralAdapter extends BaseProviderAdapter {
             },
             cost,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleOCR(request, apiKey) {
+    async handleOCR(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const mistral = getMistralClient(apiKey);
         const { data: ocr, model } = request;
@@ -313,6 +315,7 @@ export class MistralAdapter extends BaseProviderAdapter {
             },
             model: response.model || ocrModel,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }

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

@@ -1 +1 @@
-{"version":3,"file":"openai-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/openai-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,WAAW,EACX,SAAS,EACT,YAAY,EACZ,UAAU,EACV,SAAS,EACT,WAAW,EACX,YAAY,EACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAoB1E,qBAAa,aAAc,SAAQ,mBAAmB;IACpD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAmB;IAE/C,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAEF,SAAS,CAAC,mBAAmB,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAIxD;IAEF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAK1D;IAEF,SAAS,CAAC,iBAAiB,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAQpD;IAEF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAG1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAGtD;IAEF,SAAS,CAAC,iBAAiB,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAKpD;IAEF,SAAS,CAAC,mBAAmB,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAOxD;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAoB5D,UAAU;YA2GV,qBAAqB;YA6BrB,gBAAgB;YAiChB,kBAAkB;CA8BjC"}
+{"version":3,"file":"openai-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/openai-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,WAAW,EACX,SAAS,EACT,YAAY,EACZ,UAAU,EACV,SAAS,EACT,WAAW,EACX,YAAY,EACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAoB1E,qBAAa,aAAc,SAAQ,mBAAmB;IACpD,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAmB;IAE/C,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAEF,SAAS,CAAC,mBAAmB,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAIxD;IAEF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAK1D;IAEF,SAAS,CAAC,iBAAiB,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAQpD;IAEF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAG1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAGtD;IAEF,SAAS,CAAC,iBAAiB,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAKpD;IAEF,SAAS,CAAC,mBAAmB,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAOxD;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YAoB5D,UAAU;YA4GV,qBAAqB;YA8BrB,gBAAgB;YAkChB,kBAAkB;CA+BjC"}

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

@@ -74,23 +74,23 @@ export class OpenAIAdapter extends BaseProviderAdapter {
     }
     async call(request, userId) {
         // Resolve API key (BYOK → Platform key)
-        const apiKey = await resolveApiKey(this.provider, userId, process.env.OPENAI_API_KEY);
+        const resolved = await resolveApiKey(this.provider, userId, process.env.OPENAI_API_KEY);
         switch (request.type) {
             case 'chat':
-                return this.handleChat(request, apiKey);
+                return this.handleChat(request, resolved.key, resolved.usedPlatformKey);
             case 'image':
-                return this.handleImageGeneration(request, apiKey);
+                return this.handleImageGeneration(request, resolved.key, resolved.usedPlatformKey);
             case 'embeddings':
-                return this.handleEmbeddings(request, apiKey);
+                return this.handleEmbeddings(request, resolved.key, resolved.usedPlatformKey);
             case 'tts':
-                return this.handleTextToSpeech(request, apiKey);
+                return this.handleTextToSpeech(request, resolved.key, resolved.usedPlatformKey);
             case 'video':
                 throw new Error('Video generation not yet supported by OpenAI');
             default:
                 throw new Error(`Unknown modality: ${request.type}`);
         }
     }
-    async handleChat(request, apiKey) {
+    async handleChat(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getOpenAIClient(apiKey);
         const { data: chat, model } = request;
@@ -181,10 +181,11 @@ export class OpenAIAdapter extends BaseProviderAdapter {
             },
             cost,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleImageGeneration(request, apiKey) {
+    async handleImageGeneration(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getOpenAIClient(apiKey);
         const { data: image, model } = request;
@@ -206,10 +207,11 @@ export class OpenAIAdapter extends BaseProviderAdapter {
             })),
             model: model,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleEmbeddings(request, apiKey) {
+    async handleEmbeddings(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getOpenAIClient(apiKey);
         const { data: embedding, model } = request;
@@ -234,10 +236,11 @@ export class OpenAIAdapter extends BaseProviderAdapter {
             },
             cost,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
             raw: response,
         };
     }
-    async handleTextToSpeech(request, apiKey) {
+    async handleTextToSpeech(request, apiKey, usedPlatformKey) {
         const startTime = Date.now();
         const client = getOpenAIClient(apiKey);
         const { data: tts, model } = request;
@@ -260,6 +263,7 @@ export class OpenAIAdapter extends BaseProviderAdapter {
             },
             model: model,
             latencyMs: Date.now() - startTime,
+            usedPlatformKey,
         };
     }
 }

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

@@ -3,7 +3,7 @@ const adapter = new AnthropicAdapter();
 async function testChatCompletion() {
     console.log('Testing chat completion...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -25,7 +25,7 @@ async function testChatCompletion() {
 async function testChatWithVision() {
     console.log('Testing chat with vision...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -50,7 +50,7 @@ async function testToolCalls() {
     console.log('Testing tool calls...');
     // Step 1: Initial request with tool available
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {
@@ -91,7 +91,7 @@ async function testToolCalls() {
     console.log('Function arguments:', toolCall.function.arguments);
     // Step 2: Send tool response back
     const toolResponseRequest = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'claude-sonnet-4-5-20250929',
         type: 'chat',
         data: {

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

@@ -4,7 +4,7 @@ const adapter = new GoogleAdapter();
 async function testChatCompletion() {
     console.log('Testing chat completion...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gemini-2.5-flash',
         type: 'chat',
         data: {
@@ -26,7 +26,7 @@ async function testChatCompletion() {
 async function testChatWithVision() {
     console.log('Testing chat with vision...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gemini-2.5-flash',
         type: 'chat',
         data: {
@@ -52,7 +52,7 @@ async function testChatWithVision() {
 async function testImageGeneration() {
     console.log('Testing image generation...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'imagen-4.0-generate-001',
         type: 'image',
         data: {
@@ -69,7 +69,7 @@ async function testImageGeneration() {
 async function testEmbeddings() {
     console.log('Testing embeddings...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'text-embedding-004',
         type: 'embeddings',
         data: {
@@ -85,7 +85,7 @@ async function testToolCalling() {
     console.log('Testing tool calling...');
     // Step 1: Send message with tools available
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gemini-2.5-flash',
         type: 'chat',
         data: {
@@ -122,7 +122,7 @@ async function testToolCalling() {
     console.log('Function arguments:', toolCall.function.arguments);
     // Step 2: Send tool response back
     const toolResponseRequest = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gemini-2.5-flash',
         type: 'chat',
         data: {
@@ -149,7 +149,7 @@ async function testToolCalling() {
 async function testEmbeddingsMultiple() {
     console.log('Testing multiple embeddings...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'text-embedding-004',
         type: 'embeddings',
         data: {
@@ -165,7 +165,7 @@ async function testEmbeddingsMultiple() {
 async function testTextToSpeech() {
     console.log('Testing text-to-speech...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gemini-2.5-flash-preview-tts',
         type: 'tts',
         data: {
@@ -182,7 +182,7 @@ async function testTextToSpeech() {
 async function testVideoGeneration() {
     console.log('Testing video generation (this may take a few minutes)...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'veo-2.0-generate-001',
         type: 'video',
         data: {

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

@@ -6,7 +6,7 @@ const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
 async function testChatCompletion() {
     console.log('--- Testing Chat Completion ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'mistral-small-latest',
         data: {
@@ -31,7 +31,7 @@ async function testChatCompletion() {
 async function testChatWithSystemPrompt() {
     console.log('--- Testing Chat with System Prompt ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'mistral-small-latest',
         data: {
@@ -54,7 +54,7 @@ async function testChatWithSystemPrompt() {
 async function testChatWithTools() {
     console.log('--- Testing Chat with Tools ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'mistral-small-latest',
         data: {
@@ -102,7 +102,7 @@ async function testToolResponse() {
     console.log('--- Testing Tool Response Flow ---');
     // First, get the tool call
     const initialRequest = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'mistral-small-latest',
         data: {
@@ -143,7 +143,7 @@ async function testToolResponse() {
         const toolCall = initialResponse.toolCalls[0];
         await delay(2000);
         const followUpRequest = {
-            gate: 'test-gate',
+            gateId: 'test-gate',
             type: 'chat',
             model: 'mistral-small-latest',
             data: {
@@ -175,7 +175,7 @@ async function testToolResponse() {
 async function testEmbeddings() {
     console.log('--- Testing Embeddings ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'embeddings',
         model: 'mistral-embed',
         data: {
@@ -193,7 +193,7 @@ async function testVisionCapability() {
     console.log('--- Testing Vision Capability (Pixtral) ---');
     // Use a public image URL for testing
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'pixtral-large-2411',
         data: {
@@ -224,7 +224,7 @@ async function testVisionCapability() {
 async function testResponseFormat() {
     console.log('--- Testing JSON Response Format ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'mistral-small-latest',
         data: {
@@ -245,7 +245,7 @@ async function testResponseFormat() {
 async function testMultiTurn() {
     console.log('--- Testing Multi-turn Conversation ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'chat',
         model: 'mistral-small-latest',
         data: {
@@ -273,7 +273,7 @@ async function testMultiTurn() {
 async function testOCR() {
     console.log('--- Testing OCR Capability ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'ocr',
         model: 'mistral-ocr-latest',
         data: {
@@ -295,7 +295,7 @@ async function testOCR() {
 async function testUnsupportedModality() {
     console.log('--- Testing Unsupported Modality (Image Generation) ---');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         type: 'image',
         model: 'mistral-large-latest',
         data: {

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

@@ -3,7 +3,7 @@ const adapter = new OpenAIAdapter();
 async function testChatCompletion() {
     console.log('Testing chat completion...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -25,7 +25,7 @@ async function testChatCompletion() {
 async function testChatWithVision() {
     console.log('Testing chat with vision...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -50,7 +50,7 @@ async function testChatWithVision() {
 async function testImageGeneration() {
     console.log('Testing image generation...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'dall-e-3',
         type: 'image',
         data: {
@@ -69,7 +69,7 @@ async function testImageGeneration() {
 async function testEmbeddings() {
     console.log('Testing embeddings...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'text-embedding-3-small',
         type: 'embeddings',
         data: {
@@ -85,7 +85,7 @@ async function testEmbeddings() {
 async function testTextToSpeech() {
     console.log('Testing text-to-speech...');
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'tts-1',
         type: 'tts',
         data: {
@@ -104,7 +104,7 @@ async function testToolCalling() {
     console.log('Testing tool calling...');
     // Step 1: Initial request with tool available
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -145,7 +145,7 @@ async function testToolCalling() {
     console.log('Function arguments:', toolCall.function.arguments);
     // Step 2: Send tool response back
     const toolResponseRequest = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {
@@ -173,7 +173,7 @@ async function testContentAndToolCalls() {
     console.log('Testing content + tool calls in same message...');
     // This tests the fix we made - assistant messages can have BOTH content and toolCalls
     const request = {
-        gate: 'test-gate',
+        gateId: 'test-gate',
         model: 'gpt-4o-mini',
         type: 'chat',
         data: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@layer-ai/core",
-  "version": "0.9.1",
+  "version": "1.0.0",
   "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": "^1.0.2"
+    "@layer-ai/sdk": "^2.0.0"
   },
   "devDependencies": {
     "@types/bcryptjs": "^2.4.6",