ai.matey.middleware 0.2.0

package/dist/esm/validation.js

@@ -0,0 +1,494 @@
+/**
+ * Input Validation & Sanitization Middleware
+ *
+ * Validates and sanitizes requests to prevent security issues and ensure data quality.
+ *
+ * ## Separation of Concerns
+ *
+ * This middleware focuses on **SECURITY validation**:
+ * - PII detection and redaction
+ * - Prompt injection prevention
+ * - Content moderation
+ * - Message length/token limits
+ * - Sanitization
+ *
+ * For **IR format validation** (structural correctness), use ai.matey.utils/validation.ts:
+ * - Message structure and content validation
+ * - Parameter type and range validation
+ * - Request format validation
+ *
+ * @module
+ */
+import { ValidationError, ErrorCode } from 'ai.matey.errors';
+import { validateIRChatRequest, validateTemperature } from 'ai.matey.utils';
+/**
+ * Helper to create a structured validation error from simple field/value/message
+ * @internal
+ */
+function createValidationError(message, field, value, provenance) {
+    return new ValidationError({
+        code: ErrorCode.INVALID_REQUEST,
+        message,
+        validationDetails: [
+            {
+                field,
+                value,
+                reason: message,
+                expected: 'Valid value',
+            },
+        ],
+        provenance,
+    });
+}
+/**
+ * Default PII patterns
+ */
+export const DEFAULT_PII_PATTERNS = {
+    // Email addresses
+    email: /\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b/g,
+    // US Social Security Numbers
+    ssn: /\b\d{3}-\d{2}-\d{4}\b/g,
+    // Credit card numbers (basic pattern)
+    creditCard: /\b\d{4}[\s-]?\d{4}[\s-]?\d{4}[\s-]?\d{4}\b/g,
+    // US Phone numbers
+    phone: /\b(\+?1[-.\s]?)?\(?\d{3}\)?[-.\s]?\d{3}[-.\s]?\d{4}\b/g,
+    // IP addresses
+    ipAddress: /\b(?:\d{1,3}\.){3}\d{1,3}\b/g,
+    // API keys (common patterns)
+    apiKey: /\b[A-Za-z0-9]{32,}\b/g,
+};
+/**
+ * Default prompt injection patterns
+ */
+export const DEFAULT_INJECTION_PATTERNS = [
+    // Ignore previous instructions
+    /ignore\s+(previous|above|all)\s+(instructions|prompts?|commands?)/i,
+    // System prompt manipulation
+    /system\s*:\s*new\s+(instruction|prompt|role)/i,
+    // Jailbreak attempts
+    /\b(jailbreak|DAN|developer\s+mode)\b/i,
+    // Role manipulation
+    /(you\s+are\s+now|act\s+as\s+if\s+you\s+are)\s+a\s+/i,
+    // Instruction override
+    /disregard\s+(all|any|previous|above)/i,
+];
+/**
+ * Detect PII in text
+ */
+export function detectPII(text, patterns = DEFAULT_PII_PATTERNS) {
+    const matches = [];
+    const types = [];
+    for (const [type, pattern] of Object.entries(patterns)) {
+        const found = text.match(pattern);
+        if (found && found.length > 0) {
+            types.push(type);
+            for (const value of found) {
+                matches.push({ type, value });
+            }
+        }
+    }
+    return {
+        detected: matches.length > 0,
+        types: Array.from(new Set(types)),
+        matches,
+    };
+}
+/**
+ * Redact PII from text
+ */
+export function redactPII(text, patterns = DEFAULT_PII_PATTERNS) {
+    let redacted = text;
+    for (const [type, pattern] of Object.entries(patterns)) {
+        redacted = redacted.replace(pattern, `[REDACTED_${type.toUpperCase()}]`);
+    }
+    return redacted;
+}
+/**
+ * Detect prompt injection attempts
+ */
+export function detectPromptInjection(text, patterns = DEFAULT_INJECTION_PATTERNS) {
+    return patterns.some((pattern) => pattern.test(text));
+}
+/**
+ * Sanitize text content
+ */
+export function sanitizeText(text) {
+    // Remove null bytes
+    let sanitized = text.replace(/\0/g, '');
+    // Normalize whitespace (but preserve intentional formatting)
+    sanitized = sanitized.replace(/\r\n/g, '\n');
+    // Remove invisible characters (zero-width, etc.)
+    sanitized = sanitized.replace(/[\u200B-\u200D\uFEFF]/g, '');
+    return sanitized;
+}
+/**
+ * Estimate token count (rough approximation)
+ */
+function estimateTokens(text) {
+    // Rough estimate: ~4 characters per token
+    return Math.ceil(text.length / 4);
+}
+/**
+ * Extract text from message content
+ */
+function extractText(content) {
+    if (typeof content === 'string') {
+        return content;
+    }
+    return content
+        .filter((c) => c.type === 'text')
+        .map((c) => c.text)
+        .join('\n');
+}
+/**
+ * Validate request
+ */
+export async function validateRequest(request, config) {
+    const errors = [];
+    const warnings = [];
+    // Perform IR format validation first if enabled
+    if (config.validateIRFormat) {
+        try {
+            validateIRChatRequest(request);
+        }
+        catch (error) {
+            if (error instanceof ValidationError) {
+                errors.push(error);
+                // If format validation fails, return early as security checks may not make sense
+                return { valid: false, errors, warnings };
+            }
+            throw error;
+        }
+    }
+    // Validate message count
+    if (config.maxMessages && request.messages.length > config.maxMessages) {
+        errors.push(createValidationError(`Too many messages: ${request.messages.length} > ${config.maxMessages}`, 'messages', request.messages.length));
+    }
+    // Validate messages
+    let totalTokens = 0;
+    const messagesArray = Array.from(request.messages);
+    for (let i = 0; i < messagesArray.length; i++) {
+        const message = messagesArray[i];
+        if (!message) {
+            continue;
+        }
+        // Check allowed roles
+        if (config.allowedRoles &&
+            !config.allowedRoles.includes(message.role)) {
+            errors.push(createValidationError(`Invalid message role: ${message.role}`, `messages[${i}].role`, message.role));
+        }
+        // Extract text
+        const text = extractText(message.content);
+        // Check empty messages
+        if (config.blockEmptyMessages !== false && text.trim().length === 0) {
+            errors.push(createValidationError(`Empty message at index ${i}`, `messages[${i}].content`, text));
+        }
+        // Check message length
+        if (config.maxMessageLength && text.length > config.maxMessageLength) {
+            errors.push(createValidationError(`Message too long: ${text.length} > ${config.maxMessageLength}`, `messages[${i}].content`, text.length));
+        }
+        // Estimate tokens
+        const tokens = estimateTokens(text);
+        totalTokens += tokens;
+        // Check tokens per message
+        if (config.maxTokensPerMessage && tokens > config.maxTokensPerMessage) {
+            errors.push(createValidationError(`Message tokens exceed limit: ${tokens} > ${config.maxTokensPerMessage}`, `messages[${i}].content`, tokens));
+        }
+        // Detect PII
+        if (config.detectPII) {
+            const piiResult = config.piiDetector
+                ? await config.piiDetector(text)
+                : detectPII(text, config.piiPatterns);
+            if (piiResult.detected) {
+                const message = `PII detected in message ${i}: ${piiResult.types.join(', ')}`;
+                if (config.piiAction === 'block') {
+                    errors.push(createValidationError(message, `messages[${i}].content`, piiResult));
+                }
+                else if (config.piiAction === 'warn' || config.piiAction === 'log') {
+                    warnings.push(message);
+                }
+            }
+        }
+        // Detect prompt injection
+        if (config.preventPromptInjection !== false) {
+            const hasInjection = detectPromptInjection(text, config.injectionPatterns);
+            if (hasInjection) {
+                errors.push(createValidationError(`Potential prompt injection detected in message ${i}`, `messages[${i}].content`, text));
+            }
+        }
+        // Content moderation
+        if (config.moderationCallback) {
+            const modResult = await config.moderationCallback(text);
+            if (modResult.flagged) {
+                const message = `Content flagged by moderation in message ${i}: ${modResult.categories.join(', ')}`;
+                if (config.blockFlaggedContent) {
+                    errors.push(createValidationError(message, `messages[${i}].content`, modResult));
+                }
+                else {
+                    warnings.push(message);
+                }
+            }
+        }
+    }
+    // Check total tokens
+    if (config.maxTotalTokens && totalTokens > config.maxTotalTokens) {
+        errors.push(createValidationError(`Total tokens exceed limit: ${totalTokens} > ${config.maxTotalTokens}`, 'messages', totalTokens));
+    }
+    // Validate model
+    if (config.validateModel && request.parameters?.model) {
+        if (config.allowedModels && !config.allowedModels.includes(request.parameters.model)) {
+            errors.push(createValidationError(`Model not allowed: ${request.parameters.model}`, 'parameters.model', request.parameters.model));
+        }
+    }
+    // Validate temperature (deprecated - use validateIRFormat instead)
+    if (config.validateTemperature && request.parameters?.temperature !== undefined) {
+        try {
+            validateTemperature(request.parameters.temperature);
+        }
+        catch (error) {
+            if (error instanceof ValidationError) {
+                errors.push(error);
+            }
+            else {
+                throw error;
+            }
+        }
+    }
+    // Custom validation
+    if (config.customValidator) {
+        const customErrors = await config.customValidator(request);
+        errors.push(...customErrors);
+    }
+    return {
+        valid: errors.length === 0,
+        errors,
+        warnings,
+    };
+}
+/**
+ * Sanitize request
+ */
+export function sanitizeRequest(request, config) {
+    if (config.sanitizeMessages === false) {
+        return request;
+    }
+    const sanitizer = config.sanitizer || sanitizeText;
+    // Sanitize messages
+    const sanitizedMessages = request.messages.map((message) => {
+        if (typeof message.content === 'string') {
+            return {
+                ...message,
+                content: sanitizer(message.content),
+            };
+        }
+        return {
+            ...message,
+            content: message.content.map((content) => {
+                if (content.type === 'text') {
+                    return {
+                        ...content,
+                        text: sanitizer(content.text),
+                    };
+                }
+                return content;
+            }),
+        };
+    });
+    // Redact PII if configured
+    if (config.detectPII && config.piiAction === 'redact') {
+        const patterns = config.piiPatterns || DEFAULT_PII_PATTERNS;
+        return {
+            ...request,
+            messages: sanitizedMessages.map((message) => {
+                if (typeof message.content === 'string') {
+                    return {
+                        ...message,
+                        content: redactPII(message.content, patterns),
+                    };
+                }
+                return {
+                    ...message,
+                    content: message.content.map((content) => {
+                        if (content.type === 'text') {
+                            return {
+                                ...content,
+                                text: redactPII(content.text, patterns),
+                            };
+                        }
+                        return content;
+                    }),
+                };
+            }),
+        };
+    }
+    return {
+        ...request,
+        messages: sanitizedMessages,
+    };
+}
+/**
+ * Create input validation middleware
+ *
+ * Validates and sanitizes requests to prevent security issues and ensure data quality.
+ *
+ * @param config - Validation configuration
+ * @returns Middleware function
+ *
+ * @example Basic Usage
+ * ```typescript
+ * import { createValidationMiddleware } from 'ai.matey';
+ *
+ * const validation = createValidationMiddleware({
+ *   maxMessages: 100,
+ *   maxTotalTokens: 128000,
+ *   preventPromptInjection: true,
+ * });
+ *
+ * bridge.use(validation);
+ * ```
+ *
+ * @example PII Detection & Redaction
+ * ```typescript
+ * const validation = createValidationMiddleware({
+ *   detectPII: true,
+ *   piiAction: 'redact', // or 'block', 'warn', 'log'
+ *   piiPatterns: {
+ *     email: /\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b/g,
+ *     ssn: /\b\d{3}-\d{2}-\d{4}\b/g,
+ *   },
+ * });
+ * ```
+ *
+ * @example Content Moderation
+ * ```typescript
+ * const validation = createValidationMiddleware({
+ *   moderationCallback: async (content) => {
+ *     // Call external moderation API
+ *     const result = await moderationAPI.check(content);
+ *     return {
+ *       flagged: result.flagged,
+ *       categories: result.categories,
+ *       scores: result.scores,
+ *     };
+ *   },
+ *   blockFlaggedContent: true,
+ * });
+ * ```
+ *
+ * @example Custom Validation
+ * ```typescript
+ * const validation = createValidationMiddleware({
+ *   customValidator: async (request) => {
+ *     const errors: ValidationError[] = [];
+ *
+ *     // Custom business logic
+ *     if (request.messages.some(m => m.content.includes('forbidden'))) {
+ *       errors.push(new ValidationError(
+ *         'Forbidden content detected',
+ *         'messages',
+ *         'forbidden'
+ *       ));
+ *     }
+ *
+ *     return errors;
+ *   },
+ * });
+ * ```
+ *
+ * @example Production Configuration
+ * ```typescript
+ * const validation = createValidationMiddleware({
+ *   maxMessages: 100,
+ *   maxTotalTokens: 128000,
+ *   maxTokensPerMessage: 32000,
+ *   maxMessageLength: 100000,
+ *   blockEmptyMessages: true,
+ *   detectPII: true,
+ *   piiAction: 'redact',
+ *   preventPromptInjection: true,
+ *   sanitizeMessages: true,
+ *   validateModel: true,
+ *   allowedModels: ['gpt-4', 'claude-3-sonnet', 'gemini-pro'],
+ *   validateTemperature: true,
+ *   temperatureRange: [0, 2],
+ *   throwOnError: true,
+ *   logWarnings: true,
+ * });
+ * ```
+ */
+export function createValidationMiddleware(config = {}) {
+    return async (context, next) => {
+        // Validate request
+        const validationResult = await validateRequest(context.request, config);
+        // Log warnings
+        if (config.logWarnings !== false && validationResult.warnings.length > 0) {
+            for (const warning of validationResult.warnings) {
+                console.warn(`[Validation] ${warning}`);
+            }
+        }
+        // Handle validation errors
+        if (!validationResult.valid) {
+            const errorMessage = validationResult.errors.map((e) => e.message).join('; ');
+            if (config.throwOnError !== false) {
+                throw createValidationError(`Validation failed: ${errorMessage}`, 'request', validationResult.errors);
+            }
+            // Log errors but continue
+            console.error(`[Validation] Errors: ${errorMessage}`);
+        }
+        // Sanitize request
+        context.request = sanitizeRequest(context.request, config);
+        // Continue to next middleware
+        return await next();
+    };
+}
+/**
+ * Create production-ready validation middleware with strict settings
+ *
+ * @returns Middleware with production validation settings
+ *
+ * @example
+ * ```typescript
+ * import { createProductionValidationMiddleware } from 'ai.matey';
+ *
+ * bridge.use(createProductionValidationMiddleware());
+ * ```
+ */
+export function createProductionValidationMiddleware() {
+    return createValidationMiddleware({
+        maxMessages: 100,
+        maxTotalTokens: 128000,
+        maxTokensPerMessage: 32000,
+        maxMessageLength: 100000,
+        blockEmptyMessages: true,
+        detectPII: true,
+        piiAction: 'redact',
+        preventPromptInjection: true,
+        sanitizeMessages: true,
+        throwOnError: true,
+        logWarnings: true,
+    });
+}
+/**
+ * Create development-friendly validation middleware with relaxed settings
+ *
+ * @returns Middleware with development validation settings
+ *
+ * @example
+ * ```typescript
+ * import { createDevelopmentValidationMiddleware } from 'ai.matey';
+ *
+ * bridge.use(createDevelopmentValidationMiddleware());
+ * ```
+ */
+export function createDevelopmentValidationMiddleware() {
+    return createValidationMiddleware({
+        maxMessages: 1000,
+        blockEmptyMessages: false,
+        detectPII: false,
+        preventPromptInjection: false,
+        sanitizeMessages: true,
+        throwOnError: false,
+        logWarnings: true,
+    });
+}
+//# sourceMappingURL=validation.js.map

