ai.matey.middleware 0.2.0

package/dist/cjs/retry.js

@@ -0,0 +1,205 @@
+"use strict";
+/**
+ * Retry Middleware
+ *
+ * Retries failed requests with exponential backoff.
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createRetryMiddleware = createRetryMiddleware;
+exports.isRateLimitError = isRateLimitError;
+exports.isNetworkError = isNetworkError;
+exports.isServerError = isServerError;
+exports.createRetryPredicate = createRetryPredicate;
+const ai_matey_errors_1 = require("ai.matey.errors");
+// ============================================================================
+// Retry Logic
+// ============================================================================
+/**
+ * Default retry condition.
+ *
+ * Only retry transient errors (network issues, rate limits, server errors).
+ * Note: maxAttempts is enforced by the middleware loop, not this function.
+ */
+function defaultShouldRetry(error, _attempt) {
+    // Check if error has isRetryable property
+    if (error && typeof error === 'object' && 'isRetryable' in error) {
+        return error.isRetryable === true;
+    }
+    // For AdapterError, check isRetryable
+    if (error instanceof ai_matey_errors_1.AdapterError) {
+        return error.isRetryable;
+    }
+    // Default: don't retry unknown errors
+    return false;
+}
+/**
+ * Calculate retry delay with exponential backoff.
+ */
+function calculateDelay(attempt, initialDelay, backoffMultiplier, maxDelay, useJitter) {
+    // Calculate exponential backoff
+    let delay = initialDelay * Math.pow(backoffMultiplier, attempt);
+    // Cap at max delay
+    delay = Math.min(delay, maxDelay);
+    // Add jitter if enabled (random value between 0 and delay)
+    if (useJitter) {
+        delay = Math.random() * delay;
+    }
+    return Math.floor(delay);
+}
+/**
+ * Sleep for specified duration.
+ */
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+// ============================================================================
+// Middleware Factory
+// ============================================================================
+/**
+ * Create retry middleware.
+ *
+ * Retries failed requests with exponential backoff.
+ *
+ * @param config Retry configuration
+ * @returns Retry middleware
+ *
+ * @example
+ * ```typescript
+ * const retry = createRetryMiddleware({
+ *   maxAttempts: 3,
+ *   initialDelay: 1000,
+ *   backoffMultiplier: 2,
+ *   maxDelay: 30000
+ * });
+ *
+ * bridge.use(retry);
+ * ```
+ */
+function createRetryMiddleware(config = {}) {
+    const { maxAttempts = 3, initialDelay = 1000, backoffMultiplier = 2, maxDelay = 30000, useJitter = true, shouldRetry = defaultShouldRetry, onRetry, } = config;
+    return async (context, next) => {
+        let lastError;
+        let attempt = 0;
+        while (attempt < maxAttempts) {
+            try {
+                // Call next middleware/handler
+                const response = await next();
+                // Success - add retry metadata if we retried
+                if (attempt > 0) {
+                    return {
+                        ...response,
+                        metadata: {
+                            ...response.metadata,
+                            custom: {
+                                ...response.metadata.custom,
+                                retryAttempts: attempt,
+                                retrySuccess: true,
+                            },
+                        },
+                    };
+                }
+                return response;
+            }
+            catch (error) {
+                lastError = error;
+                attempt++;
+                // Check if we should retry
+                const willRetry = attempt < maxAttempts && shouldRetry(error, attempt);
+                if (!willRetry) {
+                    // No more retries - add metadata and throw
+                    if (error instanceof ai_matey_errors_1.AdapterError) {
+                        throw new ai_matey_errors_1.AdapterError({
+                            ...error,
+                            details: {
+                                ...error.details,
+                                retryAttempts: attempt,
+                                retrySuccess: false,
+                            },
+                        });
+                    }
+                    throw error;
+                }
+                // Calculate retry delay
+                const delay = calculateDelay(attempt - 1, // 0-indexed for delay calculation
+                initialDelay, backoffMultiplier, maxDelay, useJitter);
+                // Call retry callback if provided
+                if (onRetry) {
+                    onRetry(error, attempt, delay);
+                }
+                // Check if request was aborted
+                if (context.signal?.aborted) {
+                    throw error;
+                }
+                // Wait before retrying
+                await sleep(delay);
+                // Check again if request was aborted during sleep
+                if (context.signal?.aborted) {
+                    throw error;
+                }
+            }
+        }
+        // Should never reach here, but just in case
+        throw lastError;
+    };
+}
+// ============================================================================
+// Retry Utilities
+// ============================================================================
+/**
+ * Check if an error is a rate limit error.
+ */
+function isRateLimitError(error) {
+    if (error instanceof ai_matey_errors_1.AdapterError) {
+        return error.code === 'RATE_LIMIT_EXCEEDED';
+    }
+    return false;
+}
+/**
+ * Check if an error is a network error.
+ */
+function isNetworkError(error) {
+    if (error instanceof ai_matey_errors_1.AdapterError) {
+        return error.code === 'NETWORK_ERROR';
+    }
+    return false;
+}
+/**
+ * Check if an error is a server error (5xx).
+ */
+function isServerError(error) {
+    if (error instanceof ai_matey_errors_1.AdapterError) {
+        return error.code === 'PROVIDER_ERROR' || error.code === 'INTERNAL_ERROR';
+    }
+    return false;
+}
+/**
+ * Create a retry predicate that only retries specific error types.
+ * Note: maxAttempts is enforced by the middleware loop, not this function.
+ */
+function createRetryPredicate(errorTypes) {
+    return (error, _attempt) => {
+        for (const type of errorTypes) {
+            switch (type) {
+                case 'rate_limit':
+                    if (isRateLimitError(error)) {
+                        return true;
+                    }
+                    break;
+                case 'network':
+                    if (isNetworkError(error)) {
+                        return true;
+                    }
+                    break;
+                case 'server':
+                    if (isServerError(error)) {
+                        return true;
+                    }
+                    break;
+            }
+        }
+        return false;
+    };
+}
+//# sourceMappingURL=retry.js.map

