@assistant-ui/react-a2a 0.2.46 → 0.2.48

package/dist/a2aThreadState.d.ts

@@ -0,0 +1,56 @@
+import { type A2uiState } from "@assistant-ui/generative-ui/a2ui";
+import type { A2AAgentCard, A2AArtifact, A2AMessage, A2ATask, A2ATaskArtifactUpdateEvent, A2ATaskStatusUpdateEvent } from "./types.js";
+export type A2AThreadState = {
+    currentTask: A2ATask | undefined;
+    currentArtifacts: A2AArtifact[];
+    a2uiState: A2uiState;
+    a2uiMessageIds: ReadonlySet<string>;
+    agentCard: A2AAgentCard | undefined;
+    isRunning: boolean;
+    isLoading: boolean;
+    loadRequested: boolean;
+    contextId: string | undefined;
+};
+export type A2AThreadAction = {
+    type: "statusUpdate";
+    event: A2ATaskStatusUpdateEvent;
+} | {
+    type: "artifactUpdate";
+    event: A2ATaskArtifactUpdateEvent;
+} | {
+    type: "taskSnapshot";
+    task: A2ATask;
+} | {
+    type: "a2uiMessage";
+    message: A2AMessage;
+    replay?: boolean;
+} | {
+    type: "prepareRun";
+} | {
+    type: "runStarted";
+} | {
+    type: "runFinished";
+} | {
+    type: "agentCardLoaded";
+    agentCard: A2AAgentCard;
+} | {
+    type: "loadRequested";
+} | {
+    type: "loadStarted";
+} | {
+    type: "loadFinished";
+} | {
+    type: "contextLearned";
+    contextId: string | undefined;
+} | {
+    type: "resetContext";
+    contextId: string | undefined;
+} | {
+    type: "taskReplaced";
+    task: A2ATask;
+} | {
+    type: "reset";
+};
+export declare const createA2AThreadState: (contextId?: string) => A2AThreadState;
+export declare const reduceA2AThreadState: (state: A2AThreadState, action: A2AThreadAction) => A2AThreadState;
+//# sourceMappingURL=a2aThreadState.d.ts.map

package/dist/a2aThreadState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"a2aThreadState.d.ts","sourceRoot":"","sources":["../src/a2aThreadState.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,SAAS,EACf,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EACV,YAAY,EACZ,WAAW,EACX,UAAU,EAEV,OAAO,EACP,0BAA0B,EAC1B,wBAAwB,EACzB,mBAAgB;AAEjB,MAAM,MAAM,cAAc,GAAG;IAC3B,WAAW,EAAE,OAAO,GAAG,SAAS,CAAC;IACjC,gBAAgB,EAAE,WAAW,EAAE,CAAC;IAChC,SAAS,EAAE,SAAS,CAAC;IACrB,cAAc,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IACpC,SAAS,EAAE,YAAY,GAAG,SAAS,CAAC;IACpC,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,aAAa,EAAE,OAAO,CAAC;IACvB,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;CAC/B,CAAC;AAEF,MAAM,MAAM,eAAe,GACvB;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,KAAK,EAAE,wBAAwB,CAAA;CAAE,GACzD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,KAAK,EAAE,0BAA0B,CAAA;CAAE,GAC7D;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,GACvC;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,OAAO,EAAE,UAAU,CAAC;IAAC,MAAM,CAAC,EAAE,OAAO,CAAA;CAAE,GAC9D;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GACtB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,SAAS,EAAE,YAAY,CAAA;CAAE,GACpD;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GACvB;IAAE,IAAI,EAAE,cAAc,CAAA;CAAE,GACxB;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,GACzD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,GACvD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,GACvC;IAAE,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC;AAEtB,eAAO,MAAM,oBAAoB,GAAI,YAAY,MAAM,KAAG,cAUxD,CAAC;AAuCH,eAAO,MAAM,oBAAoB,GAC/B,OAAO,cAAc,EACrB,QAAQ,eAAe,KACtB,cAiJF,CAAC"}

package/dist/a2aThreadState.js

