@x12i/ai-gateway 9.2.0 → 9.3.4

package/dist-cjs/gateway-utils.cjs

@@ -1,78 +1,27 @@
-"use strict";
 /**
  * Gateway Utilities Module
  * Handles utility functions
  */
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DEFAULT_ACTIVITY_FULL_RESPONSE_MAX_CHARS = void 0;
-exports.generateMD5Hash = generateMD5Hash;
-exports.ensureTaskTypeId = ensureTaskTypeId;
-exports.mergeConfig = mergeConfig;
-exports.normalizeRouterUsageTokens = normalizeRouterUsageTokens;
-exports.extractTokenUsageFromRouterResponse = extractTokenUsageFromRouterResponse;
-exports.extractCostUsdFromRouterResponse = extractCostUsdFromRouterResponse;
-exports.hasNonZeroTokenUsage = hasNonZeroTokenUsage;
-exports.resolveActivityCostCompletion = resolveActivityCostCompletion;
-exports.resolveCostCompletionForActivity = resolveCostCompletionForActivity;
-exports.pickInvokeRoutingMetadataSlice = pickInvokeRoutingMetadataSlice;
-exports.pickEffectiveModelConfigForMetadata = pickEffectiveModelConfigForMetadata;
-exports.pickTraceMergedRouterConfig = pickTraceMergedRouterConfig;
-exports.pickEffectiveModelConfigFromInvokeRequest = pickEffectiveModelConfigFromInvokeRequest;
-exports.tryExtractRouterLikePayloadFromErrorChain = tryExtractRouterLikePayloadFromErrorChain;
-exports.pickRequestIdsFromRouterLike = pickRequestIdsFromRouterLike;
-exports.buildInvokeRejectionMetadata = buildInvokeRejectionMetadata;
-exports.attachGatewayInvokeRejectionMetadata = attachGatewayInvokeRejectionMetadata;
-exports.capActivityFullResponsePayload = capActivityFullResponsePayload;
-const crypto = __importStar(require("crypto"));
-const gateway_instructions_js_1 = require("./gateway-instructions.cjs");
-const flex_md_loader_js_1 = require("./flex-md-loader.cjs");
+import * as crypto from 'crypto';
+import { ModelResolutionError } from '@x12i/ai-tools';
+import { getPreParsedInstructions } from './gateway-instructions.js';
+import { getModelMaxTokensFromFlexMd } from './flex-md-loader.js';
+import { applyModelResolution } from './ai-tools-client.js';
+import { getGatewayOperationalMode, isProdGatewayMode, resolveGatewayDefaultModel, warnDefaultModelSubstitution } from './gateway-mode.js';
 /**
  * Generates MD5 hash of a string
  */
