@copilotkit/core 1.75.0 → 1.75.2

package/dist/index.umd.js

@@ -23,20 +23,23 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 	*
 	* `transformChunks` is still applied — message reassembly is needed either way.
 	*
-	* This mirrors the base `AbstractAgent.connectAgent` implementation exactly
-	* apart from that omission, so callers keep the same subscriber notifications,
-	* detach semantics, and `{ result, newMessages }` return shape.
+	* Connection-local replay hooks track the phase before applying events. An
+	* explicitly live RUN_ERROR clears busy without closing the connection;
+	* historical errors remain data. Connections without hooks retain the existing
+	* completion behavior. Subscriber, detach, and result contracts are preserved.
 	*
 	* TODO: Remove this in favour of the base implementation once AG-UI's
 	* AbstractAgent supports opting out of `verifyEvents` for transports whose
-	* connection life-cycle isn't a single run. As of `@ag-ui/client@0.0.57`
+	* connection life-cycle isn't a single run AND preserves the connection-local
+	* replay lifecycle and running-state behavior below. Skipping verification alone
+	* is insufficient. As of `@ag-ui/client@0.0.57`
 	* `connectAgent(parameters?, subscriber?)` takes no such option.
 	*
 	* @param agent - The agent whose `connect()` stream should be consumed.
 	* @param parameters - Run parameters, forwarded to `prepareRunAgentInput`.
 	* @param subscriber - Optional one-shot subscriber for this connect call.
 	*/