@@ -0,0 +1,152 @@
+import { a2uiPartsToOperations, isTerminalTaskState } from "./conversions.js";
+import { applyA2uiOperations } from "@assistant-ui/generative-ui/a2ui";
+//#region src/a2aThreadState.ts
+const createA2AThreadState = (contextId) => ({
+	currentTask: void 0,
+	currentArtifacts: [],
+	a2uiState: /* @__PURE__ */ new Map(),
+	a2uiMessageIds: /* @__PURE__ */ new Set(),
+	agentCard: void 0,
+	isRunning: false,
+	isLoading: false,
+	loadRequested: false,
+	contextId
+});
+const normalizeArtifact = (artifact) => ({
+	...artifact,
+	parts: Array.isArray(artifact.parts) ? artifact.parts : []
+});
+const applyParts = (state, parts) => {
+	const operations = a2uiPartsToOperations(parts);
+	if (operations.length === 0) return state;
+	return {
+		...state,
+		a2uiState: applyA2uiOperations(state.a2uiState, operations).state
+	};
+};
+const applyMessage = (state, message, replay) => {
+	if (message.messageId && replay && state.a2uiMessageIds.has(message.messageId)) return state;
+	const next = message.messageId ? {
+		...state,
+		a2uiMessageIds: /* @__PURE__ */ new Set([...state.a2uiMessageIds, message.messageId])
+	} : state;
+	return applyParts(next, message.parts);
+};
+const reduceA2AThreadState = (state, action) => {
+	switch (action.type) {
+		case "statusUpdate": {
+			const { event } = action;
+			const currentTask = state.currentTask ? {
+				...state.currentTask,
+				status: event.status
+			} : {
+				id: event.taskId,
+				contextId: event.contextId,
+				status: event.status
+			};
+			const next = {
+				...state,
+				currentTask,
+				contextId: event.contextId || state.contextId
+			};
+			return event.status.message ? applyMessage(next, event.status.message, false) : next;
+		}
+		case "artifactUpdate": {
+			const { artifact: input, append } = action.event;
+			const artifact = normalizeArtifact(input);
+			const index = state.currentArtifacts.findIndex((current) => current.artifactId === artifact.artifactId);
+			const updated = index >= 0 && append ? {
+				...state.currentArtifacts[index],
+				parts: [...state.currentArtifacts[index].parts, ...artifact.parts]
+			} : artifact;
+			const currentArtifacts = index >= 0 ? [
+				...state.currentArtifacts.slice(0, index),
+				updated,
+				...state.currentArtifacts.slice(index + 1)
+			] : [...state.currentArtifacts, updated];
+			return applyParts({
+				...state,
+				currentArtifacts
+			}, artifact.parts);
+		}
+		case "taskSnapshot": {
+			const { task } = action;
+			const artifacts = task.artifacts === void 0 ? void 0 : Array.isArray(task.artifacts) ? task.artifacts.map(normalizeArtifact) : [];
+			const history = task.history === void 0 ? void 0 : Array.isArray(task.history) ? task.history : [];
+			const complete = history !== void 0 && artifacts !== void 0;
+			const artifactsToApply = complete ? artifacts : artifacts?.filter((artifact) => !state.currentArtifacts.some(({ artifactId }) => artifactId === artifact.artifactId));
+			let next = {
+				...state,
+				currentTask: {
+					...task,
+					...artifacts === void 0 ? {} : { artifacts },
+					...history === void 0 ? {} : { history }
+				},
+				contextId: task.contextId || state.contextId,
+				currentArtifacts: complete ? artifacts : artifactsToApply ? [...state.currentArtifacts, ...artifactsToApply] : state.currentArtifacts,
+				a2uiState: complete ? /* @__PURE__ */ new Map() : state.a2uiState,
+				a2uiMessageIds: complete ? /* @__PURE__ */ new Set() : state.a2uiMessageIds
+			};
+			for (const message of history ?? []) if (message.role === "agent") next = applyMessage(next, message, true);
+			for (const artifact of artifactsToApply ?? []) next = applyParts(next, artifact.parts);
+			if (task.status.message) next = applyMessage(next, task.status.message, true);
+			return next;
+		}
+		case "a2uiMessage": return applyMessage(state, action.message, action.replay ?? false);
+		case "prepareRun": return {
+			...state,
+			currentTask: state.currentTask && isTerminalTaskState(state.currentTask.status.state) ? void 0 : state.currentTask,
+			currentArtifacts: [],
+			a2uiState: /* @__PURE__ */ new Map(),
+			a2uiMessageIds: /* @__PURE__ */ new Set()
+		};
+		case "runStarted": return state.isRunning ? state : {
+			...state,
+			isRunning: true
+		};
+		case "runFinished": return !state.isRunning ? state : {
+			...state,
+			isRunning: false
+		};
+		case "agentCardLoaded": return state.agentCard === action.agentCard ? state : {
+			...state,
+			agentCard: action.agentCard
+		};
+		case "loadRequested": return state.loadRequested ? state : {
+			...state,
+			loadRequested: true
+		};
+		case "loadStarted": return state.isLoading ? state : {
+			...state,
+			isLoading: true
+		};
+		case "loadFinished": return !state.isLoading ? state : {
+			...state,
+			isLoading: false
+		};
+		case "contextLearned": return state.contextId === action.contextId ? state : {
+			...state,
+			contextId: action.contextId
+		};
+		case "resetContext": return !state.isLoading && state.contextId === action.contextId ? state : {
+			...state,
+			isLoading: false,
+			contextId: action.contextId
+		};
+		case "taskReplaced": return state.currentTask === action.task ? state : {
+			...state,
+			currentTask: action.task
+		};
+		case "reset": return {
+			...state,
+			currentTask: void 0,
+			currentArtifacts: [],
+			a2uiState: /* @__PURE__ */ new Map(),
+			a2uiMessageIds: /* @__PURE__ */ new Set()
+		};
+	}
+};
+//#endregion
+export { createA2AThreadState, reduceA2AThreadState };
+
+//# sourceMappingURL=a2aThreadState.js.map

