@tanstack/ai 0.10.3 → 0.11.1

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 {\n  AbortInfo,\n  AfterToolCallInfo,\n  BeforeToolCallDecision,\n  ChatMiddleware,\n  ChatMiddlewareConfig,\n  ChatMiddlewareContext,\n  ErrorInfo,\n  FinishInfo,\n  IterationInfo,\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): boolean {\n  return mw.name === 'devtools'\n}\n\n/** Build the base context for middleware instrumentation events. */\nfunction instrumentCtx(ctx: ChatMiddlewareContext) {\n  return {\n    requestId: ctx.requestId,\n    streamId: ctx.streamId,\n    clientId: ctx.conversationId,\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 {\n  private readonly middlewares: ReadonlyArray<ChatMiddleware>\n\n  constructor(middlewares: ReadonlyArray<ChatMiddleware>) {\n    this.middlewares = middlewares\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,\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        }\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 as Record<string, unknown>,\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Call onStart on all middleware in order.\n   */\n  async runOnStart(ctx: ChatMiddlewareContext): 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          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,\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        const result = await mw.onChunk(ctx, c)\n        if (result === null) {\n          // Drop this chunk\n          if (!skip) {\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: c.type,\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            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: c.type,\n              resultCount: result.length,\n              wasDropped: false,\n            })\n          }\n        } else {\n          // Replace\n          nextChunks.push(result)\n          if (!skip) {\n            aiEventClient.emit('middleware:chunk:transformed', {\n              ...instrumentCtx(ctx),\n              middlewareName: mw.name || 'unnamed',\n              originalChunkType: c.type,\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,\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          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,\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          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,\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          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,\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          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(ctx: ChatMiddlewareContext, info: AbortInfo): 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          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(ctx: ChatMiddlewareContext, info: ErrorInfo): 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          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,\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          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,\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          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":";AAkBA,SAAS,0BAA0B,IAA6B;AAC9D,SAAO,GAAG,SAAS;AACrB;AAGA,SAAS,cAAc,KAA4B;AACjD,SAAO;AAAA,IACL,WAAW,IAAI;AAAA,IACf,UAAU,IAAI;AAAA,IACd,UAAU,IAAI;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,EAAI;AAExB;AAMO,MAAM,iBAAiB;AAAA,EAG5B,YAAY,aAA4C;AACtD,SAAK,cAAc;AAAA,EACrB;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;AAAA,QAC7B;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,EAKA,MAAM,WAAW,KAA2C;AAC1D,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,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;AACtB,cAAM,SAAS,MAAM,GAAG,QAAQ,KAAK,CAAC;AACtC,YAAI,WAAW,MAAM;AAEnB,cAAI,CAAC,MAAM;AACT,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB,EAAE;AAAA,cACrB,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,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB,EAAE;AAAA,cACrB,aAAa,OAAO;AAAA,cACpB,YAAY;AAAA,YAAA,CACb;AAAA,UACH;AAAA,QACF,OAAO;AAEL,qBAAW,KAAK,MAAM;AACtB,cAAI,CAAC,MAAM;AACT,0BAAc,KAAK,gCAAgC;AAAA,cACjD,GAAG,cAAc,GAAG;AAAA,cACpB,gBAAgB,GAAG,QAAQ;AAAA,cAC3B,mBAAmB,EAAE;AAAA,cACrB,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,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,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,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,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,WAAW,KAA4B,MAAgC;AAC3E,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,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,WAAW,KAA4B,MAAgC;AAC3E,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,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,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,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 {\n  AbortInfo,\n  AfterToolCallInfo,\n  BeforeToolCallDecision,\n  ChatMiddleware,\n  ChatMiddlewareConfig,\n  ChatMiddlewareContext,\n  ErrorInfo,\n  FinishInfo,\n  IterationInfo,\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): 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) {\n  return {\n    requestId: ctx.requestId,\n    streamId: ctx.streamId,\n    clientId: ctx.conversationId,\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 {\n  private readonly middlewares: ReadonlyArray<ChatMiddleware>\n\n  constructor(middlewares: ReadonlyArray<ChatMiddleware>) {\n    this.middlewares = middlewares\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,\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        }\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 as Record<string, unknown>,\n            })\n          }\n        }\n      }\n    }\n    return current\n  }\n\n  /**\n   * Call onStart on all middleware in order.\n   */\n  async runOnStart(ctx: ChatMiddlewareContext): 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          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,\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 as StreamChunk & { type: string }).type\n        const result = await mw.onChunk(ctx, c)\n        if (result === null) {\n          // Drop this chunk\n          if (!skip) {\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            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            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,\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          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,\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          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,\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          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,\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          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(ctx: ChatMiddlewareContext, info: AbortInfo): 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          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(ctx: ChatMiddlewareContext, info: ErrorInfo): 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          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,\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          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,\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          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":";AAkBA,SAAS,0BAA0B,IAA6B;AAC9D,SAAO,GAAG,SAAS,cAAc,GAAG,SAAS;AAC/C;AAGA,SAAS,cAAc,KAA4B;AACjD,SAAO;AAAA,IACL,WAAW,IAAI;AAAA,IACf,UAAU,IAAI;AAAA,IACd,UAAU,IAAI;AAAA,IACd,WAAW,KAAK,IAAA;AAAA,EAAI;AAExB;AAMO,MAAM,iBAAiB;AAAA,EAG5B,YAAY,aAA4C;AACtD,SAAK,cAAc;AAAA,EACrB;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;AAAA,QAC7B;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,EAKA,MAAM,WAAW,KAA2C;AAC1D,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,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,YAAa,EAAqC;AACxD,cAAM,SAAS,MAAM,GAAG,QAAQ,KAAK,CAAC;AACtC,YAAI,WAAW,MAAM;AAEnB,cAAI,CAAC,MAAM;AACT,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,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,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,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,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,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,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,WAAW,KAA4B,MAAgC;AAC3E,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,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,WAAW,KAA4B,MAAgC;AAC3E,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,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,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,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/stream/processor.d.ts

@@ -159,7 +159,7 @@ export declare class StreamProcessor {
      *
      * Central dispatch for all AG-UI events. Each event type maps to a specific
      * handler. Events not listed in the switch are intentionally ignored
-     * (RUN_STARTED, STEP_STARTED, STATE_DELTA).
+     * (STEP_STARTED, STATE_SNAPSHOT, STATE_DELTA).
      *
      * @see docs/chat-architecture.md#adapter-contract — Expected event types and ordering
      */
@@ -253,6 +253,14 @@ export declare class StreamProcessor {
      * @see docs/chat-architecture.md#single-shot-tool-call-response — End-to-end flow
      */
     private handleToolCallEndEvent;
+    /**
+     * Handle TOOL_CALL_RESULT event (AG-UI spec).
+     *
+     * Creates a tool-result part and updates the tool-call output field,
+     * mirroring the logic from TOOL_CALL_END when it carries a result.
+     * This is the spec-compliant path for delivering tool results to the client.
+     */
+    private handleToolCallResultEvent;
     /**
      * Handle RUN_STARTED event.
      *
@@ -284,6 +292,13 @@ export declare class StreamProcessor {
      * @see docs/chat-architecture.md#thinkingreasoning-content — Thinking flow
      */
     private handleStepFinishedEvent;
+    /**
+     * Handle REASONING_MESSAGE_CONTENT event (AG-UI reasoning protocol).
+     *
+     * Accumulates reasoning delta into thinkingContent and updates the ThinkingPart
+     * in the UIMessage.
+     */
+    private handleReasoningMessageContentEvent;
     /**
      * Handle CUSTOM event.
      *

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

@@ -229,7 +229,7 @@ class StreamProcessor {
    *
    * Central dispatch for all AG-UI events. Each event type maps to a specific
    * handler. Events not listed in the switch are intentionally ignored
-   * (RUN_STARTED, STEP_STARTED, STATE_DELTA).
+   * (STEP_STARTED, STATE_SNAPSHOT, STATE_DELTA).
    *
    * @see docs/chat-architecture.md#adapter-contract — Expected event types and ordering
    */
@@ -241,43 +241,83 @@ class StreamProcessor {
         index: this.recording.chunks.length
       });
     }
-    switch (chunk.type) {
+    const c = chunk;
+    switch (c.type) {
       // AG-UI Events
       case "TEXT_MESSAGE_START":
-        this.handleTextMessageStartEvent(chunk);
+        this.handleTextMessageStartEvent(
+          chunk
+        );
         break;
       case "TEXT_MESSAGE_CONTENT":
-        this.handleTextMessageContentEvent(chunk);
+        this.handleTextMessageContentEvent(
+          chunk
+        );
         break;
       case "TEXT_MESSAGE_END":
-        this.handleTextMessageEndEvent(chunk);
+        this.handleTextMessageEndEvent(
+          chunk
+        );
         break;
       case "TOOL_CALL_START":
-        this.handleToolCallStartEvent(chunk);
+        this.handleToolCallStartEvent(
+          chunk
+        );
         break;
       case "TOOL_CALL_ARGS":
-        this.handleToolCallArgsEvent(chunk);
+        this.handleToolCallArgsEvent(
+          chunk
+        );
         break;
       case "TOOL_CALL_END":
-        this.handleToolCallEndEvent(chunk);
+        this.handleToolCallEndEvent(
+          chunk
+        );
         break;
       case "RUN_FINISHED":
-        this.handleRunFinishedEvent(chunk);
+        this.handleRunFinishedEvent(
+          chunk
+        );
         break;
       case "RUN_ERROR":
-        this.handleRunErrorEvent(chunk);
+        this.handleRunErrorEvent(
+          chunk
+        );
         break;
       case "STEP_FINISHED":
-        this.handleStepFinishedEvent(chunk);
+        this.handleStepFinishedEvent(
+          chunk
+        );
         break;
       case "MESSAGES_SNAPSHOT":
-        this.handleMessagesSnapshotEvent(chunk);
+        this.handleMessagesSnapshotEvent(
+          chunk
+        );
         break;
       case "CUSTOM":
-        this.handleCustomEvent(chunk);
+        this.handleCustomEvent(
+          chunk
+        );
         break;
       case "RUN_STARTED":
-        this.handleRunStartedEvent(chunk);
+        this.handleRunStartedEvent(
+          chunk
+        );
+        break;
+      case "REASONING_START":
+      case "REASONING_MESSAGE_START":
+      case "REASONING_MESSAGE_END":
+      case "REASONING_END":
+        break;
+      case "REASONING_MESSAGE_CONTENT":
+        this.handleReasoningMessageContentEvent(
+          chunk
+        );
+        break;
+      case "TOOL_CALL_RESULT":
+        this.handleToolCallResultEvent(
+          chunk
+        );
         break;
     }
   }
@@ -295,6 +335,7 @@ class StreamProcessor {
       currentSegmentText: "",
       lastEmittedText: "",
       thinkingContent: "",
+      hasSeenReasoningEvents: false,
       toolCalls: /* @__PURE__ */ new Map(),
       toolCallOrder: [],
       hasToolCallsSinceTextStart: false,
@@ -380,7 +421,7 @@ class StreamProcessor {
    */
   handleTextMessageStartEvent(chunk) {
     const { messageId, role } = chunk;
-    const uiRole = role === "tool" ? "assistant" : role;
+    const uiRole = role === "user" || role === "system" ? role : "assistant";
     if (this.pendingManualMessageId) {
       const pendingId = this.pendingManualMessageId;
       this.pendingManualMessageId = null;
@@ -528,9 +569,10 @@ class StreamProcessor {
     const existingToolCall = state.toolCalls.get(toolCallId);
     if (!existingToolCall) {
       const initialState = "awaiting-input";
+      const toolName = chunk.toolCallName;
       const newToolCall = {
         id: chunk.toolCallId,
-        name: chunk.toolName,
+        name: toolName,
         arguments: "",
         state: initialState,
         parsedArguments: void 0,
@@ -541,7 +583,7 @@ class StreamProcessor {
       this.toolCallToMessage.set(toolCallId, messageId);
       this.messages = updateToolCallPart(this.messages, messageId, {
         id: chunk.toolCallId,
-        name: chunk.toolName,
+        name: toolName,
         arguments: "",
         state: initialState
       });
@@ -649,6 +691,37 @@ class StreamProcessor {
       this.emitMessagesChange();
     }
   }
+  /**
+   * Handle TOOL_CALL_RESULT event (AG-UI spec).
+   *
+   * Creates a tool-result part and updates the tool-call output field,
+   * mirroring the logic from TOOL_CALL_END when it carries a result.
+   * This is the spec-compliant path for delivering tool results to the client.
+   */
+  handleToolCallResultEvent(chunk) {
+    const messageId = this.toolCallToMessage.get(chunk.toolCallId);
+    if (!messageId) return;
+    let output;
+    try {
+      output = JSON.parse(chunk.content);
+    } catch {
+      output = chunk.content;
+    }
+    this.messages = updateToolCallWithOutput(
+      this.messages,
+      chunk.toolCallId,
+      output
+    );
+    const resultState = "complete";
+    this.messages = updateToolResultPart(
+      this.messages,
+      messageId,
+      chunk.toolCallId,
+      chunk.content,
+      resultState
+    );
+    this.emitMessagesChange();
+  }
   /**
    * Handle RUN_STARTED event.
    *
@@ -669,7 +742,7 @@ class StreamProcessor {
    * @see docs/chat-architecture.md#adapter-contract — Why RUN_FINISHED is mandatory
    */
   handleRunFinishedEvent(chunk) {
-    this.finishReason = chunk.finishReason;
+    this.finishReason = chunk.finishReason ?? null;
     this.activeRuns.delete(chunk.runId);
     if (this.activeRuns.size === 0) {
       this.isDone = true;
@@ -682,13 +755,15 @@ class StreamProcessor {
    */
   handleRunErrorEvent(chunk) {
     this.hasError = true;
-    if (chunk.runId) {
-      this.activeRuns.delete(chunk.runId);
+    const runId = chunk.runId;
+    if (runId) {
+      this.activeRuns.delete(runId);
     } else {
       this.activeRuns.clear();
     }
     this.ensureAssistantMessage();
-    this.events.onError?.(new Error(chunk.error.message || "An error occurred"));
+    const errorMessage = chunk.message || chunk.error?.message || "An error occurred";
+    this.events.onError?.(new Error(errorMessage));
   }
   /**
    * Handle STEP_FINISHED event (for thinking/reasoning content).
@@ -702,6 +777,9 @@ class StreamProcessor {
     const { messageId, state } = this.ensureAssistantMessage(
       this.getActiveAssistantMessageId() ?? void 0
     );
+    if (state.hasSeenReasoningEvents) {
+      return;
+    }
     const previous = state.thinkingContent;
     let nextThinking = previous;
     if (chunk.delta && chunk.delta !== "") {
@@ -724,6 +802,27 @@ class StreamProcessor {
     this.emitMessagesChange();
     this.events.onThinkingUpdate?.(messageId, state.thinkingContent);
   }
+  /**
+   * Handle REASONING_MESSAGE_CONTENT event (AG-UI reasoning protocol).
+   *
+   * Accumulates reasoning delta into thinkingContent and updates the ThinkingPart
+   * in the UIMessage.
+   */
+  handleReasoningMessageContentEvent(chunk) {
+    const { messageId, state } = this.ensureAssistantMessage(
+      this.getActiveAssistantMessageId() ?? void 0
+    );
+    state.hasSeenReasoningEvents = true;
+    const delta = chunk.delta || "";
+    state.thinkingContent = state.thinkingContent + delta;
+    this.messages = updateThinkingPart(
+      this.messages,
+      messageId,
+      state.thinkingContent
+    );
+    this.emitMessagesChange();
+    this.events.onThinkingUpdate?.(messageId, state.thinkingContent);
+  }
   /**
    * Handle CUSTOM event.
    *