package/dist/cjs/retry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/retry.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;AAwIH,sDA2FC;AASD,4CAMC;AAKD,wCAMC;AAKD,sCAMC;AAMD,oDA0BC;AApSD,qDAA+C;AAoD/C,+EAA+E;AAC/E,cAAc;AACd,+EAA+E;AAE/E;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,KAAc,EAAE,QAAgB;IAC1D,0CAA0C;IAC1C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,aAAa,IAAI,KAAK,EAAE,CAAC;QACjE,OAAQ,KAAkC,CAAC,WAAW,KAAK,IAAI,CAAC;IAClE,CAAC;IAED,sCAAsC;IACtC,IAAI,KAAK,YAAY,8BAAY,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC,WAAW,CAAC;IAC3B,CAAC;IAED,sCAAsC;IACtC,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CACrB,OAAe,EACf,YAAoB,EACpB,iBAAyB,EACzB,QAAgB,EAChB,SAAkB;IAElB,gCAAgC;IAChC,IAAI,KAAK,GAAG,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;IAEhE,mBAAmB;IACnB,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAElC,2DAA2D;IAC3D,IAAI,SAAS,EAAE,CAAC;QACd,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC;IAChC,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;GAmBG;AACH,SAAgB,qBAAqB,CAAC,SAAsB,EAAE;IAC5D,MAAM,EACJ,WAAW,GAAG,CAAC,EACf,YAAY,GAAG,IAAI,EACnB,iBAAiB,GAAG,CAAC,EACrB,QAAQ,GAAG,KAAK,EAChB,SAAS,GAAG,IAAI,EAChB,WAAW,GAAG,kBAAkB,EAChC,OAAO,GACR,GAAG,MAAM,CAAC;IAEX,OAAO,KAAK,EAAE,OAA0B,EAAE,IAAoB,EAA2B,EAAE;QACzF,IAAI,SAAkB,CAAC;QACvB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,OAAO,OAAO,GAAG,WAAW,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,+BAA+B;gBAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,EAAE,CAAC;gBAE9B,6CAA6C;gBAC7C,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,OAAO;wBACL,GAAG,QAAQ;wBACX,QAAQ,EAAE;4BACR,GAAG,QAAQ,CAAC,QAAQ;4BACpB,MAAM,EAAE;gCACN,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM;gCAC3B,aAAa,EAAE,OAAO;gCACtB,YAAY,EAAE,IAAI;6BACnB;yBACF;qBACF,CAAC;gBACJ,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAK,CAAC;gBAClB,OAAO,EAAE,CAAC;gBAEV,2BAA2B;gBAC3B,MAAM,SAAS,GAAG,OAAO,GAAG,WAAW,IAAI,WAAW,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBAEvE,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,2CAA2C;oBAC3C,IAAI,KAAK,YAAY,8BAAY,EAAE,CAAC;wBAClC,MAAM,IAAI,8BAAY,CAAC;4BACrB,GAAG,KAAK;4BACR,OAAO,EAAE;gCACP,GAAG,KAAK,CAAC,OAAO;gCAChB,aAAa,EAAE,OAAO;gCACtB,YAAY,EAAE,KAAK;6BACpB;yBACF,CAAC,CAAC;oBACL,CAAC;oBAED,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,wBAAwB;gBACxB,MAAM,KAAK,GAAG,cAAc,CAC1B,OAAO,GAAG,CAAC,EAAE,kCAAkC;gBAC/C,YAAY,EACZ,iBAAiB,EACjB,QAAQ,EACR,SAAS,CACV,CAAC;gBAEF,kCAAkC;gBAClC,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;gBACjC,CAAC;gBAED,+BAA+B;gBAC/B,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC5B,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,uBAAuB;gBACvB,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;gBAEnB,kDAAkD;gBAClD,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC5B,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,4CAA4C;QAC5C,MAAM,SAAS,CAAC;IAClB,CAAC,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,kBAAkB;AAClB,+EAA+E;AAE/E;;GAEG;AACH,SAAgB,gBAAgB,CAAC,KAAc;IAC7C,IAAI,KAAK,YAAY,8BAAY,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC,IAAI,KAAK,qBAAqB,CAAC;IAC9C,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAgB,cAAc,CAAC,KAAc;IAC3C,IAAI,KAAK,YAAY,8BAAY,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC,IAAI,KAAK,eAAe,CAAC;IACxC,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAgB,aAAa,CAAC,KAAc;IAC1C,IAAI,KAAK,YAAY,8BAAY,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,CAAC;IAC5E,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,SAAgB,oBAAoB,CAClC,UAAsD;IAEtD,OAAO,CAAC,KAAc,EAAE,QAAgB,EAAW,EAAE;QACnD,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,YAAY;oBACf,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC5B,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD,MAAM;gBACR,KAAK,SAAS;oBACZ,IAAI,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC1B,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD,MAAM;gBACR,KAAK,QAAQ;oBACX,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;wBACzB,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;AACJ,CAAC"}