package/dist/esm/validation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../src/validation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;GAoBG;AAIH,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE7D,OAAO,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAE5E;;;GAGG;AACH,SAAS,qBAAqB,CAC5B,OAAe,EACf,KAAa,EACb,KAAe,EACf,UAA4B;IAE5B,OAAO,IAAI,eAAe,CAAC;QACzB,IAAI,EAAE,SAAS,CAAC,eAAe;QAC/B,OAAO;QACP,iBAAiB,EAAE;YACjB;gBACE,KAAK;gBACL,KAAK;gBACL,MAAM,EAAE,OAAO;gBACf,QAAQ,EAAE,aAAa;aACxB;SACF;QACD,UAAU;KACX,CAAC,CAAC;AACL,CAAC;AAuND;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAA2B;IAC1D,kBAAkB;IAClB,KAAK,EAAE,sDAAsD;IAE7D,6BAA6B;IAC7B,GAAG,EAAE,wBAAwB;IAE7B,sCAAsC;IACtC,UAAU,EAAE,6CAA6C;IAEzD,mBAAmB;IACnB,KAAK,EAAE,wDAAwD;IAE/D,eAAe;IACf,SAAS,EAAE,8BAA8B;IAEzC,6BAA6B;IAC7B,MAAM,EAAE,uBAAuB;CAChC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAa;IAClD,+BAA+B;IAC/B,oEAAoE;IAEpE,6BAA6B;IAC7B,+CAA+C;IAE/C,qBAAqB;IACrB,uCAAuC;IAEvC,oBAAoB;IACpB,qDAAqD;IAErD,uBAAuB;IACvB,uCAAuC;CACxC,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,SAAS,CACvB,IAAY,EACZ,WAAmC,oBAAoB;IAEvD,MAAM,OAAO,GAA2C,EAAE,CAAC;IAC3D,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,KAAK,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC;QAC5B,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,SAAS,CACvB,IAAY,EACZ,WAAmC,oBAAoB;IAEvD,IAAI,QAAQ,GAAG,IAAI,CAAC;IAEpB,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvD,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,aAAa,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;IAC3E,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,IAAY,EACZ,WAAqB,0BAA0B;IAE/C,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AACxD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,IAAY;IACvC,oBAAoB;IACpB,IAAI,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAExC,6DAA6D;IAC7D,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAE7C,iDAAiD;IACjD,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;IAE5D,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY;IAClC,0CAA0C;IAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,OAA2C;IAC9D,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAsB,CAAC,IAAI,CAAC;SACxC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,OAAsB,EACtB,MAAwB;IAExB,MAAM,MAAM,GAAsB,EAAE,CAAC;IACrC,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,gDAAgD;IAChD,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,qBAAqB,CAAC,OAAO,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,iFAAiF;gBACjF,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;YAC5C,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,IAAI,MAAM,CAAC,WAAW,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QACvE,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,sBAAsB,OAAO,CAAC,QAAQ,CAAC,MAAM,MAAM,MAAM,CAAC,WAAW,EAAE,EACvE,UAAU,EACV,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CACF,CAAC;IACJ,CAAC;IAED,oBAAoB;IACpB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,SAAS;QACX,CAAC;QAED,sBAAsB;QACtB,IACE,MAAM,CAAC,YAAY;YACnB,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAuC,CAAC,EAC9E,CAAC;YACD,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,yBAAyB,OAAO,CAAC,IAAI,EAAE,EACvC,YAAY,CAAC,QAAQ,EACrB,OAAO,CAAC,IAAI,CACb,CACF,CAAC;QACJ,CAAC;QAED,eAAe;QACf,MAAM,IAAI,GAAG,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAE1C,uBAAuB;QACvB,IAAI,MAAM,CAAC,kBAAkB,KAAK,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpE,MAAM,CAAC,IAAI,CACT,qBAAqB,CAAC,0BAA0B,CAAC,EAAE,EAAE,YAAY,CAAC,WAAW,EAAE,IAAI,CAAC,CACrF,CAAC;QACJ,CAAC;QAED,uBAAuB;QACvB,IAAI,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;YACrE,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,qBAAqB,IAAI,CAAC,MAAM,MAAM,MAAM,CAAC,gBAAgB,EAAE,EAC/D,YAAY,CAAC,WAAW,EACxB,IAAI,CAAC,MAAM,CACZ,CACF,CAAC;QACJ,CAAC;QAED,kBAAkB;QAClB,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACpC,WAAW,IAAI,MAAM,CAAC;QAEtB,2BAA2B;QAC3B,IAAI,MAAM,CAAC,mBAAmB,IAAI,MAAM,GAAG,MAAM,CAAC,mBAAmB,EAAE,CAAC;YACtE,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,gCAAgC,MAAM,MAAM,MAAM,CAAC,mBAAmB,EAAE,EACxE,YAAY,CAAC,WAAW,EACxB,MAAM,CACP,CACF,CAAC;QACJ,CAAC;QAED,aAAa;QACb,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW;gBAClC,CAAC,CAAC,MAAM,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC;gBAChC,CAAC,CAAC,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;YAExC,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;gBACvB,MAAM,OAAO,GAAG,2BAA2B,CAAC,KAAK,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAE9E,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;oBACjC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,YAAY,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;gBACnF,CAAC;qBAAM,IAAI,MAAM,CAAC,SAAS,KAAK,MAAM,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;oBACrE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,MAAM,CAAC,sBAAsB,KAAK,KAAK,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,qBAAqB,CAAC,IAAI,EAAE,MAAM,CAAC,iBAAiB,CAAC,CAAC;YAE3E,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,kDAAkD,CAAC,EAAE,EACrD,YAAY,CAAC,WAAW,EACxB,IAAI,CACL,CACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,qBAAqB;QACrB,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC9B,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;YAExD,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;gBACtB,MAAM,OAAO,GAAG,4CAA4C,CAAC,KAAK,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAEpG,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;oBAC/B,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,YAAY,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;gBACnF,CAAC;qBAAM,CAAC;oBACN,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,qBAAqB;IACrB,IAAI,MAAM,CAAC,cAAc,IAAI,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;QACjE,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,8BAA8B,WAAW,MAAM,MAAM,CAAC,cAAc,EAAE,EACtE,UAAU,EACV,WAAW,CACZ,CACF,CAAC;IACJ,CAAC;IAED,iBAAiB;IACjB,IAAI,MAAM,CAAC,aAAa,IAAI,OAAO,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC;QACtD,IAAI,MAAM,CAAC,aAAa,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACrF,MAAM,CAAC,IAAI,CACT,qBAAqB,CACnB,sBAAsB,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,EAChD,kBAAkB,EAClB,OAAO,CAAC,UAAU,CAAC,KAAK,CACzB,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,IAAI,MAAM,CAAC,mBAAmB,IAAI,OAAO,CAAC,UAAU,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;QAChF,IAAI,CAAC;YACH,mBAAmB,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,oBAAoB;IACpB,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC3D,MAAM,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;IAC/B,CAAC;IAED,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,MAAM,KAAK,CAAC;QAC1B,MAAM;QACN,QAAQ;KACT,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,OAAsB,EAAE,MAAwB;IAC9E,IAAI,MAAM,CAAC,gBAAgB,KAAK,KAAK,EAAE,CAAC;QACtC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,YAAY,CAAC;IAEnD,oBAAoB;IACpB,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QACzD,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO;gBACL,GAAG,OAAO;gBACV,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;aACpC,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,OAAO;YACV,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBACvC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC5B,OAAO;wBACL,GAAG,OAAO;wBACV,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC;qBAC9B,CAAC;gBACJ,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;SACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,2BAA2B;IAC3B,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;QACtD,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,IAAI,oBAAoB,CAAC;QAE5D,OAAO;YACL,GAAG,OAAO;YACV,QAAQ,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC1C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACxC,OAAO;wBACL,GAAG,OAAO;wBACV,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC;qBAC9C,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,GAAG,OAAO;oBACV,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;wBACvC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BAC5B,OAAO;gCACL,GAAG,OAAO;gCACV,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC;6BACxC,CAAC;wBACJ,CAAC;wBACD,OAAO,OAAO,CAAC;oBACjB,CAAC,CAAC;iBACH,CAAC;YACJ,CAAC,CAAC;SACH,CAAC;IACJ,CAAC;IAED,OAAO;QACL,GAAG,OAAO;QACV,QAAQ,EAAE,iBAAiB;KAC5B,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyFG;AACH,MAAM,UAAU,0BAA0B,CAAC,SAA2B,EAAE;IACtE,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;QAC7B,mBAAmB;QACnB,MAAM,gBAAgB,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAExE,eAAe;QACf,IAAI,MAAM,CAAC,WAAW,KAAK,KAAK,IAAI,gBAAgB,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzE,KAAK,MAAM,OAAO,IAAI,gBAAgB,CAAC,QAAQ,EAAE,CAAC;gBAChD,OAAO,CAAC,IAAI,CAAC,gBAAgB,OAAO,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,2BAA2B;QAC3B,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;YAC5B,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE9E,IAAI,MAAM,CAAC,YAAY,KAAK,KAAK,EAAE,CAAC;gBAClC,MAAM,qBAAqB,CACzB,sBAAsB,YAAY,EAAE,EACpC,SAAS,EACT,gBAAgB,CAAC,MAAM,CACxB,CAAC;YACJ,CAAC;YAED,0BAA0B;YAC1B,OAAO,CAAC,KAAK,CAAC,wBAAwB,YAAY,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,mBAAmB;QACnB,OAAO,CAAC,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE3D,8BAA8B;QAC9B,OAAO,MAAM,IAAI,EAAE,CAAC;IACtB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,oCAAoC;IAClD,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,GAAG;QAChB,cAAc,EAAE,MAAM;QACtB,mBAAmB,EAAE,KAAK;QAC1B,gBAAgB,EAAE,MAAM;QACxB,kBAAkB,EAAE,IAAI;QACxB,SAAS,EAAE,IAAI;QACf,SAAS,EAAE,QAAQ;QACnB,sBAAsB,EAAE,IAAI;QAC5B,gBAAgB,EAAE,IAAI;QACtB,YAAY,EAAE,IAAI;QAClB,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,qCAAqC;IACnD,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,IAAI;QACjB,kBAAkB,EAAE,KAAK;QACzB,SAAS,EAAE,KAAK;QAChB,sBAAsB,EAAE,KAAK;QAC7B,gBAAgB,EAAE,IAAI;QACtB,YAAY,EAAE,KAAK;QACnB,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;AACL,CAAC"}

