@copilotkit/core 1.68.3-canary.1787220377 → 1.69.0

package/dist/index.cjs

@@ -572,6 +572,31 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	}
 };
 
+//#endregion
+//#region src/utils/runtime-info-error.ts
+/**
+* Build the error for a failed runtime `/info` response, folding in whatever
+* explanation the runtime sent back.
+*
+* The status alone cannot distinguish a wrong `runtimeUrl` from a transport
+* mismatch, so the runtime answers the latter with a body naming the cause
+* (see the handler's `single_route_envelope_against_multi_route_runtime`
+* diagnostic). Discarding that body — as every `/info` caller used to — left
+* the developer with a bare 404 and nothing to act on (OSS-882).
+*
+* A body that isn't JSON, or carries no string `message`, contributes nothing:
+* the status is then the whole error.
+*/
+async function runtimeInfoError(response) {
+	const base = `Runtime info request failed with status ${response.status}`;
+	let detail;
+	try {
+		const message = (await response.clone().json())?.message;
+		if (typeof message === "string" && message.trim().length > 0) detail = message.trim();
+	} catch {}
+	return new Error(detail ? `${base}: ${detail}` : base);
+}
+
 //#endregion
 //#region src/agent.ts
 function hasHeaders(agent) {
@@ -861,7 +886,7 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 			headers,
 			...this.credentials ? { credentials: this.credentials } : {}
 		});
-		if (!response.ok) throw new Error(`Runtime info request failed with status ${response.status}`);
+		if (!response.ok) throw await runtimeInfoError(response);
 		return await response.json();
 	}
 	async fetchRuntimeInfoAutoDetect(headers) {
@@ -883,7 +908,7 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 			body: JSON.stringify({ method: "info" }),
 			...this.credentials ? { credentials: this.credentials } : {}
 		});
-		if (!response.ok) throw new Error(`Runtime info request failed with status ${response.status}`);
+		if (!response.ok) throw await runtimeInfoError(response);
 		this.transport = "single";
 		this.singleEndpointUrl = this.runtimeUrl;
 		return await response.json();
@@ -1492,7 +1517,7 @@ var AgentRegistry = class {
 			headers,
 			...credentials ? { credentials } : {}
 		});
-		if (!response.ok) throw new Error(`Runtime info request failed with status ${response.status}`);
+		if (!response.ok) throw await runtimeInfoError(response);
 		return {
 			runtimeInfo: await response.json(),
 			resolvedTransport: "rest"
@@ -1505,7 +1530,7 @@ var AgentRegistry = class {
 			body: JSON.stringify({ method: "info" }),
 			...credentials ? { credentials } : {}
 		});
-		if (!response.ok) throw new Error(`Runtime info request failed with status ${response.status}`);
+		if (!response.ok) throw await runtimeInfoError(response);
 		return await response.json();
 	}
 	/**
@@ -2778,6 +2803,7 @@ const isContinuation = (input) => input.resume !== void 0 || Object.prototype.ha
 var StateManager = class {
 	stateByRun = /* @__PURE__ */ new Map();
 	messageToRun = /* @__PURE__ */ new Map();
+	rawEventByMessage = /* @__PURE__ */ new Map();
 	activeRun = /* @__PURE__ */ new Map();
 	agentSubscriptions = /* @__PURE__ */ new Map();
 	pendingContinuations = /* @__PURE__ */ new WeakMap();