package/dist/cjs/security.js

@@ -0,0 +1,175 @@
+"use strict";
+/**
+ * Security Middleware
+ *
+ * Adds security headers and implements security best practices for HTTP responses.
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_SECURITY_CONFIG = void 0;
+exports.createSecurityMiddleware = createSecurityMiddleware;
+exports.createProductionSecurityMiddleware = createProductionSecurityMiddleware;
+exports.createDevelopmentSecurityMiddleware = createDevelopmentSecurityMiddleware;
+/**
+ * Default security configuration
+ */
+exports.DEFAULT_SECURITY_CONFIG = {
+    contentSecurityPolicy: "default-src 'self'",
+    contentTypeOptions: 'nosniff',
+    frameOptions: 'DENY',
+    xssProtection: '1; mode=block',
+    hsts: 'max-age=31536000; includeSubDomains',
+    referrerPolicy: 'strict-origin-when-cross-origin',
+    permissionsPolicy: 'geolocation=(), microphone=(), camera=()',
+    poweredBy: false,
+};
+/**
+ * Create security headers middleware
+ *
+ * Adds security headers to responses to protect against common vulnerabilities.
+ *
+ * @param config - Security configuration
+ * @returns Middleware function
+ *
+ * @example
+ * ```typescript
+ * import { createSecurityMiddleware } from 'ai.matey';
+ *
+ * const security = createSecurityMiddleware({
+ *   contentSecurityPolicy: "default-src 'self'",
+ *   hsts: 'max-age=31536000',
+ * });
+ *
+ * bridge.use(security);
+ * ```
+ *
+ * @example Production Configuration
+ * ```typescript
+ * const productionSecurity = createSecurityMiddleware({
+ *   contentSecurityPolicy: "default-src 'self'; script-src 'self' 'unsafe-inline'; style-src 'self' 'unsafe-inline'",
+ *   frameOptions: 'DENY',
+ *   hsts: 'max-age=31536000; includeSubDomains; preload',
+ *   referrerPolicy: 'strict-origin-when-cross-origin',
+ *   permissionsPolicy: 'geolocation=(), microphone=(), camera=(), payment=()',
+ * });
+ * ```
+ */
+function createSecurityMiddleware(config = {}) {
+    const mergedConfig = {
+        ...exports.DEFAULT_SECURITY_CONFIG,
+        ...config,
+    };
+    return async (context, next) => {
+        // Add security headers to request metadata (will be passed through to response)
+        // Note: In ai.matey, security headers should ideally be added at the HTTP layer
+        // This middleware just marks that security headers should be added
+        const custom = context.request.metadata?.custom || {};
+        const securityHeaders = {};
+        const headers = securityHeaders;
+        // Content Security Policy
+        if (mergedConfig.contentSecurityPolicy !== false) {
+            headers['Content-Security-Policy'] = mergedConfig.contentSecurityPolicy;
+        }
+        // X-Content-Type-Options
+        if (mergedConfig.contentTypeOptions !== false) {
+            headers['X-Content-Type-Options'] = mergedConfig.contentTypeOptions;
+        }
+        // X-Frame-Options
+        if (mergedConfig.frameOptions !== false) {
+            headers['X-Frame-Options'] = mergedConfig.frameOptions;
+        }
+        // X-XSS-Protection
+        if (mergedConfig.xssProtection !== false) {
+            headers['X-XSS-Protection'] = mergedConfig.xssProtection;
+        }
+        // Strict-Transport-Security
+        if (mergedConfig.hsts !== false) {
+            headers['Strict-Transport-Security'] = mergedConfig.hsts;
+        }
+        // Referrer-Policy
+        if (mergedConfig.referrerPolicy !== false) {
+            headers['Referrer-Policy'] = mergedConfig.referrerPolicy;
+        }
+        // Permissions-Policy
+        if (mergedConfig.permissionsPolicy !== false) {
+            headers['Permissions-Policy'] = mergedConfig.permissionsPolicy;
+        }
+        // Remove X-Powered-By (or set custom value)
+        if (mergedConfig.poweredBy === false) {
+            // Mark for removal
+            headers['X-Powered-By'] = '';
+        }
+        else if (mergedConfig.poweredBy) {
+            headers['X-Powered-By'] = mergedConfig.poweredBy;
+        }
+        // Add custom headers
+        if (config.customHeaders) {
+            Object.assign(headers, config.customHeaders);
+        }
+        // Store headers in custom metadata (HTTP layer should apply these)
+        const newCustom = {
+            ...custom,
+            securityHeaders,
+        };
+        // Update request with security headers metadata
+        context.request = {
+            ...context.request,
+            metadata: {
+                ...context.request.metadata,
+                custom: newCustom,
+            },
+        };
+        // Execute next middleware/backend
+        return await next();
+    };
+}
+/**
+ * Create production-ready security middleware with strict settings
+ *
+ * @returns Middleware with production security settings
+ *
+ * @example
+ * ```typescript
+ * import { createProductionSecurityMiddleware } from 'ai.matey';
+ *
+ * bridge.use(createProductionSecurityMiddleware());
+ * ```
+ */
+function createProductionSecurityMiddleware() {
+    return createSecurityMiddleware({
+        contentSecurityPolicy: "default-src 'self'; script-src 'self'; style-src 'self'; img-src 'self' data:; font-src 'self'; connect-src 'self'; frame-ancestors 'none'",
+        frameOptions: 'DENY',
+        hsts: 'max-age=31536000; includeSubDomains; preload',
+        xssProtection: '1; mode=block',
+        contentTypeOptions: 'nosniff',
+        referrerPolicy: 'strict-origin-when-cross-origin',
+        permissionsPolicy: 'geolocation=(), microphone=(), camera=(), payment=(), usb=()',
+        poweredBy: false,
+    });
+}
+/**
+ * Create development-friendly security middleware with relaxed settings
+ *
+ * @returns Middleware with development security settings
+ *
+ * @example
+ * ```typescript
+ * import { createDevelopmentSecurityMiddleware } from 'ai.matey';
+ *
+ * bridge.use(createDevelopmentSecurityMiddleware());
+ * ```
+ */
+function createDevelopmentSecurityMiddleware() {
+    return createSecurityMiddleware({
+        contentSecurityPolicy: false, // Disable for easier development
+        frameOptions: 'SAMEORIGIN',
+        hsts: false, // Don't enforce HTTPS in development
+        xssProtection: '1; mode=block',
+        contentTypeOptions: 'nosniff',
+        referrerPolicy: 'strict-origin-when-cross-origin',
+        permissionsPolicy: false,
+        poweredBy: false,
+    });
+}
+//# sourceMappingURL=security.js.map

