@copilotkit/core 1.68.2 → 1.68.3-canary.1787220377

package/dist/index.umd.js

@@ -3148,6 +3148,129 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		}
 	};
 
+//#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 {
+		constructor(core) {
+			this.core = core;
+			_defineProperty(this, "_subagents", {});
+			_defineProperty(this, "agentSubscriptions", /* @__PURE__ */ new Map());
+		}
+		/**
+		* 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) {
+			var _this$_subagents, _this$_subagents$agen;
+			const bucket = (_this$_subagents$agen = (_this$_subagents = this._subagents)[agentId]) !== null && _this$_subagents$agen !== void 0 ? _this$_subagents$agen : _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 === null || bucket === void 0 ? void 0 : bucket[event.subagentRunId];
+			if (!bucket || !existing) return;
+			bucket[event.subagentRunId] = {
+				...existing,
+				status: "finished"
+			};
+			this.notifySubagentsChanged(agentId);
+		}
+		handleError(agentId, event) {
+			var _this$_subagents2, _this$_subagents2$age;
+			const bucket = (_this$_subagents2$age = (_this$_subagents2 = this._subagents)[agentId]) !== null && _this$_subagents2$age !== void 0 ? _this$_subagents2$age : _this$_subagents2[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) => {
+				var _subscriber$onSubagen;
+				return (_subscriber$onSubagen = subscriber.onSubagentsChanged) === null || _subscriber$onSubagen === void 0 ? void 0 : _subscriber$onSubagen.call(subscriber, {
+					copilotkit: this.core,
+					agentId,
+					subagents
+				});
+			}, "Subscriber onSubagentsChanged error:");
+		}
+	};
+
 //#endregion
 //#region src/core/thread-store-registry.ts
 	var ThreadStoreRegistry = class {
@@ -4348,6 +4471,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			_defineProperty(this, "suggestionEngine", void 0);
 			_defineProperty(this, "runHandler", void 0);
 			_defineProperty(this, "stateManager", void 0);
+			_defineProperty(this, "subagentRegistry", void 0);
 			_defineProperty(this, "threadStoreRegistry", void 0);
 			_defineProperty(this, "_memoryStore", void 0);
 			_defineProperty(this, "previousAgentIds", /* @__PURE__ */ new Set());
@@ -4360,6 +4484,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			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);
@@ -4374,7 +4499,10 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				},
 				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 {
@@ -4382,10 +4510,17 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 					} 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;
 				}
@@ -4765,6 +4900,16 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			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) {
@@ -6108,6 +6253,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;