@layer-ai/core 0.2.2 → 0.3.1

package/dist/lib/db/postgres.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../../src/lib/db/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAyB,MAAM,eAAe,CAAC;AAO/E,iBAAS,OAAO,IAAI,EAAE,CAAC,IAAI,CAqB1B;AA0BD,eAAO,MAAM,EAAE;gBAEK,MAAM,WAAW,GAAG,EAAE;0BASZ,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;oBAQnC,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;sBAQ3B,MAAM,gBAAgB,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;6BASrC,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;yBAQnC,MAAM,WAAW,MAAM,aAAa,MAAM,QAAQ,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;kCAQjE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;8BAO1B,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;qBAQnC,MAAM,UAAU,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;iCAS7B,MAAM,YAAY,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;4BAQpD,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;uBAQ7B,MAAM,QAAQ,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;oBAsBpC,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;mBAQ9B,MAAM,QAAQ,GAAG,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;mBAgCxC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;qBASvB,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;iCAgBP,MAAM,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,IAAI,CAAA;KAAE,GAAG,IAAI,CAAC;6BAQhE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;qCAehB,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAM9D,CAAC;AAEF,eAAe,OAAO,CAAC"}
+{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../../src/lib/db/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAyB,MAAM,eAAe,CAAC;AAO/E,iBAAS,OAAO,IAAI,EAAE,CAAC,IAAI,CAqB1B;AA0BD,eAAO,MAAM,EAAE;gBAEK,MAAM,WAAW,GAAG,EAAE;0BASZ,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;oBAQnC,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;sBAQ3B,MAAM,gBAAgB,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;6BASrC,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;yBAQnC,MAAM,WAAW,MAAM,aAAa,MAAM,QAAQ,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;kCAQjE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;8BAO1B,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;qBAQnC,MAAM,UAAU,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;iCAS7B,MAAM,YAAY,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;4BAQpD,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;uBAQ7B,MAAM,QAAQ,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;oBA2BpC,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;mBAQ9B,MAAM,QAAQ,GAAG,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;mBA0CxC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;qBASvB,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;iCAgBP,MAAM,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,IAAI,CAAA;KAAE,GAAG,IAAI,CAAC;6BAQhE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;qCAehB,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAM9D,CAAC;AAEF,eAAe,OAAO,CAAC"}

package/dist/lib/db/postgres.js

@@ -31,7 +31,7 @@ function toCamelCase(obj) {
         const camelKey = key.replace(/_([a-z])/g, (_, letter) => letter.toUpperCase());
         let value = obj[key];
         // Convert numeric strings to numbers for specific fields
-        if ((camelKey === 'temperature' || camelKey === 'topP') && typeof value === 'string') {
+        if ((camelKey === 'temperature' || camelKey === 'topP' || camelKey === 'costWeight' || camelKey === 'latencyWeight' || camelKey === 'qualityWeight') && typeof value === 'string') {
             value = parseFloat(value);
         }
         if (camelKey === 'maxTokens' && typeof value === 'string') {
@@ -94,11 +94,12 @@ export const db = {
         return result.rows.map(toCamelCase);
     },
     async createGate(userId, data) {
-        const result = await getPool().query(`INSERT INTO gates (user_id, name, description, model, system_prompt, allow_overrides, temperature, max_tokens, top_p, tags, routing_strategy, fallback_models)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12) RETURNING *`, [
+        const result = await getPool().query(`INSERT INTO gates (user_id, name, description, task_type, model, system_prompt, allow_overrides, temperature, max_tokens, top_p, tags, routing_strategy, fallback_models, cost_weight, latency_weight, quality_weight, reanalysis_period)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17) RETURNING *`, [
             userId,
             data.name,
             data.description,
+            data.taskType,
             data.model,
             data.systemPrompt,
             data.allowOverrides ? JSON.stringify(data.allowOverrides) : null,
@@ -107,7 +108,11 @@ export const db = {
             data.topP,
             JSON.stringify(data.tags || []),
             data.routingStrategy || 'single',
-            JSON.stringify(data.fallbackModels || [])
+            JSON.stringify(data.fallbackModels || []),
+            data.costWeight ?? 0.33,
+            data.latencyWeight ?? 0.33,
+            data.qualityWeight ?? 0.34,
+            data.reanalysisPeriod || 'never'
         ]);
         return toCamelCase(result.rows[0]);
     },
@@ -118,19 +123,25 @@ export const db = {
     async updateGate(id, data) {
         const result = await getPool().query(`UPDATE gates SET
         description = COALESCE($2, description),
-        model = COALESCE($3, model),
-        system_prompt = COALESCE($4, system_prompt),
-        allow_overrides = COALESCE($5, allow_overrides),
-        temperature = COALESCE($6, temperature),
-        max_tokens = COALESCE($7, max_tokens),
-        top_p = COALESCE($8, top_p),
-        tags = COALESCE($9, tags),
-        routing_strategy = COALESCE($10, routing_strategy),
-        fallback_models = COALESCE($11, fallback_models),
+        task_type = COALESCE($3, task_type),
+        model = COALESCE($4, model),
+        system_prompt = COALESCE($5, system_prompt),
+        allow_overrides = COALESCE($6, allow_overrides),
+        temperature = COALESCE($7, temperature),
+        max_tokens = COALESCE($8, max_tokens),
+        top_p = COALESCE($9, top_p),
+        tags = COALESCE($10, tags),
+        routing_strategy = COALESCE($11, routing_strategy),
+        fallback_models = COALESCE($12, fallback_models),
+        cost_weight = COALESCE($13, cost_weight),
+        latency_weight = COALESCE($14, latency_weight),
+        quality_weight = COALESCE($15, quality_weight),
+        reanalysis_period = COALESCE($16, reanalysis_period),
         updated_at = NOW()
       WHERE id = $1 RETURNING *`, [
             id,
             data.description,
+            data.taskType,
             data.model,
             data.systemPrompt,
             data.allowOverrides ? JSON.stringify(data.allowOverrides) : null,
@@ -140,6 +151,10 @@ export const db = {
             data.tags ? JSON.stringify(data.tags) : null,
             data.routingStrategy,
             data.fallbackModels ? JSON.stringify(data.fallbackModels) : null,
+            data.costWeight,
+            data.latencyWeight,
+            data.qualityWeight,
+            data.reanalysisPeriod,
         ]);
         return result.rows[0] ? toCamelCase(result.rows[0]) : null;
     },

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;AAUpD,QAAA,MAAM,MAAM,EAAE,UAAqB,CAAC;AAgQpC,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;AAYpD,QAAA,MAAM,MAAM,EAAE,UAAqB,CAAC;AAgQpC,eAAe,MAAM,CAAC"}

package/dist/routes/v2/complete.js

@@ -4,6 +4,7 @@ import { cache } from '../../lib/db/redis.js';
 import { authenticate } from '../../middleware/auth.js';
 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 { MODEL_REGISTRY, OverrideField } from '@layer-ai/sdk';
 const router = Router();
 // MARK:- Helper Functions
@@ -70,8 +71,8 @@ async function callProvider(request) {
             return await adapter.call(request);
         }
         case 'google':
-            // TODO: Migrate Google to use adapter pattern
-            throw new Error('Google provider not yet migrated to v2 API');
+            const adapter = new GoogleAdapter();
+            return await adapter.call(request);
         default:
             throw new Error(`Unknown provider: ${provider}`);
     }

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

@@ -1 +1 @@
-{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/services/providers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAe7D,MAAM,WAAW,yBAAyB;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,yBAAyB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAkDnG"}
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/services/providers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAe7D,MAAM,WAAW,yBAAyB;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,yBAAyB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAoDnG"}

package/dist/services/providers/anthropic.js

@@ -45,8 +45,10 @@ export async function createCompletion(params) {
     const completionTokens = response.usage.output_tokens;
     const totalTokens = promptTokens + completionTokens;
     // Calculate cost
-    const pricing = MODEL_REGISTRY[params.model].pricing;
-    const costUsd = (promptTokens / 1000 * pricing.input) + (completionTokens / 1000 * pricing.output);
+    const modelInfo = MODEL_REGISTRY[params.model];
+    const costUsd = ('pricing' in modelInfo && modelInfo.pricing?.input && modelInfo.pricing?.output)
+        ? (promptTokens / 1000 * modelInfo.pricing.input) + (completionTokens / 1000 * modelInfo.pricing.output)
+        : 0;
     return {
         content,
         promptTokens,

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;AAEvB,OAAO,EAAE,eAAe,EAAE,CAAC;AAE3B,8BAAsB,mBAAmB;IACvC,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAEpC,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,GAAG,OAAO,CAAC,aAAa,CAAC;IAE5D,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;CAMV"}
+{"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;AAEvB,OAAO,EAAE,eAAe,EAAE,CAAC;AAE3B,8BAAsB,mBAAmB;IACvC,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAEpC,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,GAAG,OAAO,CAAC,aAAa,CAAC;IAE5D,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;CAOV"}

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

@@ -82,8 +82,9 @@ export class BaseProviderAdapter {
     }
     calculateCost(model, promptTokens, completionTokens) {
         const modelInfo = MODEL_REGISTRY[model];
-        return modelInfo?.pricing
-            ? (promptTokens / 1000 * modelInfo.pricing.input) + (completionTokens / 1000 * modelInfo.pricing.output)
-            : 0;
+        if (!modelInfo || !('pricing' in modelInfo) || !modelInfo.pricing?.input || !modelInfo.pricing?.output) {
+            return 0;
+        }
+        return (promptTokens / 1000 * modelInfo.pricing.input) + (completionTokens / 1000 * modelInfo.pricing.output);
     }
 }

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

@@ -0,0 +1,21 @@
+import { FunctionCallingConfigMode } from '@google/genai';
+import { LayerRequest, LayerResponse, Role, FinishReason, VideoSize } from '@layer-ai/sdk';
+import { BaseProviderAdapter } from './base-adapter.js';
+export declare class GoogleAdapter extends BaseProviderAdapter {
+    protected provider: string;
+    protected roleMappings: Record<Role, string>;
+    protected finishReasonMappings: Record<string, FinishReason>;
+    protected toolChoiceMappings: Record<string, FunctionCallingConfigMode>;
+    protected videoSizeConfig: Record<VideoSize, {
+        aspectRatio: string;
+        resolution: string;
+    }>;
+    call(request: LayerRequest): Promise<LayerResponse>;
+    private handleChat;
+    private handleImageGeneration;
+    private handleEmbeddings;
+    private handleVideoGeneration;
+    private handleTextToSpeech;
+    private sleep;
+}
+//# sourceMappingURL=google-adapter.d.ts.map

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

@@ -0,0 +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;AAYxD,qBAAa,aAAc,SAAQ,mBAAmB;IACpD,SAAS,CAAC,QAAQ,SAAY;IAE9B,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,GAAG,OAAO,CAAC,aAAa,CAAC;YAiB3C,UAAU;YAqLV,qBAAqB;YAkCrB,gBAAgB;YAmChB,qBAAqB;YAmHrB,kBAAkB;IAwChC,OAAO,CAAC,KAAK;CAGd"}

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

@@ -0,0 +1,408 @@
+import { GoogleGenAI, FunctionCallingConfigMode, VideoGenerationReferenceType, } from '@google/genai';
+import { BaseProviderAdapter } from './base-adapter.js';
+import { ADAPTER_HANDLED } from './base-adapter.js';
+let client = null;
+function getGoogleClient() {
+    if (!client) {
+        client = new GoogleGenAI({ apiKey: process.env.GOOGLE_API_KEY || '' });
+    }
+    return client;
+}
+export class GoogleAdapter extends BaseProviderAdapter {
+    constructor() {
+        super(...arguments);
+        this.provider = 'google';
+        this.roleMappings = {
+            system: ADAPTER_HANDLED,
+            user: 'user',
+            assistant: 'model',
+            tool: 'function',
+            function: 'function',
+            model: 'model',
+            developer: 'system',
+        };
+        // Map Google finish reasons to Layer finish reasons
+        this.finishReasonMappings = {
+            STOP: 'completed',
+            MAX_TOKENS: 'length_limit',
+            SAFETY: 'filtered',
+            RECITATION: 'filtered',
+            FINISH_REASON_UNSPECIFIED: 'completed',
+            OTHER: 'completed',
+        };
+        this.toolChoiceMappings = {
+            auto: FunctionCallingConfigMode.AUTO,
+            none: FunctionCallingConfigMode.NONE,
+            required: FunctionCallingConfigMode.ANY,
+        };
+        // Map Layer VideoSize to Veo aspect ratio and resolution
+        this.videoSizeConfig = {
+            '720x1280': { aspectRatio: '9:16', resolution: '720p' },
+            '1280x720': { aspectRatio: '16:9', resolution: '720p' },
+            '1024x1792': { aspectRatio: '9:16', resolution: '1080p' },
+            '1792x1024': { aspectRatio: '16:9', resolution: '1080p' },
+        };
+    }
+    async call(request) {
+        switch (request.type) {
+            case 'chat':
+                return this.handleChat(request);
+            case 'image':
+                return this.handleImageGeneration(request);
+            case 'embeddings':
+                return this.handleEmbeddings(request);
+            case 'tts':
+                return this.handleTextToSpeech(request);
+            case 'video':
+                return this.handleVideoGeneration(request);
+            default:
+                throw new Error(`Unknown modality: ${request.type}`);
+        }
+    }
+    async handleChat(request) {
+        const startTime = Date.now();
+        const client = getGoogleClient();
+        const { data: chat, model } = request;
+        if (!model) {
+            throw new Error('Model is required for chat completion');
+        }
+        const contents = [];
+        let systemInstruction;
+        // Handle system prompt
+        if (chat.systemPrompt) {
+            systemInstruction = chat.systemPrompt;
+        }
+        // Convert messages to Google format
+        for (const msg of chat.messages) {
+            const role = this.mapRole(msg.role);
+            // Skip system messages (handled via systemInstruction)
+            if (role === 'system') {
+                systemInstruction = systemInstruction
+                    ? `${systemInstruction}\n${msg.content}`
+                    : msg.content;
+                continue;
+            }
+            const parts = [];
+            // Handle text content
+            if (msg.content) {
+                parts.push({ text: msg.content });
+            }
+            // Handle images
+            if (msg.images && msg.images.length > 0) {
+                for (const image of msg.images) {
+                    if (image.base64) {
+                        parts.push({
+                            inlineData: {
+                                mimeType: image.mimeType || 'image/jpeg',
+                                data: image.base64,
+                            },
+                        });
+                    }
+                    else if (image.url) {
+                        parts.push({
+                            fileData: {
+                                mimeType: image.mimeType || 'image/jpeg',
+                                fileUri: image.url,
+                            },
+                        });
+                    }
+                }
+            }
+            // Handle tool responses
+            if (msg.toolCallId && msg.role === 'tool') {
+                if (!msg.name) {
+                    throw new Error('Tool response messages must include the function name');
+                }
+                parts.push({
+                    functionResponse: {
+                        name: msg.name || msg.toolCallId,
+                        response: { result: msg.content },
+                    },
+                });
+            }
+            // Handle assistant messages with tool calls
+            if (msg.toolCalls && msg.toolCalls.length > 0) {
+                for (const toolCall of msg.toolCalls) {
+                    parts.push({
+                        functionCall: {
+                            name: toolCall.function.name,
+                            args: JSON.parse(toolCall.function.arguments),
+                        },
+                    });
+                }
+            }
+            if (parts.length > 0) {
+                contents.push({
+                    role: role === 'model' ? 'model' : 'user',
+                    parts,
+                });
+            }
+        }
+        // Convert tools to Google format
+        let googleTools;
+        if (chat.tools && chat.tools.length > 0) {
+            googleTools = [
+                {
+                    functionDeclarations: chat.tools.map((tool) => ({
+                        name: tool.function.name,
+                        description: tool.function.description,
+                        parametersJsonSchema: tool.function.parameters,
+                    })),
+                },
+            ];
+        }
+        // Map tool choice
+        let toolConfig;
+        if (chat.toolChoice) {
+            const mode = this.mapToolChoice(chat.toolChoice);
+            if (typeof mode === 'string') {
+                toolConfig = {
+                    functionCallingConfig: { mode: mode },
+                };
+            }
+        }
+        const response = await client.models.generateContent({
+            model: model,
+            contents,
+            config: {
+                ...(systemInstruction && { systemInstruction }),
+                ...(googleTools && { tools: googleTools }),
+                ...(toolConfig && { toolConfig }),
+                ...(chat.temperature !== undefined && { temperature: chat.temperature }),
+                ...(chat.maxTokens !== undefined && { maxOutputTokens: chat.maxTokens }),
+                ...(chat.topP !== undefined && { topP: chat.topP }),
+                ...(chat.stopSequences !== undefined && { stopSequences: chat.stopSequences }),
+            }
+        });
+        const candidate = response.candidates?.[0];
+        const content = candidate?.content;
+        const textContent = content?.parts
+            ?.filter((part) => 'text' in part)
+            .map((part) => part.text)
+            .join('');
+        // Extract tool calls
+        const toolCalls = content?.parts
+            ?.filter((part) => 'functionCall' in part)
+            .map((part, index) => {
+            const fc = part.functionCall;
+            return {
+                id: `call_${index}_${fc.name}`,
+                type: 'function',
+                function: {
+                    name: fc.name,
+                    arguments: JSON.stringify(fc.args),
+                },
+            };
+        });
+        const promptTokens = response.usageMetadata?.promptTokenCount || 0;
+        const completionTokens = response.usageMetadata?.candidatesTokenCount || 0;
+        const totalTokens = response.usageMetadata?.totalTokenCount || 0;
+        const cost = this.calculateCost(model, promptTokens, completionTokens);
+        return {
+            content: textContent || undefined,
+            toolCalls: toolCalls && toolCalls.length > 0 ? toolCalls : undefined,
+            model: response.modelVersion || model,
+            finishReason: this.mapFinishReason(candidate?.finishReason || 'STOP'),
+            rawFinishReason: candidate?.finishReason,
+            usage: {
+                promptTokens,
+                completionTokens,
+                totalTokens,
+            },
+            cost,
+            latencyMs: Date.now() - startTime,
+            raw: response,
+        };
+    }
+    async handleImageGeneration(request) {
+        const startTime = Date.now();
+        const client = getGoogleClient();
+        const { data: image, model } = request;
+        if (!model) {
+            throw new Error('Model is required for chat completion');
+        }
+        // Google's Imagen API via generateImages
+        const response = await client.models.generateImages({
+            model: model,
+            prompt: image.prompt,
+            config: {
+                numberOfImages: image.count || 1,
+                ...(image.seed !== undefined && { seed: image.seed }),
+            },
+        });
+        const images = response.generatedImages?.map((img) => ({
+            base64: img.image?.imageBytes,
+        })) || [];
+        return {
+            images,
+            model: model,
+            latencyMs: Date.now() - startTime,
+            raw: response,
+        };
+    }
+    async handleEmbeddings(request) {
+        const startTime = Date.now();
+        const client = getGoogleClient();
+        const { data: embedding, model } = request;
+        if (!model) {
+            throw new Error('Model is required for chat completion');
+        }
+        const inputs = Array.isArray(embedding.input)
+            ? embedding.input
+            : [embedding.input];
+        const response = await client.models.embedContent({
+            model: model,
+            contents: inputs.map((text) => ({ parts: [{ text }] })),
+        });
+        const embeddings = response.embeddings?.map((e) => e.values || []) || [];
+        return {
+            embeddings,
+            model: model,
+            usage: {
+                promptTokens: 0, // Google doesn't provide token count for embeddings
+                completionTokens: 0,
+                totalTokens: 0,
+            },
+            latencyMs: Date.now() - startTime,
+            raw: response,
+        };
+    }
+    async handleVideoGeneration(request) {
+        const startTime = Date.now();
+        const client = getGoogleClient();
+        const { data: video, model } = request;
+        if (!model) {
+            throw new Error('Model is required for chat completion');
+        }
+        // Derive aspect ratio and resolution from size
+        const sizeConfig = video.size ? this.videoSizeConfig[video.size] : null;
+        const aspectRatio = sizeConfig?.aspectRatio || '16:9';
+        const resolution = sizeConfig?.resolution || '720p';
+        // Parse duration - Veo accepts 4, 6, or 8 seconds
+        let durationSeconds;
+        if (video.duration !== undefined) {
+            const dur = typeof video.duration === 'string'
+                ? parseInt(video.duration, 10)
+                : video.duration;
+            // Veo supports 4, 6, or 8 seconds
+            if (dur <= 5)
+                durationSeconds = 5;
+            else if (dur <= 6)
+                durationSeconds = 6;
+            else
+                durationSeconds = 8;
+        }
+        // Build reference images for Veo 3.1
+        let referenceImages;
+        if (video.referenceImages && video.referenceImages.length > 0) {
+            const refTypeMap = {
+                subject: VideoGenerationReferenceType.ASSET,
+                style: VideoGenerationReferenceType.STYLE,
+                asset: VideoGenerationReferenceType.ASSET,
+            };
+            referenceImages = video.referenceImages.slice(0, 3).map((ref) => ({
+                image: ref.base64 ? { imageBytes: ref.base64 } : { imageUri: ref.url },
+                referenceType: refTypeMap[ref.referenceType || 'asset'],
+            }));
+        }
+        // Build starting image if provided
+        let image;
+        if (video.image) {
+            image = video.image.base64
+                ? { imageBytes: video.image.base64 }
+                : { imageUri: video.image.url };
+        }
+        // Build last frame for interpolation
+        let lastFrame;
+        if (video.lastFrame) {
+            lastFrame = video.lastFrame.base64
+                ? { imageBytes: video.lastFrame.base64 }
+                : { imageUri: video.lastFrame.url };
+        }
+        // Start the video generation operation
+        let operation = await client.models.generateVideos({
+            model: model,
+            prompt: video.prompt,
+            ...(image && { image }),
+            config: {
+                aspectRatio,
+                resolution,
+                ...(durationSeconds !== undefined && { durationSeconds }),
+                ...(video.seed !== undefined && { seed: video.seed }),
+                ...(video.negativePrompt && { negativePrompt: video.negativePrompt }),
+                ...(video.personGeneration && {
+                    personGeneration: video.personGeneration,
+                }),
+                ...(video.numberOfVideos && { numberOfVideos: video.numberOfVideos }),
+                ...(referenceImages && { referenceImages }),
+                ...(lastFrame && { lastFrame }),
+            },
+        });
+        // Poll until the operation completes
+        // Veo operations can take 11 seconds to 6 minutes
+        const pollIntervalMs = 5000; // Poll every 5 seconds
+        const maxWaitMs = 10 * 60 * 1000; // Max wait 10 minutes
+        const startPoll = Date.now();
+        while (!operation.done) {
+            if (Date.now() - startPoll > maxWaitMs) {
+                throw new Error('Video generation timed out after 10 minutes');
+            }
+            await this.sleep(pollIntervalMs);
+            operation = (await client.operations.get({
+                operation: operation,
+            }));
+        }
+        // Extract generated videos
+        const generatedVideos = operation.response?.generatedVideos || [];
+        const videos = generatedVideos.map((vid) => {
+            if (vid.video) {
+                return {
+                    url: vid.video.uri,
+                    duration: durationSeconds,
+                };
+            }
+            return {};
+        });
+        return {
+            videos: videos.filter((v) => v.url),
+            model: model,
+            latencyMs: Date.now() - startTime,
+            raw: operation.response,
+        };
+    }
+    async handleTextToSpeech(request) {
+        const startTime = Date.now();
+        const client = getGoogleClient();
+        const { data: tts, model } = request;
+        if (!model) {
+            throw new Error('Model is required for chat completion');
+        }
+        const response = await client.models.generateContent({
+            model: model,
+            contents: tts.input,
+            config: {
+                responseModalities: ['AUDIO'],
+                speechConfig: {
+                    voiceConfig: {
+                        prebuiltVoiceConfig: {
+                            voiceName: tts.voice || 'Kore',
+                        },
+                    },
+                },
+            },
+        });
+        const audioData = response.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
+        return {
+            audio: {
+                base64: audioData,
+                format: 'wav',
+            },
+            model: model,
+            latencyMs: Date.now() - startTime,
+            raw: response,
+        };
+    }
+    sleep(ms) {
+        return new Promise((resolve) => setTimeout(resolve, ms));
+    }
+}

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

@@ -1 +1 @@
-{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../../src/services/providers/google.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAY7D,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAGD,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAiChG"}
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../../src/services/providers/google.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAY7D,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAGD,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAmChG"}

package/dist/services/providers/google.js

@@ -27,8 +27,10 @@ export async function createCompletion(params) {
     const promptTokens = usageMetadata?.promptTokenCount || 0;
     const completionTokens = usageMetadata?.candidatesTokenCount || 0;
     const totalTokens = usageMetadata?.totalTokenCount || (promptTokens + completionTokens);
-    const pricing = MODEL_REGISTRY[params.model].pricing;
-    const costUsd = (promptTokens / 1000 * pricing.input) + (completionTokens / 1000 * pricing.output);
+    const modelInfo = MODEL_REGISTRY[params.model];
+    const costUsd = ('pricing' in modelInfo && modelInfo.pricing?.input && modelInfo.pricing?.output)
+        ? (promptTokens / 1000 * modelInfo.pricing.input) + (completionTokens / 1000 * modelInfo.pricing.output)
+        : 0;
     return {
         content,
         promptTokens,

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

@@ -1 +1 @@
-{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/services/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAe7D,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAmChG"}
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/services/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAe7D,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAqChG"}

package/dist/services/providers/openai.js

@@ -31,8 +31,10 @@ export async function createCompletion(params) {
     const completionTokens = response.usage?.completion_tokens || 0;
     const totalTokens = response.usage?.total_tokens || 0;
     // Calculate cost
-    const pricing = MODEL_REGISTRY[params.model].pricing;
-    const costUsd = (promptTokens / 1000 * pricing.input) + (completionTokens / 1000 * pricing.output);
+    const modelInfo = MODEL_REGISTRY[params.model];
+    const costUsd = ('pricing' in modelInfo && modelInfo.pricing?.input && modelInfo.pricing?.output)
+        ? (promptTokens / 1000 * modelInfo.pricing.input) + (completionTokens / 1000 * modelInfo.pricing.output)
+        : 0;
     return {
         content,
         promptTokens,

package/dist/services/providers/tests/test-google-adapter.d.ts

@@ -0,0 +1,2 @@
+import 'dotenv/config';
+//# sourceMappingURL=test-google-adapter.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"test-google-adapter.d.ts","sourceRoot":"","sources":["../../../../src/services/providers/tests/test-google-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,eAAe,CAAC"}

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

@@ -0,0 +1,224 @@
+import 'dotenv/config';
+import { GoogleAdapter } from '../google-adapter.js';
+const adapter = new GoogleAdapter();
+async function testChatCompletion() {
+    console.log('Testing chat completion...');
+    const request = {
+        gate: 'test-gate',
+        model: 'gemini-2.5-flash',
+        type: 'chat',
+        data: {
+            messages: [
+                { role: 'user', content: 'Say "Hello World" and nothing else.' },
+            ],
+            temperature: 0.7,
+            maxTokens: 10,
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Response:', response.content);
+    console.log('Tokens:', response.usage);
+    console.log('Cost:', response.cost);
+    console.log('Latency:', response.latencyMs + 'ms');
+    console.log('Finish reason:', response.finishReason);
+    console.log('✅ Chat completion test passed\n');
+}
+async function testChatWithVision() {
+    console.log('Testing chat with vision...');
+    const request = {
+        gate: 'test-gate',
+        model: 'gemini-2.5-flash',
+        type: 'chat',
+        data: {
+            messages: [
+                {
+                    role: 'user',
+                    content: 'What color is the sky in this image?',
+                    images: [
+                        {
+                            url: 'https://images.unsplash.com/photo-1765202659641-9ad9facfe5cf?q=80&w=1364&auto=format&fit=crop&ixlib=rb-4.1.0&ixid=M3wxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8fA%3D%3D',
+                        },
+                    ],
+                },
+            ],
+            maxTokens: 50,
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Response:', response.content);
+    console.log('Finish reason:', response.finishReason);
+    console.log('✅ Vision test passed\n');
+}
+async function testImageGeneration() {
+    console.log('Testing image generation...');
+    const request = {
+        gate: 'test-gate',
+        model: 'imagen-4.0-generate-001',
+        type: 'image',
+        data: {
+            prompt: 'A cute cat playing with a ball of yarn',
+            count: 1,
+        },
+    };
+    const response = await adapter.call(request);
+    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('✅ Image generation test passed\n');
+}
+async function testEmbeddings() {
+    console.log('Testing embeddings...');
+    const request = {
+        gate: 'test-gate',
+        model: 'text-embedding-004',
+        type: 'embeddings',
+        data: {
+            input: 'Hello world',
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Embeddings dimensions:', response.embeddings?.[0]?.length);
+    console.log('Latency:', response.latencyMs + 'ms');
+    console.log('✅ Embeddings test passed\n');
+}
+async function testToolCalling() {
+    console.log('Testing tool calling...');
+    // Step 1: Send message with tools available
+    const request = {
+        gate: 'test-gate',
+        model: 'gemini-2.5-flash',
+        type: 'chat',
+        data: {
+            messages: [
+                { role: 'user', content: 'Use the get_current_time function to tell me what time it is.' },
+            ],
+            tools: [
+                {
+                    type: 'function',
+                    function: {
+                        name: 'get_current_time',
+                        description: 'Returns the current time in ISO format',
+                        parameters: {
+                            type: 'object',
+                            properties: {},
+                        },
+                    },
+                },
+            ],
+            toolChoice: 'required',
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Response content:', response.content);
+    console.log('Tool calls:', response.toolCalls);
+    console.log('Finish reason:', response.finishReason);
+    console.log('Raw finish reason:', response.rawFinishReason);
+    if (!response.toolCalls || response.toolCalls.length === 0) {
+        console.log('Full response:', JSON.stringify(response, null, 2));
+        throw new Error('Expected tool calls but got none');
+    }
+    const toolCall = response.toolCalls[0];
+    console.log('Function called:', toolCall.function.name);
+    console.log('Function arguments:', toolCall.function.arguments);
+    // Step 2: Send tool response back
+    const toolResponseRequest = {
+        gate: 'test-gate',
+        model: 'gemini-2.5-flash',
+        type: 'chat',
+        data: {
+            messages: [
+                { role: 'user', content: 'Use the get_current_time function to tell me what time it is.' },
+                {
+                    role: 'assistant',
+                    toolCalls: response.toolCalls,
+                },
+                {
+                    role: 'tool',
+                    toolCallId: toolCall.id,
+                    name: toolCall.function.name,
+                    content: JSON.stringify({ current_time: '2025-12-20T07:30:00Z' }),
+                },
+            ],
+            tools: request.data.tools,
+        },
+    };
+    const finalResponse = await adapter.call(toolResponseRequest);
+    console.log('Final response:', finalResponse.content);
+    console.log('✅ Tool calling test passed\n');
+}
+async function testEmbeddingsMultiple() {
+    console.log('Testing multiple embeddings...');
+    const request = {
+        gate: 'test-gate',
+        model: 'text-embedding-004',
+        type: 'embeddings',
+        data: {
+            input: ['Hello world', 'Goodbye world', 'Testing embeddings'],
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Number of embeddings:', response.embeddings?.length);
+    console.log('First embedding dimensions:', response.embeddings?.[0]?.length);
+    console.log('Latency:', response.latencyMs + 'ms');
+    console.log('✅ Multiple embeddings test passed\n');
+}
+async function testTextToSpeech() {
+    console.log('Testing text-to-speech...');
+    const request = {
+        gate: 'test-gate',
+        model: 'gemini-2.5-flash-preview-tts',
+        type: 'tts',
+        data: {
+            input: 'Hello, this is a test of the Google text-to-speech capabilities.',
+            voice: 'Kore',
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Audio format:', response.audio?.format);
+    console.log('Audio base64 length:', response.audio?.base64?.length);
+    console.log('Latency:', response.latencyMs + 'ms');
+    console.log('✅ Text-to-speech test passed\n');
+}
+async function testVideoGeneration() {
+    console.log('Testing video generation (this may take a few minutes)...');
+    const request = {
+        gate: 'test-gate',
+        model: 'veo-2.0-generate-001',
+        type: 'video',
+        data: {
+            prompt: 'A serene ocean wave rolling onto a sandy beach at sunset',
+            size: '1280x720',
+            duration: 5,
+        },
+    };
+    const response = await adapter.call(request);
+    console.log('Generated videos:', response.videos?.length);
+    console.log('Video URL:', response.videos?.[0]?.url);
+    console.log('Video duration:', response.videos?.[0]?.duration);
+    console.log('Latency:', response.latencyMs + 'ms');
+    console.log('✅ Video generation test passed\n');
+}
+async function runTests() {
+    try {
+        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...');
+        await testTextToSpeech();
+        console.log('Video generation...');
+        await testVideoGeneration();
+        console.log('Tool calling...');
+        await testToolCalling();
+        console.log('✅ All tests passed!');
+    }
+    catch (error) {
+        console.error('❌ Test failed:', error);
+        process.exit(1);
+    }
+}
+runTests();

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,EAAE,MAAM,eAAe,CAAC;AAE7D,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;CACxB,GACA,OAAO,CAAC,YAAY,CAAC,CA4EvB"}
+{"version":3,"file":"task-analysis.d.ts","sourceRoot":"","sources":["../../src/services/task-analysis.ts"],"names":[],"mappings":"AACA,OAAO,EAAkB,YAAY,EAAmC,MAAM,eAAe,CAAC;AAqD9F,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;CACxB,GACA,OAAO,CAAC,YAAY,CAAC,CAqGvB"}

package/dist/services/task-analysis.js

@@ -1,15 +1,76 @@
 import Anthropic from '@anthropic-ai/sdk';
 import { MODEL_REGISTRY } from '@layer-ai/sdk';
+async function detectTaskType(description, anthropic) {
+    const prompt = `Analyze this task description and determine what TYPE of AI task it is.
+
+TASK DESCRIPTION:
+"${description}"
+
+AVAILABLE TASK TYPES:
+- chat: Conversational AI, text generation, Q&A, summarization, translation
+- image: Image generation, image creation
+- video: Video generation, video creation
+- audio: Audio/music generation
+- tts: Text-to-speech, voice synthesis
+- stt: Speech-to-text, audio transcription
+- embeddings: Text embeddings, semantic search
+- document: Document processing, OCR
+- responses: Complex reasoning tasks (o3-pro style models)
+- language-completion: Legacy text completion
+
+Return ONLY the task type as a single word, nothing else.`;
+    try {
+        const response = await anthropic.messages.create({
+            model: 'claude-haiku-4-5-20251001',
+            max_tokens: 50,
+            temperature: 0.0,
+            messages: [{
+                    role: 'user',
+                    content: prompt
+                }]
+        });
+        const responseContent = response.content[0];
+        if (responseContent.type !== 'text') {
+            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'];
+        if (validTypes.includes(detectedType)) {
+            return detectedType;
+        }
+        return 'chat';
+    }
+    catch (error) {
+        console.error('Failed to detect task type:', error);
+        return 'chat';
+    }
+}
 export async function analyzeTask(description, userPreferences) {
     const anthropic = new Anthropic({
         apiKey: process.env.ANTHROPIC_API_KEY
     });
-    const registryContext = JSON.stringify(MODEL_REGISTRY, null, 2);
     const costWeight = userPreferences?.costWeight ?? 0.33;
     const latencyWeight = userPreferences?.latencyWeight ?? 0.33;
     const qualityWeight = userPreferences?.qualityWeight ?? 0.33;
+    let taskType = 'chat';
+    try {
+        taskType = await detectTaskType(description, anthropic);
+    }
+    catch (error) {
+        console.error('Failed to detect task type, defaulting to chat:', error);
+    }
+    const filteredRegistry = {};
+    for (const [key, model] of Object.entries(MODEL_REGISTRY)) {
+        if (model.type === taskType) {
+            filteredRegistry[key] = model;
+        }
+    }
+    const registryContext = JSON.stringify(filteredRegistry, null, 2);
     const prompt = `You are analyzing a task to recommend the best AI models from our registry.
 
+TASK TYPE: ${taskType}
+All models below are specifically for ${taskType} tasks.
+
 MODEL REGISTRY (available models and their capabilities):
 ${registryContext}
 
@@ -61,11 +122,15 @@ Return JSON with:
         if (typeof (mapping) !== 'object' || Array.isArray(mapping)) {
             throw new Error('Mapping is in wrong format');
         }
-        return mapping;
+        return {
+            taskType,
+            ...mapping
+        };
     }
     catch (error) {
         console.error('Failed to find accurate task requirements', error);
         return {
+            taskType,
             primary: 'gpt-4o',
             alternatives: ['claude-sonnet-4-5-20250929', 'gemini-2.5-flash'],
             reasoning: 'Task analysis failed, returning safe defaults'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@layer-ai/core",
-  "version": "0.2.2",
+  "version": "0.3.1",
   "description": "Core API routes and services for Layer AI",
   "type": "module",
   "main": "./dist/index.js",
@@ -35,7 +35,7 @@
     "nanoid": "^5.0.4",
     "openai": "^4.24.0",
     "pg": "^8.11.3",
-    "@layer-ai/sdk": "^0.2.0"
+    "@layer-ai/sdk": "^0.3.1"
   },
   "devDependencies": {
     "@types/express": "^4.17.21",