@tanstack/ai 0.38.0 → 0.39.1

package/dist/esm/activities/chat/middleware/compose.d.ts

@@ -1,6 +1,6 @@
 import { StreamChunk } from '../../../types.js';
 import { InternalLogger } from '../../../logger/internal-logger.js';
-import { AbortInfo, AfterToolCallInfo, BeforeToolCallDecision, ChatMiddleware, ChatMiddlewareConfig, ChatMiddlewareContext, ErrorInfo, FinishInfo, IterationInfo, StructuredOutputMiddlewareConfig, ToolCallHookContext, ToolPhaseCompleteInfo, UsageInfo } from './types.js';
+import { AbortInfo, AfterToolCallInfo, BeforeToolCallDecision, ChatMiddleware, ChatMiddlewareConfig, ChatMiddlewareContext, ErrorInfo, FinishInfo, IterationInfo, SandboxFileEvent, StructuredOutputMiddlewareConfig, ToolCallHookContext, ToolPhaseCompleteInfo, UsageInfo } from './types.js';
 /**
  * Internal middleware runner that manages composed execution of middleware hooks.
  * Created once per chat() invocation.
@@ -49,6 +49,12 @@ export declare class MiddlewareRunner<TContext = unknown> {
      * - null: drop the chunk entirely
      */
     runOnChunk(ctx: ChatMiddlewareContext<TContext>, chunk: StreamChunk): Promise<Array<StreamChunk>>;
