@copilotkit/core 1.74.0 → 1.75.1

package/dist/index.cjs

@@ -25,20 +25,23 @@ let zod_to_json_schema = require("zod-to-json-schema");
 *
 * `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 {
 		agent.isRunning = true;
@@ -46,10 +49,20 @@ async function ɵconnectWithoutEventVerification(agent, parameters, subscriber)
 		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 ?? {}
 		];
@@ -59,7 +72,18 @@ async function ɵconnectWithoutEventVerification(agent, parameters, subscriber)
 		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;
@@ -206,9 +230,11 @@ function ɵobservePhoenixSocketHealth$(socketSignals$, maxConsecutiveErrors) {
 const globalFetch = (...args) => fetch(...args);
 const CLIENT_AG_UI_EVENT = "ag_ui_event";
 const REPLAY_COMPLETE_EVENT = "replay_complete";
+const REPLAY_FAILED_EVENT = "replay_failed";
 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) {
@@ -296,7 +322,7 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 			...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) {
@@ -345,17 +371,22 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	* 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) {
+				lifecycle?.onReplayStarted?.();
+				lifecycle?.onReplayFinished?.();
+				return rxjs.EMPTY;
+			}
 			const canonicalInput = this.applyCanonicalRunIdentity(input, credentials, { fallbackToInputRunId: false });
 			return this.observeThread$(canonicalInput, credentials, {
 				completeOnRunError: false,
 				streamMode: "connect",
-				replayCursor
+				replayCursor,
+				lifecycle
 			});
 		}));
 	}
@@ -455,8 +486,24 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	}
 	observeThreadSession$(input, credentials, options) {
 		return (0, rxjs.defer)(() => {
+			options.lifecycle?.onReplayStarted?.();
 			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(() => {
+					warnedAboutIdle = true;
+					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: {
@@ -490,19 +537,42 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 			let latestObservedReplayCursor = null;
 			const threadEvents$ = this.observeThreadEvents$(input.threadId, channel$, options).pipe((0, rxjs_operators.tap)((payload) => {
 				latestObservedReplayCursor = this.readEventId(payload) ?? latestObservedReplayCursor;
+				scheduleIdleWarning();
 			}), (0, rxjs_operators.share)());
-			const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT).pipe((0, rxjs_operators.shareReplay)({
-				bufferSize: 1,
-				refCount: true
+			const controlCursor = {
+				applied: null,
+				replayFailed: false
+			};
+			const replayFailed$ = channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => this.observeChannelEvent$(channel, REPLAY_FAILED_EVENT)), (0, rxjs_operators.tap)(() => {
+				controlCursor.replayFailed = true;
 			}));
-			const streamIdle$ = this.observeControlEvent$(input.threadId, channel$, STREAM_IDLE_EVENT).pipe((0, rxjs_operators.shareReplay)({
+			const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT, controlCursor).pipe((0, rxjs_operators.tap)(() => {
+				replayFinished = true;
+				scheduleIdleWarning();
+				options.lifecycle?.onReplayFinished?.();
+			}), (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)(() => {
+				replayFinished = false;
+				warnedAboutIdle = false;
+				clearIdleWarning();
+				options.lifecycle?.onReplayStarted?.();
+			}), (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)(() => {
+				this.debugLogger?.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$)), 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$) {
@@ -522,8 +592,8 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 			errorOnRunError: options.streamMode === "run"
 		})), (0, rxjs.dematerialize)());
 	}
-	observeControlEvent$(threadId, channel$, eventName) {
-		return channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => this.observeChannelEvent$(channel, eventName)), (0, rxjs_operators.tap)((payload) => this.updateLastSeenEventIdFromControl(threadId, payload)));
+	observeControlEvent$(threadId, channel$, eventName, controlCursor) {
+		return channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => this.observeChannelEvent$(channel, eventName)), (0, rxjs_operators.tap)((payload) => this.updateLastSeenEventIdFromControl(threadId, payload, controlCursor)));
 	}
 	observeChannelEvent$(channel, eventName) {
 		return ɵobservePhoenixEvent$(channel, eventName);
@@ -572,9 +642,11 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 		if (!eventId) return;
 		this.advanceLastSeenEventId(threadId, eventId);
 	}
-	updateLastSeenEventIdFromControl(threadId, payload) {
+	updateLastSeenEventIdFromControl(threadId, payload, controlCursor) {
 		const eventId = this.readControlEventId(payload);
 		if (!eventId) return;
+		if (eventId === controlCursor.applied && !controlCursor.replayFailed) return;
+		controlCursor.applied = eventId;
 		this.advanceLastSeenEventId(threadId, eventId);
 	}
 	advanceLastSeenEventId(threadId, eventId) {
@@ -982,7 +1054,7 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 		});
 	}
 	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;
@@ -1003,7 +1075,10 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 				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));
@@ -1018,9 +1093,9 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 			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 this.#connectViaDelegate(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 this.#connectViaDelegate(input, lifecycle);
 		return this.#connectViaHttp(input);
 	}
 	run(input) {
@@ -1028,8 +1103,8 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 		if (this.runtimeMode === _copilotkit_shared.RUNTIME_MODE_INTELLIGENCE) return this.#runViaDelegate(input);
 		return this.#runViaHttp(input);
 	}
-	#connectViaDelegate(input) {
-		return (0, rxjs.defer)(() => (0, rxjs.from)(this.resolveDelegate())).pipe((0, rxjs_operators.switchMap)((delegate) => withAbortErrorHandling(delegate.connect(input))));
+	#connectViaDelegate(input, lifecycle) {
+		return (0, rxjs.defer)(() => (0, rxjs.from)(this.resolveDelegate())).pipe((0, rxjs_operators.switchMap)((delegate) => withAbortErrorHandling(delegate.connect(input, lifecycle))));
 	}
 	#connectViaHttp(unfiltered) {
 		const input = this.#applyMessageFilter(unfiltered);
@@ -1223,6 +1298,7 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 		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 };
@@ -5394,6 +5470,7 @@ const ALLOWED_KEYS = new Set([
 	"onMessagesChanged",
 	"onStateChanged",
 	"onRunInitialized",
+	"onRunStartedEvent",
 	"onRunFinalized",
 	"onRunFailed",
 	"onRunErrorEvent"
@@ -5995,6 +6072,10 @@ var CopilotKitCore = class {
 				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);
@@ -6046,6 +6127,7 @@ var CopilotKitCore = class {
 		});
 		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;