package/dist/types/caching.d.ts

@@ -0,0 +1,98 @@
+/**
+ * Caching Middleware
+ *
+ * Caches responses with TTL-based expiration and LRU eviction.
+ *
+ * @module
+ */
+import type { Middleware, CacheStorage } from 'ai.matey.types';
+import type { IRChatRequest, IRChatResponse } from 'ai.matey.types';
+/**
+ * Configuration for caching middleware.
+ */
+export interface CachingConfig {
+    /**
+     * Cache key generator function.
+     * @default Default implementation based on request hash
+     */
+    keyGenerator?: (request: IRChatRequest) => string;
+    /**
+     * Cache TTL in milliseconds.
+     * @default 3600000 (1 hour)
+     */
+    ttl?: number;
+    /**
+     * Maximum cache size (number of entries).
+     * @default 1000
+     */
+    maxSize?: number;
+    /**
+     * Cache storage implementation.
+     * @default InMemoryCacheStorage
+     */
+    storage?: CacheStorage;
+    /**
+     * Whether to cache streaming responses.
+     * @default false
+     */
+    cacheStreaming?: boolean;
+}
+/**
+ * In-memory cache storage with LRU eviction.
+ */
+export declare class InMemoryCacheStorage implements CacheStorage {
+    private cache;
+    private accessOrder;
+    private maxSize;
+    constructor(maxSize?: number);
+    get(key: string): Promise<IRChatResponse | undefined>;
+    set(key: string, value: IRChatResponse, ttl?: number): Promise<void>;
+    has(key: string): Promise<boolean>;
+    delete(key: string): Promise<boolean>;
+    clear(): Promise<void>;
+    /**
+     * Get cache statistics.
+     */
+    getStats(): {
+        size: number;
+        maxSize: number;
+        hitRate?: number;
+    };
+    /**
+     * Update LRU access order.
+     */
+    private updateAccessOrder;
+    /**
+     * Remove key from access order.
+     */
+    private removeFromAccessOrder;
+    /**
+     * Evict least recently used entry.
+     */
+    private evictLRU;
+    /**
+     * Clean up expired entries.
+     */
+    cleanup(): void;
+}
+/**
+ * Create caching middleware.
+ *
+ * Caches responses with TTL-based expiration and LRU eviction.
+ *
+ * @param config Caching configuration
+ * @returns Caching middleware
+ *
+ * @example
+ * ```typescript
+ * const caching = createCachingMiddleware({
+ *   ttl: 3600000, // 1 hour
+ *   maxSize: 1000,
+ *   cacheStreaming: false
+ * });
+ *
+ * bridge.use(caching);
+ * ```
+ */
+export declare function createCachingMiddleware(config?: CachingConfig): Middleware;
+//# sourceMappingURL=caching.d.ts.map

