@copilotkit/core 1.68.1 → 1.68.3-canary.1786716392

package/dist/index.cjs

@@ -3003,6 +3003,124 @@ var StateManager = class {
 	}
 };
 
+//#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:");
+	}
+};
+
 //#endregion
 //#region src/core/thread-store-registry.ts
 var ThreadStoreRegistry = class {
@@ -4191,6 +4309,7 @@ var CopilotKitCore = class {
 	suggestionEngine;
 	runHandler;
 	stateManager;
+	subagentRegistry;
 	threadStoreRegistry;
 	/**
 	* The single core-owned memory store, created lazily on first
@@ -4216,6 +4335,7 @@ 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);
@@ -4230,7 +4350,10 @@ var CopilotKitCore = class {
 			},
 			onAgentsChanged: ({ agents }) => {
 				Object.values(agents).forEach((agent) => {
-					if (agent.agentId) this.stateManager.subscribeToAgent(agent);
+					if (agent.agentId) {
+						this.stateManager.subscribeToAgent(agent);
+						this.subagentRegistry.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 {
@@ -4238,10 +4361,17 @@ 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);
+				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);
+					}
 				}
 				this.previousAgentIds = currentAgentIds;
 			}
@@ -4611,6 +4741,16 @@ 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) {
@@ -5905,6 +6045,7 @@ exports.MemoryError = MemoryError;
 exports.ProxiedCopilotRuntimeAgent = ProxiedCopilotRuntimeAgent;
 exports.RunHandler = RunHandler;
 exports.StateManager = StateManager;
+exports.SubagentRegistry = SubagentRegistry;
 exports.SuggestionEngine = SuggestionEngine;
 exports.ToolCallStatus = ToolCallStatus;
 exports.completePartialMarkdown = completePartialMarkdown;