@copilotkit/core 1.73.2 → 1.74.0

package/dist/index.cjs

@@ -209,6 +209,7 @@ const REPLAY_COMPLETE_EVENT = "replay_complete";
 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");
@@ -430,15 +431,27 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 		};
 	}
 	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)(() => {
@@ -489,14 +502,16 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 			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$)), 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?.();
+		})), 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) => {