@assistant-ui/react-ag-ui 0.0.35 → 0.0.36

package/dist/runtime/adapter/run-aggregator.js

@@ -1,6 +1,8 @@
 "use client";
+import { isMcpAppUri } from "@assistant-ui/core";
 //#region src/runtime/adapter/run-aggregator.ts
 const AG_UI_METADATA_NAMESPACE = "agui";
+const MCP_APPS_ACTIVITY_TYPE = "mcp-apps";
 /**
 * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.
 *
@@ -20,6 +22,7 @@ var RunAggregator = class {
 	activeReasoningKey;
 	reasoningPartCounter = 0;
 	toolCalls = /* @__PURE__ */ new Map();
+	lastResolvedToolCallId;
 	partOrder = [];
 	textPartCounter = 0;
 	serverMessageIdReported = false;
@@ -40,6 +43,7 @@ var RunAggregator = class {
 				this.activeReasoningKey = void 0;
 				this.reasoningPartCounter = 0;
 				this.toolCalls.clear();
+				this.lastResolvedToolCallId = void 0;
 				this.partOrder.length = 0;
 				this.textPartCounter = 0;
 				this.activeTextMessageId = void 0;
@@ -153,6 +157,17 @@ var RunAggregator = class {
 				this.finishToolCall(event.toolCallId, event.content ?? "", event.role === "tool" ? false : void 0, event.messageId);
 				this.emit();
 				break;
+			case "ACTIVITY_SNAPSHOT": {
+				if (event.activityType !== MCP_APPS_ACTIVITY_TYPE) break;
+				const id = this.lastResolvedToolCallId;
+				const entry = id ? this.toolCalls.get(id) : void 0;
+				const resourceUri = event.content.resourceUri;
+				if (entry && typeof resourceUri === "string" && isMcpAppUri(resourceUri)) {
+					entry.mcpAppResourceUri = resourceUri;
+					this.emit();
+				}
+				break;
+			}
 			default: this.logger.debug?.("[agui] aggregator ignored event", event);
 		}
 	}
