@assistant-ui/react-a2a 0.2.26 → 0.2.28

package/dist/A2AThreadRuntimeCore.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"A2AThreadRuntimeCore.d.ts","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"mappings":";;;;KAgCY;EACV,QAAQ;EACR;EACA,gBAAgB;EAChB,YAAY,OAAO;EACnB;EACA,uBAAuB,UAAU;EACjC,UAAU;EACV;;cAQW;UACH;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UACA;UAGA;UACA;UACA;mBAGS;mBACA;UACT;UACA;EAEI,YAAA,SAAS;EAWrB,cAAc,SAAS,KAAK;EAU5B,cAAc,SAAS;EAIvB;EASA,cAAc;EAId,wBAAwB;EAIxB,wBAAwB;UAKhB;UAQA;UAUA;UAIA;UAQA;UAKA;UAKA;UAKA;EAYR,WAAW;EAIX,yBAAyB;EAIzB,gBAAgB;EAIhB;MAII;EAIJ,mBAAmB;EA+Bb,OAAO,SAAS,gBAAgB;EAuBhC,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAW,YAAY;MACtB;EAaG,UAAU;UAiBR;UAgBA;EAWR,sBAAsB,mBAAmB;UAiCjC;UA6DM;UAwEA;UA4BA;UA4BN;UAiBA;UA6BA;UAqCA;UAQA;UAuBA;UAmCA;UAsBA;UAUA;UAaA;UAQA;UAKA;UASA;UAIA;UAQA;UAeA"}
+{"version":3,"file":"A2AThreadRuntimeCore.d.ts","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"mappings":";;;;KAgCY;EACV,QAAQ;EACR;EACA,gBAAgB;EAChB,YAAY,OAAO;EACnB;EACA,uBAAuB,UAAU;EACjC,UAAU;EACV;;cAsCW;UACH;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UACA;UAGA;UACA;UACA;mBAGS;mBACA;UACT;UACA;EAEI,YAAA,SAAS;EAWrB,cAAc,SAAS,KAAK;EAU5B,cAAc,SAAS;EAIvB;EASA,cAAc;EAId,wBAAwB;EAIxB,wBAAwB;UAKhB;UAQA;UAUA;UAIA;UAQA;UAKA;UAKA;UAKA;EAYR,WAAW;EAIX,yBAAyB;EAIzB,gBAAgB;EAIhB;MAII;EAIJ,mBAAmB;EAiCb,OAAO,SAAS,gBAAgB;EAuBhC,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAW,YAAY;MACtB;EAaG,UAAU;UAiBR;UAgBA;EAWR,sBAAsB,mBAAmB;UAiCjC;UA6DM;UAwEA;UA4BA;UA4BN;UAiBA;UA6BA;UAyCA;UAQA;UAuBA;UAmCA;UAsBA;UAUA;UAaA;UAQA;UAKA;UAQA;UAIA;UAQA;UAeA"}

package/dist/A2AThreadRuntimeCore.js

@@ -7,6 +7,20 @@ const FALLBACK_USER_STATUS = {
 	type: "complete",
 	reason: "unknown"
 };
+const reportCallbackError = (name, error) => {
+	console.error(`[react-a2a] ${name} callback threw an error`, error);
+};
+const invokeRuntimeCallback = (name, callback, ...args) => {
+	if (!callback) return;
+	try {
+		const result = callback(...args);
+		if (result !== null && (typeof result === "object" || typeof result === "function") && "then" in result && typeof result.then === "function") Promise.resolve(result).catch((error) => {
+			reportCallbackError(name, error);
+		});
+	} catch (error) {
+		reportCallbackError(name, error);
+	}
+};
 var A2AThreadRuntimeCore = class {
 	client;
 	contextId;
@@ -133,7 +147,7 @@ var A2AThreadRuntimeCore = class {
 			if (agentCard) this.agentCardValue = agentCard;
 			if (repo) this.applyExternalMessageRepository(repo);
 		}).catch((error) => {
-			this.onError?.(error instanceof Error ? error : new Error(String(error)));
+			invokeRuntimeCallback("onError", this.onError, error instanceof Error ? error : new Error(String(error)));
 		}).finally(() => {
 			this._isLoading = false;
 			this.notifyUpdate();
@@ -284,7 +298,7 @@ var A2AThreadRuntimeCore = class {
 				reason: "cancelled"
 			});
 			this.finishRun(abortController);
-			this.onCancel?.();
+			invokeRuntimeCallback("onCancel", this.onCancel);
 		}, { once: true });
 		this.setRunning(true);
 		const supportsStreaming = this.agentCardValue?.capabilities?.streaming !== false;
@@ -298,7 +312,7 @@ var A2AThreadRuntimeCore = class {
 					type: "incomplete",
 					reason: "error"
 				});