+    /**
+     * Dispatch a sandbox file event to every middleware's `sandbox` hooks, in
+     * array order: the catch-all `onFile` then the type-specific hook. Errors are
+     * logged and swallowed so one bad hook can't break the run.
+     */
+    runSandboxFile(ctx: ChatMiddlewareContext<TContext>, event: SandboxFileEvent): Promise<void>;
     /**
      * Run onBeforeToolCall through middleware in order.
      * Returns the first non-void decision, or undefined to continue normally.

package/dist/esm/activities/chat/middleware/compose.js

@@ -287,6 +287,33 @@ class MiddlewareRunner {
     }
     return chunks;
   }
+  /**
+   * Dispatch a sandbox file event to every middleware's `sandbox` hooks, in
+   * array order: the catch-all `onFile` then the type-specific hook. Errors are
+   * logged and swallowed so one bad hook can't break the run.
+   */
+  async runSandboxFile(ctx, event) {
+    const typed = {
+      create: "onFileCreate",
+      change: "onFileChange",
+      delete: "onFileDelete"
+    }[event.type];
+    for (const mw of this.middlewares) {
+      const hooks = mw.sandbox;
+      if (!hooks) continue;
+      for (const fn of [hooks.onFile, hooks[typed]]) {
+        if (!fn) continue;
+        try {
+          await fn(ctx, event);
+        } catch (error) {
+          this.logger.sandbox(
+            `hook=${typed} middleware=${mw.name ?? "unnamed"} threw`,
+            { middleware: mw.name ?? "unnamed", error }
+          );
+        }
+      }
+    }
+  }
   /**
    * Run onBeforeToolCall through middleware in order.
    * Returns the first non-void decision, or undefined to continue normally.

package/dist/esm/activities/chat/middleware/compose.js.map

@@ -1 +1 @@
-{"version":3,"file":"compose.js","sources":["../../../../../src/activities/chat/middleware/compose.ts"],"sourcesContent":["import { aiEventClient } from '@tanstack/ai-event-client'\nimport type { StreamChunk } from '../../../types'\nimport type { InternalLogger } from '../../../logger/internal-logger'\nimport type {\n  AbortInfo,\n  AfterToolCallInfo,\n  BeforeToolCallDecision,\n  ChatMiddleware,\n  ChatMiddlewareConfig,\n  ChatMiddlewareContext,\n  ErrorInfo,\n  FinishInfo,\n  IterationInfo,\n  StructuredOutputMiddlewareConfig,\n  ToolCallHookContext,\n  ToolPhaseCompleteInfo,\n  UsageInfo,\n} from './types'\n\n/** Check if a middleware should be skipped for instrumentation events. */\nfunction shouldSkipInstrumentation(mw: ChatMiddleware<any>): boolean {\n  return mw.name === 'devtools' || mw.name === 'strip-to-spec'\n}\n\n/** Build the base context for middleware instrumentation events. */\nfunction instrumentCtx(ctx: ChatMiddlewareContext<any>) {\n  return {\n    requestId: ctx.requestId,\n    streamId: ctx.streamId,\n    clientId: ctx.threadId,\n    timestamp: Date.now(),\n  }\n}\n\n/**\n * Internal middleware runner that manages composed execution of middleware hooks.\n * Created once per chat() invocation.\n */\nexport class MiddlewareRunner<TContext = unknown> {\n  private readonly middlewares: ReadonlyArray<ChatMiddleware<TContext>>\n  private readonly logger: InternalLogger\n\n  constructor(\n    middlewares: ReadonlyArray<ChatMiddleware<TContext>>,\n    logger: InternalLogger,\n  ) {\n    this.middlewares = middlewares\n    this.logger = logger\n  }\n\n  get hasMiddleware(): boolean {\n    return this.middlewares.length > 0\n  }\n\n  /**\n   * Pipe config through all middleware onConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   */\n  async runOnConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: ChatMiddlewareConfig,\n  ): Promise<ChatMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              changes: result,\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Pipe config through all middleware onStructuredOutputConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   *\n   * Called once at the structured-output boundary, before runOnConfig at the\n   * same boundary (which receives a ChatMiddlewareConfig view, no outputSchema).\n   */\n  async runOnStructuredOutputConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: StructuredOutputMiddlewareConfig,\n  ): Promise<StructuredOutputMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onStructuredOutputConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onStructuredOutputConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStructuredOutputConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              // `result` is `Partial<StructuredOutputMiddlewareConfig>` —\n              // Object.fromEntries(Object.entries(result)) yields the\n              // structural `Record<string, unknown>` the event emitter wants\n              // without an `as` cast.\n              changes: Object.fromEntries(Object.entries(result)),\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Run all `setup` hooks in array order, then assert every declared `provides`\n   * capability was actually provided. Wires the last-wins duplicate-provide\n   * warning into the registry. Runs before init `onConfig`.\n   *\n   * Takes the full `ChatMiddlewareContext` — the same stable context the engine\n   * threads through every other hook — because it both forwards `ctx` to each\n   * `setup` hook and emits instrumentation events from it.\n   */\n  async runSetup(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    ctx.capabilities.setOnDuplicate((name) => {\n      this.logger.warn(\n        `capability \"${name}\" was provided more than once; last provider wins`,\n        { capability: name },\n      )\n    })\n\n    for (const mw of this.middlewares) {\n      if (mw.setup) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.setup(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=setup middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'setup' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'setup',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n\n    for (const mw of this.middlewares) {\n      for (const handle of mw.provides ?? []) {\n        if (!ctx.capabilities.has(handle)) {\n          throw new Error(\n            `Middleware \"${mw.name ?? 'unnamed'}\" declares it provides ` +\n              `\"${handle.capabilityName}\" but never called provide() in setup().`,\n          )\n        }\n      }\n    }\n  }\n\n  /**\n   * Call onStart on all middleware in order.\n   */\n  async runOnStart(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onStart) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onStart(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onStart middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onStart' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStart',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Pipe a single chunk through all middleware onChunk hooks in order.\n   * Returns the resulting chunks (0..N) to yield to the consumer.\n   *\n   * - void: pass through unchanged\n   * - chunk: replace with this chunk\n   * - chunk[]: expand to multiple chunks\n   * - null: drop the chunk entirely\n   */\n  async runOnChunk(\n    ctx: ChatMiddlewareContext<TContext>,\n    chunk: StreamChunk,\n  ): Promise<Array<StreamChunk>> {\n    let chunks: Array<StreamChunk> = [chunk]\n\n    for (const mw of this.middlewares) {\n      if (!mw.onChunk) continue\n      const skip = shouldSkipInstrumentation(mw)\n\n      const nextChunks: Array<StreamChunk> = []\n      for (const c of chunks) {\n        // Cast: @ag-ui/core Zod passthrough types prevent direct `.type` access\n        const chunkType = c.type\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onChunk', in: c },\n          )\n        }\n        const result = await mw.onChunk(ctx, c)\n        if (result === null) {\n          // Drop this chunk\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=<dropped>`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                dropped: true,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 0,\n              wasDropped: true,\n            })\n          }\n          continue\n        } else if (result === undefined) {\n          // Pass through — no instrumentation for pass-throughs\n          nextChunks.push(c)\n        } else if (Array.isArray(result)) {\n          // Expand\n          nextChunks.push(...result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=[${result.map((r: StreamChunk) => r.type).join(',')}]`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: result.length,\n              wasDropped: false,\n            })\n          }\n        } else {\n          // Replace\n          nextChunks.push(result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=${result.type}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 1,\n              wasDropped: false,\n            })\n          }\n        }\n      }\n      chunks = nextChunks\n    }\n\n    return chunks\n  }\n\n  /**\n   * Run onBeforeToolCall through middleware in order.\n   * Returns the first non-void decision, or undefined to continue normally.\n   */\n  async runOnBeforeToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    hookCtx: ToolCallHookContext,\n  ): Promise<BeforeToolCallDecision> {\n    for (const mw of this.middlewares) {\n      if (mw.onBeforeToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const decision = await mw.onBeforeToolCall(ctx, hookCtx)\n        const hasTransform = decision !== undefined && decision !== null\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onBeforeToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onBeforeToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onBeforeToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n        }\n        if (hasTransform) {\n          return decision\n        }\n      }\n    }\n    return undefined\n  }\n\n  /**\n   * Run onAfterToolCall on all middleware in order.\n   */\n  async runOnAfterToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AfterToolCallInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAfterToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAfterToolCall(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAfterToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAfterToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAfterToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onUsage on all middleware in order.\n   */\n  async runOnUsage(\n    ctx: ChatMiddlewareContext<TContext>,\n    usage: UsageInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onUsage) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onUsage(ctx, usage)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onUsage middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onUsage' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onUsage',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onFinish on all middleware in order.\n   */\n  async runOnFinish(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: FinishInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onFinish) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onFinish(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onFinish middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onFinish' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onFinish',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onAbort on all middleware in order.\n   */\n  async runOnAbort(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AbortInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAbort) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAbort(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAbort middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAbort' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAbort',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onError on all middleware in order.\n   */\n  async runOnError(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ErrorInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onError) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onError(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onError middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onError' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onError',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onIteration on all middleware in order.\n   * Called at the start of each agent loop iteration.\n   */\n  async runOnIteration(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: IterationInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onIteration) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onIteration(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onIteration middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onIteration' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onIteration',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onToolPhaseComplete on all middleware in order.\n   * Called after all tool calls in an iteration have been processed.\n   */\n  async runOnToolPhaseComplete(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ToolPhaseCompleteInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onToolPhaseComplete) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onToolPhaseComplete(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onToolPhaseComplete middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onToolPhaseComplete' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onToolPhaseComplete',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n}\n"],"names":[],"mappings":";AAoBA,SAAS,0BAA0B,IAAkC;AACnE,SAAO,GAAG,SAAS,cAAc,GAAG,SAAS;AAC/C;AAGA,SAAS,cAAc,KAAiC;AACtD,SAAO;AAAA,IACL,WAAW,IAAI;AAAA,IACf,UAAU,IAAI;AAAA,IACd,UAAU,IAAI;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,EAAI;AAExB;AAMO,MAAM,iBAAqC;AAAA,EAC/B;AAAA,EACA;AAAA,EAEjB,YACE,aACA,QACA;AACA,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,gBAAyB;AAC3B,WAAO,KAAK,YAAY,SAAS;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,YACJ,KACA,QAC+B;AAC/B,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,SAAS,KAAK,OAAO;AAC7C,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA,cACf,SAAS;AAAA,YAAA,CACV;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,4BACJ,KACA,QAC2C;AAC3C,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,0BAA0B;AAC/B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,yBAAyB,KAAK,OAAO;AAC7D,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA;AAAA;AAAA;AAAA;AAAA,cAKf,SAAS,OAAO,YAAY,OAAO,QAAQ,MAAM,CAAC;AAAA,YAAA,CACnD;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,SAAS,KAAqD;AAClE,QAAI,aAAa,eAAe,CAAC,SAAS;AACxC,WAAK,OAAO;AAAA,QACV,eAAe,IAAI;AAAA,QACnB,EAAE,YAAY,KAAA;AAAA,MAAK;AAAA,IAEvB,CAAC;AAED,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,OAAO;AACZ,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,MAAM,GAAG;AAClB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,yBAAyB,GAAG,QAAQ,SAAS;AAAA,YAC7C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,QAAA;AAAA,UAAQ;AAEpD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAEA,eAAW,MAAM,KAAK,aAAa;AACjC,iBAAW,UAAU,GAAG,YAAY,CAAA,GAAI;AACtC,YAAI,CAAC,IAAI,aAAa,IAAI,MAAM,GAAG;AACjC,gBAAM,IAAI;AAAA,YACR,eAAe,GAAG,QAAQ,SAAS,2BAC7B,OAAO,cAAc;AAAA,UAAA;AAAA,QAE/B;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WAAW,KAAqD;AACpE,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,GAAG;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,WACJ,KACA,OAC6B;AAC7B,QAAI,SAA6B,CAAC,KAAK;AAEvC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,CAAC,GAAG,QAAS;AACjB,YAAM,OAAO,0BAA0B,EAAE;AAEzC,YAAM,aAAiC,CAAA;AACvC,iBAAW,KAAK,QAAQ;AAEtB,cAAM,YAAY,EAAE;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,YAC/D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAW,IAAI,EAAA;AAAA,UAAE;AAAA,QAE/D;AACA,cAAM,SAAS,MAAM,GAAG,QAAQ,KAAK,CAAC;AACtC,YAAI,WAAW,MAAM;AAEnB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,cAC/D;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,SAAS;AAAA,cAAA;AAAA,YACX;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AACA;AAAA,QACF,WAAW,WAAW,QAAW;AAE/B,qBAAW,KAAK,CAAC;AAAA,QACnB,WAAW,MAAM,QAAQ,MAAM,GAAG;AAEhC,qBAAW,KAAK,GAAG,MAAM;AACzB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,SAAS,OAAO,IAAI,CAAC,MAAmB,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC;AAAA,cACxH;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa,OAAO;AAAA,cACpB,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF,OAAO;AAEL,qBAAW,KAAK,MAAM;AACtB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,QAAQ,OAAO,IAAI;AAAA,cAClF;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF;AAAA,MACF;AACA,eAAS;AAAA,IACX;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,oBACJ,KACA,SACiC;AACjC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,kBAAkB;AACvB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,WAAW,MAAM,GAAG,iBAAiB,KAAK,OAAO;AACvD,cAAM,eAAe,aAAa,UAAa,aAAa;AAC5D,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,oCAAoC,GAAG,QAAQ,SAAS;AAAA,YACxD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,mBAAA;AAAA,UAAmB;AAE/D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AAAA,QACH;AACA,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,mBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,iBAAiB;AACtB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,gBAAgB,KAAK,IAAI;AAClC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,mCAAmC,GAAG,QAAQ,SAAS;AAAA,YACvD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,kBAAA;AAAA,UAAkB;AAE9D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,OACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,KAAK;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,SAAS,KAAK,IAAI;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,4BAA4B,GAAG,QAAQ,SAAS;AAAA,YAChD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAA;AAAA,UAAW;AAEvD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,eACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,aAAa;AAClB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,YAAY,KAAK,IAAI;AAC9B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,+BAA+B,GAAG,QAAQ,SAAS;AAAA,YACnD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,cAAA;AAAA,UAAc;AAE1D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,uBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,qBAAqB;AAC1B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,oBAAoB,KAAK,IAAI;AACtC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,uCAAuC,GAAG,QAAQ,SAAS;AAAA,YAC3D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,sBAAA;AAAA,UAAsB;AAElE,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;"}
+{"version":3,"file":"compose.js","sources":["../../../../../src/activities/chat/middleware/compose.ts"],"sourcesContent":["import { aiEventClient } from '@tanstack/ai-event-client'\nimport type { StreamChunk } from '../../../types'\nimport type { InternalLogger } from '../../../logger/internal-logger'\nimport type {\n  AbortInfo,\n  AfterToolCallInfo,\n  BeforeToolCallDecision,\n  ChatMiddleware,\n  ChatMiddlewareConfig,\n  ChatMiddlewareContext,\n  ErrorInfo,\n  FinishInfo,\n  IterationInfo,\n  SandboxFileEvent,\n  StructuredOutputMiddlewareConfig,\n  ToolCallHookContext,\n  ToolPhaseCompleteInfo,\n  UsageInfo,\n} from './types'\n\n/** Check if a middleware should be skipped for instrumentation events. */\nfunction shouldSkipInstrumentation(mw: ChatMiddleware<any>): boolean {\n  return mw.name === 'devtools' || mw.name === 'strip-to-spec'\n}\n\n/** Build the base context for middleware instrumentation events. */\nfunction instrumentCtx(ctx: ChatMiddlewareContext<any>) {\n  return {\n    requestId: ctx.requestId,\n    streamId: ctx.streamId,\n    clientId: ctx.threadId,\n    timestamp: Date.now(),\n  }\n}\n\n/**\n * Internal middleware runner that manages composed execution of middleware hooks.\n * Created once per chat() invocation.\n */\nexport class MiddlewareRunner<TContext = unknown> {\n  private readonly middlewares: ReadonlyArray<ChatMiddleware<TContext>>\n  private readonly logger: InternalLogger\n\n  constructor(\n    middlewares: ReadonlyArray<ChatMiddleware<TContext>>,\n    logger: InternalLogger,\n  ) {\n    this.middlewares = middlewares\n    this.logger = logger\n  }\n\n  get hasMiddleware(): boolean {\n    return this.middlewares.length > 0\n  }\n\n  /**\n   * Pipe config through all middleware onConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   */\n  async runOnConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: ChatMiddlewareConfig,\n  ): Promise<ChatMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              changes: result,\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Pipe config through all middleware onStructuredOutputConfig hooks in order.\n   * Each middleware receives the merged config from previous middleware.\n   * Partial returns are shallow-merged with the current config.\n   *\n   * Called once at the structured-output boundary, before runOnConfig at the\n   * same boundary (which receives a ChatMiddlewareConfig view, no outputSchema).\n   */\n  async runOnStructuredOutputConfig(\n    ctx: ChatMiddlewareContext<TContext>,\n    config: StructuredOutputMiddlewareConfig,\n  ): Promise<StructuredOutputMiddlewareConfig> {\n    let current = config\n    for (const mw of this.middlewares) {\n      if (mw.onStructuredOutputConfig) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const result = await mw.onStructuredOutputConfig(ctx, current)\n        const hasTransform = result !== undefined && result !== null\n        if (hasTransform) {\n          current = { ...current, ...result }\n          if (!skip) {\n            this.logger.config(\n              `middleware=${mw.name ?? 'unnamed'} keys=${Object.keys(result).join(',')}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                changes: result,\n              },\n            )\n          }\n        }\n        if (!skip) {\n          const base = instrumentCtx(ctx)\n          aiEventClient.emit('middleware:hook:executed', {\n            ...base,\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStructuredOutputConfig',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n          if (hasTransform) {\n            aiEventClient.emit('middleware:config:transformed', {\n              ...base,\n              middlewareName: mw.name || 'unnamed',\n              iteration: ctx.iteration,\n              // `result` is `Partial<StructuredOutputMiddlewareConfig>` —\n              // Object.fromEntries(Object.entries(result)) yields the\n              // structural `Record<string, unknown>` the event emitter wants\n              // without an `as` cast.\n              changes: Object.fromEntries(Object.entries(result)),\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Run all `setup` hooks in array order, then assert every declared `provides`\n   * capability was actually provided. Wires the last-wins duplicate-provide\n   * warning into the registry. Runs before init `onConfig`.\n   *\n   * Takes the full `ChatMiddlewareContext` — the same stable context the engine\n   * threads through every other hook — because it both forwards `ctx` to each\n   * `setup` hook and emits instrumentation events from it.\n   */\n  async runSetup(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    ctx.capabilities.setOnDuplicate((name) => {\n      this.logger.warn(\n        `capability \"${name}\" was provided more than once; last provider wins`,\n        { capability: name },\n      )\n    })\n\n    for (const mw of this.middlewares) {\n      if (mw.setup) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.setup(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=setup middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'setup' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'setup',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n\n    for (const mw of this.middlewares) {\n      for (const handle of mw.provides ?? []) {\n        if (!ctx.capabilities.has(handle)) {\n          throw new Error(\n            `Middleware \"${mw.name ?? 'unnamed'}\" declares it provides ` +\n              `\"${handle.capabilityName}\" but never called provide() in setup().`,\n          )\n        }\n      }\n    }\n  }\n\n  /**\n   * Call onStart on all middleware in order.\n   */\n  async runOnStart(ctx: ChatMiddlewareContext<TContext>): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onStart) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onStart(ctx)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onStart middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onStart' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onStart',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Pipe a single chunk through all middleware onChunk hooks in order.\n   * Returns the resulting chunks (0..N) to yield to the consumer.\n   *\n   * - void: pass through unchanged\n   * - chunk: replace with this chunk\n   * - chunk[]: expand to multiple chunks\n   * - null: drop the chunk entirely\n   */\n  async runOnChunk(\n    ctx: ChatMiddlewareContext<TContext>,\n    chunk: StreamChunk,\n  ): Promise<Array<StreamChunk>> {\n    let chunks: Array<StreamChunk> = [chunk]\n\n    for (const mw of this.middlewares) {\n      if (!mw.onChunk) continue\n      const skip = shouldSkipInstrumentation(mw)\n\n      const nextChunks: Array<StreamChunk> = []\n      for (const c of chunks) {\n        // Cast: @ag-ui/core Zod passthrough types prevent direct `.type` access\n        const chunkType = c.type\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onChunk', in: c },\n          )\n        }\n        const result = await mw.onChunk(ctx, c)\n        if (result === null) {\n          // Drop this chunk\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=<dropped>`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                dropped: true,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 0,\n              wasDropped: true,\n            })\n          }\n          continue\n        } else if (result === undefined) {\n          // Pass through — no instrumentation for pass-throughs\n          nextChunks.push(c)\n        } else if (Array.isArray(result)) {\n          // Expand\n          nextChunks.push(...result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=[${result.map((r: StreamChunk) => r.type).join(',')}]`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: result.length,\n              wasDropped: false,\n            })\n          }\n        } else {\n          // Replace\n          nextChunks.push(result)\n          if (!skip) {\n            this.logger.middleware(\n              `hook=onChunk middleware=${mw.name ?? 'unnamed'} in=${chunkType} out=${result.type}`,\n              {\n                middleware: mw.name ?? 'unnamed',\n                hook: 'onChunk',\n                in: c,\n                out: result,\n              },\n            )\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: chunkType,\n              resultCount: 1,\n              wasDropped: false,\n            })\n          }\n        }\n      }\n      chunks = nextChunks\n    }\n\n    return chunks\n  }\n\n  /**\n   * Dispatch a sandbox file event to every middleware's `sandbox` hooks, in\n   * array order: the catch-all `onFile` then the type-specific hook. Errors are\n   * logged and swallowed so one bad hook can't break the run.\n   */\n  async runSandboxFile(\n    ctx: ChatMiddlewareContext<TContext>,\n    event: SandboxFileEvent,\n  ): Promise<void> {\n    const typed = (\n      {\n        create: 'onFileCreate',\n        change: 'onFileChange',\n        delete: 'onFileDelete',\n      } as const\n    )[event.type]\n    for (const mw of this.middlewares) {\n      const hooks = mw.sandbox\n      if (!hooks) continue\n      for (const fn of [hooks.onFile, hooks[typed]]) {\n        if (!fn) continue\n        try {\n          await fn(ctx, event)\n        } catch (error) {\n          this.logger.sandbox(\n            `hook=${typed} middleware=${mw.name ?? 'unnamed'} threw`,\n            { middleware: mw.name ?? 'unnamed', error },\n          )\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onBeforeToolCall through middleware in order.\n   * Returns the first non-void decision, or undefined to continue normally.\n   */\n  async runOnBeforeToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    hookCtx: ToolCallHookContext,\n  ): Promise<BeforeToolCallDecision> {\n    for (const mw of this.middlewares) {\n      if (mw.onBeforeToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        const decision = await mw.onBeforeToolCall(ctx, hookCtx)\n        const hasTransform = decision !== undefined && decision !== null\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onBeforeToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onBeforeToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onBeforeToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform,\n          })\n        }\n        if (hasTransform) {\n          return decision\n        }\n      }\n    }\n    return undefined\n  }\n\n  /**\n   * Run onAfterToolCall on all middleware in order.\n   */\n  async runOnAfterToolCall(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AfterToolCallInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAfterToolCall) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAfterToolCall(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAfterToolCall middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAfterToolCall' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAfterToolCall',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onUsage on all middleware in order.\n   */\n  async runOnUsage(\n    ctx: ChatMiddlewareContext<TContext>,\n    usage: UsageInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onUsage) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onUsage(ctx, usage)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onUsage middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onUsage' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onUsage',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onFinish on all middleware in order.\n   */\n  async runOnFinish(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: FinishInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onFinish) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onFinish(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onFinish middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onFinish' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onFinish',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onAbort on all middleware in order.\n   */\n  async runOnAbort(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: AbortInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onAbort) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onAbort(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onAbort middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onAbort' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onAbort',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onError on all middleware in order.\n   */\n  async runOnError(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ErrorInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onError) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onError(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onError middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onError' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onError',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onIteration on all middleware in order.\n   * Called at the start of each agent loop iteration.\n   */\n  async runOnIteration(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: IterationInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onIteration) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onIteration(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onIteration middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onIteration' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onIteration',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n\n  /**\n   * Run onToolPhaseComplete on all middleware in order.\n   * Called after all tool calls in an iteration have been processed.\n   */\n  async runOnToolPhaseComplete(\n    ctx: ChatMiddlewareContext<TContext>,\n    info: ToolPhaseCompleteInfo,\n  ): Promise<void> {\n    for (const mw of this.middlewares) {\n      if (mw.onToolPhaseComplete) {\n        const skip = shouldSkipInstrumentation(mw)\n        const start = Date.now()\n        await mw.onToolPhaseComplete(ctx, info)\n        if (!skip) {\n          this.logger.middleware(\n            `hook=onToolPhaseComplete middleware=${mw.name ?? 'unnamed'}`,\n            { middleware: mw.name ?? 'unnamed', hook: 'onToolPhaseComplete' },\n          )\n          aiEventClient.emit('middleware:hook:executed', {\n            ...instrumentCtx(ctx),\n            middlewareName: mw.name || 'unnamed',\n            hookName: 'onToolPhaseComplete',\n            iteration: ctx.iteration,\n            duration: Date.now() - start,\n            hasTransform: false,\n          })\n        }\n      }\n    }\n  }\n}\n"],"names":[],"mappings":";AAqBA,SAAS,0BAA0B,IAAkC;AACnE,SAAO,GAAG,SAAS,cAAc,GAAG,SAAS;AAC/C;AAGA,SAAS,cAAc,KAAiC;AACtD,SAAO;AAAA,IACL,WAAW,IAAI;AAAA,IACf,UAAU,IAAI;AAAA,IACd,UAAU,IAAI;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,EAAI;AAExB;AAMO,MAAM,iBAAqC;AAAA,EAC/B;AAAA,EACA;AAAA,EAEjB,YACE,aACA,QACA;AACA,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,gBAAyB;AAC3B,WAAO,KAAK,YAAY,SAAS;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,YACJ,KACA,QAC+B;AAC/B,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,SAAS,KAAK,OAAO;AAC7C,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA,cACf,SAAS;AAAA,YAAA,CACV;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,4BACJ,KACA,QAC2C;AAC3C,QAAI,UAAU;AACd,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,0BAA0B;AAC/B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,SAAS,MAAM,GAAG,yBAAyB,KAAK,OAAO;AAC7D,cAAM,eAAe,WAAW,UAAa,WAAW;AACxD,YAAI,cAAc;AAChB,oBAAU,EAAE,GAAG,SAAS,GAAG,OAAA;AAC3B,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,cAAc,GAAG,QAAQ,SAAS,SAAS,OAAO,KAAK,MAAM,EAAE,KAAK,GAAG,CAAC;AAAA,cACxE;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,SAAS;AAAA,cAAA;AAAA,YACX;AAAA,UAEJ;AAAA,QACF;AACA,YAAI,CAAC,MAAM;AACT,gBAAM,OAAO,cAAc,GAAG;AAC9B,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG;AAAA,YACH,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AACD,cAAI,cAAc;AAChB,0BAAc,KAAK,iCAAiC;AAAA,cAClD,GAAG;AAAA,cACH,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,WAAW,IAAI;AAAA;AAAA;AAAA;AAAA;AAAA,cAKf,SAAS,OAAO,YAAY,OAAO,QAAQ,MAAM,CAAC;AAAA,YAAA,CACnD;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,SAAS,KAAqD;AAClE,QAAI,aAAa,eAAe,CAAC,SAAS;AACxC,WAAK,OAAO;AAAA,QACV,eAAe,IAAI;AAAA,QACnB,EAAE,YAAY,KAAA;AAAA,MAAK;AAAA,IAEvB,CAAC;AAED,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,OAAO;AACZ,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,MAAM,GAAG;AAClB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,yBAAyB,GAAG,QAAQ,SAAS;AAAA,YAC7C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,QAAA;AAAA,UAAQ;AAEpD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAEA,eAAW,MAAM,KAAK,aAAa;AACjC,iBAAW,UAAU,GAAG,YAAY,CAAA,GAAI;AACtC,YAAI,CAAC,IAAI,aAAa,IAAI,MAAM,GAAG;AACjC,gBAAM,IAAI;AAAA,YACR,eAAe,GAAG,QAAQ,SAAS,2BAC7B,OAAO,cAAc;AAAA,UAAA;AAAA,QAE/B;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WAAW,KAAqD;AACpE,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,GAAG;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,WACJ,KACA,OAC6B;AAC7B,QAAI,SAA6B,CAAC,KAAK;AAEvC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,CAAC,GAAG,QAAS;AACjB,YAAM,OAAO,0BAA0B,EAAE;AAEzC,YAAM,aAAiC,CAAA;AACvC,iBAAW,KAAK,QAAQ;AAEtB,cAAM,YAAY,EAAE;AACpB,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,YAC/D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAW,IAAI,EAAA;AAAA,UAAE;AAAA,QAE/D;AACA,cAAM,SAAS,MAAM,GAAG,QAAQ,KAAK,CAAC;AACtC,YAAI,WAAW,MAAM;AAEnB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS;AAAA,cAC/D;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,SAAS;AAAA,cAAA;AAAA,YACX;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AACA;AAAA,QACF,WAAW,WAAW,QAAW;AAE/B,qBAAW,KAAK,CAAC;AAAA,QACnB,WAAW,MAAM,QAAQ,MAAM,GAAG;AAEhC,qBAAW,KAAK,GAAG,MAAM;AACzB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,SAAS,OAAO,IAAI,CAAC,MAAmB,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC;AAAA,cACxH;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa,OAAO;AAAA,cACpB,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF,OAAO;AAEL,qBAAW,KAAK,MAAM;AACtB,cAAI,CAAC,MAAM;AACT,iBAAK,OAAO;AAAA,cACV,2BAA2B,GAAG,QAAQ,SAAS,OAAO,SAAS,QAAQ,OAAO,IAAI;AAAA,cAClF;AAAA,gBACE,YAAY,GAAG,QAAQ;AAAA,gBACvB,MAAM;AAAA,gBACN,IAAI;AAAA,gBACJ,KAAK;AAAA,cAAA;AAAA,YACP;AAEF,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB;AAAA,cACnB,aAAa;AAAA,cACb,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF;AAAA,MACF;AACA,eAAS;AAAA,IACX;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,eACJ,KACA,OACe;AACf,UAAM,QACJ;AAAA,MACE,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR,QAAQ;AAAA,IAAA,EAEV,MAAM,IAAI;AACZ,eAAW,MAAM,KAAK,aAAa;AACjC,YAAM,QAAQ,GAAG;AACjB,UAAI,CAAC,MAAO;AACZ,iBAAW,MAAM,CAAC,MAAM,QAAQ,MAAM,KAAK,CAAC,GAAG;AAC7C,YAAI,CAAC,GAAI;AACT,YAAI;AACF,gBAAM,GAAG,KAAK,KAAK;AAAA,QACrB,SAAS,OAAO;AACd,eAAK,OAAO;AAAA,YACV,QAAQ,KAAK,eAAe,GAAG,QAAQ,SAAS;AAAA,YAChD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAA;AAAA,UAAM;AAAA,QAE9C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,oBACJ,KACA,SACiC;AACjC,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,kBAAkB;AACvB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,WAAW,MAAM,GAAG,iBAAiB,KAAK,OAAO;AACvD,cAAM,eAAe,aAAa,UAAa,aAAa;AAC5D,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,oCAAoC,GAAG,QAAQ,SAAS;AAAA,YACxD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,mBAAA;AAAA,UAAmB;AAE/D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB;AAAA,UAAA,CACD;AAAA,QACH;AACA,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,mBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,iBAAiB;AACtB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,gBAAgB,KAAK,IAAI;AAClC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,mCAAmC,GAAG,QAAQ,SAAS;AAAA,YACvD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,kBAAA;AAAA,UAAkB;AAE9D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,OACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,KAAK;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,UAAU;AACf,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,SAAS,KAAK,IAAI;AAC3B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,4BAA4B,GAAG,QAAQ,SAAS;AAAA,YAChD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,WAAA;AAAA,UAAW;AAEvD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,WACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,SAAS;AACd,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,QAAQ,KAAK,IAAI;AAC1B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,2BAA2B,GAAG,QAAQ,SAAS;AAAA,YAC/C,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,UAAA;AAAA,UAAU;AAEtD,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,eACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,aAAa;AAClB,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,YAAY,KAAK,IAAI;AAC9B,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,+BAA+B,GAAG,QAAQ,SAAS;AAAA,YACnD,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,cAAA;AAAA,UAAc;AAE1D,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,uBACJ,KACA,MACe;AACf,eAAW,MAAM,KAAK,aAAa;AACjC,UAAI,GAAG,qBAAqB;AAC1B,cAAM,OAAO,0BAA0B,EAAE;AACzC,cAAM,QAAQ,KAAK,IAAA;AACnB,cAAM,GAAG,oBAAoB,KAAK,IAAI;AACtC,YAAI,CAAC,MAAM;AACT,eAAK,OAAO;AAAA,YACV,uCAAuC,GAAG,QAAQ,SAAS;AAAA,YAC3D,EAAE,YAAY,GAAG,QAAQ,WAAW,MAAM,sBAAA;AAAA,UAAsB;AAElE,wBAAc,KAAK,4BAA4B;AAAA,YAC7C,GAAG,cAAc,GAAG;AAAA,YACpB,gBAAgB,GAAG,QAAQ;AAAA,YAC3B,UAAU;AAAA,YACV,WAAW,IAAI;AAAA,YACf,UAAU,KAAK,IAAA,IAAQ;AAAA,YACvB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;"}