@@ -261,6 +276,7 @@ var RunAggregator = class {
 		entry.result = this.tryParseJSON(content);
 		entry.isError = isError;
 		if (toolMessageId) entry.toolMessageId = toolMessageId;
+		this.lastResolvedToolCallId = id;
 	}
 	tryParseJSON(value) {
 		if (!value) return value;
@@ -301,6 +317,7 @@ var RunAggregator = class {
 				argsText: entry.argsText,
 				...entry.result !== void 0 ? { result: entry.result } : {},
 				...entry.isError !== void 0 ? { isError: entry.isError } : {},
+				...entry.mcpAppResourceUri ? { mcp: { app: { resourceUri: entry.mcpAppResourceUri } } } : {},
 				...entry.parentMessageId ? { parentId: entry.parentMessageId } : {},
 				...entry.toolMessageId ? { unstable_toolMessageId: entry.toolMessageId } : {}
 			};

package/dist/runtime/adapter/run-aggregator.js.map

@@ -1 +1 @@
-{"version":3,"file":"run-aggregator.js","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  ChatModelRunResult,\n  MessageTiming,\n  ThreadAssistantMessagePart,\n  ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { AgUiEvent, AgUiInterrupt } from \"../types\";\nimport type { Logger } from \"../logger\";\n\nexport const AG_UI_METADATA_NAMESPACE = \"agui\";\n\nexport type AgUiCustomMetadata = {\n  interrupts?: AgUiInterrupt[];\n};\n\ntype Emit = (update: ChatModelRunResult) => void;\n\ntype ToolCallState = {\n  toolCallId: string;\n  toolCallName: string;\n  argsText: string;\n  parsedArgs: Record<string, unknown> | undefined;\n  result: unknown;\n  isError: boolean | undefined;\n  parentMessageId?: string;\n  toolMessageId?: string;\n};\n\nexport type RunAggregatorOptions = {\n  showThinking: boolean;\n  logger: Logger;\n  emit: Emit;\n  onServerMessageId?: (messageId: string) => void;\n};\n\n/**\n * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.\n *\n * The aggregator keeps a single assistant message worth of parts. Each incoming event updates the parts and\n * emits a fresh snapshot through the provided `emit` callback.\n */\nexport class RunAggregator {\n  private readonly emitUpdate: Emit;\n  private readonly showThinking: boolean;\n  private readonly logger: Logger;\n  private readonly onServerMessageId: ((messageId: string) => void) | undefined;\n\n  private status: ChatModelRunResult[\"status\"] | undefined;\n  private interrupts: AgUiInterrupt[] | undefined;\n  private readonly textParts = new Map<\n    string,\n    { buffer: string; touched: boolean }\n  >();\n  private activeTextMessageId: string | undefined;\n  private readonly reasoningParts = new Map<string, string>(); // key → buffer\n  private activeReasoningKey: string | undefined;\n  private reasoningPartCounter = 0;\n  private readonly toolCalls = new Map<string, ToolCallState>();\n  private readonly partOrder: (\n    | { kind: \"text\"; key: string }\n    | { kind: \"reasoning\"; key: string }\n    | { kind: \"tool-call\"; toolCallId: string }\n  )[] = [];\n  private textPartCounter = 0;\n  private serverMessageIdReported = false;\n\n  private streamStartTime: number | undefined;\n  private firstTokenTime: number | undefined;\n  private totalChunks = 0;\n\n  constructor(options: RunAggregatorOptions) {\n    this.emitUpdate = options.emit;\n    this.showThinking = options.showThinking;\n    this.logger = options.logger;\n    this.onServerMessageId = options.onServerMessageId;\n  }\n\n  handle(event: AgUiEvent): void {\n    switch (event.type) {\n      case \"RUN_STARTED\": {\n        this.clearTextParts();\n        this.reasoningParts.clear();\n        this.activeReasoningKey = undefined;\n        this.reasoningPartCounter = 0;\n        this.toolCalls.clear();\n        this.partOrder.length = 0;\n        this.textPartCounter = 0;\n        this.activeTextMessageId = undefined;\n        this.interrupts = undefined;\n        this.serverMessageIdReported = false;\n        this.streamStartTime = Date.now();\n        this.firstTokenTime = undefined;\n        this.totalChunks = 0;\n        this.status = { type: \"running\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_FINISHED\": {\n        if (event.outcome?.type === \"interrupt\") {\n          this.interrupts = event.outcome.interrupts;\n          this.status = { type: \"requires-action\", reason: \"interrupt\" };\n          this.emit();\n          break;\n        }\n\n        this.interrupts = undefined;\n        const hasUnresolvedToolCalls = Array.from(this.toolCalls.values()).some(\n          (tc) => tc.result === undefined,\n        );\n\n        this.status =\n          event.outcome?.type === \"success\" || !hasUnresolvedToolCalls\n            ? { type: \"complete\", reason: \"unknown\" }\n            : { type: \"requires-action\", reason: \"tool-calls\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_ERROR\": {\n        this.status = {\n          type: \"incomplete\",\n          reason: \"error\",\n          ...(event.message !== undefined ? { error: event.message } : {}),\n        };\n        this.emit();\n        break;\n      }\n      case \"RUN_CANCELLED\": {\n        this.status = { type: \"incomplete\", reason: \"cancelled\" };\n        this.emit();\n        break;\n      }\n\n      case \"TEXT_MESSAGE_START\": {\n        this.reportServerMessageId(event.messageId);\n        const id = this.startTextMessage(event.messageId);\n        if (id) {\n          this.markTextPartTouched(id);\n        }\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_CONTENT\":\n      case \"TEXT_MESSAGE_CHUNK\": {\n        const incomingId = \"messageId\" in event ? event.messageId : undefined;\n        this.reportServerMessageId(incomingId);\n        if (!event.delta) break;\n        this.recordFirstToken();\n        const id = this.resolveTextMessageId(incomingId);\n        this.appendText(id, event.delta);\n        this.totalChunks++;\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_END\": {\n        this.reportServerMessageId(event.messageId);\n        if (event.messageId && this.activeTextMessageId === event.messageId) {\n          this.activeTextMessageId = undefined;\n        }\n        this.emit();\n        break;\n      }\n\n      case \"THINKING_START\":\n      case \"THINKING_TEXT_MESSAGE_START\":\n      case \"REASONING_START\":\n      case \"REASONING_MESSAGE_START\":\n        this.handleReasoningStart(\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        break;\n      case \"THINKING_TEXT_MESSAGE_CONTENT\":\n        this.handleReasoningContent(event.delta);\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"REASONING_MESSAGE_CONTENT\":\n        this.handleReasoningContent(\n          event.delta,\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"THINKING_TEXT_MESSAGE_END\":\n      case \"THINKING_END\":\n      case \"REASONING_MESSAGE_END\":\n      case \"REASONING_END\":\n        this.handleReasoningEnd();\n        break;\n\n      case \"TOOL_CALL_START\": {\n        this.reportServerMessageId(event.parentMessageId);\n        this.startToolCall(\n          event.toolCallId,\n          event.toolCallName,\n          event.parentMessageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_ARGS\":\n      case \"TOOL_CALL_CHUNK\": {\n        if (event.type === \"TOOL_CALL_CHUNK\") {\n          this.reportServerMessageId(event.parentMessageId);\n        }\n        if (!event.delta) break;\n        this.appendToolArgs(event.toolCallId, event.delta);\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_END\": {\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_RESULT\": {\n        this.finishToolCall(\n          event.toolCallId,\n          event.content ?? \"\",\n          event.role === \"tool\" ? false : undefined,\n          event.messageId,\n        );\n        this.emit();\n        break;\n      }\n\n      default: {\n        this.logger.debug?.(\"[agui] aggregator ignored event\", event);\n      }\n    }\n  }\n\n  private reportServerMessageId(messageId: string | undefined): void {\n    if (this.serverMessageIdReported || !messageId) return;\n    this.serverMessageIdReported = true;\n    this.onServerMessageId?.(messageId);\n  }\n\n  private clearTextParts(): void {\n    this.textParts.clear();\n  }\n\n  private generateTextKey(): string {\n    this.textPartCounter += 1;\n    return `text-${this.textPartCounter}`;\n  }\n\n  private startTextMessage(messageId?: string): string {\n    const id = messageId ?? this.generateTextKey();\n    this.ensureTextPart(id);\n    this.activeTextMessageId = id;\n    return id;\n  }\n\n  private resolveTextMessageId(messageId?: string): string {\n    if (messageId) {\n      this.ensureTextPart(messageId);\n      this.activeTextMessageId = messageId;\n      return messageId;\n    }\n\n    if (this.activeTextMessageId) {\n      return this.activeTextMessageId;\n    }\n\n    const generated = this.generateTextKey();\n    this.ensureTextPart(generated);\n    this.activeTextMessageId = generated;\n    return generated;\n  }\n\n  private ensureTextPart(id: string): void {\n    if (!this.textParts.has(id)) {\n      this.textParts.set(id, { buffer: \"\", touched: false });\n      if (\n        !this.partOrder.some((part) => part.kind === \"text\" && part.key === id)\n      ) {\n        this.partOrder.push({ kind: \"text\", key: id });\n      }\n    }\n  }\n\n  private markTextPartTouched(id: string): void {\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.touched = true;\n  }\n\n  private appendText(id: string, delta: string): void {\n    this.ensureTextPart(id);\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.buffer += delta;\n    entry.touched = true;\n  }\n\n  private startToolCall(\n    id: string | undefined,\n    name?: string,\n    parentMessageId?: string,\n  ) {\n    if (!id) return;\n    // A new tool call acts as a boundary: any anonymous text that arrives\n    // after it should be a new part, not appended to the pre-tool text.\n    this.activeTextMessageId = undefined;\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    const state: ToolCallState = {\n      toolCallId: id,\n      toolCallName: name ?? \"tool\",\n      argsText: \"\",\n      parsedArgs: undefined,\n      result: undefined,\n      isError: undefined,\n    };\n    if (parentMessageId) {\n      state.parentMessageId = parentMessageId;\n    }\n    this.toolCalls.set(id, state);\n  }\n\n  private appendToolArgs(id: string | undefined, delta: string) {\n    const entry = id ? this.toolCalls.get(id) : undefined;\n    if (!entry) return;\n    entry.argsText += delta;\n    try {\n      const parsed = JSON.parse(entry.argsText);\n      if (parsed && typeof parsed === \"object\") {\n        entry.parsedArgs = parsed as Record<string, unknown>;\n      } else {\n        entry.parsedArgs = undefined;\n      }\n    } catch {\n      entry.parsedArgs = undefined;\n    }\n  }\n\n  private finishToolCall(\n    id: string,\n    content: string,\n    isError?: boolean,\n    toolMessageId?: string,\n  ) {\n    if (!id) return;\n    let entry = this.toolCalls.get(id);\n    if (!entry) {\n      entry = {\n        toolCallId: id,\n        toolCallName: \"tool\",\n        argsText: \"\",\n        parsedArgs: undefined,\n        result: undefined,\n        isError: undefined,\n      };\n      this.toolCalls.set(id, entry);\n    }\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    entry.result = this.tryParseJSON(content);\n    entry.isError = isError;\n    if (toolMessageId) {\n      entry.toolMessageId = toolMessageId;\n    }\n  }\n\n  private tryParseJSON(value: string): unknown {\n    if (!value) return value;\n    try {\n      return JSON.parse(value);\n    } catch {\n      return value;\n    }\n  }\n\n  private emit(): void {\n    const snapshot: ThreadAssistantMessagePart[] = [];\n\n    for (const part of this.partOrder) {\n      if (part.kind === \"reasoning\") {\n        if (this.showThinking) {\n          const buffer = this.reasoningParts.get(part.key) ?? \"\";\n          if (buffer.length > 0 || this.activeReasoningKey === part.key) {\n            snapshot.push({ type: \"reasoning\", text: buffer } as const);\n          }\n        }\n        continue;\n      }\n\n      if (part.kind === \"text\") {\n        const entry = this.textParts.get(part.key);\n        if (entry?.touched) {\n          snapshot.push({ type: \"text\", text: entry.buffer } as const);\n        }\n        continue;\n      }\n\n      const entry = this.toolCalls.get(part.toolCallId);\n      if (!entry) continue;\n      const toolPart: ToolCallMessagePart = {\n        type: \"tool-call\",\n        toolCallId: entry.toolCallId,\n        toolName: entry.toolCallName,\n        args: (entry.parsedArgs ?? {}) as any,\n        argsText: entry.argsText,\n        ...(entry.result !== undefined ? { result: entry.result } : {}),\n        ...(entry.isError !== undefined ? { isError: entry.isError } : {}),\n        ...(entry.parentMessageId ? { parentId: entry.parentMessageId } : {}),\n        ...(entry.toolMessageId\n          ? { unstable_toolMessageId: entry.toolMessageId }\n          : {}),\n      } as ToolCallMessagePart & { unstable_toolMessageId?: string };\n      snapshot.push(toolPart);\n    }\n\n    const timing = this.getTiming();\n    const metadata = {\n      ...(timing ? { timing } : {}),\n      ...(this.interrupts\n        ? {\n            custom: {\n              [AG_UI_METADATA_NAMESPACE]: {\n                interrupts: this.interrupts,\n              } satisfies AgUiCustomMetadata,\n            },\n          }\n        : {}),\n    };\n    const result: ChatModelRunResult = {\n      content: snapshot,\n      ...(this.status ? { status: this.status } : undefined),\n      ...(Object.keys(metadata).length > 0 ? { metadata } : undefined),\n    };\n    this.emitUpdate(result);\n  }\n\n  private recordFirstToken(): void {\n    if (\n      this.firstTokenTime === undefined &&\n      this.streamStartTime !== undefined\n    ) {\n      this.firstTokenTime = Date.now() - this.streamStartTime;\n    }\n  }\n\n  private getTiming(): MessageTiming | undefined {\n    if (this.streamStartTime === undefined) return undefined;\n\n    const now = Date.now();\n    const totalStreamTime = now - this.streamStartTime;\n    const tokenCount =\n      this.totalChunks > 0\n        ? Math.ceil(\n            Array.from(this.textParts.values()).reduce(\n              (sum, p) => sum + p.buffer.length,\n              0,\n            ) / 4,\n          )\n        : undefined;\n    const tokensPerSecond =\n      tokenCount && totalStreamTime > 0\n        ? (tokenCount / totalStreamTime) * 1000\n        : undefined;\n\n    return {\n      streamStartTime: this.streamStartTime,\n      ...(this.firstTokenTime !== undefined\n        ? { firstTokenTime: this.firstTokenTime }\n        : {}),\n      totalStreamTime,\n      ...(tokenCount !== undefined ? { tokenCount } : {}),\n      ...(tokensPerSecond !== undefined ? { tokensPerSecond } : {}),\n      totalChunks: this.totalChunks,\n      toolCallCount: this.toolCalls.size,\n    };\n  }\n\n  private handleReasoningStart(messageId?: string): void {\n    if (!this.showThinking) return;\n    // A reasoning block acts as a boundary: anonymous text arriving after it\n    // should be a new part, not appended to any pre-reasoning text.\n    this.activeTextMessageId = undefined;\n    const key = messageId ?? `__auto-reasoning-${++this.reasoningPartCounter}`;\n    if (!this.reasoningParts.has(key)) {\n      this.reasoningParts.set(key, \"\");\n      this.partOrder.push({ kind: \"reasoning\", key });\n    }\n    this.activeReasoningKey = key;\n    this.emit();\n  }\n\n  private handleReasoningContent(delta: string, messageId?: string): void {\n    if (!this.showThinking || !delta) return;\n    if (!this.activeReasoningKey) {\n      // Content arrived without a preceding START — create the slot lazily.\n      this.handleReasoningStart(messageId);\n    }\n    const key = this.activeReasoningKey;\n    if (!key) return;\n    this.reasoningParts.set(key, (this.reasoningParts.get(key) ?? \"\") + delta);\n    this.emit();\n  }\n\n  private handleReasoningEnd(): void {\n    if (!this.showThinking) return;\n    this.activeReasoningKey = undefined;\n    this.emit();\n  }\n}\n"],"mappings":";;AAWA,MAAa,2BAA2B;;;;;;;AAgCxC,IAAa,gBAAb,MAA2B;CACzB;CACA;CACA;CACA;CAEA;CACA;CACA,4BAA6B,IAAI,IAG/B;CACF;CACA,iCAAkC,IAAI,IAAoB;CAC1D;CACA,uBAA+B;CAC/B,4BAA6B,IAAI,IAA2B;CAC5D,YAIM,CAAC;CACP,kBAA0B;CAC1B,0BAAkC;CAElC;CACA;CACA,cAAsB;CAEtB,YAAY,SAA+B;EACzC,KAAK,aAAa,QAAQ;EAC1B,KAAK,eAAe,QAAQ;EAC5B,KAAK,SAAS,QAAQ;EACtB,KAAK,oBAAoB,QAAQ;CACnC;CAEA,OAAO,OAAwB;EAC7B,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,eAAe;IACpB,KAAK,eAAe,MAAM;IAC1B,KAAK,qBAAqB,KAAA;IAC1B,KAAK,uBAAuB;IAC5B,KAAK,UAAU,MAAM;IACrB,KAAK,UAAU,SAAS;IACxB,KAAK,kBAAkB;IACvB,KAAK,sBAAsB,KAAA;IAC3B,KAAK,aAAa,KAAA;IAClB,KAAK,0BAA0B;IAC/B,KAAK,kBAAkB,KAAK,IAAI;IAChC,KAAK,iBAAiB,KAAA;IACtB,KAAK,cAAc;IACnB,KAAK,SAAS,EAAE,MAAM,UAAU;IAChC,KAAK,KAAK;IACV;GAEF,KAAK,gBAAgB;IACnB,IAAI,MAAM,SAAS,SAAS,aAAa;KACvC,KAAK,aAAa,MAAM,QAAQ;KAChC,KAAK,SAAS;MAAE,MAAM;MAAmB,QAAQ;KAAY;KAC7D,KAAK,KAAK;KACV;IACF;IAEA,KAAK,aAAa,KAAA;IAClB,MAAM,yBAAyB,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,EAAE,MAChE,OAAO,GAAG,WAAW,KAAA,CACxB;IAEA,KAAK,SACH,MAAM,SAAS,SAAS,aAAa,CAAC,yBAClC;KAAE,MAAM;KAAY,QAAQ;IAAU,IACtC;KAAE,MAAM;KAAmB,QAAQ;IAAa;IACtD,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,SAAS;KACZ,MAAM;KACN,QAAQ;KACR,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,OAAO,MAAM,QAAQ,IAAI,CAAC;IAChE;IACA,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,SAAS;KAAE,MAAM;KAAc,QAAQ;IAAY;IACxD,KAAK,KAAK;IACV;GAGF,KAAK,sBAAsB;IACzB,KAAK,sBAAsB,MAAM,SAAS;IAC1C,MAAM,KAAK,KAAK,iBAAiB,MAAM,SAAS;IAChD,IAAI,IACF,KAAK,oBAAoB,EAAE;IAE7B,KAAK,KAAK;IACV;GACF;GACA,KAAK;GACL,KAAK,sBAAsB;IACzB,MAAM,aAAa,eAAe,QAAQ,MAAM,YAAY,KAAA;IAC5D,KAAK,sBAAsB,UAAU;IACrC,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,iBAAiB;IACtB,MAAM,KAAK,KAAK,qBAAqB,UAAU;IAC/C,KAAK,WAAW,IAAI,MAAM,KAAK;IAC/B,KAAK;IACL,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,sBAAsB,MAAM,SAAS;IAC1C,IAAI,MAAM,aAAa,KAAK,wBAAwB,MAAM,WACxD,KAAK,sBAAsB,KAAA;IAE7B,KAAK,KAAK;IACV;GAGF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,qBACH,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA;GACF,KAAK;IACH,KAAK,uBAAuB,MAAM,KAAK;IACvC,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;IACH,KAAK,uBACH,MAAM,OACN,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,mBAAmB;IACxB;GAEF,KAAK;IACH,KAAK,sBAAsB,MAAM,eAAe;IAChD,KAAK,cACH,MAAM,YACN,MAAM,cACN,MAAM,eACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK;GACL,KAAK;IACH,IAAI,MAAM,SAAS,mBACjB,KAAK,sBAAsB,MAAM,eAAe;IAElD,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,eAAe,MAAM,YAAY,MAAM,KAAK;IACjD,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,eACH,MAAM,YACN,MAAM,WAAW,IACjB,MAAM,SAAS,SAAS,QAAQ,KAAA,GAChC,MAAM,SACR;IACA,KAAK,KAAK;IACV;GAGF,SACE,KAAK,OAAO,QAAQ,mCAAmC,KAAK;EAEhE;CACF;CAEA,sBAA8B,WAAqC;EACjE,IAAI,KAAK,2BAA2B,CAAC,WAAW;EAChD,KAAK,0BAA0B;EAC/B,KAAK,oBAAoB,SAAS;CACpC;CAEA,iBAA+B;EAC7B,KAAK,UAAU,MAAM;CACvB;CAEA,kBAAkC;EAChC,KAAK,mBAAmB;EACxB,OAAO,QAAQ,KAAK;CACtB;CAEA,iBAAyB,WAA4B;EACnD,MAAM,KAAK,aAAa,KAAK,gBAAgB;EAC7C,KAAK,eAAe,EAAE;EACtB,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,qBAA6B,WAA4B;EACvD,IAAI,WAAW;GACb,KAAK,eAAe,SAAS;GAC7B,KAAK,sBAAsB;GAC3B,OAAO;EACT;EAEA,IAAI,KAAK,qBACP,OAAO,KAAK;EAGd,MAAM,YAAY,KAAK,gBAAgB;EACvC,KAAK,eAAe,SAAS;EAC7B,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,eAAuB,IAAkB;EACvC,IAAI,CAAC,KAAK,UAAU,IAAI,EAAE,GAAG;GAC3B,KAAK,UAAU,IAAI,IAAI;IAAE,QAAQ;IAAI,SAAS;GAAM,CAAC;GACrD,IACE,CAAC,KAAK,UAAU,MAAM,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,EAAE,GAEtE,KAAK,UAAU,KAAK;IAAE,MAAM;IAAQ,KAAK;GAAG,CAAC;EAEjD;CACF;CAEA,oBAA4B,IAAkB;EAC5C,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;CAClB;CAEA,WAAmB,IAAY,OAAqB;EAClD,KAAK,eAAe,EAAE;EACtB,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;EAChB,MAAM,UAAU;CAClB;CAEA,cACE,IACA,MACA,iBACA;EACA,IAAI,CAAC,IAAI;EAGT,KAAK,sBAAsB,KAAA;EAC3B,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,QAAuB;GAC3B,YAAY;GACZ,cAAc,QAAQ;GACtB,UAAU;GACV,YAAY,KAAA;GACZ,QAAQ,KAAA;GACR,SAAS,KAAA;EACX;EACA,IAAI,iBACF,MAAM,kBAAkB;EAE1B,KAAK,UAAU,IAAI,IAAI,KAAK;CAC9B;CAEA,eAAuB,IAAwB,OAAe;EAC5D,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;EAC5C,IAAI,CAAC,OAAO;EACZ,MAAM,YAAY;EAClB,IAAI;GACF,MAAM,SAAS,KAAK,MAAM,MAAM,QAAQ;GACxC,IAAI,UAAU,OAAO,WAAW,UAC9B,MAAM,aAAa;QAEnB,MAAM,aAAa,KAAA;EAEvB,QAAQ;GACN,MAAM,aAAa,KAAA;EACrB;CACF;CAEA,eACE,IACA,SACA,SACA,eACA;EACA,IAAI,CAAC,IAAI;EACT,IAAI,QAAQ,KAAK,UAAU,IAAI,EAAE;EACjC,IAAI,CAAC,OAAO;GACV,QAAQ;IACN,YAAY;IACZ,cAAc;IACd,UAAU;IACV,YAAY,KAAA;IACZ,QAAQ,KAAA;IACR,SAAS,KAAA;GACX;GACA,KAAK,UAAU,IAAI,IAAI,KAAK;EAC9B;EACA,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,SAAS,KAAK,aAAa,OAAO;EACxC,MAAM,UAAU;EAChB,IAAI,eACF,MAAM,gBAAgB;CAE1B;CAEA,aAAqB,OAAwB;EAC3C,IAAI,CAAC,OAAO,OAAO;EACnB,IAAI;GACF,OAAO,KAAK,MAAM,KAAK;EACzB,QAAQ;GACN,OAAO;EACT;CACF;CAEA,OAAqB;EACnB,MAAM,WAAyC,CAAC;EAEhD,KAAK,MAAM,QAAQ,KAAK,WAAW;GACjC,IAAI,KAAK,SAAS,aAAa;IAC7B,IAAI,KAAK,cAAc;KACrB,MAAM,SAAS,KAAK,eAAe,IAAI,KAAK,GAAG,KAAK;KACpD,IAAI,OAAO,SAAS,KAAK,KAAK,uBAAuB,KAAK,KACxD,SAAS,KAAK;MAAE,MAAM;MAAa,MAAM;KAAO,CAAU;IAE9D;IACA;GACF;GAEA,IAAI,KAAK,SAAS,QAAQ;IACxB,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,GAAG;IACzC,IAAI,OAAO,SACT,SAAS,KAAK;KAAE,MAAM;KAAQ,MAAM,MAAM;IAAO,CAAU;IAE7D;GACF;GAEA,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,UAAU;GAChD,IAAI,CAAC,OAAO;GACZ,MAAM,WAAgC;IACpC,MAAM;IACN,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,MAAO,MAAM,cAAc,CAAC;IAC5B,UAAU,MAAM;IAChB,GAAI,MAAM,WAAW,KAAA,IAAY,EAAE,QAAQ,MAAM,OAAO,IAAI,CAAC;IAC7D,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;IAChE,GAAI,MAAM,kBAAkB,EAAE,UAAU,MAAM,gBAAgB,IAAI,CAAC;IACnE,GAAI,MAAM,gBACN,EAAE,wBAAwB,MAAM,cAAc,IAC9C,CAAC;GACP;GACA,SAAS,KAAK,QAAQ;EACxB;EAEA,MAAM,SAAS,KAAK,UAAU;EAC9B,MAAM,WAAW;GACf,GAAI,SAAS,EAAE,OAAO,IAAI,CAAC;GAC3B,GAAI,KAAK,aACL,EACE,QAAQ,GACL,2BAA2B,EAC1B,YAAY,KAAK,WACnB,EACF,EACF,IACA,CAAC;EACP;EACA,MAAM,SAA6B;GACjC,SAAS;GACT,GAAI,KAAK,SAAS,EAAE,QAAQ,KAAK,OAAO,IAAI,KAAA;GAC5C,GAAI,OAAO,KAAK,QAAQ,EAAE,SAAS,IAAI,EAAE,SAAS,IAAI,KAAA;EACxD;EACA,KAAK,WAAW,MAAM;CACxB;CAEA,mBAAiC;EAC/B,IACE,KAAK,mBAAmB,KAAA,KACxB,KAAK,oBAAoB,KAAA,GAEzB,KAAK,iBAAiB,KAAK,IAAI,IAAI,KAAK;CAE5C;CAEA,YAA+C;EAC7C,IAAI,KAAK,oBAAoB,KAAA,GAAW,OAAO,KAAA;EAG/C,MAAM,kBADM,KAAK,IACS,IAAI,KAAK;EACnC,MAAM,aACJ,KAAK,cAAc,IACf,KAAK,KACH,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,EAAE,QACjC,KAAK,MAAM,MAAM,EAAE,OAAO,QAC3B,CACF,IAAI,CACN,IACA,KAAA;EACN,MAAM,kBACJ,cAAc,kBAAkB,IAC3B,aAAa,kBAAmB,MACjC,KAAA;EAEN,OAAO;GACL,iBAAiB,KAAK;GACtB,GAAI,KAAK,mBAAmB,KAAA,IACxB,EAAE,gBAAgB,KAAK,eAAe,IACtC,CAAC;GACL;GACA,GAAI,eAAe,KAAA,IAAY,EAAE,WAAW,IAAI,CAAC;GACjD,GAAI,oBAAoB,KAAA,IAAY,EAAE,gBAAgB,IAAI,CAAC;GAC3D,aAAa,KAAK;GAClB,eAAe,KAAK,UAAU;EAChC;CACF;CAEA,qBAA6B,WAA0B;EACrD,IAAI,CAAC,KAAK,cAAc;EAGxB,KAAK,sBAAsB,KAAA;EAC3B,MAAM,MAAM,aAAa,oBAAoB,EAAE,KAAK;EACpD,IAAI,CAAC,KAAK,eAAe,IAAI,GAAG,GAAG;GACjC,KAAK,eAAe,IAAI,KAAK,EAAE;GAC/B,KAAK,UAAU,KAAK;IAAE,MAAM;IAAa;GAAI,CAAC;EAChD;EACA,KAAK,qBAAqB;EAC1B,KAAK,KAAK;CACZ;CAEA,uBAA+B,OAAe,WAA0B;EACtE,IAAI,CAAC,KAAK,gBAAgB,CAAC,OAAO;EAClC,IAAI,CAAC,KAAK,oBAER,KAAK,qBAAqB,SAAS;EAErC,MAAM,MAAM,KAAK;EACjB,IAAI,CAAC,KAAK;EACV,KAAK,eAAe,IAAI,MAAM,KAAK,eAAe,IAAI,GAAG,KAAK,MAAM,KAAK;EACzE,KAAK,KAAK;CACZ;CAEA,qBAAmC;EACjC,IAAI,CAAC,KAAK,cAAc;EACxB,KAAK,qBAAqB,KAAA;EAC1B,KAAK,KAAK;CACZ;AACF"}
+{"version":3,"file":"run-aggregator.js","names":[],"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  isMcpAppUri,\n  type ChatModelRunResult,\n  type MessageTiming,\n  type ThreadAssistantMessagePart,\n  type ToolCallMessagePart,\n} from \"@assistant-ui/core\";\nimport type { AgUiEvent, AgUiInterrupt } from \"../types\";\nimport type { Logger } from \"../logger\";\n\nexport const AG_UI_METADATA_NAMESPACE = \"agui\";\n\nexport type AgUiCustomMetadata = {\n  interrupts?: AgUiInterrupt[];\n};\n\ntype Emit = (update: ChatModelRunResult) => void;\n\ntype ToolCallState = {\n  toolCallId: string;\n  toolCallName: string;\n  argsText: string;\n  parsedArgs: Record<string, unknown> | undefined;\n  result: unknown;\n  isError: boolean | undefined;\n  parentMessageId?: string;\n  toolMessageId?: string;\n  mcpAppResourceUri?: string;\n};\n\nconst MCP_APPS_ACTIVITY_TYPE = \"mcp-apps\";\n\nexport type RunAggregatorOptions = {\n  showThinking: boolean;\n  logger: Logger;\n  emit: Emit;\n  onServerMessageId?: (messageId: string) => void;\n};\n\n/**\n * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.\n *\n * The aggregator keeps a single assistant message worth of parts. Each incoming event updates the parts and\n * emits a fresh snapshot through the provided `emit` callback.\n */\nexport class RunAggregator {\n  private readonly emitUpdate: Emit;\n  private readonly showThinking: boolean;\n  private readonly logger: Logger;\n  private readonly onServerMessageId: ((messageId: string) => void) | undefined;\n\n  private status: ChatModelRunResult[\"status\"] | undefined;\n  private interrupts: AgUiInterrupt[] | undefined;\n  private readonly textParts = new Map<\n    string,\n    { buffer: string; touched: boolean }\n  >();\n  private activeTextMessageId: string | undefined;\n  private readonly reasoningParts = new Map<string, string>(); // key → buffer\n  private activeReasoningKey: string | undefined;\n  private reasoningPartCounter = 0;\n  private readonly toolCalls = new Map<string, ToolCallState>();\n  private lastResolvedToolCallId: string | undefined;\n  private readonly partOrder: (\n    | { kind: \"text\"; key: string }\n    | { kind: \"reasoning\"; key: string }\n    | { kind: \"tool-call\"; toolCallId: string }\n  )[] = [];\n  private textPartCounter = 0;\n  private serverMessageIdReported = false;\n\n  private streamStartTime: number | undefined;\n  private firstTokenTime: number | undefined;\n  private totalChunks = 0;\n\n  constructor(options: RunAggregatorOptions) {\n    this.emitUpdate = options.emit;\n    this.showThinking = options.showThinking;\n    this.logger = options.logger;\n    this.onServerMessageId = options.onServerMessageId;\n  }\n\n  handle(event: AgUiEvent): void {\n    switch (event.type) {\n      case \"RUN_STARTED\": {\n        this.clearTextParts();\n        this.reasoningParts.clear();\n        this.activeReasoningKey = undefined;\n        this.reasoningPartCounter = 0;\n        this.toolCalls.clear();\n        this.lastResolvedToolCallId = undefined;\n        this.partOrder.length = 0;\n        this.textPartCounter = 0;\n        this.activeTextMessageId = undefined;\n        this.interrupts = undefined;\n        this.serverMessageIdReported = false;\n        this.streamStartTime = Date.now();\n        this.firstTokenTime = undefined;\n        this.totalChunks = 0;\n        this.status = { type: \"running\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_FINISHED\": {\n        if (event.outcome?.type === \"interrupt\") {\n          this.interrupts = event.outcome.interrupts;\n          this.status = { type: \"requires-action\", reason: \"interrupt\" };\n          this.emit();\n          break;\n        }\n\n        this.interrupts = undefined;\n        const hasUnresolvedToolCalls = Array.from(this.toolCalls.values()).some(\n          (tc) => tc.result === undefined,\n        );\n\n        this.status =\n          event.outcome?.type === \"success\" || !hasUnresolvedToolCalls\n            ? { type: \"complete\", reason: \"unknown\" }\n            : { type: \"requires-action\", reason: \"tool-calls\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_ERROR\": {\n        this.status = {\n          type: \"incomplete\",\n          reason: \"error\",\n          ...(event.message !== undefined ? { error: event.message } : {}),\n        };\n        this.emit();\n        break;\n      }\n      case \"RUN_CANCELLED\": {\n        this.status = { type: \"incomplete\", reason: \"cancelled\" };\n        this.emit();\n        break;\n      }\n\n      case \"TEXT_MESSAGE_START\": {\n        this.reportServerMessageId(event.messageId);\n        const id = this.startTextMessage(event.messageId);\n        if (id) {\n          this.markTextPartTouched(id);\n        }\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_CONTENT\":\n      case \"TEXT_MESSAGE_CHUNK\": {\n        const incomingId = \"messageId\" in event ? event.messageId : undefined;\n        this.reportServerMessageId(incomingId);\n        if (!event.delta) break;\n        this.recordFirstToken();\n        const id = this.resolveTextMessageId(incomingId);\n        this.appendText(id, event.delta);\n        this.totalChunks++;\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_END\": {\n        this.reportServerMessageId(event.messageId);\n        if (event.messageId && this.activeTextMessageId === event.messageId) {\n          this.activeTextMessageId = undefined;\n        }\n        this.emit();\n        break;\n      }\n\n      case \"THINKING_START\":\n      case \"THINKING_TEXT_MESSAGE_START\":\n      case \"REASONING_START\":\n      case \"REASONING_MESSAGE_START\":\n        this.handleReasoningStart(\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        break;\n      case \"THINKING_TEXT_MESSAGE_CONTENT\":\n        this.handleReasoningContent(event.delta);\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"REASONING_MESSAGE_CONTENT\":\n        this.handleReasoningContent(\n          event.delta,\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        this.totalChunks++;\n        this.recordFirstToken();\n        break;\n      case \"THINKING_TEXT_MESSAGE_END\":\n      case \"THINKING_END\":\n      case \"REASONING_MESSAGE_END\":\n      case \"REASONING_END\":\n        this.handleReasoningEnd();\n        break;\n\n      case \"TOOL_CALL_START\": {\n        this.reportServerMessageId(event.parentMessageId);\n        this.startToolCall(\n          event.toolCallId,\n          event.toolCallName,\n          event.parentMessageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_ARGS\":\n      case \"TOOL_CALL_CHUNK\": {\n        if (event.type === \"TOOL_CALL_CHUNK\") {\n          this.reportServerMessageId(event.parentMessageId);\n        }\n        if (!event.delta) break;\n        this.appendToolArgs(event.toolCallId, event.delta);\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_END\": {\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_RESULT\": {\n        this.finishToolCall(\n          event.toolCallId,\n          event.content ?? \"\",\n          event.role === \"tool\" ? false : undefined,\n          event.messageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"ACTIVITY_SNAPSHOT\": {\n        if (event.activityType !== MCP_APPS_ACTIVITY_TYPE) break;\n        const id = this.lastResolvedToolCallId;\n        const entry = id ? this.toolCalls.get(id) : undefined;\n        const resourceUri = event.content.resourceUri;\n        if (\n          entry &&\n          typeof resourceUri === \"string\" &&\n          isMcpAppUri(resourceUri)\n        ) {\n          entry.mcpAppResourceUri = resourceUri;\n          this.emit();\n        }\n        break;\n      }\n\n      default: {\n        this.logger.debug?.(\"[agui] aggregator ignored event\", event);\n      }\n    }\n  }\n\n  private reportServerMessageId(messageId: string | undefined): void {\n    if (this.serverMessageIdReported || !messageId) return;\n    this.serverMessageIdReported = true;\n    this.onServerMessageId?.(messageId);\n  }\n\n  private clearTextParts(): void {\n    this.textParts.clear();\n  }\n\n  private generateTextKey(): string {\n    this.textPartCounter += 1;\n    return `text-${this.textPartCounter}`;\n  }\n\n  private startTextMessage(messageId?: string): string {\n    const id = messageId ?? this.generateTextKey();\n    this.ensureTextPart(id);\n    this.activeTextMessageId = id;\n    return id;\n  }\n\n  private resolveTextMessageId(messageId?: string): string {\n    if (messageId) {\n      this.ensureTextPart(messageId);\n      this.activeTextMessageId = messageId;\n      return messageId;\n    }\n\n    if (this.activeTextMessageId) {\n      return this.activeTextMessageId;\n    }\n\n    const generated = this.generateTextKey();\n    this.ensureTextPart(generated);\n    this.activeTextMessageId = generated;\n    return generated;\n  }\n\n  private ensureTextPart(id: string): void {\n    if (!this.textParts.has(id)) {\n      this.textParts.set(id, { buffer: \"\", touched: false });\n      if (\n        !this.partOrder.some((part) => part.kind === \"text\" && part.key === id)\n      ) {\n        this.partOrder.push({ kind: \"text\", key: id });\n      }\n    }\n  }\n\n  private markTextPartTouched(id: string): void {\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.touched = true;\n  }\n\n  private appendText(id: string, delta: string): void {\n    this.ensureTextPart(id);\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.buffer += delta;\n    entry.touched = true;\n  }\n\n  private startToolCall(\n    id: string | undefined,\n    name?: string,\n    parentMessageId?: string,\n  ) {\n    if (!id) return;\n    // A new tool call acts as a boundary: any anonymous text that arrives\n    // after it should be a new part, not appended to the pre-tool text.\n    this.activeTextMessageId = undefined;\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    const state: ToolCallState = {\n      toolCallId: id,\n      toolCallName: name ?? \"tool\",\n      argsText: \"\",\n      parsedArgs: undefined,\n      result: undefined,\n      isError: undefined,\n    };\n    if (parentMessageId) {\n      state.parentMessageId = parentMessageId;\n    }\n    this.toolCalls.set(id, state);\n  }\n\n  private appendToolArgs(id: string | undefined, delta: string) {\n    const entry = id ? this.toolCalls.get(id) : undefined;\n    if (!entry) return;\n    entry.argsText += delta;\n    try {\n      const parsed = JSON.parse(entry.argsText);\n      if (parsed && typeof parsed === \"object\") {\n        entry.parsedArgs = parsed as Record<string, unknown>;\n      } else {\n        entry.parsedArgs = undefined;\n      }\n    } catch {\n      entry.parsedArgs = undefined;\n    }\n  }\n\n  private finishToolCall(\n    id: string,\n    content: string,\n    isError?: boolean,\n    toolMessageId?: string,\n  ) {\n    if (!id) return;\n    let entry = this.toolCalls.get(id);\n    if (!entry) {\n      entry = {\n        toolCallId: id,\n        toolCallName: \"tool\",\n        argsText: \"\",\n        parsedArgs: undefined,\n        result: undefined,\n        isError: undefined,\n      };\n      this.toolCalls.set(id, entry);\n    }\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    entry.result = this.tryParseJSON(content);\n    entry.isError = isError;\n    if (toolMessageId) {\n      entry.toolMessageId = toolMessageId;\n    }\n    this.lastResolvedToolCallId = id;\n  }\n\n  private tryParseJSON(value: string): unknown {\n    if (!value) return value;\n    try {\n      return JSON.parse(value);\n    } catch {\n      return value;\n    }\n  }\n\n  private emit(): void {\n    const snapshot: ThreadAssistantMessagePart[] = [];\n\n    for (const part of this.partOrder) {\n      if (part.kind === \"reasoning\") {\n        if (this.showThinking) {\n          const buffer = this.reasoningParts.get(part.key) ?? \"\";\n          if (buffer.length > 0 || this.activeReasoningKey === part.key) {\n            snapshot.push({ type: \"reasoning\", text: buffer } as const);\n          }\n        }\n        continue;\n      }\n\n      if (part.kind === \"text\") {\n        const entry = this.textParts.get(part.key);\n        if (entry?.touched) {\n          snapshot.push({ type: \"text\", text: entry.buffer } as const);\n        }\n        continue;\n      }\n\n      const entry = this.toolCalls.get(part.toolCallId);\n      if (!entry) continue;\n      const toolPart: ToolCallMessagePart = {\n        type: \"tool-call\",\n        toolCallId: entry.toolCallId,\n        toolName: entry.toolCallName,\n        args: (entry.parsedArgs ?? {}) as any,\n        argsText: entry.argsText,\n        ...(entry.result !== undefined ? { result: entry.result } : {}),\n        ...(entry.isError !== undefined ? { isError: entry.isError } : {}),\n        ...(entry.mcpAppResourceUri\n          ? { mcp: { app: { resourceUri: entry.mcpAppResourceUri } } }\n          : {}),\n        ...(entry.parentMessageId ? { parentId: entry.parentMessageId } : {}),\n        ...(entry.toolMessageId\n          ? { unstable_toolMessageId: entry.toolMessageId }\n          : {}),\n      } as ToolCallMessagePart & { unstable_toolMessageId?: string };\n      snapshot.push(toolPart);\n    }\n\n    const timing = this.getTiming();\n    const metadata = {\n      ...(timing ? { timing } : {}),\n      ...(this.interrupts\n        ? {\n            custom: {\n              [AG_UI_METADATA_NAMESPACE]: {\n                interrupts: this.interrupts,\n              } satisfies AgUiCustomMetadata,\n            },\n          }\n        : {}),\n    };\n    const result: ChatModelRunResult = {\n      content: snapshot,\n      ...(this.status ? { status: this.status } : undefined),\n      ...(Object.keys(metadata).length > 0 ? { metadata } : undefined),\n    };\n    this.emitUpdate(result);\n  }\n\n  private recordFirstToken(): void {\n    if (\n      this.firstTokenTime === undefined &&\n      this.streamStartTime !== undefined\n    ) {\n      this.firstTokenTime = Date.now() - this.streamStartTime;\n    }\n  }\n\n  private getTiming(): MessageTiming | undefined {\n    if (this.streamStartTime === undefined) return undefined;\n\n    const now = Date.now();\n    const totalStreamTime = now - this.streamStartTime;\n    const tokenCount =\n      this.totalChunks > 0\n        ? Math.ceil(\n            Array.from(this.textParts.values()).reduce(\n              (sum, p) => sum + p.buffer.length,\n              0,\n            ) / 4,\n          )\n        : undefined;\n    const tokensPerSecond =\n      tokenCount && totalStreamTime > 0\n        ? (tokenCount / totalStreamTime) * 1000\n        : undefined;\n\n    return {\n      streamStartTime: this.streamStartTime,\n      ...(this.firstTokenTime !== undefined\n        ? { firstTokenTime: this.firstTokenTime }\n        : {}),\n      totalStreamTime,\n      ...(tokenCount !== undefined ? { tokenCount } : {}),\n      ...(tokensPerSecond !== undefined ? { tokensPerSecond } : {}),\n      totalChunks: this.totalChunks,\n      toolCallCount: this.toolCalls.size,\n    };\n  }\n\n  private handleReasoningStart(messageId?: string): void {\n    if (!this.showThinking) return;\n    // A reasoning block acts as a boundary: anonymous text arriving after it\n    // should be a new part, not appended to any pre-reasoning text.\n    this.activeTextMessageId = undefined;\n    const key = messageId ?? `__auto-reasoning-${++this.reasoningPartCounter}`;\n    if (!this.reasoningParts.has(key)) {\n      this.reasoningParts.set(key, \"\");\n      this.partOrder.push({ kind: \"reasoning\", key });\n    }\n    this.activeReasoningKey = key;\n    this.emit();\n  }\n\n  private handleReasoningContent(delta: string, messageId?: string): void {\n    if (!this.showThinking || !delta) return;\n    if (!this.activeReasoningKey) {\n      // Content arrived without a preceding START — create the slot lazily.\n      this.handleReasoningStart(messageId);\n    }\n    const key = this.activeReasoningKey;\n    if (!key) return;\n    this.reasoningParts.set(key, (this.reasoningParts.get(key) ?? \"\") + delta);\n    this.emit();\n  }\n\n  private handleReasoningEnd(): void {\n    if (!this.showThinking) return;\n    this.activeReasoningKey = undefined;\n    this.emit();\n  }\n}\n"],"mappings":";;;AAYA,MAAa,2BAA2B;AAoBxC,MAAM,yBAAyB;;;;;;;AAe/B,IAAa,gBAAb,MAA2B;CACzB;CACA;CACA;CACA;CAEA;CACA;CACA,4BAA6B,IAAI,IAG/B;CACF;CACA,iCAAkC,IAAI,IAAoB;CAC1D;CACA,uBAA+B;CAC/B,4BAA6B,IAAI,IAA2B;CAC5D;CACA,YAIM,CAAC;CACP,kBAA0B;CAC1B,0BAAkC;CAElC;CACA;CACA,cAAsB;CAEtB,YAAY,SAA+B;EACzC,KAAK,aAAa,QAAQ;EAC1B,KAAK,eAAe,QAAQ;EAC5B,KAAK,SAAS,QAAQ;EACtB,KAAK,oBAAoB,QAAQ;CACnC;CAEA,OAAO,OAAwB;EAC7B,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,eAAe;IACpB,KAAK,eAAe,MAAM;IAC1B,KAAK,qBAAqB,KAAA;IAC1B,KAAK,uBAAuB;IAC5B,KAAK,UAAU,MAAM;IACrB,KAAK,yBAAyB,KAAA;IAC9B,KAAK,UAAU,SAAS;IACxB,KAAK,kBAAkB;IACvB,KAAK,sBAAsB,KAAA;IAC3B,KAAK,aAAa,KAAA;IAClB,KAAK,0BAA0B;IAC/B,KAAK,kBAAkB,KAAK,IAAI;IAChC,KAAK,iBAAiB,KAAA;IACtB,KAAK,cAAc;IACnB,KAAK,SAAS,EAAE,MAAM,UAAU;IAChC,KAAK,KAAK;IACV;GAEF,KAAK,gBAAgB;IACnB,IAAI,MAAM,SAAS,SAAS,aAAa;KACvC,KAAK,aAAa,MAAM,QAAQ;KAChC,KAAK,SAAS;MAAE,MAAM;MAAmB,QAAQ;KAAY;KAC7D,KAAK,KAAK;KACV;IACF;IAEA,KAAK,aAAa,KAAA;IAClB,MAAM,yBAAyB,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,CAAC,CAAC,MAChE,OAAO,GAAG,WAAW,KAAA,CACxB;IAEA,KAAK,SACH,MAAM,SAAS,SAAS,aAAa,CAAC,yBAClC;KAAE,MAAM;KAAY,QAAQ;IAAU,IACtC;KAAE,MAAM;KAAmB,QAAQ;IAAa;IACtD,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,SAAS;KACZ,MAAM;KACN,QAAQ;KACR,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,OAAO,MAAM,QAAQ,IAAI,CAAC;IAChE;IACA,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,SAAS;KAAE,MAAM;KAAc,QAAQ;IAAY;IACxD,KAAK,KAAK;IACV;GAGF,KAAK,sBAAsB;IACzB,KAAK,sBAAsB,MAAM,SAAS;IAC1C,MAAM,KAAK,KAAK,iBAAiB,MAAM,SAAS;IAChD,IAAI,IACF,KAAK,oBAAoB,EAAE;IAE7B,KAAK,KAAK;IACV;GACF;GACA,KAAK;GACL,KAAK,sBAAsB;IACzB,MAAM,aAAa,eAAe,QAAQ,MAAM,YAAY,KAAA;IAC5D,KAAK,sBAAsB,UAAU;IACrC,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,iBAAiB;IACtB,MAAM,KAAK,KAAK,qBAAqB,UAAU;IAC/C,KAAK,WAAW,IAAI,MAAM,KAAK;IAC/B,KAAK;IACL,KAAK,KAAK;IACV;GACF;GACA,KAAK;IACH,KAAK,sBAAsB,MAAM,SAAS;IAC1C,IAAI,MAAM,aAAa,KAAK,wBAAwB,MAAM,WACxD,KAAK,sBAAsB,KAAA;IAE7B,KAAK,KAAK;IACV;GAGF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,qBACH,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA;GACF,KAAK;IACH,KAAK,uBAAuB,MAAM,KAAK;IACvC,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;IACH,KAAK,uBACH,MAAM,OACN,eAAe,QAAQ,MAAM,YAAY,KAAA,CAC3C;IACA,KAAK;IACL,KAAK,iBAAiB;IACtB;GACF,KAAK;GACL,KAAK;GACL,KAAK;GACL,KAAK;IACH,KAAK,mBAAmB;IACxB;GAEF,KAAK;IACH,KAAK,sBAAsB,MAAM,eAAe;IAChD,KAAK,cACH,MAAM,YACN,MAAM,cACN,MAAM,eACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK;GACL,KAAK;IACH,IAAI,MAAM,SAAS,mBACjB,KAAK,sBAAsB,MAAM,eAAe;IAElD,IAAI,CAAC,MAAM,OAAO;IAClB,KAAK,eAAe,MAAM,YAAY,MAAM,KAAK;IACjD,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,KAAK;IACV;GAEF,KAAK;IACH,KAAK,eACH,MAAM,YACN,MAAM,WAAW,IACjB,MAAM,SAAS,SAAS,QAAQ,KAAA,GAChC,MAAM,SACR;IACA,KAAK,KAAK;IACV;GAEF,KAAK,qBAAqB;IACxB,IAAI,MAAM,iBAAiB,wBAAwB;IACnD,MAAM,KAAK,KAAK;IAChB,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;IAC5C,MAAM,cAAc,MAAM,QAAQ;IAClC,IACE,SACA,OAAO,gBAAgB,YACvB,YAAY,WAAW,GACvB;KACA,MAAM,oBAAoB;KAC1B,KAAK,KAAK;IACZ;IACA;GACF;GAEA,SACE,KAAK,OAAO,QAAQ,mCAAmC,KAAK;EAEhE;CACF;CAEA,sBAA8B,WAAqC;EACjE,IAAI,KAAK,2BAA2B,CAAC,WAAW;EAChD,KAAK,0BAA0B;EAC/B,KAAK,oBAAoB,SAAS;CACpC;CAEA,iBAA+B;EAC7B,KAAK,UAAU,MAAM;CACvB;CAEA,kBAAkC;EAChC,KAAK,mBAAmB;EACxB,OAAO,QAAQ,KAAK;CACtB;CAEA,iBAAyB,WAA4B;EACnD,MAAM,KAAK,aAAa,KAAK,gBAAgB;EAC7C,KAAK,eAAe,EAAE;EACtB,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,qBAA6B,WAA4B;EACvD,IAAI,WAAW;GACb,KAAK,eAAe,SAAS;GAC7B,KAAK,sBAAsB;GAC3B,OAAO;EACT;EAEA,IAAI,KAAK,qBACP,OAAO,KAAK;EAGd,MAAM,YAAY,KAAK,gBAAgB;EACvC,KAAK,eAAe,SAAS;EAC7B,KAAK,sBAAsB;EAC3B,OAAO;CACT;CAEA,eAAuB,IAAkB;EACvC,IAAI,CAAC,KAAK,UAAU,IAAI,EAAE,GAAG;GAC3B,KAAK,UAAU,IAAI,IAAI;IAAE,QAAQ;IAAI,SAAS;GAAM,CAAC;GACrD,IACE,CAAC,KAAK,UAAU,MAAM,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,EAAE,GAEtE,KAAK,UAAU,KAAK;IAAE,MAAM;IAAQ,KAAK;GAAG,CAAC;EAEjD;CACF;CAEA,oBAA4B,IAAkB;EAC5C,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;CAClB;CAEA,WAAmB,IAAY,OAAqB;EAClD,KAAK,eAAe,EAAE;EACtB,MAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;EACnC,IAAI,CAAC,OAAO;EACZ,MAAM,UAAU;EAChB,MAAM,UAAU;CAClB;CAEA,cACE,IACA,MACA,iBACA;EACA,IAAI,CAAC,IAAI;EAGT,KAAK,sBAAsB,KAAA;EAC3B,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,QAAuB;GAC3B,YAAY;GACZ,cAAc,QAAQ;GACtB,UAAU;GACV,YAAY,KAAA;GACZ,QAAQ,KAAA;GACR,SAAS,KAAA;EACX;EACA,IAAI,iBACF,MAAM,kBAAkB;EAE1B,KAAK,UAAU,IAAI,IAAI,KAAK;CAC9B;CAEA,eAAuB,IAAwB,OAAe;EAC5D,MAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI,KAAA;EAC5C,IAAI,CAAC,OAAO;EACZ,MAAM,YAAY;EAClB,IAAI;GACF,MAAM,SAAS,KAAK,MAAM,MAAM,QAAQ;GACxC,IAAI,UAAU,OAAO,WAAW,UAC9B,MAAM,aAAa;QAEnB,MAAM,aAAa,KAAA;EAEvB,QAAQ;GACN,MAAM,aAAa,KAAA;EACrB;CACF;CAEA,eACE,IACA,SACA,SACA,eACA;EACA,IAAI,CAAC,IAAI;EACT,IAAI,QAAQ,KAAK,UAAU,IAAI,EAAE;EACjC,IAAI,CAAC,OAAO;GACV,QAAQ;IACN,YAAY;IACZ,cAAc;IACd,UAAU;IACV,YAAY,KAAA;IACZ,QAAQ,KAAA;IACR,SAAS,KAAA;GACX;GACA,KAAK,UAAU,IAAI,IAAI,KAAK;EAC9B;EACA,IACE,CAAC,KAAK,UAAU,MACb,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe,EAC7D,GAEA,KAAK,UAAU,KAAK;GAAE,MAAM;GAAa,YAAY;EAAG,CAAC;EAE3D,MAAM,SAAS,KAAK,aAAa,OAAO;EACxC,MAAM,UAAU;EAChB,IAAI,eACF,MAAM,gBAAgB;EAExB,KAAK,yBAAyB;CAChC;CAEA,aAAqB,OAAwB;EAC3C,IAAI,CAAC,OAAO,OAAO;EACnB,IAAI;GACF,OAAO,KAAK,MAAM,KAAK;EACzB,QAAQ;GACN,OAAO;EACT;CACF;CAEA,OAAqB;EACnB,MAAM,WAAyC,CAAC;EAEhD,KAAK,MAAM,QAAQ,KAAK,WAAW;GACjC,IAAI,KAAK,SAAS,aAAa;IAC7B,IAAI,KAAK,cAAc;KACrB,MAAM,SAAS,KAAK,eAAe,IAAI,KAAK,GAAG,KAAK;KACpD,IAAI,OAAO,SAAS,KAAK,KAAK,uBAAuB,KAAK,KACxD,SAAS,KAAK;MAAE,MAAM;MAAa,MAAM;KAAO,CAAU;IAE9D;IACA;GACF;GAEA,IAAI,KAAK,SAAS,QAAQ;IACxB,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,GAAG;IACzC,IAAI,OAAO,SACT,SAAS,KAAK;KAAE,MAAM;KAAQ,MAAM,MAAM;IAAO,CAAU;IAE7D;GACF;GAEA,MAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,UAAU;GAChD,IAAI,CAAC,OAAO;GACZ,MAAM,WAAgC;IACpC,MAAM;IACN,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,MAAO,MAAM,cAAc,CAAC;IAC5B,UAAU,MAAM;IAChB,GAAI,MAAM,WAAW,KAAA,IAAY,EAAE,QAAQ,MAAM,OAAO,IAAI,CAAC;IAC7D,GAAI,MAAM,YAAY,KAAA,IAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;IAChE,GAAI,MAAM,oBACN,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,MAAM,kBAAkB,EAAE,EAAE,IACzD,CAAC;IACL,GAAI,MAAM,kBAAkB,EAAE,UAAU,MAAM,gBAAgB,IAAI,CAAC;IACnE,GAAI,MAAM,gBACN,EAAE,wBAAwB,MAAM,cAAc,IAC9C,CAAC;GACP;GACA,SAAS,KAAK,QAAQ;EACxB;EAEA,MAAM,SAAS,KAAK,UAAU;EAC9B,MAAM,WAAW;GACf,GAAI,SAAS,EAAE,OAAO,IAAI,CAAC;GAC3B,GAAI,KAAK,aACL,EACE,QAAQ,GACL,2BAA2B,EAC1B,YAAY,KAAK,WACnB,EACF,EACF,IACA,CAAC;EACP;EACA,MAAM,SAA6B;GACjC,SAAS;GACT,GAAI,KAAK,SAAS,EAAE,QAAQ,KAAK,OAAO,IAAI,KAAA;GAC5C,GAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,SAAS,IAAI,EAAE,SAAS,IAAI,KAAA;EACxD;EACA,KAAK,WAAW,MAAM;CACxB;CAEA,mBAAiC;EAC/B,IACE,KAAK,mBAAmB,KAAA,KACxB,KAAK,oBAAoB,KAAA,GAEzB,KAAK,iBAAiB,KAAK,IAAI,IAAI,KAAK;CAE5C;CAEA,YAA+C;EAC7C,IAAI,KAAK,oBAAoB,KAAA,GAAW,OAAO,KAAA;EAG/C,MAAM,kBADM,KAAK,IACS,IAAI,KAAK;EACnC,MAAM,aACJ,KAAK,cAAc,IACf,KAAK,KACH,MAAM,KAAK,KAAK,UAAU,OAAO,CAAC,CAAC,CAAC,QACjC,KAAK,MAAM,MAAM,EAAE,OAAO,QAC3B,CACF,IAAI,CACN,IACA,KAAA;EACN,MAAM,kBACJ,cAAc,kBAAkB,IAC3B,aAAa,kBAAmB,MACjC,KAAA;EAEN,OAAO;GACL,iBAAiB,KAAK;GACtB,GAAI,KAAK,mBAAmB,KAAA,IACxB,EAAE,gBAAgB,KAAK,eAAe,IACtC,CAAC;GACL;GACA,GAAI,eAAe,KAAA,IAAY,EAAE,WAAW,IAAI,CAAC;GACjD,GAAI,oBAAoB,KAAA,IAAY,EAAE,gBAAgB,IAAI,CAAC;GAC3D,aAAa,KAAK;GAClB,eAAe,KAAK,UAAU;EAChC;CACF;CAEA,qBAA6B,WAA0B;EACrD,IAAI,CAAC,KAAK,cAAc;EAGxB,KAAK,sBAAsB,KAAA;EAC3B,MAAM,MAAM,aAAa,oBAAoB,EAAE,KAAK;EACpD,IAAI,CAAC,KAAK,eAAe,IAAI,GAAG,GAAG;GACjC,KAAK,eAAe,IAAI,KAAK,EAAE;GAC/B,KAAK,UAAU,KAAK;IAAE,MAAM;IAAa;GAAI,CAAC;EAChD;EACA,KAAK,qBAAqB;EAC1B,KAAK,KAAK;CACZ;CAEA,uBAA+B,OAAe,WAA0B;EACtE,IAAI,CAAC,KAAK,gBAAgB,CAAC,OAAO;EAClC,IAAI,CAAC,KAAK,oBAER,KAAK,qBAAqB,SAAS;EAErC,MAAM,MAAM,KAAK;EACjB,IAAI,CAAC,KAAK;EACV,KAAK,eAAe,IAAI,MAAM,KAAK,eAAe,IAAI,GAAG,KAAK,MAAM,KAAK;EACzE,KAAK,KAAK;CACZ;CAEA,qBAAmC;EACjC,IAAI,CAAC,KAAK,cAAc;EACxB,KAAK,qBAAqB,KAAA;EAC1B,KAAK,KAAK;CACZ;AACF"}

