ai.matey.middleware 0.2.0

package/dist/esm/security.js

@@ -0,0 +1,169 @@
+/**
+ * Security Middleware
+ *
+ * Adds security headers and implements security best practices for HTTP responses.
+ *
+ * @module
+ */
+/**
+ * Default security configuration
+ */
+export const 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=()',
+ * });
+ * ```
+ */
+export function createSecurityMiddleware(config = {}) {
+    const mergedConfig = {
+        ...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());
+ * ```
+ */
+export 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());
+ * ```
+ */
+export 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/esm/security.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"security.js","sourceRoot":"","sources":["../../src/security.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAuEH;;GAEG;AACH,MAAM,CAAC,MAAM,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,MAAM,UAAU,wBAAwB,CAAC,SAAyB,EAAE;IAClE,MAAM,YAAY,GAAG;QACnB,GAAG,uBAAuB;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,MAAM,UAAU,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,MAAM,UAAU,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/esm/telemetry.js

@@ -0,0 +1,210 @@
+/**
+ * Telemetry Middleware
+ *
+ * Tracks metrics, events, and performance data for requests and responses.
+ *
+ * @module
+ */
+/**
+ * Metric names.
+ */
+export const 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.
+ */
+export const 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);
+ * ```
+ */
+export 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(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(MetricNames.REQUEST_COUNT, 1, {
+                        ...responseTags,
+                        status: 'success',
+                    });
+                }
+                // Track latency
+                if (trackLatencies) {
+                    sink.recordMetric(MetricNames.REQUEST_DURATION, duration, responseTags);
+                }
+                // Track token usage
+                if (trackTokens && response.usage) {
+                    if (response.usage.promptTokens) {
+                        sink.recordMetric(MetricNames.TOKEN_PROMPT, response.usage.promptTokens, responseTags);
+                    }
+                    if (response.usage.completionTokens) {
+                        sink.recordMetric(MetricNames.TOKEN_COMPLETION, response.usage.completionTokens, responseTags);
+                    }
+                    if (response.usage.totalTokens) {
+                        sink.recordMetric(MetricNames.TOKEN_TOTAL, response.usage.totalTokens, responseTags);
+                    }
+                }
+                // Record completion event
+                sink.recordEvent(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(MetricNames.REQUEST_COUNT, 1, {
+                        ...errorTags,
+                        status: 'error',
+                    });
+                }
+                if (trackLatencies) {
+                    sink.recordMetric(MetricNames.REQUEST_DURATION, duration, errorTags);
+                }
+                if (trackErrors) {
+                    sink.recordMetric(MetricNames.REQUEST_ERROR, 1, errorTags);
+                }
+                // Record error event
+                sink.recordEvent(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.
+ */
+export class ConsoleTelemetrySink {
+    recordMetric(name, value, tags) {
+        console.warn('[Telemetry Metric]', { name, value, tags });
+    }
+    recordEvent(name, data) {
+        console.warn('[Telemetry Event]', { name, data });
+    }
+}
+/**
+ * In-memory telemetry sink for testing.
+ */
+export 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 = [];
+    }
+}
+//# sourceMappingURL=telemetry.js.map

package/dist/esm/telemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../src/telemetry.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAsDH;;GAEG;AACH,MAAM,CAAC,MAAM,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;AACH,MAAM,CAAC,MAAM,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,MAAM,UAAU,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,UAAU,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,WAAW,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,WAAW,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,WAAW,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,WAAW,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,WAAW,CAAC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;oBACvF,CAAC;gBACH,CAAC;gBAED,0BAA0B;gBAC1B,IAAI,CAAC,WAAW,CAAC,UAAU,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,WAAW,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,WAAW,CAAC,gBAAgB,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;gBACvE,CAAC;gBAED,IAAI,WAAW,EAAE,CAAC;oBAChB,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;gBAC7D,CAAC;gBAED,qBAAqB;gBACrB,IAAI,CAAC,WAAW,CAAC,UAAU,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,MAAM,OAAO,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;AAED;;GAEG;AACH,MAAM,OAAO,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"}

package/dist/esm/transform.js