package/dist/esm/activities/chat/middleware/index.d.ts

@@ -1,4 +1,4 @@
-export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, StructuredOutputMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, } from './types.js';
+export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, StructuredOutputMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, SandboxFileEvent, ChatSandboxHooks, } from './types.js';
 export { MiddlewareRunner } from './compose.js';
 export { createCapability, CapabilityRegistry } from './capabilities.js';
 export type { Capability, CapabilityHandle, CapabilityContext, CapabilityGetter, CapabilityProvider, CapabilityGetOptions, } from './capabilities.js';

package/dist/esm/activities/chat/middleware/sandbox-runtime.d.ts

@@ -0,0 +1,8 @@
+import { InternalLogger } from '../../../logger/internal-logger.js';
+import { SandboxFileEvent } from './types.js';
+export interface SandboxRuntime {
+    emit: (event: SandboxFileEvent) => void;
+    logger: InternalLogger;
+}
+export declare const SandboxRuntimeCapability: import('./capabilities.js').Capability<SandboxRuntime, "sandbox-runtime">;
+export declare const getSandboxRuntime: import('./capabilities.js').CapabilityGetter<SandboxRuntime>, provideSandboxRuntime: import('./capabilities.js').CapabilityProvider<SandboxRuntime>;

package/dist/esm/activities/chat/middleware/sandbox-runtime.js