package/dist/runtime/adapter/subscriber.d.ts

@@ -64,6 +64,9 @@ type Subscriber = {
   onToolCallResultEvent?: (payload: {
     event: unknown;
   }) => void;
+  onActivitySnapshotEvent?: (payload: {
+    event: unknown;
+  }) => void;
   onStateSnapshotEvent?: (payload: {
     event: unknown;
   }) => void;

package/dist/runtime/adapter/subscriber.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"subscriber.d.ts","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"mappings":";;;;KAMK,QAAA,IAAY,KAAgB,EAAT,SAAS;AAAA,KAE5B,UAAA;EACH,OAAA,IAAW,OAAA;IAAW,KAAA;EAAA;EACtB,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,yBAAA,IAA6B,OAAA;IAAW,KAAA;EAAA;EACxC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,+BAAA,IAAmC,OAAA;IAAW,KAAA;EAAA;EAC9C,iCAAA,IAAqC,OAAA;IAAW,KAAA;EAAA;EAChD,6BAAA,IAAiC,OAAA;IAAW,KAAA;EAAA;EAC5C,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,4BAAA,IAAgC,OAAA;IAAW,KAAA;EAAA;EAC3C,8BAAA,IAAkC,OAAA;IAAW,KAAA;EAAA;EAC7C,0BAAA,IAA8B,OAAA;IAAW,KAAA;EAAA;EACzC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,iBAAA,IAAqB,OAAA;IAAW,KAAA;EAAA;EAChC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,aAAA,IAAiB,OAAA;IAAW,KAAA;EAAA;EAC5B,UAAA,IAAc,OAAA;IAAW,KAAA;EAAA;EACzB,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,cAAA;EACA,WAAA,IAAe,OAAA;IAAW,KAAA,EAAO,KAAK;EAAA;AAAA;AAAA,KAuBnC,iBAAA;EACH,QAAA,EAAU,QAAA;EACV,KAAA;EACA,MAAA,GAAS,MAAA;EACT,WAAA,IAAe,KAAA,EAAO,KAAA;AAAA;AAAA,cAGX,oBAAA,GACX,OAAA,EAAS,iBAAA,KACR,UAwFF;AAAA,KAEW,cAAA,GAAiB,UAAU,QAAQ,oBAAA"}
+{"version":3,"file":"subscriber.d.ts","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"mappings":";;;;KAMK,QAAA,IAAY,KAAgB,EAAT,SAAS;AAAA,KAE5B,UAAA;EACH,OAAA,IAAW,OAAA;IAAW,KAAA;EAAA;EACtB,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,yBAAA,IAA6B,OAAA;IAAW,KAAA;EAAA;EACxC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,+BAAA,IAAmC,OAAA;IAAW,KAAA;EAAA;EAC9C,iCAAA,IAAqC,OAAA;IAAW,KAAA;EAAA;EAChD,6BAAA,IAAiC,OAAA;IAAW,KAAA;EAAA;EAC5C,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,4BAAA,IAAgC,OAAA;IAAW,KAAA;EAAA;EAC3C,8BAAA,IAAkC,OAAA;IAAW,KAAA;EAAA;EAC7C,0BAAA,IAA8B,OAAA;IAAW,KAAA;EAAA;EACzC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,mBAAA,IAAuB,OAAA;IAAW,KAAA;EAAA;EAClC,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,qBAAA,IAAyB,OAAA;IAAW,KAAA;EAAA;EACpC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,oBAAA,IAAwB,OAAA;IAAW,KAAA;EAAA;EACnC,iBAAA,IAAqB,OAAA;IAAW,KAAA;EAAA;EAChC,uBAAA,IAA2B,OAAA;IAAW,KAAA;EAAA;EACtC,aAAA,IAAiB,OAAA;IAAW,KAAA;EAAA;EAC5B,UAAA,IAAc,OAAA;IAAW,KAAA;EAAA;EACzB,kBAAA,IAAsB,OAAA;IAAW,KAAA;EAAA;EACjC,cAAA;EACA,WAAA,IAAe,OAAA;IAAW,KAAA,EAAO,KAAK;EAAA;AAAA;AAAA,KAuBnC,iBAAA;EACH,QAAA,EAAU,QAAA;EACV,KAAA;EACA,MAAA,GAAS,MAAA;EACT,WAAA,IAAe,KAAA,EAAO,KAAA;AAAA;AAAA,cAGX,oBAAA,GACX,OAAA,EAAS,iBAAA,KACR,UA0FF;AAAA,KAEW,cAAA,GAAiB,UAAU,QAAQ,oBAAA"}

package/dist/runtime/adapter/subscriber.js

@@ -47,6 +47,7 @@ const createAgUiSubscriber = (options) => {
 		onToolCallEndEvent: ({ event }) => dispatchIfValid(event, "TOOL_CALL_END"),
 		onToolCallChunkEvent: ({ event }) => dispatchIfValid(event, "TOOL_CALL_CHUNK"),
 		onToolCallResultEvent: ({ event }) => dispatchIfValid(event, "TOOL_CALL_RESULT"),
+		onActivitySnapshotEvent: ({ event }) => dispatchIfValid(event, "ACTIVITY_SNAPSHOT"),
 		onStateSnapshotEvent: ({ event }) => dispatchIfValid(event, "STATE_SNAPSHOT"),
 		onStateDeltaEvent: ({ event }) => dispatchIfValid(event, "STATE_DELTA"),
 		onMessagesSnapshotEvent: ({ event }) => dispatchIfValid(event, "MESSAGES_SNAPSHOT"),

package/dist/runtime/adapter/subscriber.js.map

@@ -1 +1 @@
-{"version":3,"file":"subscriber.js","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"sourcesContent":["\"use client\";\n\nimport type { AgUiEvent } from \"../types\";\nimport { parseAgUiEvent } from \"../event-parser\";\nimport type { Logger } from \"../logger\";\n\ntype Dispatch = (event: AgUiEvent) => void;\n\ntype Subscriber = {\n  onEvent?: (payload: { event: unknown }) => void;\n  onTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onTextMessageChunkEvent?: (payload: { event: unknown }) => void;\n  onThinkingStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingEndEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageContentEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallStartEvent?: (payload: { event: unknown }) => void;\n  onToolCallArgsEvent?: (payload: { event: unknown }) => void;\n  onToolCallEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallChunkEvent?: (payload: { event: unknown }) => void;\n  onToolCallResultEvent?: (payload: { event: unknown }) => void;\n  onStateSnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateDeltaEvent?: (payload: { event: unknown }) => void;\n  onMessagesSnapshotEvent?: (payload: { event: unknown }) => void;\n  onCustomEvent?: (payload: { event: unknown }) => void;\n  onRawEvent?: (payload: { event: unknown }) => void;\n  onRunFinishedEvent?: (payload: { event: unknown }) => void;\n  onRunFinalized?: () => void;\n  onRunFailed?: (payload: { error: Error }) => void;\n};\n\nconst ensureEvent = (\n  raw: unknown,\n  type: AgUiEvent[\"type\"],\n  logger?: Logger,\n): AgUiEvent | null => {\n  const parseOptions = logger ? { logger } : undefined;\n  let parsed: AgUiEvent | null;\n  if (raw && typeof raw === \"object\") {\n    const payload = raw as Record<string, unknown>;\n    if (typeof payload.type === \"string\") {\n      parsed = parseAgUiEvent(payload, parseOptions);\n    } else {\n      parsed = parseAgUiEvent({ type, ...payload }, parseOptions);\n    }\n  } else {\n    parsed = parseAgUiEvent({ type }, parseOptions);\n  }\n  return parsed && parsed.type === type ? parsed : null;\n};\n\ntype SubscriberOptions = {\n  dispatch: Dispatch;\n  runId: string;\n  logger?: Logger;\n  onRunFailed?: (error: Error) => void;\n};\n\nexport const createAgUiSubscriber = (\n  options: SubscriberOptions,\n): Subscriber => {\n  const { dispatch, runId, logger, onRunFailed } = options;\n  let runFinishedDispatched = false;\n  const dispatchIfValid = (raw: unknown, type: AgUiEvent[\"type\"]) => {\n    const event = ensureEvent(raw, type, logger);\n    if (!event) return;\n    dispatch(event);\n  };\n  return {\n    onEvent: ({ event }) => {\n      const typeCandidate =\n        event && typeof event === \"object\"\n          ? (event as Record<string, unknown>).type\n          : undefined;\n      if (typeof typeCandidate === \"string\") {\n        // Typed handlers will receive this via the discriminated callbacks; avoid duplicates.\n        return;\n      }\n      const parsed = parseAgUiEvent(event);\n      if (parsed) dispatch(parsed);\n    },\n    onTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_START\"),\n    onTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CONTENT\"),\n    onTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_END\"),\n    onTextMessageChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CHUNK\"),\n    onThinkingStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_START\"),\n    onThinkingEndEvent: ({ event }) => dispatchIfValid(event, \"THINKING_END\"),\n    onThinkingTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_START\"),\n    onThinkingTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_CONTENT\"),\n    onThinkingTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_END\"),\n    onReasoningStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_START\"),\n    onReasoningEndEvent: ({ event }) => dispatchIfValid(event, \"REASONING_END\"),\n    onReasoningMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_START\"),\n    onReasoningMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_CONTENT\"),\n    onReasoningMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_END\"),\n    onToolCallStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_START\"),\n    onToolCallArgsEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_ARGS\"),\n    onToolCallEndEvent: ({ event }) => dispatchIfValid(event, \"TOOL_CALL_END\"),\n    onToolCallChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_CHUNK\"),\n    onToolCallResultEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_RESULT\"),\n    onStateSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"STATE_SNAPSHOT\"),\n    onStateDeltaEvent: ({ event }) => dispatchIfValid(event, \"STATE_DELTA\"),\n    onMessagesSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"MESSAGES_SNAPSHOT\"),\n    onCustomEvent: ({ event }) => dispatchIfValid(event, \"CUSTOM\"),\n    onRawEvent: ({ event }) => dispatchIfValid(event, \"RAW\"),\n    onRunFinishedEvent: ({ event }) => {\n      const parsed = ensureEvent(event, \"RUN_FINISHED\", logger);\n      if (!parsed) return;\n      runFinishedDispatched = true;\n      dispatch(parsed);\n    },\n    onRunFinalized: () => {\n      if (runFinishedDispatched) return;\n      dispatch({ type: \"RUN_FINISHED\", runId });\n    },\n    onRunFailed: ({ error }) => {\n      onRunFailed?.(error);\n      const message =\n        typeof error.message === \"string\" ? error.message : \"Run failed\";\n      const code =\n        typeof (error as any)?.code === \"string\"\n          ? (error as any).code\n          : undefined;\n      dispatch({\n        type: \"RUN_ERROR\" as const,\n        ...(message !== undefined ? { message } : {}),\n        ...(code !== undefined ? { code } : {}),\n      } satisfies AgUiEvent);\n    },\n  };\n};\n\nexport type AgUiSubscriber = ReturnType<typeof createAgUiSubscriber>;\n"],"mappings":";;;AAuCA,MAAM,eACJ,KACA,MACA,WACqB;CACrB,MAAM,eAAe,SAAS,EAAE,OAAO,IAAI,KAAA;CAC3C,IAAI;CACJ,IAAI,OAAO,OAAO,QAAQ,UAAU;EAClC,MAAM,UAAU;EAChB,IAAI,OAAO,QAAQ,SAAS,UAC1B,SAAS,eAAe,SAAS,YAAY;OAE7C,SAAS,eAAe;GAAE;GAAM,GAAG;EAAQ,GAAG,YAAY;CAE9D,OACE,SAAS,eAAe,EAAE,KAAK,GAAG,YAAY;CAEhD,OAAO,UAAU,OAAO,SAAS,OAAO,SAAS;AACnD;AASA,MAAa,wBACX,YACe;CACf,MAAM,EAAE,UAAU,OAAO,QAAQ,gBAAgB;CACjD,IAAI,wBAAwB;CAC5B,MAAM,mBAAmB,KAAc,SAA4B;EACjE,MAAM,QAAQ,YAAY,KAAK,MAAM,MAAM;EAC3C,IAAI,CAAC,OAAO;EACZ,SAAS,KAAK;CAChB;CACA,OAAO;EACL,UAAU,EAAE,YAAY;GAKtB,IAAI,QAHF,SAAS,OAAO,UAAU,WACrB,MAAkC,OACnC,KAAA,OACuB,UAE3B;GAEF,MAAM,SAAS,eAAe,KAAK;GACnC,IAAI,QAAQ,SAAS,MAAM;EAC7B;EACA,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,4BAA4B,EAAE,YAC5B,gBAAgB,OAAO,sBAAsB;EAC/C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,cAAc;EACxE,kCAAkC,EAAE,YAClC,gBAAgB,OAAO,6BAA6B;EACtD,oCAAoC,EAAE,YACpC,gBAAgB,OAAO,+BAA+B;EACxD,gCAAgC,EAAE,YAChC,gBAAgB,OAAO,2BAA2B;EACpD,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EAC1E,+BAA+B,EAAE,YAC/B,gBAAgB,OAAO,yBAAyB;EAClD,iCAAiC,EAAE,YACjC,gBAAgB,OAAO,2BAA2B;EACpD,6BAA6B,EAAE,YAC7B,gBAAgB,OAAO,uBAAuB;EAChD,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YACtB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EACzE,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,oBAAoB,EAAE,YAAY,gBAAgB,OAAO,aAAa;EACtE,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,gBAAgB,EAAE,YAAY,gBAAgB,OAAO,QAAQ;EAC7D,aAAa,EAAE,YAAY,gBAAgB,OAAO,KAAK;EACvD,qBAAqB,EAAE,YAAY;GACjC,MAAM,SAAS,YAAY,OAAO,gBAAgB,MAAM;GACxD,IAAI,CAAC,QAAQ;GACb,wBAAwB;GACxB,SAAS,MAAM;EACjB;EACA,sBAAsB;GACpB,IAAI,uBAAuB;GAC3B,SAAS;IAAE,MAAM;IAAgB;GAAM,CAAC;EAC1C;EACA,cAAc,EAAE,YAAY;GAC1B,cAAc,KAAK;GACnB,MAAM,UACJ,OAAO,MAAM,YAAY,WAAW,MAAM,UAAU;GACtD,MAAM,OACJ,OAAQ,OAAe,SAAS,WAC3B,MAAc,OACf,KAAA;GACN,SAAS;IACP,MAAM;IACN,GAAI,YAAY,KAAA,IAAY,EAAE,QAAQ,IAAI,CAAC;IAC3C,GAAI,SAAS,KAAA,IAAY,EAAE,KAAK,IAAI,CAAC;GACvC,CAAqB;EACvB;CACF;AACF"}
+{"version":3,"file":"subscriber.js","names":[],"sources":["../../../src/runtime/adapter/subscriber.ts"],"sourcesContent":["\"use client\";\n\nimport type { AgUiEvent } from \"../types\";\nimport { parseAgUiEvent } from \"../event-parser\";\nimport type { Logger } from \"../logger\";\n\ntype Dispatch = (event: AgUiEvent) => void;\n\ntype Subscriber = {\n  onEvent?: (payload: { event: unknown }) => void;\n  onTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onTextMessageChunkEvent?: (payload: { event: unknown }) => void;\n  onThinkingStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingEndEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningEndEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageStartEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageContentEvent?: (payload: { event: unknown }) => void;\n  onReasoningMessageEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallStartEvent?: (payload: { event: unknown }) => void;\n  onToolCallArgsEvent?: (payload: { event: unknown }) => void;\n  onToolCallEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallChunkEvent?: (payload: { event: unknown }) => void;\n  onToolCallResultEvent?: (payload: { event: unknown }) => void;\n  onActivitySnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateSnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateDeltaEvent?: (payload: { event: unknown }) => void;\n  onMessagesSnapshotEvent?: (payload: { event: unknown }) => void;\n  onCustomEvent?: (payload: { event: unknown }) => void;\n  onRawEvent?: (payload: { event: unknown }) => void;\n  onRunFinishedEvent?: (payload: { event: unknown }) => void;\n  onRunFinalized?: () => void;\n  onRunFailed?: (payload: { error: Error }) => void;\n};\n\nconst ensureEvent = (\n  raw: unknown,\n  type: AgUiEvent[\"type\"],\n  logger?: Logger,\n): AgUiEvent | null => {\n  const parseOptions = logger ? { logger } : undefined;\n  let parsed: AgUiEvent | null;\n  if (raw && typeof raw === \"object\") {\n    const payload = raw as Record<string, unknown>;\n    if (typeof payload.type === \"string\") {\n      parsed = parseAgUiEvent(payload, parseOptions);\n    } else {\n      parsed = parseAgUiEvent({ type, ...payload }, parseOptions);\n    }\n  } else {\n    parsed = parseAgUiEvent({ type }, parseOptions);\n  }\n  return parsed && parsed.type === type ? parsed : null;\n};\n\ntype SubscriberOptions = {\n  dispatch: Dispatch;\n  runId: string;\n  logger?: Logger;\n  onRunFailed?: (error: Error) => void;\n};\n\nexport const createAgUiSubscriber = (\n  options: SubscriberOptions,\n): Subscriber => {\n  const { dispatch, runId, logger, onRunFailed } = options;\n  let runFinishedDispatched = false;\n  const dispatchIfValid = (raw: unknown, type: AgUiEvent[\"type\"]) => {\n    const event = ensureEvent(raw, type, logger);\n    if (!event) return;\n    dispatch(event);\n  };\n  return {\n    onEvent: ({ event }) => {\n      const typeCandidate =\n        event && typeof event === \"object\"\n          ? (event as Record<string, unknown>).type\n          : undefined;\n      if (typeof typeCandidate === \"string\") {\n        // Typed handlers will receive this via the discriminated callbacks; avoid duplicates.\n        return;\n      }\n      const parsed = parseAgUiEvent(event);\n      if (parsed) dispatch(parsed);\n    },\n    onTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_START\"),\n    onTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CONTENT\"),\n    onTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_END\"),\n    onTextMessageChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TEXT_MESSAGE_CHUNK\"),\n    onThinkingStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_START\"),\n    onThinkingEndEvent: ({ event }) => dispatchIfValid(event, \"THINKING_END\"),\n    onThinkingTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_START\"),\n    onThinkingTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_CONTENT\"),\n    onThinkingTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"THINKING_TEXT_MESSAGE_END\"),\n    onReasoningStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_START\"),\n    onReasoningEndEvent: ({ event }) => dispatchIfValid(event, \"REASONING_END\"),\n    onReasoningMessageStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_START\"),\n    onReasoningMessageContentEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_CONTENT\"),\n    onReasoningMessageEndEvent: ({ event }) =>\n      dispatchIfValid(event, \"REASONING_MESSAGE_END\"),\n    onToolCallStartEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_START\"),\n    onToolCallArgsEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_ARGS\"),\n    onToolCallEndEvent: ({ event }) => dispatchIfValid(event, \"TOOL_CALL_END\"),\n    onToolCallChunkEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_CHUNK\"),\n    onToolCallResultEvent: ({ event }) =>\n      dispatchIfValid(event, \"TOOL_CALL_RESULT\"),\n    onActivitySnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"ACTIVITY_SNAPSHOT\"),\n    onStateSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"STATE_SNAPSHOT\"),\n    onStateDeltaEvent: ({ event }) => dispatchIfValid(event, \"STATE_DELTA\"),\n    onMessagesSnapshotEvent: ({ event }) =>\n      dispatchIfValid(event, \"MESSAGES_SNAPSHOT\"),\n    onCustomEvent: ({ event }) => dispatchIfValid(event, \"CUSTOM\"),\n    onRawEvent: ({ event }) => dispatchIfValid(event, \"RAW\"),\n    onRunFinishedEvent: ({ event }) => {\n      const parsed = ensureEvent(event, \"RUN_FINISHED\", logger);\n      if (!parsed) return;\n      runFinishedDispatched = true;\n      dispatch(parsed);\n    },\n    onRunFinalized: () => {\n      if (runFinishedDispatched) return;\n      dispatch({ type: \"RUN_FINISHED\", runId });\n    },\n    onRunFailed: ({ error }) => {\n      onRunFailed?.(error);\n      const message =\n        typeof error.message === \"string\" ? error.message : \"Run failed\";\n      const code =\n        typeof (error as any)?.code === \"string\"\n          ? (error as any).code\n          : undefined;\n      dispatch({\n        type: \"RUN_ERROR\" as const,\n        ...(message !== undefined ? { message } : {}),\n        ...(code !== undefined ? { code } : {}),\n      } satisfies AgUiEvent);\n    },\n  };\n};\n\nexport type AgUiSubscriber = ReturnType<typeof createAgUiSubscriber>;\n"],"mappings":";;;AAwCA,MAAM,eACJ,KACA,MACA,WACqB;CACrB,MAAM,eAAe,SAAS,EAAE,OAAO,IAAI,KAAA;CAC3C,IAAI;CACJ,IAAI,OAAO,OAAO,QAAQ,UAAU;EAClC,MAAM,UAAU;EAChB,IAAI,OAAO,QAAQ,SAAS,UAC1B,SAAS,eAAe,SAAS,YAAY;OAE7C,SAAS,eAAe;GAAE;GAAM,GAAG;EAAQ,GAAG,YAAY;CAE9D,OACE,SAAS,eAAe,EAAE,KAAK,GAAG,YAAY;CAEhD,OAAO,UAAU,OAAO,SAAS,OAAO,SAAS;AACnD;AASA,MAAa,wBACX,YACe;CACf,MAAM,EAAE,UAAU,OAAO,QAAQ,gBAAgB;CACjD,IAAI,wBAAwB;CAC5B,MAAM,mBAAmB,KAAc,SAA4B;EACjE,MAAM,QAAQ,YAAY,KAAK,MAAM,MAAM;EAC3C,IAAI,CAAC,OAAO;EACZ,SAAS,KAAK;CAChB;CACA,OAAO;EACL,UAAU,EAAE,YAAY;GAKtB,IAAI,QAHF,SAAS,OAAO,UAAU,WACrB,MAAkC,OACnC,KAAA,OACuB,UAE3B;GAEF,MAAM,SAAS,eAAe,KAAK;GACnC,IAAI,QAAQ,SAAS,MAAM;EAC7B;EACA,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,4BAA4B,EAAE,YAC5B,gBAAgB,OAAO,sBAAsB;EAC/C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,oBAAoB;EAC7C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,cAAc;EACxE,kCAAkC,EAAE,YAClC,gBAAgB,OAAO,6BAA6B;EACtD,oCAAoC,EAAE,YACpC,gBAAgB,OAAO,+BAA+B;EACxD,gCAAgC,EAAE,YAChC,gBAAgB,OAAO,2BAA2B;EACpD,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EAC1E,+BAA+B,EAAE,YAC/B,gBAAgB,OAAO,yBAAyB;EAClD,iCAAiC,EAAE,YACjC,gBAAgB,OAAO,2BAA2B;EACpD,6BAA6B,EAAE,YAC7B,gBAAgB,OAAO,uBAAuB;EAChD,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,sBAAsB,EAAE,YACtB,gBAAgB,OAAO,gBAAgB;EACzC,qBAAqB,EAAE,YAAY,gBAAgB,OAAO,eAAe;EACzE,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,iBAAiB;EAC1C,wBAAwB,EAAE,YACxB,gBAAgB,OAAO,kBAAkB;EAC3C,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,uBAAuB,EAAE,YACvB,gBAAgB,OAAO,gBAAgB;EACzC,oBAAoB,EAAE,YAAY,gBAAgB,OAAO,aAAa;EACtE,0BAA0B,EAAE,YAC1B,gBAAgB,OAAO,mBAAmB;EAC5C,gBAAgB,EAAE,YAAY,gBAAgB,OAAO,QAAQ;EAC7D,aAAa,EAAE,YAAY,gBAAgB,OAAO,KAAK;EACvD,qBAAqB,EAAE,YAAY;GACjC,MAAM,SAAS,YAAY,OAAO,gBAAgB,MAAM;GACxD,IAAI,CAAC,QAAQ;GACb,wBAAwB;GACxB,SAAS,MAAM;EACjB;EACA,sBAAsB;GACpB,IAAI,uBAAuB;GAC3B,SAAS;IAAE,MAAM;IAAgB;GAAM,CAAC;EAC1C;EACA,cAAc,EAAE,YAAY;GAC1B,cAAc,KAAK;GACnB,MAAM,UACJ,OAAO,MAAM,YAAY,WAAW,MAAM,UAAU;GACtD,MAAM,OACJ,OAAQ,OAAe,SAAS,WAC3B,MAAc,OACf,KAAA;GACN,SAAS;IACP,MAAM;IACN,GAAI,YAAY,KAAA,IAAY,EAAE,QAAQ,IAAI,CAAC;IAC3C,GAAI,SAAS,KAAA,IAAY,EAAE,KAAK,IAAI,CAAC;GACvC,CAAqB;EACvB;CACF;AACF"}

package/dist/runtime/event-parser.js

@@ -158,6 +158,15 @@ const parseAgUiEvent = (event, options) => {
 			type: "MESSAGES_SNAPSHOT",
 			messages: Array.isArray(payload.messages) ? payload.messages : []
 		};
+		case "ACTIVITY_SNAPSHOT": {
+			const activityType = getString("activityType");
+			if (!activityType || !isPlainObject(payload.content)) return null;
+			return {
+				type: "ACTIVITY_SNAPSHOT",
+				activityType,
+				content: payload.content
+			};
+		}
 		case "RAW": return withOptional({
 			type: "RAW",
 			event: payload.event

package/dist/runtime/event-parser.js.map

@@ -1 +1 @@
-{"version":3,"file":"event-parser.js","names":[],"sources":["../../src/runtime/event-parser.ts"],"sourcesContent":["import type { AgUiEvent, AgUiInterrupt, AgUiRunFinishedOutcome } from \"./types\";\nimport type { Logger } from \"./logger\";\n\nexport type ParseAgUiEventOptions = {\n  logger?: Logger;\n};\n\nconst isString = (value: unknown): value is string => typeof value === \"string\";\nconst isNonEmptyString = (value: unknown): value is string =>\n  isString(value) && value.length > 0;\n\nconst withOptional = <T extends object>(\n  base: T,\n  optionals: Record<string, unknown>,\n) => {\n  const definedEntries = Object.entries(optionals).filter(\n    ([, value]) => value !== undefined,\n  );\n  return definedEntries.length === 0\n    ? base\n    : ({ ...base, ...Object.fromEntries(definedEntries) } as T);\n};\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  !!value && typeof value === \"object\" && !Array.isArray(value);\n\nconst parseInterrupt = (raw: unknown): AgUiInterrupt | null => {\n  if (!isPlainObject(raw)) return null;\n  const id = raw.id;\n  const reason = raw.reason;\n  if (typeof id !== \"string\" || typeof reason !== \"string\") return null;\n  const interrupt: AgUiInterrupt = { id, reason };\n  if (typeof raw.message === \"string\") interrupt.message = raw.message;\n  if (typeof raw.toolCallId === \"string\") interrupt.toolCallId = raw.toolCallId;\n  if (typeof raw.expiresAt === \"string\") interrupt.expiresAt = raw.expiresAt;\n  if (isPlainObject(raw.responseSchema))\n    interrupt.responseSchema = raw.responseSchema;\n  if (isPlainObject(raw.metadata)) interrupt.metadata = raw.metadata;\n  return interrupt;\n};\n\nconst parseRunFinishedOutcome = (\n  raw: unknown,\n  logger: Logger | undefined,\n): AgUiRunFinishedOutcome | undefined => {\n  if (!isPlainObject(raw)) return undefined;\n  if (raw.type === \"success\") return { type: \"success\" };\n  if (raw.type === \"interrupt\") {\n    if (!Array.isArray(raw.interrupts)) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome missing interrupts array\",\n        raw,\n      );\n      return undefined;\n    }\n    const parsed = raw.interrupts\n      .map((entry) => parseInterrupt(entry))\n      .filter((entry): entry is AgUiInterrupt => entry !== null);\n    if (parsed.length === 0) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome has no valid interrupts\",\n        raw.interrupts,\n      );\n      return undefined;\n    }\n    return { type: \"interrupt\", interrupts: parsed };\n  }\n  return undefined;\n};\n\nexport const parseAgUiEvent = (\n  event: unknown,\n  options?: ParseAgUiEventOptions,\n): AgUiEvent | null => {\n  if (!event || typeof event !== \"object\") return null;\n  const payload = event as Record<string, unknown>;\n  const typeValue = payload.type;\n  if (!isString(typeValue)) return null;\n\n  const getString = (key: string) =>\n    isString(payload[key]) ? (payload[key] as string) : undefined;\n\n  switch (typeValue) {\n    case \"RUN_STARTED\": {\n      const runId = getString(\"runId\");\n      return runId ? { type: \"RUN_STARTED\", runId } : null;\n    }\n    case \"RUN_FINISHED\": {\n      const runId = getString(\"runId\");\n      if (!runId) return null;\n      return withOptional(\n        { type: \"RUN_FINISHED\" as const, runId },\n        {\n          outcome: parseRunFinishedOutcome(payload.outcome, options?.logger),\n        },\n      );\n    }\n    case \"RUN_CANCELLED\": {\n      const runId = getString(\"runId\");\n      return withOptional({ type: \"RUN_CANCELLED\" as const }, { runId });\n    }\n    case \"RUN_ERROR\": {\n      return withOptional(\n        { type: \"RUN_ERROR\" as const },\n        {\n          message: getString(\"message\"),\n          code: getString(\"code\"),\n        },\n      );\n    }\n    case \"TEXT_MESSAGE_START\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\");\n      if (!isNonEmptyString(delta)) return null;\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"TEXT_MESSAGE_END\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CHUNK\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CHUNK\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"THINKING_START\":\n      return withOptional(\n        { type: \"THINKING_START\" as const },\n        { title: getString(\"title\") },\n      );\n    case \"THINKING_TEXT_MESSAGE_START\":\n      return { type: \"THINKING_TEXT_MESSAGE_START\" };\n    case \"THINKING_TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"THINKING_TEXT_MESSAGE_CONTENT\", delta };\n    }\n    case \"THINKING_TEXT_MESSAGE_END\":\n      return { type: \"THINKING_TEXT_MESSAGE_END\" };\n    case \"THINKING_END\":\n      return { type: \"THINKING_END\" };\n    case \"REASONING_START\":\n      return withOptional(\n        { type: \"REASONING_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_START\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"REASONING_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"REASONING_MESSAGE_END\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_END\":\n      return withOptional(\n        { type: \"REASONING_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TOOL_CALL_START\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        { type: \"TOOL_CALL_START\" as const, toolCallId },\n        {\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n        },\n      );\n    }\n    case \"TOOL_CALL_ARGS\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"TOOL_CALL_ARGS\", toolCallId, delta };\n    }\n    case \"TOOL_CALL_END\": {\n      const toolCallId = getString(\"toolCallId\");\n      return toolCallId ? { type: \"TOOL_CALL_END\", toolCallId } : null;\n    }\n    case \"TOOL_CALL_CHUNK\":\n      return withOptional(\n        { type: \"TOOL_CALL_CHUNK\" as const },\n        {\n          toolCallId: getString(\"toolCallId\"),\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n          delta: getString(\"delta\"),\n        },\n      );\n    case \"TOOL_CALL_RESULT\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        {\n          type: \"TOOL_CALL_RESULT\" as const,\n          toolCallId,\n          content: getString(\"content\") ?? \"\",\n        },\n        {\n          messageId: getString(\"messageId\"),\n          role: payload.role === \"tool\" ? \"tool\" : undefined,\n        },\n      );\n    }\n    case \"STATE_SNAPSHOT\":\n      return { type: \"STATE_SNAPSHOT\", snapshot: payload.snapshot };\n    case \"STATE_DELTA\":\n      return {\n        type: \"STATE_DELTA\",\n        delta: Array.isArray(payload.delta) ? (payload.delta as any[]) : [],\n      };\n    case \"MESSAGES_SNAPSHOT\":\n      return {\n        type: \"MESSAGES_SNAPSHOT\",\n        messages: Array.isArray(payload.messages)\n          ? (payload.messages as any[])\n          : [],\n      };\n    case \"RAW\":\n      return withOptional(\n        { type: \"RAW\" as const, event: payload.event },\n        { source: getString(\"source\") },\n      );\n    case \"CUSTOM\": {\n      const name = getString(\"name\");\n      if (!name) return null;\n      return { type: \"CUSTOM\", name, value: payload.value };\n    }\n    default:\n      return withOptional(\n        { type: \"RAW\" as const, event: payload },\n        {\n          source: isString(payload.type) ? (payload.type as string) : undefined,\n        },\n      );\n  }\n};\n"],"mappings":";AAOA,MAAM,YAAY,UAAoC,OAAO,UAAU;AACvE,MAAM,oBAAoB,UACxB,SAAS,KAAK,KAAK,MAAM,SAAS;AAEpC,MAAM,gBACJ,MACA,cACG;CACH,MAAM,iBAAiB,OAAO,QAAQ,SAAS,EAAE,QAC9C,GAAG,WAAW,UAAU,KAAA,CAC3B;CACA,OAAO,eAAe,WAAW,IAC7B,OACC;EAAE,GAAG;EAAM,GAAG,OAAO,YAAY,cAAc;CAAE;AACxD;AAEA,MAAM,iBAAiB,UACrB,CAAC,CAAC,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAE9D,MAAM,kBAAkB,QAAuC;CAC7D,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO;CAChC,MAAM,KAAK,IAAI;CACf,MAAM,SAAS,IAAI;CACnB,IAAI,OAAO,OAAO,YAAY,OAAO,WAAW,UAAU,OAAO;CACjE,MAAM,YAA2B;EAAE;EAAI;CAAO;CAC9C,IAAI,OAAO,IAAI,YAAY,UAAU,UAAU,UAAU,IAAI;CAC7D,IAAI,OAAO,IAAI,eAAe,UAAU,UAAU,aAAa,IAAI;CACnE,IAAI,OAAO,IAAI,cAAc,UAAU,UAAU,YAAY,IAAI;CACjE,IAAI,cAAc,IAAI,cAAc,GAClC,UAAU,iBAAiB,IAAI;CACjC,IAAI,cAAc,IAAI,QAAQ,GAAG,UAAU,WAAW,IAAI;CAC1D,OAAO;AACT;AAEA,MAAM,2BACJ,KACA,WACuC;CACvC,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO,KAAA;CAChC,IAAI,IAAI,SAAS,WAAW,OAAO,EAAE,MAAM,UAAU;CACrD,IAAI,IAAI,SAAS,aAAa;EAC5B,IAAI,CAAC,MAAM,QAAQ,IAAI,UAAU,GAAG;GAClC,QAAQ,QACN,kEACA,GACF;GACA;EACF;EACA,MAAM,SAAS,IAAI,WAChB,KAAK,UAAU,eAAe,KAAK,CAAC,EACpC,QAAQ,UAAkC,UAAU,IAAI;EAC3D,IAAI,OAAO,WAAW,GAAG;GACvB,QAAQ,QACN,iEACA,IAAI,UACN;GACA;EACF;EACA,OAAO;GAAE,MAAM;GAAa,YAAY;EAAO;CACjD;AAEF;AAEA,MAAa,kBACX,OACA,YACqB;CACrB,IAAI,CAAC,SAAS,OAAO,UAAU,UAAU,OAAO;CAChD,MAAM,UAAU;CAChB,MAAM,YAAY,QAAQ;CAC1B,IAAI,CAAC,SAAS,SAAS,GAAG,OAAO;CAEjC,MAAM,aAAa,QACjB,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA;CAEtD,QAAQ,WAAR;EACE,KAAK,eAAe;GAClB,MAAM,QAAQ,UAAU,OAAO;GAC/B,OAAO,QAAQ;IAAE,MAAM;IAAe;GAAM,IAAI;EAClD;EACA,KAAK,gBAAgB;GACnB,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,OAAO,OAAO;GACnB,OAAO,aACL;IAAE,MAAM;IAAyB;GAAM,GACvC,EACE,SAAS,wBAAwB,QAAQ,SAAS,SAAS,MAAM,EACnE,CACF;EACF;EACA,KAAK,iBAEH,OAAO,aAAa,EAAE,MAAM,gBAAyB,GAAG,EAAE,OAD5C,UAAU,OACsC,EAAE,CAAC;EAEnE,KAAK,aACH,OAAO,aACL,EAAE,MAAM,YAAqB,GAC7B;GACE,SAAS,UAAU,SAAS;GAC5B,MAAM,UAAU,MAAM;EACxB,CACF;EAEF,KAAK,sBACH,OAAO,aACL,EAAE,MAAM,qBAA8B,GACtC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,wBAAwB;GAC3B,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,iBAAiB,KAAK,GAAG,OAAO;GACrC,OAAO,aACL;IAAE,MAAM;IAAiC;GAAM,GAC/C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF;EACA,KAAK,oBACH,OAAO,aACL,EAAE,MAAM,mBAA4B,GACpC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,sBAEH,OAAO,aACL;GAAE,MAAM;GAA+B,OAF3B,UAAU,OAAO,KAAK;EAEW,GAC7C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EAEF,KAAK,kBACH,OAAO,aACL,EAAE,MAAM,iBAA0B,GAClC,EAAE,OAAO,UAAU,OAAO,EAAE,CAC9B;EACF,KAAK,+BACH,OAAO,EAAE,MAAM,8BAA8B;EAC/C,KAAK,iCAEH,OAAO;GAAE,MAAM;GAAiC,OADlC,UAAU,OAAO,KAAK;EACkB;EAExD,KAAK,6BACH,OAAO,EAAE,MAAM,4BAA4B;EAC7C,KAAK,gBACH,OAAO,EAAE,MAAM,eAAe;EAChC,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,2BACH,OAAO,aACL,EAAE,MAAM,0BAAmC,GAC3C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,6BAEH,OAAO,aACL;GAAE,MAAM;GAAsC,OAFlC,UAAU,OAAO,KAAK;EAEkB,GACpD,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EAEF,KAAK,yBACH,OAAO,aACL,EAAE,MAAM,wBAAiC,GACzC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,iBACH,OAAO,aACL,EAAE,MAAM,gBAAyB,GACjC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,mBAAmB;GACtB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IAAE,MAAM;IAA4B;GAAW,GAC/C;IACE,cAAc,UAAU,cAAc;IACtC,iBAAiB,UAAU,iBAAiB;GAC9C,CACF;EACF;EACA,KAAK,kBAAkB;GACrB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GAExB,OAAO;IAAE,MAAM;IAAkB;IAAY,OAD/B,UAAU,OAAO,KAAK;GACe;EACrD;EACA,KAAK,iBAAiB;GACpB,MAAM,aAAa,UAAU,YAAY;GACzC,OAAO,aAAa;IAAE,MAAM;IAAiB;GAAW,IAAI;EAC9D;EACA,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC;GACE,YAAY,UAAU,YAAY;GAClC,cAAc,UAAU,cAAc;GACtC,iBAAiB,UAAU,iBAAiB;GAC5C,OAAO,UAAU,OAAO;EAC1B,CACF;EACF,KAAK,oBAAoB;GACvB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IACE,MAAM;IACN;IACA,SAAS,UAAU,SAAS,KAAK;GACnC,GACA;IACE,WAAW,UAAU,WAAW;IAChC,MAAM,QAAQ,SAAS,SAAS,SAAS,KAAA;GAC3C,CACF;EACF;EACA,KAAK,kBACH,OAAO;GAAE,MAAM;GAAkB,UAAU,QAAQ;EAAS;EAC9D,KAAK,eACH,OAAO;GACL,MAAM;GACN,OAAO,MAAM,QAAQ,QAAQ,KAAK,IAAK,QAAQ,QAAkB,CAAC;EACpE;EACF,KAAK,qBACH,OAAO;GACL,MAAM;GACN,UAAU,MAAM,QAAQ,QAAQ,QAAQ,IACnC,QAAQ,WACT,CAAC;EACP;EACF,KAAK,OACH,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO,QAAQ;EAAM,GAC7C,EAAE,QAAQ,UAAU,QAAQ,EAAE,CAChC;EACF,KAAK,UAAU;GACb,MAAM,OAAO,UAAU,MAAM;GAC7B,IAAI,CAAC,MAAM,OAAO;GAClB,OAAO;IAAE,MAAM;IAAU;IAAM,OAAO,QAAQ;GAAM;EACtD;EACA,SACE,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO;EAAQ,GACvC,EACE,QAAQ,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA,EAC9D,CACF;CACJ;AACF"}
+{"version":3,"file":"event-parser.js","names":[],"sources":["../../src/runtime/event-parser.ts"],"sourcesContent":["import type { AgUiEvent, AgUiInterrupt, AgUiRunFinishedOutcome } from \"./types\";\nimport type { Logger } from \"./logger\";\n\nexport type ParseAgUiEventOptions = {\n  logger?: Logger;\n};\n\nconst isString = (value: unknown): value is string => typeof value === \"string\";\nconst isNonEmptyString = (value: unknown): value is string =>\n  isString(value) && value.length > 0;\n\nconst withOptional = <T extends object>(\n  base: T,\n  optionals: Record<string, unknown>,\n) => {\n  const definedEntries = Object.entries(optionals).filter(\n    ([, value]) => value !== undefined,\n  );\n  return definedEntries.length === 0\n    ? base\n    : ({ ...base, ...Object.fromEntries(definedEntries) } as T);\n};\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  !!value && typeof value === \"object\" && !Array.isArray(value);\n\nconst parseInterrupt = (raw: unknown): AgUiInterrupt | null => {\n  if (!isPlainObject(raw)) return null;\n  const id = raw.id;\n  const reason = raw.reason;\n  if (typeof id !== \"string\" || typeof reason !== \"string\") return null;\n  const interrupt: AgUiInterrupt = { id, reason };\n  if (typeof raw.message === \"string\") interrupt.message = raw.message;\n  if (typeof raw.toolCallId === \"string\") interrupt.toolCallId = raw.toolCallId;\n  if (typeof raw.expiresAt === \"string\") interrupt.expiresAt = raw.expiresAt;\n  if (isPlainObject(raw.responseSchema))\n    interrupt.responseSchema = raw.responseSchema;\n  if (isPlainObject(raw.metadata)) interrupt.metadata = raw.metadata;\n  return interrupt;\n};\n\nconst parseRunFinishedOutcome = (\n  raw: unknown,\n  logger: Logger | undefined,\n): AgUiRunFinishedOutcome | undefined => {\n  if (!isPlainObject(raw)) return undefined;\n  if (raw.type === \"success\") return { type: \"success\" };\n  if (raw.type === \"interrupt\") {\n    if (!Array.isArray(raw.interrupts)) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome missing interrupts array\",\n        raw,\n      );\n      return undefined;\n    }\n    const parsed = raw.interrupts\n      .map((entry) => parseInterrupt(entry))\n      .filter((entry): entry is AgUiInterrupt => entry !== null);\n    if (parsed.length === 0) {\n      logger?.debug?.(\n        \"[agui] RUN_FINISHED interrupt outcome has no valid interrupts\",\n        raw.interrupts,\n      );\n      return undefined;\n    }\n    return { type: \"interrupt\", interrupts: parsed };\n  }\n  return undefined;\n};\n\nexport const parseAgUiEvent = (\n  event: unknown,\n  options?: ParseAgUiEventOptions,\n): AgUiEvent | null => {\n  if (!event || typeof event !== \"object\") return null;\n  const payload = event as Record<string, unknown>;\n  const typeValue = payload.type;\n  if (!isString(typeValue)) return null;\n\n  const getString = (key: string) =>\n    isString(payload[key]) ? (payload[key] as string) : undefined;\n\n  switch (typeValue) {\n    case \"RUN_STARTED\": {\n      const runId = getString(\"runId\");\n      return runId ? { type: \"RUN_STARTED\", runId } : null;\n    }\n    case \"RUN_FINISHED\": {\n      const runId = getString(\"runId\");\n      if (!runId) return null;\n      return withOptional(\n        { type: \"RUN_FINISHED\" as const, runId },\n        {\n          outcome: parseRunFinishedOutcome(payload.outcome, options?.logger),\n        },\n      );\n    }\n    case \"RUN_CANCELLED\": {\n      const runId = getString(\"runId\");\n      return withOptional({ type: \"RUN_CANCELLED\" as const }, { runId });\n    }\n    case \"RUN_ERROR\": {\n      return withOptional(\n        { type: \"RUN_ERROR\" as const },\n        {\n          message: getString(\"message\"),\n          code: getString(\"code\"),\n        },\n      );\n    }\n    case \"TEXT_MESSAGE_START\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\");\n      if (!isNonEmptyString(delta)) return null;\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"TEXT_MESSAGE_END\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CHUNK\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CHUNK\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"THINKING_START\":\n      return withOptional(\n        { type: \"THINKING_START\" as const },\n        { title: getString(\"title\") },\n      );\n    case \"THINKING_TEXT_MESSAGE_START\":\n      return { type: \"THINKING_TEXT_MESSAGE_START\" };\n    case \"THINKING_TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"THINKING_TEXT_MESSAGE_CONTENT\", delta };\n    }\n    case \"THINKING_TEXT_MESSAGE_END\":\n      return { type: \"THINKING_TEXT_MESSAGE_END\" };\n    case \"THINKING_END\":\n      return { type: \"THINKING_END\" };\n    case \"REASONING_START\":\n      return withOptional(\n        { type: \"REASONING_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_START\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"REASONING_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"REASONING_MESSAGE_END\":\n      return withOptional(\n        { type: \"REASONING_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"REASONING_END\":\n      return withOptional(\n        { type: \"REASONING_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TOOL_CALL_START\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        { type: \"TOOL_CALL_START\" as const, toolCallId },\n        {\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n        },\n      );\n    }\n    case \"TOOL_CALL_ARGS\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"TOOL_CALL_ARGS\", toolCallId, delta };\n    }\n    case \"TOOL_CALL_END\": {\n      const toolCallId = getString(\"toolCallId\");\n      return toolCallId ? { type: \"TOOL_CALL_END\", toolCallId } : null;\n    }\n    case \"TOOL_CALL_CHUNK\":\n      return withOptional(\n        { type: \"TOOL_CALL_CHUNK\" as const },\n        {\n          toolCallId: getString(\"toolCallId\"),\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n          delta: getString(\"delta\"),\n        },\n      );\n    case \"TOOL_CALL_RESULT\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        {\n          type: \"TOOL_CALL_RESULT\" as const,\n          toolCallId,\n          content: getString(\"content\") ?? \"\",\n        },\n        {\n          messageId: getString(\"messageId\"),\n          role: payload.role === \"tool\" ? \"tool\" : undefined,\n        },\n      );\n    }\n    case \"STATE_SNAPSHOT\":\n      return { type: \"STATE_SNAPSHOT\", snapshot: payload.snapshot };\n    case \"STATE_DELTA\":\n      return {\n        type: \"STATE_DELTA\",\n        delta: Array.isArray(payload.delta) ? (payload.delta as any[]) : [],\n      };\n    case \"MESSAGES_SNAPSHOT\":\n      return {\n        type: \"MESSAGES_SNAPSHOT\",\n        messages: Array.isArray(payload.messages)\n          ? (payload.messages as any[])\n          : [],\n      };\n    case \"ACTIVITY_SNAPSHOT\": {\n      const activityType = getString(\"activityType\");\n      if (!activityType || !isPlainObject(payload.content)) return null;\n      return {\n        type: \"ACTIVITY_SNAPSHOT\" as const,\n        activityType,\n        content: payload.content,\n      };\n    }\n    case \"RAW\":\n      return withOptional(\n        { type: \"RAW\" as const, event: payload.event },\n        { source: getString(\"source\") },\n      );\n    case \"CUSTOM\": {\n      const name = getString(\"name\");\n      if (!name) return null;\n      return { type: \"CUSTOM\", name, value: payload.value };\n    }\n    default:\n      return withOptional(\n        { type: \"RAW\" as const, event: payload },\n        {\n          source: isString(payload.type) ? (payload.type as string) : undefined,\n        },\n      );\n  }\n};\n"],"mappings":";AAOA,MAAM,YAAY,UAAoC,OAAO,UAAU;AACvE,MAAM,oBAAoB,UACxB,SAAS,KAAK,KAAK,MAAM,SAAS;AAEpC,MAAM,gBACJ,MACA,cACG;CACH,MAAM,iBAAiB,OAAO,QAAQ,SAAS,CAAC,CAAC,QAC9C,GAAG,WAAW,UAAU,KAAA,CAC3B;CACA,OAAO,eAAe,WAAW,IAC7B,OACC;EAAE,GAAG;EAAM,GAAG,OAAO,YAAY,cAAc;CAAE;AACxD;AAEA,MAAM,iBAAiB,UACrB,CAAC,CAAC,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAE9D,MAAM,kBAAkB,QAAuC;CAC7D,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO;CAChC,MAAM,KAAK,IAAI;CACf,MAAM,SAAS,IAAI;CACnB,IAAI,OAAO,OAAO,YAAY,OAAO,WAAW,UAAU,OAAO;CACjE,MAAM,YAA2B;EAAE;EAAI;CAAO;CAC9C,IAAI,OAAO,IAAI,YAAY,UAAU,UAAU,UAAU,IAAI;CAC7D,IAAI,OAAO,IAAI,eAAe,UAAU,UAAU,aAAa,IAAI;CACnE,IAAI,OAAO,IAAI,cAAc,UAAU,UAAU,YAAY,IAAI;CACjE,IAAI,cAAc,IAAI,cAAc,GAClC,UAAU,iBAAiB,IAAI;CACjC,IAAI,cAAc,IAAI,QAAQ,GAAG,UAAU,WAAW,IAAI;CAC1D,OAAO;AACT;AAEA,MAAM,2BACJ,KACA,WACuC;CACvC,IAAI,CAAC,cAAc,GAAG,GAAG,OAAO,KAAA;CAChC,IAAI,IAAI,SAAS,WAAW,OAAO,EAAE,MAAM,UAAU;CACrD,IAAI,IAAI,SAAS,aAAa;EAC5B,IAAI,CAAC,MAAM,QAAQ,IAAI,UAAU,GAAG;GAClC,QAAQ,QACN,kEACA,GACF;GACA;EACF;EACA,MAAM,SAAS,IAAI,WAChB,KAAK,UAAU,eAAe,KAAK,CAAC,CAAC,CACrC,QAAQ,UAAkC,UAAU,IAAI;EAC3D,IAAI,OAAO,WAAW,GAAG;GACvB,QAAQ,QACN,iEACA,IAAI,UACN;GACA;EACF;EACA,OAAO;GAAE,MAAM;GAAa,YAAY;EAAO;CACjD;AAEF;AAEA,MAAa,kBACX,OACA,YACqB;CACrB,IAAI,CAAC,SAAS,OAAO,UAAU,UAAU,OAAO;CAChD,MAAM,UAAU;CAChB,MAAM,YAAY,QAAQ;CAC1B,IAAI,CAAC,SAAS,SAAS,GAAG,OAAO;CAEjC,MAAM,aAAa,QACjB,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA;CAEtD,QAAQ,WAAR;EACE,KAAK,eAAe;GAClB,MAAM,QAAQ,UAAU,OAAO;GAC/B,OAAO,QAAQ;IAAE,MAAM;IAAe;GAAM,IAAI;EAClD;EACA,KAAK,gBAAgB;GACnB,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,OAAO,OAAO;GACnB,OAAO,aACL;IAAE,MAAM;IAAyB;GAAM,GACvC,EACE,SAAS,wBAAwB,QAAQ,SAAS,SAAS,MAAM,EACnE,CACF;EACF;EACA,KAAK,iBAEH,OAAO,aAAa,EAAE,MAAM,gBAAyB,GAAG,EAAE,OAD5C,UAAU,OACsC,EAAE,CAAC;EAEnE,KAAK,aACH,OAAO,aACL,EAAE,MAAM,YAAqB,GAC7B;GACE,SAAS,UAAU,SAAS;GAC5B,MAAM,UAAU,MAAM;EACxB,CACF;EAEF,KAAK,sBACH,OAAO,aACL,EAAE,MAAM,qBAA8B,GACtC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,wBAAwB;GAC3B,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,CAAC,iBAAiB,KAAK,GAAG,OAAO;GACrC,OAAO,aACL;IAAE,MAAM;IAAiC;GAAM,GAC/C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF;EACA,KAAK,oBACH,OAAO,aACL,EAAE,MAAM,mBAA4B,GACpC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,sBAEH,OAAO,aACL;GAAE,MAAM;GAA+B,OAF3B,UAAU,OAAO,KAAK;EAEW,GAC7C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EAEF,KAAK,kBACH,OAAO,aACL,EAAE,MAAM,iBAA0B,GAClC,EAAE,OAAO,UAAU,OAAO,EAAE,CAC9B;EACF,KAAK,+BACH,OAAO,EAAE,MAAM,8BAA8B;EAC/C,KAAK,iCAEH,OAAO;GAAE,MAAM;GAAiC,OADlC,UAAU,OAAO,KAAK;EACkB;EAExD,KAAK,6BACH,OAAO,EAAE,MAAM,4BAA4B;EAC7C,KAAK,gBACH,OAAO,EAAE,MAAM,eAAe;EAChC,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,2BACH,OAAO,aACL,EAAE,MAAM,0BAAmC,GAC3C,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,6BAEH,OAAO,aACL;GAAE,MAAM;GAAsC,OAFlC,UAAU,OAAO,KAAK;EAEkB,GACpD,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EAEF,KAAK,yBACH,OAAO,aACL,EAAE,MAAM,wBAAiC,GACzC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,iBACH,OAAO,aACL,EAAE,MAAM,gBAAyB,GACjC,EAAE,WAAW,UAAU,WAAW,EAAE,CACtC;EACF,KAAK,mBAAmB;GACtB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IAAE,MAAM;IAA4B;GAAW,GAC/C;IACE,cAAc,UAAU,cAAc;IACtC,iBAAiB,UAAU,iBAAiB;GAC9C,CACF;EACF;EACA,KAAK,kBAAkB;GACrB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GAExB,OAAO;IAAE,MAAM;IAAkB;IAAY,OAD/B,UAAU,OAAO,KAAK;GACe;EACrD;EACA,KAAK,iBAAiB;GACpB,MAAM,aAAa,UAAU,YAAY;GACzC,OAAO,aAAa;IAAE,MAAM;IAAiB;GAAW,IAAI;EAC9D;EACA,KAAK,mBACH,OAAO,aACL,EAAE,MAAM,kBAA2B,GACnC;GACE,YAAY,UAAU,YAAY;GAClC,cAAc,UAAU,cAAc;GACtC,iBAAiB,UAAU,iBAAiB;GAC5C,OAAO,UAAU,OAAO;EAC1B,CACF;EACF,KAAK,oBAAoB;GACvB,MAAM,aAAa,UAAU,YAAY;GACzC,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,aACL;IACE,MAAM;IACN;IACA,SAAS,UAAU,SAAS,KAAK;GACnC,GACA;IACE,WAAW,UAAU,WAAW;IAChC,MAAM,QAAQ,SAAS,SAAS,SAAS,KAAA;GAC3C,CACF;EACF;EACA,KAAK,kBACH,OAAO;GAAE,MAAM;GAAkB,UAAU,QAAQ;EAAS;EAC9D,KAAK,eACH,OAAO;GACL,MAAM;GACN,OAAO,MAAM,QAAQ,QAAQ,KAAK,IAAK,QAAQ,QAAkB,CAAC;EACpE;EACF,KAAK,qBACH,OAAO;GACL,MAAM;GACN,UAAU,MAAM,QAAQ,QAAQ,QAAQ,IACnC,QAAQ,WACT,CAAC;EACP;EACF,KAAK,qBAAqB;GACxB,MAAM,eAAe,UAAU,cAAc;GAC7C,IAAI,CAAC,gBAAgB,CAAC,cAAc,QAAQ,OAAO,GAAG,OAAO;GAC7D,OAAO;IACL,MAAM;IACN;IACA,SAAS,QAAQ;GACnB;EACF;EACA,KAAK,OACH,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO,QAAQ;EAAM,GAC7C,EAAE,QAAQ,UAAU,QAAQ,EAAE,CAChC;EACF,KAAK,UAAU;GACb,MAAM,OAAO,UAAU,MAAM;GAC7B,IAAI,CAAC,MAAM,OAAO;GAClB,OAAO;IAAE,MAAM;IAAU;IAAM,OAAO,QAAQ;GAAM;EACtD;EACA,SACE,OAAO,aACL;GAAE,MAAM;GAAgB,OAAO;EAAQ,GACvC,EACE,QAAQ,SAAS,QAAQ,IAAI,IAAK,QAAQ,OAAkB,KAAA,EAC9D,CACF;CACJ;AACF"}

package/dist/runtime/types.d.ts

@@ -138,6 +138,10 @@ type AgUiEvent = {
   toolCallId: string;
   content: string;
   role?: "tool";
+} | {
+  type: "ACTIVITY_SNAPSHOT";
+  activityType: string;
+  content: Record<string, unknown>;
 } | {
   type: "RAW";
   event: any;

package/dist/runtime/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/runtime/types.ts"],"mappings":";;;;;;;;AAiBA;KAAY,wBAAA;EACV,QAAA;EACA,mBAAA,UAA6B,OAAA;EAC7B,gBAAA,KACM,QAAA,aACE,OAAA;IACE,QAAA,WAAmB,aAAA;IACnB,KAAA,GAAQ,iBAAA;EAAA;IAER,QAAA,WAAmB,aAAA;IAAiB,KAAA,GAAQ,iBAAA;EAAA;AAAA;AAAA,KAI5C,sBAAA;EACV,WAAA,GAAc,iBAAA;EACd,MAAA,GAAS,sBAAA;EACT,SAAA,GAAY,gBAAA;EACZ,KAAA,GAAQ,oBAAA;EACR,QAAA,GAAW,eAAA;EACX,OAAA,GAAU,oBAAA;EAZA;;;EAgBV,UAAA,GAAa,wBAAA;AAAA;AAAA,KAGH,qBAAA,GAAwB,0BAAA;EAClC,KAAA,EAAO,SAAA;EACP,MAAA,GAAS,OAAA,CAAQ,MAAA;EACjB,YAAA;EACA,OAAA,IAAW,CAAA,EAAG,KAAA;EACd,QAAA;EACA,QAAA,GAAW,sBAAA;AAAA;AAAA,KAGD,mBAAA;AAAA,KAMA,aAAA;EACV,EAAA;EACA,MAAA,EAAQ,mBAAA;EACR,OAAA;EACA,UAAA;EACA,cAAA,GAAiB,MAAA;EACjB,SAAA;EACA,QAAA,GAAW,MAAA;AAAA;AAAA,KAGD,eAAA;EACV,WAAA;EACA,MAAA;EACA,OAAA;AAAA;AAAA,KAGU,sBAAA;EACN,IAAA;AAAA;EACA,IAAA;EAAmB,UAAA,EAAY,aAAa;AAAA;AAAA,KAEtC,SAAA;EACN,IAAA;EAAqB,KAAA;AAAA;EAErB,IAAA;EACA,KAAA;EACA,OAAA,GAAU,sBAAsB;AAAA;EAEhC,IAAA;EAAuB,KAAA;AAAA;EACvB,IAAA;EAAmB,OAAA;EAAkB,IAAA;AAAA;EACrC,IAAA;EAA4B,SAAA;AAAA;EAC5B,IAAA;EAA8B,SAAA;EAAoB,KAAA;AAAA;EAClD,IAAA;EAA0B,SAAA;AAAA;EAC1B,IAAA;EAA4B,KAAA;AAAA;EAC5B,IAAA;EAAwB,KAAA;AAAA;EACxB,IAAA;AAAA;EACA,IAAA;EAAuC,KAAA;AAAA;EACvC,IAAA;AAAA;EACA,IAAA;AAAA;EACA,IAAA;EAAyB,SAAA;AAAA;EACzB,IAAA;EAAiC,SAAA;AAAA;EACjC,IAAA;EAAmC,SAAA;EAAoB,KAAA;AAAA;EACvD,IAAA;EAA+B,SAAA;AAAA;EAC/B,IAAA;EAAuB,SAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;AAAA;EAEA,IAAA;EAAwB,UAAA;EAAoB,KAAA;AAAA;EAC5C,IAAA;EAAuB,UAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;EACA,KAAA;AAAA;EAGA,IAAA;EACA,SAAA;EACA,UAAA;EACA,OAAA;EACA,IAAA;AAAA;EAEA,IAAA;EAAa,KAAA;EAAY,MAAA;AAAA;EACzB,IAAA;EAAgB,IAAA;EAAc,KAAA;AAAA;EAC9B,IAAA;EAAwB,QAAA;AAAA;EACxB,IAAA;EAAqB,KAAA;AAAA;EACrB,IAAA;EAA2B,QAAA;AAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/runtime/types.ts"],"mappings":";;;;;;;;AAiBA;KAAY,wBAAA;EACV,QAAA;EACA,mBAAA,UAA6B,OAAA;EAC7B,gBAAA,KACM,QAAA,aACE,OAAA;IACE,QAAA,WAAmB,aAAA;IACnB,KAAA,GAAQ,iBAAA;EAAA;IAER,QAAA,WAAmB,aAAA;IAAiB,KAAA,GAAQ,iBAAA;EAAA;AAAA;AAAA,KAI5C,sBAAA;EACV,WAAA,GAAc,iBAAA;EACd,MAAA,GAAS,sBAAA;EACT,SAAA,GAAY,gBAAA;EACZ,KAAA,GAAQ,oBAAA;EACR,QAAA,GAAW,eAAA;EACX,OAAA,GAAU,oBAAA;EAZA;;;EAgBV,UAAA,GAAa,wBAAA;AAAA;AAAA,KAGH,qBAAA,GAAwB,0BAAA;EAClC,KAAA,EAAO,SAAA;EACP,MAAA,GAAS,OAAA,CAAQ,MAAA;EACjB,YAAA;EACA,OAAA,IAAW,CAAA,EAAG,KAAA;EACd,QAAA;EACA,QAAA,GAAW,sBAAA;AAAA;AAAA,KAGD,mBAAA;AAAA,KAMA,aAAA;EACV,EAAA;EACA,MAAA,EAAQ,mBAAA;EACR,OAAA;EACA,UAAA;EACA,cAAA,GAAiB,MAAA;EACjB,SAAA;EACA,QAAA,GAAW,MAAA;AAAA;AAAA,KAGD,eAAA;EACV,WAAA;EACA,MAAA;EACA,OAAA;AAAA;AAAA,KAGU,sBAAA;EACN,IAAA;AAAA;EACA,IAAA;EAAmB,UAAA,EAAY,aAAa;AAAA;AAAA,KAEtC,SAAA;EACN,IAAA;EAAqB,KAAA;AAAA;EAErB,IAAA;EACA,KAAA;EACA,OAAA,GAAU,sBAAA;AAAA;EAEV,IAAA;EAAuB,KAAA;AAAA;EACvB,IAAA;EAAmB,OAAA;EAAkB,IAAA;AAAA;EACrC,IAAA;EAA4B,SAAA;AAAA;EAC5B,IAAA;EAA8B,SAAA;EAAoB,KAAA;AAAA;EAClD,IAAA;EAA0B,SAAA;AAAA;EAC1B,IAAA;EAA4B,KAAA;AAAA;EAC5B,IAAA;EAAwB,KAAA;AAAA;EACxB,IAAA;AAAA;EACA,IAAA;EAAuC,KAAA;AAAA;EACvC,IAAA;AAAA;EACA,IAAA;AAAA;EACA,IAAA;EAAyB,SAAA;AAAA;EACzB,IAAA;EAAiC,SAAA;AAAA;EACjC,IAAA;EAAmC,SAAA;EAAoB,KAAA;AAAA;EACvD,IAAA;EAA+B,SAAA;AAAA;EAC/B,IAAA;EAAuB,SAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;AAAA;EAEA,IAAA;EAAwB,UAAA;EAAoB,KAAA;AAAA;EAC5C,IAAA;EAAuB,UAAA;AAAA;EAEvB,IAAA;EACA,UAAA;EACA,YAAA;EACA,eAAA;EACA,KAAA;AAAA;EAGA,IAAA;EACA,SAAA;EACA,UAAA;EACA,OAAA;EACA,IAAA;AAAA;EAGA,IAAA;EACA,YAAA;EACA,OAAA,EAAS,MAAM;AAAA;EAEf,IAAA;EAAa,KAAA;EAAY,MAAA;AAAA;EACzB,IAAA;EAAgB,IAAA;EAAc,KAAA;AAAA;EAC9B,IAAA;EAAwB,QAAA;AAAA;EACxB,IAAA;EAAqB,KAAA;AAAA;EACrB,IAAA;EAA2B,QAAA;AAAA"}

package/dist/useAgUiRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAgUiRuntime.js","names":[],"sources":["../src/useAgUiRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { makeLogger } from \"./runtime/logger\";\nimport type {\n  AgUiInterrupt,\n  AgUiResumeEntry,\n  UseAgUiRuntimeOptions,\n} from \"./runtime/types\";\nimport { AgUiThreadRuntimeCore } from \"./runtime/AgUiThreadRuntimeCore\";\n\nexport type AgUiAssistantRuntime = AssistantRuntime & {\n  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];\n  unstable_submitInterruptResponses: (\n    responses: readonly AgUiResumeEntry[],\n  ) => Promise<void>;\n};\n\nexport function useAgUiRuntime(\n  options: UseAgUiRuntimeOptions,\n): AgUiAssistantRuntime {\n  const logger = useMemo(() => makeLogger(options.logger), [options.logger]);\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const coreRef = useRef<AgUiThreadRuntimeCore | null>(null);\n  const runtimeAdapters = useRuntimeAdapters();\n\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  if (!coreRef.current) {\n    coreRef.current = new AgUiThreadRuntimeCore({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n      notifyUpdate,\n    });\n  }\n\n  const core = coreRef.current;\n  core.updateOptions({\n    agent: options.agent,\n    logger,\n    showThinking: options.showThinking ?? true,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(historyAdapter && { history: historyAdapter }),\n  });\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            await onSwitchToNewThread();\n            core.applyExternalMessages([]);\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const result = await onSwitchToThread(threadId);\n            core.applyExternalMessages(result.messages);\n            if (result.state) {\n              core.loadExternalState(result.state);\n            }\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(\n    () => {\n      void _version; // rerender on version change\n\n      return {\n        ...shared,\n        isLoading: core.isLoading,\n        messages: core.getMessages(),\n        state: core.getState(),\n        isRunning: core.isRunning() || hasExecutingTools,\n        unstable_enableToolInvocations: true,\n        setToolStatuses,\n        onNew: (message: AppendMessage) => core.append(message),\n        onEdit: (message: AppendMessage) => core.edit(message),\n        onReload: (parentId: string | null, config: { runConfig?: any }) =>\n          core.reload(parentId, config),\n        onCancel: async () => {\n          core.cancel();\n        },\n        onAddToolResult: (options) => core.addToolResult(options),\n        onResume: (config) => core.resume(config),\n        setMessages: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onImport: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onLoadExternalState: (state: ReadonlyJSONValue) =>\n          core.loadExternalState(state),\n        adapters: adapterAdapters,\n      } satisfies ExternalStoreAdapter<ThreadMessage>;\n    },\n    // _version is intentionally included to trigger re-computation when core state changes via notifyUpdate\n    // toolInvocations intentionally excluded: abort/resume use refs internally and work with stale captures\n    [adapterAdapters, core, _version, hasExecutingTools, shared],\n  );\n\n  const baseRuntime = useExternalStoreRuntime(store);\n\n  const runtime = useMemo<AgUiAssistantRuntime>(() => {\n    const wrapper = Object.create(baseRuntime) as AgUiAssistantRuntime;\n    wrapper.unstable_getPendingInterrupts = () =>\n      core.getPendingInterrupts()?.interrupts ?? [];\n    wrapper.unstable_submitInterruptResponses = (responses) =>\n      core.submitInterruptResponses(responses);\n    return wrapper;\n  }, [baseRuntime, core]);\n\n  useEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;AA+BA,SAAgB,eACd,SACsB;CACtB,MAAM,SAAS,cAAc,WAAW,QAAQ,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC;CACzE,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,UAAU,OAAqC,IAAI;CACzD,MAAM,kBAAkB,mBAAmB;CAE3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,sBAAsB;EAC1C,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAChD;CACF,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,KAAK,cAAc;EACjB,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD,CAAC;CAED,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CAEJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CAEA,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,oBAAoB;IAC1B,KAAK,sBAAsB,CAAC,CAAC;GAC/B,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,KAAK,sBAAsB,OAAO,QAAQ;IAC1C,IAAI,OAAO,OACT,KAAK,kBAAkB,OAAO,KAAK;GAEvC,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAE5B,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAEA,MAAM,SAAS,8BAA8B,OAAO;CAoCpD,MAAM,cAAc,wBAnCN,cACN;EAGJ,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,UAAU,KAAK,YAAY;GAC3B,OAAO,KAAK,SAAS;GACrB,WAAW,KAAK,UAAU,KAAK;GAC/B,gCAAgC;GAChC;GACA,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,UAAyB,WAClC,KAAK,OAAO,UAAU,MAAM;GAC9B,UAAU,YAAY;IACpB,KAAK,OAAO;GACd;GACA,kBAAkB,YAAY,KAAK,cAAc,OAAO;GACxD,WAAW,WAAW,KAAK,OAAO,MAAM;GACxC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,sBAAsB,UACpB,KAAK,kBAAkB,KAAK;GAC9B,UAAU;EACZ;CACF,GAGA;EAAC;EAAiB;EAAM;EAAU;EAAmB;CAAM,CAGb,CAAC;CAEjD,MAAM,UAAU,cAAoC;EAClD,MAAM,UAAU,OAAO,OAAO,WAAW;EACzC,QAAQ,sCACN,KAAK,qBAAqB,GAAG,cAAc,CAAC;EAC9C,QAAQ,qCAAqC,cAC3C,KAAK,yBAAyB,SAAS;EACzC,OAAO;CACT,GAAG,CAAC,aAAa,IAAI,CAAC;CAEtB,gBAAgB;EACd,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}
+{"version":3,"file":"useAgUiRuntime.js","names":[],"sources":["../src/useAgUiRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { makeLogger } from \"./runtime/logger\";\nimport type {\n  AgUiInterrupt,\n  AgUiResumeEntry,\n  UseAgUiRuntimeOptions,\n} from \"./runtime/types\";\nimport { AgUiThreadRuntimeCore } from \"./runtime/AgUiThreadRuntimeCore\";\n\nexport type AgUiAssistantRuntime = AssistantRuntime & {\n  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];\n  unstable_submitInterruptResponses: (\n    responses: readonly AgUiResumeEntry[],\n  ) => Promise<void>;\n};\n\nexport function useAgUiRuntime(\n  options: UseAgUiRuntimeOptions,\n): AgUiAssistantRuntime {\n  const logger = useMemo(() => makeLogger(options.logger), [options.logger]);\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const coreRef = useRef<AgUiThreadRuntimeCore | null>(null);\n  const runtimeAdapters = useRuntimeAdapters();\n\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  if (!coreRef.current) {\n    coreRef.current = new AgUiThreadRuntimeCore({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n      notifyUpdate,\n    });\n  }\n\n  const core = coreRef.current;\n  core.updateOptions({\n    agent: options.agent,\n    logger,\n    showThinking: options.showThinking ?? true,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(historyAdapter && { history: historyAdapter }),\n  });\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            await onSwitchToNewThread();\n            core.applyExternalMessages([]);\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const result = await onSwitchToThread(threadId);\n            core.applyExternalMessages(result.messages);\n            if (result.state) {\n              core.loadExternalState(result.state);\n            }\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(\n    () => {\n      void _version; // rerender on version change\n\n      return {\n        ...shared,\n        isLoading: core.isLoading,\n        messages: core.getMessages(),\n        state: core.getState(),\n        isRunning: core.isRunning() || hasExecutingTools,\n        unstable_enableToolInvocations: true,\n        setToolStatuses,\n        onNew: (message: AppendMessage) => core.append(message),\n        onEdit: (message: AppendMessage) => core.edit(message),\n        onReload: (parentId: string | null, config: { runConfig?: any }) =>\n          core.reload(parentId, config),\n        onCancel: async () => {\n          core.cancel();\n        },\n        onAddToolResult: (options) => core.addToolResult(options),\n        onResume: (config) => core.resume(config),\n        setMessages: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onImport: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onLoadExternalState: (state: ReadonlyJSONValue) =>\n          core.loadExternalState(state),\n        adapters: adapterAdapters,\n      } satisfies ExternalStoreAdapter<ThreadMessage>;\n    },\n    // _version is intentionally included to trigger re-computation when core state changes via notifyUpdate\n    // toolInvocations intentionally excluded: abort/resume use refs internally and work with stale captures\n    [adapterAdapters, core, _version, hasExecutingTools, shared],\n  );\n\n  const baseRuntime = useExternalStoreRuntime(store);\n\n  const runtime = useMemo<AgUiAssistantRuntime>(() => {\n    const wrapper = Object.create(baseRuntime) as AgUiAssistantRuntime;\n    wrapper.unstable_getPendingInterrupts = () =>\n      core.getPendingInterrupts()?.interrupts ?? [];\n    wrapper.unstable_submitInterruptResponses = (responses) =>\n      core.submitInterruptResponses(responses);\n    return wrapper;\n  }, [baseRuntime, core]);\n\n  useEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;AA+BA,SAAgB,eACd,SACsB;CACtB,MAAM,SAAS,cAAc,WAAW,QAAQ,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC;CACzE,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,UAAU,OAAqC,IAAI;CACzD,MAAM,kBAAkB,mBAAmB;CAE3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,sBAAsB;EAC1C,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAChD;CACF,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,KAAK,cAAc;EACjB,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD,CAAC;CAED,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CAEJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CAEA,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,oBAAoB;IAC1B,KAAK,sBAAsB,CAAC,CAAC;GAC/B,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,SAAS,MAAM,iBAAiB,QAAQ;IAC9C,KAAK,sBAAsB,OAAO,QAAQ;IAC1C,IAAI,OAAO,OACT,KAAK,kBAAkB,OAAO,KAAK;GAEvC,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAE5B,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAEA,MAAM,SAAS,8BAA8B,OAAO;CAoCpD,MAAM,cAAc,wBAnCN,cACN;EAGJ,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,UAAU,KAAK,YAAY;GAC3B,OAAO,KAAK,SAAS;GACrB,WAAW,KAAK,UAAU,KAAK;GAC/B,gCAAgC;GAChC;GACA,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,UAAyB,WAClC,KAAK,OAAO,UAAU,MAAM;GAC9B,UAAU,YAAY;IACpB,KAAK,OAAO;GACd;GACA,kBAAkB,YAAY,KAAK,cAAc,OAAO;GACxD,WAAW,WAAW,KAAK,OAAO,MAAM;GACxC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,sBAAsB,UACpB,KAAK,kBAAkB,KAAK;GAC9B,UAAU;EACZ;CACF,GAGA;EAAC;EAAiB;EAAM;EAAU;EAAmB;CAAM,CAGb,CAAC;CAEjD,MAAM,UAAU,cAAoC;EAClD,MAAM,UAAU,OAAO,OAAO,WAAW;EACzC,QAAQ,sCACN,KAAK,qBAAqB,CAAC,EAAE,cAAc,CAAC;EAC9C,QAAQ,qCAAqC,cAC3C,KAAK,yBAAyB,SAAS;EACzC,OAAO;CACT,GAAG,CAAC,aAAa,IAAI,CAAC;CAEtB,gBAAgB;EACd,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ag-ui",
-  "version": "0.0.35",
+  "version": "0.0.36",
   "description": "AG-UI protocol adapter for assistant-ui",
   "keywords": [
     "ag-ui",
@@ -29,9 +29,9 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ag-ui/client": "^0.0.54",
-    "@assistant-ui/core": "^0.2.10",
-    "assistant-stream": "^0.3.20",
+    "@ag-ui/client": "^0.0.55",
+    "@assistant-ui/core": "^0.2.11",
+    "assistant-stream": "^0.3.21",
     "fast-json-patch": "^3.1.1"
   },
   "peerDependencies": {
@@ -44,11 +44,11 @@
     }
   },
   "devDependencies": {
-    "@types/react": "^19.2.15",
-    "react": "^19.2.6",
-    "vitest": "^4.1.7",
+    "@types/react": "^19.2.17",
+    "react": "^19.2.7",
+    "vitest": "^4.1.8",
     "zod": "^4.4.3",
-    "@assistant-ui/x-buildutils": "0.0.11"
+    "@assistant-ui/x-buildutils": "0.0.12"
   },
   "publishConfig": {
     "access": "public",

package/src/index.ts

@@ -1,5 +1,7 @@
 export { useAgUiRuntime } from "./useAgUiRuntime";
 export type { AgUiAssistantRuntime } from "./useAgUiRuntime";
+export { fromAgUiMessages } from "./runtime/adapter/conversions";
+export type { FromAgUiMessagesOptions } from "./runtime/adapter/conversions";
 export type {
   AgUiInterrupt,
   AgUiInterruptReason,

package/src/runtime/AgUiThreadRuntimeCore.ts

@@ -911,16 +911,14 @@ export class AgUiThreadRuntimeCore {
 
   private importMessagesSnapshot(rawMessages: readonly unknown[]) {
     try {
-      const normalized = fromAgUiMessages(rawMessages);
+      const normalized = fromAgUiMessages(rawMessages, {
+        showThinking: this.showThinking,
+      });
       const converted: ThreadMessage[] = [];
       for (const message of normalized) {
         try {
           converted.push(
-            fromThreadMessageLike(
-              message as any,
-              generateId(),
-              FALLBACK_USER_STATUS,
-            ),
+            fromThreadMessageLike(message, generateId(), FALLBACK_USER_STATUS),
           );
         } catch (error) {
           this.logger.error?.(