@@ -0,0 +1,272 @@
+/**
+ * Transform Middleware
+ *
+ * Transforms requests and responses with custom functions.
+ *
+ * @module
+ */
+// ============================================================================
+// Middleware Factory
+// ============================================================================
+/**
+ * Create transform middleware.
+ *
+ * Applies custom transformations to requests and responses.
+ *
+ * @param config Transform configuration
+ * @returns Transform middleware
+ *
+ * @example
+ * ```typescript
+ * // Add system message prefix
+ * const transform = createTransformMiddleware({
+ *   transformRequest: (request) => ({
+ *     ...request,
+ *     messages: [
+ *       { role: 'system', content: 'You are helpful.' },
+ *       ...request.messages
+ *     ]
+ *   })
+ * });
+ *
+ * bridge.use(transform);
+ * ```
+ */
+export function createTransformMiddleware(config = {}) {
+    const { transformRequest, transformResponse, transformMessages } = config;
+    return async (context, next) => {
+        let request = context.request;
+        // Apply message transformations
+        if (transformMessages) {
+            const transformedMessages = await Promise.all(request.messages.map((msg) => Promise.resolve(transformMessages(msg))));
+            request = {
+                ...request,
+                messages: transformedMessages,
+            };
+        }
+        // Apply request transformation
+        if (transformRequest) {
+            request = await transformRequest(request);
+        }
+        // Update context with transformed request
+        context.request = request;
+        // Call next middleware/handler
+        let response = await next();
+        // Apply response transformation
+        if (transformResponse) {
+            response = await transformResponse(response);
+        }
+        return response;
+    };
+}
+// ============================================================================
+// Built-in Transformers
+// ============================================================================
+/**
+ * Create a prompt rewriting transformer.
+ *
+ * @param rewriter Function to rewrite prompt text
+ * @returns Message transformer
+ *
+ * @example
+ * ```typescript
+ * const transformer = createPromptRewriter((text) => {
+ *   return text.replace(/foo/g, 'bar');
+ * });
+ * ```
+ */
+export function createPromptRewriter(rewriter) {
+    return async (message) => {
+        // Only transform text content
+        if (typeof message.content === 'string') {
+            return {
+                ...message,
+                content: await rewriter(message.content),
+            };
+        }
+        // Transform text in content blocks
+        if (Array.isArray(message.content)) {
+            const transformedContent = await Promise.all(message.content.map(async (block) => {
+                if (block.type === 'text') {
+                    return {
+                        ...block,
+                        text: await rewriter(block.text),
+                    };
+                }
+                return block;
+            }));
+            return {
+                ...message,
+                content: transformedContent,
+            };
+        }
+        return message;
+    };
+}
+/**
+ * Create a parameter modifier transformer.
+ *
+ * @param modifier Function to modify parameters
+ * @returns Request transformer
+ *
+ * @example
+ * ```typescript
+ * const transformer = createParameterModifier((params) => ({
+ *   ...params,
+ *   temperature: Math.min(params.temperature ?? 0.7, 0.9)
+ * }));
+ * ```
+ */
+export function createParameterModifier(modifier) {
+    return async (request) => {
+        return {
+            ...request,
+            parameters: await modifier(request.parameters),
+        };
+    };
+}
+/**
+ * Create a response filter transformer.
+ *
+ * @param filter Function to filter/modify response
+ * @returns Response transformer
+ *
+ * @example
+ * ```typescript
+ * const transformer = createResponseFilter((response) => {
+ *   // Remove custom metadata
+ *   const { custom, ...metadata } = response.metadata;
+ *   return { ...response, metadata };
+ * });
+ * ```
+ */
+export function createResponseFilter(filter) {
+    return filter;
+}
+/**
+ * Create a system message injector.
+ *
+ * @param systemMessage System message to inject
+ * @param position Where to inject ('start' or 'end')
+ * @returns Request transformer
+ *
+ * @example
+ * ```typescript
+ * const transformer = createSystemMessageInjector(
+ *   'You are a helpful assistant.',
+ *   'start'
+ * );
+ * ```
+ */
+export function createSystemMessageInjector(systemMessage, position = 'start') {
+    return (request) => {
+        const systemMsg = {
+            role: 'system',
+            content: systemMessage,
+        };
+        const messages = position === 'start' ? [systemMsg, ...request.messages] : [...request.messages, systemMsg];
+        return {
+            ...request,
+            messages,
+        };
+    };
+}
+/**
+ * Create a message filter transformer.
+ *
+ * @param predicate Function to determine if message should be kept
+ * @returns Request transformer
+ *
+ * @example
+ * ```typescript
+ * // Remove all system messages
+ * const transformer = createMessageFilter(
+ *   (msg) => msg.role !== 'system'
+ * );
+ * ```
+ */
+export function createMessageFilter(predicate) {
+    return async (request) => {
+        const filteredMessages = [];
+        for (const message of request.messages) {
+            if (await predicate(message)) {
+                filteredMessages.push(message);
+            }
+        }
+        return {
+            ...request,
+            messages: filteredMessages,
+        };
+    };
+}
+/**
+ * Create a content sanitizer transformer.
+ *
+ * @param sanitizer Function to sanitize message content
+ * @returns Message transformer
+ *
+ * @example
+ * ```typescript
+ * // Remove sensitive data
+ * const transformer = createContentSanitizer((text) => {
+ *   return text.replace(/\b\d{3}-\d{2}-\d{4}\b/g, '[SSN]');
+ * });
+ * ```
+ */
+export function createContentSanitizer(sanitizer) {
+    return createPromptRewriter(sanitizer);
+}
+/**
+ * Compose multiple request transformers.
+ *
+ * @param transformers Request transformers to compose
+ * @returns Composed request transformer
+ *
+ * @example
+ * ```typescript
+ * const composed = composeRequestTransformers(
+ *   createSystemMessageInjector('Be helpful'),
+ *   createParameterModifier(params => ({ ...params, temperature: 0.7 }))
+ * );
+ * ```
+ */
+export function composeRequestTransformers(...transformers) {
+    return async (request) => {
+        let result = request;
+        for (const transformer of transformers) {
+            result = await transformer(result);
+        }
+        return result;
+    };
+}
+/**
+ * Compose multiple response transformers.
+ *
+ * @param transformers Response transformers to compose
+ * @returns Composed response transformer
+ */
+export function composeResponseTransformers(...transformers) {
+    return async (response) => {
+        let result = response;
+        for (const transformer of transformers) {
+            result = await transformer(result);
+        }
+        return result;
+    };
+}
+/**
+ * Compose multiple message transformers.
+ *
+ * @param transformers Message transformers to compose
+ * @returns Composed message transformer
+ */
+export function composeMessageTransformers(...transformers) {
+    return async (message) => {
+        let result = message;
+        for (const transformer of transformers) {
+            result = await transformer(result);
+        }
+        return result;
+    };
+}
+//# sourceMappingURL=transform.js.map