@@ -0,0 +1,9 @@
+import { createCapability } from "./capabilities.js";
+const SandboxRuntimeCapability = createCapability()("sandbox-runtime");
+const [getSandboxRuntime, provideSandboxRuntime] = SandboxRuntimeCapability;
+export {
+  SandboxRuntimeCapability,
+  getSandboxRuntime,
+  provideSandboxRuntime
+};
+//# sourceMappingURL=sandbox-runtime.js.map

package/dist/esm/activities/chat/middleware/sandbox-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sandbox-runtime.js","sources":["../../../../../src/activities/chat/middleware/sandbox-runtime.ts"],"sourcesContent":["/**\n * Internal runtime seam the chat engine PROVIDES so the sandbox middleware can\n * surface file events without a public ctx method. `emit` runs every\n * middleware's `sandbox` hooks AND emits a CUSTOM `sandbox.file` chunk into the\n * stream; `logger` lets the sandbox layer log under the `sandbox` debug\n * category. Consumed (optionally) by `withSandbox` in `@tanstack/ai-sandbox`.\n */\nimport { createCapability } from './capabilities'\nimport type { InternalLogger } from '../../../logger/internal-logger'\nimport type { SandboxFileEvent } from './types'\n\nexport interface SandboxRuntime {\n  emit: (event: SandboxFileEvent) => void\n  logger: InternalLogger\n}\n\nexport const SandboxRuntimeCapability =\n  createCapability<SandboxRuntime>()('sandbox-runtime')\n\nexport const [getSandboxRuntime, provideSandboxRuntime] =\n  SandboxRuntimeCapability\n"],"names":[],"mappings":";AAgBO,MAAM,2BACX,iBAAA,EAAmC,iBAAiB;AAE/C,MAAM,CAAC,mBAAmB,qBAAqB,IACpD;"}

