@copilotkit/core 1.73.3 → 1.75.0

package/dist/index.umd.js

@@ -247,9 +247,11 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 	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 MAX_UNOPENED_CREDENTIAL_REFRESHES = 2;
 	var AgentThreadLockedError = class extends Error {
 		constructor(threadId) {
 			super(threadId ? `Thread ${threadId} is locked` : "Thread is locked");
@@ -472,15 +474,27 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			};
 		}
 		observeThread$(input, credentials, options) {
-			return this.observeThreadSession$(input, credentials, options).pipe((0, rxjs_operators.catchError)((error) => {
-				if (!this.isSocketReconnectExhaustedError(error)) return (0, rxjs.throwError)(() => error);
-				const replayCursor = this.getReconnectCursor(input);
-				return this.requestJoinCredentials$("connect", input, replayCursor).pipe((0, rxjs.switchMap)((refreshedCredentials) => refreshedCredentials === null ? rxjs.EMPTY : this.observeThread$(this.applyCanonicalRunIdentity(input, refreshedCredentials, { fallbackToInputRunId: options.streamMode === "run" }), refreshedCredentials, {
-					...options,
-					channelMode: "connect",
-					replayCursor
-				})));
-			}));
+			const { unopenedRefreshes: previousUnopened = 0, ...sessionOptions } = options;
+			return (0, rxjs.defer)(() => {
+				let socketOpened = false;
+				return this.observeThreadSession$(input, credentials, {
+					...sessionOptions,
+					onSocketOpen: () => {
+						socketOpened = true;
+					}
+				}).pipe((0, rxjs_operators.catchError)((error) => {
+					if (!this.isSocketReconnectExhaustedError(error)) return (0, rxjs.throwError)(() => error);
+					const unopenedRefreshes = socketOpened ? 0 : previousUnopened + 1;
+					if (options.streamMode === "run" && unopenedRefreshes > MAX_UNOPENED_CREDENTIAL_REFRESHES) return (0, rxjs.throwError)(() => /* @__PURE__ */ new Error(`Realtime connection to ${credentials.realtime.clientUrl} never opened in ${unopenedRefreshes} connection attempts. The realtime endpoint is unavailable.`));
+					const replayCursor = this.getReconnectCursor(input);
+					return this.requestJoinCredentials$("connect", input, replayCursor).pipe((0, rxjs.switchMap)((refreshedCredentials) => refreshedCredentials === null ? rxjs.EMPTY : this.observeThread$(this.applyCanonicalRunIdentity(input, refreshedCredentials, { fallbackToInputRunId: options.streamMode === "run" }), refreshedCredentials, {
+						...sessionOptions,
+						channelMode: "connect",
+						replayCursor,
+						unopenedRefreshes
+					})));
+				}));
+			});
 		}
 		observeThreadSession$(input, credentials, options) {
 			return (0, rxjs.defer)(() => {
@@ -522,25 +536,34 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 					var _this$readEventId;
 					latestObservedReplayCursor = (_this$readEventId = this.readEventId(payload)) !== null && _this$readEventId !== void 0 ? _this$readEventId : latestObservedReplayCursor;
 				}), (0, rxjs_operators.share)());
-				const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT).pipe((0, rxjs_operators.shareReplay)({
+				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 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).pipe((0, rxjs_operators.shareReplay)({
+				const streamIdle$ = this.observeControlEvent$(input.threadId, channel$, STREAM_IDLE_EVENT, controlCursor).pipe((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;
 				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$).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)(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)));
 			});
 		}
 		joinThreadChannel$(channel$) {
 			return ɵjoinPhoenixChannel$(channel$);
 		}
-		observeSocketHealth$(socket$) {
-			return ɵobservePhoenixSocketHealth$(ɵobservePhoenixSocketSignals$(socket$), 5);
+		observeSocketHealth$(socket$, onSocketOpen) {
+			return ɵobservePhoenixSocketHealth$(ɵobservePhoenixSocketSignals$(socket$).pipe((0, rxjs_operators.tap)((signal) => {
+				if (signal.type === "open") onSocketOpen === null || onSocketOpen === void 0 || onSocketOpen();
+			})), 5);
 		}
 		observeThreadEvents$(threadId, channel$, options) {
 			return channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => this.observeChannelEvent$(channel, CLIENT_AG_UI_EVENT)), (0, rxjs_operators.tap)((payload) => {
@@ -551,8 +574,8 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				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);
@@ -603,9 +626,11 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			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) {