@assistant-ui/react-ag-ui 0.0.48 → 0.0.49

package/dist/agUiExtras.d.ts

@@ -1,6 +1,6 @@
 import { AgUiRuntimeExtras } from "./runtime/types.js";
 //#region src/agUiExtras.d.ts
-declare const agUiExtras: import("@assistant-ui/core/internal").RuntimeExtras<AgUiRuntimeExtras>;
+declare const agUiExtras: import("@assistant-ui/core/react").RuntimeExtras<AgUiRuntimeExtras>;
 //#endregion
 export { agUiExtras };
 //# sourceMappingURL=agUiExtras.d.ts.map

package/dist/agUiExtras.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agUiExtras.d.ts","names":[],"sources":["../src/agUiExtras.ts"],"mappings":";;cAGa,kDAAU,cAAA"}
+{"version":3,"file":"agUiExtras.d.ts","names":[],"sources":["../src/agUiExtras.ts"],"mappings":";;cAGa,+CAAU,cAAA"}

package/dist/agUiExtras.js

@@ -1,4 +1,4 @@
-import { createRuntimeExtras } from "@assistant-ui/core/internal";
+import { createRuntimeExtras } from "@assistant-ui/core/react";
 //#region src/agUiExtras.ts
 const agUiExtras = createRuntimeExtras("useAgUiRuntime");
 //#endregion

package/dist/agUiExtras.js.map

@@ -1 +1 @@
-{"version":3,"file":"agUiExtras.js","names":[],"sources":["../src/agUiExtras.ts"],"sourcesContent":["import { createRuntimeExtras } from \"@assistant-ui/core/internal\";\nimport type { AgUiRuntimeExtras } from \"./runtime/types\";\n\nexport const agUiExtras =\n  createRuntimeExtras<AgUiRuntimeExtras>(\"useAgUiRuntime\");\n"],"mappings":";;AAGA,MAAa,aACX,oBAAuC,gBAAgB"}
+{"version":3,"file":"agUiExtras.js","names":[],"sources":["../src/agUiExtras.ts"],"sourcesContent":["import { createRuntimeExtras } from \"@assistant-ui/core/react\";\nimport type { AgUiRuntimeExtras } from \"./runtime/types\";\n\nexport const agUiExtras =\n  createRuntimeExtras<AgUiRuntimeExtras>(\"useAgUiRuntime\");\n"],"mappings":";;AAGA,MAAa,aACX,oBAAuC,gBAAgB"}

package/dist/hooks.d.ts

@@ -7,6 +7,11 @@ import { ReadonlyJSONValue } from "assistant-stream/utils";
  * array when no interrupts are pending, so consumers can `.map` without a guard.
  */
 declare const useAgUiInterrupts: () => readonly AgUiInterrupt[];