package/dist/esm/activities/chat/middleware/types.d.ts

@@ -1,6 +1,23 @@
 import { JSONSchema, ModelMessage, StreamChunk, TokenUsage, Tool, ToolCall } from '../../../types.js';
 import { SystemPrompt } from '../../../system-prompts.js';
 import { Capability, CapabilityHandle, CapabilityRegistry } from './capabilities.js';
+/** A file change observed inside a sandbox during a chat run. */
+export interface SandboxFileEvent {
+    type: 'create' | 'change' | 'delete';
+    /** Absolute path inside the sandbox (under the workspace root). */
+    path: string;
+    timestamp: number;
+}
+/**
+ * Sandbox file-event hooks a chat middleware can declare. Fire server-side for
+ * every file create/change/delete observed in the sandbox during the run.
+ */
+export interface ChatSandboxHooks<TContext = unknown> {
+    onFile?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+    onFileCreate?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+    onFileChange?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+    onFileDelete?: (ctx: ChatMiddlewareContext<TContext>, e: SandboxFileEvent) => void | Promise<void>;
+}
 /**
  * Phase of the chat middleware lifecycle.
  * - 'init': Initial config transform before the chat engine starts
@@ -400,6 +417,11 @@ export interface ChatMiddleware<TContext = unknown> {
      * Exactly one of onFinish/onAbort/onError will be called per run.
      */
     onError?: (ctx: ChatMiddlewareContext<TContext>, info: ErrorInfo) => void | Promise<void>;
