@copilotkit/core 1.62.1 → 1.62.2

package/dist/index.cjs

@@ -6,7 +6,6 @@ let rxjs = require("rxjs");
 let rxjs_operators = require("rxjs/operators");
 let phoenix = require("phoenix");
 let zod_to_json_schema = require("zod-to-json-schema");
-let rxjs_fetch = require("rxjs/fetch");
 
 //#region src/utils/phoenix-observable.ts
 /**
@@ -131,6 +130,7 @@ const CLIENT_AG_UI_EVENT = "ag_ui_event";
 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;
 var AgentThreadLockedError = class extends Error {
 	constructor(threadId) {
 		super(threadId ? `Thread ${threadId} is locked` : "Thread is locked");
@@ -280,12 +280,14 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	connect(input) {
 		this.threadId = input.threadId;
 		this.canonicalRunId = null;
-		return (0, rxjs.defer)(() => this.requestJoinCredentials$("connect", input)).pipe((0, rxjs.switchMap)((credentials) => {
+		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;
 			const canonicalInput = this.applyCanonicalRunIdentity(input, credentials, { fallbackToInputRunId: false });
 			return this.observeThread$(canonicalInput, credentials, {
 				completeOnRunError: false,
-				streamMode: "connect"
+				streamMode: "connect",
+				replayCursor
 			});
 		}));
 	}
@@ -308,7 +310,7 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	cleanup() {
 		this.cleanupOwned(this.activeChannel, this.socket);
 	}
-	requestJoinCredentials$(mode, input) {
+	requestJoinCredentials$(mode, input, replayCursor) {
 		return (0, rxjs.defer)(async () => {
 			try {
 				const response = await fetch(this.buildRuntimeUrl(mode), {
@@ -325,7 +327,7 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 						context: input.context,
 						state: input.state,
 						forwardedProps: input.forwardedProps,
-						...mode === "connect" ? { lastSeenEventId: this.getReconnectCursor(input) } : {}
+						...mode === "connect" ? { lastSeenEventId: replayCursor === void 0 ? this.getReconnectCursor(input) : replayCursor } : {}
 					}),
 					...this.config.credentials ? { credentials: this.config.credentials } : {}
 				});
@@ -361,10 +363,11 @@ 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);
-			return this.requestJoinCredentials$("connect", input).pipe((0, rxjs.switchMap)((refreshedCredentials) => refreshedCredentials === null ? rxjs.EMPTY : this.observeThread$(this.applyCanonicalRunIdentity(input, refreshedCredentials, { fallbackToInputRunId: options.streamMode === "run" }), refreshedCredentials, {
+			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: this.getReconnectCursor(input)
+				replayCursor
 			})));
 		}));
 	}
@@ -401,15 +404,23 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 				bufferSize: 1,
 				refCount: true
 			}));
-			const threadEvents$ = this.observeThreadEvents$(input.threadId, channel$, options).pipe((0, rxjs_operators.share)());
-			const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT).pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.share)());
+			const reconnectCursor = this.readDurableEventId(options.replayCursor ?? this.getReconnectCursor(input));
+			let latestObservedReplayCursor = null;
+			const threadEvents$ = this.observeThreadEvents$(input.threadId, channel$, options).pipe((0, rxjs_operators.tap)((payload) => {
+				latestObservedReplayCursor = this.readEventId(payload) ?? latestObservedReplayCursor;
+			}), (0, rxjs_operators.share)());
+			const replayComplete$ = this.observeControlEvent$(input.threadId, channel$, REPLAY_COMPLETE_EVENT).pipe((0, rxjs_operators.shareReplay)({
+				bufferSize: 1,
+				refCount: true
+			}));
 			const streamIdle$ = this.observeControlEvent$(input.threadId, channel$, STREAM_IDLE_EVENT).pipe((0, rxjs_operators.shareReplay)({
 				bufferSize: 1,
 				refCount: true
 			}));
-			const streamIdleCompletion$ = options.streamMode === "connect" ? streamIdle$.pipe((0, rxjs_operators.take)(1)) : rxjs.EMPTY;
-			const terminal$ = (0, rxjs.merge)(threadEvents$.pipe((0, rxjs_operators.ignoreElements)(), (0, rxjs_operators.endWith)(null), (0, rxjs_operators.take)(1)), 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.takeUntil)(terminal$)), streamIdleCompletion$.pipe((0, rxjs_operators.ignoreElements)())).pipe((0, rxjs_operators.finalize)(() => this.cleanupOwned(ownChannel, ownSocket)));
+			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)));
 		});
 	}
 	joinThreadChannel$(channel$) {
@@ -475,23 +486,36 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	updateLastSeenEventId(threadId, payload) {
 		const eventId = this.readEventId(payload);
 		if (!eventId) return;
-		this.sharedState.lastSeenEventIds.set(threadId, eventId);
+		this.advanceLastSeenEventId(threadId, eventId);
 	}
 	updateLastSeenEventIdFromControl(threadId, payload) {
 		const eventId = this.readControlEventId(payload);
 		if (!eventId) return;
+		this.advanceLastSeenEventId(threadId, eventId);
+	}
+	advanceLastSeenEventId(threadId, eventId) {
 		this.sharedState.lastSeenEventIds.set(threadId, eventId);
 	}
 	readEventId(payload) {
 		const metadata = payload.metadata;
 		if (!metadata || typeof metadata !== "object") return null;
-		const runnerEventId = metadata.cpki_event_id;
-		return typeof runnerEventId === "string" ? runnerEventId : null;
+		const eventMetadata = metadata;
+		return this.readDurableEventId(eventMetadata.cpki_event_id);
 	}
 	readControlEventId(payload) {
 		if (!payload || typeof payload !== "object") return null;
-		const latestEventId = payload.latestEventId;
-		return typeof latestEventId === "string" ? latestEventId : null;
+		const controlPayload = payload;
+		const latestEventId = controlPayload.latestEventId ?? controlPayload.latest_event_id;
+		return this.readDurableEventId(latestEventId);
+	}
+	readDurableEventId(eventId) {
+		if (typeof eventId !== "string" || eventId.trim() === "") return null;
+		return eventId.startsWith("cpki_ingested") ? null : eventId;
+	}
+	canFallbackCompleteConnect(streamIdlePayload, reconnectCursor, latestObservedReplayCursor) {
+		const idleCursor = this.readControlEventId(streamIdlePayload);
+		if (!idleCursor) return true;
+		return idleCursor === reconnectCursor || idleCursor === latestObservedReplayCursor;
 	}
 	applyCanonicalRunIdentity(input, credentials, options) {
 		this.threadId = credentials.threadId;
@@ -1660,7 +1684,9 @@ var RunHandler = class {
 	* downstream churn into duplicate `cpki_event_id` rows in the
 	* inspector and intermittent "Message not found" toasts.
 	*/