+/**
+ * Returns a stable function that sends an A2UI action back to the agent and
+ * resumes the run.
+ */
+declare const useAgUiSendA2uiAction: () => (action: Record<string, unknown>) => void;
 /**
  * Returns a function that submits responses (each `resolved` or `cancelled`)
  * for the pending AG-UI interrupts and resumes the run.
@@ -29,5 +34,5 @@ declare const useAgUiSteerAway: () => (message: CreateAppendMessage, responses?:
 declare const useAgUiState: <TState = ReadonlyJSONValue>() => TState | undefined;
 declare const useAgUiSetState: <TState = ReadonlyJSONValue>() => (next: TState | ((prev: TState | undefined) => TState)) => void;
 //#endregion
-export { useAgUiInterrupts, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };
+export { useAgUiInterrupts, useAgUiSendA2uiAction, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };
 //# sourceMappingURL=hooks.d.ts.map

package/dist/hooks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"hooks.d.ts","names":[],"sources":["../src/hooks.ts"],"mappings":";;;;;;;;cAca,kCAAiC;;;;;cAOjC,wCAEH,oBAAoB,sBAAoB;;;;;;;;;;;;;;cAiBrC,yBAGT,SAAS,qBACT,qBAAqB,sBACpB;cAGQ,eAAgB,SAAS,wBAClC;cAIS,kBAAmB,SAAS,yBAE/B,MAAM,WAAW,MAAM,uBAAuB"}
+{"version":3,"file":"hooks.d.ts","names":[],"sources":["../src/hooks.ts"],"mappings":";;;;;;;;cAea,kCAAiC;;;;;cAOjC,8BAAqB,QAGrB;;;;;cAUA,wCAEH,oBAAoB,sBAAoB;;;;;;;;;;;;;;cAiBrC,yBAGT,SAAS,qBACT,qBAAqB,sBACpB;cAGQ,eAAgB,SAAS,wBAClC;cAIS,kBAAmB,SAAS,yBAE/B,MAAM,WAAW,MAAM,uBAAuB"}

package/dist/hooks.js

@@ -1,5 +1,6 @@
 "use client";
 import { agUiExtras } from "./agUiExtras.js";
+import { useCallback } from "react";
 import { useAui } from "@assistant-ui/store";
 //#region src/hooks.ts
 const EMPTY_INTERRUPTS = [];
@@ -9,6 +10,14 @@ const EMPTY_INTERRUPTS = [];
 */
 const useAgUiInterrupts = () => agUiExtras.use((e) => e.interrupts, EMPTY_INTERRUPTS);
 /**
+* Returns a stable function that sends an A2UI action back to the agent and
+* resumes the run.
+*/
+const useAgUiSendA2uiAction = () => {
+	const aui = useAui();
+	return useCallback((action) => agUiExtras.get(aui).sendA2uiAction(action), [aui]);
+};
+/**
 * Returns a function that submits responses (each `resolved` or `cancelled`)
 * for the pending AG-UI interrupts and resumes the run.
 */
@@ -39,6 +48,6 @@ const useAgUiSetState = () => {
 	return (next) => agUiExtras.get(aui).setState(next);
 };
 //#endregion
-export { useAgUiInterrupts, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };
+export { useAgUiInterrupts, useAgUiSendA2uiAction, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };
 
 //# sourceMappingURL=hooks.js.map

package/dist/hooks.js.map