package/dist/esm/transform.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../src/transform.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AA8CH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,UAAU,yBAAyB,CAAC,SAA0B,EAAE;IACpE,MAAM,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,GAAG,MAAM,CAAC;IAE1E,OAAO,KAAK,EAAE,OAA0B,EAAE,IAAoB,EAA2B,EAAE;QACzF,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAE9B,gCAAgC;QAChC,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,mBAAmB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC3C,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CACvE,CAAC;YAEF,OAAO,GAAG;gBACR,GAAG,OAAO;gBACV,QAAQ,EAAE,mBAAmB;aAC9B,CAAC;QACJ,CAAC;QAED,+BAA+B;QAC/B,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,GAAG,MAAM,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;QAED,0CAA0C;QAC1C,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC;QAE1B,+BAA+B;QAC/B,IAAI,QAAQ,GAAG,MAAM,IAAI,EAAE,CAAC;QAE5B,gCAAgC;QAChC,IAAI,iBAAiB,EAAE,CAAC;YACtB,QAAQ,GAAG,MAAM,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,wBAAwB;AACxB,+EAA+E;AAE/E;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,oBAAoB,CAClC,QAAoD;IAEpD,OAAO,KAAK,EAAE,OAAkB,EAAsB,EAAE;QACtD,8BAA8B;QAC9B,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO;gBACL,GAAG,OAAO;gBACV,OAAO,EAAE,MAAM,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC;aACzC,CAAC;QACJ,CAAC;QAED,mCAAmC;QACnC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACnC,MAAM,kBAAkB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;gBAClC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,OAAO;wBACL,GAAG,KAAK;wBACR,IAAI,EAAE,MAAM,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;qBACjC,CAAC;gBACJ,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CACH,CAAC;YAEF,OAAO;gBACL,GAAG,OAAO;gBACV,OAAO,EAAE,kBAAkB;aAC5B,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,uBAAuB,CACrC,QAEuE;IAEvE,OAAO,KAAK,EAAE,OAAsB,EAA0B,EAAE;QAC9D,OAAO;YACL,GAAG,OAAO;YACV,UAAU,EAAE,MAAM,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC;SAC/C,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAA8E;IAE9E,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,2BAA2B,CACzC,aAAqB,EACrB,WAA4B,OAAO;IAEnC,OAAO,CAAC,OAAsB,EAAiB,EAAE;QAC/C,MAAM,SAAS,GAAc;YAC3B,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,aAAa;SACvB,CAAC;QAEF,MAAM,QAAQ,GACZ,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAE7F,OAAO;YACL,GAAG,OAAO;YACV,QAAQ;SACT,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,mBAAmB,CACjC,SAA6D;IAE7D,OAAO,KAAK,EAAE,OAAsB,EAA0B,EAAE;QAC9D,MAAM,gBAAgB,GAAG,EAAE,CAAC;QAE5B,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvC,IAAI,MAAM,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7B,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAED,OAAO;YACL,GAAG,OAAO;YACV,QAAQ,EAAE,gBAAgB;SAC3B,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,sBAAsB,CACpC,SAAqD;IAErD,OAAO,oBAAoB,CAAC,SAAS,CAAC,CAAC;AACzC,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,0BAA0B,CACxC,GAAG,YAAkC;IAErC,OAAO,KAAK,EAAE,OAAsB,EAA0B,EAAE;QAC9D,IAAI,MAAM,GAAG,OAAO,CAAC;QAErB,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,2BAA2B,CACzC,GAAG,YAAmC;IAEtC,OAAO,KAAK,EAAE,QAAwB,EAA2B,EAAE;QACjE,IAAI,MAAM,GAAG,QAAQ,CAAC;QAEtB,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,0BAA0B,CACxC,GAAG,YAAkC;IAErC,OAAO,KAAK,EAAE,OAAkB,EAAsB,EAAE;QACtD,IAAI,MAAM,GAAG,OAAO,CAAC;QAErB,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;AACJ,CAAC"}