package/dist/cjs/security.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"security.js","sourceRoot":"","sources":["../../src/security.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAoHH,4DAgFC;AAcD,gFAYC;AAcD,kFAWC;AAhLD;;GAEG;AACU,QAAA,uBAAuB,GAAoD;IACtF,qBAAqB,EAAE,oBAAoB;IAC3C,kBAAkB,EAAE,SAAS;IAC7B,YAAY,EAAE,MAAM;IACpB,aAAa,EAAE,eAAe;IAC9B,IAAI,EAAE,qCAAqC;IAC3C,cAAc,EAAE,iCAAiC;IACjD,iBAAiB,EAAE,0CAA0C;IAC7D,SAAS,EAAE,KAAK;CACjB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,SAAgB,wBAAwB,CAAC,SAAyB,EAAE;IAClE,MAAM,YAAY,GAAG;QACnB,GAAG,+BAAuB;QAC1B,GAAG,MAAM;KACV,CAAC;IAEF,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;QAC7B,gFAAgF;QAChF,gFAAgF;QAChF,mEAAmE;QACnE,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,CAAC;QACtD,MAAM,eAAe,GAA2B,EAAE,CAAC;QACnD,MAAM,OAAO,GAAG,eAAe,CAAC;QAEhC,0BAA0B;QAC1B,IAAI,YAAY,CAAC,qBAAqB,KAAK,KAAK,EAAE,CAAC;YACjD,OAAO,CAAC,yBAAyB,CAAC,GAAG,YAAY,CAAC,qBAAqB,CAAC;QAC1E,CAAC;QAED,yBAAyB;QACzB,IAAI,YAAY,CAAC,kBAAkB,KAAK,KAAK,EAAE,CAAC;YAC9C,OAAO,CAAC,wBAAwB,CAAC,GAAG,YAAY,CAAC,kBAAkB,CAAC;QACtE,CAAC;QAED,kBAAkB;QAClB,IAAI,YAAY,CAAC,YAAY,KAAK,KAAK,EAAE,CAAC;YACxC,OAAO,CAAC,iBAAiB,CAAC,GAAG,YAAY,CAAC,YAAY,CAAC;QACzD,CAAC;QAED,mBAAmB;QACnB,IAAI,YAAY,CAAC,aAAa,KAAK,KAAK,EAAE,CAAC;YACzC,OAAO,CAAC,kBAAkB,CAAC,GAAG,YAAY,CAAC,aAAa,CAAC;QAC3D,CAAC;QAED,4BAA4B;QAC5B,IAAI,YAAY,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAChC,OAAO,CAAC,2BAA2B,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC;QAC3D,CAAC;QAED,kBAAkB;QAClB,IAAI,YAAY,CAAC,cAAc,KAAK,KAAK,EAAE,CAAC;YAC1C,OAAO,CAAC,iBAAiB,CAAC,GAAG,YAAY,CAAC,cAAc,CAAC;QAC3D,CAAC;QAED,qBAAqB;QACrB,IAAI,YAAY,CAAC,iBAAiB,KAAK,KAAK,EAAE,CAAC;YAC7C,OAAO,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC,iBAAiB,CAAC;QACjE,CAAC;QAED,4CAA4C;QAC5C,IAAI,YAAY,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;YACrC,mBAAmB;YACnB,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,CAAC;QAC/B,CAAC;aAAM,IAAI,YAAY,CAAC,SAAS,EAAE,CAAC;YAClC,OAAO,CAAC,cAAc,CAAC,GAAG,YAAY,CAAC,SAAS,CAAC;QACnD,CAAC;QAED,qBAAqB;QACrB,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QAC/C,CAAC;QAED,mEAAmE;QACnE,MAAM,SAAS,GAAG;YAChB,GAAG,MAAM;YACT,eAAe;SAChB,CAAC;QAEF,gDAAgD;QAChD,OAAO,CAAC,OAAO,GAAG;YAChB,GAAG,OAAO,CAAC,OAAO;YAClB,QAAQ,EAAE;gBACR,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ;gBAC3B,MAAM,EAAE,SAAS;aAClB;SACF,CAAC;QAEF,kCAAkC;QAClC,OAAO,MAAM,IAAI,EAAE,CAAC;IACtB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;GAWG;AACH,SAAgB,kCAAkC;IAChD,OAAO,wBAAwB,CAAC;QAC9B,qBAAqB,EACnB,4IAA4I;QAC9I,YAAY,EAAE,MAAM;QACpB,IAAI,EAAE,8CAA8C;QACpD,aAAa,EAAE,eAAe;QAC9B,kBAAkB,EAAE,SAAS;QAC7B,cAAc,EAAE,iCAAiC;QACjD,iBAAiB,EAAE,8DAA8D;QACjF,SAAS,EAAE,KAAK;KACjB,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;;;;GAWG;AACH,SAAgB,mCAAmC;IACjD,OAAO,wBAAwB,CAAC;QAC9B,qBAAqB,EAAE,KAAK,EAAE,iCAAiC;QAC/D,YAAY,EAAE,YAAY;QAC1B,IAAI,EAAE,KAAK,EAAE,qCAAqC;QAClD,aAAa,EAAE,eAAe;QAC9B,kBAAkB,EAAE,SAAS;QAC7B,cAAc,EAAE,iCAAiC;QACjD,iBAAiB,EAAE,KAAK;QACxB,SAAS,EAAE,KAAK;KACjB,CAAC,CAAC;AACL,CAAC"}

