@tanstack/ai 0.35.0 → 0.36.0

package/dist/esm/logger/resolve.js

@@ -1,54 +0,0 @@
-import { ConsoleLogger } from "./console-logger.js";
-import { InternalLogger } from "./internal-logger.js";
-const ALL_OFF = {
-  provider: false,
-  output: false,
-  middleware: false,
-  tools: false,
-  agentLoop: false,
-  config: false,
-  errors: false,
-  request: false
-};
-const ALL_ON = {
-  provider: true,
-  output: true,
-  middleware: true,
-  tools: true,
-  agentLoop: true,
-  config: true,
-  errors: true,
-  request: true
-};
-const errorsOnlyCategories = () => ({
-  ...ALL_OFF,
-  errors: true
-});
-const resolveCategoriesFromPartial = (partial) => ({
-  provider: partial.provider ?? true,
-  output: partial.output ?? true,
-  middleware: partial.middleware ?? true,
-  tools: partial.tools ?? true,
-  agentLoop: partial.agentLoop ?? true,
-  config: partial.config ?? true,
-  errors: partial.errors ?? true,
-  request: partial.request ?? true
-});
-function resolveDebugOption(debug) {
-  if (debug === void 0) {
-    return new InternalLogger(new ConsoleLogger(), errorsOnlyCategories());
-  }
-  if (debug === true) {
-    return new InternalLogger(new ConsoleLogger(), ALL_ON);
-  }
-  if (debug === false) {
-    return new InternalLogger(new ConsoleLogger(), ALL_OFF);
-  }
-  const { logger, ...cats } = debug;
-  const userLogger = logger ?? new ConsoleLogger();
-  return new InternalLogger(userLogger, resolveCategoriesFromPartial(cats));
-}
-export {
-  resolveDebugOption
-};
-//# sourceMappingURL=resolve.js.map

package/dist/esm/logger/resolve.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"resolve.js","sources":["../../../src/logger/resolve.ts"],"sourcesContent":["import { ConsoleLogger } from './console-logger'\nimport { InternalLogger } from './internal-logger'\nimport type { ResolvedCategories } from './internal-logger'\nimport type { DebugCategories, DebugConfig, DebugOption, Logger } from './types'\n\nconst ALL_OFF: ResolvedCategories = {\n  provider: false,\n  output: false,\n  middleware: false,\n  tools: false,\n  agentLoop: false,\n  config: false,\n  errors: false,\n  request: false,\n}\n\nconst ALL_ON: ResolvedCategories = {\n  provider: true,\n  output: true,\n  middleware: true,\n  tools: true,\n  agentLoop: true,\n  config: true,\n  errors: true,\n  request: true,\n}\n\nconst errorsOnlyCategories = (): ResolvedCategories => ({\n  ...ALL_OFF,\n  errors: true,\n})\n\nconst resolveCategoriesFromPartial = (\n  partial: DebugCategories,\n): ResolvedCategories => ({\n  provider: partial.provider ?? true,\n  output: partial.output ?? true,\n  middleware: partial.middleware ?? true,\n  tools: partial.tools ?? true,\n  agentLoop: partial.agentLoop ?? true,\n  config: partial.config ?? true,\n  errors: partial.errors ?? true,\n  request: partial.request ?? true,\n})\n\n/**\n * Normalize a `DebugOption` into an `InternalLogger` ready to be threaded\n * through the library's activities and adapters. See the `DebugOption`\n * resolution table in the spec for the complete rules.\n *\n * - `undefined`: only the `errors` category is enabled; default `ConsoleLogger`.\n * - `true`: all categories enabled; default `ConsoleLogger`.\n * - `false`: all categories disabled (including `errors`); default `ConsoleLogger`.\n * - `DebugConfig`: each unspecified category defaults to `true`; an optional\n *   `logger` replaces the default `ConsoleLogger`.\n */\nexport function resolveDebugOption(\n  debug: DebugOption | undefined,\n): InternalLogger {\n  if (debug === undefined) {\n    return new InternalLogger(new ConsoleLogger(), errorsOnlyCategories())\n  }\n  if (debug === true) {\n    return new InternalLogger(new ConsoleLogger(), ALL_ON)\n  }\n  if (debug === false) {\n    return new InternalLogger(new ConsoleLogger(), ALL_OFF)\n  }\n  const { logger, ...cats }: DebugConfig = debug\n  const userLogger: Logger = logger ?? new ConsoleLogger()\n  return new InternalLogger(userLogger, resolveCategoriesFromPartial(cats))\n}\n"],"names":[],"mappings":";;AAKA,MAAM,UAA8B;AAAA,EAClC,UAAU;AAAA,EACV,QAAQ;AAAA,EACR,YAAY;AAAA,EACZ,OAAO;AAAA,EACP,WAAW;AAAA,EACX,QAAQ;AAAA,EACR,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,MAAM,SAA6B;AAAA,EACjC,UAAU;AAAA,EACV,QAAQ;AAAA,EACR,YAAY;AAAA,EACZ,OAAO;AAAA,EACP,WAAW;AAAA,EACX,QAAQ;AAAA,EACR,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,MAAM,uBAAuB,OAA2B;AAAA,EACtD,GAAG;AAAA,EACH,QAAQ;AACV;AAEA,MAAM,+BAA+B,CACnC,aACwB;AAAA,EACxB,UAAU,QAAQ,YAAY;AAAA,EAC9B,QAAQ,QAAQ,UAAU;AAAA,EAC1B,YAAY,QAAQ,cAAc;AAAA,EAClC,OAAO,QAAQ,SAAS;AAAA,EACxB,WAAW,QAAQ,aAAa;AAAA,EAChC,QAAQ,QAAQ,UAAU;AAAA,EAC1B,QAAQ,QAAQ,UAAU;AAAA,EAC1B,SAAS,QAAQ,WAAW;AAC9B;AAaO,SAAS,mBACd,OACgB;AAChB,MAAI,UAAU,QAAW;AACvB,WAAO,IAAI,eAAe,IAAI,cAAA,GAAiB,sBAAsB;AAAA,EACvE;AACA,MAAI,UAAU,MAAM;AAClB,WAAO,IAAI,eAAe,IAAI,cAAA,GAAiB,MAAM;AAAA,EACvD;AACA,MAAI,UAAU,OAAO;AACnB,WAAO,IAAI,eAAe,IAAI,cAAA,GAAiB,OAAO;AAAA,EACxD;AACA,QAAM,EAAE,QAAQ,GAAG,KAAA,IAAsB;AACzC,QAAM,aAAqB,UAAU,IAAI,cAAA;AACzC,SAAO,IAAI,eAAe,YAAY,6BAA6B,IAAI,CAAC;AAC1E;"}