-	_lastConnectedThreadId = null;
+	_lastConnectedThreadIdsByAgent = /* @__PURE__ */ new Map();
+	_anonymousAgentIds = /* @__PURE__ */ new WeakMap();
+	_nextAnonymousAgentId = 0;
 	constructor(core) {
 		this.core = core;
 	}
@@ -1680,6 +1706,20 @@ var RunHandler = class {
 		return this.core;
 	}
 	/**
+	* Return a stable restore-tracking key for the logical agent being
+	* connected. Named agents share their last-thread marker across proxy
+	* instances; anonymous agents fall back to object identity.
+	*/
+	getConnectRestoreKey(agent) {
+		if (agent.agentId) return `agent:${agent.agentId}`;
+		const existing = this._anonymousAgentIds.get(agent);
+		if (existing) return existing;
+		const anonymousId = `anonymous:${this._nextAnonymousAgentId}`;
+		this._nextAnonymousAgentId += 1;
+		this._anonymousAgentIds.set(agent, anonymousId);
+		return anonymousId;
+	}
+	/**
 	* Get all tools as a readonly array
 	*/
 	get tools() {
@@ -1735,14 +1775,18 @@ var RunHandler = class {
 	async connectAgent({ agent }) {
 		try {
 			const incomingThreadId = agent.threadId ?? null;
-			const isFreshRestore = incomingThreadId !== this._lastConnectedThreadId;
-			this._lastConnectedThreadId = incomingThreadId;
+			const restoreKey = this.getConnectRestoreKey(agent);
+			const isFreshRestore = incomingThreadId !== (this._lastConnectedThreadIdsByAgent.get(restoreKey) ?? null);
+			this._lastConnectedThreadIdsByAgent.set(restoreKey, incomingThreadId);
 			await agent.detachActiveRun();
 			if (isFreshRestore) {
 				agent.setMessages([]);
 				agent.setState({});
-				const proxied = agent;
-				if (incomingThreadId && typeof proxied.clearReplayCursor === "function") proxied.clearReplayCursor(incomingThreadId);
+				const cursorAware = agent;
+				if (incomingThreadId) {
+					if (typeof cursorAware.clearReplayCursor === "function") cursorAware.clearReplayCursor(incomingThreadId);
+					if (typeof cursorAware.clearReconnectCursor === "function") cursorAware.clearReconnectCursor(incomingThreadId);
+				}
 			}
 			this._internal.applyHeadersToAgent(agent);
 			await this._internal.notifySubscribers((subscriber) => subscriber.onAgentRunStarted?.({
@@ -1756,7 +1800,8 @@ var RunHandler = class {
 			}, this.createAgentErrorSubscriber(agent));
 			return this.processAgentResult({
 				runAgentResult,
-				agent
+				agent,
+				executeFrontendTools: false
 			});
 		} catch (error) {
 			const connectError = error instanceof Error ? error : new Error(String(error));
@@ -1778,7 +1823,7 @@ var RunHandler = class {
 	/**
 	* Run an agent
 	*/
-	async runAgent({ agent, forwardedProps, resume }) {
+	async runAgent({ agent, forwardedProps, resume, runId }) {
 		if (agent.agentId) this._internal.suggestionEngine.clearSuggestions(agent.agentId);
 		this._internal.applyHeadersToAgent(agent);
 		if (agent.detachActiveRun) await agent.detachActiveRun();
@@ -1805,6 +1850,7 @@ var RunHandler = class {
 					...forwardedProps
 				},
 				...resume !== void 0 ? { resume } : {},
+				...runId !== void 0 ? { runId } : {},
 				tools: this.buildFrontendTools(agent.agentId),
 				context: this._internal.getContextForAgent(agent.agentId)
 			}, this.createAgentErrorSubscriber(agent));
@@ -1833,25 +1879,27 @@ var RunHandler = class {
 	/**
 	* Process agent result and execute tools
 	*/
-	async processAgentResult({ runAgentResult, agent }) {
+	async processAgentResult({ runAgentResult, agent, executeFrontendTools = true }) {
 		const { newMessages } = runAgentResult;
 		const agentId = agent.agentId;
 		let needsFollowUp = false;
-		for (const message of newMessages) if (message.role === "assistant") {
-			for (const toolCall of message.toolCalls || []) if (newMessages.findIndex((m) => m.role === "tool" && m.toolCallId === toolCall.id) === -1) {
-				const tool = this.getTool({
-					toolName: toolCall.function.name,
-					agentId: agent.agentId
-				});
-				if (tool) {
-					if (await this.executeSpecificTool(tool, toolCall, message, agent, agentId)) needsFollowUp = true;
-				} else {
-					const wildcardTool = this.getTool({
-						toolName: "*",
+		if (executeFrontendTools) {
+			for (const message of newMessages) if (message.role === "assistant") {
+				for (const toolCall of message.toolCalls || []) if (newMessages.findIndex((m) => m.role === "tool" && m.toolCallId === toolCall.id) === -1) {
+					const tool = this.getTool({
+						toolName: toolCall.function.name,
 						agentId: agent.agentId
 					});
-					if (wildcardTool) {
-						if (await this.executeWildcardTool(wildcardTool, toolCall, message, agent, agentId)) needsFollowUp = true;
+					if (tool) {
+						if (await this.executeSpecificTool(tool, toolCall, message, agent, agentId)) needsFollowUp = true;
+					} else {
+						const wildcardTool = this.getTool({
+							toolName: "*",
+							agentId: agent.agentId
+						});
+						if (wildcardTool) {
+							if (await this.executeWildcardTool(wildcardTool, toolCall, message, agent, agentId)) needsFollowUp = true;
+						}
 					}
 				}
 			}
@@ -3585,6 +3633,7 @@ function createStore(options) {
 //#endregion
 //#region src/threads.ts
 const THREADS_CHANNEL_EVENT = "thread_metadata";
+const THREAD_RUN_ACTIVITY_CHANNEL_EVENT = "thread_run_activity";
 const THREAD_SUBSCRIBE_PATH = "/threads/subscribe";
 const MAX_SOCKET_RETRIES = 5;
 const REQUEST_TIMEOUT_MS = 15e3;
@@ -3628,7 +3677,8 @@ const threadSocketEvents = createActionGroup("Thread Socket", {
 	errored: props(),
 	joinFailed: props(),
 	joinTimedOut: props(),
-	metadataReceived: props()
+	metadataReceived: props(),
+	runActivityReceived: props()
 });
 const threadDomainEvents = createActionGroup("Thread Domain", {
 	threadUpserted: props(),
@@ -3647,6 +3697,28 @@ function upsertThread(threads, thread) {
 	next[existingIndex] = thread;
 	return sortThreadsByRecency(next);
 }
+/**
+* Returns a non-empty string payload field or undefined for absent fields.
+*/
+function optionalString(value) {
+	return typeof value === "string" && value.length > 0 ? value : void 0;
+}
+/**
+* Converts gateway run-activity payloads into the internal TypeScript shape.
+*/
+function normalizeThreadRunActivityNotification(payload) {
+	const threadId = optionalString(payload.threadId ?? payload.thread_id);
+	const eventType = optionalString(payload.eventType ?? payload.event_type);
+	if (!threadId || !eventType) return null;
+	return {
+		type: "thread_run_activity",
+		threadId,
+		agentId: optionalString(payload.agentId ?? payload.agent_id),
+		runId: optionalString(payload.runId ?? payload.run_id),
+		eventType,
+		latestEventId: optionalString(payload.latestEventId ?? payload.latest_event_id)
+	};
+}
 const threadReducer = createReducer(initialThreadState, on(threadAdapterEvents.contextChanged, (state, { context }) => ({
 	...state,
 	context,
@@ -3680,12 +3752,14 @@ const threadReducer = createReducer(initialThreadState, on(threadAdapterEvents.c
 	};
 }), on(threadRestEvents.listSucceeded, (state, { sessionId, threads, joinCode, nextCursor }) => {
 	if (sessionId !== state.sessionId) return state;
+	const joinCodeChanged = joinCode !== state.metadataJoinCode;
 	return {
 		...state,
 		threads: sortThreadsByRecency(threads),
 		isLoading: false,
 		error: null,
 		metadataJoinCode: joinCode,
+		metadataCredentialsRequested: joinCodeChanged ? false : state.metadataCredentialsRequested,
 		nextCursor
 	};
 }), on(threadRestEvents.listFailed, (state, { sessionId, error }) => {
@@ -3714,7 +3788,10 @@ const threadReducer = createReducer(initialThreadState, on(threadAdapterEvents.c
 	};
 }), on(threadRestEvents.metadataCredentialsFailed, (state, { sessionId }) => {
 	if (sessionId !== state.sessionId) return state;
-	return state;
+	return {
+		...state,
+		metadataCredentialsRequested: false
+	};
 }), on(threadRestEvents.metadataCredentialsRequested, (state, { sessionId }) => {
 	if (sessionId !== state.sessionId) return state;
 	return {
@@ -3872,7 +3949,27 @@ function createThreadRequestId() {
 	return `thread-request-${threadRequestId}`;
 }
 function threadFromFetch(input, init) {
-	return (0, rxjs_fetch.fromFetch)(input, init);
+	return new rxjs.Observable((subscriber) => {
+		const { fetch: fetchImpl, selector, signal, ...requestInit } = init;
+		const controller = new AbortController();
+		const abortRequest = () => controller.abort();
+		if (signal?.aborted) abortRequest();
+		else signal?.addEventListener("abort", abortRequest, { once: true });
+		fetchImpl(input, {
+			...requestInit,
+			signal: controller.signal
+		}).then((response) => selector(response)).then((value) => {
+			if (subscriber.closed) return;
+			subscriber.next(value);
+			subscriber.complete();
+		}).catch((error) => {
+			if (!subscriber.closed) subscriber.error(error);
+		});
+		return () => {
+			signal?.removeEventListener("abort", abortRequest);
+			abortRequest();
+		};
+	});
 }
 function createThreadFetchObservable(environment, context, sessionId) {
 	return (0, rxjs.defer)(() => {
@@ -4018,6 +4115,9 @@ function createThreadStore(environment) {
 					return (0, rxjs.merge)(socketSignals$.pipe((0, rxjs_operators.map)((signal) => signal.type === "open" ? threadSocketEvents.opened({ sessionId: action.sessionId }) : threadSocketEvents.errored({ sessionId: action.sessionId }))), channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => ɵobservePhoenixEvent$(channel, THREADS_CHANNEL_EVENT)), (0, rxjs_operators.map)((payload) => threadSocketEvents.metadataReceived({
 						sessionId: action.sessionId,
 						payload
+					}))), channel$.pipe((0, rxjs_operators.switchMap)(({ channel }) => ɵobservePhoenixEvent$(channel, THREAD_RUN_ACTIVITY_CHANNEL_EVENT)), (0, rxjs_operators.map)((payload) => normalizeThreadRunActivityNotification(payload)), (0, rxjs_operators.filter)((notification) => notification !== null), (0, rxjs_operators.map)((notification) => threadSocketEvents.runActivityReceived({
+						sessionId: action.sessionId,
+						notification
 					}))), ɵobservePhoenixJoinOutcome$(channel$).pipe((0, rxjs_operators.filter)((outcome) => outcome.type !== "joined"), (0, rxjs_operators.map)((outcome) => outcome.type === "timeout" ? threadSocketEvents.joinTimedOut({ sessionId: action.sessionId }) : threadSocketEvents.joinFailed({ sessionId: action.sessionId })))).pipe((0, rxjs_operators.takeUntil)((0, rxjs.merge)(shutdown$, fatalSocketShutdown$)));
 				});
 			}))),
@@ -4195,6 +4295,9 @@ function createThreadStore(environment) {
 				threadId
 			}));
 		},
+		subscribeToRunActivity(callback) {
+			return store.actions$.pipe(ofType(threadSocketEvents.runActivityReceived), (0, rxjs_operators.filter)((action) => action.sessionId === store.getState().sessionId), (0, rxjs_operators.map)((action) => action.notification)).subscribe(callback);
+		},
 		getState() {
 			return store.getState();
 		},