@tanstack/ai 0.42.0 → 0.43.1

package/dist/esm/middlewares/content-guard.js

@@ -1,156 +1,150 @@
+//#region src/middlewares/content-guard.ts
+/**
+* Apply all rules to a string, returning the transformed result.
+*/
 function applyRules(text, rules) {
-  let result = text;
-  for (const rule of rules) {
-    if ("pattern" in rule) {
-      result = result.replace(rule.pattern, rule.replacement);
-    } else {
-      result = rule.fn(result);
-    }
-  }
-  return result;
+	let result = text;
+	for (const rule of rules) if ("pattern" in rule) result = result.replace(rule.pattern, rule.replacement);
+	else result = rule.fn(result);
+	return result;
 }
+/**
+* Creates a middleware that filters or transforms streamed text content.
+*
+* @example
+* ```ts
+* import { contentGuardMiddleware } from '@tanstack/ai/middlewares'
+*
+* const guard = contentGuardMiddleware({
+*   rules: [
+*     { pattern: /\b\d{3}-\d{2}-\d{4}\b/g, replacement: '[SSN REDACTED]' },
+*   ],
+*   strategy: 'buffered',
+* })
+* ```
+*/
 function contentGuardMiddleware(options) {
-  const {
-    rules,
-    strategy = "buffered",
-    bufferSize = 50,
-    blockOnMatch = false,
-    onFiltered
-  } = options;
-  if (strategy === "delta") {
-    return createDeltaStrategy(rules, blockOnMatch, onFiltered);
-  }
-  return createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered);
+	const { rules, strategy = "buffered", bufferSize = 50, blockOnMatch = false, onFiltered } = options;
+	if (strategy === "delta") return createDeltaStrategy(rules, blockOnMatch, onFiltered);
+	return createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered);
 }
 function createDeltaStrategy(rules, blockOnMatch, onFiltered) {
-  return {
-    name: "content-guard",
-    onChunk(_ctx, chunk) {
-      if (chunk.type !== "TEXT_MESSAGE_CONTENT") return;
-      const original = chunk.delta;
-      const filtered = applyRules(original, rules);
-      if (filtered === original) return;
-      if (onFiltered) {
-        onFiltered({
-          messageId: chunk.messageId,
-          original,
-          filtered,
-          strategy: "delta"
-        });
-      }
-      if (blockOnMatch) return null;
-      const { content: _strippedContent, ...rest } = chunk;
-      return {
-        ...rest,
-        delta: filtered
-      };
-    }
-  };
+	return {
+		name: "content-guard",
+		onChunk(_ctx, chunk) {
+			if (chunk.type !== "TEXT_MESSAGE_CONTENT") return;
+			const original = chunk.delta;
+			const filtered = applyRules(original, rules);
+			if (filtered === original) return;
+			if (onFiltered) onFiltered({
+				messageId: chunk.messageId,
+				original,
+				filtered,
+				strategy: "delta"
+			});
+			if (blockOnMatch) return null;
+			const { content: _strippedContent, ...rest } = chunk;
+			return {
+				...rest,
+				delta: filtered
+			};
+		}
+	};
 }
 function createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered) {
-  let rawAccumulated = "";
-  let emittedFilteredLength = 0;
-  let lastMessageId = "";
-  function resetState() {
-    rawAccumulated = "";
-    emittedFilteredLength = 0;
-    lastMessageId = "";
-  }
-  function flushBuffer() {
-    if (rawAccumulated.length === 0) return null;
-    const filtered = applyRules(rawAccumulated, rules);
-    if (blockOnMatch && filtered !== rawAccumulated) {
-      if (onFiltered) {
-        onFiltered({
-          messageId: lastMessageId,
-          original: rawAccumulated,
-          filtered,
-          strategy: "buffered"
-        });
-      }
-      resetState();
-      return null;
-    }
-    const remaining = filtered.slice(emittedFilteredLength);
-    if (remaining.length > 0) {
-      if (filtered !== rawAccumulated && onFiltered) {
-        onFiltered({
-          messageId: lastMessageId,
-          original: rawAccumulated,
-          filtered,
-          strategy: "buffered"
-        });
-      }
-      const flushed = {
-        type: "TEXT_MESSAGE_CONTENT",
-        messageId: lastMessageId,
-        delta: remaining,
-        content: filtered,
-        timestamp: Date.now()
-      };
-      resetState();
-      return flushed;
-    }
-    resetState();
-    return null;
-  }
-  return {
-    name: "content-guard",
-    onStart() {
-      resetState();
-    },
-    onChunk(_ctx, chunk) {
-      if (chunk.type === "TEXT_MESSAGE_END" || chunk.type === "RUN_FINISHED") {
-        const flushed = flushBuffer();
-        if (flushed) return [flushed, chunk];
-        return;
-      }
-      if (chunk.type !== "TEXT_MESSAGE_CONTENT") return;
-      const pending = [];
-      if (lastMessageId && chunk.messageId !== lastMessageId) {
-        const flushed = flushBuffer();
-        if (flushed) pending.push(flushed);
-      }
-      rawAccumulated += chunk.delta;
-      lastMessageId = chunk.messageId;
-      const filtered = applyRules(rawAccumulated, rules);
-      const safeFilteredEnd = Math.max(0, filtered.length - bufferSize);
-      if (safeFilteredEnd <= emittedFilteredLength) {
-        return pending.length > 0 ? pending : null;
-      }
-      if (blockOnMatch && filtered !== rawAccumulated) {
-        if (onFiltered) {
-          onFiltered({
-            messageId: chunk.messageId,
-            original: rawAccumulated,
-            filtered,
-            strategy: "buffered"
-          });
-        }
-        return pending.length > 0 ? pending : null;
-      }
-      const newDelta = filtered.slice(emittedFilteredLength, safeFilteredEnd);
-      if (filtered !== rawAccumulated && onFiltered) {
-        onFiltered({
-          messageId: chunk.messageId,
-          original: rawAccumulated,
-          filtered,
-          strategy: "buffered"
-        });
-      }
-      emittedFilteredLength = safeFilteredEnd;
-      const emitChunk = {
-        ...chunk,
-        delta: newDelta,
-        content: filtered.slice(0, safeFilteredEnd)
-      };
-      const wasEmpty = pending.length === 0;
-      pending.push(emitChunk);
-      return wasEmpty ? emitChunk : pending;
-    }
-  };
+	let rawAccumulated = "";
+	let emittedFilteredLength = 0;
+	let lastMessageId = "";
+	function resetState() {
+		rawAccumulated = "";
+		emittedFilteredLength = 0;
+		lastMessageId = "";
+	}
+	function flushBuffer() {
+		if (rawAccumulated.length === 0) return null;
+		const filtered = applyRules(rawAccumulated, rules);
+		if (blockOnMatch && filtered !== rawAccumulated) {
+			if (onFiltered) onFiltered({
+				messageId: lastMessageId,
+				original: rawAccumulated,
+				filtered,
+				strategy: "buffered"
+			});
+			resetState();
+			return null;
+		}
+		const remaining = filtered.slice(emittedFilteredLength);
+		if (remaining.length > 0) {
+			if (filtered !== rawAccumulated && onFiltered) onFiltered({
+				messageId: lastMessageId,
+				original: rawAccumulated,
+				filtered,
+				strategy: "buffered"
+			});
+			const flushed = {
+				type: "TEXT_MESSAGE_CONTENT",
+				messageId: lastMessageId,
+				delta: remaining,
+				content: filtered,
+				timestamp: Date.now()
+			};
+			resetState();
+			return flushed;
+		}
+		resetState();
+		return null;
+	}
+	return {
+		name: "content-guard",
+		onStart() {
+			resetState();
+		},
+		onChunk(_ctx, chunk) {
+			if (chunk.type === "TEXT_MESSAGE_END" || chunk.type === "RUN_FINISHED") {
+				const flushed = flushBuffer();
+				if (flushed) return [flushed, chunk];
+				return;
+			}
+			if (chunk.type !== "TEXT_MESSAGE_CONTENT") return;
+			const pending = [];
+			if (lastMessageId && chunk.messageId !== lastMessageId) {
+				const flushed = flushBuffer();
+				if (flushed) pending.push(flushed);
+			}
+			rawAccumulated += chunk.delta;
+			lastMessageId = chunk.messageId;
+			const filtered = applyRules(rawAccumulated, rules);
+			const safeFilteredEnd = Math.max(0, filtered.length - bufferSize);
+			if (safeFilteredEnd <= emittedFilteredLength) return pending.length > 0 ? pending : null;
+			if (blockOnMatch && filtered !== rawAccumulated) {
+				if (onFiltered) onFiltered({
+					messageId: chunk.messageId,
+					original: rawAccumulated,
+					filtered,
+					strategy: "buffered"
+				});
+				return pending.length > 0 ? pending : null;
+			}
+			const newDelta = filtered.slice(emittedFilteredLength, safeFilteredEnd);
+			if (filtered !== rawAccumulated && onFiltered) onFiltered({
+				messageId: chunk.messageId,
+				original: rawAccumulated,
+				filtered,
+				strategy: "buffered"
+			});
+			emittedFilteredLength = safeFilteredEnd;
+			const emitChunk = {
+				...chunk,
+				delta: newDelta,
+				content: filtered.slice(0, safeFilteredEnd)
+			};
+			const wasEmpty = pending.length === 0;
+			pending.push(emitChunk);
+			return wasEmpty ? emitChunk : pending;
+		}
+	};
 }