@@ -2829,12 +2855,13 @@ var StateManager = class {
 			runId: subRunId ?? input.runId
 		});
 		const { unsubscribe } = agent.subscribe({
-			onRunStartedEvent: ({ input, state }) => {
+			onRunStartedEvent: ({ event, input, state }) => {
 				if (revoked) return;
 				const internalContinuation = [...this.pendingContinuations.get(agent) ?? []].find((pending) => pending.expectedInput === input);
 				internalContinuation?.cancel();
-				if (runFinished && input.runId === subRunId && !internalContinuation && !isContinuation(input)) subRunId = (0, _ag_ui_client.randomUUID)();
-				else subRunId = internalContinuation?.expectedRunId ?? input.runId;
+				if (internalContinuation) subRunId = internalContinuation.expectedRunId ?? event.runId ?? input.runId;
+				else if (runFinished && input.runId === subRunId && !isContinuation(input) && (event.runId == null || event.runId === subRunId)) subRunId = (0, _ag_ui_client.randomUUID)();
+				else subRunId = event.runId || input.runId;
 				runFinished = false;
 				this.handleRunStarted(agent, effectiveInput(input), state);
 			},
@@ -2856,13 +2883,22 @@ var StateManager = class {
 				if (revoked) return;
 				this.handleStateDelta(agent, event, effectiveInput(input), state);
 			},
+			onTextMessageStartEvent: ({ event, input }) => {
+				if (revoked) return;
+				this.handleTextMessageStart(agent, event, effectiveInput(input));
+			},
 			onMessagesSnapshotEvent: ({ event, input, messages }) => {
 				if (revoked) return;
 				this.handleMessagesSnapshot(agent, event, effectiveInput(input), messages);
+				this.pruneRawEvents(agent.agentId, input.threadId, event.messages, effectiveInput(input));
 			},
 			onNewMessage: ({ message, input }) => {
 				if (revoked) return;
 				this.handleNewMessage(agent, message, input ? effectiveInput(input) : void 0);
+			},
+			onMessagesChanged: ({ messages, input }) => {
+				if (revoked) return;
+				if (!input) this.pruneRawEvents(agent.agentId, agent.threadId, messages);
 			}
 		});
 		this.agentSubscriptions.set(agentId, () => {
@@ -2880,6 +2916,7 @@ var StateManager = class {
 			unsubscribe();
 			this.agentSubscriptions.delete(agentId);
 		}
+		this.rawEventByMessage.delete(agentId);
 	}
 	/**
 	* Get state for a specific run
@@ -2897,6 +2934,13 @@ var StateManager = class {
 		return this.messageToRun.get(agentId)?.get(threadId)?.get(messageId);
 	}
 	/**
+	* Get direct text-start metadata associated with a message.
+	*/
+	getRawEventForMessage(agentId, threadId, messageId) {
+		const rawEvent = this.rawEventByMessage.get(agentId)?.get(threadId)?.get(messageId);
+		return rawEvent === void 0 ? void 0 : (0, _ag_ui_client.structuredClone_)(rawEvent);
+	}
+	/**
 	* Get all states for an agent's thread
 	*/
 	getStatesForThread(agentId, threadId) {
@@ -2948,12 +2992,30 @@ var StateManager = class {
 		this.saveState(agent.agentId, threadId, runId, state);
 	}
 	/**
+	* Capture only defined metadata from a normalized direct text-start event.
+	*/
+	handleTextMessageStart(agent, event, input) {
+		if (!agent.agentId) return;
+		const { threadId } = input;
+		if (event.rawEvent === void 0) {
+			const threadEvents = this.rawEventByMessage.get(agent.agentId)?.get(threadId);
+			threadEvents?.delete(event.messageId);
+			if (threadEvents?.size === 0) this.rawEventByMessage.get(agent.agentId)?.delete(threadId);
+			if (this.rawEventByMessage.get(agent.agentId)?.size === 0) this.rawEventByMessage.delete(agent.agentId);
+			return;
+		}
+		if (!this.rawEventByMessage.has(agent.agentId)) this.rawEventByMessage.set(agent.agentId, /* @__PURE__ */ new Map());
+		const agentEvents = this.rawEventByMessage.get(agent.agentId);
+		if (!agentEvents.has(threadId)) agentEvents.set(threadId, /* @__PURE__ */ new Map());
+		agentEvents.get(threadId).set(event.messageId, event.rawEvent);
+	}
+	/**
 	* Handle messages snapshot event
 	*/
 	handleMessagesSnapshot(agent, event, input, messages) {
 		if (!agent.agentId) return;
 		const { threadId, runId } = input;
-		for (const message of event.messages) this.associateMessageWithRun(agent.agentId, threadId, message.id, runId);
+		for (const message of event.messages) if (this.getRunIdForMessage(agent.agentId, threadId, message.id) === void 0) this.associateMessageWithRun(agent.agentId, threadId, message.id, runId);
 	}
 	/**
 	* Handle new message event
@@ -2987,12 +3049,23 @@ var StateManager = class {
 		if (!agentMessages.has(threadId)) agentMessages.set(threadId, /* @__PURE__ */ new Map());
 		agentMessages.get(threadId).set(messageId, runId);
 	}
+	pruneRawEvents(agentId, fallbackThreadId, messages, input) {
+		const threadId = input?.threadId ?? fallbackThreadId;
+		if (!threadId) return;
+		const threadEvents = this.rawEventByMessage.get(agentId)?.get(threadId);
+		if (!threadEvents) return;
+		const messageIds = new Set(messages.map((message) => message.id));
+		for (const messageId of threadEvents.keys()) if (!messageIds.has(messageId)) threadEvents.delete(messageId);
+		if (threadEvents.size === 0) this.rawEventByMessage.get(agentId)?.delete(threadId);
+		if (this.rawEventByMessage.get(agentId)?.size === 0) this.rawEventByMessage.delete(agentId);
+	}
 	/**
 	* Clear all state for an agent
 	*/
 	clearAgentState(agentId) {
 		this.stateByRun.delete(agentId);
 		this.messageToRun.delete(agentId);
+		this.rawEventByMessage.delete(agentId);
 	}
 	/**
 	* Clear all state for a thread
@@ -3000,124 +3073,7 @@ var StateManager = class {
 	clearThreadState(agentId, threadId) {
 		this.stateByRun.get(agentId)?.delete(threadId);
 		this.messageToRun.get(agentId)?.delete(threadId);
-	}
-};
-
-//#endregion
-//#region src/core/subagent-registry.ts
-/**
-* Tracks subagent lifecycle events (SUBAGENT_STARTED / FINISHED / ERROR) per
-* owning agent so consumers can resolve a message's `subagentRunId` to a name,
-* description, and running status.
-*
-* Modeled on {@link SuggestionEngine}: an in-memory `Record` keyed by (owning)
-* agentId, mutated from streamed AG-UI events, made reactive by calling
-* `notifySubscribers(onSubagentsChanged)` after every change. It owns its own
-* per-agent `agent.subscribe` (like {@link StateManager}) so it is fully
-* self-contained and adds no coupling to the state/message tracking path.
-*
-* No-op for runs without subagents (a normal LangGraph run never emits
-* SUBAGENT_* events), so the registry simply stays empty.
-*/
-var SubagentRegistry = class {
-	_subagents = {};
-	agentSubscriptions = /* @__PURE__ */ new Map();
-	constructor(core) {
-		this.core = core;
-	}
-	/**
-	* Subscribe to an agent's subagent lifecycle events. Idempotent per agentId:
-	* an existing subscription for the same agent is revoked and replaced.
-	*/
-	subscribeToAgent(agent) {
-		if (!agent.agentId) return;
-		const agentId = agent.agentId;
-		const existingUnsubscribe = this.agentSubscriptions.get(agentId);
-		if (existingUnsubscribe) {
-			existingUnsubscribe();
-			this.agentSubscriptions.delete(agentId);
-		}
-		let revoked = false;
-		const { unsubscribe } = agent.subscribe({
-			onSubagentStartedEvent: ({ event }) => {
-				if (revoked) return;
-				this.handleStarted(agentId, event);
-			},
-			onSubagentFinishedEvent: ({ event }) => {
-				if (revoked) return;
-				this.handleFinished(agentId, event);
-			},
-			onSubagentErrorEvent: ({ event }) => {
-				if (revoked) return;
-				this.handleError(agentId, event);
-			}
-		});
-		this.agentSubscriptions.set(agentId, () => {
-			revoked = true;
-			unsubscribe();
-		});
-	}
-	/**
-	* Unsubscribe an agent's subscription (symmetric with StateManager cleanup).
-	*/
-	unsubscribeFromAgent(agentId) {
-		const unsubscribe = this.agentSubscriptions.get(agentId);
-		if (unsubscribe) {
-			unsubscribe();
-			this.agentSubscriptions.delete(agentId);
-		}
-	}
-	/**
-	* Get the subagents observed for an owning agent, keyed by subagentRunId.
-	* Returns a fresh shallow copy each call so consumers (and React state) see
-	* a new reference when anything changed.
-	*/
-	getSubagents(agentId) {
-		return { ...this._subagents[agentId] };
-	}
-	handleStarted(agentId, event) {
-		const bucket = this._subagents[agentId] ??= {};
-		bucket[event.subagentRunId] = {
-			subagentRunId: event.subagentRunId,
-			name: event.name,
-			description: event.description,
-			parentSubagentRunId: event.parentSubagentRunId,
-			status: "running"
-		};
-		this.notifySubagentsChanged(agentId);
-	}
-	handleFinished(agentId, event) {
-		const bucket = this._subagents[agentId];
-		const existing = bucket?.[event.subagentRunId];
-		if (!bucket || !existing) return;
-		bucket[event.subagentRunId] = {
-			...existing,
-			status: "finished"
-		};
-		this.notifySubagentsChanged(agentId);
-	}
-	handleError(agentId, event) {
-		const bucket = this._subagents[agentId] ??= {};
-		const existing = bucket[event.subagentRunId];
-		bucket[event.subagentRunId] = existing ? {
-			...existing,
-			status: "error",
-			error: event.message
-		} : {
-			subagentRunId: event.subagentRunId,
-			name: event.subagentRunId,
-			status: "error",
-			error: event.message
-		};
-		this.notifySubagentsChanged(agentId);
-	}
-	async notifySubagentsChanged(agentId) {
-		const subagents = this.getSubagents(agentId);
-		await this.core.notifySubscribers((subscriber) => subscriber.onSubagentsChanged?.({
-			copilotkit: this.core,
-			agentId,
-			subagents
-		}), "Subscriber onSubagentsChanged error:");
+		this.rawEventByMessage.get(agentId)?.delete(threadId);
 	}
 };
 
@@ -4309,7 +4265,6 @@ var CopilotKitCore = class {
 	suggestionEngine;
 	runHandler;
 	stateManager;
-	subagentRegistry;
 	threadStoreRegistry;
 	/**
 	* The single core-owned memory store, created lazily on first
@@ -4335,7 +4290,6 @@ var CopilotKitCore = class {
 		this.suggestionEngine = new SuggestionEngine(this);
 		this.runHandler = new RunHandler(this);
 		this.stateManager = new StateManager(this);
-		this.subagentRegistry = new SubagentRegistry(this);
 		this.threadStoreRegistry = new ThreadStoreRegistry(this);
 		this.agentRegistry.initialize(agents__unsafe_dev_only);
 		this.runHandler.initialize(tools);
@@ -4350,10 +4304,7 @@ var CopilotKitCore = class {
 			},
 			onAgentsChanged: ({ agents }) => {
 				Object.values(agents).forEach((agent) => {
-					if (agent.agentId) {
-						this.stateManager.subscribeToAgent(agent);
-						this.subagentRegistry.subscribeToAgent(agent);
-					}
+					if (agent.agentId) this.stateManager.subscribeToAgent(agent);
 				});
 				const currentAgentIds = new Set(Object.keys(agents));
 				for (const agentId of Object.keys(this.threadStoreRegistry.getAll())) if (this.previousAgentIds.has(agentId) && !currentAgentIds.has(agentId)) try {
@@ -4361,17 +4312,10 @@ var CopilotKitCore = class {
 				} catch (err) {
 					console.error(`CopilotKitCore.onAgentsChanged: threadStoreRegistry.unregister failed for "${agentId}":`, err);
 				}
-				for (const agentId of this.previousAgentIds) if (!currentAgentIds.has(agentId)) {
-					try {
-						this.stateManager.unsubscribeFromAgent(agentId);
-					} catch (err) {
-						console.error(`CopilotKitCore.onAgentsChanged: stateManager.unsubscribeFromAgent failed for "${agentId}":`, err);
-					}
-					try {
-						this.subagentRegistry.unsubscribeFromAgent(agentId);
-					} catch (err) {
-						console.error(`CopilotKitCore.onAgentsChanged: subagentRegistry.unsubscribeFromAgent failed for "${agentId}":`, err);
-					}
+				for (const agentId of this.previousAgentIds) if (!currentAgentIds.has(agentId)) try {
+					this.stateManager.unsubscribeFromAgent(agentId);
+				} catch (err) {
+					console.error(`CopilotKitCore.onAgentsChanged: stateManager.unsubscribeFromAgent failed for "${agentId}":`, err);
 				}
 				this.previousAgentIds = currentAgentIds;
 			}
@@ -4741,16 +4685,6 @@ var CopilotKitCore = class {
 		return this.suggestionEngine.getSuggestions(agentId);
 	}
 	/**
-	* Subagent registry (delegated to SubagentRegistry).
-	*
-	* Returns the subagents observed for an owning agent during its run(s),
-	* keyed by `subagentRunId`, each carrying `name` / `description` / `status`.
-	* Empty for runs that never emit SUBAGENT_* events.
-	*/
-	getSubagents(agentId) {
-		return this.subagentRegistry.getSubagents(agentId);
-	}
-	/**
 	* Tool management (delegated to RunHandler)
 	*/
 	addTool(tool) {
@@ -4969,6 +4903,9 @@ var CopilotKitCore = class {
 	getRunIdForMessage(agentId, threadId, messageId) {
 		return this.stateManager.getRunIdForMessage(agentId, threadId, messageId);
 	}
+	getRawEventForMessage(agentId, threadId, messageId) {
+		return this.stateManager.getRawEventForMessage(agentId, threadId, messageId);
+	}
 	getRunIdsForThread(agentId, threadId) {
 		return this.stateManager.getRunIdsForThread(agentId, threadId);
 	}
@@ -5253,6 +5190,16 @@ function optionalString(value) {
 	return typeof value === "string" && value.length > 0 ? value : void 0;
 }
 /**
+* REST `/threads` is agent-scoped. The Phoenix `user_meta` channel is not.
+* Drop live upserts that belong to a different agent, or one Inspector
+* `all-agents` flatten shows the same thread twice until a hard refresh.
+*/
+function threadMetadataTargetsAgent(payload, agentId) {
+	if (!agentId || payload.operation === "deleted") return true;
+	const threadAgentId = optionalString(payload.thread.agentId);
+	return !threadAgentId || threadAgentId === agentId;
+}
+/**
 * Converts gateway run-activity payloads into the internal TypeScript shape.
 */
 function normalizeThreadRunActivityNotification(payload) {
@@ -5678,7 +5625,7 @@ function createThreadStore(environment) {
 					}))), ɵobservePhoenixJoinOutcome$(channel$).pipe((0, rxjs_operators.filter)((outcome) => outcome.type !== "joined"), (0, rxjs_operators.map)((outcome) => outcome.type === "timeout" ? threadSocketEvents.joinTimedOut({ sessionId: action.sessionId }) : threadSocketEvents.joinFailed({ sessionId: action.sessionId })))).pipe((0, rxjs_operators.takeUntil)((0, rxjs.merge)(shutdown$, fatalSocketShutdown$)));
 				});
 			}))),
