@layer-ai/core 0.7.2 → 0.7.3

package/dist/routes/v2/tests/test-byok-completion.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Test BYOK completion flow
+ *
+ * This test verifies that when making a completion request:
+ * 1. The system checks for user's BYOK keys
+ * 2. Uses BYOK keys when available
+ * 3. Falls back to platform keys when BYOK not configured
+ */
+export {};
+//# sourceMappingURL=test-byok-completion.d.ts.map

package/dist/routes/v2/tests/test-byok-completion.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-byok-completion.d.ts","sourceRoot":"","sources":["../../../../src/routes/v2/tests/test-byok-completion.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG"}

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

@@ -0,0 +1,82 @@
+/**
+ * Test BYOK completion flow
+ *
+ * This test verifies that when making a completion request:
+ * 1. The system checks for user's BYOK keys
+ * 2. Uses BYOK keys when available
+ * 3. Falls back to platform keys when BYOK not configured
+ */
+import { OpenAIAdapter } from '../../../services/providers/openai-adapter.js';
+import { AnthropicAdapter } from '../../../services/providers/anthropic-adapter.js';
+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',
+    model: 'gpt-4o-mini',
+    type: 'chat',
+    data: {
+        messages: [
+            {
+                role: 'user',
+                content: 'Say "BYOK test successful" and nothing else.',
+            },
+        ],
+        maxTokens: 20,
+    },
+};
+async function testBYOKCompletion() {
+    console.log('\n🧪 Testing BYOK Completion Flow...\n');
+    try {
+        // Test 1: OpenAI with BYOK (should use user's key)
+        console.log('Test 1: OpenAI completion with BYOK key');
+        console.log('----------------------------------------');
+        const openaiAdapter = new OpenAIAdapter();
+        const openaiRequest = { ...SIMPLE_REQUEST, model: 'gpt-4o-mini' };
+        console.log(`Making request to OpenAI with userId: ${TEST_USER_ID}`);
+        const openaiResult = await openaiAdapter.call(openaiRequest, TEST_USER_ID);
+        console.log('✓ OpenAI request successful');
+        console.log(`Response: ${openaiResult.content}`);
+        console.log('Note: This request used your BYOK OpenAI key\n');
+        // Test 2: Anthropic with BYOK (should use user's key)
+        console.log('Test 2: Anthropic completion with BYOK key');
+        console.log('------------------------------------------');
+        const anthropicAdapter = new AnthropicAdapter();
+        const anthropicRequest = { ...SIMPLE_REQUEST, model: 'claude-3-haiku-20240307' };
+        console.log(`Making request to Anthropic with userId: ${TEST_USER_ID}`);
+        const anthropicResult = await anthropicAdapter.call(anthropicRequest, TEST_USER_ID);
+        console.log('✓ Anthropic request successful');
+        console.log(`Response: ${anthropicResult.content}`);
+        console.log('Note: This request used your BYOK Anthropic key\n');
+        // Test 3: Google without BYOK (should use platform key)
+        console.log('Test 3: Google completion without BYOK key');
+        console.log('------------------------------------------');
+        const googleAdapter = new GoogleAdapter();
+        const googleRequest = { ...SIMPLE_REQUEST, model: 'gemini-2.0-flash' };
+        console.log(`Making request to Google with userId: ${TEST_USER_ID}`);
+        const googleResult = await googleAdapter.call(googleRequest, TEST_USER_ID);
+        console.log('✓ Google request successful');
+        console.log(`Response: ${googleResult.content}`);
+        console.log('Note: This request used the platform Google key (no BYOK configured)\n');
+        console.log('✅ All BYOK completion tests passed!\n');
+        console.log('Summary:');
+        console.log('- OpenAI: Used BYOK key ✓');
+        console.log('- Anthropic: Used BYOK key ✓');
+        console.log('- Google: Used platform key (fallback) ✓');
+    }
+    catch (error) {
+        console.error('\n❌ BYOK completion test failed:', error);
+        if (error instanceof Error) {
+            console.error('Error message:', error.message);
+            console.error('Stack:', error.stack);
+        }
+        throw error;
+    }
+}
+// Run the test
+testBYOKCompletion()
+    .then(() => process.exit(0))
+    .catch((error) => {
+    console.error('Test failed:', error);
+    process.exit(1);
+});