package/dist/a2aThreadState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"a2aThreadState.js","names":[],"sources":["../src/a2aThreadState.ts"],"sourcesContent":["import {\n  applyA2uiOperations,\n  type A2uiState,\n} from \"@assistant-ui/generative-ui/a2ui\";\nimport { a2uiPartsToOperations, isTerminalTaskState } from \"./conversions\";\nimport type {\n  A2AAgentCard,\n  A2AArtifact,\n  A2AMessage,\n  A2APart,\n  A2ATask,\n  A2ATaskArtifactUpdateEvent,\n  A2ATaskStatusUpdateEvent,\n} from \"./types\";\n\nexport type A2AThreadState = {\n  currentTask: A2ATask | undefined;\n  currentArtifacts: A2AArtifact[];\n  a2uiState: A2uiState;\n  a2uiMessageIds: ReadonlySet<string>;\n  agentCard: A2AAgentCard | undefined;\n  isRunning: boolean;\n  isLoading: boolean;\n  loadRequested: boolean;\n  contextId: string | undefined;\n};\n\nexport type A2AThreadAction =\n  | { type: \"statusUpdate\"; event: A2ATaskStatusUpdateEvent }\n  | { type: \"artifactUpdate\"; event: A2ATaskArtifactUpdateEvent }\n  | { type: \"taskSnapshot\"; task: A2ATask }\n  | { type: \"a2uiMessage\"; message: A2AMessage; replay?: boolean }\n  | { type: \"prepareRun\" }\n  | { type: \"runStarted\" }\n  | { type: \"runFinished\" }\n  | { type: \"agentCardLoaded\"; agentCard: A2AAgentCard }\n  | { type: \"loadRequested\" }\n  | { type: \"loadStarted\" }\n  | { type: \"loadFinished\" }\n  | { type: \"contextLearned\"; contextId: string | undefined }\n  | { type: \"resetContext\"; contextId: string | undefined }\n  | { type: \"taskReplaced\"; task: A2ATask }\n  | { type: \"reset\" };\n\nexport const createA2AThreadState = (contextId?: string): A2AThreadState => ({\n  currentTask: undefined,\n  currentArtifacts: [],\n  a2uiState: new Map(),\n  a2uiMessageIds: new Set(),\n  agentCard: undefined,\n  isRunning: false,\n  isLoading: false,\n  loadRequested: false,\n  contextId,\n});\n\nconst normalizeArtifact = (artifact: A2AArtifact): A2AArtifact => ({\n  ...artifact,\n  parts: Array.isArray(artifact.parts) ? artifact.parts : [],\n});\n\nconst applyParts = (\n  state: A2AThreadState,\n  parts: readonly A2APart[],\n): A2AThreadState => {\n  const operations = a2uiPartsToOperations(parts);\n  if (operations.length === 0) return state;\n  return {\n    ...state,\n    a2uiState: applyA2uiOperations(state.a2uiState, operations).state,\n  };\n};\n\nconst applyMessage = (\n  state: A2AThreadState,\n  message: A2AMessage,\n  replay: boolean,\n): A2AThreadState => {\n  if (\n    message.messageId &&\n    replay &&\n    state.a2uiMessageIds.has(message.messageId)\n  )\n    return state;\n  const next = message.messageId\n    ? {\n        ...state,\n        a2uiMessageIds: new Set([...state.a2uiMessageIds, message.messageId]),\n      }\n    : state;\n  return applyParts(next, message.parts);\n};\n\nexport const reduceA2AThreadState = (\n  state: A2AThreadState,\n  action: A2AThreadAction,\n): A2AThreadState => {\n  switch (action.type) {\n    case \"statusUpdate\": {\n      const { event } = action;\n      const currentTask = state.currentTask\n        ? { ...state.currentTask, status: event.status }\n        : {\n            id: event.taskId,\n            contextId: event.contextId,\n            status: event.status,\n          };\n      const next = {\n        ...state,\n        currentTask,\n        contextId: event.contextId || state.contextId,\n      };\n      return event.status.message\n        ? applyMessage(next, event.status.message, false)\n        : next;\n    }\n    case \"artifactUpdate\": {\n      const { artifact: input, append } = action.event;\n      const artifact = normalizeArtifact(input);\n      const index = state.currentArtifacts.findIndex(\n        (current) => current.artifactId === artifact.artifactId,\n      );\n      const updated =\n        index >= 0 && append\n          ? {\n              ...state.currentArtifacts[index]!,\n              parts: [\n                ...state.currentArtifacts[index]!.parts,\n                ...artifact.parts,\n              ],\n            }\n          : artifact;\n      const currentArtifacts =\n        index >= 0\n          ? [\n              ...state.currentArtifacts.slice(0, index),\n              updated,\n              ...state.currentArtifacts.slice(index + 1),\n            ]\n          : [...state.currentArtifacts, updated];\n      return applyParts({ ...state, currentArtifacts }, artifact.parts);\n    }\n    case \"taskSnapshot\": {\n      const { task } = action;\n      const artifacts =\n        task.artifacts === undefined\n          ? undefined\n          : Array.isArray(task.artifacts)\n            ? task.artifacts.map(normalizeArtifact)\n            : [];\n      const history =\n        task.history === undefined\n          ? undefined\n          : Array.isArray(task.history)\n            ? task.history\n            : [];\n      const complete = history !== undefined && artifacts !== undefined;\n      const artifactsToApply = complete\n        ? artifacts\n        : artifacts?.filter(\n            (artifact) =>\n              !state.currentArtifacts.some(\n                ({ artifactId }) => artifactId === artifact.artifactId,\n              ),\n          );\n      let next: A2AThreadState = {\n        ...state,\n        currentTask: {\n          ...task,\n          ...(artifacts === undefined ? {} : { artifacts }),\n          ...(history === undefined ? {} : { history }),\n        },\n        contextId: task.contextId || state.contextId,\n        currentArtifacts: complete\n          ? artifacts!\n          : artifactsToApply\n            ? [...state.currentArtifacts, ...artifactsToApply]\n            : state.currentArtifacts,\n        a2uiState: complete ? new Map() : state.a2uiState,\n        a2uiMessageIds: complete ? new Set() : state.a2uiMessageIds,\n      };\n      for (const message of history ?? []) {\n        if (message.role === \"agent\") next = applyMessage(next, message, true);\n      }\n      for (const artifact of artifactsToApply ?? []) {\n        next = applyParts(next, artifact.parts);\n      }\n      if (task.status.message) {\n        next = applyMessage(next, task.status.message, true);\n      }\n      return next;\n    }\n    case \"a2uiMessage\":\n      return applyMessage(state, action.message, action.replay ?? false);\n    case \"prepareRun\":\n      return {\n        ...state,\n        currentTask:\n          state.currentTask &&\n          isTerminalTaskState(state.currentTask.status.state)\n            ? undefined\n            : state.currentTask,\n        currentArtifacts: [],\n        a2uiState: new Map(),\n        a2uiMessageIds: new Set(),\n      };\n    case \"runStarted\":\n      return state.isRunning ? state : { ...state, isRunning: true };\n    case \"runFinished\":\n      return !state.isRunning ? state : { ...state, isRunning: false };\n    case \"agentCardLoaded\":\n      return state.agentCard === action.agentCard\n        ? state\n        : { ...state, agentCard: action.agentCard };\n    case \"loadRequested\":\n      return state.loadRequested ? state : { ...state, loadRequested: true };\n    case \"loadStarted\":\n      return state.isLoading ? state : { ...state, isLoading: true };\n    case \"loadFinished\":\n      return !state.isLoading ? state : { ...state, isLoading: false };\n    case \"contextLearned\":\n      return state.contextId === action.contextId\n        ? state\n        : { ...state, contextId: action.contextId };\n    case \"resetContext\":\n      return !state.isLoading && state.contextId === action.contextId\n        ? state\n        : { ...state, isLoading: false, contextId: action.contextId };\n    case \"taskReplaced\":\n      return state.currentTask === action.task\n        ? state\n        : { ...state, currentTask: action.task };\n    case \"reset\":\n      return {\n        ...state,\n        currentTask: undefined,\n        currentArtifacts: [],\n        a2uiState: new Map(),\n        a2uiMessageIds: new Set(),\n      };\n  }\n};\n"],"mappings":";;;AA4CA,MAAa,wBAAwB,eAAwC;CAC3E,aAAa,KAAA;CACb,kBAAkB,CAAC;CACnB,2BAAW,IAAI,IAAI;CACnB,gCAAgB,IAAI,IAAI;CACxB,WAAW,KAAA;CACX,WAAW;CACX,WAAW;CACX,eAAe;CACf;AACF;AAEA,MAAM,qBAAqB,cAAwC;CACjE,GAAG;CACH,OAAO,MAAM,QAAQ,SAAS,KAAK,IAAI,SAAS,QAAQ,CAAC;AAC3D;AAEA,MAAM,cACJ,OACA,UACmB;CACnB,MAAM,aAAa,sBAAsB,KAAK;CAC9C,IAAI,WAAW,WAAW,GAAG,OAAO;CACpC,OAAO;EACL,GAAG;EACH,WAAW,oBAAoB,MAAM,WAAW,UAAU,CAAC,CAAC;CAC9D;AACF;AAEA,MAAM,gBACJ,OACA,SACA,WACmB;CACnB,IACE,QAAQ,aACR,UACA,MAAM,eAAe,IAAI,QAAQ,SAAS,GAE1C,OAAO;CACT,MAAM,OAAO,QAAQ,YACjB;EACE,GAAG;EACH,gCAAgB,IAAI,IAAI,CAAC,GAAG,MAAM,gBAAgB,QAAQ,SAAS,CAAC;CACtE,IACA;CACJ,OAAO,WAAW,MAAM,QAAQ,KAAK;AACvC;AAEA,MAAa,wBACX,OACA,WACmB;CACnB,QAAQ,OAAO,MAAf;EACE,KAAK,gBAAgB;GACnB,MAAM,EAAE,UAAU;GAClB,MAAM,cAAc,MAAM,cACtB;IAAE,GAAG,MAAM;IAAa,QAAQ,MAAM;GAAO,IAC7C;IACE,IAAI,MAAM;IACV,WAAW,MAAM;IACjB,QAAQ,MAAM;GAChB;GACJ,MAAM,OAAO;IACX,GAAG;IACH;IACA,WAAW,MAAM,aAAa,MAAM;GACtC;GACA,OAAO,MAAM,OAAO,UAChB,aAAa,MAAM,MAAM,OAAO,SAAS,KAAK,IAC9C;EACN;EACA,KAAK,kBAAkB;GACrB,MAAM,EAAE,UAAU,OAAO,WAAW,OAAO;GAC3C,MAAM,WAAW,kBAAkB,KAAK;GACxC,MAAM,QAAQ,MAAM,iBAAiB,WAClC,YAAY,QAAQ,eAAe,SAAS,UAC/C;GACA,MAAM,UACJ,SAAS,KAAK,SACV;IACE,GAAG,MAAM,iBAAiB;IAC1B,OAAO,CACL,GAAG,MAAM,iBAAiB,MAAM,CAAE,OAClC,GAAG,SAAS,KACd;GACF,IACA;GACN,MAAM,mBACJ,SAAS,IACL;IACE,GAAG,MAAM,iBAAiB,MAAM,GAAG,KAAK;IACxC;IACA,GAAG,MAAM,iBAAiB,MAAM,QAAQ,CAAC;GAC3C,IACA,CAAC,GAAG,MAAM,kBAAkB,OAAO;GACzC,OAAO,WAAW;IAAE,GAAG;IAAO;GAAiB,GAAG,SAAS,KAAK;EAClE;EACA,KAAK,gBAAgB;GACnB,MAAM,EAAE,SAAS;GACjB,MAAM,YACJ,KAAK,cAAc,KAAA,IACf,KAAA,IACA,MAAM,QAAQ,KAAK,SAAS,IAC1B,KAAK,UAAU,IAAI,iBAAiB,IACpC,CAAC;GACT,MAAM,UACJ,KAAK,YAAY,KAAA,IACb,KAAA,IACA,MAAM,QAAQ,KAAK,OAAO,IACxB,KAAK,UACL,CAAC;GACT,MAAM,WAAW,YAAY,KAAA,KAAa,cAAc,KAAA;GACxD,MAAM,mBAAmB,WACrB,YACA,WAAW,QACR,aACC,CAAC,MAAM,iBAAiB,MACrB,EAAE,iBAAiB,eAAe,SAAS,UAC9C,CACJ;GACJ,IAAI,OAAuB;IACzB,GAAG;IACH,aAAa;KACX,GAAG;KACH,GAAI,cAAc,KAAA,IAAY,CAAC,IAAI,EAAE,UAAU;KAC/C,GAAI,YAAY,KAAA,IAAY,CAAC,IAAI,EAAE,QAAQ;IAC7C;IACA,WAAW,KAAK,aAAa,MAAM;IACnC,kBAAkB,WACd,YACA,mBACE,CAAC,GAAG,MAAM,kBAAkB,GAAG,gBAAgB,IAC/C,MAAM;IACZ,WAAW,2BAAW,IAAI,IAAI,IAAI,MAAM;IACxC,gBAAgB,2BAAW,IAAI,IAAI,IAAI,MAAM;GAC/C;GACA,KAAK,MAAM,WAAW,WAAW,CAAC,GAChC,IAAI,QAAQ,SAAS,SAAS,OAAO,aAAa,MAAM,SAAS,IAAI;GAEvE,KAAK,MAAM,YAAY,oBAAoB,CAAC,GAC1C,OAAO,WAAW,MAAM,SAAS,KAAK;GAExC,IAAI,KAAK,OAAO,SACd,OAAO,aAAa,MAAM,KAAK,OAAO,SAAS,IAAI;GAErD,OAAO;EACT;EACA,KAAK,eACH,OAAO,aAAa,OAAO,OAAO,SAAS,OAAO,UAAU,KAAK;EACnE,KAAK,cACH,OAAO;GACL,GAAG;GACH,aACE,MAAM,eACN,oBAAoB,MAAM,YAAY,OAAO,KAAK,IAC9C,KAAA,IACA,MAAM;GACZ,kBAAkB,CAAC;GACnB,2BAAW,IAAI,IAAI;GACnB,gCAAgB,IAAI,IAAI;EAC1B;EACF,KAAK,cACH,OAAO,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAK;EAC/D,KAAK,eACH,OAAO,CAAC,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAM;EACjE,KAAK,mBACH,OAAO,MAAM,cAAc,OAAO,YAC9B,QACA;GAAE,GAAG;GAAO,WAAW,OAAO;EAAU;EAC9C,KAAK,iBACH,OAAO,MAAM,gBAAgB,QAAQ;GAAE,GAAG;GAAO,eAAe;EAAK;EACvE,KAAK,eACH,OAAO,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAK;EAC/D,KAAK,gBACH,OAAO,CAAC,MAAM,YAAY,QAAQ;GAAE,GAAG;GAAO,WAAW;EAAM;EACjE,KAAK,kBACH,OAAO,MAAM,cAAc,OAAO,YAC9B,QACA;GAAE,GAAG;GAAO,WAAW,OAAO;EAAU;EAC9C,KAAK,gBACH,OAAO,CAAC,MAAM,aAAa,MAAM,cAAc,OAAO,YAClD,QACA;GAAE,GAAG;GAAO,WAAW;GAAO,WAAW,OAAO;EAAU;EAChE,KAAK,gBACH,OAAO,MAAM,gBAAgB,OAAO,OAChC,QACA;GAAE,GAAG;GAAO,aAAa,OAAO;EAAK;EAC3C,KAAK,SACH,OAAO;GACL,GAAG;GACH,aAAa,KAAA;GACb,kBAAkB,CAAC;GACnB,2BAAW,IAAI,IAAI;GACnB,gCAAgB,IAAI,IAAI;EAC1B;CACJ;AACF"}