-	async function ɵconnectWithoutEventVerification(agent, parameters, subscriber) {
+	async function ɵconnectWithoutEventVerification(agent, parameters, subscriber, connect) {
 		const self = agent;
 		try {
 			var _agent$agentId;
@@ -45,10 +48,20 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			const input = self.prepareRunAgentInput(parameters);
 			let result;
 			const previousMessageIds = new Set(agent.messages.map((m) => m.id));
+			let isReplaying = true;
+			const liveErrors = /* @__PURE__ */ new WeakSet();
 			const subscribers = [
-				{ onRunFinishedEvent: (event) => {
-					if (event.outcome === "success") result = event.result;
-				} },
+				{
+					onRunStartedEvent: () => {
+						agent.isRunning = true;
+					},
+					onRunErrorEvent: ({ event }) => {
+						if (liveErrors.has(event)) agent.isRunning = false;
+					},
+					onRunFinishedEvent: (event) => {
+						if (event.outcome === "success") result = event.result;
+					}
+				},
 				...agent.subscribers,
 				subscriber !== null && subscriber !== void 0 ? subscriber : {}
 			];
@@ -58,7 +71,18 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			self.activeRunCompletionPromise = new Promise((resolve) => {
 				resolveCompletion = resolve;
 			});
-			const source$ = (0, rxjs.defer)(() => self.connect(input)).pipe((0, _ag_ui_client.transformChunks)(self.debugLogger), (0, rxjs_operators.takeUntil)(self.activeRunDetach$));
+			const lifecycle = {
+				onReplayStarted: () => {
+					isReplaying = true;
+					agent.isRunning = true;
+				},
+				onReplayFinished: () => {
+					isReplaying = false;
+				}
+			};
+			const source$ = (0, rxjs.defer)(() => connect ? connect(input, lifecycle) : self.connect(input)).pipe((0, rxjs_operators.tap)((event) => {
+				if (!isReplaying && event.type === _ag_ui_client.EventType.RUN_ERROR) liveErrors.add(event);
+			}), (0, _ag_ui_client.transformChunks)(self.debugLogger), (0, rxjs_operators.takeUntil)(self.activeRunDetach$));
 			const applied$ = self.apply(input, source$, subscribers);
 			await (0, rxjs.lastValueFrom)(self.processApplyEvents(input, applied$, subscribers).pipe((0, rxjs_operators.catchError)((error) => {
 				agent.isRunning = false;
@@ -251,6 +275,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 	const STREAM_IDLE_EVENT = "stream_idle";
 	const STOP_RUN_EVENT = "stop_run";
 	const CONNECT_STREAM_IDLE_REPLAY_FALLBACK_MS = 100;
+	const CONNECT_IDLE_WARNING_MS = 3e4;
 	const MAX_UNOPENED_CREDENTIAL_REFRESHES = 2;
 	var AgentThreadLockedError = class extends Error {
 		constructor(threadId) {
@@ -338,7 +363,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				...parameters,
 				runId: this.canonicalRunId
 			};
-			return ɵconnectWithoutEventVerification(this, effectiveParameters, subscriber);
+			return ɵconnectWithoutEventVerification(this, effectiveParameters, subscriber, (input, lifecycle) => this.connect(input, lifecycle));
 		}
 		abortRun() {
 			if (this.activeChannel && this.canonicalRunId) {
@@ -387,17 +412,23 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		* gateway only streams events past it instead of replaying the
 		* entire history every time the chat re-opens a socket.
 		*/
-		connect(input) {
+		connect(input, lifecycle) {
 			this.threadId = input.threadId;
 			this.canonicalRunId = null;
 			const replayCursor = this.getReconnectCursor(input);
 			return (0, rxjs.defer)(() => this.requestJoinCredentials$("connect", input, replayCursor)).pipe((0, rxjs.switchMap)((credentials) => {
-				if (credentials === null) return rxjs.EMPTY;
+				if (credentials === null) {
+					var _lifecycle$onReplaySt, _lifecycle$onReplayFi;
+					lifecycle === null || lifecycle === void 0 || (_lifecycle$onReplaySt = lifecycle.onReplayStarted) === null || _lifecycle$onReplaySt === void 0 || _lifecycle$onReplaySt.call(lifecycle);
+					lifecycle === null || lifecycle === void 0 || (_lifecycle$onReplayFi = lifecycle.onReplayFinished) === null || _lifecycle$onReplayFi === void 0 || _lifecycle$onReplayFi.call(lifecycle);
+					return rxjs.EMPTY;
+				}
 				const canonicalInput = this.applyCanonicalRunIdentity(input, credentials, { fallbackToInputRunId: false });
 				return this.observeThread$(canonicalInput, credentials, {
 					completeOnRunError: false,
 					streamMode: "connect",
-					replayCursor
+					replayCursor,
+					lifecycle
 				});
 			}));
 		}
@@ -498,9 +529,26 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		}
 		observeThreadSession$(input, credentials, options) {
 			return (0, rxjs.defer)(() => {
-				var _options$channelMode, _options$replayCursor;
+				var _options$lifecycle, _options$lifecycle$on, _options$channelMode, _options$replayCursor;
+				(_options$lifecycle = options.lifecycle) === null || _options$lifecycle === void 0 || (_options$lifecycle$on = _options$lifecycle.onReplayStarted) === null || _options$lifecycle$on === void 0 || _options$lifecycle$on.call(_options$lifecycle);
 				let ownSocket = null;
 				let ownChannel = null;
+				let replayFinished = false;
+				let warnedAboutIdle = false;
+				let idleWarning;
+				const clearIdleWarning = () => {
+					clearTimeout(idleWarning);
+					idleWarning = void 0;
+				};
+				const scheduleIdleWarning = () => {
+					clearIdleWarning();
+					if (options.streamMode !== "connect" || !replayFinished || warnedAboutIdle) return;
+					idleWarning = setTimeout(() => {
+						var _this$debugLogger;
+						warnedAboutIdle = true;
+						(_this$debugLogger = this.debugLogger) === null || _this$debugLogger === void 0 || _this$debugLogger.lifecycle("INTELLIGENCE", "No progress or stream_idle for 30 seconds after replay_complete; the run may still be active or its status unavailable.", { threadId: input.threadId });
+					}, CONNECT_IDLE_WARNING_MS);
+				};
 				const socket$ = ɵphoenixSocket$({
 					url: credentials.realtime.clientUrl,
 					options: {
@@ -535,6 +583,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				const threadEvents$ = this.observeThreadEvents$(input.threadId, channel$, options).pipe((0, rxjs_operators.tap)((payload) => {
 					var _this$readEventId;
 					latestObservedReplayCursor = (_this$readEventId = this.readEventId(payload)) !== null && _this$readEventId !== void 0 ? _this$readEventId : latestObservedReplayCursor;
+					scheduleIdleWarning();
 				}), (0, rxjs_operators.share)());
 				const controlCursor = {
 					applied: null,
@@ -543,18 +592,36 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				const replayFailed$ = channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => this.observeChannelEvent$(channel, REPLAY_FAILED_EVENT)), (0, rxjs_operators.tap)(() => {
 					controlCursor.replayFailed = true;
 				}));
-				const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT, controlCursor).pipe((0, rxjs_operators.shareReplay)({
-					bufferSize: 1,
-					refCount: true
-				}));
-				const streamIdle$ = this.observeControlEvent$(input.threadId, channel$, STREAM_IDLE_EVENT, controlCursor).pipe((0, rxjs_operators.shareReplay)({
+				const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT, controlCursor).pipe((0, rxjs_operators.tap)(() => {
+					var _options$lifecycle2, _options$lifecycle2$o;
+					replayFinished = true;
+					scheduleIdleWarning();
+					(_options$lifecycle2 = options.lifecycle) === null || _options$lifecycle2 === void 0 || (_options$lifecycle2$o = _options$lifecycle2.onReplayFinished) === null || _options$lifecycle2$o === void 0 || _options$lifecycle2$o.call(_options$lifecycle2);
+				}), (0, rxjs_operators.share)());
+				const streamIdle$ = this.observeControlEvent$(input.threadId, channel$, STREAM_IDLE_EVENT, controlCursor).pipe((0, rxjs_operators.share)());
+				const replayRestart$ = (0, rxjs.merge)(this.observeControlEvent$(input.threadId, channel$, "phx_error", controlCursor), ɵobservePhoenixSocketSignals$(socket$).pipe((0, rxjs_operators.filter)((signal) => signal.type === "error"))).pipe((0, rxjs_operators.tap)(() => {
+					var _options$lifecycle3, _options$lifecycle3$o;
+					replayFinished = false;
+					warnedAboutIdle = false;
+					clearIdleWarning();
+					(_options$lifecycle3 = options.lifecycle) === null || _options$lifecycle3 === void 0 || (_options$lifecycle3$o = _options$lifecycle3.onReplayStarted) === null || _options$lifecycle3$o === void 0 || _options$lifecycle3$o.call(_options$lifecycle3);
+				}), (0, rxjs_operators.share)());
+				const streamIdleCompletion$ = options.streamMode === "connect" ? replayRestart$.pipe((0, rxjs_operators.startWith)(null), (0, rxjs.switchMap)(() => (0, rxjs.merge)((0, rxjs.combineLatest)([replayComplete$.pipe((0, rxjs_operators.take)(1)), streamIdle$.pipe((0, rxjs_operators.take)(1))]), streamIdle$.pipe((0, rxjs_operators.take)(1), (0, rxjs_operators.filter)((payload) => this.canFallbackCompleteConnect(payload, reconnectCursor, latestObservedReplayCursor)), (0, rxjs_operators.delay)(CONNECT_STREAM_IDLE_REPLAY_FALLBACK_MS), (0, rxjs_operators.tap)(() => {
+					var _this$debugLogger2;
+					(_this$debugLogger2 = this.debugLogger) === null || _this$debugLogger2 === void 0 || _this$debugLogger2.lifecycle("INTELLIGENCE", "Completing connection on stream_idle fallback without replay_complete.", {
+						threadId: input.threadId,
+						fallbackMs: CONNECT_STREAM_IDLE_REPLAY_FALLBACK_MS
+					});
+				})))), (0, rxjs_operators.take)(1), (0, rxjs_operators.shareReplay)({
 					bufferSize: 1,
 					refCount: true
-				}));
-				const streamIdleCompletion$ = options.streamMode === "connect" ? (0, rxjs.merge)((0, rxjs.combineLatest)([replayComplete$.pipe((0, rxjs_operators.take)(1)), streamIdle$.pipe((0, rxjs_operators.take)(1))]), streamIdle$.pipe((0, rxjs_operators.take)(1), (0, rxjs_operators.filter)((payload) => this.canFallbackCompleteConnect(payload, reconnectCursor, latestObservedReplayCursor)), (0, rxjs_operators.delay)(CONNECT_STREAM_IDLE_REPLAY_FALLBACK_MS))).pipe((0, rxjs_operators.take)(1)) : rxjs.EMPTY;
+				})) : rxjs.EMPTY;
 				const threadCompleted$ = threadEvents$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.endWith)(null), (0, rxjs_operators.take)(1));
 				const terminal$ = (0, rxjs.merge)(threadCompleted$, streamIdleCompletion$);
-				return (0, rxjs.merge)(this.joinThreadChannel$(channel$), this.observeSocketHealth$(socket$, options.onSocketOpen).pipe((0, rxjs_operators.takeUntil)(terminal$)), threadEvents$.pipe((0, rxjs_operators.takeUntil)(streamIdleCompletion$)), replayFailed$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.takeUntil)(terminal$)), replayComplete$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.takeUntil)(terminal$)), streamIdleCompletion$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.takeUntil)(threadCompleted$))).pipe((0, rxjs_operators.finalize)(() => this.cleanupOwned(ownChannel, ownSocket)));
+				return (0, rxjs.merge)(replayFailed$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.takeUntil)(terminal$)), replayComplete$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.takeUntil)(terminal$)), this.joinThreadChannel$(channel$), this.observeSocketHealth$(socket$, options.onSocketOpen).pipe((0, rxjs_operators.takeUntil)(terminal$)), threadEvents$.pipe((0, rxjs_operators.takeUntil)(streamIdleCompletion$)), streamIdleCompletion$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.takeUntil)(threadCompleted$))).pipe((0, rxjs_operators.finalize)(() => {
+					clearIdleWarning();
+					this.cleanupOwned(ownChannel, ownSocket);
+				}));
 			});
 		}
 		joinThreadChannel$(channel$) {
@@ -1041,7 +1108,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			});
 		}
 		async connectAgent(parameters, subscriber) {
-			if (this.runtimeMode !== _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return ɵconnectWithoutEventVerification(this, parameters, subscriber);
+			if (this.runtimeMode !== _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return ɵconnectWithoutEventVerification(this, parameters, subscriber, (input, lifecycle) => this.connect(input, lifecycle));
 			if (this.delegate) await this.delegate.detachActiveRun();
 			await this.resolveDelegate();
 			const delegate = this.delegate;
@@ -1062,7 +1129,10 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 					this.isRunning = false;
 				},
 				onRunErrorEvent: () => {
-					this.isRunning = false;
+					this.isRunning = delegate.isRunning;
+				},
+				onRunStartedEvent: () => {
+					this.isRunning = delegate.isRunning;
 				}
 			});
 			const forwardedSubs = this.subscribers.map((s) => delegate.subscribe(s));