+    /**
+     * Sandbox file-event hooks. Fire when a sandbox provided by `withSandbox` is
+     * active during the run and a file is created/changed/deleted. Server-side.
+     */
+    sandbox?: ChatSandboxHooks<TContext>;
 }
 /** A `ChatMiddleware` with a permissive context — for use as a constraint. */
 export type AnyChatMiddleware = ChatMiddleware<any>;

package/dist/esm/activities/chat/stream/processor.d.ts

@@ -220,6 +220,24 @@ export declare class StreamProcessor {
      * Handle MESSAGES_SNAPSHOT event
      */
     private handleMessagesSnapshotEvent;
+    /**
+     * Reconcile a freshly normalized snapshot with the pre-snapshot message
+     * state so unreconstructable tool-call metadata is preserved.
+     *
+     * Post-pass (a): anchor `tool-result`-only assistant messages (the shape
+     * `aguiSnapshotMessageToUIMessage` emits for AG-UI `role: 'tool'` wire
+     * messages) into the message containing the matching `tool-call` part, or —
+     * when the snapshot supplies no such part — the nearest earlier anchorable
+     * assistant message, matching the in-stream fan-out shape
+     * `assistant: [text, tool-call, tool-result, ...]`. Detached messages with
+     * no earlier anchorable assistant are kept verbatim.
+     *
+     * Post-pass (b): when a `tool-result` part references a `toolCallId` whose
+     * `tool-call` part is absent from the snapshot, carry the `tool-call` part
+     * forward from the pre-snapshot state (state and output untouched) so a
+     * subsequent `addToolResult(toolCallId)` can still locate the call.
+     */
+    private reconcileSnapshotToolCalls;
     /**
      * Handle TEXT_MESSAGE_CONTENT event.
      *

package/dist/esm/activities/chat/stream/processor.js

@@ -1,5 +1,6 @@
 import { generateMessageId, uiMessageToModelMessages, aguiSnapshotMessageToUIMessage } from "../messages.js";
 import { normalizeToolResult } from "../../../utilities/tool-result.js";
+import { isProviderExecutedToolCall } from "../../../utilities/provider-executed.js";
 import { defaultJSONParser } from "./json-parser.js";
 import { updateToolCallWithOutput, updateToolResultPart, updateToolCallApprovalResponse, appendStructuredOutputDelta, updateToolCallPart, errorStructuredOutputPart, updateThinkingPart, completeStructuredOutputPart, updateToolCallApproval, updateTextPart } from "./message-updaters.js";
 import { ImmediateStrategy } from "./strategies.js";
@@ -197,7 +198,7 @@ class StreamProcessor {
       lastAssistant.parts.filter((p) => p.type === "tool-result").map((p) => p.toolCallId)
     );
     return toolParts.every(
-      (part) => part.state === "complete" || part.state === "approval-responded" || part.output !== void 0 && !part.approval || toolResultIds.has(part.id)
+      (part) => part.state === "complete" || part.state === "approval-responded" || part.output !== void 0 && !part.approval || toolResultIds.has(part.id) || isProviderExecutedToolCall(part)
     );
   }
   /**
@@ -548,9 +549,87 @@ class StreamProcessor {
    */
   handleMessagesSnapshotEvent(chunk) {
     this.resetStreamState();
-    this.messages = chunk.messages.map(aguiSnapshotMessageToUIMessage);
+    const prevMessages = this.messages;
+    const normalized = chunk.messages.map(aguiSnapshotMessageToUIMessage);
+    this.messages = this.reconcileSnapshotToolCalls(normalized, prevMessages);
     this.emitMessagesChange();
   }
+  /**
+   * Reconcile a freshly normalized snapshot with the pre-snapshot message
+   * state so unreconstructable tool-call metadata is preserved.
+   *
+   * Post-pass (a): anchor `tool-result`-only assistant messages (the shape
+   * `aguiSnapshotMessageToUIMessage` emits for AG-UI `role: 'tool'` wire
+   * messages) into the message containing the matching `tool-call` part, or —
+   * when the snapshot supplies no such part — the nearest earlier anchorable
+   * assistant message, matching the in-stream fan-out shape
+   * `assistant: [text, tool-call, tool-result, ...]`. Detached messages with
+   * no earlier anchorable assistant are kept verbatim.
+   *
+   * Post-pass (b): when a `tool-result` part references a `toolCallId` whose
+   * `tool-call` part is absent from the snapshot, carry the `tool-call` part
+   * forward from the pre-snapshot state (state and output untouched) so a
+   * subsequent `addToolResult(toolCallId)` can still locate the call.
+   */
+  reconcileSnapshotToolCalls(snapshot, prevMessages) {
+    const prevToolCalls = /* @__PURE__ */ new Map();
+    for (const msg of prevMessages) {
+      for (const part of msg.parts) {
+        if (part.type === "tool-call") {
+          prevToolCalls.set(part.id, part);
+        }
+      }
+    }
+    const snapshotToolCallIds = /* @__PURE__ */ new Set();
+    for (const msg of snapshot) {
+      for (const part of msg.parts) {
+        if (part.type === "tool-call") {
+          snapshotToolCallIds.add(part.id);
+        }
+      }
+    }
+    const reconciled = [];
+    for (const msg of snapshot) {
+      const toolResultPart = msg.role === "assistant" && msg.parts.length === 1 ? msg.parts.find((p) => p.type === "tool-result") : void 0;
+      if (!toolResultPart) {
+        reconciled.push(msg);
+        continue;
+      }
+      const target = reconciled.findLast(
+        (m) => m.parts.some(
+          (p) => p.type === "tool-call" && p.id === toolResultPart.toolCallId
+        )
+      ) ?? reconciled.findLast(
+        (m) => m.role === "assistant" && !(m.parts.length === 1 && m.parts[0]?.type === "tool-result")
+      );
+      if (!target) {
+        if (!snapshotToolCallIds.has(toolResultPart.toolCallId)) {
+          console.warn(
+            `[StreamProcessor] MESSAGES_SNAPSHOT contains a tool-result for "${toolResultPart.toolCallId}" but no matching tool-call exists in the snapshot, and there is no assistant message to anchor into; addToolResult("${toolResultPart.toolCallId}") will not be able to locate this call`
+          );
+        }
+        reconciled.push(msg);
+        continue;
+      }
+      const parts = [...target.parts];
+      if (!snapshotToolCallIds.has(toolResultPart.toolCallId) && !parts.some(
+        (p) => p.type === "tool-call" && p.id === toolResultPart.toolCallId
+      )) {
+        const prev = prevToolCalls.get(toolResultPart.toolCallId);
+        if (prev) {
+          parts.push({ ...prev });
+          snapshotToolCallIds.add(prev.id);
+        } else {
+          console.warn(
+            `[StreamProcessor] MESSAGES_SNAPSHOT contains a tool-result for "${toolResultPart.toolCallId}" but no matching tool-call exists in the snapshot or the pre-snapshot state; addToolResult("${toolResultPart.toolCallId}") will not be able to locate this call`
+          );
+        }
+      }
+      parts.push(toolResultPart);
+      target.parts = parts;
+    }
+    return reconciled;
+  }
   /**
    * Handle TEXT_MESSAGE_CONTENT event.
    *
@@ -1185,8 +1264,11 @@ class StreamProcessor {
    */
   isToolCallPartErrored(toolCallId) {
     return this.messages.some(
-      (msg) => msg.parts.some(
-        (part) => part.type === "tool-call" && part.id === toolCallId && part.state === "error"
+      (msg) => (
+        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition -- `parts` is typed as required, but seeded ModelMessage-shaped messages can lack it at runtime.
+        msg.parts?.some(
+          (part) => part.type === "tool-call" && part.id === toolCallId && part.state === "error"
+        )
       )
     );
   }