package/dist/types/caching.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"caching.d.ts","sourceRoot":"","sources":["../../src/caching.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAqC,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAClG,OAAO,KAAK,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAOpE;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;;OAGG;IACH,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,MAAM,CAAC;IAElD;;;OAGG;IACH,GAAG,CAAC,EAAE,MAAM,CAAC;IAEb;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,OAAO,CAAC,EAAE,YAAY,CAAC;IAEvB;;;OAGG;IACH,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAwCD;;GAEG;AACH,qBAAa,oBAAqB,YAAW,YAAY;IACvD,OAAO,CAAC,KAAK,CAAiC;IAC9C,OAAO,CAAC,WAAW,CAAgB;IACnC,OAAO,CAAC,OAAO,CAAS;gBAEZ,OAAO,GAAE,MAAa;IAIlC,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IAoBrD,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,GAAE,MAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAkB7E,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAiBlC,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKrC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAMtB;;OAEG;IACH,QAAQ,IAAI;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE;IAO/D;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAQzB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAO7B;;OAEG;IACH,OAAO,CAAC,QAAQ;IAchB;;OAEG;IACH,OAAO,IAAI,IAAI;CAgBhB;AAMD;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAgB,uBAAuB,CAAC,MAAM,GAAE,aAAkB,GAAG,UAAU,CAuD9E"}