-			createEffect((actions$, state$) => actions$.pipe(ofType(threadSocketEvents.metadataReceived), (0, rxjs_operators.withLatestFrom)(state$), (0, rxjs_operators.filter)(([action, state]) => action.sessionId === state.sessionId), (0, rxjs_operators.map)(([action, state]) => {
+			createEffect((actions$, state$) => actions$.pipe(ofType(threadSocketEvents.metadataReceived), (0, rxjs_operators.withLatestFrom)(state$), (0, rxjs_operators.filter)(([action, state]) => action.sessionId === state.sessionId && threadMetadataTargetsAgent(action.payload, state.context?.agentId)), (0, rxjs_operators.map)(([action, state]) => {
 				if (action.payload.operation === "deleted") return threadDomainEvents.threadDeleted({
 					sessionId: action.sessionId,
 					threadId: action.payload.deleted.id
@@ -6045,7 +5992,6 @@ exports.MemoryError = MemoryError;
 exports.ProxiedCopilotRuntimeAgent = ProxiedCopilotRuntimeAgent;
 exports.RunHandler = RunHandler;
 exports.StateManager = StateManager;
-exports.SubagentRegistry = SubagentRegistry;
 exports.SuggestionEngine = SuggestionEngine;
 exports.ToolCallStatus = ToolCallStatus;
 exports.completePartialMarkdown = completePartialMarkdown;