package/dist/cjs/telemetry.js

@@ -0,0 +1,216 @@
+"use strict";
+/**
+ * Telemetry Middleware
+ *
+ * Tracks metrics, events, and performance data for requests and responses.
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryTelemetrySink = exports.ConsoleTelemetrySink = exports.EventNames = exports.MetricNames = void 0;
+exports.createTelemetryMiddleware = createTelemetryMiddleware;
+/**
+ * Metric names.
+ */
+exports.MetricNames = {
+    REQUEST_COUNT: 'ai.adapter.request.count',
+    REQUEST_DURATION: 'ai.adapter.request.duration',
+    REQUEST_ERROR: 'ai.adapter.request.error',
+    TOKEN_PROMPT: 'ai.adapter.tokens.prompt',
+    TOKEN_COMPLETION: 'ai.adapter.tokens.completion',
+    TOKEN_TOTAL: 'ai.adapter.tokens.total',
+};
+/**
+ * Event names.
+ */
+exports.EventNames = {
+    REQUEST_START: 'ai.adapter.request.start',
+    REQUEST_COMPLETE: 'ai.adapter.request.complete',
+    REQUEST_ERROR: 'ai.adapter.request.error',
+};
+// ============================================================================
+// Sampling
+// ============================================================================
+/**
+ * Determine if this request should be sampled.
+ */
+function shouldSample(sampleRate) {
+    return Math.random() < sampleRate;
+}
+// ============================================================================
+// Middleware Factory
+// ============================================================================
+/**
+ * Create telemetry middleware.
+ *
+ * Tracks metrics and events for monitoring and observability.
+ *
+ * @param config Telemetry configuration
+ * @returns Telemetry middleware
+ *
+ * @example
+ * ```typescript
+ * const telemetry = createTelemetryMiddleware({
+ *   sink: {
+ *     recordMetric: (name, value, tags) => {
+ *       console.log(`Metric: ${name} = ${value}`, tags);
+ *     },
+ *     recordEvent: (name, data) => {
+ *       console.log(`Event: ${name}`, data);
+ *     }
+ *   },
+ *   trackCounts: true,
+ *   trackLatencies: true,
+ *   trackTokens: true
+ * });
+ *
+ * bridge.use(telemetry);
+ * ```
+ */
+function createTelemetryMiddleware(config) {
+    const { sink, trackCounts = true, trackLatencies = true, trackErrors = true, trackTokens = true, tags = {}, sampleRate = 1.0, } = config;
+    return async (context, next) => {
+        // Check if we should sample this request
+        const sampled = shouldSample(sampleRate);
+        const startTime = Date.now();
+        const requestId = context.request.metadata.requestId;
+        // Build base tags
+        const baseTags = {
+            ...tags,
+            request_id: requestId,
+            model: context.request.parameters?.model ?? 'unknown',
+            stream: String(context.request.stream ?? false),
+            frontend: context.request.metadata.provenance?.frontend ?? 'unknown',
+        };
+        // Record request start event
+        if (sampled) {
+            sink.recordEvent(exports.EventNames.REQUEST_START, {
+                requestId,
+                model: context.request.parameters?.model,
+                messageCount: context.request.messages.length,
+                stream: context.request.stream,
+                timestamp: startTime,
+            });
+        }
+        try {
+            // Call next middleware/handler
+            const response = await next();
+            const duration = Date.now() - startTime;
+            // Add backend to tags
+            const responseTags = {
+                ...baseTags,
+                backend: response.metadata.provenance?.backend ?? 'unknown',
+                finish_reason: response.finishReason,
+            };
+            // Record metrics
+            if (sampled) {
+                // Track request count
+                if (trackCounts) {
+                    sink.recordMetric(exports.MetricNames.REQUEST_COUNT, 1, {
+                        ...responseTags,
+                        status: 'success',
+                    });
+                }
+                // Track latency
+                if (trackLatencies) {
+                    sink.recordMetric(exports.MetricNames.REQUEST_DURATION, duration, responseTags);
+                }
+                // Track token usage
+                if (trackTokens && response.usage) {
+                    if (response.usage.promptTokens) {
+                        sink.recordMetric(exports.MetricNames.TOKEN_PROMPT, response.usage.promptTokens, responseTags);
+                    }
+                    if (response.usage.completionTokens) {
+                        sink.recordMetric(exports.MetricNames.TOKEN_COMPLETION, response.usage.completionTokens, responseTags);
+                    }
+                    if (response.usage.totalTokens) {
+                        sink.recordMetric(exports.MetricNames.TOKEN_TOTAL, response.usage.totalTokens, responseTags);
+                    }
+                }
+                // Record completion event
+                sink.recordEvent(exports.EventNames.REQUEST_COMPLETE, {
+                    requestId,
+                    duration,
+                    backend: response.metadata.provenance?.backend,
+                    finishReason: response.finishReason,
+                    usage: response.usage,
+                    timestamp: Date.now(),
+                });
+            }
+            return response;
+        }
+        catch (error) {
+            const duration = Date.now() - startTime;
+            // Add error info to tags
+            const errorTags = {
+                ...baseTags,
+                error_type: error instanceof Error ? error.name : 'unknown',
+            };
+            // Record error metrics
+            if (sampled) {
+                if (trackCounts) {
+                    sink.recordMetric(exports.MetricNames.REQUEST_COUNT, 1, {
+                        ...errorTags,
+                        status: 'error',
+                    });
+                }
+                if (trackLatencies) {
+                    sink.recordMetric(exports.MetricNames.REQUEST_DURATION, duration, errorTags);
+                }
+                if (trackErrors) {
+                    sink.recordMetric(exports.MetricNames.REQUEST_ERROR, 1, errorTags);
+                }
+                // Record error event
+                sink.recordEvent(exports.EventNames.REQUEST_ERROR, {
+                    requestId,
+                    duration,
+                    error: error instanceof Error ? error.message : String(error),
+                    errorType: error instanceof Error ? error.name : 'unknown',
+                    timestamp: Date.now(),
+                });
+            }
+            // Re-throw error
+            throw error;
+        }
+    };
+}
+// ============================================================================
+// Built-in Sinks
+// ============================================================================
+/**
+ * Console telemetry sink for development/debugging.
+ */
+class ConsoleTelemetrySink {
+    recordMetric(name, value, tags) {
+        console.warn('[Telemetry Metric]', { name, value, tags });
+    }
+    recordEvent(name, data) {
+        console.warn('[Telemetry Event]', { name, data });
+    }
+}
+exports.ConsoleTelemetrySink = ConsoleTelemetrySink;
+/**
+ * In-memory telemetry sink for testing.
+ */
+class InMemoryTelemetrySink {
+    metrics = [];
+    events = [];
+    recordMetric(name, value, tags) {
+        this.metrics.push({ name, value, tags });
+    }
+    recordEvent(name, data) {
+        this.events.push({ name, data });
+    }
+    getMetrics() {
+        return [...this.metrics];
+    }
+    getEvents() {
+        return [...this.events];
+    }
+    clear() {
+        this.metrics = [];
+        this.events = [];
+    }
+}
+exports.InMemoryTelemetrySink = InMemoryTelemetrySink;
+//# sourceMappingURL=telemetry.js.map