@@ -1077,9 +1147,9 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				for (const sub of forwardedSubs) sub.unsubscribe();
 			}
 		}
-		connect(input) {
-			if (this.runtimeMode === "pending" || this.transport === "auto" && this.runtimeMode !== _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return (0, rxjs.defer)(() => (0, rxjs.from)(this.ensureRuntimeConfiguration())).pipe((0, rxjs_operators.switchMap)(() => this.connect(input)));
-			if (this.runtimeMode === _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return _assertClassBrand(_ProxiedCopilotRuntimeAgent_brand, this, _connectViaDelegate).call(this, input);
+		connect(input, lifecycle) {
+			if (this.runtimeMode === "pending" || this.transport === "auto" && this.runtimeMode !== _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return (0, rxjs.defer)(() => (0, rxjs.from)(this.ensureRuntimeConfiguration())).pipe((0, rxjs_operators.switchMap)(() => this.connect(input, lifecycle)));
+			if (this.runtimeMode === _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return _assertClassBrand(_ProxiedCopilotRuntimeAgent_brand, this, _connectViaDelegate).call(this, input, lifecycle);
 			return _assertClassBrand(_ProxiedCopilotRuntimeAgent_brand, this, _connectViaHttp).call(this, input);
 		}
 		run(input) {
@@ -1252,6 +1322,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			delegate.agentId = this.routedAgentId();
 			delegate.description = this.description;
 			delegate.threadId = this.threadId;
+			delegate.debugLogger = this.debugLogger;
 			delegate.setMessages(this.messages);
 			delegate.setState(this.state);
 			if (hasHeaders(delegate)) delegate.headers = { ...this.headers };
@@ -1307,8 +1378,8 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			return input;
 		}
 	}
-	function _connectViaDelegate(input) {
-		return (0, rxjs.defer)(() => (0, rxjs.from)(this.resolveDelegate())).pipe((0, rxjs_operators.switchMap)((delegate) => withAbortErrorHandling(delegate.connect(input))));
+	function _connectViaDelegate(input, lifecycle) {
+		return (0, rxjs.defer)(() => (0, rxjs.from)(this.resolveDelegate())).pipe((0, rxjs_operators.switchMap)((delegate) => withAbortErrorHandling(delegate.connect(input, lifecycle))));
 	}
 	function _connectViaHttp(unfiltered) {
 		const input = _assertClassBrand(_ProxiedCopilotRuntimeAgent_brand, this, _applyMessageFilter).call(this, unfiltered);
@@ -5562,6 +5633,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		"onMessagesChanged",
 		"onStateChanged",
 		"onRunInitialized",
+		"onRunStartedEvent",
 		"onRunFinalized",
 		"onRunFailed",
 		"onRunErrorEvent"
@@ -6168,6 +6240,10 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 					const fn = sub.onRunInitialized;
 					guarded.onRunInitialized = (params) => safeCall("onRunInitialized", fn, params);
 				}
+				if (sub.onRunStartedEvent) {
+					const fn = sub.onRunStartedEvent;
+					guarded.onRunStartedEvent = (params) => safeCall("onRunStartedEvent", fn, params);
+				}
 				if (sub.onRunFinalized) {
 					const fn = sub.onRunFinalized;
 					guarded.onRunFinalized = (params) => safeCall("onRunFinalized", fn, params);
@@ -6219,6 +6295,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			});
 			const lifecycleOnly = {};
 			if (subscriber.onRunInitialized) lifecycleOnly.onRunInitialized = subscriber.onRunInitialized;
+			if (subscriber.onRunStartedEvent) lifecycleOnly.onRunStartedEvent = subscriber.onRunStartedEvent;
 			if (subscriber.onRunFinalized) lifecycleOnly.onRunFinalized = subscriber.onRunFinalized;
 			if (subscriber.onRunFailed) lifecycleOnly.onRunFailed = subscriber.onRunFailed;
 			if (subscriber.onRunErrorEvent) lifecycleOnly.onRunErrorEvent = subscriber.onRunErrorEvent;