@@ -1 +1 @@
-{"version":3,"file":"hooks.js","names":[],"sources":["../src/hooks.ts"],"sourcesContent":["\"use client\";\n\nimport { useAui } from \"@assistant-ui/store\";\nimport type { CreateAppendMessage } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { agUiExtras } from \"./agUiExtras\";\nimport type { AgUiInterrupt, AgUiResumeEntry } from \"./runtime/types\";\n\nconst EMPTY_INTERRUPTS: readonly AgUiInterrupt[] = [];\n\n/**\n * Read the pending AG-UI interrupts on the current thread. Returns an empty\n * array when no interrupts are pending, so consumers can `.map` without a guard.\n */\nexport const useAgUiInterrupts = (): readonly AgUiInterrupt[] =>\n  agUiExtras.use((e) => e.interrupts, EMPTY_INTERRUPTS);\n\n/**\n * Returns a function that submits responses (each `resolved` or `cancelled`)\n * for the pending AG-UI interrupts and resumes the run.\n */\nexport const useAgUiSubmitInterruptResponses = () => {\n  const aui = useAui();\n  return (responses: readonly AgUiResumeEntry[]): Promise<void> =>\n    agUiExtras.get(aui).submitInterruptResponses(responses);\n};\n\n/**\n * Returns a function that discards the pending AG-UI interrupts or pending\n * client-side tool calls and appends the given user message, starting a fresh\n * run. Use this when the user sends a new message instead of resolving them:\n * every open interrupt is reported to the agent as cancelled, and every\n * unresolved client-side tool call receives an error result. The message\n * accepts a plain string or a partial `AppendMessage` (the parent defaults to\n * the current head). Pass `responses` to override the status of specific\n * interrupts; the rest still default to cancelled. `responses` throws when\n * tool calls rather than interrupts are pending. Unlike the\n * `autoCancelPendingToolCalls` runtime option, this hook cancels regardless\n * of that option's value.\n */\nexport const useAgUiSteerAway = () => {\n  const aui = useAui();\n  return (\n    message: CreateAppendMessage,\n    responses?: readonly AgUiResumeEntry[],\n  ): Promise<void> => agUiExtras.get(aui).steerAway(message, responses);\n};\n\nexport const useAgUiState = <TState = ReadonlyJSONValue>():\n  | TState\n  | undefined =>\n  agUiExtras.use((e) => e.state as TState | undefined, undefined);\n\nexport const useAgUiSetState = <TState = ReadonlyJSONValue>() => {\n  const aui = useAui();\n  return (next: TState | ((prev: TState | undefined) => TState)): void =>\n    agUiExtras\n      .get(aui)\n      .setState(\n        next as\n          | ReadonlyJSONValue\n          | ((prev: ReadonlyJSONValue | undefined) => ReadonlyJSONValue),\n      );\n};\n"],"mappings":";;;;AAQA,MAAM,mBAA6C,CAAC;;;;;AAMpD,MAAa,0BACX,WAAW,KAAK,MAAM,EAAE,YAAY,gBAAgB;;;;;AAMtD,MAAa,wCAAwC;CACnD,MAAM,MAAM,OAAO;CACnB,QAAQ,cACN,WAAW,IAAI,GAAG,CAAC,CAAC,yBAAyB,SAAS;AAC1D;;;;;;;;;;;;;;AAeA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,QACE,SACA,cACkB,WAAW,IAAI,GAAG,CAAC,CAAC,UAAU,SAAS,SAAS;AACtE;AAEA,MAAa,qBAGX,WAAW,KAAK,MAAM,EAAE,OAA6B,KAAA,CAAS;AAEhE,MAAa,wBAAoD;CAC/D,MAAM,MAAM,OAAO;CACnB,QAAQ,SACN,WACG,IAAI,GAAG,CAAC,CACR,SACC,IAGF;AACN"}
+{"version":3,"file":"hooks.js","names":[],"sources":["../src/hooks.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport type { CreateAppendMessage } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { agUiExtras } from \"./agUiExtras\";\nimport type { AgUiInterrupt, AgUiResumeEntry } from \"./runtime/types\";\n\nconst EMPTY_INTERRUPTS: readonly AgUiInterrupt[] = [];\n\n/**\n * Read the pending AG-UI interrupts on the current thread. Returns an empty\n * array when no interrupts are pending, so consumers can `.map` without a guard.\n */\nexport const useAgUiInterrupts = (): readonly AgUiInterrupt[] =>\n  agUiExtras.use((e) => e.interrupts, EMPTY_INTERRUPTS);\n\n/**\n * Returns a stable function that sends an A2UI action back to the agent and\n * resumes the run.\n */\nexport const useAgUiSendA2uiAction = () => {\n  const aui = useAui();\n  return useCallback(\n    (action: Record<string, unknown>): void =>\n      agUiExtras.get(aui).sendA2uiAction(action),\n    [aui],\n  );\n};\n\n/**\n * Returns a function that submits responses (each `resolved` or `cancelled`)\n * for the pending AG-UI interrupts and resumes the run.\n */\nexport const useAgUiSubmitInterruptResponses = () => {\n  const aui = useAui();\n  return (responses: readonly AgUiResumeEntry[]): Promise<void> =>\n    agUiExtras.get(aui).submitInterruptResponses(responses);\n};\n\n/**\n * Returns a function that discards the pending AG-UI interrupts or pending\n * client-side tool calls and appends the given user message, starting a fresh\n * run. Use this when the user sends a new message instead of resolving them:\n * every open interrupt is reported to the agent as cancelled, and every\n * unresolved client-side tool call receives an error result. The message\n * accepts a plain string or a partial `AppendMessage` (the parent defaults to\n * the current head). Pass `responses` to override the status of specific\n * interrupts; the rest still default to cancelled. `responses` throws when\n * tool calls rather than interrupts are pending. Unlike the\n * `autoCancelPendingToolCalls` runtime option, this hook cancels regardless\n * of that option's value.\n */\nexport const useAgUiSteerAway = () => {\n  const aui = useAui();\n  return (\n    message: CreateAppendMessage,\n    responses?: readonly AgUiResumeEntry[],\n  ): Promise<void> => agUiExtras.get(aui).steerAway(message, responses);\n};\n\nexport const useAgUiState = <TState = ReadonlyJSONValue>():\n  | TState\n  | undefined =>\n  agUiExtras.use((e) => e.state as TState | undefined, undefined);\n\nexport const useAgUiSetState = <TState = ReadonlyJSONValue>() => {\n  const aui = useAui();\n  return (next: TState | ((prev: TState | undefined) => TState)): void =>\n    agUiExtras\n      .get(aui)\n      .setState(\n        next as\n          | ReadonlyJSONValue\n          | ((prev: ReadonlyJSONValue | undefined) => ReadonlyJSONValue),\n      );\n};\n"],"mappings":";;;;;AASA,MAAM,mBAA6C,CAAC;;;;;AAMpD,MAAa,0BACX,WAAW,KAAK,MAAM,EAAE,YAAY,gBAAgB;;;;;AAMtD,MAAa,8BAA8B;CACzC,MAAM,MAAM,OAAO;CACnB,OAAO,aACJ,WACC,WAAW,IAAI,GAAG,CAAC,CAAC,eAAe,MAAM,GAC3C,CAAC,GAAG,CACN;AACF;;;;;AAMA,MAAa,wCAAwC;CACnD,MAAM,MAAM,OAAO;CACnB,QAAQ,cACN,WAAW,IAAI,GAAG,CAAC,CAAC,yBAAyB,SAAS;AAC1D;;;;;;;;;;;;;;AAeA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,QACE,SACA,cACkB,WAAW,IAAI,GAAG,CAAC,CAAC,UAAU,SAAS,SAAS;AACtE;AAEA,MAAa,qBAGX,WAAW,KAAK,MAAM,EAAE,OAA6B,KAAA,CAAS;AAEhE,MAAa,wBAAoD;CAC/D,MAAM,MAAM,OAAO;CACnB,QAAQ,SACN,WACG,IAAI,GAAG,CAAC,CACR,SACC,IAGF;AACN"}

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import { AgUiInterrupt, AgUiInterruptReason, AgUiResumeEntry, AgUiRunFinishedOutcome, UseAgUiRuntimeAdapters, UseAgUiRuntimeOptions, UseAgUiThreadListAdapter } from "./runtime/types.js";
-import { useAgUiInterrupts, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses } from "./hooks.js";
+import { useAgUiInterrupts, useAgUiSendA2uiAction, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses } from "./hooks.js";
 import { AgUiAssistantRuntime, useAgUiRuntime } from "./useAgUiRuntime.js";
 import { FromAgUiMessagesOptions, fromAgUiMessages } from "./runtime/adapter/conversions.js";