-function generateMD5Hash(text) {
+export function generateMD5Hash(text) {
     return crypto.createHash('md5').update(text).digest('hex');
 }
 /**
  * Auto-generates taskTypeId from MD5 hash of pre-parsed instructions if not provided
  */
-async function ensureTaskTypeId(request, logger) {
+export async function ensureTaskTypeId(request, logger) {
     if (request.taskTypeId) {
         return request.taskTypeId;
     }
-    const preParsedInstructions = (0, gateway_instructions_js_1.getPreParsedInstructions)(request.instructions);
+    const preParsedInstructions = getPreParsedInstructions(request.instructions);
     // Generate MD5 hash
     const taskTypeId = generateMD5Hash(preParsedInstructions);
     logger.debug('Auto-generated taskTypeId from instructions MD5 hash', {
@@ -83,11 +32,34 @@ async function ensureTaskTypeId(request, logger) {
     });
     return taskTypeId;
 }
+function applyGatewayDefaultToMerged(merged, defaults, config) {
+    merged.model = defaults.model;
+    if (defaults.provider) {
+        merged.provider = defaults.provider;
+    }
+    else if (!merged.provider) {
+        merged.provider = config.defaultEngine;
+    }
+}
+async function substituteGatewayDefaultModel(merged, request, config, logger, mergeOptions, reason, original) {
+    const operationalMode = getGatewayOperationalMode(config);
+    const defaults = resolveGatewayDefaultModel(mergeOptions?.defaultModelConfig, config.defaultEngine);
+    warnDefaultModelSubstitution(logger, request.identity, {
+        reason,
+        mode: operationalMode,
+        defaultSource: defaults.source,
+        defaultProvider: defaults.provider ?? merged.provider,
+        defaultModel: defaults.model,
+        originalProvider: original?.provider ?? merged.provider,
+        originalModel: original?.model
+    });
+    applyGatewayDefaultToMerged(merged, defaults, config);
+}
 /**
  * Merges config with defaults
  * Supports using internal system action defaults (internalSkill or skillAudit) when useInternalDefaults is set
  */
-async function mergeConfig(request, config, logger) {
+export async function mergeConfig(request, config, logger, mergeOptions) {
     const useInternalDefaults = request.useInternalDefaults;
     const internalDefaults = useInternalDefaults
         ? (useInternalDefaults === 'skill'
@@ -106,8 +78,8 @@ async function mergeConfig(request, config, logger) {
         useInternalDefaults,
         hasInternalDefaults: !!internalDefaults
     });
-    // Default model to "gpt-5-nano" if nothing is provided (most permissive - always works)
-    const defaultModel = 'gpt-5-nano';
+    const operationalMode = getGatewayOperationalMode(config);
+    const resolveModels = config.aiTools?.resolveModels !== false;
     // Priority: modelConfig > request.config > internalSystemActions[useInternalDefaults] > gateway defaults
     // First, merge modelConfig into a config-like object if present
     const modelConfigAsConfig = request.modelConfig ? {
@@ -141,18 +113,67 @@ async function mergeConfig(request, config, logger) {
         ...request.config,
         // ModelConfig overrides (highest priority) - merge only defined values
         ...(modelConfigAsConfig ? Object.fromEntries(Object.entries(modelConfigAsConfig).filter(([_, value]) => value !== undefined)) : {}),
-        // Ensure model is set: modelConfig > request.config > internalDefaults > default
-        model: modelConfigAsConfig?.model || request.config?.model || internalDefaults?.model || defaultModel,
+        // Model resolved below (catalog, default chain, or explicit pass-through)
+        model: modelConfigAsConfig?.model || request.config?.model || internalDefaults?.model,
         // Ensure provider is set: modelConfig > request.config > internalDefaults > gateway default
-        // Provider is required for router to know which provider to use
         provider: modelConfigAsConfig?.provider || request.config?.provider || internalDefaults?.engine || config.defaultEngine
     };
-    // Log if using default model
-    if (!request.config?.model && !internalDefaults?.model) {
-        logger.info('Using default model: gpt-5-nano (no model provided in request)', {
-            jobId: request.identity.jobId,
-            note: 'Default model ensures requests always work regardless of configuration'
-        });
+    const explicitModel = merged.model;
+    const originalProvider = merged.provider;
+    const originalModel = explicitModel;
+    if (!explicitModel) {
+        await substituteGatewayDefaultModel(merged, request, config, logger, mergeOptions, 'no_model_provided');
+    }
+    else if (resolveModels && mergeOptions?.catalog) {
+        try {
+            const resolution = await mergeOptions.catalog.resolveModel({
+                provider: merged.provider,
+                model: explicitModel
+            });
+            if (resolution.found) {
+                applyModelResolution(merged, resolution, config.defaultEngine);
+                request._modelResolution = {
+                    modelId: resolution.modelId,
+                    routedViaOpenRouter: resolution.routedViaOpenRouter,
+                    confidence: resolution.confidence,
+                    resolvedVia: resolution.resolvedVia,
+                    originalProvider,
+                    originalModel
+                };
+                logger.verbose('Catalog resolved model name', {
+                    jobId: request.identity.jobId,
+                    originalModel,
+                    resolvedModelId: resolution.modelId,
+                    provider: merged.provider,
+                    model: merged.model,
+                    confidence: resolution.confidence,
+                    resolvedVia: resolution.resolvedVia
+                });
+            }
+            else if (isProdGatewayMode(operationalMode)) {
+                await substituteGatewayDefaultModel(merged, request, config, logger, mergeOptions, 'model_resolution_failed', { provider: originalProvider, model: originalModel });
+            }
+            else {
+                throw new ModelResolutionError({ provider: merged.provider, model: explicitModel }, resolution);
+            }
+        }
+        catch (error) {
+            if (error instanceof ModelResolutionError) {
+                throw error;
+            }
+            if (isProdGatewayMode(operationalMode)) {
+                await substituteGatewayDefaultModel(merged, request, config, logger, mergeOptions, 'ai_tools_unavailable', { provider: originalProvider, model: originalModel });
+            }
+            else {
+                throw error;
+            }
+        }
+    }
+    else if (resolveModels && !mergeOptions?.catalog && isProdGatewayMode(operationalMode)) {
+        await substituteGatewayDefaultModel(merged, request, config, logger, mergeOptions, 'ai_tools_unavailable', { provider: originalProvider, model: originalModel });
+    }
+    if (!merged.model) {
+        await substituteGatewayDefaultModel(merged, request, config, logger, mergeOptions, 'no_model_provided');
     }
     // Auto-get maxTokens from flex-md if not explicitly set in ANY config source
     // Check all possible sources: request.config, internalDefaults, gateway config
@@ -162,7 +183,7 @@ async function mergeConfig(request, config, logger) {
     if (!maxTokensExplicitlySet && merged.model && merged.provider) {
         // Try to get maxTokens from flex-md
         try {
-            const flexMdMaxTokens = await (0, flex_md_loader_js_1.getModelMaxTokensFromFlexMd)(merged.provider, merged.model);
+            const flexMdMaxTokens = await getModelMaxTokensFromFlexMd(merged.provider, merged.model);
             if (flexMdMaxTokens && flexMdMaxTokens > 0) {
                 merged.maxTokens = flexMdMaxTokens;
                 logger.debug('Using maxTokens from flex-md', {
@@ -252,7 +273,7 @@ function isNonZeroTokenCount(n) {
  * Maps provider/router usage objects to gateway token counts (`metadata.tokens`, Activix, trace attempts).
  * Handles promptTokens/inputTokens, OpenAI-style snake_case, Responses-style input/output tokens, and missing total (sum prompt+completion).
  */
-function normalizeRouterUsageTokens(usage) {
+export function normalizeRouterUsageTokens(usage) {
     if (usage == null || typeof usage !== 'object')
         return undefined;
     const u = usage;
@@ -300,7 +321,7 @@ function firstNonZeroUsageFromBuckets(buckets) {
  * Reads token usage from every stable location the router may populate (see docs/PROVIDERS_ROUTER_DIAGNOSTICS_TRACE_REQUIREMENTS.md).
  * Prefers the raw/provider body (`rawResponse` / `raw`) when it carries non-zero usage before re-reading the outer envelope.
  */
-function extractTokenUsageFromRouterResponse(routerResponse) {
+export function extractTokenUsageFromRouterResponse(routerResponse) {
     const zeros = { prompt: 0, completion: 0, total: 0 };
     if (routerResponse == null || typeof routerResponse !== 'object') {
         return zeros;
@@ -327,7 +348,7 @@ function extractTokenUsageFromRouterResponse(routerResponse) {
  * metadata.attempts[].costUsd, response root, then common raw payload locations.
  * Does not compute cost from tokens — adapters must populate normalized fields or raw usage.cost-style keys.
  */
-function extractCostUsdFromRouterResponse(routerResponse) {
+export function extractCostUsdFromRouterResponse(routerResponse) {
     if (routerResponse == null || typeof routerResponse !== 'object')
         return undefined;
     const r = routerResponse;
@@ -369,7 +390,7 @@ function extractCostUsdFromRouterResponse(routerResponse) {
     }
     return undefined;
 }
-function hasNonZeroTokenUsage(tokens) {
+export function hasNonZeroTokenUsage(tokens) {
     return !!(tokens.prompt || tokens.completion || tokens.total);
 }
 function pickRouterCostStatus(routerResponse) {
@@ -386,7 +407,7 @@ function pickRouterCostStatus(routerResponse) {
  * Gateway fallback when the router does not set `metadata.costStatus`.
  * Prefer {@link resolveCostCompletionForActivity} at invoke boundaries.
  */
-function resolveActivityCostCompletion(tokens, costUsd) {
+export function resolveActivityCostCompletion(tokens, costUsd) {
     if (typeof costUsd === 'number' && Number.isFinite(costUsd)) {
         return { cost: costUsd, costStatus: 'priced' };
     }
@@ -399,7 +420,7 @@ function resolveActivityCostCompletion(tokens, costUsd) {
  * Activity cost slice for Activix: router `metadata.costStatus` / cost wins when present;
  * otherwise gateway applies the G8 fallback (usage + no price → `unpriced`).
  */
-function resolveCostCompletionForActivity(routerResponse, tokens) {
+export function resolveCostCompletionForActivity(routerResponse, tokens) {
     const routerStatus = pickRouterCostStatus(routerResponse);
     const costUsd = extractCostUsdFromRouterResponse(routerResponse);
     if (routerStatus === 'priced') {
@@ -413,11 +434,61 @@ function resolveCostCompletionForActivity(routerResponse, tokens) {
     }
     return resolveActivityCostCompletion(tokens, costUsd);
 }
+/**
+ * Router cost passthrough, then optional @x12i/ai-tools catalog pricing when still unpriced.
+ */
+export async function resolveCostCompletionWithAiTools(routerResponse, tokens, options) {
+    const routerStatus = pickRouterCostStatus(routerResponse);
+    const base = resolveCostCompletionForActivity(routerResponse, tokens);
+    if (base.costStatus === 'priced') {
+        return base;
+    }
+    if (routerStatus === 'unpriced') {
+        return base;
+    }
+    if (options?.calculateCost === false || !options?.calculator) {
+        return base;
+    }
+    if (!hasNonZeroTokenUsage(tokens)) {
+        return base;
+    }
+    const routing = pickInvokeRoutingMetadataSlice(routerResponse, options.mergedConfig);
+    const cfg = options.mergedConfig != null && typeof options.mergedConfig === 'object'
+        ? options.mergedConfig
+        : {};
+    const provider = routing.provider ?? cfg.provider;
+    const modelUsed = routing.modelUsed ?? cfg.model;
+    if (!provider || !modelUsed) {
+        return base;
+    }
+    try {
+        const result = await options.calculator.calculate({
+            tokens: {
+                prompt: tokens.prompt,
+                completion: tokens.completion,
+                total: tokens.total
+            },
+            provider,
+            modelUsed
+        });
+        if (typeof result.cost === 'number' && Number.isFinite(result.cost)) {
+            return {
+                cost: result.cost,
+                costStatus: 'priced',
+                ...(result.breakdown ? { costBreakdown: result.breakdown } : {})
+            };
+        }
+    }
+    catch {
+        // Keep router/gateway unpriced fallback
+    }
+    return base;
+}
 /**
  * Stable routing facts for gateway response metadata (router metadata + merged config fallbacks).
  * Matches trace-mode resolution; intended for every successful invoke(), not only diagnostics.trace.
  */
-function pickInvokeRoutingMetadataSlice(routerResponse, mergedConfig) {
+export function pickInvokeRoutingMetadataSlice(routerResponse, mergedConfig) {
     const rr = routerResponse != null && typeof routerResponse === 'object' ? routerResponse : {};
     const meta = rr.metadata != null && typeof rr.metadata === 'object' ? rr.metadata : {};
     const cfg = mergedConfig != null && typeof mergedConfig === 'object' ? mergedConfig : {};
@@ -443,7 +514,7 @@ function pickInvokeRoutingMetadataSlice(routerResponse, mergedConfig) {
 /**
  * Allowlisted generation profile from merged config for client introspection (no secrets, no arbitrary extras).
  */
-function pickEffectiveModelConfigForMetadata(mergedConfig) {
+export function pickEffectiveModelConfigForMetadata(mergedConfig) {
     if (mergedConfig == null || typeof mergedConfig !== 'object')
         return undefined;
     const c = mergedConfig;
@@ -466,7 +537,7 @@ const TRACE_MERGED_ROUTER_NUMERIC_KEYS = [
 /**
  * Allowlisted snapshot of merged router config for diagnostics trace responses (no arbitrary extras).
  */
-function pickTraceMergedRouterConfig(mergedConfig) {
+export function pickTraceMergedRouterConfig(mergedConfig) {
     if (mergedConfig == null || typeof mergedConfig !== 'object')
         return undefined;
     const c = mergedConfig;
@@ -495,7 +566,7 @@ const EFFECTIVE_MODEL_CONFIG_KEYS = ['model', 'modelId', 'provider', 'temperatur
  * Allowlisted generation fields from request only (before mergeConfig / flex-md).
  * Priority matches mergeConfig: modelConfig overrides request.config per key.
  */
-function pickEffectiveModelConfigFromInvokeRequest(request) {
+export function pickEffectiveModelConfigFromInvokeRequest(request) {
     const cfg = (request.config ?? {});
     const mc = (request.modelConfig ?? {});
     const out = {};
@@ -523,7 +594,7 @@ function isRouterLikeEnvelope(value) {
  * Walk `error`, optional `error.cause`, and common adapter fields (`response`, `routerResponse`, …)
  * to find a router-shaped object for token / correlation extraction.
  */
-function tryExtractRouterLikePayloadFromErrorChain(error, maxDepth = 8) {
+export function tryExtractRouterLikePayloadFromErrorChain(error, maxDepth = 8) {
     const seen = new Set();
     let cur = error;
     for (let i = 0; i < maxDepth && cur != null; i++) {
@@ -544,7 +615,7 @@ function tryExtractRouterLikePayloadFromErrorChain(error, maxDepth = 8) {
     }
     return undefined;
 }
-function pickRequestIdsFromRouterLike(gatewayAiRequestId, routerLike) {
+export function pickRequestIdsFromRouterLike(gatewayAiRequestId, routerLike) {
     if (typeof gatewayAiRequestId !== 'string' || gatewayAiRequestId.length === 0) {
         return undefined;
     }
@@ -570,7 +641,7 @@ function pickRequestIdsFromRouterLike(gatewayAiRequestId, routerLike) {
     }
     return out;
 }
-function buildInvokeRejectionMetadata(args) {
+export function buildInvokeRejectionMetadata(args) {
     const gid = args.gatewayAiRequestId ?? args.request.aiRequestId;
     const partial = args.partialRouterPayload;
     const mc = args.mergedConfig;
@@ -595,16 +666,16 @@ function buildInvokeRejectionMetadata(args) {
         ...(mc === undefined ? { mergeConfigUnavailable: true } : {})
     };
 }
-function attachGatewayInvokeRejectionMetadata(err, metadata) {
+export function attachGatewayInvokeRejectionMetadata(err, metadata) {
     err.metadata = metadata;
 }
 /** Default JSON string length cap for Activix `content.fullResponse` when diagnostics allow storing it. */
-exports.DEFAULT_ACTIVITY_FULL_RESPONSE_MAX_CHARS = 512_000;
+export const DEFAULT_ACTIVITY_FULL_RESPONSE_MAX_CHARS = 512_000;
 /**
  * Size-cap a provider/router payload before storing on an activity record.
  * Non-serializable values become a small marker object instead of throwing.
  */
-function capActivityFullResponsePayload(payload, maxChars = exports.DEFAULT_ACTIVITY_FULL_RESPONSE_MAX_CHARS) {
+export function capActivityFullResponsePayload(payload, maxChars = DEFAULT_ACTIVITY_FULL_RESPONSE_MAX_CHARS) {
     if (payload == null)
         return payload;
     let serialized;

package/dist-cjs/gateway-utils.d.ts

@@ -4,6 +4,7 @@
  */
 import type { AIInvokeRequest, ChatRequest, GatewayConfig, GatewayInvokeRejectionMetadata, GatewayTraceMergedConfig, GatewayTraceRequestIds, ModelConfig } from './types.js';
 import type { Logxer } from '@x12i/logxer';
+import { type AiModelsCatalogClient, type CostCalculator } from '@x12i/ai-tools';
 /**
  * Generates MD5 hash of a string
  */
@@ -12,13 +13,17 @@ export declare function generateMD5Hash(text: string): string;
  * Auto-generates taskTypeId from MD5 hash of pre-parsed instructions if not provided
  */
 export declare function ensureTaskTypeId(request: ChatRequest, logger: Logxer): Promise<string>;
+export type MergeConfigOptions = {
+    defaultModelConfig?: Record<string, unknown>;
+    catalog?: AiModelsCatalogClient | null;
+};
 /**
  * Merges config with defaults
  * Supports using internal system action defaults (internalSkill or skillAudit) when useInternalDefaults is set
  */
 export declare function mergeConfig(request: ChatRequest & {
     useInternalDefaults?: 'skill' | 'audit';
-}, config: GatewayConfig, logger: Logxer): Promise<ChatRequest['config']>;
+}, config: GatewayConfig, logger: Logxer, mergeOptions?: MergeConfigOptions): Promise<ChatRequest['config']>;
 /**
  * Maps provider/router usage objects to gateway token counts (`metadata.tokens`, Activix, trace attempts).
  * Handles promptTokens/inputTokens, OpenAI-style snake_case, Responses-style input/output tokens, and missing total (sum prompt+completion).
@@ -48,6 +53,15 @@ export type ActivityCostStatus = 'priced' | 'unpriced';
 export type ResolvedActivityCost = {
     cost?: number;
     costStatus?: ActivityCostStatus;
+    costBreakdown?: {
+        promptCostUsd: number;
+        completionCostUsd: number;
+        cachingCostUsd?: number;
+        reasoningCostUsd?: number;
+        audioCostUsd?: number;
+        imageCostUsd?: number;
+        requestFlatCostUsd?: number;
+    };
 };
 export declare function hasNonZeroTokenUsage(tokens: {
     prompt: number;
@@ -72,6 +86,19 @@ export declare function resolveCostCompletionForActivity(routerResponse: unknown
     completion: number;
     total: number;
 }): ResolvedActivityCost;
+export type ResolveCostCompletionOptions = {
+    mergedConfig?: unknown;
+    calculator?: CostCalculator | null;
+    calculateCost?: boolean;
+};
+/**
+ * Router cost passthrough, then optional @x12i/ai-tools catalog pricing when still unpriced.
+ */
+export declare function resolveCostCompletionWithAiTools(routerResponse: unknown, tokens: {
+    prompt: number;
+    completion: number;
+    total: number;
+}, options?: ResolveCostCompletionOptions): Promise<ResolvedActivityCost>;
 /**
  * Stable routing facts for gateway response metadata (router metadata + merged config fallbacks).
  * Matches trace-mode resolution; intended for every successful invoke(), not only diagnostics.trace.

package/dist-cjs/gateway-validation.cjs

@@ -1,11 +1,7 @@
-"use strict";
 /**
  * Gateway Validation Module
  * Basic validation for clean proxy implementation
  */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.validateChatRequest = validateChatRequest;
-exports.validateAIRequest = validateAIRequest;
 function validateMandatoryRuntimeIdentity(request) {
     const id = request.identity;
     if (id === undefined || id === null || typeof id !== 'object') {
@@ -16,7 +12,7 @@ function validateMandatoryRuntimeIdentity(request) {
 /**
  * Validates ChatRequest has required fields
  */
-function validateChatRequest(request) {
+export function validateChatRequest(request) {
     if (!request.aiRequestId) {
         throw new Error('aiRequestId is required');
     }
@@ -40,7 +36,7 @@ const GATEWAY_ACTION_TYPES = ['skill', 'preSkill', 'postSkill'];
 /**
  * Validates AIInvokeRequest has required fields
  */
-function validateAIRequest(request) {
+export function validateAIRequest(request) {
     if (!request.aiRequestId) {
         throw new Error('aiRequestId is required for AI requests');
     }