@copilotkit/core 1.66.2-canary.1785878925 → 1.66.3

package/dist/index.umd.js

@@ -2084,10 +2084,15 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		/**
 		* Run an agent
 		*/
-		async runAgent({ agent, forwardedProps, resume, runId }) {
+		async runAgent({ agent, forwardedProps, resume, runId }, continuationHandoff) {
 			if (agent.agentId) this._internal.suggestionEngine.clearSuggestions(agent.agentId);
 			this._internal.applyHeadersToAgent(agent);
-			if (agent.detachActiveRun) await agent.detachActiveRun();
+			if (agent.detachActiveRun) try {
+				await agent.detachActiveRun();
+			} catch (error) {
+				continuationHandoff === null || continuationHandoff === void 0 || continuationHandoff.cancel();
+				throw error;
+			}
 			const isTopLevel = this._runDepth === 0;
 			let originalAbortRun;
 			if (isTopLevel) {
@@ -2108,21 +2113,40 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			}
 			this._runDepth++;
 			try {
-				const runAgentResult = await agent.runAgent({
+				let logicalRunId = runId;
+				const agentSubscriber = this.createAgentErrorSubscriber(agent);
+				let started = false;
+				const onRunStartedEvent = agentSubscriber.onRunStartedEvent;
+				const onRunInitialized = agentSubscriber.onRunInitialized;
+				agentSubscriber.onRunInitialized = async (params) => {
+					continuationHandoff === null || continuationHandoff === void 0 || continuationHandoff.bind(params.input);
+					return onRunInitialized === null || onRunInitialized === void 0 ? void 0 : onRunInitialized(params);
+				};
+				agentSubscriber.onRunStartedEvent = async (params) => {
+					started = true;
+					if (!continuationHandoff) logicalRunId = params.input.runId;
+					return onRunStartedEvent === null || onRunStartedEvent === void 0 ? void 0 : onRunStartedEvent(params);
+				};
+				const pinRunIdOnWire = runId !== void 0 && !continuationHandoff;
+				const agentRunInput = {
 					forwardedProps: {
 						...this._internal.properties,
 						...forwardedProps
 					},
 					...resume !== void 0 ? { resume } : {},
-					...runId !== void 0 ? { runId } : {},
+					...pinRunIdOnWire ? { runId } : {},
 					tools: this.buildFrontendTools(agent.agentId),
 					context: this._internal.getContextForAgent(agent.agentId)
-				}, this.createAgentErrorSubscriber(agent));
+				};
+				const runAgentResult = await agent.runAgent(agentRunInput, agentSubscriber);
+				if (!started) continuationHandoff === null || continuationHandoff === void 0 || continuationHandoff.cancel();
 				return await this.processAgentResult({
 					runAgentResult,
-					agent
+					agent,
+					runId: logicalRunId
 				});
 			} catch (error) {
+				continuationHandoff === null || continuationHandoff === void 0 || continuationHandoff.cancel();
 				const runError = error instanceof Error ? error : new Error(String(error));
 				const context = {};
 				if (agent.agentId) context.agentId = agent.agentId;
@@ -2143,7 +2167,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		/**
 		* Process agent result and execute tools
 		*/
-		async processAgentResult({ runAgentResult, agent, executeFrontendTools = true }) {
+		async processAgentResult({ runAgentResult, agent, runId, executeFrontendTools = true }) {
 			var _this$_runAbortContro2;
 			const { newMessages } = runAgentResult;
 			const agentId = agent.agentId;
@@ -2185,7 +2209,11 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			if (needsFollowUp && !((_this$_runAbortContro2 = this._runAbortController) === null || _this$_runAbortContro2 === void 0 ? void 0 : _this$_runAbortContro2.signal.aborted)) if (this._runDepth >= MAX_FOLLOW_UP_DEPTH) _copilotkit_shared.logger.warn(`[CopilotKit] Follow-up depth limit (${MAX_FOLLOW_UP_DEPTH}) reached for agent "${agentId}". Stopping recursive follow-up runs to prevent an infinite loop. This usually indicates a tool that keeps requesting a follow-up (e.g. the LLM repeatedly calling the same tool). Consider setting "followUp: false" on the offending tool.`);
 			else {
 				await this._internal.waitForPendingFrameworkUpdates();
-				return await this.runAgent({ agent });
+				const continuationHandoff = this._internal.stateManager.markNextRunAsContinuation(agent, runId);
+				return await this.runAgent({
+					agent,
+					...runId !== void 0 ? { runId } : {}
+				}, continuationHandoff);
 			}
 			this._internal.suggestionEngine.reloadSuggestions(agentId);
 			return runAgentResult;
@@ -2642,7 +2670,11 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 
 //#endregion
 //#region src/core/state-manager.ts
-/**
+	const isContinuation = (input) => {
+		var _command, _input$forwardedProps;
+		return input.resume !== void 0 || Object.prototype.hasOwnProperty.call((_command = (_input$forwardedProps = input.forwardedProps) === null || _input$forwardedProps === void 0 ? void 0 : _input$forwardedProps.command) !== null && _command !== void 0 ? _command : {}, "resume");
+	};
+	/**
 	* Manages state and message tracking by run for CopilotKitCore.
 	* Tracks agent state snapshots and message-to-run associations.
 	*/
@@ -2653,11 +2685,34 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			_defineProperty(this, "messageToRun", /* @__PURE__ */ new Map());
 			_defineProperty(this, "activeRun", /* @__PURE__ */ new Map());
 			_defineProperty(this, "agentSubscriptions", /* @__PURE__ */ new Map());
+			_defineProperty(this, "pendingContinuations", /* @__PURE__ */ new WeakMap());
 		}
 		/**
 		* Initialize state tracking for an agent
 		*/
 		initialize() {}
+		markNextRunAsContinuation(agent, expectedRunId) {
+			let pendingForAgent = this.pendingContinuations.get(agent);
+			if (!pendingForAgent) {
+				pendingForAgent = /* @__PURE__ */ new Set();
+				this.pendingContinuations.set(agent, pendingForAgent);
+			}
+			const pending = {
+				active: true,
+				expectedRunId,
+				bind: (input) => {
+					if (pending.active) pending.expectedInput = input;
+				},
+				cancel: () => {
+					if (!pending.active) return;
+					pending.active = false;
+					pendingForAgent.delete(pending);
+					if (pendingForAgent.size === 0) this.pendingContinuations.delete(agent);
+				}
+			};
+			pendingForAgent.add(pending);
+			return pending;
+		}
 		/**
 		* Subscribe to an agent's events to track state and messages.
 		*/
@@ -2682,8 +2737,14 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			const { unsubscribe } = agent.subscribe({
 				onRunStartedEvent: ({ input, state }) => {
 					if (revoked) return;
-					if (runFinished && input.runId === subRunId) subRunId = (0, _ag_ui_client.randomUUID)();
-					else subRunId = input.runId;
+					const pendingForAgent = this.pendingContinuations.get(agent);
+					const internalContinuation = [...pendingForAgent !== null && pendingForAgent !== void 0 ? pendingForAgent : []].find((pending) => pending.expectedInput === input);
+					internalContinuation === null || internalContinuation === void 0 || internalContinuation.cancel();
+					if (runFinished && input.runId === subRunId && !internalContinuation && !isContinuation(input)) subRunId = (0, _ag_ui_client.randomUUID)();
+					else {
+						var _internalContinuation;
+						subRunId = (_internalContinuation = internalContinuation === null || internalContinuation === void 0 ? void 0 : internalContinuation.expectedRunId) !== null && _internalContinuation !== void 0 ? _internalContinuation : input.runId;
+					}
 					runFinished = false;
 					this.handleRunStarted(agent, effectiveInput(input), state);
 				},
@@ -2716,6 +2777,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			});
 			this.agentSubscriptions.set(agentId, () => {
 				revoked = true;
+				this.pendingContinuations.delete(agent);
 				unsubscribe();
 			});
 		}