package/dist/routes/v2/tests/test-byok.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Test BYOK (Bring Your Own Keys) functionality
+ *
+ * This test verifies that when a user has configured their own provider keys,
+ * those keys are used instead of the platform keys.
+ */
+export {};
+//# sourceMappingURL=test-byok.d.ts.map

package/dist/routes/v2/tests/test-byok.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-byok.d.ts","sourceRoot":"","sources":["../../../../src/routes/v2/tests/test-byok.ts"],"names":[],"mappings":"AAAA;;;;;GAKG"}

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

@@ -0,0 +1,73 @@
+/**
+ * Test BYOK (Bring Your Own Keys) functionality
+ *
+ * This test verifies that when a user has configured their own provider keys,
+ * those keys are used instead of the platform keys.
+ */
+import { db } from '../../../lib/db/postgres.js';
+import { decrypt } from '../../../lib/encryption.js';
+async function testBYOK() {
+    console.log('\n🔑 Testing BYOK Key Resolution...\n');
+    // Test user ID from the database query
+    const userId = 'ebd64998-465d-4211-ad67-87b4e01ad0da';
+    try {
+        // Test 1: Retrieve OpenAI key
+        console.log('Test 1: Retrieving OpenAI BYOK key...');
+        const openaiKey = await db.getProviderKey(userId, 'openai');
+        if (openaiKey) {
+            console.log('✓ OpenAI key found in database');
+            console.log(`  Provider: ${openaiKey.provider}`);
+            console.log(`  Key Prefix: ${openaiKey.keyPrefix}`);
+            console.log(`  Created: ${openaiKey.createdAt}`);
+            // Decrypt and verify the key
+            const encryptionKey = process.env.ENCRYPTION_KEY;
+            if (!encryptionKey) {
+                throw new Error('ENCRYPTION_KEY not found in environment');
+            }
+            const decryptedKey = decrypt(openaiKey.encryptedKey, encryptionKey);
+            console.log(`  Decrypted key starts with: ${decryptedKey.substring(0, 10)}...`);
+            console.log(`  Decrypted key length: ${decryptedKey.length} characters`);
+        }
+        else {
+            console.log('✗ No OpenAI key found');
+        }
+        console.log('\nTest 2: Retrieving Anthropic BYOK key...');
+        const anthropicKey = await db.getProviderKey(userId, 'anthropic');
+        if (anthropicKey) {
+            console.log('✓ Anthropic key found in database');
+            console.log(`  Provider: ${anthropicKey.provider}`);
+            console.log(`  Key Prefix: ${anthropicKey.keyPrefix}`);
+            console.log(`  Created: ${anthropicKey.createdAt}`);
+            const encryptionKey = process.env.ENCRYPTION_KEY;
+            if (!encryptionKey) {
+                throw new Error('ENCRYPTION_KEY not found in environment');
+            }
+            const decryptedKey = decrypt(anthropicKey.encryptedKey, encryptionKey);
+            console.log(`  Decrypted key starts with: ${decryptedKey.substring(0, 14)}...`);
+            console.log(`  Decrypted key length: ${decryptedKey.length} characters`);
+        }
+        else {
+            console.log('✗ No Anthropic key found');
+        }
+        console.log('\nTest 3: Retrieving Google BYOK key (should not exist)...');
+        const googleKey = await db.getProviderKey(userId, 'google');
+        if (googleKey) {
+            console.log('✓ Google key found in database');
+        }
+        else {
+            console.log('✓ No Google key found (as expected)');
+        }
+        console.log('\n✅ BYOK test completed successfully!\n');
+    }
+    catch (error) {
+        console.error('\n❌ BYOK test failed:', error);
+        throw error;
+    }
+}
+// Run the test
+testBYOK()
+    .then(() => process.exit(0))
+    .catch((error) => {
+    console.error('Test failed:', error);
+    process.exit(1);
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@layer-ai/core",
-  "version": "0.7.2",
+  "version": "0.7.3",
   "description": "Core API routes and services for Layer AI",
   "type": "module",
   "main": "./dist/index.js",