ai.matey.middleware 0.2.0

package/dist/cjs/transform.js

@@ -0,0 +1,284 @@
+"use strict";
+/**
+ * Transform Middleware
+ *
+ * Transforms requests and responses with custom functions.
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTransformMiddleware = createTransformMiddleware;
+exports.createPromptRewriter = createPromptRewriter;
+exports.createParameterModifier = createParameterModifier;
+exports.createResponseFilter = createResponseFilter;
+exports.createSystemMessageInjector = createSystemMessageInjector;
+exports.createMessageFilter = createMessageFilter;
+exports.createContentSanitizer = createContentSanitizer;
+exports.composeRequestTransformers = composeRequestTransformers;
+exports.composeResponseTransformers = composeResponseTransformers;
+exports.composeMessageTransformers = composeMessageTransformers;
+// ============================================================================
+// 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);
+ * ```
+ */
+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');
+ * });
+ * ```
+ */
+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)
+ * }));
+ * ```
+ */
+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 };
+ * });
+ * ```
+ */
+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'
+ * );
+ * ```
+ */
+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'
+ * );
+ * ```
+ */
+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]');
+ * });
+ * ```
+ */
+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 }))
+ * );
+ * ```
+ */
+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
+ */
+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
+ */
+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/cjs/transform.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../src/transform.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;AA0EH,8DAoCC;AAmBD,oDAkCC;AAgBD,0DAWC;AAiBD,oDAIC;AAiBD,kEAkBC;AAgBD,kDAiBC;AAgBD,wDAIC;AAgBD,gEAYC;AAQD,kEAYC;AAQD,gEAYC;AAjUD,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,SAAgB,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,SAAgB,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,SAAgB,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,SAAgB,oBAAoB,CAClC,MAA8E;IAE9E,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,SAAgB,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,SAAgB,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,SAAgB,sBAAsB,CACpC,SAAqD;IAErD,OAAO,oBAAoB,CAAC,SAAS,CAAC,CAAC;AACzC,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,SAAgB,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,SAAgB,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,SAAgB,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"}