-export {
-  contentGuardMiddleware
-};
-//# sourceMappingURL=content-guard.js.map
+//#endregion
+export { contentGuardMiddleware };
+
+//# sourceMappingURL=content-guard.js.map

package/dist/esm/middlewares/content-guard.js.map

@@ -1 +1 @@
-{"version":3,"file":"content-guard.js","sources":["../../../src/middlewares/content-guard.ts"],"sourcesContent":["import type {\n  ChatMiddleware,\n  ChatMiddlewareContext,\n} from '../activities/chat/middleware/types'\nimport type { StreamChunk } from '../types'\n\n/**\n * A content guard rule — either a regex pattern with replacement, or a transform function.\n */\nexport type ContentGuardRule =\n  | { pattern: RegExp; replacement: string }\n  | { fn: (text: string) => string }\n\n/**\n * Information passed to the onFiltered callback.\n */\nexport interface ContentFilteredInfo {\n  /** The message ID being filtered */\n  messageId: string\n  /** The original text before filtering */\n  original: string\n  /** The filtered text after rules applied */\n  filtered: string\n  /** Which strategy was used */\n  strategy: 'delta' | 'buffered'\n}\n\n/**\n * Options for the content guard middleware.\n */\nexport interface ContentGuardMiddlewareOptions {\n  /**\n   * Rules to apply to text content. Each rule is either a regex pattern\n   * with a replacement string, or a custom transform function.\n   * Rules are applied in order. Each rule receives the output of the previous.\n   */\n  rules: Array<ContentGuardRule>\n\n  /**\n   * Matching strategy:\n   * - 'delta': Apply rules to each delta as it arrives. Fast, real-time,\n   *   but patterns spanning chunk boundaries may be missed.\n   * - 'buffered': Accumulate content and apply rules to settled portions,\n   *   holding back a look-behind buffer to catch cross-boundary patterns.\n   *\n   * @default 'buffered'\n   */\n  strategy?: 'delta' | 'buffered'\n\n  /**\n   * Number of characters to hold back before emitting (buffered strategy only).\n   * Should be at least as long as the longest pattern you expect to match.\n   * Buffer is flushed when the stream ends.\n   *\n   * @default 50\n   */\n  bufferSize?: number\n\n  /**\n   * If true, drop the entire chunk when any rule changes the content.\n   * @default false\n   */\n  blockOnMatch?: boolean\n\n  /**\n   * Callback when content is filtered by any rule.\n   */\n  onFiltered?: (info: ContentFilteredInfo) => void\n}\n\n/**\n * Apply all rules to a string, returning the transformed result.\n */\nfunction applyRules(text: string, rules: Array<ContentGuardRule>): string {\n  let result = text\n  for (const rule of rules) {\n    if ('pattern' in rule) {\n      result = result.replace(rule.pattern, rule.replacement)\n    } else {\n      result = rule.fn(result)\n    }\n  }\n  return result\n}\n\n/**\n * Creates a middleware that filters or transforms streamed text content.\n *\n * @example\n * ```ts\n * import { contentGuardMiddleware } from '@tanstack/ai/middlewares'\n *\n * const guard = contentGuardMiddleware({\n *   rules: [\n *     { pattern: /\\b\\d{3}-\\d{2}-\\d{4}\\b/g, replacement: '[SSN REDACTED]' },\n *   ],\n *   strategy: 'buffered',\n * })\n * ```\n */\nexport function contentGuardMiddleware(\n  options: ContentGuardMiddlewareOptions,\n): ChatMiddleware {\n  const {\n    rules,\n    strategy = 'buffered',\n    bufferSize = 50,\n    blockOnMatch = false,\n    onFiltered,\n  } = options\n\n  if (strategy === 'delta') {\n    return createDeltaStrategy(rules, blockOnMatch, onFiltered)\n  }\n  return createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered)\n}\n\nfunction createDeltaStrategy(\n  rules: Array<ContentGuardRule>,\n  blockOnMatch: boolean,\n  onFiltered?: (info: ContentFilteredInfo) => void,\n): ChatMiddleware {\n  return {\n    name: 'content-guard',\n\n    onChunk(_ctx: ChatMiddlewareContext, chunk: StreamChunk) {\n      if (chunk.type !== 'TEXT_MESSAGE_CONTENT') return\n\n      const original = chunk.delta\n      const filtered = applyRules(original, rules)\n\n      if (filtered === original) return // unchanged, pass through\n\n      if (onFiltered) {\n        onFiltered({\n          messageId: chunk.messageId,\n          original,\n          filtered,\n          strategy: 'delta',\n        })\n      }\n\n      if (blockOnMatch) return null // drop chunk\n\n      // Strip out the previous `content` field by destructuring it away — with\n      // `exactOptionalPropertyTypes` we can't assign `content: undefined`\n      // against `content?: string`. The replacement event carries only the\n      // filtered delta.\n      const { content: _strippedContent, ...rest } = chunk\n      void _strippedContent\n      return {\n        ...rest,\n        delta: filtered,\n      }\n    },\n  }\n}\n\nfunction createBufferedStrategy(\n  rules: Array<ContentGuardRule>,\n  bufferSize: number,\n  blockOnMatch: boolean,\n  onFiltered?: (info: ContentFilteredInfo) => void,\n): ChatMiddleware {\n  let rawAccumulated = ''\n  let emittedFilteredLength = 0\n  let lastMessageId = ''\n\n  function resetState() {\n    rawAccumulated = ''\n    emittedFilteredLength = 0\n    lastMessageId = ''\n  }\n\n  function flushBuffer(): StreamChunk | null {\n    if (rawAccumulated.length === 0) return null\n\n    const filtered = applyRules(rawAccumulated, rules)\n\n    if (blockOnMatch && filtered !== rawAccumulated) {\n      if (onFiltered) {\n        onFiltered({\n          messageId: lastMessageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n      resetState()\n      return null\n    }\n\n    const remaining = filtered.slice(emittedFilteredLength)\n    if (remaining.length > 0) {\n      if (filtered !== rawAccumulated && onFiltered) {\n        onFiltered({\n          messageId: lastMessageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n\n      const flushed = {\n        type: 'TEXT_MESSAGE_CONTENT',\n        messageId: lastMessageId,\n        delta: remaining,\n        content: filtered,\n        timestamp: Date.now(),\n      } as StreamChunk\n\n      resetState()\n      return flushed\n    }\n\n    resetState()\n    return null\n  }\n\n  return {\n    name: 'content-guard',\n\n    onStart() {\n      resetState()\n    },\n\n    onChunk(_ctx: ChatMiddlewareContext, chunk: StreamChunk) {\n      // Flush buffer on stream end events\n      if (chunk.type === 'TEXT_MESSAGE_END' || chunk.type === 'RUN_FINISHED') {\n        const flushed = flushBuffer()\n        if (flushed) return [flushed, chunk]\n        return // pass through end event\n      }\n\n      if (chunk.type !== 'TEXT_MESSAGE_CONTENT') return // pass through\n\n      // Flush buffer on message boundary change\n      const pending: Array<StreamChunk> = []\n      if (lastMessageId && chunk.messageId !== lastMessageId) {\n        const flushed = flushBuffer()\n        if (flushed) pending.push(flushed)\n      }\n\n      rawAccumulated += chunk.delta\n      lastMessageId = chunk.messageId\n\n      // Apply rules to full accumulated text, buffer in filtered space\n      const filtered = applyRules(rawAccumulated, rules)\n      const safeFilteredEnd = Math.max(0, filtered.length - bufferSize)\n\n      if (safeFilteredEnd <= emittedFilteredLength) {\n        return pending.length > 0 ? pending : null\n      }\n\n      if (blockOnMatch && filtered !== rawAccumulated) {\n        if (onFiltered) {\n          onFiltered({\n            messageId: chunk.messageId,\n            original: rawAccumulated,\n            filtered,\n            strategy: 'buffered',\n          })\n        }\n        return pending.length > 0 ? pending : null\n      }\n\n      const newDelta = filtered.slice(emittedFilteredLength, safeFilteredEnd)\n\n      if (filtered !== rawAccumulated && onFiltered) {\n        onFiltered({\n          messageId: chunk.messageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n\n      emittedFilteredLength = safeFilteredEnd\n\n      const emitChunk = {\n        ...chunk,\n        delta: newDelta,\n        content: filtered.slice(0, safeFilteredEnd),\n      } as StreamChunk\n\n      // `pending` was empty before this push iff `emitChunk` is now the only\n      // entry — return it directly without re-indexing through `pending[0]`.\n      const wasEmpty = pending.length === 0\n      pending.push(emitChunk)\n      return wasEmpty ? emitChunk : pending\n    },\n  }\n}\n"],"names":[],"mappings":"AAyEA,SAAS,WAAW,MAAc,OAAwC;AACxE,MAAI,SAAS;AACb,aAAW,QAAQ,OAAO;AACxB,QAAI,aAAa,MAAM;AACrB,eAAS,OAAO,QAAQ,KAAK,SAAS,KAAK,WAAW;AAAA,IACxD,OAAO;AACL,eAAS,KAAK,GAAG,MAAM;AAAA,IACzB;AAAA,EACF;AACA,SAAO;AACT;AAiBO,SAAS,uBACd,SACgB;AAChB,QAAM;AAAA,IACJ;AAAA,IACA,WAAW;AAAA,IACX,aAAa;AAAA,IACb,eAAe;AAAA,IACf;AAAA,EAAA,IACE;AAEJ,MAAI,aAAa,SAAS;AACxB,WAAO,oBAAoB,OAAO,cAAc,UAAU;AAAA,EAC5D;AACA,SAAO,uBAAuB,OAAO,YAAY,cAAc,UAAU;AAC3E;AAEA,SAAS,oBACP,OACA,cACA,YACgB;AAChB,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,QAAQ,MAA6B,OAAoB;AACvD,UAAI,MAAM,SAAS,uBAAwB;AAE3C,YAAM,WAAW,MAAM;AACvB,YAAM,WAAW,WAAW,UAAU,KAAK;AAE3C,UAAI,aAAa,SAAU;AAE3B,UAAI,YAAY;AACd,mBAAW;AAAA,UACT,WAAW,MAAM;AAAA,UACjB;AAAA,UACA;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AAEA,UAAI,aAAc,QAAO;AAMzB,YAAM,EAAE,SAAS,kBAAkB,GAAG,SAAS;AAE/C,aAAO;AAAA,QACL,GAAG;AAAA,QACH,OAAO;AAAA,MAAA;AAAA,IAEX;AAAA,EAAA;AAEJ;AAEA,SAAS,uBACP,OACA,YACA,cACA,YACgB;AAChB,MAAI,iBAAiB;AACrB,MAAI,wBAAwB;AAC5B,MAAI,gBAAgB;AAEpB,WAAS,aAAa;AACpB,qBAAiB;AACjB,4BAAwB;AACxB,oBAAgB;AAAA,EAClB;AAEA,WAAS,cAAkC;AACzC,QAAI,eAAe,WAAW,EAAG,QAAO;AAExC,UAAM,WAAW,WAAW,gBAAgB,KAAK;AAEjD,QAAI,gBAAgB,aAAa,gBAAgB;AAC/C,UAAI,YAAY;AACd,mBAAW;AAAA,UACT,WAAW;AAAA,UACX,UAAU;AAAA,UACV;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AACA,iBAAA;AACA,aAAO;AAAA,IACT;AAEA,UAAM,YAAY,SAAS,MAAM,qBAAqB;AACtD,QAAI,UAAU,SAAS,GAAG;AACxB,UAAI,aAAa,kBAAkB,YAAY;AAC7C,mBAAW;AAAA,UACT,WAAW;AAAA,UACX,UAAU;AAAA,UACV;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AAEA,YAAM,UAAU;AAAA,QACd,MAAM;AAAA,QACN,WAAW;AAAA,QACX,OAAO;AAAA,QACP,SAAS;AAAA,QACT,WAAW,KAAK,IAAA;AAAA,MAAI;AAGtB,iBAAA;AACA,aAAO;AAAA,IACT;AAEA,eAAA;AACA,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,UAAU;AACR,iBAAA;AAAA,IACF;AAAA,IAEA,QAAQ,MAA6B,OAAoB;AAEvD,UAAI,MAAM,SAAS,sBAAsB,MAAM,SAAS,gBAAgB;AACtE,cAAM,UAAU,YAAA;AAChB,YAAI,QAAS,QAAO,CAAC,SAAS,KAAK;AACnC;AAAA,MACF;AAEA,UAAI,MAAM,SAAS,uBAAwB;AAG3C,YAAM,UAA8B,CAAA;AACpC,UAAI,iBAAiB,MAAM,cAAc,eAAe;AACtD,cAAM,UAAU,YAAA;AAChB,YAAI,QAAS,SAAQ,KAAK,OAAO;AAAA,MACnC;AAEA,wBAAkB,MAAM;AACxB,sBAAgB,MAAM;AAGtB,YAAM,WAAW,WAAW,gBAAgB,KAAK;AACjD,YAAM,kBAAkB,KAAK,IAAI,GAAG,SAAS,SAAS,UAAU;AAEhE,UAAI,mBAAmB,uBAAuB;AAC5C,eAAO,QAAQ,SAAS,IAAI,UAAU;AAAA,MACxC;AAEA,UAAI,gBAAgB,aAAa,gBAAgB;AAC/C,YAAI,YAAY;AACd,qBAAW;AAAA,YACT,WAAW,MAAM;AAAA,YACjB,UAAU;AAAA,YACV;AAAA,YACA,UAAU;AAAA,UAAA,CACX;AAAA,QACH;AACA,eAAO,QAAQ,SAAS,IAAI,UAAU;AAAA,MACxC;AAEA,YAAM,WAAW,SAAS,MAAM,uBAAuB,eAAe;AAEtE,UAAI,aAAa,kBAAkB,YAAY;AAC7C,mBAAW;AAAA,UACT,WAAW,MAAM;AAAA,UACjB,UAAU;AAAA,UACV;AAAA,UACA,UAAU;AAAA,QAAA,CACX;AAAA,MACH;AAEA,8BAAwB;AAExB,YAAM,YAAY;AAAA,QAChB,GAAG;AAAA,QACH,OAAO;AAAA,QACP,SAAS,SAAS,MAAM,GAAG,eAAe;AAAA,MAAA;AAK5C,YAAM,WAAW,QAAQ,WAAW;AACpC,cAAQ,KAAK,SAAS;AACtB,aAAO,WAAW,YAAY;AAAA,IAChC;AAAA,EAAA;AAEJ;"}
+{"version":3,"file":"content-guard.js","names":[],"sources":["../../../src/middlewares/content-guard.ts"],"sourcesContent":["import type {\n  ChatMiddleware,\n  ChatMiddlewareContext,\n} from '../activities/chat/middleware/types'\nimport type { StreamChunk } from '../types'\n\n/**\n * A content guard rule — either a regex pattern with replacement, or a transform function.\n */\nexport type ContentGuardRule =\n  | { pattern: RegExp; replacement: string }\n  | { fn: (text: string) => string }\n\n/**\n * Information passed to the onFiltered callback.\n */\nexport interface ContentFilteredInfo {\n  /** The message ID being filtered */\n  messageId: string\n  /** The original text before filtering */\n  original: string\n  /** The filtered text after rules applied */\n  filtered: string\n  /** Which strategy was used */\n  strategy: 'delta' | 'buffered'\n}\n\n/**\n * Options for the content guard middleware.\n */\nexport interface ContentGuardMiddlewareOptions {\n  /**\n   * Rules to apply to text content. Each rule is either a regex pattern\n   * with a replacement string, or a custom transform function.\n   * Rules are applied in order. Each rule receives the output of the previous.\n   */\n  rules: Array<ContentGuardRule>\n\n  /**\n   * Matching strategy:\n   * - 'delta': Apply rules to each delta as it arrives. Fast, real-time,\n   *   but patterns spanning chunk boundaries may be missed.\n   * - 'buffered': Accumulate content and apply rules to settled portions,\n   *   holding back a look-behind buffer to catch cross-boundary patterns.\n   *\n   * @default 'buffered'\n   */\n  strategy?: 'delta' | 'buffered'\n\n  /**\n   * Number of characters to hold back before emitting (buffered strategy only).\n   * Should be at least as long as the longest pattern you expect to match.\n   * Buffer is flushed when the stream ends.\n   *\n   * @default 50\n   */\n  bufferSize?: number\n\n  /**\n   * If true, drop the entire chunk when any rule changes the content.\n   * @default false\n   */\n  blockOnMatch?: boolean\n\n  /**\n   * Callback when content is filtered by any rule.\n   */\n  onFiltered?: (info: ContentFilteredInfo) => void\n}\n\n/**\n * Apply all rules to a string, returning the transformed result.\n */\nfunction applyRules(text: string, rules: Array<ContentGuardRule>): string {\n  let result = text\n  for (const rule of rules) {\n    if ('pattern' in rule) {\n      result = result.replace(rule.pattern, rule.replacement)\n    } else {\n      result = rule.fn(result)\n    }\n  }\n  return result\n}\n\n/**\n * Creates a middleware that filters or transforms streamed text content.\n *\n * @example\n * ```ts\n * import { contentGuardMiddleware } from '@tanstack/ai/middlewares'\n *\n * const guard = contentGuardMiddleware({\n *   rules: [\n *     { pattern: /\\b\\d{3}-\\d{2}-\\d{4}\\b/g, replacement: '[SSN REDACTED]' },\n *   ],\n *   strategy: 'buffered',\n * })\n * ```\n */\nexport function contentGuardMiddleware(\n  options: ContentGuardMiddlewareOptions,\n): ChatMiddleware {\n  const {\n    rules,\n    strategy = 'buffered',\n    bufferSize = 50,\n    blockOnMatch = false,\n    onFiltered,\n  } = options\n\n  if (strategy === 'delta') {\n    return createDeltaStrategy(rules, blockOnMatch, onFiltered)\n  }\n  return createBufferedStrategy(rules, bufferSize, blockOnMatch, onFiltered)\n}\n\nfunction createDeltaStrategy(\n  rules: Array<ContentGuardRule>,\n  blockOnMatch: boolean,\n  onFiltered?: (info: ContentFilteredInfo) => void,\n): ChatMiddleware {\n  return {\n    name: 'content-guard',\n\n    onChunk(_ctx: ChatMiddlewareContext, chunk: StreamChunk) {\n      if (chunk.type !== 'TEXT_MESSAGE_CONTENT') return\n\n      const original = chunk.delta\n      const filtered = applyRules(original, rules)\n\n      if (filtered === original) return // unchanged, pass through\n\n      if (onFiltered) {\n        onFiltered({\n          messageId: chunk.messageId,\n          original,\n          filtered,\n          strategy: 'delta',\n        })\n      }\n\n      if (blockOnMatch) return null // drop chunk\n\n      // Strip out the previous `content` field by destructuring it away — with\n      // `exactOptionalPropertyTypes` we can't assign `content: undefined`\n      // against `content?: string`. The replacement event carries only the\n      // filtered delta.\n      const { content: _strippedContent, ...rest } = chunk\n      void _strippedContent\n      return {\n        ...rest,\n        delta: filtered,\n      }\n    },\n  }\n}\n\nfunction createBufferedStrategy(\n  rules: Array<ContentGuardRule>,\n  bufferSize: number,\n  blockOnMatch: boolean,\n  onFiltered?: (info: ContentFilteredInfo) => void,\n): ChatMiddleware {\n  let rawAccumulated = ''\n  let emittedFilteredLength = 0\n  let lastMessageId = ''\n\n  function resetState() {\n    rawAccumulated = ''\n    emittedFilteredLength = 0\n    lastMessageId = ''\n  }\n\n  function flushBuffer(): StreamChunk | null {\n    if (rawAccumulated.length === 0) return null\n\n    const filtered = applyRules(rawAccumulated, rules)\n\n    if (blockOnMatch && filtered !== rawAccumulated) {\n      if (onFiltered) {\n        onFiltered({\n          messageId: lastMessageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n      resetState()\n      return null\n    }\n\n    const remaining = filtered.slice(emittedFilteredLength)\n    if (remaining.length > 0) {\n      if (filtered !== rawAccumulated && onFiltered) {\n        onFiltered({\n          messageId: lastMessageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n\n      const flushed = {\n        type: 'TEXT_MESSAGE_CONTENT',\n        messageId: lastMessageId,\n        delta: remaining,\n        content: filtered,\n        timestamp: Date.now(),\n      } as StreamChunk\n\n      resetState()\n      return flushed\n    }\n\n    resetState()\n    return null\n  }\n\n  return {\n    name: 'content-guard',\n\n    onStart() {\n      resetState()\n    },\n\n    onChunk(_ctx: ChatMiddlewareContext, chunk: StreamChunk) {\n      // Flush buffer on stream end events\n      if (chunk.type === 'TEXT_MESSAGE_END' || chunk.type === 'RUN_FINISHED') {\n        const flushed = flushBuffer()\n        if (flushed) return [flushed, chunk]\n        return // pass through end event\n      }\n\n      if (chunk.type !== 'TEXT_MESSAGE_CONTENT') return // pass through\n\n      // Flush buffer on message boundary change\n      const pending: Array<StreamChunk> = []\n      if (lastMessageId && chunk.messageId !== lastMessageId) {\n        const flushed = flushBuffer()\n        if (flushed) pending.push(flushed)\n      }\n\n      rawAccumulated += chunk.delta\n      lastMessageId = chunk.messageId\n\n      // Apply rules to full accumulated text, buffer in filtered space\n      const filtered = applyRules(rawAccumulated, rules)\n      const safeFilteredEnd = Math.max(0, filtered.length - bufferSize)\n\n      if (safeFilteredEnd <= emittedFilteredLength) {\n        return pending.length > 0 ? pending : null\n      }\n\n      if (blockOnMatch && filtered !== rawAccumulated) {\n        if (onFiltered) {\n          onFiltered({\n            messageId: chunk.messageId,\n            original: rawAccumulated,\n            filtered,\n            strategy: 'buffered',\n          })\n        }\n        return pending.length > 0 ? pending : null\n      }\n\n      const newDelta = filtered.slice(emittedFilteredLength, safeFilteredEnd)\n\n      if (filtered !== rawAccumulated && onFiltered) {\n        onFiltered({\n          messageId: chunk.messageId,\n          original: rawAccumulated,\n          filtered,\n          strategy: 'buffered',\n        })\n      }\n\n      emittedFilteredLength = safeFilteredEnd\n\n      const emitChunk = {\n        ...chunk,\n        delta: newDelta,\n        content: filtered.slice(0, safeFilteredEnd),\n      } as StreamChunk\n\n      // `pending` was empty before this push iff `emitChunk` is now the only\n      // entry — return it directly without re-indexing through `pending[0]`.\n      const wasEmpty = pending.length === 0\n      pending.push(emitChunk)\n      return wasEmpty ? emitChunk : pending\n    },\n  }\n}\n"],"mappings":";;;;AAyEA,SAAS,WAAW,MAAc,OAAwC;CACxE,IAAI,SAAS;CACb,KAAK,MAAM,QAAQ,OACjB,IAAI,aAAa,MACf,SAAS,OAAO,QAAQ,KAAK,SAAS,KAAK,WAAW;MAEtD,SAAS,KAAK,GAAG,MAAM;CAG3B,OAAO;AACT;;;;;;;;;;;;;;;;AAiBA,SAAgB,uBACd,SACgB;CAChB,MAAM,EACJ,OACA,WAAW,YACX,aAAa,IACb,eAAe,OACf,eACE;CAEJ,IAAI,aAAa,SACf,OAAO,oBAAoB,OAAO,cAAc,UAAU;CAE5D,OAAO,uBAAuB,OAAO,YAAY,cAAc,UAAU;AAC3E;AAEA,SAAS,oBACP,OACA,cACA,YACgB;CAChB,OAAO;EACL,MAAM;EAEN,QAAQ,MAA6B,OAAoB;GACvD,IAAI,MAAM,SAAS,wBAAwB;GAE3C,MAAM,WAAW,MAAM;GACvB,MAAM,WAAW,WAAW,UAAU,KAAK;GAE3C,IAAI,aAAa,UAAU;GAE3B,IAAI,YACF,WAAW;IACT,WAAW,MAAM;IACjB;IACA;IACA,UAAU;GACZ,CAAC;GAGH,IAAI,cAAc,OAAO;GAMzB,MAAM,EAAE,SAAS,kBAAkB,GAAG,SAAS;GAE/C,OAAO;IACL,GAAG;IACH,OAAO;GACT;EACF;CACF;AACF;AAEA,SAAS,uBACP,OACA,YACA,cACA,YACgB;CAChB,IAAI,iBAAiB;CACrB,IAAI,wBAAwB;CAC5B,IAAI,gBAAgB;CAEpB,SAAS,aAAa;EACpB,iBAAiB;EACjB,wBAAwB;EACxB,gBAAgB;CAClB;CAEA,SAAS,cAAkC;EACzC,IAAI,eAAe,WAAW,GAAG,OAAO;EAExC,MAAM,WAAW,WAAW,gBAAgB,KAAK;EAEjD,IAAI,gBAAgB,aAAa,gBAAgB;GAC/C,IAAI,YACF,WAAW;IACT,WAAW;IACX,UAAU;IACV;IACA,UAAU;GACZ,CAAC;GAEH,WAAW;GACX,OAAO;EACT;EAEA,MAAM,YAAY,SAAS,MAAM,qBAAqB;EACtD,IAAI,UAAU,SAAS,GAAG;GACxB,IAAI,aAAa,kBAAkB,YACjC,WAAW;IACT,WAAW;IACX,UAAU;IACV;IACA,UAAU;GACZ,CAAC;GAGH,MAAM,UAAU;IACd,MAAM;IACN,WAAW;IACX,OAAO;IACP,SAAS;IACT,WAAW,KAAK,IAAI;GACtB;GAEA,WAAW;GACX,OAAO;EACT;EAEA,WAAW;EACX,OAAO;CACT;CAEA,OAAO;EACL,MAAM;EAEN,UAAU;GACR,WAAW;EACb;EAEA,QAAQ,MAA6B,OAAoB;GAEvD,IAAI,MAAM,SAAS,sBAAsB,MAAM,SAAS,gBAAgB;IACtE,MAAM,UAAU,YAAY;IAC5B,IAAI,SAAS,OAAO,CAAC,SAAS,KAAK;IACnC;GACF;GAEA,IAAI,MAAM,SAAS,wBAAwB;GAG3C,MAAM,UAA8B,CAAC;GACrC,IAAI,iBAAiB,MAAM,cAAc,eAAe;IACtD,MAAM,UAAU,YAAY;IAC5B,IAAI,SAAS,QAAQ,KAAK,OAAO;GACnC;GAEA,kBAAkB,MAAM;GACxB,gBAAgB,MAAM;GAGtB,MAAM,WAAW,WAAW,gBAAgB,KAAK;GACjD,MAAM,kBAAkB,KAAK,IAAI,GAAG,SAAS,SAAS,UAAU;GAEhE,IAAI,mBAAmB,uBACrB,OAAO,QAAQ,SAAS,IAAI,UAAU;GAGxC,IAAI,gBAAgB,aAAa,gBAAgB;IAC/C,IAAI,YACF,WAAW;KACT,WAAW,MAAM;KACjB,UAAU;KACV;KACA,UAAU;IACZ,CAAC;IAEH,OAAO,QAAQ,SAAS,IAAI,UAAU;GACxC;GAEA,MAAM,WAAW,SAAS,MAAM,uBAAuB,eAAe;GAEtE,IAAI,aAAa,kBAAkB,YACjC,WAAW;IACT,WAAW,MAAM;IACjB,UAAU;IACV;IACA,UAAU;GACZ,CAAC;GAGH,wBAAwB;GAExB,MAAM,YAAY;IAChB,GAAG;IACH,OAAO;IACP,SAAS,SAAS,MAAM,GAAG,eAAe;GAC5C;GAIA,MAAM,WAAW,QAAQ,WAAW;GACpC,QAAQ,KAAK,SAAS;GACtB,OAAO,WAAW,YAAY;EAChC;CACF;AACF"}

package/dist/esm/middlewares/index.js

@@ -1,7 +1,3 @@
-import { contentGuardMiddleware } from "./content-guard.js";
 import { toolCacheMiddleware } from "../activities/chat/middleware/tool-cache-middleware.js";
-export {
-  contentGuardMiddleware,
-  toolCacheMiddleware
-};
-//# sourceMappingURL=index.js.map
+import { contentGuardMiddleware } from "./content-guard.js";
+export { contentGuardMiddleware, toolCacheMiddleware };

package/dist/esm/middlewares/otel.d.ts

@@ -5,7 +5,9 @@ import { GenerationMiddleware, GenerationMiddlewareContext } from '../activities
  * Scope (role) of an OTel span emitted by this middleware.
  *
  * - `chat` — the root span for a single `chat()` call
- * - `iteration` — one per agent-loop iteration (one model call)
+ * - `iteration` — one per provider model call (agent-loop `beforeModel`
+ *   turn, or the separate `structuredOutput` finalization when
+ *   `outputSchema` skips the agent loop — see #1054)
  * - `tool` — one per tool execution inside an iteration
  * - `generation` — the single span for a media activity call
  *   (`generateImage`, `generateVideo`, `generateSpeech`, …)