package/dist/useA2ARuntime.js

@@ -1,6 +1,6 @@
 "use client";
 import { A2AClient } from "./A2AClient.js";
-import { A2AThreadRuntimeCore } from "./A2AThreadRuntimeCore.js";
+import { A2AThreadController } from "./A2AThreadController.js";
 import { a2aExtras } from "./a2aExtras.js";
 import { useExternalStoreRuntime, useExternalStoreSharedOptions, useRuntimeAdapters } from "@assistant-ui/core/react";
 import { useCallback, useEffect, useInsertionEffect, useMemo, useRef, useState, useSyncExternalStore } from "react";
@@ -66,7 +66,7 @@ function useA2ARuntime(options) {
 	};
 	const coreOptionsRef = useRef(coreOptions);
 	coreOptionsRef.current = coreOptions;
-	const createCore = () => new A2AThreadRuntimeCore({
+	const createCore = () => new A2AThreadController({
 		...coreOptionsRef.current,
 		client
 	});

package/dist/useA2ARuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useA2ARuntime.js","names":[],"sources":["../src/useA2ARuntime.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport { useReplaySafeEffect } from \"@assistant-ui/store/internal\";\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 runtimeAdapters = useRuntimeAdapters();\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  const headersRef = useRef(options.headers);\n  useInsertionEffect(() => {\n    headersRef.current = options.headers;\n  });\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 createClient = () => {\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  };\n  const [pinnedClient, setPinnedClient] = useState(() => ({\n    key: managedClientOptionsKey,\n    provided: options.client,\n    client: createClient(),\n  }));\n  let currentClient = pinnedClient;\n  if (\n    pinnedClient.key !== managedClientOptionsKey ||\n    pinnedClient.provided !== options.client\n  ) {\n    currentClient = {\n      key: managedClientOptionsKey,\n      provided: options.client,\n      client: createClient(),\n    };\n    setPinnedClient(currentClient);\n  }\n  const client = currentClient.client;\n\n  const coreOptions = {\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  const coreOptionsRef = useRef(coreOptions);\n  coreOptionsRef.current = coreOptions;\n\n  const createCore = () =>\n    new A2AThreadRuntimeCore({\n      ...coreOptionsRef.current,\n      client,\n    });\n  const [pinnedCore, setPinnedCore] = useState(() => ({\n    client,\n    core: createCore(),\n  }));\n  let currentCore = pinnedCore;\n  if (pinnedCore.client !== client) {\n    currentCore = { client, core: createCore() };\n    setPinnedCore(currentCore);\n  }\n  const core = currentCore.core;\n  const snapshot = useSyncExternalStore(\n    core.subscribe,\n    core.getSnapshot,\n    core.getSnapshot,\n  );\n\n  useEffect(() => {\n    core.updateOptions(coreOptions);\n  });\n\n  // Thread list\n  const threadSwitchGenerationRef = useRef(0);\n  const switchingGenerationRef = useRef<number | null>(null);\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    const releaseSwitch = (generation: number) => {\n      if (switchingGenerationRef.current === generation) {\n        switchingGenerationRef.current = null;\n      }\n    };\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetContext();\n              await onSwitchToNewThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              // Apply first so the abort inside resetContext finds an already\n              // cleared repository and cannot persist the old thread's partial\n              // assistant message.\n              core.applyExternalMessages([]);\n              core.resetContext();\n            } finally {\n              releaseSwitch(generation);\n            }\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetContext();\n              const result = await onSwitchToThread(threadId);\n              if (generation !== threadSwitchGenerationRef.current) return;\n              core.applyExternalMessages([]);\n              core.applyExternalMessages(result.messages);\n              core.resetContext();\n            } finally {\n              releaseSwitch(generation);\n            }\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    return {\n      ...shared,\n      isLoading: snapshot.isLoading,\n      messageRepository: snapshot.messageRepository,\n      isRunning: snapshot.isRunning,\n      unstable_persistsHistory: true,\n      extras: a2aExtras.provide({\n        task: snapshot.task,\n        artifacts: snapshot.artifacts,\n        agentCard: snapshot.agentCard,\n      }),\n      onNew: (message: AppendMessage) =>\n        switchingGenerationRef.current === null\n          ? core.append(message)\n          : Promise.resolve(),\n      onVoiceTranscript: (message: ThreadMessage) =>\n        core.appendVoiceTranscript(message),\n      onEdit: (message: AppendMessage) =>\n        switchingGenerationRef.current === null\n          ? core.edit(message)\n          : Promise.resolve(),\n      onReload: (parentId: string | null) =>\n        switchingGenerationRef.current === null\n          ? core.reload(parentId)\n          : Promise.resolve(),\n      onCancel: () => core.cancel(),\n      unstable_onRecordToolInteraction: (options) =>\n        core.recordToolInteraction(options),\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, snapshot, shared]);\n\n  const runtime = useExternalStoreRuntime(store);\n\n  useReplaySafeEffect(() => {\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":";;;;;;;;AA8BA,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,kBAAkB,mBAAmB;CAC3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,MAAM,aAAa,OAAO,QAAQ,OAAO;CACzC,yBAAyB;EACvB,WAAW,UAAU,QAAQ;CAC/B,CAAC;CACD,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,qBAAqB;EACzB,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;CACA,MAAM,CAAC,cAAc,mBAAmB,gBAAgB;EACtD,KAAK;EACL,UAAU,QAAQ;EAClB,QAAQ,aAAa;CACvB,EAAE;CACF,IAAI,gBAAgB;CACpB,IACE,aAAa,QAAQ,2BACrB,aAAa,aAAa,QAAQ,QAClC;EACA,gBAAgB;GACd,KAAK;GACL,UAAU,QAAQ;GAClB,QAAQ,aAAa;EACvB;EACA,gBAAgB,aAAa;CAC/B;CACA,MAAM,SAAS,cAAc;CAE7B,MAAM,cAAc;EAClB;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;CACA,MAAM,iBAAiB,OAAO,WAAW;CACzC,eAAe,UAAU;CAEzB,MAAM,mBACJ,IAAI,qBAAqB;EACvB,GAAG,eAAe;EAClB;CACF,CAAC;CACH,MAAM,CAAC,YAAY,iBAAiB,gBAAgB;EAClD;EACA,MAAM,WAAW;CACnB,EAAE;CACF,IAAI,cAAc;CAClB,IAAI,WAAW,WAAW,QAAQ;EAChC,cAAc;GAAE;GAAQ,MAAM,WAAW;EAAE;EAC3C,cAAc,WAAW;CAC3B;CACA,MAAM,OAAO,YAAY;CACzB,MAAM,WAAW,qBACf,KAAK,WACL,KAAK,aACL,KAAK,WACP;CAEA,gBAAgB;EACd,KAAK,cAAc,WAAW;CAChC,CAAC;CAGD,MAAM,4BAA4B,OAAO,CAAC;CAC1C,MAAM,yBAAyB,OAAsB,IAAI;CACzD,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,MAAM,iBAAiB,eAAuB;GAC5C,IAAI,uBAAuB,YAAY,YACrC,uBAAuB,UAAU;EAErC;EAEA,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KAGF,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,aAAa;KAClB,MAAM,oBAAoB;KAC1B,IAAI,eAAe,0BAA0B,SAAS;KAItD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,aAAa;IACpB,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KAGF,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,aAAa;KAClB,MAAM,SAAS,MAAM,iBAAiB,QAAQ;KAC9C,IAAI,eAAe,0BAA0B,SAAS;KACtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,sBAAsB,OAAO,QAAQ;KAC1C,KAAK,aAAa;IACpB,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,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;EAC1B,OAAO;GACL,GAAG;GACH,WAAW,SAAS;GACpB,mBAAmB,SAAS;GAC5B,WAAW,SAAS;GACpB,0BAA0B;GAC1B,QAAQ,UAAU,QAAQ;IACxB,MAAM,SAAS;IACf,WAAW,SAAS;IACpB,WAAW,SAAS;GACtB,CAAC;GACD,QAAQ,YACN,uBAAuB,YAAY,OAC/B,KAAK,OAAO,OAAO,IACnB,QAAQ,QAAQ;GACtB,oBAAoB,YAClB,KAAK,sBAAsB,OAAO;GACpC,SAAS,YACP,uBAAuB,YAAY,OAC/B,KAAK,KAAK,OAAO,IACjB,QAAQ,QAAQ;GACtB,WAAW,aACT,uBAAuB,YAAY,OAC/B,KAAK,OAAO,QAAQ,IACpB,QAAQ,QAAQ;GACtB,gBAAgB,KAAK,OAAO;GAC5B,mCAAmC,YACjC,KAAK,sBAAsB,OAAO;GACpC,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,0BAA0B;EACxB,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 {\n  useCallback,\n  useEffect,\n  useInsertionEffect,\n  useMemo,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport { useReplaySafeEffect } from \"@assistant-ui/store/internal\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport { A2AClient, type A2AClientOptions } from \"./A2AClient\";\nimport { A2AThreadController } from \"./A2AThreadController\";\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 runtimeAdapters = useRuntimeAdapters();\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  const headersRef = useRef(options.headers);\n  useInsertionEffect(() => {\n    headersRef.current = options.headers;\n  });\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 createClient = () => {\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  };\n  const [pinnedClient, setPinnedClient] = useState(() => ({\n    key: managedClientOptionsKey,\n    provided: options.client,\n    client: createClient(),\n  }));\n  let currentClient = pinnedClient;\n  if (\n    pinnedClient.key !== managedClientOptionsKey ||\n    pinnedClient.provided !== options.client\n  ) {\n    currentClient = {\n      key: managedClientOptionsKey,\n      provided: options.client,\n      client: createClient(),\n    };\n    setPinnedClient(currentClient);\n  }\n  const client = currentClient.client;\n\n  const coreOptions = {\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  const coreOptionsRef = useRef(coreOptions);\n  coreOptionsRef.current = coreOptions;\n\n  const createCore = () =>\n    new A2AThreadController({\n      ...coreOptionsRef.current,\n      client,\n    });\n  const [pinnedCore, setPinnedCore] = useState(() => ({\n    client,\n    core: createCore(),\n  }));\n  let currentCore = pinnedCore;\n  if (pinnedCore.client !== client) {\n    currentCore = { client, core: createCore() };\n    setPinnedCore(currentCore);\n  }\n  const core = currentCore.core;\n  const snapshot = useSyncExternalStore(\n    core.subscribe,\n    core.getSnapshot,\n    core.getSnapshot,\n  );\n\n  useEffect(() => {\n    core.updateOptions(coreOptions);\n  });\n\n  // Thread list\n  const threadSwitchGenerationRef = useRef(0);\n  const switchingGenerationRef = useRef<number | null>(null);\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread } = threadListAdapter;\n\n    const releaseSwitch = (generation: number) => {\n      if (switchingGenerationRef.current === generation) {\n        switchingGenerationRef.current = null;\n      }\n    };\n\n    return {\n      threadId: threadListAdapter.threadId,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetContext();\n              await onSwitchToNewThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              // Apply first so the abort inside resetContext finds an already\n              // cleared repository and cannot persist the old thread's partial\n              // assistant message.\n              core.applyExternalMessages([]);\n              core.resetContext();\n            } finally {\n              releaseSwitch(generation);\n            }\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetContext();\n              const result = await onSwitchToThread(threadId);\n              if (generation !== threadSwitchGenerationRef.current) return;\n              core.applyExternalMessages([]);\n              core.applyExternalMessages(result.messages);\n              core.resetContext();\n            } finally {\n              releaseSwitch(generation);\n            }\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    return {\n      ...shared,\n      isLoading: snapshot.isLoading,\n      messageRepository: snapshot.messageRepository,\n      isRunning: snapshot.isRunning,\n      unstable_persistsHistory: true,\n      extras: a2aExtras.provide({\n        task: snapshot.task,\n        artifacts: snapshot.artifacts,\n        agentCard: snapshot.agentCard,\n      }),\n      onNew: (message: AppendMessage) =>\n        switchingGenerationRef.current === null\n          ? core.append(message)\n          : Promise.resolve(),\n      onVoiceTranscript: (message: ThreadMessage) =>\n        core.appendVoiceTranscript(message),\n      onEdit: (message: AppendMessage) =>\n        switchingGenerationRef.current === null\n          ? core.edit(message)\n          : Promise.resolve(),\n      onReload: (parentId: string | null) =>\n        switchingGenerationRef.current === null\n          ? core.reload(parentId)\n          : Promise.resolve(),\n      onCancel: () => core.cancel(),\n      unstable_onRecordToolInteraction: (options) =>\n        core.recordToolInteraction(options),\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, snapshot, shared]);\n\n  const runtime = useExternalStoreRuntime(store);\n\n  useReplaySafeEffect(() => {\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":";;;;;;;;AA8BA,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,kBAAkB,mBAAmB;CAC3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,MAAM,aAAa,OAAO,QAAQ,OAAO;CACzC,yBAAyB;EACvB,WAAW,UAAU,QAAQ;CAC/B,CAAC;CACD,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,qBAAqB;EACzB,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;CACA,MAAM,CAAC,cAAc,mBAAmB,gBAAgB;EACtD,KAAK;EACL,UAAU,QAAQ;EAClB,QAAQ,aAAa;CACvB,EAAE;CACF,IAAI,gBAAgB;CACpB,IACE,aAAa,QAAQ,2BACrB,aAAa,aAAa,QAAQ,QAClC;EACA,gBAAgB;GACd,KAAK;GACL,UAAU,QAAQ;GAClB,QAAQ,aAAa;EACvB;EACA,gBAAgB,aAAa;CAC/B;CACA,MAAM,SAAS,cAAc;CAE7B,MAAM,cAAc;EAClB;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;CACA,MAAM,iBAAiB,OAAO,WAAW;CACzC,eAAe,UAAU;CAEzB,MAAM,mBACJ,IAAI,oBAAoB;EACtB,GAAG,eAAe;EAClB;CACF,CAAC;CACH,MAAM,CAAC,YAAY,iBAAiB,gBAAgB;EAClD;EACA,MAAM,WAAW;CACnB,EAAE;CACF,IAAI,cAAc;CAClB,IAAI,WAAW,WAAW,QAAQ;EAChC,cAAc;GAAE;GAAQ,MAAM,WAAW;EAAE;EAC3C,cAAc,WAAW;CAC3B;CACA,MAAM,OAAO,YAAY;CACzB,MAAM,WAAW,qBACf,KAAK,WACL,KAAK,aACL,KAAK,WACP;CAEA,gBAAgB;EACd,KAAK,cAAc,WAAW;CAChC,CAAC;CAGD,MAAM,4BAA4B,OAAO,CAAC;CAC1C,MAAM,yBAAyB,OAAsB,IAAI;CACzD,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,qBAAqB;EAElD,MAAM,iBAAiB,eAAuB;GAC5C,IAAI,uBAAuB,YAAY,YACrC,uBAAuB,UAAU;EAErC;EAEA,OAAO;GACL,UAAU,kBAAkB;GAC5B,qBAAqB,sBACjB,YAAY;IACV,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KAGF,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,aAAa;KAClB,MAAM,oBAAoB;KAC1B,IAAI,eAAe,0BAA0B,SAAS;KAItD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,aAAa;IACpB,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KAGF,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,aAAa;KAClB,MAAM,SAAS,MAAM,iBAAiB,QAAQ;KAC9C,IAAI,eAAe,0BAA0B,SAAS;KACtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,sBAAsB,OAAO,QAAQ;KAC1C,KAAK,aAAa;IACpB,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,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;EAC1B,OAAO;GACL,GAAG;GACH,WAAW,SAAS;GACpB,mBAAmB,SAAS;GAC5B,WAAW,SAAS;GACpB,0BAA0B;GAC1B,QAAQ,UAAU,QAAQ;IACxB,MAAM,SAAS;IACf,WAAW,SAAS;IACpB,WAAW,SAAS;GACtB,CAAC;GACD,QAAQ,YACN,uBAAuB,YAAY,OAC/B,KAAK,OAAO,OAAO,IACnB,QAAQ,QAAQ;GACtB,oBAAoB,YAClB,KAAK,sBAAsB,OAAO;GACpC,SAAS,YACP,uBAAuB,YAAY,OAC/B,KAAK,KAAK,OAAO,IACjB,QAAQ,QAAQ;GACtB,WAAW,aACT,uBAAuB,YAAY,OAC/B,KAAK,OAAO,QAAQ,IACpB,QAAQ,QAAQ;GACtB,gBAAgB,KAAK,OAAO;GAC5B,mCAAmC,YACjC,KAAK,sBAAsB,OAAO;GACpC,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,0BAA0B;EACxB,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.46",
+  "version": "0.2.48",
   "description": "A2A (Agent-to-Agent) v1.0 protocol adapter for assistant-ui",
   "keywords": [
     "a2a",
@@ -29,10 +29,10 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.29",
-    "@assistant-ui/generative-ui": "^0.0.5",
-    "@assistant-ui/store": "^0.3.21",
-    "assistant-stream": "^0.3.52"
+    "@assistant-ui/core": "^0.3.31",
+    "@assistant-ui/generative-ui": "^0.0.7",
+    "@assistant-ui/store": "^0.3.22",
+    "assistant-stream": "^0.3.54"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -44,15 +44,15 @@
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.30",
+    "@assistant-ui/x-buildutils": "0.0.31",
     "@assistant-ui/x-react18": "0.0.0",
     "@testing-library/dom": "^10.4.2",
     "@testing-library/react": "^16.3.3",
     "@types/react": "^19.3.0",
-    "jsdom": "^30.1.2",
-    "react": "^19.3.0",
-    "react-dom": "^19.3.0",
-    "vitest": "^5.0.3"
+    "jsdom": "^30.1.3",
+    "react": "^19.3.1",
+    "react-dom": "^19.3.1",
+    "vitest": "^5.0.5"
   },
   "publishConfig": {
     "access": "public",

package/src/{A2AThreadRuntimeCore.test.ts → A2AThreadController.test.ts}

@@ -3,7 +3,7 @@ import {
   applyA2uiOperations,
   convertSurfaceToUISpec,
 } from "@assistant-ui/generative-ui/a2ui";
-import { A2AThreadRuntimeCore } from "./A2AThreadRuntimeCore";
+import { A2AThreadController } from "./A2AThreadController";
 import type { A2AClient } from "./A2AClient";
 import type {
   A2AAgentCard,
@@ -154,7 +154,7 @@ const toolInteraction = {
   payload: { value: "selected" },
 };
 
-describe("A2AThreadRuntimeCore", () => {
+describe("A2AThreadController", () => {
   let onUpdate = vi.fn<() => void>();
 
   beforeEach(() => {
@@ -169,7 +169,7 @@ describe("A2AThreadRuntimeCore", () => {
     clientOverrides: Partial<A2AClient> = {},
     coreOverrides: Record<string, unknown> = {},
   ) {
-    const core = new A2AThreadRuntimeCore({
+    const core = new A2AThreadController({
       client: createMockClient(clientOverrides),
       ...coreOverrides,
     });
@@ -634,7 +634,7 @@ describe("A2AThreadRuntimeCore", () => {
         yield statusUpdateEvent("completed", "Answer");
       });
       const client = createMockClient({ streamMessage });
-      const core = new A2AThreadRuntimeCore({
+      const core = new A2AThreadController({
         client,
       });
 
@@ -653,7 +653,7 @@ describe("A2AThreadRuntimeCore", () => {
         yield statusUpdateEvent("completed", "Answer");
       });
       const client = createMockClient({ streamMessage });
-      const core = new A2AThreadRuntimeCore({
+      const core = new A2AThreadController({
         client,
       });
 
@@ -706,7 +706,7 @@ describe("A2AThreadRuntimeCore", () => {
         yield statusUpdateEvent("completed", "Answer");
       });
       const client = createMockClient({ streamMessage });
-      const core = new A2AThreadRuntimeCore({
+      const core = new A2AThreadController({
         client,
       });
 
@@ -725,7 +725,7 @@ describe("A2AThreadRuntimeCore", () => {
         yield statusUpdateEvent("completed", "Answer");
       });
       const client = createMockClient({ streamMessage });
-      const core = new A2AThreadRuntimeCore({
+      const core = new A2AThreadController({
         client,
       });
 
@@ -1581,8 +1581,9 @@ describe("A2AThreadRuntimeCore", () => {
 
       const core = createCore({ sendMessage, streamMessage });
       // Simulate agent card with streaming: false
-      (core as any).agentCardValue = {
-        capabilities: { streaming: false },
+      (core as any).state = {
+        ...(core as any).state,
+        agentCard: { capabilities: { streaming: false } },
       };
 
       await core.append(createUserAppendMessage("Hello"));
@@ -1605,8 +1606,9 @@ describe("A2AThreadRuntimeCore", () => {
       } satisfies A2AMessage);
 
       const core = createCore({ sendMessage });
-      (core as any).agentCardValue = {
-        capabilities: { streaming: false },
+      (core as any).state = {
+        ...(core as any).state,
+        agentCard: { capabilities: { streaming: false } },
       };
 
       await core.append(createUserAppendMessage("Hello"));
@@ -1763,7 +1765,7 @@ describe("A2AThreadRuntimeCore", () => {
     it("calls onArtifactComplete when lastChunk=true", async () => {
       const onArtifactComplete = vi.fn();
 
-      const core = new A2AThreadRuntimeCore({
+      const core = new A2AThreadController({
         client: createMockClient({
           streamMessage: vi.fn().mockImplementation(async function* () {
             yield artifactUpdateEvent("a1", [{ text: "code" }], {
@@ -1781,6 +1783,32 @@ describe("A2AThreadRuntimeCore", () => {
       expect(onArtifactComplete.mock.calls[0]![0].artifactId).toBe("a1");
     });
 
+    it("passes the completed artifact even if a subscriber clears the thread", async () => {
+      const onArtifactComplete = vi.fn();
+
+      const core = new A2AThreadController({
+        client: createMockClient({
+          streamMessage: vi.fn().mockImplementation(async function* () {
+            yield artifactUpdateEvent("a1", [{ text: "code" }], {
+              lastChunk: true,
+            });
+          }),
+        }),
+        onArtifactComplete,
+      });
+      let cleared = false;
+      core.subscribe(() => {
+        if (cleared || core.getArtifacts().length === 0) return;
+        cleared = true;
+        core.applyExternalMessages([]);
+      });
+
+      await core.append(createUserAppendMessage("Go"));
+
+      expect(onArtifactComplete).toHaveBeenCalledTimes(1);
+      expect(onArtifactComplete.mock.calls[0]![0]?.artifactId).toBe("a1");
+    });
+
     it.each(["throws", "rejects"] as const)(
       "continues streaming when onArtifactComplete %s",
       async (failureMode) => {
@@ -2377,7 +2405,7 @@ describe("A2AThreadRuntimeCore", () => {
           signal.addEventListener("abort", () => resolve(), { once: true });
         });
       });
-      let core!: A2AThreadRuntimeCore;
+      let core!: A2AThreadController;
       let replacementRun: Promise<void> | undefined;
       let startReplacement = true;
       core = createCore(
@@ -2428,9 +2456,9 @@ describe("A2AThreadRuntimeCore", () => {
       const core = createCore({ cancelTask });
 
       // Manually set task state to simulate a running task
-      (core as any).currentTask = {
-        id: "t1",
-        status: { state: "working" },
+      (core as any).state = {
+        ...(core as any).state,
+        currentTask: { id: "t1", status: { state: "working" } },
       };
       (core as any).abortController = new AbortController();
       onUpdate.mockClear();
@@ -2556,7 +2584,7 @@ describe("A2AThreadRuntimeCore", () => {
         });
       });
       let restarted = false;
-      let core!: A2AThreadRuntimeCore;
+      let core!: A2AThreadController;
       core = createCore(
         { streamMessage },
         {
@@ -2621,7 +2649,7 @@ describe("A2AThreadRuntimeCore", () => {
             signal.addEventListener("abort", () => resolve(), { once: true });
         });
       });
-      let core!: A2AThreadRuntimeCore;
+      let core!: A2AThreadController;
       core = createCore(
         { cancelTask, streamMessage },
         { onCancel: () => core.applyExternalMessages([]) },
@@ -2704,7 +2732,7 @@ describe("A2AThreadRuntimeCore", () => {
     it("sets error status and re-throws on stream failure", async () => {
       const onError = vi.fn();
 
-      const core = new A2AThreadRuntimeCore({
+      const core = new A2AThreadController({
         client: createMockClient({
           streamMessage: vi.fn().mockImplementation(() => ({
             async next() {
@@ -2959,7 +2987,7 @@ describe("outbound message conversion", () => {
     const streamMessage = vi.fn().mockImplementation(async function* () {
       yield statusUpdateEvent("completed", "Done");
     });
-    const core = new A2AThreadRuntimeCore({
+    const core = new A2AThreadController({
       client: createMockClient({ streamMessage }),
     });
     return { core, streamMessage };