-export { type AgUiAssistantRuntime, type AgUiInterrupt, type AgUiInterruptReason, type AgUiResumeEntry, type AgUiRunFinishedOutcome, type FromAgUiMessagesOptions, type UseAgUiRuntimeAdapters, type UseAgUiRuntimeOptions, type UseAgUiThreadListAdapter, fromAgUiMessages, useAgUiInterrupts, useAgUiRuntime, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };
+export { type AgUiAssistantRuntime, type AgUiInterrupt, type AgUiInterruptReason, type AgUiResumeEntry, type AgUiRunFinishedOutcome, type FromAgUiMessagesOptions, type UseAgUiRuntimeAdapters, type UseAgUiRuntimeOptions, type UseAgUiThreadListAdapter, fromAgUiMessages, useAgUiInterrupts, useAgUiRuntime, useAgUiSendA2uiAction, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };

package/dist/index.js

@@ -1,4 +1,4 @@
-import { useAgUiInterrupts, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses } from "./hooks.js";
+import { useAgUiInterrupts, useAgUiSendA2uiAction, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses } from "./hooks.js";
 import { fromAgUiMessages } from "./runtime/adapter/conversions.js";
 import { useAgUiRuntime } from "./useAgUiRuntime.js";
-export { fromAgUiMessages, useAgUiInterrupts, useAgUiRuntime, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };
+export { fromAgUiMessages, useAgUiInterrupts, useAgUiRuntime, useAgUiSendA2uiAction, useAgUiSetState, useAgUiState, useAgUiSteerAway, useAgUiSubmitInterruptResponses };

package/dist/runtime/AgUiThreadRuntimeCore.d.ts