package/dist/esm/logger/types.d.ts

@@ -1,75 +0,0 @@
-/**
- * Pluggable logger interface consumed by every `@tanstack/ai` activity when `debug` is enabled. Supply a custom implementation via `debug: { logger }` on `chat()`, `summarize()`, `generateImage()`, etc. The four methods correspond to log levels: use `debug` for chunk-level diagnostic output, `info`/`warn` for notable events, `error` for caught exceptions.
- */
-export interface Logger {
-    /**
-     * Called for chunk-level diagnostic output (raw provider chunks, per-chunk output, agent-loop iteration markers).
-     * @param meta Structured data forwarded to the underlying logger. Loggers like pino will preserve this as a structured record; the default `ConsoleLogger` renders it with a runtime-appropriate strategy (depth-unlimited `console.dir` on Node, JSON appended to the message on Cloudflare Workers, a second `console.<level>` argument elsewhere).
-     */
-    debug: (message: string, meta?: Record<string, unknown>) => void;
-    /**
-     * Called for notable informational events (outgoing requests, tool invocations, middleware transitions).
-     * @param meta Structured data forwarded to the underlying logger. Loggers like pino will preserve this as a structured record; the default `ConsoleLogger` renders it with a runtime-appropriate strategy (depth-unlimited `console.dir` on Node, JSON appended to the message on Cloudflare Workers, a second `console.<level>` argument elsewhere).
-     */
-    info: (message: string, meta?: Record<string, unknown>) => void;
-    /**
-     * Called for notable warnings that don't halt execution (deprecations, recoverable anomalies).
-     * @param meta Structured data forwarded to the underlying logger. Loggers like pino will preserve this as a structured record; the default `ConsoleLogger` renders it with a runtime-appropriate strategy (depth-unlimited `console.dir` on Node, JSON appended to the message on Cloudflare Workers, a second `console.<level>` argument elsewhere).
-     */
-    warn: (message: string, meta?: Record<string, unknown>) => void;
-    /**
-     * Called for caught exceptions throughout the pipeline.
-     * @param meta Structured data forwarded to the underlying logger. Loggers like pino will preserve this as a structured record; the default `ConsoleLogger` renders it with a runtime-appropriate strategy (depth-unlimited `console.dir` on Node, JSON appended to the message on Cloudflare Workers, a second `console.<level>` argument elsewhere).
-     */
-    error: (message: string, meta?: Record<string, unknown>) => void;
-}
-/**
- * Per-category toggles for debug logging. Each flag enables or disables one class of log message. Unspecified flags default to `true` when `DebugConfig` is partially specified; `undefined` on the `debug` option defaults all flags to `false` except `errors`.
- */
-export interface DebugCategories {
-    /**
-     * Raw chunks/frames received from a provider SDK (OpenAI, Anthropic, Gemini, Ollama, Grok, Groq, OpenRouter, fal, ElevenLabs). Emitted inside every streaming adapter's chunk loop.
-     */
-    provider?: boolean;
-    /**
-     * Chunks/results yielded to the consumer after all middleware. For streaming activities this fires per chunk; for non-streaming activities it fires once per result.
-     */
-    output?: boolean;
-    /**
-     * Inputs and outputs around each middleware hook invocation. Chat-only.
-     */
-    middleware?: boolean;
-    /**
-     * Before/after tool-call execution in the chat agent loop. Chat-only.
-     */
-    tools?: boolean;
-    /**
-     * Iteration markers and phase transitions in the chat agent loop. Chat-only.
-     */
-    agentLoop?: boolean;
-    /**
-     * Config transforms returned by middleware `onConfig` hooks. Chat-only.
-     */
-    config?: boolean;
-    /**
-     * Caught errors throughout the pipeline. Unlike other categories, defaults to `true` even when `debug` is unspecified. Explicitly set `errors: false` or `debug: false` to silence.
-     */
-    errors?: boolean;
-    /**
-     * Outgoing call metadata (provider, model, message/tool counts) emitted before each adapter SDK call.
-     */
-    request?: boolean;
-}
-/**
- * Granular debug configuration combining per-category toggles with an optional custom logger. Any unspecified category flag defaults to `true`.
- */
-export interface DebugConfig extends DebugCategories {
-    /**
-     * Custom `Logger` implementation. When omitted, a default `ConsoleLogger` routes output to `console.debug`/`info`/`warn`/`error`.
-     */
-    logger?: Logger;
-}
-/**
- * The shape accepted by the `debug` option on every `@tanstack/ai` activity. Pass `true` to enable all categories with the default console logger; `false` to silence everything including errors; an object for granular control.
- */
-export type DebugOption = boolean | DebugConfig;