-				this.onError?.(err);
+				invokeRuntimeCallback("onError", this.onError, err);
 				this.pendingError = this.pendingError ?? err;
 			}
 		} finally {
@@ -346,9 +360,7 @@ var A2AThreadRuntimeCore = class {
 			case "message":
 				this.handleMessage(assistantId, event.message);
 				break;
-			case "task":
-				this.handleTaskSnapshot(assistantId, event.task);
-				break;
+			case "task": this.handleTaskSnapshot(assistantId, event.task);
 		}
 	}
 	handleStatusUpdate(assistantId, event) {
@@ -396,7 +408,7 @@ var A2AThreadRuntimeCore = class {
 			updated = artifact;
 			this.currentArtifacts = [...this.currentArtifacts, updated];
 		}
-		if (lastChunk) this.onArtifactComplete?.(updated);
+		if (lastChunk) invokeRuntimeCallback("onArtifactComplete", this.onArtifactComplete, updated);
 		this.notifyUpdate();
 	}
 	handleMessage(assistantId, message) {
@@ -484,7 +496,8 @@ var A2AThreadRuntimeCore = class {
 		this.notifyUpdate();
 	}
 	finishRun(controller) {
-		if (this.abortController === controller) this.abortController = null;
+		if (this.abortController !== controller) return;
+		this.abortController = null;
 		this.setRunning(false);
 	}
 	recordHistoryEntry(parentId, message) {

package/dist/A2AThreadRuntimeCore.js.map

@@ -1 +1 @@
-{"version":3,"file":"A2AThreadRuntimeCore.js","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"sourcesContent":["\"use client\";\n\nimport { generateId, fromThreadMessageLike } from \"@assistant-ui/core\";\nimport type {\n  AppendMessage,\n  AssistantRuntime,\n  ExportedMessageRepository,\n  MessageStatus,\n  ThreadAssistantMessage,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport type { A2AClient } from \"./A2AClient\";\nimport type {\n  A2AArtifact,\n  A2AAgentCard,\n  A2AMessage,\n  A2APart,\n  A2ASendMessageConfiguration,\n  A2AStreamEvent,\n  A2ATask,\n  A2ATaskArtifactUpdateEvent,\n  A2ATaskStatusUpdateEvent,\n} from \"./types\";\nimport {\n  a2aMessageToContent,\n  contentPartsToA2AParts,\n  isTerminalTaskState,\n  taskStateToMessageStatus,\n} from \"./conversions\";\n\nexport type A2AThreadRuntimeCoreOptions = {\n  client: A2AClient;\n  contextId?: string | undefined;\n  configuration?: A2ASendMessageConfiguration | undefined;\n  onError?: ((error: Error) => void) | undefined;\n  onCancel?: (() => void) | undefined;\n  onArtifactComplete?: ((artifact: A2AArtifact) => void) | undefined;\n  history?: ThreadHistoryAdapter | undefined;\n  notifyUpdate: () => void;\n};\n\nconst FALLBACK_USER_STATUS = {\n  type: \"complete\",\n  reason: \"unknown\",\n} as const;\n\nexport class A2AThreadRuntimeCore {\n  private client: A2AClient;\n  private contextId: string | undefined;\n  private configuration: A2ASendMessageConfiguration | undefined;\n  private onError: ((error: Error) => void) | undefined;\n  private onCancel: (() => void) | undefined;\n  private onArtifactComplete: ((artifact: A2AArtifact) => void) | undefined;\n  private history: ThreadHistoryAdapter | undefined;\n  private readonly notifyUpdate: () => void;\n\n  private runtime: AssistantRuntime | undefined;\n  private readonly repository = new MessageRepository();\n  private exportedRepository: ExportedMessageRepository | undefined;\n  private isRunningFlag = false;\n  private abortController: AbortController | null = null;\n  private pendingError: Error | null = null;\n\n  // A2A-specific state\n  private currentTask: A2ATask | undefined;\n  private currentArtifacts: A2AArtifact[] = [];\n  private agentCardValue: A2AAgentCard | undefined;\n\n  // History tracking\n  private readonly assistantHistoryParents = new Map<string, string | null>();\n  private readonly recordedHistoryIds = new Set<string>();\n  private _isLoading = false;\n  private _loadPromise: Promise<void> | undefined;\n\n  constructor(options: A2AThreadRuntimeCoreOptions) {\n    this.client = options.client;\n    this.contextId = options.contextId;\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    this.history = options.history;\n    this.notifyUpdate = options.notifyUpdate;\n  }\n\n  updateOptions(options: Omit<A2AThreadRuntimeCoreOptions, \"notifyUpdate\">) {\n    this.client = options.client;\n    this.contextId = options.contextId;\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    this.history = options.history;\n  }\n\n  attachRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  detachRuntime() {\n    this.runtime = undefined;\n    // Abort in-flight requests on unmount\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n  }\n\n  getRuntime(): AssistantRuntime | undefined {\n    return this.runtime;\n  }\n\n  getMessages(): readonly ThreadMessage[] {\n    return this.repository.getMessages();\n  }\n\n  getMessageRepository(): ExportedMessageRepository {\n    this.exportedRepository ??= this.repository.export();\n    return this.exportedRepository;\n  }\n\n  private tryGetMessage(messageId: string) {\n    try {\n      return this.repository.getMessage(messageId);\n    } catch {\n      return undefined;\n    }\n  }\n\n  private tryGetMessages(\n    messageId: string,\n  ): readonly ThreadMessage[] | undefined {\n    try {\n      return this.repository.getMessages(messageId);\n    } catch {\n      return undefined;\n    }\n  }\n\n  private hasMessage(messageId: string): boolean {\n    return this.tryGetMessage(messageId) !== undefined;\n  }\n\n  private addOrUpdateMessage(\n    parentId: string | null,\n    message: ThreadMessage,\n  ): void {\n    this.repository.addOrUpdateMessage(parentId, message);\n    this.exportedRepository = undefined;\n  }\n\n  private switchToBranch(messageId: string): void {\n    this.repository.switchToBranch(messageId);\n    this.exportedRepository = undefined;\n  }\n\n  private resetRepositoryHead(messageId: string | null): void {\n    this.repository.resetHead(messageId);\n    this.exportedRepository = undefined;\n  }\n\n  private clearRepository(): void {\n    this.repository.clear();\n    this.exportedRepository = undefined;\n  }\n\n  private updateMessage(\n    messageId: string,\n    updater: (message: ThreadMessage) => ThreadMessage,\n  ): boolean {\n    const item = this.tryGetMessage(messageId);\n    if (!item) return false;\n    const message = updater(item.message);\n    if (message === item.message) return false;\n    this.addOrUpdateMessage(item.parentId, message);\n    return true;\n  }\n\n  getTask(): A2ATask | undefined {\n    return this.currentTask;\n  }\n\n  getArtifacts(): readonly A2AArtifact[] {\n    return this.currentArtifacts;\n  }\n\n  getAgentCard(): A2AAgentCard | undefined {\n    return this.agentCardValue;\n  }\n\n  isRunning(): boolean {\n    return this.isRunningFlag;\n  }\n\n  get isLoading(): boolean {\n    return this._isLoading;\n  }\n\n  __internal_load(): Promise<void> {\n    if (this._loadPromise) return this._loadPromise;\n\n    this._isLoading = true;\n\n    const historyPromise = this.history?.load() ?? Promise.resolve(null);\n    const agentCardPromise = this.client.getAgentCard().catch(() => undefined);\n\n    this._loadPromise = Promise.all([historyPromise, agentCardPromise])\n      .then(([repo, agentCard]) => {\n        if (agentCard) {\n          this.agentCardValue = agentCard;\n        }\n        if (repo) {\n          this.applyExternalMessageRepository(repo);\n        }\n      })\n      .catch((error) => {\n        this.onError?.(\n          error instanceof Error ? error : new Error(String(error)),\n        );\n      })\n      .finally(() => {\n        this._isLoading = false;\n        this.notifyUpdate();\n      });\n\n    this.notifyUpdate();\n    return this._loadPromise;\n  }\n\n  async append(message: AppendMessage): Promise<void> {\n    const startRun = message.startRun ?? message.role === \"user\";\n\n    const threadMessage = fromThreadMessageLike(\n      message as any,\n      generateId(),\n      FALLBACK_USER_STATUS,\n    );\n    const parentId =\n      message.parentId === null\n        ? null\n        : message.parentId && this.hasMessage(message.parentId)\n          ? message.parentId\n          : this.repository.headId;\n    this.addOrUpdateMessage(parentId, threadMessage);\n    this.switchToBranch(threadMessage.id);\n    this.notifyUpdate();\n    this.recordHistoryEntry(parentId, threadMessage);\n\n    if (!startRun) return;\n    await this.startRun(threadMessage);\n  }\n\n  async edit(message: AppendMessage): Promise<void> {\n    await this.append(message);\n  }\n\n  async reload(\n    parentId: string | null,\n    _config: { runConfig?: Record<string, unknown> } = {},\n  ): Promise<void> {\n    const messages =\n      parentId === null\n        ? []\n        : (this.tryGetMessages(parentId) ?? this.getMessages());\n    for (let i = messages.length - 1; i >= 0; i--) {\n      if (messages[i]!.role === \"user\") {\n        await this.startRun(messages[i]!);\n        return;\n      }\n    }\n  }\n\n  async cancel(): Promise<void> {\n    if (!this.abortController) return;\n\n    // Abort locally first so the stream stops immediately\n    this.abortController.abort();\n\n    // Then try to cancel the task on the server\n    if (this.currentTask?.id) {\n      try {\n        const updated = await this.client.cancelTask(this.currentTask.id);\n        this.currentTask = updated;\n      } catch {\n        // Server cancel failed; local abort already handled\n      }\n    }\n  }\n\n  private appendLinearChain(messages: readonly ThreadMessage[]): string | null {\n    let parentId: string | null = null;\n    let lastId: string | null = null;\n    const seen = new Set<string>();\n\n    for (const message of messages) {\n      if (seen.has(message.id)) continue;\n      seen.add(message.id);\n      this.addOrUpdateMessage(parentId, message);\n      parentId = message.id;\n      lastId = message.id;\n    }\n\n    return lastId;\n  }\n\n  private finalizeExternalApply(): void {\n    this.assistantHistoryParents.clear();\n    this.recordedHistoryIds.clear();\n    for (const { message } of this.getMessageRepository().messages) {\n      this.recordedHistoryIds.add(message.id);\n    }\n    this.currentTask = undefined;\n    this.currentArtifacts = [];\n    this.notifyUpdate();\n  }\n\n  applyExternalMessages(messages: readonly ThreadMessage[]): void {\n    if (messages.length === 0) {\n      this.clearRepository();\n    } else {\n      let expectedParentId: string | null = null;\n      let lastAppliedId: string | null = null;\n      let hardReplace = false;\n      const seen = new Set<string>();\n\n      for (const message of messages) {\n        if (seen.has(message.id)) continue;\n        seen.add(message.id);\n        const existing = this.tryGetMessage(message.id);\n        if (existing && existing.parentId !== expectedParentId) {\n          hardReplace = true;\n          break;\n        }\n        this.addOrUpdateMessage(expectedParentId, message);\n        expectedParentId = message.id;\n        lastAppliedId = message.id;\n      }\n\n      if (hardReplace) {\n        this.clearRepository();\n        lastAppliedId = this.appendLinearChain(messages);\n      }\n\n      this.resetRepositoryHead(lastAppliedId);\n    }\n\n    this.finalizeExternalApply();\n  }\n\n  private applyExternalMessageRepository(\n    loaded: ExportedMessageRepository,\n  ): void {\n    const headId = loaded.headId ?? loaded.messages.at(-1)?.message.id ?? null;\n    const ids = new Set<string>();\n    let degenerate = false;\n    for (const { message } of loaded.messages) {\n      if (ids.has(message.id)) {\n        degenerate = true;\n        break;\n      }\n      ids.add(message.id);\n    }\n    if (headId !== null && !ids.has(headId)) degenerate = true;\n\n    if (!degenerate) {\n      this.clearRepository();\n      let pending = [...loaded.messages];\n      const importedIds = new Set<string>();\n\n      while (pending.length > 0) {\n        const unresolved: typeof pending = [];\n        let progressed = false;\n        for (const item of pending) {\n          if (item.parentId !== null && !importedIds.has(item.parentId)) {\n            unresolved.push(item);\n            continue;\n          }\n          this.addOrUpdateMessage(item.parentId, item.message);\n          importedIds.add(item.message.id);\n          progressed = true;\n        }\n        if (!progressed) {\n          degenerate = true;\n          break;\n        }\n        pending = unresolved;\n      }\n    }\n\n    if (degenerate) {\n      this.clearRepository();\n      let previousId: string | null = null;\n      for (const { message } of loaded.messages) {\n        const existing = this.tryGetMessage(message.id);\n        this.addOrUpdateMessage(\n          existing ? existing.parentId : previousId,\n          message,\n        );\n        previousId = message.id;\n      }\n      this.resetRepositoryHead(previousId);\n    } else {\n      this.resetRepositoryHead(headId);\n    }\n\n    this.finalizeExternalApply();\n  }\n\n  // --- Run logic ---\n\n  private async startRun(userThreadMessage: ThreadMessage): Promise<void> {\n    // Cancel any in-progress run before starting a new one\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n\n    const a2aMessage = this.threadMessageToA2AMessage(userThreadMessage);\n\n    // Clear task if previous task reached terminal state\n    if (\n      this.currentTask &&\n      isTerminalTaskState(this.currentTask.status.state)\n    ) {\n      this.currentTask = undefined;\n    }\n\n    this.currentArtifacts = [];\n\n    const assistantParentId = userThreadMessage.id;\n    const assistantId = this.insertAssistantPlaceholder(assistantParentId);\n    this.markPendingAssistantHistory(assistantId, assistantParentId);\n\n    const abortController = new AbortController();\n    this.abortController = abortController;\n\n    abortController.signal.addEventListener(\n      \"abort\",\n      () => {\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"cancelled\",\n        });\n        this.finishRun(abortController);\n        this.onCancel?.();\n      },\n      { once: true },\n    );\n\n    this.setRunning(true);\n\n    // Check if agent supports streaming; fall back to sync sendMessage if not\n    const supportsStreaming =\n      this.agentCardValue?.capabilities?.streaming !== false;\n\n    try {\n      if (supportsStreaming) {\n        await this.runStreaming(a2aMessage, assistantId, abortController);\n      } else {\n        await this.runSync(a2aMessage, assistantId, abortController);\n      }\n    } catch (error) {\n      if (!abortController.signal.aborted) {\n        const err = error instanceof Error ? error : new Error(String(error));\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"error\",\n        });\n        this.onError?.(err);\n        this.pendingError = this.pendingError ?? err;\n      }\n    } finally {\n      this.finishRun(abortController);\n    }\n\n    if (this.pendingError) {\n      const err = this.pendingError;\n      this.pendingError = null;\n      throw err;\n    }\n  }\n\n  private async runStreaming(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const stream = this.client.streamMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    for await (const event of stream) {\n      if (abortController.signal.aborted) break;\n      this.handleStreamEvent(assistantId, event);\n    }\n\n    if (!abortController.signal.aborted) {\n      const lastStatus = this.getAssistantStatus(assistantId);\n      if (lastStatus?.type === \"running\") {\n        this.updateAssistantStatus(assistantId, {\n          type: \"complete\",\n          reason: \"stop\",\n        });\n      }\n    }\n  }\n\n  private async runSync(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const result = await this.client.sendMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    if (abortController.signal.aborted) return;\n\n    // Result is either A2ATask or A2AMessage\n    if (\"id\" in result && \"status\" in result) {\n      // It's a Task\n      this.handleTaskSnapshot(assistantId, result as A2ATask);\n    } else if (\"messageId\" in result && \"parts\" in result) {\n      // It's a Message\n      this.handleMessage(assistantId, result as A2AMessage);\n      this.updateAssistantStatus(assistantId, {\n        type: \"complete\",\n        reason: \"stop\",\n      });\n    }\n  }\n\n  private handleStreamEvent(assistantId: string, event: A2AStreamEvent) {\n    switch (event.type) {\n      case \"statusUpdate\":\n        this.handleStatusUpdate(assistantId, event.event);\n        break;\n      case \"artifactUpdate\":\n        this.handleArtifactUpdate(event.event);\n        break;\n      case \"message\":\n        this.handleMessage(assistantId, event.message);\n        break;\n      case \"task\":\n        this.handleTaskSnapshot(assistantId, event.task);\n        break;\n    }\n  }\n\n  private handleStatusUpdate(\n    assistantId: string,\n    event: A2ATaskStatusUpdateEvent,\n  ) {\n    if (!this.currentTask) {\n      this.currentTask = {\n        id: event.taskId,\n        contextId: event.contextId,\n        status: event.status,\n      };\n    } else {\n      this.currentTask = { ...this.currentTask, status: event.status };\n    }\n\n    if (event.contextId) {\n      this.contextId = event.contextId;\n    }\n\n    if (event.status.message) {\n      const content = a2aMessageToContent(event.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(event.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  private handleArtifactUpdate(event: A2ATaskArtifactUpdateEvent) {\n    const { artifact, append, lastChunk } = event;\n    const existingIdx = this.currentArtifacts.findIndex(\n      (a) => a.artifactId === artifact.artifactId,\n    );\n\n    let updated: A2AArtifact;\n    if (existingIdx >= 0 && append) {\n      const existing = this.currentArtifacts[existingIdx]!;\n      updated = {\n        ...existing,\n        parts: [...existing.parts, ...artifact.parts],\n      };\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else if (existingIdx >= 0) {\n      updated = artifact;\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else {\n      updated = artifact;\n      this.currentArtifacts = [...this.currentArtifacts, updated];\n    }\n\n    if (lastChunk) {\n      this.onArtifactComplete?.(updated);\n    }\n\n    this.notifyUpdate();\n  }\n\n  private handleMessage(assistantId: string, message: A2AMessage) {\n    if (message.role !== \"agent\") return;\n\n    const content = a2aMessageToContent(message);\n    this.updateAssistantContent(assistantId, content);\n    this.notifyUpdate();\n  }\n\n  private handleTaskSnapshot(assistantId: string, task: A2ATask) {\n    this.currentTask = task;\n\n    if (task.contextId) {\n      this.contextId = task.contextId;\n    }\n    if (task.artifacts) {\n      this.currentArtifacts = task.artifacts;\n    }\n\n    if (task.status.message) {\n      const content = a2aMessageToContent(task.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(task.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  // --- Message helpers ---\n\n  private threadMessageToA2AMessage(message: ThreadMessage): A2AMessage {\n    const parts: A2APart[] = [];\n\n    if (message.role === \"user\") {\n      parts.push(...contentPartsToA2AParts(message.content));\n      for (const attachment of message.attachments ?? []) {\n        parts.push(\n          ...contentPartsToA2AParts(\n            attachment.content ?? [],\n            attachment.contentType,\n          ),\n        );\n      }\n    }\n\n    const a2aMsg: A2AMessage = {\n      messageId: message.id,\n      role: \"user\",\n      parts,\n    };\n\n    if (this.contextId) {\n      a2aMsg.contextId = this.contextId;\n    }\n    // Only attach taskId if current task is NOT in terminal state\n    if (\n      this.currentTask?.id &&\n      !isTerminalTaskState(this.currentTask.status.state)\n    ) {\n      a2aMsg.taskId = this.currentTask.id;\n    }\n\n    return a2aMsg;\n  }\n\n  private insertAssistantPlaceholder(parentId: string): string {\n    const id = generateId();\n    const assistant: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      createdAt: new Date(),\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: null,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n    };\n    this.addOrUpdateMessage(parentId, assistant);\n    this.switchToBranch(id);\n    this.notifyUpdate();\n    return id;\n  }\n\n  private updateAssistantContent(\n    messageId: string,\n    content: ThreadAssistantMessage[\"content\"],\n  ) {\n    this.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, content };\n    });\n  }\n\n  private updateAssistantStatus(messageId: string, status: MessageStatus) {\n    const touched = this.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, status };\n    });\n    if (touched) {\n      this.notifyUpdate();\n      if (status.type === \"complete\" || status.type === \"incomplete\") {\n        this.persistAssistantHistory(messageId);\n      }\n    }\n  }\n\n  private getAssistantStatus(messageId: string): MessageStatus | undefined {\n    const msg = this.tryGetMessage(messageId)?.message;\n    if (msg?.role !== \"assistant\") return undefined;\n    return msg.status;\n  }\n\n  // --- Lifecycle helpers ---\n\n  private setRunning(running: boolean) {\n    this.isRunningFlag = running;\n    this.notifyUpdate();\n  }\n\n  private finishRun(controller: AbortController | null) {\n    if (this.abortController === controller) {\n      this.abortController = null;\n    }\n    this.setRunning(false);\n  }\n\n  // --- History persistence ---\n\n  private recordHistoryEntry(parentId: string | null, message: ThreadMessage) {\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private markPendingAssistantHistory(\n    messageId: string,\n    parentId: string | null,\n  ) {\n    if (!this.history) return;\n    this.assistantHistoryParents.set(messageId, parentId);\n  }\n\n  private persistAssistantHistory(messageId: string) {\n    if (!this.history) return;\n    const parentId = this.assistantHistoryParents.get(messageId);\n    if (parentId === undefined) return;\n    const message = this.tryGetMessage(messageId)?.message;\n    if (!message || message.role !== \"assistant\") return;\n    if (\n      message.status?.type !== \"complete\" &&\n      message.status?.type !== \"incomplete\"\n    )\n      return;\n    this.assistantHistoryParents.delete(messageId);\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private appendHistoryItem(parentId: string | null, message: ThreadMessage) {\n    if (!this.history || this.recordedHistoryIds.has(message.id)) return;\n    this.recordedHistoryIds.add(message.id);\n    void this.history.append({ parentId, message }).catch(() => {\n      this.recordedHistoryIds.delete(message.id);\n    });\n  }\n}\n"],"mappings":";;;;;AA2CA,MAAM,uBAAuB;CAC3B,MAAM;CACN,QAAQ;AACV;AAEA,IAAa,uBAAb,MAAkC;CAChC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA;CACA,aAA8B,IAAI,kBAAkB;CACpD;CACA,gBAAwB;CACxB,kBAAkD;CAClD,eAAqC;CAGrC;CACA,mBAA0C,CAAC;CAC3C;CAGA,0CAA2C,IAAI,IAA2B;CAC1E,qCAAsC,IAAI,IAAY;CACtD,aAAqB;CACrB;CAEA,YAAY,SAAsC;EAChD,KAAK,SAAS,QAAQ;EACtB,KAAK,YAAY,QAAQ;EACzB,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,UAAU,QAAQ;EACvB,KAAK,eAAe,QAAQ;CAC9B;CAEA,cAAc,SAA4D;EACxE,KAAK,SAAS,QAAQ;EACtB,KAAK,YAAY,QAAQ;EACzB,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,UAAU,QAAQ;CACzB;CAEA,cAAc,SAA2B;EACvC,KAAK,UAAU;CACjB;CAEA,gBAAgB;EACd,KAAK,UAAU,KAAA;EAEf,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;CACF;CAEA,aAA2C;EACzC,OAAO,KAAK;CACd;CAEA,cAAwC;EACtC,OAAO,KAAK,WAAW,YAAY;CACrC;CAEA,uBAAkD;EAChD,KAAK,uBAAuB,KAAK,WAAW,OAAO;EACnD,OAAO,KAAK;CACd;CAEA,cAAsB,WAAmB;EACvC,IAAI;GACF,OAAO,KAAK,WAAW,WAAW,SAAS;EAC7C,QAAQ;GACN;EACF;CACF;CAEA,eACE,WACsC;EACtC,IAAI;GACF,OAAO,KAAK,WAAW,YAAY,SAAS;EAC9C,QAAQ;GACN;EACF;CACF;CAEA,WAAmB,WAA4B;EAC7C,OAAO,KAAK,cAAc,SAAS,MAAM,KAAA;CAC3C;CAEA,mBACE,UACA,SACM;EACN,KAAK,WAAW,mBAAmB,UAAU,OAAO;EACpD,KAAK,qBAAqB,KAAA;CAC5B;CAEA,eAAuB,WAAyB;EAC9C,KAAK,WAAW,eAAe,SAAS;EACxC,KAAK,qBAAqB,KAAA;CAC5B;CAEA,oBAA4B,WAAgC;EAC1D,KAAK,WAAW,UAAU,SAAS;EACnC,KAAK,qBAAqB,KAAA;CAC5B;CAEA,kBAAgC;EAC9B,KAAK,WAAW,MAAM;EACtB,KAAK,qBAAqB,KAAA;CAC5B;CAEA,cACE,WACA,SACS;EACT,MAAM,OAAO,KAAK,cAAc,SAAS;EACzC,IAAI,CAAC,MAAM,OAAO;EAClB,MAAM,UAAU,QAAQ,KAAK,OAAO;EACpC,IAAI,YAAY,KAAK,SAAS,OAAO;EACrC,KAAK,mBAAmB,KAAK,UAAU,OAAO;EAC9C,OAAO;CACT;CAEA,UAA+B;EAC7B,OAAO,KAAK;CACd;CAEA,eAAuC;EACrC,OAAO,KAAK;CACd;CAEA,eAAyC;EACvC,OAAO,KAAK;CACd;CAEA,YAAqB;EACnB,OAAO,KAAK;CACd;CAEA,IAAI,YAAqB;EACvB,OAAO,KAAK;CACd;CAEA,kBAAiC;EAC/B,IAAI,KAAK,cAAc,OAAO,KAAK;EAEnC,KAAK,aAAa;EAElB,MAAM,iBAAiB,KAAK,SAAS,KAAK,KAAK,QAAQ,QAAQ,IAAI;EACnE,MAAM,mBAAmB,KAAK,OAAO,aAAa,CAAC,CAAC,YAAY,KAAA,CAAS;EAEzE,KAAK,eAAe,QAAQ,IAAI,CAAC,gBAAgB,gBAAgB,CAAC,CAAC,CAChE,MAAM,CAAC,MAAM,eAAe;GAC3B,IAAI,WACF,KAAK,iBAAiB;GAExB,IAAI,MACF,KAAK,+BAA+B,IAAI;EAE5C,CAAC,CAAC,CACD,OAAO,UAAU;GAChB,KAAK,UACH,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,CAAC,CACD,cAAc;GACb,KAAK,aAAa;GAClB,KAAK,aAAa;EACpB,CAAC;EAEH,KAAK,aAAa;EAClB,OAAO,KAAK;CACd;CAEA,MAAM,OAAO,SAAuC;EAClD,MAAM,WAAW,QAAQ,YAAY,QAAQ,SAAS;EAEtD,MAAM,gBAAgB,sBACpB,SACA,WAAW,GACX,oBACF;EACA,MAAM,WACJ,QAAQ,aAAa,OACjB,OACA,QAAQ,YAAY,KAAK,WAAW,QAAQ,QAAQ,IAClD,QAAQ,WACR,KAAK,WAAW;EACxB,KAAK,mBAAmB,UAAU,aAAa;EAC/C,KAAK,eAAe,cAAc,EAAE;EACpC,KAAK,aAAa;EAClB,KAAK,mBAAmB,UAAU,aAAa;EAE/C,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,SAAS,aAAa;CACnC;CAEA,MAAM,KAAK,SAAuC;EAChD,MAAM,KAAK,OAAO,OAAO;CAC3B;CAEA,MAAM,OACJ,UACA,UAAmD,CAAC,GACrC;EACf,MAAM,WACJ,aAAa,OACT,CAAC,IACA,KAAK,eAAe,QAAQ,KAAK,KAAK,YAAY;EACzD,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KACxC,IAAI,SAAS,EAAE,CAAE,SAAS,QAAQ;GAChC,MAAM,KAAK,SAAS,SAAS,EAAG;GAChC;EACF;CAEJ;CAEA,MAAM,SAAwB;EAC5B,IAAI,CAAC,KAAK,iBAAiB;EAG3B,KAAK,gBAAgB,MAAM;EAG3B,IAAI,KAAK,aAAa,IACpB,IAAI;GACF,MAAM,UAAU,MAAM,KAAK,OAAO,WAAW,KAAK,YAAY,EAAE;GAChE,KAAK,cAAc;EACrB,QAAQ,CAER;CAEJ;CAEA,kBAA0B,UAAmD;EAC3E,IAAI,WAA0B;EAC9B,IAAI,SAAwB;EAC5B,MAAM,uBAAO,IAAI,IAAY;EAE7B,KAAK,MAAM,WAAW,UAAU;GAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;GAC1B,KAAK,IAAI,QAAQ,EAAE;GACnB,KAAK,mBAAmB,UAAU,OAAO;GACzC,WAAW,QAAQ;GACnB,SAAS,QAAQ;EACnB;EAEA,OAAO;CACT;CAEA,wBAAsC;EACpC,KAAK,wBAAwB,MAAM;EACnC,KAAK,mBAAmB,MAAM;EAC9B,KAAK,MAAM,EAAE,aAAa,KAAK,qBAAqB,CAAC,CAAC,UACpD,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EAExC,KAAK,cAAc,KAAA;EACnB,KAAK,mBAAmB,CAAC;EACzB,KAAK,aAAa;CACpB;CAEA,sBAAsB,UAA0C;EAC9D,IAAI,SAAS,WAAW,GACtB,KAAK,gBAAgB;OAChB;GACL,IAAI,mBAAkC;GACtC,IAAI,gBAA+B;GACnC,IAAI,cAAc;GAClB,MAAM,uBAAO,IAAI,IAAY;GAE7B,KAAK,MAAM,WAAW,UAAU;IAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;IAC1B,KAAK,IAAI,QAAQ,EAAE;IACnB,MAAM,WAAW,KAAK,cAAc,QAAQ,EAAE;IAC9C,IAAI,YAAY,SAAS,aAAa,kBAAkB;KACtD,cAAc;KACd;IACF;IACA,KAAK,mBAAmB,kBAAkB,OAAO;IACjD,mBAAmB,QAAQ;IAC3B,gBAAgB,QAAQ;GAC1B;GAEA,IAAI,aAAa;IACf,KAAK,gBAAgB;IACrB,gBAAgB,KAAK,kBAAkB,QAAQ;GACjD;GAEA,KAAK,oBAAoB,aAAa;EACxC;EAEA,KAAK,sBAAsB;CAC7B;CAEA,+BACE,QACM;EACN,MAAM,SAAS,OAAO,UAAU,OAAO,SAAS,GAAG,EAAE,CAAC,EAAE,QAAQ,MAAM;EACtE,MAAM,sBAAM,IAAI,IAAY;EAC5B,IAAI,aAAa;EACjB,KAAK,MAAM,EAAE,aAAa,OAAO,UAAU;GACzC,IAAI,IAAI,IAAI,QAAQ,EAAE,GAAG;IACvB,aAAa;IACb;GACF;GACA,IAAI,IAAI,QAAQ,EAAE;EACpB;EACA,IAAI,WAAW,QAAQ,CAAC,IAAI,IAAI,MAAM,GAAG,aAAa;EAEtD,IAAI,CAAC,YAAY;GACf,KAAK,gBAAgB;GACrB,IAAI,UAAU,CAAC,GAAG,OAAO,QAAQ;GACjC,MAAM,8BAAc,IAAI,IAAY;GAEpC,OAAO,QAAQ,SAAS,GAAG;IACzB,MAAM,aAA6B,CAAC;IACpC,IAAI,aAAa;IACjB,KAAK,MAAM,QAAQ,SAAS;KAC1B,IAAI,KAAK,aAAa,QAAQ,CAAC,YAAY,IAAI,KAAK,QAAQ,GAAG;MAC7D,WAAW,KAAK,IAAI;MACpB;KACF;KACA,KAAK,mBAAmB,KAAK,UAAU,KAAK,OAAO;KACnD,YAAY,IAAI,KAAK,QAAQ,EAAE;KAC/B,aAAa;IACf;IACA,IAAI,CAAC,YAAY;KACf,aAAa;KACb;IACF;IACA,UAAU;GACZ;EACF;EAEA,IAAI,YAAY;GACd,KAAK,gBAAgB;GACrB,IAAI,aAA4B;GAChC,KAAK,MAAM,EAAE,aAAa,OAAO,UAAU;IACzC,MAAM,WAAW,KAAK,cAAc,QAAQ,EAAE;IAC9C,KAAK,mBACH,WAAW,SAAS,WAAW,YAC/B,OACF;IACA,aAAa,QAAQ;GACvB;GACA,KAAK,oBAAoB,UAAU;EACrC,OACE,KAAK,oBAAoB,MAAM;EAGjC,KAAK,sBAAsB;CAC7B;CAIA,MAAc,SAAS,mBAAiD;EAEtE,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;EAEA,MAAM,aAAa,KAAK,0BAA0B,iBAAiB;EAGnE,IACE,KAAK,eACL,oBAAoB,KAAK,YAAY,OAAO,KAAK,GAEjD,KAAK,cAAc,KAAA;EAGrB,KAAK,mBAAmB,CAAC;EAEzB,MAAM,oBAAoB,kBAAkB;EAC5C,MAAM,cAAc,KAAK,2BAA2B,iBAAiB;EACrE,KAAK,4BAA4B,aAAa,iBAAiB;EAE/D,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,KAAK,kBAAkB;EAEvB,gBAAgB,OAAO,iBACrB,eACM;GACJ,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;GACD,KAAK,UAAU,eAAe;GAC9B,KAAK,WAAW;EAClB,GACA,EAAE,MAAM,KAAK,CACf;EAEA,KAAK,WAAW,IAAI;EAGpB,MAAM,oBACJ,KAAK,gBAAgB,cAAc,cAAc;EAEnD,IAAI;GACF,IAAI,mBACF,MAAM,KAAK,aAAa,YAAY,aAAa,eAAe;QAEhE,MAAM,KAAK,QAAQ,YAAY,aAAa,eAAe;EAE/D,SAAS,OAAO;GACd,IAAI,CAAC,gBAAgB,OAAO,SAAS;IACnC,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IACpE,KAAK,sBAAsB,aAAa;KACtC,MAAM;KACN,QAAQ;IACV,CAAC;IACD,KAAK,UAAU,GAAG;IAClB,KAAK,eAAe,KAAK,gBAAgB;GAC3C;EACF,UAAU;GACR,KAAK,UAAU,eAAe;EAChC;EAEA,IAAI,KAAK,cAAc;GACrB,MAAM,MAAM,KAAK;GACjB,KAAK,eAAe;GACpB,MAAM;EACR;CACF;CAEA,MAAc,aACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,KAAK,OAAO,cACzB,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,WAAW,MAAM,SAAS,QAAQ;GAChC,IAAI,gBAAgB,OAAO,SAAS;GACpC,KAAK,kBAAkB,aAAa,KAAK;EAC3C;EAEA,IAAI,CAAC,gBAAgB,OAAO,SACP;OAAA,KAAK,mBAAmB,WAC9B,CAAC,EAAE,SAAS,WACvB,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EAAA;CAGP;CAEA,MAAc,QACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,MAAM,KAAK,OAAO,YAC/B,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,IAAI,gBAAgB,OAAO,SAAS;EAGpC,IAAI,QAAQ,UAAU,YAAY,QAEhC,KAAK,mBAAmB,aAAa,MAAiB;OACjD,IAAI,eAAe,UAAU,WAAW,QAAQ;GAErD,KAAK,cAAc,aAAa,MAAoB;GACpD,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EACH;CACF;CAEA,kBAA0B,aAAqB,OAAuB;EACpE,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,mBAAmB,aAAa,MAAM,KAAK;IAChD;GACF,KAAK;IACH,KAAK,qBAAqB,MAAM,KAAK;IACrC;GACF,KAAK;IACH,KAAK,cAAc,aAAa,MAAM,OAAO;IAC7C;GACF,KAAK;IACH,KAAK,mBAAmB,aAAa,MAAM,IAAI;IAC/C;EACJ;CACF;CAEA,mBACE,aACA,OACA;EACA,IAAI,CAAC,KAAK,aACR,KAAK,cAAc;GACjB,IAAI,MAAM;GACV,WAAW,MAAM;GACjB,QAAQ,MAAM;EAChB;OAEA,KAAK,cAAc;GAAE,GAAG,KAAK;GAAa,QAAQ,MAAM;EAAO;EAGjE,IAAI,MAAM,WACR,KAAK,YAAY,MAAM;EAGzB,IAAI,MAAM,OAAO,SAAS;GACxB,MAAM,UAAU,oBAAoB,MAAM,OAAO,OAAO;GACxD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,MAAM,OAAO,KAAK;EAC1D,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAEA,qBAA6B,OAAmC;EAC9D,MAAM,EAAE,UAAU,QAAQ,cAAc;EACxC,MAAM,cAAc,KAAK,iBAAiB,WACvC,MAAM,EAAE,eAAe,SAAS,UACnC;EAEA,IAAI;EACJ,IAAI,eAAe,KAAK,QAAQ;GAC9B,MAAM,WAAW,KAAK,iBAAiB;GACvC,UAAU;IACR,GAAG;IACH,OAAO,CAAC,GAAG,SAAS,OAAO,GAAG,SAAS,KAAK;GAC9C;GACA,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO,IAAI,eAAe,GAAG;GAC3B,UAAU;GACV,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO;GACL,UAAU;GACV,KAAK,mBAAmB,CAAC,GAAG,KAAK,kBAAkB,OAAO;EAC5D;EAEA,IAAI,WACF,KAAK,qBAAqB,OAAO;EAGnC,KAAK,aAAa;CACpB;CAEA,cAAsB,aAAqB,SAAqB;EAC9D,IAAI,QAAQ,SAAS,SAAS;EAE9B,MAAM,UAAU,oBAAoB,OAAO;EAC3C,KAAK,uBAAuB,aAAa,OAAO;EAChD,KAAK,aAAa;CACpB;CAEA,mBAA2B,aAAqB,MAAe;EAC7D,KAAK,cAAc;EAEnB,IAAI,KAAK,WACP,KAAK,YAAY,KAAK;EAExB,IAAI,KAAK,WACP,KAAK,mBAAmB,KAAK;EAG/B,IAAI,KAAK,OAAO,SAAS;GACvB,MAAM,UAAU,oBAAoB,KAAK,OAAO,OAAO;GACvD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,KAAK,OAAO,KAAK;EACzD,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAIA,0BAAkC,SAAoC;EACpE,MAAM,QAAmB,CAAC;EAE1B,IAAI,QAAQ,SAAS,QAAQ;GAC3B,MAAM,KAAK,GAAG,uBAAuB,QAAQ,OAAO,CAAC;GACrD,KAAK,MAAM,cAAc,QAAQ,eAAe,CAAC,GAC/C,MAAM,KACJ,GAAG,uBACD,WAAW,WAAW,CAAC,GACvB,WAAW,WACb,CACF;EAEJ;EAEA,MAAM,SAAqB;GACzB,WAAW,QAAQ;GACnB,MAAM;GACN;EACF;EAEA,IAAI,KAAK,WACP,OAAO,YAAY,KAAK;EAG1B,IACE,KAAK,aAAa,MAClB,CAAC,oBAAoB,KAAK,YAAY,OAAO,KAAK,GAElD,OAAO,SAAS,KAAK,YAAY;EAGnC,OAAO;CACT;CAEA,2BAAmC,UAA0B;EAC3D,MAAM,KAAK,WAAW;EACtB,MAAM,YAAoC;GACxC;GACA,MAAM;GACN,2BAAW,IAAI,KAAK;GACpB,QAAQ,EAAE,MAAM,UAAU;GAC1B,SAAS,CAAC;GACV,UAAU;IACR,gBAAgB;IAChB,sBAAsB,CAAC;IACvB,eAAe,CAAC;IAChB,OAAO,CAAC;IACR,QAAQ,CAAC;GACX;EACF;EACA,KAAK,mBAAmB,UAAU,SAAS;EAC3C,KAAK,eAAe,EAAE;EACtB,KAAK,aAAa;EAClB,OAAO;CACT;CAEA,uBACE,WACA,SACA;EACA,KAAK,cAAc,YAAY,YAAY;GACzC,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAQ;EAC/B,CAAC;CACH;CAEA,sBAA8B,WAAmB,QAAuB;EAKtE,IAJgB,KAAK,cAAc,YAAY,YAAY;GACzD,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAO;EAC9B,CACU,GAAG;GACX,KAAK,aAAa;GAClB,IAAI,OAAO,SAAS,cAAc,OAAO,SAAS,cAChD,KAAK,wBAAwB,SAAS;EAE1C;CACF;CAEA,mBAA2B,WAA8C;EACvE,MAAM,MAAM,KAAK,cAAc,SAAS,CAAC,EAAE;EAC3C,IAAI,KAAK,SAAS,aAAa,OAAO,KAAA;EACtC,OAAO,IAAI;CACb;CAIA,WAAmB,SAAkB;EACnC,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,UAAkB,YAAoC;EACpD,IAAI,KAAK,oBAAoB,YAC3B,KAAK,kBAAkB;EAEzB,KAAK,WAAW,KAAK;CACvB;CAIA,mBAA2B,UAAyB,SAAwB;EAC1E,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,4BACE,WACA,UACA;EACA,IAAI,CAAC,KAAK,SAAS;EACnB,KAAK,wBAAwB,IAAI,WAAW,QAAQ;CACtD;CAEA,wBAAgC,WAAmB;EACjD,IAAI,CAAC,KAAK,SAAS;EACnB,MAAM,WAAW,KAAK,wBAAwB,IAAI,SAAS;EAC3D,IAAI,aAAa,KAAA,GAAW;EAC5B,MAAM,UAAU,KAAK,cAAc,SAAS,CAAC,EAAE;EAC/C,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,IACE,QAAQ,QAAQ,SAAS,cACzB,QAAQ,QAAQ,SAAS,cAEzB;EACF,KAAK,wBAAwB,OAAO,SAAS;EAC7C,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,kBAA0B,UAAyB,SAAwB;EACzE,IAAI,CAAC,KAAK,WAAW,KAAK,mBAAmB,IAAI,QAAQ,EAAE,GAAG;EAC9D,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EACtC,KAAU,QAAQ,OAAO;GAAE;GAAU;EAAQ,CAAC,CAAC,CAAC,YAAY;GAC1D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;EAC3C,CAAC;CACH;AACF"}
+{"version":3,"file":"A2AThreadRuntimeCore.js","names":[],"sources":["../src/A2AThreadRuntimeCore.ts"],"sourcesContent":["\"use client\";\n\nimport { generateId, fromThreadMessageLike } from \"@assistant-ui/core\";\nimport type {\n  AppendMessage,\n  AssistantRuntime,\n  ExportedMessageRepository,\n  MessageStatus,\n  ThreadAssistantMessage,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport type { A2AClient } from \"./A2AClient\";\nimport type {\n  A2AArtifact,\n  A2AAgentCard,\n  A2AMessage,\n  A2APart,\n  A2ASendMessageConfiguration,\n  A2AStreamEvent,\n  A2ATask,\n  A2ATaskArtifactUpdateEvent,\n  A2ATaskStatusUpdateEvent,\n} from \"./types\";\nimport {\n  a2aMessageToContent,\n  contentPartsToA2AParts,\n  isTerminalTaskState,\n  taskStateToMessageStatus,\n} from \"./conversions\";\n\nexport type A2AThreadRuntimeCoreOptions = {\n  client: A2AClient;\n  contextId?: string | undefined;\n  configuration?: A2ASendMessageConfiguration | undefined;\n  onError?: ((error: Error) => void) | undefined;\n  onCancel?: (() => void) | undefined;\n  onArtifactComplete?: ((artifact: A2AArtifact) => void) | undefined;\n  history?: ThreadHistoryAdapter | undefined;\n  notifyUpdate: () => void;\n};\n\nconst FALLBACK_USER_STATUS = {\n  type: \"complete\",\n  reason: \"unknown\",\n} as const;\n\ntype A2ARuntimeCallbackName = \"onError\" | \"onCancel\" | \"onArtifactComplete\";\n\nconst reportCallbackError = (name: A2ARuntimeCallbackName, error: unknown) => {\n  console.error(`[react-a2a] ${name} callback threw an error`, error);\n};\n\nconst invokeRuntimeCallback = <TArgs extends unknown[]>(\n  name: A2ARuntimeCallbackName,\n  callback: ((...args: TArgs) => void) | undefined,\n  ...args: TArgs\n) => {\n  if (!callback) return;\n\n  try {\n    const result = callback(...args) as unknown;\n    if (\n      result !== null &&\n      (typeof result === \"object\" || typeof result === \"function\") &&\n      \"then\" in result &&\n      typeof result.then === \"function\"\n    ) {\n      void Promise.resolve(result).catch((error) => {\n        reportCallbackError(name, error);\n      });\n    }\n  } catch (error) {\n    reportCallbackError(name, error);\n  }\n};\n\nexport class A2AThreadRuntimeCore {\n  private client: A2AClient;\n  private contextId: string | undefined;\n  private configuration: A2ASendMessageConfiguration | undefined;\n  private onError: ((error: Error) => void) | undefined;\n  private onCancel: (() => void) | undefined;\n  private onArtifactComplete: ((artifact: A2AArtifact) => void) | undefined;\n  private history: ThreadHistoryAdapter | undefined;\n  private readonly notifyUpdate: () => void;\n\n  private runtime: AssistantRuntime | undefined;\n  private readonly repository = new MessageRepository();\n  private exportedRepository: ExportedMessageRepository | undefined;\n  private isRunningFlag = false;\n  private abortController: AbortController | null = null;\n  private pendingError: Error | null = null;\n\n  // A2A-specific state\n  private currentTask: A2ATask | undefined;\n  private currentArtifacts: A2AArtifact[] = [];\n  private agentCardValue: A2AAgentCard | undefined;\n\n  // History tracking\n  private readonly assistantHistoryParents = new Map<string, string | null>();\n  private readonly recordedHistoryIds = new Set<string>();\n  private _isLoading = false;\n  private _loadPromise: Promise<void> | undefined;\n\n  constructor(options: A2AThreadRuntimeCoreOptions) {\n    this.client = options.client;\n    this.contextId = options.contextId;\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    this.history = options.history;\n    this.notifyUpdate = options.notifyUpdate;\n  }\n\n  updateOptions(options: Omit<A2AThreadRuntimeCoreOptions, \"notifyUpdate\">) {\n    this.client = options.client;\n    this.contextId = options.contextId;\n    this.configuration = options.configuration;\n    this.onError = options.onError;\n    this.onCancel = options.onCancel;\n    this.onArtifactComplete = options.onArtifactComplete;\n    this.history = options.history;\n  }\n\n  attachRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  detachRuntime() {\n    this.runtime = undefined;\n    // Abort in-flight requests on unmount\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n  }\n\n  getRuntime(): AssistantRuntime | undefined {\n    return this.runtime;\n  }\n\n  getMessages(): readonly ThreadMessage[] {\n    return this.repository.getMessages();\n  }\n\n  getMessageRepository(): ExportedMessageRepository {\n    this.exportedRepository ??= this.repository.export();\n    return this.exportedRepository;\n  }\n\n  private tryGetMessage(messageId: string) {\n    try {\n      return this.repository.getMessage(messageId);\n    } catch {\n      return undefined;\n    }\n  }\n\n  private tryGetMessages(\n    messageId: string,\n  ): readonly ThreadMessage[] | undefined {\n    try {\n      return this.repository.getMessages(messageId);\n    } catch {\n      return undefined;\n    }\n  }\n\n  private hasMessage(messageId: string): boolean {\n    return this.tryGetMessage(messageId) !== undefined;\n  }\n\n  private addOrUpdateMessage(\n    parentId: string | null,\n    message: ThreadMessage,\n  ): void {\n    this.repository.addOrUpdateMessage(parentId, message);\n    this.exportedRepository = undefined;\n  }\n\n  private switchToBranch(messageId: string): void {\n    this.repository.switchToBranch(messageId);\n    this.exportedRepository = undefined;\n  }\n\n  private resetRepositoryHead(messageId: string | null): void {\n    this.repository.resetHead(messageId);\n    this.exportedRepository = undefined;\n  }\n\n  private clearRepository(): void {\n    this.repository.clear();\n    this.exportedRepository = undefined;\n  }\n\n  private updateMessage(\n    messageId: string,\n    updater: (message: ThreadMessage) => ThreadMessage,\n  ): boolean {\n    const item = this.tryGetMessage(messageId);\n    if (!item) return false;\n    const message = updater(item.message);\n    if (message === item.message) return false;\n    this.addOrUpdateMessage(item.parentId, message);\n    return true;\n  }\n\n  getTask(): A2ATask | undefined {\n    return this.currentTask;\n  }\n\n  getArtifacts(): readonly A2AArtifact[] {\n    return this.currentArtifacts;\n  }\n\n  getAgentCard(): A2AAgentCard | undefined {\n    return this.agentCardValue;\n  }\n\n  isRunning(): boolean {\n    return this.isRunningFlag;\n  }\n\n  get isLoading(): boolean {\n    return this._isLoading;\n  }\n\n  __internal_load(): Promise<void> {\n    if (this._loadPromise) return this._loadPromise;\n\n    this._isLoading = true;\n\n    const historyPromise = this.history?.load() ?? Promise.resolve(null);\n    const agentCardPromise = this.client.getAgentCard().catch(() => undefined);\n\n    this._loadPromise = Promise.all([historyPromise, agentCardPromise])\n      .then(([repo, agentCard]) => {\n        if (agentCard) {\n          this.agentCardValue = agentCard;\n        }\n        if (repo) {\n          this.applyExternalMessageRepository(repo);\n        }\n      })\n      .catch((error) => {\n        invokeRuntimeCallback(\n          \"onError\",\n          this.onError,\n          error instanceof Error ? error : new Error(String(error)),\n        );\n      })\n      .finally(() => {\n        this._isLoading = false;\n        this.notifyUpdate();\n      });\n\n    this.notifyUpdate();\n    return this._loadPromise;\n  }\n\n  async append(message: AppendMessage): Promise<void> {\n    const startRun = message.startRun ?? message.role === \"user\";\n\n    const threadMessage = fromThreadMessageLike(\n      message as any,\n      generateId(),\n      FALLBACK_USER_STATUS,\n    );\n    const parentId =\n      message.parentId === null\n        ? null\n        : message.parentId && this.hasMessage(message.parentId)\n          ? message.parentId\n          : this.repository.headId;\n    this.addOrUpdateMessage(parentId, threadMessage);\n    this.switchToBranch(threadMessage.id);\n    this.notifyUpdate();\n    this.recordHistoryEntry(parentId, threadMessage);\n\n    if (!startRun) return;\n    await this.startRun(threadMessage);\n  }\n\n  async edit(message: AppendMessage): Promise<void> {\n    await this.append(message);\n  }\n\n  async reload(\n    parentId: string | null,\n    _config: { runConfig?: Record<string, unknown> } = {},\n  ): Promise<void> {\n    const messages =\n      parentId === null\n        ? []\n        : (this.tryGetMessages(parentId) ?? this.getMessages());\n    for (let i = messages.length - 1; i >= 0; i--) {\n      if (messages[i]!.role === \"user\") {\n        await this.startRun(messages[i]!);\n        return;\n      }\n    }\n  }\n\n  async cancel(): Promise<void> {\n    if (!this.abortController) return;\n\n    // Abort locally first so the stream stops immediately\n    this.abortController.abort();\n\n    // Then try to cancel the task on the server\n    if (this.currentTask?.id) {\n      try {\n        const updated = await this.client.cancelTask(this.currentTask.id);\n        this.currentTask = updated;\n      } catch {\n        // Server cancel failed; local abort already handled\n      }\n    }\n  }\n\n  private appendLinearChain(messages: readonly ThreadMessage[]): string | null {\n    let parentId: string | null = null;\n    let lastId: string | null = null;\n    const seen = new Set<string>();\n\n    for (const message of messages) {\n      if (seen.has(message.id)) continue;\n      seen.add(message.id);\n      this.addOrUpdateMessage(parentId, message);\n      parentId = message.id;\n      lastId = message.id;\n    }\n\n    return lastId;\n  }\n\n  private finalizeExternalApply(): void {\n    this.assistantHistoryParents.clear();\n    this.recordedHistoryIds.clear();\n    for (const { message } of this.getMessageRepository().messages) {\n      this.recordedHistoryIds.add(message.id);\n    }\n    this.currentTask = undefined;\n    this.currentArtifacts = [];\n    this.notifyUpdate();\n  }\n\n  applyExternalMessages(messages: readonly ThreadMessage[]): void {\n    if (messages.length === 0) {\n      this.clearRepository();\n    } else {\n      let expectedParentId: string | null = null;\n      let lastAppliedId: string | null = null;\n      let hardReplace = false;\n      const seen = new Set<string>();\n\n      for (const message of messages) {\n        if (seen.has(message.id)) continue;\n        seen.add(message.id);\n        const existing = this.tryGetMessage(message.id);\n        if (existing && existing.parentId !== expectedParentId) {\n          hardReplace = true;\n          break;\n        }\n        this.addOrUpdateMessage(expectedParentId, message);\n        expectedParentId = message.id;\n        lastAppliedId = message.id;\n      }\n\n      if (hardReplace) {\n        this.clearRepository();\n        lastAppliedId = this.appendLinearChain(messages);\n      }\n\n      this.resetRepositoryHead(lastAppliedId);\n    }\n\n    this.finalizeExternalApply();\n  }\n\n  private applyExternalMessageRepository(\n    loaded: ExportedMessageRepository,\n  ): void {\n    const headId = loaded.headId ?? loaded.messages.at(-1)?.message.id ?? null;\n    const ids = new Set<string>();\n    let degenerate = false;\n    for (const { message } of loaded.messages) {\n      if (ids.has(message.id)) {\n        degenerate = true;\n        break;\n      }\n      ids.add(message.id);\n    }\n    if (headId !== null && !ids.has(headId)) degenerate = true;\n\n    if (!degenerate) {\n      this.clearRepository();\n      let pending = [...loaded.messages];\n      const importedIds = new Set<string>();\n\n      while (pending.length > 0) {\n        const unresolved: typeof pending = [];\n        let progressed = false;\n        for (const item of pending) {\n          if (item.parentId !== null && !importedIds.has(item.parentId)) {\n            unresolved.push(item);\n            continue;\n          }\n          this.addOrUpdateMessage(item.parentId, item.message);\n          importedIds.add(item.message.id);\n          progressed = true;\n        }\n        if (!progressed) {\n          degenerate = true;\n          break;\n        }\n        pending = unresolved;\n      }\n    }\n\n    if (degenerate) {\n      this.clearRepository();\n      let previousId: string | null = null;\n      for (const { message } of loaded.messages) {\n        const existing = this.tryGetMessage(message.id);\n        this.addOrUpdateMessage(\n          existing ? existing.parentId : previousId,\n          message,\n        );\n        previousId = message.id;\n      }\n      this.resetRepositoryHead(previousId);\n    } else {\n      this.resetRepositoryHead(headId);\n    }\n\n    this.finalizeExternalApply();\n  }\n\n  // --- Run logic ---\n\n  private async startRun(userThreadMessage: ThreadMessage): Promise<void> {\n    // Cancel any in-progress run before starting a new one\n    if (this.abortController) {\n      this.abortController.abort();\n      this.abortController = null;\n    }\n\n    const a2aMessage = this.threadMessageToA2AMessage(userThreadMessage);\n\n    // Clear task if previous task reached terminal state\n    if (\n      this.currentTask &&\n      isTerminalTaskState(this.currentTask.status.state)\n    ) {\n      this.currentTask = undefined;\n    }\n\n    this.currentArtifacts = [];\n\n    const assistantParentId = userThreadMessage.id;\n    const assistantId = this.insertAssistantPlaceholder(assistantParentId);\n    this.markPendingAssistantHistory(assistantId, assistantParentId);\n\n    const abortController = new AbortController();\n    this.abortController = abortController;\n\n    abortController.signal.addEventListener(\n      \"abort\",\n      () => {\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"cancelled\",\n        });\n        this.finishRun(abortController);\n        invokeRuntimeCallback(\"onCancel\", this.onCancel);\n      },\n      { once: true },\n    );\n\n    this.setRunning(true);\n\n    // Check if agent supports streaming; fall back to sync sendMessage if not\n    const supportsStreaming =\n      this.agentCardValue?.capabilities?.streaming !== false;\n\n    try {\n      if (supportsStreaming) {\n        await this.runStreaming(a2aMessage, assistantId, abortController);\n      } else {\n        await this.runSync(a2aMessage, assistantId, abortController);\n      }\n    } catch (error) {\n      if (!abortController.signal.aborted) {\n        const err = error instanceof Error ? error : new Error(String(error));\n        this.updateAssistantStatus(assistantId, {\n          type: \"incomplete\",\n          reason: \"error\",\n        });\n        invokeRuntimeCallback(\"onError\", this.onError, err);\n        this.pendingError = this.pendingError ?? err;\n      }\n    } finally {\n      this.finishRun(abortController);\n    }\n\n    if (this.pendingError) {\n      const err = this.pendingError;\n      this.pendingError = null;\n      throw err;\n    }\n  }\n\n  private async runStreaming(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const stream = this.client.streamMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    for await (const event of stream) {\n      if (abortController.signal.aborted) break;\n      this.handleStreamEvent(assistantId, event);\n    }\n\n    if (!abortController.signal.aborted) {\n      const lastStatus = this.getAssistantStatus(assistantId);\n      if (lastStatus?.type === \"running\") {\n        this.updateAssistantStatus(assistantId, {\n          type: \"complete\",\n          reason: \"stop\",\n        });\n      }\n    }\n  }\n\n  private async runSync(\n    a2aMessage: A2AMessage,\n    assistantId: string,\n    abortController: AbortController,\n  ): Promise<void> {\n    const result = await this.client.sendMessage(\n      a2aMessage,\n      this.configuration,\n      undefined, // metadata\n      abortController.signal,\n    );\n\n    if (abortController.signal.aborted) return;\n\n    // Result is either A2ATask or A2AMessage\n    if (\"id\" in result && \"status\" in result) {\n      // It's a Task\n      this.handleTaskSnapshot(assistantId, result as A2ATask);\n    } else if (\"messageId\" in result && \"parts\" in result) {\n      // It's a Message\n      this.handleMessage(assistantId, result as A2AMessage);\n      this.updateAssistantStatus(assistantId, {\n        type: \"complete\",\n        reason: \"stop\",\n      });\n    }\n  }\n\n  private handleStreamEvent(assistantId: string, event: A2AStreamEvent) {\n    switch (event.type) {\n      case \"statusUpdate\":\n        this.handleStatusUpdate(assistantId, event.event);\n        break;\n      case \"artifactUpdate\":\n        this.handleArtifactUpdate(event.event);\n        break;\n      case \"message\":\n        this.handleMessage(assistantId, event.message);\n        break;\n      case \"task\":\n        this.handleTaskSnapshot(assistantId, event.task);\n        break;\n    }\n  }\n\n  private handleStatusUpdate(\n    assistantId: string,\n    event: A2ATaskStatusUpdateEvent,\n  ) {\n    if (!this.currentTask) {\n      this.currentTask = {\n        id: event.taskId,\n        contextId: event.contextId,\n        status: event.status,\n      };\n    } else {\n      this.currentTask = { ...this.currentTask, status: event.status };\n    }\n\n    if (event.contextId) {\n      this.contextId = event.contextId;\n    }\n\n    if (event.status.message) {\n      const content = a2aMessageToContent(event.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(event.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  private handleArtifactUpdate(event: A2ATaskArtifactUpdateEvent) {\n    const { artifact, append, lastChunk } = event;\n    const existingIdx = this.currentArtifacts.findIndex(\n      (a) => a.artifactId === artifact.artifactId,\n    );\n\n    let updated: A2AArtifact;\n    if (existingIdx >= 0 && append) {\n      const existing = this.currentArtifacts[existingIdx]!;\n      updated = {\n        ...existing,\n        parts: [...existing.parts, ...artifact.parts],\n      };\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else if (existingIdx >= 0) {\n      updated = artifact;\n      this.currentArtifacts = [\n        ...this.currentArtifacts.slice(0, existingIdx),\n        updated,\n        ...this.currentArtifacts.slice(existingIdx + 1),\n      ];\n    } else {\n      updated = artifact;\n      this.currentArtifacts = [...this.currentArtifacts, updated];\n    }\n\n    if (lastChunk) {\n      invokeRuntimeCallback(\n        \"onArtifactComplete\",\n        this.onArtifactComplete,\n        updated,\n      );\n    }\n\n    this.notifyUpdate();\n  }\n\n  private handleMessage(assistantId: string, message: A2AMessage) {\n    if (message.role !== \"agent\") return;\n\n    const content = a2aMessageToContent(message);\n    this.updateAssistantContent(assistantId, content);\n    this.notifyUpdate();\n  }\n\n  private handleTaskSnapshot(assistantId: string, task: A2ATask) {\n    this.currentTask = task;\n\n    if (task.contextId) {\n      this.contextId = task.contextId;\n    }\n    if (task.artifacts) {\n      this.currentArtifacts = task.artifacts;\n    }\n\n    if (task.status.message) {\n      const content = a2aMessageToContent(task.status.message);\n      this.updateAssistantContent(assistantId, content);\n    }\n\n    const status = taskStateToMessageStatus(task.status.state);\n    this.updateAssistantStatus(assistantId, status);\n\n    this.notifyUpdate();\n  }\n\n  // --- Message helpers ---\n\n  private threadMessageToA2AMessage(message: ThreadMessage): A2AMessage {\n    const parts: A2APart[] = [];\n\n    if (message.role === \"user\") {\n      parts.push(...contentPartsToA2AParts(message.content));\n      for (const attachment of message.attachments ?? []) {\n        parts.push(\n          ...contentPartsToA2AParts(\n            attachment.content ?? [],\n            attachment.contentType,\n          ),\n        );\n      }\n    }\n\n    const a2aMsg: A2AMessage = {\n      messageId: message.id,\n      role: \"user\",\n      parts,\n    };\n\n    if (this.contextId) {\n      a2aMsg.contextId = this.contextId;\n    }\n    // Only attach taskId if current task is NOT in terminal state\n    if (\n      this.currentTask?.id &&\n      !isTerminalTaskState(this.currentTask.status.state)\n    ) {\n      a2aMsg.taskId = this.currentTask.id;\n    }\n\n    return a2aMsg;\n  }\n\n  private insertAssistantPlaceholder(parentId: string): string {\n    const id = generateId();\n    const assistant: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      createdAt: new Date(),\n      status: { type: \"running\" },\n      content: [],\n      metadata: {\n        unstable_state: null,\n        unstable_annotations: [],\n        unstable_data: [],\n        steps: [],\n        custom: {},\n      },\n    };\n    this.addOrUpdateMessage(parentId, assistant);\n    this.switchToBranch(id);\n    this.notifyUpdate();\n    return id;\n  }\n\n  private updateAssistantContent(\n    messageId: string,\n    content: ThreadAssistantMessage[\"content\"],\n  ) {\n    this.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, content };\n    });\n  }\n\n  private updateAssistantStatus(messageId: string, status: MessageStatus) {\n    const touched = this.updateMessage(messageId, (message) => {\n      if (message.role !== \"assistant\") return message;\n      return { ...message, status };\n    });\n    if (touched) {\n      this.notifyUpdate();\n      if (status.type === \"complete\" || status.type === \"incomplete\") {\n        this.persistAssistantHistory(messageId);\n      }\n    }\n  }\n\n  private getAssistantStatus(messageId: string): MessageStatus | undefined {\n    const msg = this.tryGetMessage(messageId)?.message;\n    if (msg?.role !== \"assistant\") return undefined;\n    return msg.status;\n  }\n\n  // --- Lifecycle helpers ---\n\n  private setRunning(running: boolean) {\n    this.isRunningFlag = running;\n    this.notifyUpdate();\n  }\n\n  private finishRun(controller: AbortController | null) {\n    if (this.abortController !== controller) return;\n    this.abortController = null;\n    this.setRunning(false);\n  }\n\n  // --- History persistence ---\n\n  private recordHistoryEntry(parentId: string | null, message: ThreadMessage) {\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private markPendingAssistantHistory(\n    messageId: string,\n    parentId: string | null,\n  ) {\n    if (!this.history) return;\n    this.assistantHistoryParents.set(messageId, parentId);\n  }\n\n  private persistAssistantHistory(messageId: string) {\n    if (!this.history) return;\n    const parentId = this.assistantHistoryParents.get(messageId);\n    if (parentId === undefined) return;\n    const message = this.tryGetMessage(messageId)?.message;\n    if (!message || message.role !== \"assistant\") return;\n    if (\n      message.status?.type !== \"complete\" &&\n      message.status?.type !== \"incomplete\"\n    )\n      return;\n    this.assistantHistoryParents.delete(messageId);\n    this.appendHistoryItem(parentId, message);\n  }\n\n  private appendHistoryItem(parentId: string | null, message: ThreadMessage) {\n    if (!this.history || this.recordedHistoryIds.has(message.id)) return;\n    this.recordedHistoryIds.add(message.id);\n    void this.history.append({ parentId, message }).catch(() => {\n      this.recordedHistoryIds.delete(message.id);\n    });\n  }\n}\n"],"mappings":";;;;;AA2CA,MAAM,uBAAuB;CAC3B,MAAM;CACN,QAAQ;AACV;AAIA,MAAM,uBAAuB,MAA8B,UAAmB;CAC5E,QAAQ,MAAM,eAAe,KAAK,2BAA2B,KAAK;AACpE;AAEA,MAAM,yBACJ,MACA,UACA,GAAG,SACA;CACH,IAAI,CAAC,UAAU;CAEf,IAAI;EACF,MAAM,SAAS,SAAS,GAAG,IAAI;EAC/B,IACE,WAAW,SACV,OAAO,WAAW,YAAY,OAAO,WAAW,eACjD,UAAU,UACV,OAAO,OAAO,SAAS,YAEvB,QAAa,QAAQ,MAAM,CAAC,CAAC,OAAO,UAAU;GAC5C,oBAAoB,MAAM,KAAK;EACjC,CAAC;CAEL,SAAS,OAAO;EACd,oBAAoB,MAAM,KAAK;CACjC;AACF;AAEA,IAAa,uBAAb,MAAkC;CAChC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA;CACA,aAA8B,IAAI,kBAAkB;CACpD;CACA,gBAAwB;CACxB,kBAAkD;CAClD,eAAqC;CAGrC;CACA,mBAA0C,CAAC;CAC3C;CAGA,0CAA2C,IAAI,IAA2B;CAC1E,qCAAsC,IAAI,IAAY;CACtD,aAAqB;CACrB;CAEA,YAAY,SAAsC;EAChD,KAAK,SAAS,QAAQ;EACtB,KAAK,YAAY,QAAQ;EACzB,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,UAAU,QAAQ;EACvB,KAAK,eAAe,QAAQ;CAC9B;CAEA,cAAc,SAA4D;EACxE,KAAK,SAAS,QAAQ;EACtB,KAAK,YAAY,QAAQ;EACzB,KAAK,gBAAgB,QAAQ;EAC7B,KAAK,UAAU,QAAQ;EACvB,KAAK,WAAW,QAAQ;EACxB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,UAAU,QAAQ;CACzB;CAEA,cAAc,SAA2B;EACvC,KAAK,UAAU;CACjB;CAEA,gBAAgB;EACd,KAAK,UAAU,KAAA;EAEf,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;CACF;CAEA,aAA2C;EACzC,OAAO,KAAK;CACd;CAEA,cAAwC;EACtC,OAAO,KAAK,WAAW,YAAY;CACrC;CAEA,uBAAkD;EAChD,KAAK,uBAAuB,KAAK,WAAW,OAAO;EACnD,OAAO,KAAK;CACd;CAEA,cAAsB,WAAmB;EACvC,IAAI;GACF,OAAO,KAAK,WAAW,WAAW,SAAS;EAC7C,QAAQ;GACN;EACF;CACF;CAEA,eACE,WACsC;EACtC,IAAI;GACF,OAAO,KAAK,WAAW,YAAY,SAAS;EAC9C,QAAQ;GACN;EACF;CACF;CAEA,WAAmB,WAA4B;EAC7C,OAAO,KAAK,cAAc,SAAS,MAAM,KAAA;CAC3C;CAEA,mBACE,UACA,SACM;EACN,KAAK,WAAW,mBAAmB,UAAU,OAAO;EACpD,KAAK,qBAAqB,KAAA;CAC5B;CAEA,eAAuB,WAAyB;EAC9C,KAAK,WAAW,eAAe,SAAS;EACxC,KAAK,qBAAqB,KAAA;CAC5B;CAEA,oBAA4B,WAAgC;EAC1D,KAAK,WAAW,UAAU,SAAS;EACnC,KAAK,qBAAqB,KAAA;CAC5B;CAEA,kBAAgC;EAC9B,KAAK,WAAW,MAAM;EACtB,KAAK,qBAAqB,KAAA;CAC5B;CAEA,cACE,WACA,SACS;EACT,MAAM,OAAO,KAAK,cAAc,SAAS;EACzC,IAAI,CAAC,MAAM,OAAO;EAClB,MAAM,UAAU,QAAQ,KAAK,OAAO;EACpC,IAAI,YAAY,KAAK,SAAS,OAAO;EACrC,KAAK,mBAAmB,KAAK,UAAU,OAAO;EAC9C,OAAO;CACT;CAEA,UAA+B;EAC7B,OAAO,KAAK;CACd;CAEA,eAAuC;EACrC,OAAO,KAAK;CACd;CAEA,eAAyC;EACvC,OAAO,KAAK;CACd;CAEA,YAAqB;EACnB,OAAO,KAAK;CACd;CAEA,IAAI,YAAqB;EACvB,OAAO,KAAK;CACd;CAEA,kBAAiC;EAC/B,IAAI,KAAK,cAAc,OAAO,KAAK;EAEnC,KAAK,aAAa;EAElB,MAAM,iBAAiB,KAAK,SAAS,KAAK,KAAK,QAAQ,QAAQ,IAAI;EACnE,MAAM,mBAAmB,KAAK,OAAO,aAAa,CAAC,CAAC,YAAY,KAAA,CAAS;EAEzE,KAAK,eAAe,QAAQ,IAAI,CAAC,gBAAgB,gBAAgB,CAAC,CAAC,CAChE,MAAM,CAAC,MAAM,eAAe;GAC3B,IAAI,WACF,KAAK,iBAAiB;GAExB,IAAI,MACF,KAAK,+BAA+B,IAAI;EAE5C,CAAC,CAAC,CACD,OAAO,UAAU;GAChB,sBACE,WACA,KAAK,SACL,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAC1D;EACF,CAAC,CAAC,CACD,cAAc;GACb,KAAK,aAAa;GAClB,KAAK,aAAa;EACpB,CAAC;EAEH,KAAK,aAAa;EAClB,OAAO,KAAK;CACd;CAEA,MAAM,OAAO,SAAuC;EAClD,MAAM,WAAW,QAAQ,YAAY,QAAQ,SAAS;EAEtD,MAAM,gBAAgB,sBACpB,SACA,WAAW,GACX,oBACF;EACA,MAAM,WACJ,QAAQ,aAAa,OACjB,OACA,QAAQ,YAAY,KAAK,WAAW,QAAQ,QAAQ,IAClD,QAAQ,WACR,KAAK,WAAW;EACxB,KAAK,mBAAmB,UAAU,aAAa;EAC/C,KAAK,eAAe,cAAc,EAAE;EACpC,KAAK,aAAa;EAClB,KAAK,mBAAmB,UAAU,aAAa;EAE/C,IAAI,CAAC,UAAU;EACf,MAAM,KAAK,SAAS,aAAa;CACnC;CAEA,MAAM,KAAK,SAAuC;EAChD,MAAM,KAAK,OAAO,OAAO;CAC3B;CAEA,MAAM,OACJ,UACA,UAAmD,CAAC,GACrC;EACf,MAAM,WACJ,aAAa,OACT,CAAC,IACA,KAAK,eAAe,QAAQ,KAAK,KAAK,YAAY;EACzD,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KACxC,IAAI,SAAS,EAAE,CAAE,SAAS,QAAQ;GAChC,MAAM,KAAK,SAAS,SAAS,EAAG;GAChC;EACF;CAEJ;CAEA,MAAM,SAAwB;EAC5B,IAAI,CAAC,KAAK,iBAAiB;EAG3B,KAAK,gBAAgB,MAAM;EAG3B,IAAI,KAAK,aAAa,IACpB,IAAI;GACF,MAAM,UAAU,MAAM,KAAK,OAAO,WAAW,KAAK,YAAY,EAAE;GAChE,KAAK,cAAc;EACrB,QAAQ,CAER;CAEJ;CAEA,kBAA0B,UAAmD;EAC3E,IAAI,WAA0B;EAC9B,IAAI,SAAwB;EAC5B,MAAM,uBAAO,IAAI,IAAY;EAE7B,KAAK,MAAM,WAAW,UAAU;GAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;GAC1B,KAAK,IAAI,QAAQ,EAAE;GACnB,KAAK,mBAAmB,UAAU,OAAO;GACzC,WAAW,QAAQ;GACnB,SAAS,QAAQ;EACnB;EAEA,OAAO;CACT;CAEA,wBAAsC;EACpC,KAAK,wBAAwB,MAAM;EACnC,KAAK,mBAAmB,MAAM;EAC9B,KAAK,MAAM,EAAE,aAAa,KAAK,qBAAqB,CAAC,CAAC,UACpD,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EAExC,KAAK,cAAc,KAAA;EACnB,KAAK,mBAAmB,CAAC;EACzB,KAAK,aAAa;CACpB;CAEA,sBAAsB,UAA0C;EAC9D,IAAI,SAAS,WAAW,GACtB,KAAK,gBAAgB;OAChB;GACL,IAAI,mBAAkC;GACtC,IAAI,gBAA+B;GACnC,IAAI,cAAc;GAClB,MAAM,uBAAO,IAAI,IAAY;GAE7B,KAAK,MAAM,WAAW,UAAU;IAC9B,IAAI,KAAK,IAAI,QAAQ,EAAE,GAAG;IAC1B,KAAK,IAAI,QAAQ,EAAE;IACnB,MAAM,WAAW,KAAK,cAAc,QAAQ,EAAE;IAC9C,IAAI,YAAY,SAAS,aAAa,kBAAkB;KACtD,cAAc;KACd;IACF;IACA,KAAK,mBAAmB,kBAAkB,OAAO;IACjD,mBAAmB,QAAQ;IAC3B,gBAAgB,QAAQ;GAC1B;GAEA,IAAI,aAAa;IACf,KAAK,gBAAgB;IACrB,gBAAgB,KAAK,kBAAkB,QAAQ;GACjD;GAEA,KAAK,oBAAoB,aAAa;EACxC;EAEA,KAAK,sBAAsB;CAC7B;CAEA,+BACE,QACM;EACN,MAAM,SAAS,OAAO,UAAU,OAAO,SAAS,GAAG,EAAE,CAAC,EAAE,QAAQ,MAAM;EACtE,MAAM,sBAAM,IAAI,IAAY;EAC5B,IAAI,aAAa;EACjB,KAAK,MAAM,EAAE,aAAa,OAAO,UAAU;GACzC,IAAI,IAAI,IAAI,QAAQ,EAAE,GAAG;IACvB,aAAa;IACb;GACF;GACA,IAAI,IAAI,QAAQ,EAAE;EACpB;EACA,IAAI,WAAW,QAAQ,CAAC,IAAI,IAAI,MAAM,GAAG,aAAa;EAEtD,IAAI,CAAC,YAAY;GACf,KAAK,gBAAgB;GACrB,IAAI,UAAU,CAAC,GAAG,OAAO,QAAQ;GACjC,MAAM,8BAAc,IAAI,IAAY;GAEpC,OAAO,QAAQ,SAAS,GAAG;IACzB,MAAM,aAA6B,CAAC;IACpC,IAAI,aAAa;IACjB,KAAK,MAAM,QAAQ,SAAS;KAC1B,IAAI,KAAK,aAAa,QAAQ,CAAC,YAAY,IAAI,KAAK,QAAQ,GAAG;MAC7D,WAAW,KAAK,IAAI;MACpB;KACF;KACA,KAAK,mBAAmB,KAAK,UAAU,KAAK,OAAO;KACnD,YAAY,IAAI,KAAK,QAAQ,EAAE;KAC/B,aAAa;IACf;IACA,IAAI,CAAC,YAAY;KACf,aAAa;KACb;IACF;IACA,UAAU;GACZ;EACF;EAEA,IAAI,YAAY;GACd,KAAK,gBAAgB;GACrB,IAAI,aAA4B;GAChC,KAAK,MAAM,EAAE,aAAa,OAAO,UAAU;IACzC,MAAM,WAAW,KAAK,cAAc,QAAQ,EAAE;IAC9C,KAAK,mBACH,WAAW,SAAS,WAAW,YAC/B,OACF;IACA,aAAa,QAAQ;GACvB;GACA,KAAK,oBAAoB,UAAU;EACrC,OACE,KAAK,oBAAoB,MAAM;EAGjC,KAAK,sBAAsB;CAC7B;CAIA,MAAc,SAAS,mBAAiD;EAEtE,IAAI,KAAK,iBAAiB;GACxB,KAAK,gBAAgB,MAAM;GAC3B,KAAK,kBAAkB;EACzB;EAEA,MAAM,aAAa,KAAK,0BAA0B,iBAAiB;EAGnE,IACE,KAAK,eACL,oBAAoB,KAAK,YAAY,OAAO,KAAK,GAEjD,KAAK,cAAc,KAAA;EAGrB,KAAK,mBAAmB,CAAC;EAEzB,MAAM,oBAAoB,kBAAkB;EAC5C,MAAM,cAAc,KAAK,2BAA2B,iBAAiB;EACrE,KAAK,4BAA4B,aAAa,iBAAiB;EAE/D,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,KAAK,kBAAkB;EAEvB,gBAAgB,OAAO,iBACrB,eACM;GACJ,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;GACD,KAAK,UAAU,eAAe;GAC9B,sBAAsB,YAAY,KAAK,QAAQ;EACjD,GACA,EAAE,MAAM,KAAK,CACf;EAEA,KAAK,WAAW,IAAI;EAGpB,MAAM,oBACJ,KAAK,gBAAgB,cAAc,cAAc;EAEnD,IAAI;GACF,IAAI,mBACF,MAAM,KAAK,aAAa,YAAY,aAAa,eAAe;QAEhE,MAAM,KAAK,QAAQ,YAAY,aAAa,eAAe;EAE/D,SAAS,OAAO;GACd,IAAI,CAAC,gBAAgB,OAAO,SAAS;IACnC,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IACpE,KAAK,sBAAsB,aAAa;KACtC,MAAM;KACN,QAAQ;IACV,CAAC;IACD,sBAAsB,WAAW,KAAK,SAAS,GAAG;IAClD,KAAK,eAAe,KAAK,gBAAgB;GAC3C;EACF,UAAU;GACR,KAAK,UAAU,eAAe;EAChC;EAEA,IAAI,KAAK,cAAc;GACrB,MAAM,MAAM,KAAK;GACjB,KAAK,eAAe;GACpB,MAAM;EACR;CACF;CAEA,MAAc,aACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,KAAK,OAAO,cACzB,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,WAAW,MAAM,SAAS,QAAQ;GAChC,IAAI,gBAAgB,OAAO,SAAS;GACpC,KAAK,kBAAkB,aAAa,KAAK;EAC3C;EAEA,IAAI,CAAC,gBAAgB,OAAO,SACP;OAAA,KAAK,mBAAmB,WAC9B,CAAC,EAAE,SAAS,WACvB,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EAAA;CAGP;CAEA,MAAc,QACZ,YACA,aACA,iBACe;EACf,MAAM,SAAS,MAAM,KAAK,OAAO,YAC/B,YACA,KAAK,eACL,KAAA,GACA,gBAAgB,MAClB;EAEA,IAAI,gBAAgB,OAAO,SAAS;EAGpC,IAAI,QAAQ,UAAU,YAAY,QAEhC,KAAK,mBAAmB,aAAa,MAAiB;OACjD,IAAI,eAAe,UAAU,WAAW,QAAQ;GAErD,KAAK,cAAc,aAAa,MAAoB;GACpD,KAAK,sBAAsB,aAAa;IACtC,MAAM;IACN,QAAQ;GACV,CAAC;EACH;CACF;CAEA,kBAA0B,aAAqB,OAAuB;EACpE,QAAQ,MAAM,MAAd;GACE,KAAK;IACH,KAAK,mBAAmB,aAAa,MAAM,KAAK;IAChD;GACF,KAAK;IACH,KAAK,qBAAqB,MAAM,KAAK;IACrC;GACF,KAAK;IACH,KAAK,cAAc,aAAa,MAAM,OAAO;IAC7C;GACF,KAAK,QACH,KAAK,mBAAmB,aAAa,MAAM,IAAI;EAEnD;CACF;CAEA,mBACE,aACA,OACA;EACA,IAAI,CAAC,KAAK,aACR,KAAK,cAAc;GACjB,IAAI,MAAM;GACV,WAAW,MAAM;GACjB,QAAQ,MAAM;EAChB;OAEA,KAAK,cAAc;GAAE,GAAG,KAAK;GAAa,QAAQ,MAAM;EAAO;EAGjE,IAAI,MAAM,WACR,KAAK,YAAY,MAAM;EAGzB,IAAI,MAAM,OAAO,SAAS;GACxB,MAAM,UAAU,oBAAoB,MAAM,OAAO,OAAO;GACxD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,MAAM,OAAO,KAAK;EAC1D,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAEA,qBAA6B,OAAmC;EAC9D,MAAM,EAAE,UAAU,QAAQ,cAAc;EACxC,MAAM,cAAc,KAAK,iBAAiB,WACvC,MAAM,EAAE,eAAe,SAAS,UACnC;EAEA,IAAI;EACJ,IAAI,eAAe,KAAK,QAAQ;GAC9B,MAAM,WAAW,KAAK,iBAAiB;GACvC,UAAU;IACR,GAAG;IACH,OAAO,CAAC,GAAG,SAAS,OAAO,GAAG,SAAS,KAAK;GAC9C;GACA,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO,IAAI,eAAe,GAAG;GAC3B,UAAU;GACV,KAAK,mBAAmB;IACtB,GAAG,KAAK,iBAAiB,MAAM,GAAG,WAAW;IAC7C;IACA,GAAG,KAAK,iBAAiB,MAAM,cAAc,CAAC;GAChD;EACF,OAAO;GACL,UAAU;GACV,KAAK,mBAAmB,CAAC,GAAG,KAAK,kBAAkB,OAAO;EAC5D;EAEA,IAAI,WACF,sBACE,sBACA,KAAK,oBACL,OACF;EAGF,KAAK,aAAa;CACpB;CAEA,cAAsB,aAAqB,SAAqB;EAC9D,IAAI,QAAQ,SAAS,SAAS;EAE9B,MAAM,UAAU,oBAAoB,OAAO;EAC3C,KAAK,uBAAuB,aAAa,OAAO;EAChD,KAAK,aAAa;CACpB;CAEA,mBAA2B,aAAqB,MAAe;EAC7D,KAAK,cAAc;EAEnB,IAAI,KAAK,WACP,KAAK,YAAY,KAAK;EAExB,IAAI,KAAK,WACP,KAAK,mBAAmB,KAAK;EAG/B,IAAI,KAAK,OAAO,SAAS;GACvB,MAAM,UAAU,oBAAoB,KAAK,OAAO,OAAO;GACvD,KAAK,uBAAuB,aAAa,OAAO;EAClD;EAEA,MAAM,SAAS,yBAAyB,KAAK,OAAO,KAAK;EACzD,KAAK,sBAAsB,aAAa,MAAM;EAE9C,KAAK,aAAa;CACpB;CAIA,0BAAkC,SAAoC;EACpE,MAAM,QAAmB,CAAC;EAE1B,IAAI,QAAQ,SAAS,QAAQ;GAC3B,MAAM,KAAK,GAAG,uBAAuB,QAAQ,OAAO,CAAC;GACrD,KAAK,MAAM,cAAc,QAAQ,eAAe,CAAC,GAC/C,MAAM,KACJ,GAAG,uBACD,WAAW,WAAW,CAAC,GACvB,WAAW,WACb,CACF;EAEJ;EAEA,MAAM,SAAqB;GACzB,WAAW,QAAQ;GACnB,MAAM;GACN;EACF;EAEA,IAAI,KAAK,WACP,OAAO,YAAY,KAAK;EAG1B,IACE,KAAK,aAAa,MAClB,CAAC,oBAAoB,KAAK,YAAY,OAAO,KAAK,GAElD,OAAO,SAAS,KAAK,YAAY;EAGnC,OAAO;CACT;CAEA,2BAAmC,UAA0B;EAC3D,MAAM,KAAK,WAAW;EACtB,MAAM,YAAoC;GACxC;GACA,MAAM;GACN,2BAAW,IAAI,KAAK;GACpB,QAAQ,EAAE,MAAM,UAAU;GAC1B,SAAS,CAAC;GACV,UAAU;IACR,gBAAgB;IAChB,sBAAsB,CAAC;IACvB,eAAe,CAAC;IAChB,OAAO,CAAC;IACR,QAAQ,CAAC;GACX;EACF;EACA,KAAK,mBAAmB,UAAU,SAAS;EAC3C,KAAK,eAAe,EAAE;EACtB,KAAK,aAAa;EAClB,OAAO;CACT;CAEA,uBACE,WACA,SACA;EACA,KAAK,cAAc,YAAY,YAAY;GACzC,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAQ;EAC/B,CAAC;CACH;CAEA,sBAA8B,WAAmB,QAAuB;EAKtE,IAJgB,KAAK,cAAc,YAAY,YAAY;GACzD,IAAI,QAAQ,SAAS,aAAa,OAAO;GACzC,OAAO;IAAE,GAAG;IAAS;GAAO;EAC9B,CACU,GAAG;GACX,KAAK,aAAa;GAClB,IAAI,OAAO,SAAS,cAAc,OAAO,SAAS,cAChD,KAAK,wBAAwB,SAAS;EAE1C;CACF;CAEA,mBAA2B,WAA8C;EACvE,MAAM,MAAM,KAAK,cAAc,SAAS,CAAC,EAAE;EAC3C,IAAI,KAAK,SAAS,aAAa,OAAO,KAAA;EACtC,OAAO,IAAI;CACb;CAIA,WAAmB,SAAkB;EACnC,KAAK,gBAAgB;EACrB,KAAK,aAAa;CACpB;CAEA,UAAkB,YAAoC;EACpD,IAAI,KAAK,oBAAoB,YAAY;EACzC,KAAK,kBAAkB;EACvB,KAAK,WAAW,KAAK;CACvB;CAIA,mBAA2B,UAAyB,SAAwB;EAC1E,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,4BACE,WACA,UACA;EACA,IAAI,CAAC,KAAK,SAAS;EACnB,KAAK,wBAAwB,IAAI,WAAW,QAAQ;CACtD;CAEA,wBAAgC,WAAmB;EACjD,IAAI,CAAC,KAAK,SAAS;EACnB,MAAM,WAAW,KAAK,wBAAwB,IAAI,SAAS;EAC3D,IAAI,aAAa,KAAA,GAAW;EAC5B,MAAM,UAAU,KAAK,cAAc,SAAS,CAAC,EAAE;EAC/C,IAAI,CAAC,WAAW,QAAQ,SAAS,aAAa;EAC9C,IACE,QAAQ,QAAQ,SAAS,cACzB,QAAQ,QAAQ,SAAS,cAEzB;EACF,KAAK,wBAAwB,OAAO,SAAS;EAC7C,KAAK,kBAAkB,UAAU,OAAO;CAC1C;CAEA,kBAA0B,UAAyB,SAAwB;EACzE,IAAI,CAAC,KAAK,WAAW,KAAK,mBAAmB,IAAI,QAAQ,EAAE,GAAG;EAC9D,KAAK,mBAAmB,IAAI,QAAQ,EAAE;EACtC,KAAU,QAAQ,OAAO;GAAE;GAAU;EAAQ,CAAC,CAAC,CAAC,YAAY;GAC1D,KAAK,mBAAmB,OAAO,QAAQ,EAAE;EAC3C,CAAC;CACH;AACF"}

package/dist/conversions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"conversions.d.ts","names":[],"sources":["../src/conversions.ts"],"mappings":";;;iBAUgB,iBACd,MAAM,UACL;iBA+Ba,kBACd,OAAO,YACN;iBAgBa,oBAAoB,OAAO;iBAI3B,uBAAuB,OAAO;iBAI9B,yBAAyB,OAAO,eAAe;iBAoB/C,uBACd,SAAS;EACP;EACA;EACA;EACA;EACA;EACA;EACA;EACA;IAAU;IAAc;;IAE1B,4BACC;iBAuEa,oBACd,SAAS,aACR"}
+{"version":3,"file":"conversions.d.ts","names":[],"sources":["../src/conversions.ts"],"mappings":";;;iBAUgB,iBACd,MAAM,UACL;iBAyCa,kBACd,OAAO,YACN;iBAgBa,oBAAoB,OAAO;iBAI3B,uBAAuB,OAAO;iBAI9B,yBAAyB,OAAO,eAAe;iBAoB/C,uBACd,SAAS;EACP;EACA;EACA;EACA;EACA;EACA;EACA;EACA;IAAU;IAAc;;IAE1B,4BACC;iBAuEa,oBACd,SAAS,aACR"}