package/dist/cjs/telemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../src/telemetry.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAqHH,8DAuIC;AAtMD;;GAEG;AACU,QAAA,WAAW,GAAG;IACzB,aAAa,EAAE,0BAA0B;IACzC,gBAAgB,EAAE,6BAA6B;IAC/C,aAAa,EAAE,0BAA0B;IACzC,YAAY,EAAE,0BAA0B;IACxC,gBAAgB,EAAE,8BAA8B;IAChD,WAAW,EAAE,yBAAyB;CAC9B,CAAC;AAEX;;GAEG;AACU,QAAA,UAAU,GAAG;IACxB,aAAa,EAAE,0BAA0B;IACzC,gBAAgB,EAAE,6BAA6B;IAC/C,aAAa,EAAE,0BAA0B;CACjC,CAAC;AAEX,+EAA+E;AAC/E,WAAW;AACX,+EAA+E;AAE/E;;GAEG;AACH,SAAS,YAAY,CAAC,UAAkB;IACtC,OAAO,IAAI,CAAC,MAAM,EAAE,GAAG,UAAU,CAAC;AACpC,CAAC;AAED,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,SAAgB,yBAAyB,CAAC,MAAuB;IAC/D,MAAM,EACJ,IAAI,EACJ,WAAW,GAAG,IAAI,EAClB,cAAc,GAAG,IAAI,EACrB,WAAW,GAAG,IAAI,EAClB,WAAW,GAAG,IAAI,EAClB,IAAI,GAAG,EAAE,EACT,UAAU,GAAG,GAAG,GACjB,GAAG,MAAM,CAAC;IAEX,OAAO,KAAK,EAAE,OAA0B,EAAE,IAAoB,EAA2B,EAAE;QACzF,yCAAyC;QACzC,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC;QAEzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;QAErD,kBAAkB;QAClB,MAAM,QAAQ,GAA2B;YACvC,GAAG,IAAI;YACP,UAAU,EAAE,SAAS;YACrB,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,KAAK,IAAI,SAAS;YACrD,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,IAAI,KAAK,CAAC;YAC/C,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,SAAS;SACrE,CAAC;QAEF,6BAA6B;QAC7B,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,WAAW,CAAC,kBAAU,CAAC,aAAa,EAAE;gBACzC,SAAS;gBACT,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,KAAK;gBACxC,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAC7C,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM;gBAC9B,SAAS,EAAE,SAAS;aACrB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,+BAA+B;YAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,EAAE,CAAC;YAE9B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,sBAAsB;YACtB,MAAM,YAAY,GAA2B;gBAC3C,GAAG,QAAQ;gBACX,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,IAAI,SAAS;gBAC3D,aAAa,EAAE,QAAQ,CAAC,YAAY;aACrC,CAAC;YAEF,iBAAiB;YACjB,IAAI,OAAO,EAAE,CAAC;gBACZ,sBAAsB;gBACtB,IAAI,WAAW,EAAE,CAAC;oBAChB,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,aAAa,EAAE,CAAC,EAAE;wBAC9C,GAAG,YAAY;wBACf,MAAM,EAAE,SAAS;qBAClB,CAAC,CAAC;gBACL,CAAC;gBAED,gBAAgB;gBAChB,IAAI,cAAc,EAAE,CAAC;oBACnB,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,gBAAgB,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;gBAC1E,CAAC;gBAED,oBAAoB;gBACpB,IAAI,WAAW,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;oBAClC,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;wBAChC,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;oBACzF,CAAC;oBACD,IAAI,QAAQ,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;wBACpC,IAAI,CAAC,YAAY,CACf,mBAAW,CAAC,gBAAgB,EAC5B,QAAQ,CAAC,KAAK,CAAC,gBAAgB,EAC/B,YAAY,CACb,CAAC;oBACJ,CAAC;oBACD,IAAI,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;wBAC/B,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;oBACvF,CAAC;gBACH,CAAC;gBAED,0BAA0B;gBAC1B,IAAI,CAAC,WAAW,CAAC,kBAAU,CAAC,gBAAgB,EAAE;oBAC5C,SAAS;oBACT,QAAQ;oBACR,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO;oBAC9C,YAAY,EAAE,QAAQ,CAAC,YAAY;oBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,yBAAyB;YACzB,MAAM,SAAS,GAA2B;gBACxC,GAAG,QAAQ;gBACX,UAAU,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC;YAEF,uBAAuB;YACvB,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,WAAW,EAAE,CAAC;oBAChB,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,aAAa,EAAE,CAAC,EAAE;wBAC9C,GAAG,SAAS;wBACZ,MAAM,EAAE,OAAO;qBAChB,CAAC,CAAC;gBACL,CAAC;gBAED,IAAI,cAAc,EAAE,CAAC;oBACnB,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,gBAAgB,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;gBACvE,CAAC;gBAED,IAAI,WAAW,EAAE,CAAC;oBAChB,IAAI,CAAC,YAAY,CAAC,mBAAW,CAAC,aAAa,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;gBAC7D,CAAC;gBAED,qBAAqB;gBACrB,IAAI,CAAC,WAAW,CAAC,kBAAU,CAAC,aAAa,EAAE;oBACzC,SAAS;oBACT,QAAQ;oBACR,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;oBAC7D,SAAS,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;oBAC1D,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;YAED,iBAAiB;YACjB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAa,oBAAoB;IAC/B,YAAY,CAAC,IAAY,EAAE,KAAa,EAAE,IAA6B;QACrE,OAAO,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,WAAW,CAAC,IAAY,EAAE,IAA8B;QACtD,OAAO,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;CACF;AARD,oDAQC;AAED;;GAEG;AACH,MAAa,qBAAqB;IACxB,OAAO,GAA0E,EAAE,CAAC;IACpF,MAAM,GAA4D,EAAE,CAAC;IAE7E,YAAY,CAAC,IAAY,EAAE,KAAa,EAAE,IAA6B;QACrE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,WAAW,CAAC,IAAY,EAAE,IAA8B;QACtD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,SAAS;QACP,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;CACF;AAxBD,sDAwBC"}