package/dist/esm/middlewares/content-guard.d.ts

@@ -1,77 +0,0 @@
-import { ChatMiddleware } from '../activities/chat/middleware/types.js';
-/**
- * A content guard rule — either a regex pattern with replacement, or a transform function.
- */
-export type ContentGuardRule = {
-    pattern: RegExp;
-    replacement: string;
-} | {
-    fn: (text: string) => string;
-};
-/**
- * Information passed to the onFiltered callback.
- */
-export interface ContentFilteredInfo {
-    /** The message ID being filtered */
-    messageId: string;
-    /** The original text before filtering */
-    original: string;
-    /** The filtered text after rules applied */
-    filtered: string;
-    /** Which strategy was used */
-    strategy: 'delta' | 'buffered';
-}
-/**
- * Options for the content guard middleware.
- */
-export interface ContentGuardMiddlewareOptions {
-    /**
-     * Rules to apply to text content. Each rule is either a regex pattern
-     * with a replacement string, or a custom transform function.
-     * Rules are applied in order. Each rule receives the output of the previous.
-     */
-    rules: Array<ContentGuardRule>;
-    /**
-     * Matching strategy:
-     * - 'delta': Apply rules to each delta as it arrives. Fast, real-time,
-     *   but patterns spanning chunk boundaries may be missed.
-     * - 'buffered': Accumulate content and apply rules to settled portions,
-     *   holding back a look-behind buffer to catch cross-boundary patterns.
-     *
-     * @default 'buffered'
-     */
-    strategy?: 'delta' | 'buffered';
-    /**
-     * Number of characters to hold back before emitting (buffered strategy only).
-     * Should be at least as long as the longest pattern you expect to match.
-     * Buffer is flushed when the stream ends.
-     *
-     * @default 50
-     */
-    bufferSize?: number;
-    /**
-     * If true, drop the entire chunk when any rule changes the content.
-     * @default false
-     */
-    blockOnMatch?: boolean;
-    /**
-     * Callback when content is filtered by any rule.
-     */
-    onFiltered?: (info: ContentFilteredInfo) => void;
-}
-/**
- * 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',
- * })
- * ```
- */
-export declare function contentGuardMiddleware(options: ContentGuardMiddlewareOptions): ChatMiddleware;

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