package/dist/conversions.js

@@ -12,21 +12,28 @@ function a2aPartToContent(part) {
 	if (part.url !== void 0) {
 		if (isImageMediaType(part.mediaType)) return {
 			type: "image",
-			image: part.url
+			image: part.url,
+			...part.filename && { filename: part.filename }
 		};
 		return {
-			type: "text",
-			text: part.filename ? `[${part.filename}](${part.url})` : part.url
+			type: "file",
+			data: part.url,
+			mimeType: part.mediaType ?? "application/octet-stream",
+			sourceType: "url",
+			...part.filename && { filename: part.filename }
 		};
 	}
 	if (part.raw !== void 0) {
 		if (isImageMediaType(part.mediaType)) return {
 			type: "image",
-			image: `data:${part.mediaType};base64,${part.raw}`
+			image: `data:${part.mediaType};base64,${part.raw}`,
+			...part.filename && { filename: part.filename }
 		};
 		return {
-			type: "text",
-			text: `[File: ${part.filename ?? "download"}]`
+			type: "file",
+			data: part.raw,
+			mimeType: part.mediaType ?? "application/octet-stream",
+			...part.filename && { filename: part.filename }
 		};
 	}
 	if (part.data !== void 0) return {

package/dist/conversions.js.map

@@ -1 +1 @@
-{"version":3,"file":"conversions.js","names":[],"sources":["../src/conversions.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageStatus, ThreadAssistantMessage } from \"@assistant-ui/core\";\nimport { httpUrlPattern, parseDataUrl } from \"@assistant-ui/core/internal\";\nimport type { A2AMessage, A2APart, A2ATaskState } from \"./types\";\n\nfunction isImageMediaType(mediaType?: string): boolean {\n  return !!mediaType && mediaType.startsWith(\"image/\");\n}\n\nexport function a2aPartToContent(\n  part: A2APart,\n): ThreadAssistantMessage[\"content\"][number] {\n  if (part.text !== undefined) {\n    return { type: \"text\", text: part.text };\n  }\n  if (part.url !== undefined) {\n    if (isImageMediaType(part.mediaType)) {\n      return { type: \"image\", image: part.url };\n    }\n    return {\n      type: \"text\",\n      text: part.filename ? `[${part.filename}](${part.url})` : part.url,\n    };\n  }\n  if (part.raw !== undefined) {\n    if (isImageMediaType(part.mediaType)) {\n      return {\n        type: \"image\",\n        image: `data:${part.mediaType};base64,${part.raw}`,\n      };\n    }\n    return {\n      type: \"text\",\n      text: `[File: ${part.filename ?? \"download\"}]`,\n    };\n  }\n  if (part.data !== undefined) {\n    return { type: \"text\", text: JSON.stringify(part.data, null, 2) };\n  }\n  return { type: \"text\", text: \"\" };\n}\n\nexport function a2aPartsToContent(\n  parts: A2APart[],\n): ThreadAssistantMessage[\"content\"] {\n  return parts.map(a2aPartToContent);\n}\n\nconst TERMINAL_STATES = new Set<A2ATaskState>([\n  \"completed\",\n  \"failed\",\n  \"canceled\",\n  \"rejected\",\n]);\n\nconst INTERRUPTED_STATES = new Set<A2ATaskState>([\n  \"input_required\",\n  \"auth_required\",\n]);\n\nexport function isTerminalTaskState(state: A2ATaskState): boolean {\n  return TERMINAL_STATES.has(state);\n}\n\nexport function isInterruptedTaskState(state: A2ATaskState): boolean {\n  return INTERRUPTED_STATES.has(state);\n}\n\nexport function taskStateToMessageStatus(state: A2ATaskState): MessageStatus {\n  switch (state) {\n    case \"submitted\":\n    case \"working\":\n      return { type: \"running\" };\n    case \"completed\":\n      return { type: \"complete\", reason: \"stop\" };\n    case \"failed\":\n    case \"rejected\":\n      return { type: \"incomplete\", reason: \"error\" };\n    case \"canceled\":\n      return { type: \"incomplete\", reason: \"cancelled\" };\n    case \"input_required\":\n    case \"auth_required\":\n      return { type: \"requires-action\", reason: \"interrupt\" };\n    default:\n      return { type: \"running\" };\n  }\n}\n\nexport function contentPartsToA2AParts(\n  content: ReadonlyArray<{\n    type: string;\n    text?: string | undefined;\n    image?: string | undefined;\n    data?: unknown;\n    mimeType?: string | undefined;\n    filename?: string | undefined;\n    sourceType?: \"url\" | \"id\" | undefined;\n    audio?: { data: string; format: string } | undefined;\n  }>,\n  fallbackMimeType?: string,\n): A2APart[] {\n  return content\n    .map((part): A2APart | null => {\n      switch (part.type) {\n        case \"text\":\n          return { text: part.text ?? \"\" };\n        case \"image\": {\n          if (!part.image) return null;\n          const parsed = parseDataUrl(part.image);\n          if (parsed) {\n            return {\n              raw: parsed.data,\n              mediaType: parsed.mimeType,\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          return {\n            url: part.image,\n            ...(fallbackMimeType && { mediaType: fallbackMimeType }),\n            ...(part.filename && { filename: part.filename }),\n          };\n        }\n        case \"file\": {\n          if (typeof part.data !== \"string\" || !part.data) return null;\n          const declaredMimeType = part.mimeType || fallbackMimeType;\n          if (part.sourceType === \"url\" || httpUrlPattern.test(part.data)) {\n            return {\n              url: part.data,\n              ...(declaredMimeType && { mediaType: declaredMimeType }),\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          const parsed = parseDataUrl(part.data);\n          if (parsed) {\n            return {\n              raw: parsed.data,\n              mediaType: parsed.mimeType,\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          if (/^data:/i.test(part.data)) {\n            return {\n              url: part.data,\n              ...(declaredMimeType && { mediaType: declaredMimeType }),\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          return {\n            raw: part.data,\n            ...(declaredMimeType && { mediaType: declaredMimeType }),\n            ...(part.filename && { filename: part.filename }),\n          };\n        }\n        case \"audio\": {\n          if (!part.audio) return null;\n          return {\n            raw: parseDataUrl(part.audio.data)?.data ?? part.audio.data,\n            mediaType: `audio/${part.audio.format}`,\n          };\n        }\n        case \"data\": {\n          if (part.data === undefined) return null;\n          return { data: part.data };\n        }\n        default:\n          return null;\n      }\n    })\n    .filter((p): p is A2APart => p !== null);\n}\n\nexport function a2aMessageToContent(\n  message: A2AMessage,\n): ThreadAssistantMessage[\"content\"] {\n  return a2aPartsToContent(message.parts);\n}\n"],"mappings":";;;AAMA,SAAS,iBAAiB,WAA6B;CACrD,OAAO,CAAC,CAAC,aAAa,UAAU,WAAW,QAAQ;AACrD;AAEA,SAAgB,iBACd,MAC2C;CAC3C,IAAI,KAAK,SAAS,KAAA,GAChB,OAAO;EAAE,MAAM;EAAQ,MAAM,KAAK;CAAK;CAEzC,IAAI,KAAK,QAAQ,KAAA,GAAW;EAC1B,IAAI,iBAAiB,KAAK,SAAS,GACjC,OAAO;GAAE,MAAM;GAAS,OAAO,KAAK;EAAI;EAE1C,OAAO;GACL,MAAM;GACN,MAAM,KAAK,WAAW,IAAI,KAAK,SAAS,IAAI,KAAK,IAAI,KAAK,KAAK;EACjE;CACF;CACA,IAAI,KAAK,QAAQ,KAAA,GAAW;EAC1B,IAAI,iBAAiB,KAAK,SAAS,GACjC,OAAO;GACL,MAAM;GACN,OAAO,QAAQ,KAAK,UAAU,UAAU,KAAK;EAC/C;EAEF,OAAO;GACL,MAAM;GACN,MAAM,UAAU,KAAK,YAAY,WAAW;EAC9C;CACF;CACA,IAAI,KAAK,SAAS,KAAA,GAChB,OAAO;EAAE,MAAM;EAAQ,MAAM,KAAK,UAAU,KAAK,MAAM,MAAM,CAAC;CAAE;CAElE,OAAO;EAAE,MAAM;EAAQ,MAAM;CAAG;AAClC;AAEA,SAAgB,kBACd,OACmC;CACnC,OAAO,MAAM,IAAI,gBAAgB;AACnC;AAEA,MAAM,kCAAkB,IAAI,IAAkB;CAC5C;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,qCAAqB,IAAI,IAAkB,CAC/C,kBACA,eACF,CAAC;AAED,SAAgB,oBAAoB,OAA8B;CAChE,OAAO,gBAAgB,IAAI,KAAK;AAClC;AAEA,SAAgB,uBAAuB,OAA8B;CACnE,OAAO,mBAAmB,IAAI,KAAK;AACrC;AAEA,SAAgB,yBAAyB,OAAoC;CAC3E,QAAQ,OAAR;EACE,KAAK;EACL,KAAK,WACH,OAAO,EAAE,MAAM,UAAU;EAC3B,KAAK,aACH,OAAO;GAAE,MAAM;GAAY,QAAQ;EAAO;EAC5C,KAAK;EACL,KAAK,YACH,OAAO;GAAE,MAAM;GAAc,QAAQ;EAAQ;EAC/C,KAAK,YACH,OAAO;GAAE,MAAM;GAAc,QAAQ;EAAY;EACnD,KAAK;EACL,KAAK,iBACH,OAAO;GAAE,MAAM;GAAmB,QAAQ;EAAY;EACxD,SACE,OAAO,EAAE,MAAM,UAAU;CAC7B;AACF;AAEA,SAAgB,uBACd,SAUA,kBACW;CACX,OAAO,QACJ,KAAK,SAAyB;EAC7B,QAAQ,KAAK,MAAb;GACE,KAAK,QACH,OAAO,EAAE,MAAM,KAAK,QAAQ,GAAG;GACjC,KAAK,SAAS;IACZ,IAAI,CAAC,KAAK,OAAO,OAAO;IACxB,MAAM,SAAS,aAAa,KAAK,KAAK;IACtC,IAAI,QACF,OAAO;KACL,KAAK,OAAO;KACZ,WAAW,OAAO;KAClB,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;GACF;GACA,KAAK,QAAQ;IACX,IAAI,OAAO,KAAK,SAAS,YAAY,CAAC,KAAK,MAAM,OAAO;IACxD,MAAM,mBAAmB,KAAK,YAAY;IAC1C,IAAI,KAAK,eAAe,SAAS,eAAe,KAAK,KAAK,IAAI,GAC5D,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,MAAM,SAAS,aAAa,KAAK,IAAI;IACrC,IAAI,QACF,OAAO;KACL,KAAK,OAAO;KACZ,WAAW,OAAO;KAClB,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,IAAI,UAAU,KAAK,KAAK,IAAI,GAC1B,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;GACF;GACA,KAAK;IACH,IAAI,CAAC,KAAK,OAAO,OAAO;IACxB,OAAO;KACL,KAAK,aAAa,KAAK,MAAM,IAAI,CAAC,EAAE,QAAQ,KAAK,MAAM;KACvD,WAAW,SAAS,KAAK,MAAM;IACjC;GAEF,KAAK;IACH,IAAI,KAAK,SAAS,KAAA,GAAW,OAAO;IACpC,OAAO,EAAE,MAAM,KAAK,KAAK;GAE3B,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,QAAQ,MAAoB,MAAM,IAAI;AAC3C;AAEA,SAAgB,oBACd,SACmC;CACnC,OAAO,kBAAkB,QAAQ,KAAK;AACxC"}
+{"version":3,"file":"conversions.js","names":[],"sources":["../src/conversions.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageStatus, ThreadAssistantMessage } from \"@assistant-ui/core\";\nimport { httpUrlPattern, parseDataUrl } from \"@assistant-ui/core/internal\";\nimport type { A2AMessage, A2APart, A2ATaskState } from \"./types\";\n\nfunction isImageMediaType(mediaType?: string): boolean {\n  return !!mediaType && mediaType.startsWith(\"image/\");\n}\n\nexport function a2aPartToContent(\n  part: A2APart,\n): ThreadAssistantMessage[\"content\"][number] {\n  if (part.text !== undefined) {\n    return { type: \"text\", text: part.text };\n  }\n  if (part.url !== undefined) {\n    if (isImageMediaType(part.mediaType)) {\n      return {\n        type: \"image\",\n        image: part.url,\n        ...(part.filename && { filename: part.filename }),\n      };\n    }\n    return {\n      type: \"file\",\n      data: part.url,\n      mimeType: part.mediaType ?? \"application/octet-stream\",\n      sourceType: \"url\",\n      ...(part.filename && { filename: part.filename }),\n    };\n  }\n  if (part.raw !== undefined) {\n    if (isImageMediaType(part.mediaType)) {\n      return {\n        type: \"image\",\n        image: `data:${part.mediaType};base64,${part.raw}`,\n        ...(part.filename && { filename: part.filename }),\n      };\n    }\n    return {\n      type: \"file\",\n      data: part.raw,\n      mimeType: part.mediaType ?? \"application/octet-stream\",\n      ...(part.filename && { filename: part.filename }),\n    };\n  }\n  if (part.data !== undefined) {\n    return { type: \"text\", text: JSON.stringify(part.data, null, 2) };\n  }\n  return { type: \"text\", text: \"\" };\n}\n\nexport function a2aPartsToContent(\n  parts: A2APart[],\n): ThreadAssistantMessage[\"content\"] {\n  return parts.map(a2aPartToContent);\n}\n\nconst TERMINAL_STATES = new Set<A2ATaskState>([\n  \"completed\",\n  \"failed\",\n  \"canceled\",\n  \"rejected\",\n]);\n\nconst INTERRUPTED_STATES = new Set<A2ATaskState>([\n  \"input_required\",\n  \"auth_required\",\n]);\n\nexport function isTerminalTaskState(state: A2ATaskState): boolean {\n  return TERMINAL_STATES.has(state);\n}\n\nexport function isInterruptedTaskState(state: A2ATaskState): boolean {\n  return INTERRUPTED_STATES.has(state);\n}\n\nexport function taskStateToMessageStatus(state: A2ATaskState): MessageStatus {\n  switch (state) {\n    case \"submitted\":\n    case \"working\":\n      return { type: \"running\" };\n    case \"completed\":\n      return { type: \"complete\", reason: \"stop\" };\n    case \"failed\":\n    case \"rejected\":\n      return { type: \"incomplete\", reason: \"error\" };\n    case \"canceled\":\n      return { type: \"incomplete\", reason: \"cancelled\" };\n    case \"input_required\":\n    case \"auth_required\":\n      return { type: \"requires-action\", reason: \"interrupt\" };\n    default:\n      return { type: \"running\" };\n  }\n}\n\nexport function contentPartsToA2AParts(\n  content: ReadonlyArray<{\n    type: string;\n    text?: string | undefined;\n    image?: string | undefined;\n    data?: unknown;\n    mimeType?: string | undefined;\n    filename?: string | undefined;\n    sourceType?: \"url\" | \"id\" | undefined;\n    audio?: { data: string; format: string } | undefined;\n  }>,\n  fallbackMimeType?: string,\n): A2APart[] {\n  return content\n    .map((part): A2APart | null => {\n      switch (part.type) {\n        case \"text\":\n          return { text: part.text ?? \"\" };\n        case \"image\": {\n          if (!part.image) return null;\n          const parsed = parseDataUrl(part.image);\n          if (parsed) {\n            return {\n              raw: parsed.data,\n              mediaType: parsed.mimeType,\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          return {\n            url: part.image,\n            ...(fallbackMimeType && { mediaType: fallbackMimeType }),\n            ...(part.filename && { filename: part.filename }),\n          };\n        }\n        case \"file\": {\n          if (typeof part.data !== \"string\" || !part.data) return null;\n          const declaredMimeType = part.mimeType || fallbackMimeType;\n          if (part.sourceType === \"url\" || httpUrlPattern.test(part.data)) {\n            return {\n              url: part.data,\n              ...(declaredMimeType && { mediaType: declaredMimeType }),\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          const parsed = parseDataUrl(part.data);\n          if (parsed) {\n            return {\n              raw: parsed.data,\n              mediaType: parsed.mimeType,\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          if (/^data:/i.test(part.data)) {\n            return {\n              url: part.data,\n              ...(declaredMimeType && { mediaType: declaredMimeType }),\n              ...(part.filename && { filename: part.filename }),\n            };\n          }\n          return {\n            raw: part.data,\n            ...(declaredMimeType && { mediaType: declaredMimeType }),\n            ...(part.filename && { filename: part.filename }),\n          };\n        }\n        case \"audio\": {\n          if (!part.audio) return null;\n          return {\n            raw: parseDataUrl(part.audio.data)?.data ?? part.audio.data,\n            mediaType: `audio/${part.audio.format}`,\n          };\n        }\n        case \"data\": {\n          if (part.data === undefined) return null;\n          return { data: part.data };\n        }\n        default:\n          return null;\n      }\n    })\n    .filter((p): p is A2APart => p !== null);\n}\n\nexport function a2aMessageToContent(\n  message: A2AMessage,\n): ThreadAssistantMessage[\"content\"] {\n  return a2aPartsToContent(message.parts);\n}\n"],"mappings":";;;AAMA,SAAS,iBAAiB,WAA6B;CACrD,OAAO,CAAC,CAAC,aAAa,UAAU,WAAW,QAAQ;AACrD;AAEA,SAAgB,iBACd,MAC2C;CAC3C,IAAI,KAAK,SAAS,KAAA,GAChB,OAAO;EAAE,MAAM;EAAQ,MAAM,KAAK;CAAK;CAEzC,IAAI,KAAK,QAAQ,KAAA,GAAW;EAC1B,IAAI,iBAAiB,KAAK,SAAS,GACjC,OAAO;GACL,MAAM;GACN,OAAO,KAAK;GACZ,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;EACjD;EAEF,OAAO;GACL,MAAM;GACN,MAAM,KAAK;GACX,UAAU,KAAK,aAAa;GAC5B,YAAY;GACZ,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;EACjD;CACF;CACA,IAAI,KAAK,QAAQ,KAAA,GAAW;EAC1B,IAAI,iBAAiB,KAAK,SAAS,GACjC,OAAO;GACL,MAAM;GACN,OAAO,QAAQ,KAAK,UAAU,UAAU,KAAK;GAC7C,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;EACjD;EAEF,OAAO;GACL,MAAM;GACN,MAAM,KAAK;GACX,UAAU,KAAK,aAAa;GAC5B,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;EACjD;CACF;CACA,IAAI,KAAK,SAAS,KAAA,GAChB,OAAO;EAAE,MAAM;EAAQ,MAAM,KAAK,UAAU,KAAK,MAAM,MAAM,CAAC;CAAE;CAElE,OAAO;EAAE,MAAM;EAAQ,MAAM;CAAG;AAClC;AAEA,SAAgB,kBACd,OACmC;CACnC,OAAO,MAAM,IAAI,gBAAgB;AACnC;AAEA,MAAM,kCAAkB,IAAI,IAAkB;CAC5C;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,qCAAqB,IAAI,IAAkB,CAC/C,kBACA,eACF,CAAC;AAED,SAAgB,oBAAoB,OAA8B;CAChE,OAAO,gBAAgB,IAAI,KAAK;AAClC;AAEA,SAAgB,uBAAuB,OAA8B;CACnE,OAAO,mBAAmB,IAAI,KAAK;AACrC;AAEA,SAAgB,yBAAyB,OAAoC;CAC3E,QAAQ,OAAR;EACE,KAAK;EACL,KAAK,WACH,OAAO,EAAE,MAAM,UAAU;EAC3B,KAAK,aACH,OAAO;GAAE,MAAM;GAAY,QAAQ;EAAO;EAC5C,KAAK;EACL,KAAK,YACH,OAAO;GAAE,MAAM;GAAc,QAAQ;EAAQ;EAC/C,KAAK,YACH,OAAO;GAAE,MAAM;GAAc,QAAQ;EAAY;EACnD,KAAK;EACL,KAAK,iBACH,OAAO;GAAE,MAAM;GAAmB,QAAQ;EAAY;EACxD,SACE,OAAO,EAAE,MAAM,UAAU;CAC7B;AACF;AAEA,SAAgB,uBACd,SAUA,kBACW;CACX,OAAO,QACJ,KAAK,SAAyB;EAC7B,QAAQ,KAAK,MAAb;GACE,KAAK,QACH,OAAO,EAAE,MAAM,KAAK,QAAQ,GAAG;GACjC,KAAK,SAAS;IACZ,IAAI,CAAC,KAAK,OAAO,OAAO;IACxB,MAAM,SAAS,aAAa,KAAK,KAAK;IACtC,IAAI,QACF,OAAO;KACL,KAAK,OAAO;KACZ,WAAW,OAAO;KAClB,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;GACF;GACA,KAAK,QAAQ;IACX,IAAI,OAAO,KAAK,SAAS,YAAY,CAAC,KAAK,MAAM,OAAO;IACxD,MAAM,mBAAmB,KAAK,YAAY;IAC1C,IAAI,KAAK,eAAe,SAAS,eAAe,KAAK,KAAK,IAAI,GAC5D,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,MAAM,SAAS,aAAa,KAAK,IAAI;IACrC,IAAI,QACF,OAAO;KACL,KAAK,OAAO;KACZ,WAAW,OAAO;KAClB,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,IAAI,UAAU,KAAK,KAAK,IAAI,GAC1B,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;IAEF,OAAO;KACL,KAAK,KAAK;KACV,GAAI,oBAAoB,EAAE,WAAW,iBAAiB;KACtD,GAAI,KAAK,YAAY,EAAE,UAAU,KAAK,SAAS;IACjD;GACF;GACA,KAAK;IACH,IAAI,CAAC,KAAK,OAAO,OAAO;IACxB,OAAO;KACL,KAAK,aAAa,KAAK,MAAM,IAAI,CAAC,EAAE,QAAQ,KAAK,MAAM;KACvD,WAAW,SAAS,KAAK,MAAM;IACjC;GAEF,KAAK;IACH,IAAI,KAAK,SAAS,KAAA,GAAW,OAAO;IACpC,OAAO,EAAE,MAAM,KAAK,KAAK;GAE3B,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,QAAQ,MAAoB,MAAM,IAAI;AAC3C;AAEA,SAAgB,oBACd,SACmC;CACnC,OAAO,kBAAkB,QAAQ,KAAK;AACxC"}

package/dist/useA2ARuntime.js

@@ -85,7 +85,7 @@ function useA2ARuntime(options) {
 		threadList
 	]);
 	const shared = useExternalStoreSharedOptions(options);
-	const runtime = useExternalStoreRuntime(useMemo(() => {
+	const store = useMemo(() => {
 		return {
 			...shared,
 			isLoading: core.isLoading,
@@ -109,7 +109,8 @@ function useA2ARuntime(options) {
 		core,
 		_version,
 		shared
-	]));
+	]);
+	const runtime = useExternalStoreRuntime(store);
 	useEffect(() => {
 		core.attachRuntime(runtime);
 		return () => {

package/dist/useA2ARuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useA2ARuntime.js","names":[],"sources":["../src/useA2ARuntime.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 {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { A2AClient, type A2AClientOptions } from \"./A2AClient\";\nimport { A2AThreadRuntimeCore } from \"./A2AThreadRuntimeCore\";\nimport { a2aExtras } from \"./a2aExtras\";\nimport type { UseA2ARuntimeOptions } from \"./types\";\n\ntype ManagedA2AClientOptions = Omit<A2AClientOptions, \"headers\">;\n\nconst serializeManagedClientOptions = (\n  options: ManagedA2AClientOptions,\n): string => {\n  const fetchOptions = Object.fromEntries(\n    Object.entries(options.fetchOptions ?? {}).sort(([a], [b]) =>\n      a.localeCompare(b),\n    ),\n  );\n\n  return JSON.stringify({\n    ...options,\n    fetchOptions,\n  });\n};\n\nexport function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const runtimeAdapters = useRuntimeAdapters();\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  const headersRef = useRef(options.headers);\n  headersRef.current = options.headers;\n  const resolveHeaders = useCallback(() => {\n    const headers = headersRef.current;\n    return typeof headers === \"function\" ? headers() : (headers ?? {});\n  }, []);\n\n  const managedClientOptionsKey = options.client\n    ? null\n    : options.baseUrl\n      ? serializeManagedClientOptions({\n          baseUrl: options.baseUrl,\n          basePath: options.basePath,\n          tenant: options.tenant,\n          extensions: options.extensions,\n          fetchOptions: options.fetchOptions,\n        })\n      : null;\n\n  const client = useMemo(() => {\n    if (options.client) return options.client;\n    if (!managedClientOptionsKey) {\n      throw new Error(\"useA2ARuntime requires either `client` or `baseUrl`\");\n    }\n\n    return new A2AClient({\n      ...(JSON.parse(managedClientOptionsKey) as ManagedA2AClientOptions),\n      headers: resolveHeaders,\n    });\n  }, [managedClientOptionsKey, options.client, resolveHeaders]);\n\n  const core = useMemo(\n    () =>\n      new A2AThreadRuntimeCore({\n        client,\n        notifyUpdate,\n      }),\n    [client, notifyUpdate],\n  );\n\n  core.updateOptions({\n    client,\n    contextId: options.contextId,\n    configuration: options.configuration,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(options.onArtifactComplete && {\n      onArtifactComplete: options.onArtifactComplete,\n    }),\n    ...(historyAdapter && { history: historyAdapter }),\n  });\n\n  // Thread list\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          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  // Adapters\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  // Build store adapter\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(() => {\n    void _version;\n\n    return {\n      ...shared,\n      isLoading: core.isLoading,\n      messageRepository: core.getMessageRepository(),\n      isRunning: core.isRunning(),\n      extras: a2aExtras.provide({\n        task: core.getTask(),\n        artifacts: core.getArtifacts(),\n        agentCard: core.getAgentCard(),\n      }),\n      onNew: (message: AppendMessage) => core.append(message),\n      onEdit: (message: AppendMessage) => core.edit(message),\n      onReload: (parentId: string | null) => core.reload(parentId),\n      onCancel: () => core.cancel(),\n      setMessages: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      onImport: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      adapters: adapterAdapters,\n    } satisfies ExternalStoreAdapter<ThreadMessage>;\n  }, [adapterAdapters, core, _version, shared]);\n\n  const runtime = useExternalStoreRuntime(store);\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":";;;;;;;AAqBA,MAAM,iCACJ,YACW;CACX,MAAM,eAAe,OAAO,YAC1B,OAAO,QAAQ,QAAQ,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OACrD,EAAE,cAAc,CAAC,CACnB,CACF;CAEA,OAAO,KAAK,UAAU;EACpB,GAAG;EACH;CACF,CAAC;AACH;AAEA,SAAgB,cAAc,SAAiD;CAC7E,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,MAAM,aAAa,OAAO,QAAQ,OAAO;CACzC,WAAW,UAAU,QAAQ;CAC7B,MAAM,iBAAiB,kBAAkB;EACvC,MAAM,UAAU,WAAW;EAC3B,OAAO,OAAO,YAAY,aAAa,QAAQ,IAAK,WAAW,CAAC;CAClE,GAAG,CAAC,CAAC;CAEL,MAAM,0BAA0B,QAAQ,SACpC,OACA,QAAQ,UACN,8BAA8B;EAC5B,SAAS,QAAQ;EACjB,UAAU,QAAQ;EAClB,QAAQ,QAAQ;EAChB,YAAY,QAAQ;EACpB,cAAc,QAAQ;CACxB,CAAC,IACD;CAEN,MAAM,SAAS,cAAc;EAC3B,IAAI,QAAQ,QAAQ,OAAO,QAAQ;EACnC,IAAI,CAAC,yBACH,MAAM,IAAI,MAAM,qDAAqD;EAGvE,OAAO,IAAI,UAAU;GACnB,GAAI,KAAK,MAAM,uBAAuB;GACtC,SAAS;EACX,CAAC;CACH,GAAG;EAAC;EAAyB,QAAQ;EAAQ;CAAc,CAAC;CAE5D,MAAM,OAAO,cAET,IAAI,qBAAqB;EACvB;EACA;CACF,CAAC,GACH,CAAC,QAAQ,YAAY,CACvB;CAEA,KAAK,cAAc;EACjB;EACA,WAAW,QAAQ;EACnB,eAAe,QAAQ;EACvB,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,QAAQ,sBAAsB,EAChC,oBAAoB,QAAQ,mBAC9B;EACA,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD,CAAC;CAGD,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;GAC5C,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAG5B,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;CAGA,MAAM,SAAS,8BAA8B,OAAO;CA0BpD,MAAM,UAAU,wBAzBF,cAAc;EAG1B,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,mBAAmB,KAAK,qBAAqB;GAC7C,WAAW,KAAK,UAAU;GAC1B,QAAQ,UAAU,QAAQ;IACxB,MAAM,KAAK,QAAQ;IACnB,WAAW,KAAK,aAAa;IAC7B,WAAW,KAAK,aAAa;GAC/B,CAAC;GACD,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,aAA4B,KAAK,OAAO,QAAQ;GAC3D,gBAAgB,KAAK,OAAO;GAC5B,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,UAAU;EACZ;CACF,GAAG;EAAC;EAAiB;EAAM;EAAU;CAAM,CAEC,CAAC;CAE7C,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":"useA2ARuntime.js","names":[],"sources":["../src/useA2ARuntime.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 {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { A2AClient, type A2AClientOptions } from \"./A2AClient\";\nimport { A2AThreadRuntimeCore } from \"./A2AThreadRuntimeCore\";\nimport { a2aExtras } from \"./a2aExtras\";\nimport type { UseA2ARuntimeOptions } from \"./types\";\n\ntype ManagedA2AClientOptions = Omit<A2AClientOptions, \"headers\">;\n\nconst serializeManagedClientOptions = (\n  options: ManagedA2AClientOptions,\n): string => {\n  const fetchOptions = Object.fromEntries(\n    Object.entries(options.fetchOptions ?? {}).sort(([a], [b]) =>\n      a.localeCompare(b),\n    ),\n  );\n\n  return JSON.stringify({\n    ...options,\n    fetchOptions,\n  });\n};\n\nexport function useA2ARuntime(options: UseA2ARuntimeOptions): AssistantRuntime {\n  const [_version, setVersion] = useState(0);\n  const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);\n  const runtimeAdapters = useRuntimeAdapters();\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  const headersRef = useRef(options.headers);\n  headersRef.current = options.headers;\n  const resolveHeaders = useCallback(() => {\n    const headers = headersRef.current;\n    return typeof headers === \"function\" ? headers() : (headers ?? {});\n  }, []);\n\n  const managedClientOptionsKey = options.client\n    ? null\n    : options.baseUrl\n      ? serializeManagedClientOptions({\n          baseUrl: options.baseUrl,\n          basePath: options.basePath,\n          tenant: options.tenant,\n          extensions: options.extensions,\n          fetchOptions: options.fetchOptions,\n        })\n      : null;\n\n  const client = useMemo(() => {\n    if (options.client) return options.client;\n    if (!managedClientOptionsKey) {\n      throw new Error(\"useA2ARuntime requires either `client` or `baseUrl`\");\n    }\n\n    return new A2AClient({\n      ...(JSON.parse(managedClientOptionsKey) as ManagedA2AClientOptions),\n      headers: resolveHeaders,\n    });\n  }, [managedClientOptionsKey, options.client, resolveHeaders]);\n\n  const core = useMemo(\n    () =>\n      new A2AThreadRuntimeCore({\n        client,\n        notifyUpdate,\n      }),\n    [client, notifyUpdate],\n  );\n\n  core.updateOptions({\n    client,\n    contextId: options.contextId,\n    configuration: options.configuration,\n    ...(options.onError && { onError: options.onError }),\n    ...(options.onCancel && { onCancel: options.onCancel }),\n    ...(options.onArtifactComplete && {\n      onArtifactComplete: options.onArtifactComplete,\n    }),\n    ...(historyAdapter && { history: historyAdapter }),\n  });\n\n  // Thread list\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          }\n        : undefined,\n    };\n  }, [threadListAdapter, core]);\n\n  // Adapters\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  // Build store adapter\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(() => {\n    void _version;\n\n    return {\n      ...shared,\n      isLoading: core.isLoading,\n      messageRepository: core.getMessageRepository(),\n      isRunning: core.isRunning(),\n      extras: a2aExtras.provide({\n        task: core.getTask(),\n        artifacts: core.getArtifacts(),\n        agentCard: core.getAgentCard(),\n      }),\n      onNew: (message: AppendMessage) => core.append(message),\n      onEdit: (message: AppendMessage) => core.edit(message),\n      onReload: (parentId: string | null) => core.reload(parentId),\n      onCancel: () => core.cancel(),\n      setMessages: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      onImport: (messages: readonly ThreadMessage[]) =>\n        core.applyExternalMessages(messages),\n      adapters: adapterAdapters,\n    } satisfies ExternalStoreAdapter<ThreadMessage>;\n  }, [adapterAdapters, core, _version, shared]);\n\n  const runtime = useExternalStoreRuntime(store);\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":";;;;;;;AAqBA,MAAM,iCACJ,YACW;CACX,MAAM,eAAe,OAAO,YAC1B,OAAO,QAAQ,QAAQ,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OACrD,EAAE,cAAc,CAAC,CACnB,CACF;CAEA,OAAO,KAAK,UAAU;EACpB,GAAG;EACH;CACF,CAAC;AACH;AAEA,SAAgB,cAAc,SAAiD;CAC7E,MAAM,CAAC,UAAU,cAAc,SAAS,CAAC;CACzC,MAAM,eAAe,kBAAkB,YAAY,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;CACnE,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,MAAM,aAAa,OAAO,QAAQ,OAAO;CACzC,WAAW,UAAU,QAAQ;CAC7B,MAAM,iBAAiB,kBAAkB;EACvC,MAAM,UAAU,WAAW;EAC3B,OAAO,OAAO,YAAY,aAAa,QAAQ,IAAK,WAAW,CAAC;CAClE,GAAG,CAAC,CAAC;CAEL,MAAM,0BAA0B,QAAQ,SACpC,OACA,QAAQ,UACN,8BAA8B;EAC5B,SAAS,QAAQ;EACjB,UAAU,QAAQ;EAClB,QAAQ,QAAQ;EAChB,YAAY,QAAQ;EACpB,cAAc,QAAQ;CACxB,CAAC,IACD;CAEN,MAAM,SAAS,cAAc;EAC3B,IAAI,QAAQ,QAAQ,OAAO,QAAQ;EACnC,IAAI,CAAC,yBACH,MAAM,IAAI,MAAM,qDAAqD;EAGvE,OAAO,IAAI,UAAU;GACnB,GAAI,KAAK,MAAM,uBAAuB;GACtC,SAAS;EACX,CAAC;CACH,GAAG;EAAC;EAAyB,QAAQ;EAAQ;CAAc,CAAC;CAE5D,MAAM,OAAO,cAET,IAAI,qBAAqB;EACvB;EACA;CACF,CAAC,GACH,CAAC,QAAQ,YAAY,CACvB;CAEA,KAAK,cAAc;EACjB;EACA,WAAW,QAAQ;EACnB,eAAe,QAAQ;EACvB,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,QAAQ,sBAAsB,EAChC,oBAAoB,QAAQ,mBAC9B;EACA,GAAI,kBAAkB,EAAE,SAAS,eAAe;CAClD,CAAC;CAGD,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;GAC5C,IACA,KAAA;EACN;CACF,GAAG,CAAC,mBAAmB,IAAI,CAAC;CAG5B,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;CAGA,MAAM,SAAS,8BAA8B,OAAO;CACpD,MAAM,QAAQ,cAAc;EAG1B,OAAO;GACL,GAAG;GACH,WAAW,KAAK;GAChB,mBAAmB,KAAK,qBAAqB;GAC7C,WAAW,KAAK,UAAU;GAC1B,QAAQ,UAAU,QAAQ;IACxB,MAAM,KAAK,QAAQ;IACnB,WAAW,KAAK,aAAa;IAC7B,WAAW,KAAK,aAAa;GAC/B,CAAC;GACD,QAAQ,YAA2B,KAAK,OAAO,OAAO;GACtD,SAAS,YAA2B,KAAK,KAAK,OAAO;GACrD,WAAW,aAA4B,KAAK,OAAO,QAAQ;GAC3D,gBAAgB,KAAK,OAAO;GAC5B,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,UAAU;EACZ;CACF,GAAG;EAAC;EAAiB;EAAM;EAAU;CAAM,CAAC;CAE5C,MAAM,UAAU,wBAAwB,KAAK;CAE7C,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-a2a",
-  "version": "0.2.26",
+  "version": "0.2.28",
   "description": "A2A (Agent-to-Agent) v1.0 protocol adapter for assistant-ui",
   "keywords": [
     "a2a",
@@ -29,9 +29,9 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.4",
-    "@assistant-ui/store": "^0.3.3",
-    "assistant-stream": "^0.3.32"
+    "@assistant-ui/core": "^0.3.10",
+    "@assistant-ui/store": "^0.3.8",
+    "assistant-stream": "^0.3.36"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -45,12 +45,12 @@
   "devDependencies": {
     "@testing-library/dom": "^10.4.1",
     "@testing-library/react": "^16.3.2",
-    "@types/react": "^19.2.17",
-    "jsdom": "^29.1.1",
+    "@types/react": "^19.2.18",
+    "jsdom": "^30.0.1",
     "react": "^19.2.8",
     "react-dom": "^19.2.8",
     "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.20"
+    "@assistant-ui/x-buildutils": "0.0.22"
   },
   "publishConfig": {
     "access": "public",

package/src/A2AThreadRuntimeCore.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
 import { A2AThreadRuntimeCore } from "./A2AThreadRuntimeCore";
 import type { A2AClient } from "./A2AClient";
 import type { A2AMessage, A2AStreamEvent, A2ATask } from "./types";
@@ -135,6 +135,10 @@ describe("A2AThreadRuntimeCore", () => {
     notifyUpdate = vi.fn();
   });
 
+  afterEach(() => {
+    vi.restoreAllMocks();
+  });
+
   function createCore(
     clientOverrides: Partial<A2AClient> = {},
     coreOverrides: Record<string, unknown> = {},
@@ -700,6 +704,53 @@ describe("A2AThreadRuntimeCore", () => {
       expect(onArtifactComplete.mock.calls[0]![0].artifactId).toBe("a1");
     });
 
+    it.each(["throws", "rejects"] as const)(
+      "continues streaming when onArtifactComplete %s",
+      async (failureMode) => {
+        const callbackError = new Error("artifact callback failed");
+        const consoleError = vi
+          .spyOn(console, "error")
+          .mockImplementation(() => {});
+        const onError = vi.fn();
+        const core = createCore(
+          {
+            streamMessage: vi.fn().mockImplementation(async function* () {
+              yield artifactUpdateEvent("a1", [{ text: "code" }], {
+                lastChunk: true,
+              });
+              yield statusUpdateEvent("completed", "Done");
+            }),
+          },
+          {
+            onArtifactComplete: () => {
+              if (failureMode === "throws") throw callbackError;
+              return Promise.reject(callbackError);
+            },
+            onError,
+          },
+        );
+
+        await expect(
+          core.append(createUserAppendMessage("Go")),
+        ).resolves.toBeUndefined();
+
+        expect(onError).not.toHaveBeenCalled();
+        expect(core.getMessages()[1]!.content).toEqual([
+          { type: "text", text: "Done" },
+        ]);
+        expect(core.getMessages()[1]!.status).toEqual({
+          type: "complete",
+          reason: "stop",
+        });
+        await vi.waitFor(() => {
+          expect(consoleError).toHaveBeenCalledWith(
+            "[react-a2a] onArtifactComplete callback threw an error",
+            callbackError,
+          );
+        });
+      },
+    );
+
     it("resets artifacts on new run", async () => {
       let runCount = 0;
 
@@ -890,6 +941,61 @@ describe("A2AThreadRuntimeCore", () => {
 
       expect(cancelTask).not.toHaveBeenCalled();
     });
+
+    it.each(["throws", "rejects"] as const)(
+      "isolates onCancel callbacks that %s",
+      async (failureMode) => {
+        const callbackError = new Error("cancel callback failed");
+        const consoleError = vi
+          .spyOn(console, "error")
+          .mockImplementation(() => {});
+        let signalStreamStarted!: () => void;
+        const streamStarted = new Promise<void>((resolve) => {
+          signalStreamStarted = resolve;
+        });
+        const core = createCore(
+          {
+            streamMessage: vi.fn().mockImplementation(async function* (
+              _message,
+              _configuration,
+              _metadata,
+              signal: AbortSignal,
+            ) {
+              signalStreamStarted();
+              await new Promise<void>((resolve) => {
+                if (signal.aborted) resolve();
+                else
+                  signal.addEventListener("abort", () => resolve(), {
+                    once: true,
+                  });
+              });
+            }),
+          },
+          {
+            onCancel: () => {
+              if (failureMode === "throws") throw callbackError;
+              return Promise.reject(callbackError);
+            },
+          },
+        );
+
+        const appendPromise = core.append(createUserAppendMessage("Go"));
+        await streamStarted;
+        await expect(core.cancel()).resolves.toBeUndefined();
+        await expect(appendPromise).resolves.toBeUndefined();
+
+        expect(core.getMessages()[1]!.status).toEqual({
+          type: "incomplete",
+          reason: "cancelled",
+        });
+        await vi.waitFor(() => {
+          expect(consoleError).toHaveBeenCalledWith(
+            "[react-a2a] onCancel callback threw an error",
+            callbackError,
+          );
+        });
+      },
+    );
   });
 
   // --- Error handling ---
@@ -927,6 +1033,50 @@ describe("A2AThreadRuntimeCore", () => {
       });
     });
 
+    it.each(["throws", "rejects"] as const)(
+      "preserves the stream error when onError %s",
+      async (failureMode) => {
+        const streamError = new Error("Network error");
+        const callbackError = new Error("error callback failed");
+        const consoleError = vi
+          .spyOn(console, "error")
+          .mockImplementation(() => {});
+        const core = createCore(
+          {
+            streamMessage: vi.fn().mockImplementation(() => ({
+              async next() {
+                throw streamError;
+              },
+              [Symbol.asyncIterator]() {
+                return this;
+              },
+            })),
+          },
+          {
+            onError: () => {
+              if (failureMode === "throws") throw callbackError;
+              return Promise.reject(callbackError);
+            },
+          },
+        );
+
+        await expect(core.append(createUserAppendMessage("Go"))).rejects.toBe(
+          streamError,
+        );
+
+        expect(core.getMessages()[1]!.status).toEqual({
+          type: "incomplete",
+          reason: "error",
+        });
+        await vi.waitFor(() => {
+          expect(consoleError).toHaveBeenCalledWith(
+            "[react-a2a] onError callback threw an error",
+            callbackError,
+          );
+        });
+      },
+    );
+
     it("marks complete when stream ends without terminal status", async () => {
       const core = createCore({
         streamMessage: vi.fn().mockImplementation(async function* () {
@@ -987,6 +1137,41 @@ describe("A2AThreadRuntimeCore", () => {
       // Second run should have completed
       expect(core.isRunning()).toBe(false);
     });
+
+    it("stays running when an aborted run settles after its replacement starts", async () => {
+      let resolveFirst!: () => void;
+      let resolveSecond!: () => void;
+      const firstPending = new Promise<void>((resolve) => {
+        resolveFirst = resolve;
+      });
+      const secondPending = new Promise<void>((resolve) => {
+        resolveSecond = resolve;
+      });
+      const streamMessage = vi.fn().mockImplementation(() => {
+        const pending =
+          streamMessage.mock.calls.length === 1 ? firstPending : secondPending;
+        return (async function* () {
+          await pending;
+        })();
+      });
+      const core = createCore({ streamMessage });
+
+      const first = core.append(createUserAppendMessage("First"));
+      await vi.waitFor(() => expect(streamMessage).toHaveBeenCalledTimes(1));
+
+      const second = core.append(createUserAppendMessage("Second"));
+      await vi.waitFor(() => expect(streamMessage).toHaveBeenCalledTimes(2));
+      expect(core.isRunning()).toBe(true);
+
+      resolveFirst();
+      await first;
+
+      expect(core.isRunning()).toBe(true);
+
+      resolveSecond();
+      await second;
+      expect(core.isRunning()).toBe(false);
+    });
   });
 
   // --- applyExternalMessages ---

package/src/A2AThreadRuntimeCore.ts

@@ -46,6 +46,36 @@ const FALLBACK_USER_STATUS = {
   reason: "unknown",
 } as const;
 
+type A2ARuntimeCallbackName = "onError" | "onCancel" | "onArtifactComplete";
+
+const reportCallbackError = (name: A2ARuntimeCallbackName, error: unknown) => {
+  console.error(`[react-a2a] ${name} callback threw an error`, error);
+};
+
+const invokeRuntimeCallback = <TArgs extends unknown[]>(
+  name: A2ARuntimeCallbackName,
+  callback: ((...args: TArgs) => void) | undefined,
+  ...args: TArgs
+) => {
+  if (!callback) return;
+
+  try {
+    const result = callback(...args) as unknown;
+    if (
+      result !== null &&
+      (typeof result === "object" || typeof result === "function") &&
+      "then" in result &&
+      typeof result.then === "function"
+    ) {
+      void Promise.resolve(result).catch((error) => {
+        reportCallbackError(name, error);
+      });
+    }
+  } catch (error) {
+    reportCallbackError(name, error);
+  }
+};
+
 export class A2AThreadRuntimeCore {
   private client: A2AClient;
   private contextId: string | undefined;
@@ -216,7 +246,9 @@ export class A2AThreadRuntimeCore {
         }
       })
       .catch((error) => {
-        this.onError?.(
+        invokeRuntimeCallback(
+          "onError",
+          this.onError,
           error instanceof Error ? error : new Error(String(error)),
         );
       })
@@ -444,7 +476,7 @@ export class A2AThreadRuntimeCore {
           reason: "cancelled",
         });
         this.finishRun(abortController);
-        this.onCancel?.();
+        invokeRuntimeCallback("onCancel", this.onCancel);
       },
       { once: true },
     );
@@ -468,7 +500,7 @@ export class A2AThreadRuntimeCore {
           type: "incomplete",
           reason: "error",
         });
-        this.onError?.(err);
+        invokeRuntimeCallback("onError", this.onError, err);
         this.pendingError = this.pendingError ?? err;
       }
     } finally {
@@ -615,7 +647,11 @@ export class A2AThreadRuntimeCore {
     }
 
     if (lastChunk) {
-      this.onArtifactComplete?.(updated);
+      invokeRuntimeCallback(
+        "onArtifactComplete",
+        this.onArtifactComplete,
+        updated,
+      );
     }
 
     this.notifyUpdate();
@@ -746,9 +782,8 @@ export class A2AThreadRuntimeCore {
   }
 
   private finishRun(controller: AbortController | null) {
-    if (this.abortController === controller) {
-      this.abortController = null;
-    }
+    if (this.abortController !== controller) return;
+    this.abortController = null;
     this.setRunning(false);
   }
 

package/src/conversions.test.ts

@@ -30,26 +30,54 @@ describe("a2aPartToContent", () => {
     });
   });
 
-  it("converts non-image URL as text link", () => {
+  it("converts non-image URL to a url-sourced file part", () => {
     const part: A2APart = {
       url: "https://example.com/doc.pdf",
       mediaType: "application/pdf",
       filename: "doc.pdf",
     };
     expect(a2aPartToContent(part)).toEqual({
-      type: "text",
-      text: "[doc.pdf](https://example.com/doc.pdf)",
+      type: "file",
+      data: "https://example.com/doc.pdf",
+      mimeType: "application/pdf",
+      sourceType: "url",
+      filename: "doc.pdf",
     });
   });
 
-  it("converts URL without filename as plain text", () => {
+  it("converts URL without filename to a file part without filename", () => {
     const part: A2APart = {
       url: "https://example.com/doc.pdf",
       mediaType: "application/pdf",
     };
     expect(a2aPartToContent(part)).toEqual({
-      type: "text",
-      text: "https://example.com/doc.pdf",
+      type: "file",
+      data: "https://example.com/doc.pdf",
+      mimeType: "application/pdf",
+      sourceType: "url",
+    });
+  });
+
+  it("falls back to application/octet-stream for a URL without mediaType", () => {
+    const part: A2APart = { url: "https://example.com/download" };
+    expect(a2aPartToContent(part)).toEqual({
+      type: "file",
+      data: "https://example.com/download",
+      mimeType: "application/octet-stream",
+      sourceType: "url",
+    });
+  });
+
+  it("keeps the wire filename on an image URL part", () => {
+    const part: A2APart = {
+      url: "https://example.com/img.png",
+      mediaType: "image/png",
+      filename: "img.png",
+    };
+    expect(a2aPartToContent(part)).toEqual({
+      type: "image",
+      image: "https://example.com/img.png",
+      filename: "img.png",
     });
   });
 
@@ -64,26 +92,51 @@ describe("a2aPartToContent", () => {
     });
   });
 
-  it("converts raw non-image bytes as file reference", () => {
+  it("keeps the wire filename on raw image bytes", () => {
+    const part: A2APart = {
+      raw: "iVBORw0KGgo=",
+      mediaType: "image/png",
+      filename: "img.png",
+    };
+    expect(a2aPartToContent(part)).toEqual({
+      type: "image",
+      image: "data:image/png;base64,iVBORw0KGgo=",
+      filename: "img.png",
+    });
+  });
+
+  it("converts raw non-image bytes to a base64 file part", () => {
     const part: A2APart = {
       raw: "AAAA",
       mediaType: "application/pdf",
       filename: "report.pdf",
     };
     expect(a2aPartToContent(part)).toEqual({
-      type: "text",
-      text: "[File: report.pdf]",
+      type: "file",
+      data: "AAAA",
+      mimeType: "application/pdf",
+      filename: "report.pdf",
     });
   });
 
-  it("converts raw bytes without filename", () => {
+  it("converts raw audio bytes to a file part with the audio mime type", () => {
     const part: A2APart = {
-      raw: "AAAA",
-      mediaType: "application/octet-stream",
+      raw: "SGVsbG8=",
+      mediaType: "audio/mp3",
     };
     expect(a2aPartToContent(part)).toEqual({
-      type: "text",
-      text: "[File: download]",
+      type: "file",
+      data: "SGVsbG8=",
+      mimeType: "audio/mp3",
+    });
+  });
+
+  it("falls back to application/octet-stream for raw bytes without mediaType", () => {
+    const part: A2APart = { raw: "AAAA" };
+    expect(a2aPartToContent(part)).toEqual({
+      type: "file",
+      data: "AAAA",
+      mimeType: "application/octet-stream",
     });
   });
 
@@ -103,6 +156,60 @@ describe("a2aPartToContent", () => {
   });
 });
 
+describe("inbound file part round trip", () => {
+  const restoreFilePart = (part: A2APart) => {
+    const restored = a2aPartToContent(part);
+    if (restored.type !== "file") throw new Error("expected a file part");
+    return restored;
+  };
+
+  it("reproduces a url wire part through the outbound converter", () => {
+    const part: A2APart = {
+      url: "https://example.com/doc.pdf",
+      mediaType: "application/pdf",
+      filename: "doc.pdf",
+    };
+    expect(contentPartsToA2AParts([restoreFilePart(part)])).toEqual([part]);
+  });
+
+  it("reproduces a raw wire part through the outbound converter", () => {
+    const part: A2APart = {
+      raw: "AAAA",
+      mediaType: "application/pdf",
+      filename: "report.pdf",
+    };
+    expect(contentPartsToA2AParts([restoreFilePart(part)])).toEqual([part]);
+  });
+
+  it("reproduces a raw image wire part through the outbound converter", () => {
+    const part: A2APart = {
+      raw: "iVBORw0KGgo=",
+      mediaType: "image/png",
+      filename: "img.png",
+    };
+    const restored = a2aPartToContent(part);
+    if (restored.type !== "image") throw new Error("expected an image part");
+    expect(contentPartsToA2AParts([restored])).toEqual([part]);
+  });
+
+  it("resends a url part without mediaType with the octet-stream fallback", () => {
+    const part: A2APart = { url: "https://example.com/download" };
+    expect(contentPartsToA2AParts([restoreFilePart(part)])).toEqual([
+      {
+        url: "https://example.com/download",
+        mediaType: "application/octet-stream",
+      },
+    ]);
+  });
+
+  it("resends a raw part without mediaType with the octet-stream fallback", () => {
+    const part: A2APart = { raw: "AAAA" };
+    expect(contentPartsToA2AParts([restoreFilePart(part)])).toEqual([
+      { raw: "AAAA", mediaType: "application/octet-stream" },
+    ]);
+  });
+});
+
 describe("a2aPartsToContent", () => {
   it("converts multiple parts", () => {
     const parts: A2APart[] = [

package/src/conversions.ts

@@ -16,11 +16,18 @@ export function a2aPartToContent(
   }
   if (part.url !== undefined) {
     if (isImageMediaType(part.mediaType)) {
-      return { type: "image", image: part.url };
+      return {
+        type: "image",
+        image: part.url,
+        ...(part.filename && { filename: part.filename }),
+      };
     }
     return {
-      type: "text",
-      text: part.filename ? `[${part.filename}](${part.url})` : part.url,
+      type: "file",
+      data: part.url,
+      mimeType: part.mediaType ?? "application/octet-stream",
+      sourceType: "url",
+      ...(part.filename && { filename: part.filename }),
     };
   }
   if (part.raw !== undefined) {
@@ -28,11 +35,14 @@ export function a2aPartToContent(
       return {
         type: "image",
         image: `data:${part.mediaType};base64,${part.raw}`,
+        ...(part.filename && { filename: part.filename }),
       };
     }
     return {
-      type: "text",
-      text: `[File: ${part.filename ?? "download"}]`,
+      type: "file",
+      data: part.raw,
+      mimeType: part.mediaType ?? "application/octet-stream",
+      ...(part.filename && { filename: part.filename }),
     };
   }
   if (part.data !== undefined) {