@@ -44,6 +44,8 @@ declare class AgUiThreadRuntimeCore {
   private _isLoading;
   private _loadPromise;
   private pendingResumeMessageId;
+  private pendingA2uiResume;
+  private pendingA2uiAction;
   constructor(options: CoreOptions);
   updateOptions(options: Omit<CoreOptions, "notifyUpdate">): void;
   attachRuntime(runtime: AssistantRuntime): void;
@@ -93,6 +95,7 @@ declare class AgUiThreadRuntimeCore {
   findMessageIdForToolCall(toolCallId: string): string | undefined;
   private cancelUnresolvedToolCalls;
   addToolResult(options: AddToolResultOptions): void;
+  sendA2uiAction(action: Record<string, unknown>): void;
   private maybeResumeAfterToolResults;
   private maybeCompleteAfterToolResults;
   private startResumeRun;

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

@@ -1 +1 @@
-{"version":3,"file":"AgUiThreadRuntimeCore.d.ts","names":[],"sources":["../../src/runtime/AgUiThreadRuntimeCore.ts"],"mappings":";;;;;;KAsDK,YAAY,YAAY;KACxB,gBACH,SAAS,wBACN,eAAe;KACf;EACH;EACA;EACA,WAAW;EACX,SAAS;;KAGN;EACH,OAAO;EACP,QAAQ;EACR;EACA;EACA,WAAW,OAAO;EAClB;EACA,UAAU;EACV;;cAKW;UACH;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;mBACA;UACT;UACA;UACA;EAEI,YAAA,SAAS;EAYrB,cAAc,SAAS,KAAK;EAW5B,cAAc,SAAS;EAIvB;EAIA,wBAAwB;EAIxB,wBAAwB;UAmBhB;UAQA;UAUA;UAIA;UAQA;UAQA;UAMA;UAKA;UAKA;UAKA;EAYR,YAAY;EAIZ;MAII;EAIJ,mBAAmB;EA2Cb,OAAO,SAAS,gBAAgB;UAW9B;UAQA;EAiBF,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAU,YAAY;MACrB;EAMG,UAAU;EAKV,OAAO,QAAQ,kBAAkB;EAUjC,kBAAkB,mBAAmB,kBAAkB;UAwBrD;UAOA;EAgBR;IACE;IACA,qBAAqB;;EAavB;IACE;IACA;;EAcI,yBACJ,oBAAoB,oBACnB;EAoDG,UACJ,SAAS,qBACT,qBAAqB,oBACpB;UAyCK;UAkBA;UAiCA;UA0BA;EA+BR,yBAAyB;UAkBjB;EAiBR,cAAc,SAAS;UA4Bf;UAYA;UA6BA;EAMR,sBAAsB,mBAAmB;UAgDjC;EAgER,kBAAkB,OAAO;EAKzB,SACE,MACI,sBACE,MAAM,kCAAkC;EAOhD;UAKc;UAkJA;UAyDN;UA6BA;UAqBA;UAKA;UAOA;UAuBA;UA2CA;UAiDA;UA4BA;UA6BA;UA4DA;UA2EA;UA8DA;UAwBA;UAQA;UAIA;UAKA;UAIA;UAQA;UAWA"}
+{"version":3,"file":"AgUiThreadRuntimeCore.d.ts","names":[],"sources":["../../src/runtime/AgUiThreadRuntimeCore.ts"],"mappings":";;;;;;KAsDK,YAAY,YAAY;KACxB,gBACH,SAAS,wBACN,eAAe;KACf;EACH;EACA;EACA,WAAW;EACX,SAAS;;KAGN;EACH,OAAO;EACP,QAAQ;EACR;EACA;EACA,WAAW,OAAO;EAClB;EACA,UAAU;EACV;;cAKW;UACH;UACA;UACA;UACA;UACA;UACA;mBACS;UAET;mBACS;UACT;UACA;UACA;UACA;UACA;UACA;UACA;mBACS;mBACA;UACT;UACA;UACA;UACA;UACA;EAEI,YAAA,SAAS;EAYrB,cAAc,SAAS,KAAK;EAW5B,cAAc,SAAS;EAIvB;EAIA,wBAAwB;EAIxB,wBAAwB;UAmBhB;UAQA;UAUA;UAIA;UAQA;UAQA;UAMA;UAKA;UAKA;UAKA;EAYR,YAAY;EAIZ;MAII;EAIJ,mBAAmB;EA2Cb,OAAO,SAAS,gBAAgB;UAW9B;UAQA;EAiBF,KAAK,SAAS,gBAAgB;EAI9B,OACJ,yBACA;IAAU,YAAY;MACrB;EAMG,UAAU;EAKV,OAAO,QAAQ,kBAAkB;EAUjC,kBAAkB,mBAAmB,kBAAkB;UAwBrD;UAOA;EAgBR;IACE;IACA,qBAAqB;;EAavB;IACE;IACA;;EAcI,yBACJ,oBAAoB,oBACnB;EAoDG,UACJ,SAAS,qBACT,qBAAqB,oBACpB;UAyCK;UAkBA;UAiCA;UA0BA;EA+BR,yBAAyB;UAkBjB;EAiBR,cAAc,SAAS;EAyBvB,eAAe,QAAQ;UA4Bf;UAYA;UA6BA;EAMR,sBAAsB,mBAAmB;UAkDjC;EAgER,kBAAkB,OAAO;EAKzB,SACE,MACI,sBACE,MAAM,kCAAkC;EAOhD;UAKc;UAoMA;UA2DN;UAkCA;UAqBA;UAKA;UAOA;UAyBA;UA2CA;UAiDA;UA4BA;UA6BA;UAgEA;UA2EA;UA8DA;UAgDA;UAQA;UAIA;UAKA;UAIA;UAQA;UAWA"}

package/dist/runtime/AgUiThreadRuntimeCore.js

@@ -3,8 +3,8 @@ import { readMcpAppResourceUri } from "./mcp-tool-result.js";
 import { AG_UI_METADATA_NAMESPACE, RunAggregator, isPlainObject, tryParseJSON } from "./adapter/run-aggregator.js";
 import { fromAgUiMessages, toAgUiMessages, toAgUiTools } from "./adapter/conversions.js";
 import { createAgUiSubscriber } from "./adapter/subscriber.js";
-import { MessageRepository } from "@assistant-ui/core/internal";
 import { fromThreadMessageLike, generateId, isMcpAppUri } from "@assistant-ui/core";
+import { MessageRepository } from "@assistant-ui/core/internal";
 import jsonpatch from "fast-json-patch";
 //#region src/runtime/AgUiThreadRuntimeCore.ts
 const optimisticPrefix = "__optimistic__";
@@ -38,6 +38,8 @@ var AgUiThreadRuntimeCore = class {
 	_isLoading = false;
 	_loadPromise;
 	pendingResumeMessageId = null;
+	pendingA2uiResume = false;
+	pendingA2uiAction;
 	constructor(options) {
 		this.agent = options.agent;
 		this.logger = options.logger;
@@ -424,6 +426,24 @@ var AgUiThreadRuntimeCore = class {
 		this.notifyUpdate();
 		this.maybeResumeAfterToolResults(options.messageId);
 	}
+	sendA2uiAction(action) {
+		this.assertNoPendingInterrupts();
+		this.maybeAutoCancelPendingToolCalls();
+		const parentId = this.repository.headId;
+		if (parentId === null) {
+			this.logger.debug("[agui] sendA2uiAction: no messages to resume, dropping action");
+			return;
+		}
+		const userAction = { ...action };
+		delete userAction.type;
+		if (!("timestamp" in userAction)) userAction.timestamp = (/* @__PURE__ */ new Date()).toISOString();
+		this.pendingA2uiAction = userAction;
+		if (this.isRunningFlag) {
+			this.pendingA2uiResume = true;
+			return;
+		}
+		this.startResumeRun(parentId);
+	}
 	maybeResumeAfterToolResults(messageId) {
 		if (!this.maybeCompleteAfterToolResults(messageId)) return;
 		if (this.isRunningFlag) {
@@ -455,6 +475,8 @@ var AgUiThreadRuntimeCore = class {
 		});
 	}
 	applyExternalMessages(messages) {
+		this.pendingA2uiResume = false;
+		this.pendingA2uiAction = void 0;
 		this.assistantHistoryParents.clear();
 		if (messages.length === 0) this.clearRepository();
 		else {
@@ -564,16 +586,21 @@ var AgUiThreadRuntimeCore = class {
 		this.pendingError = null;
 		const assistantParentId = parent ? parentId : this.repository.headId;
 		let assistantMessageId;
-		const ensureAssistant = () => {
-			if (assistantMessageId) return assistantMessageId;
-			const created = this.insertAssistantPlaceholder(assistantParentId ?? null);
+		let assistantCollided = false;
+		const ensureAssistant = (allowRecreate = false) => {
+			const cached = assistantMessageId;
+			if (cached !== void 0 && this.tryGetMessage(cached)) return cached;
+			if (cached !== void 0 && (assistantCollided || !allowRecreate)) return cached;
+			const parentId = cached === void 0 && assistantParentId && this.hasMessage(assistantParentId) ? assistantParentId : this.repository.headId;
+			const created = this.insertAssistantPlaceholder(parentId, cached);
 			assistantMessageId = created;
-			this.markPendingAssistantHistory(created, assistantParentId ?? null);
+			this.markPendingAssistantHistory(created, parentId);
 			return created;
 		};
 		if (shouldEagerlyInsertAssistant) ensureAssistant();
 		const applyUpdate = (update) => {
-			const resolved = this.updateAssistantMessage(ensureAssistant(), update);
+			const hasContent = Array.isArray(update.content) && update.content.length > 0;
+			const resolved = this.updateAssistantMessage(ensureAssistant(hasContent), update);
 			if (resolved !== assistantMessageId) assistantMessageId = resolved;
 		};
 		const aggregator = new RunAggregator({
@@ -581,12 +608,13 @@ var AgUiThreadRuntimeCore = class {
 			logger: this.logger,
 			emit: applyUpdate,
 			onServerMessageId: (serverId) => {
-				const placeholder = ensureAssistant();
+				const placeholder = ensureAssistant(true);
 				if (placeholder === serverId) return;
 				if (this.reassignAssistantId(placeholder, serverId)) assistantMessageId = serverId;
+				else assistantCollided = true;
 			}
 		});
-		const dispatch = (event) => this.handleEvent(aggregator, event);
+		const dispatch = (event) => this.handleEvent(aggregator, event, assistantMessageId);
 		const abortController = new AbortController();
 		const abortSignal = abortController.signal;
 		this.abortController = abortController;
@@ -653,15 +681,37 @@ var AgUiThreadRuntimeCore = class {
 			const err = this.pendingError;
 			this.pendingError = null;
 			this.pendingResumeMessageId = null;
+			this.pendingA2uiResume = false;
+			this.pendingA2uiAction = void 0;
 			throw err;
 		}
 		if (this.pendingResumeMessageId !== null) {
 			const resumeMessageId = this.pendingResumeMessageId;
 			this.pendingResumeMessageId = null;
 			if (!abortSignal.aborted) this.startResumeRun(resumeMessageId);
+			else this.pendingA2uiAction = void 0;
+		}
+		if (this.pendingA2uiResume) {
+			this.pendingA2uiResume = false;
+			if (!abortSignal.aborted && this.pendingA2uiAction !== void 0) {
+				if (this.getPendingInterrupts()) {
+					this.pendingA2uiAction = void 0;
+					this.logger.debug("[agui] sendA2uiAction: pending interrupts, dropping action");
+					return;
+				}
+				this.maybeAutoCancelPendingToolCalls();
+				const parentId = this.repository.headId;
+				if (parentId !== null) this.startResumeRun(parentId);
+				else {
+					this.pendingA2uiAction = void 0;
+					this.logger.debug("[agui] sendA2uiAction: no messages to resume, dropping action");
+				}
+			} else this.pendingA2uiAction = void 0;
 		}
 	}
 	async consumeResumeStream(stream, ctx) {
+		this.pendingA2uiAction = void 0;
+		this.pendingA2uiResume = false;
 		const assistantId = ctx.ensureAssistant();
 		const currentId = () => ctx.getAssistantMessageId() ?? assistantId;
 		const options = {
@@ -706,7 +756,7 @@ var AgUiThreadRuntimeCore = class {
 		const threadId = this.agent.threadId || "main";
 		const messages = toAgUiMessages(historyMessages ?? this.repository.getMessages());
 		const context = this.runtime?.thread.getModelContext();
-		return {
+		const input = {
 			threadId,
 			runId,
 			state: this.stateSnapshot ?? null,
@@ -719,10 +769,13 @@ var AgUiThreadRuntimeCore = class {
 			forwardedProps: {
 				...context?.callSettings ?? {},
 				...context?.config ?? {},
-				...runConfig?.custom ? { runConfig: runConfig.custom } : {}
+				...runConfig?.custom ? { runConfig: runConfig.custom } : {},
+				...this.pendingA2uiAction ? { a2uiAction: { userAction: this.pendingA2uiAction } } : {}
 			},
 			...resume !== void 0 ? { resume } : {}
 		};
+		this.pendingA2uiAction = void 0;
+		return input;
 	}
 	installResumeShim() {
 		const agent = this.agent;
@@ -747,8 +800,7 @@ var AgUiThreadRuntimeCore = class {
 		if (this.abortController === controller) this.abortController = null;
 		this.setRunning(false);
 	}
-	insertAssistantPlaceholder(parentId) {
-		const id = generateOptimisticId();
+	insertAssistantPlaceholder(parentId, id = generateOptimisticId()) {
 		const assistant = {
 			id,
 			role: "assistant",
@@ -760,7 +812,7 @@ var AgUiThreadRuntimeCore = class {
 				unstable_annotations: [],
 				unstable_data: [],
 				steps: [],
-				isOptimistic: true,
+				isOptimistic: isOptimisticId(id),
 				custom: {}
 			}
 		};
@@ -863,7 +915,7 @@ var AgUiThreadRuntimeCore = class {
 			} : current.custom
 		};
 	}
-	handleEvent(aggregator, event) {
+	handleEvent(aggregator, event, activeAssistantId) {
 		switch (event.type) {
 			case "STATE_SNAPSHOT":
 				this.stateSnapshot = event.snapshot;
@@ -881,7 +933,7 @@ var AgUiThreadRuntimeCore = class {
 				}
 				return;
 			case "MESSAGES_SNAPSHOT":
-				this.importMessagesSnapshot(event.messages);
+				this.importMessagesSnapshot(event.messages, activeAssistantId);
 				return;
 			case "TOOL_CALL_RESULT":
 				if (!aggregator.hasToolCall(event.toolCallId)) {
@@ -986,8 +1038,10 @@ var AgUiThreadRuntimeCore = class {
 		this.notifyUpdate();
 		this.maybeCompleteAfterToolResults(messageId);
 	}
-	importMessagesSnapshot(rawMessages) {
+	importMessagesSnapshot(rawMessages, activeAssistantId) {
 		try {
+			const activeMessage = activeAssistantId ? this.tryGetMessage(activeAssistantId)?.message : void 0;
+			const activeAssistant = activeMessage?.role === "assistant" ? activeMessage : void 0;
 			const normalized = fromAgUiMessages(rawMessages, { showThinking: this.showThinking });
 			const converted = [];
 			for (const message of normalized) try {
@@ -995,7 +1049,15 @@ var AgUiThreadRuntimeCore = class {
 			} catch (error) {
 				this.logger.error?.("[agui] failed to import message from snapshot", error);
 			}
+			const snapshotHeadId = converted.at(-1)?.id ?? null;
+			const snapshotContainsActiveAssistant = converted.some((message) => message.id === activeAssistant?.id);
+			const preservesActiveAssistant = activeAssistant !== void 0 && !snapshotContainsActiveAssistant && (activeAssistant.metadata.isOptimistic !== true || converted.at(-1)?.role !== "assistant");
+			if (preservesActiveAssistant) converted.push(activeAssistant);
 			this.applyExternalMessages(converted);
+			if (preservesActiveAssistant) {
+				this.recordedHistoryIds.delete(activeAssistant.id);
+				this.markPendingAssistantHistory(activeAssistant.id, snapshotHeadId);
+			}
 		} catch (error) {
 			this.logger.error?.("[agui] failed to import messages snapshot", error);
 		}