@@ -1,156 +0,0 @@
-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;
-}
-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);
-}
-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
-      };
-    }
-  };
-}
-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;
-    }
-  };
-}
-export {
-  contentGuardMiddleware
-};
-//# sourceMappingURL=content-guard.js.map

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

@@ -1 +0,0 @@
-{"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;"}

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

@@ -1,2 +0,0 @@
-export { toolCacheMiddleware, type ToolCacheMiddlewareOptions, type ToolCacheStorage, type ToolCacheEntry, } from './tool-cache.js';
-export { contentGuardMiddleware, type ContentGuardMiddlewareOptions, type ContentGuardRule, type ContentFilteredInfo, } from './content-guard.js';

package/dist/esm/middlewares/index.js

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

package/dist/esm/middlewares/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;"}

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

@@ -1,81 +0,0 @@
-import { AttributeValue, Meter, Span, SpanOptions, Tracer } from '@opentelemetry/api';
-import { ChatMiddleware, ChatMiddlewareContext } from '../activities/chat/middleware/types.js';
-import { GenerationMiddleware, GenerationMiddlewareContext } from '../activities/middleware/types.js';
-/**
- * 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)
- * - `tool` — one per tool execution inside an iteration
- * - `generation` — the single span for a media activity call
- *   (`generateImage`, `generateVideo`, `generateSpeech`, …)
- */
-export type OtelSpanScope = 'chat' | 'iteration' | 'tool' | 'generation';
-/**
- * Alias retained for backwards compatibility. Prefer {@link OtelSpanScope}.
- *
- * @deprecated Use `OtelSpanScope` instead — the name shadows OTel's built-in
- * `SpanKind` which is also imported by integrations of this middleware.
- */
-export type OtelSpanKind = OtelSpanScope;
-/**
- * Span metadata passed to `spanNameFormatter`, `attributeEnricher`,
- * `onBeforeSpanStart`, and `onSpanEnd`. Discriminated by `kind` so that
- * tool-only fields narrow automatically inside callback bodies.
- */
-export type OtelSpanInfo<TScope extends OtelSpanScope = OtelSpanScope> = TScope extends 'chat' ? {
-    kind: 'chat';
-    ctx: ChatMiddlewareContext;
-} : TScope extends 'iteration' ? {
-    kind: 'iteration';
-    ctx: ChatMiddlewareContext;
-    iteration: number;
-} : TScope extends 'tool' ? {
-    kind: 'tool';
-    ctx: ChatMiddlewareContext;
-    iteration: number;
-    toolName: string;
-    toolCallId: string;
-} : TScope extends 'generation' ? {
-    kind: 'generation';
-    ctx: GenerationMiddlewareContext;
-} : never;
-export interface OtelMiddlewareOptions {
-    /** OTel `Tracer` used to start root, iteration, and tool spans. */
-    tracer: Tracer;
-    /**
-     * Optional OTel `Meter`. When provided, the middleware records
-     * `gen_ai.client.operation.duration` and `gen_ai.client.token.usage`
-     * histograms. Omit to disable metrics without disabling tracing.
-     */
-    meter?: Meter;
-    /**
-     * When `true`, prompt and completion content is attached to iteration spans
-     * as `gen_ai.*.message` / `gen_ai.choice` events. Defaults to `false` so
-     * that PII never lands on a span by accident.
-     */
-    captureContent?: boolean;
-    /**
-     * Invoked on every captured content string before it lands on a span.
-     * Return a redacted version. If this function throws, the middleware emits
-     * the literal sentinel `"[redaction_failed]"` instead of the original text
-     * — it never falls back to raw content.
-     */
-    redact?: (text: string) => string;
-    /**
-     * Maximum characters kept in the per-iteration assistant text buffer used
-     * to emit `gen_ai.choice` events. Extra characters are truncated with a
-     * trailing `"…"` marker. Defaults to 100 000. Set to `0` to disable the
-     * cap. Exporters typically truncate long attribute values anyway.
-     */
-    maxContentLength?: number;
-    /** Override the default span name for each `kind`. */
-    spanNameFormatter?: (info: OtelSpanInfo) => string;
-    /** Add extra attributes to each span. */
-    attributeEnricher?: (info: OtelSpanInfo) => Record<string, AttributeValue>;
-    /** Mutate `SpanOptions` immediately before `tracer.startSpan(...)`. */
-    onBeforeSpanStart?: (info: OtelSpanInfo, options: SpanOptions) => SpanOptions;
-    /** Fires just before every `span.end()`. */
-    onSpanEnd?: (info: OtelSpanInfo, span: Span) => void;
-}
-export declare function otelMiddleware(options: OtelMiddlewareOptions): GenerationMiddleware & ChatMiddleware;