@copilotkit/core 1.68.3 → 1.69.0

package/dist/index.cjs

@@ -2803,6 +2803,7 @@ const isContinuation = (input) => input.resume !== void 0 || Object.prototype.ha
 var StateManager = class {
 	stateByRun = /* @__PURE__ */ new Map();
 	messageToRun = /* @__PURE__ */ new Map();
+	rawEventByMessage = /* @__PURE__ */ new Map();
 	activeRun = /* @__PURE__ */ new Map();
 	agentSubscriptions = /* @__PURE__ */ new Map();
 	pendingContinuations = /* @__PURE__ */ new WeakMap();
@@ -2854,12 +2855,13 @@ var StateManager = class {
 			runId: subRunId ?? input.runId
 		});
 		const { unsubscribe } = agent.subscribe({
-			onRunStartedEvent: ({ input, state }) => {
+			onRunStartedEvent: ({ event, input, state }) => {
 				if (revoked) return;
 				const internalContinuation = [...this.pendingContinuations.get(agent) ?? []].find((pending) => pending.expectedInput === input);
 				internalContinuation?.cancel();
-				if (runFinished && input.runId === subRunId && !internalContinuation && !isContinuation(input)) subRunId = (0, _ag_ui_client.randomUUID)();
-				else subRunId = internalContinuation?.expectedRunId ?? input.runId;
+				if (internalContinuation) subRunId = internalContinuation.expectedRunId ?? event.runId ?? input.runId;
+				else if (runFinished && input.runId === subRunId && !isContinuation(input) && (event.runId == null || event.runId === subRunId)) subRunId = (0, _ag_ui_client.randomUUID)();
+				else subRunId = event.runId || input.runId;
 				runFinished = false;
 				this.handleRunStarted(agent, effectiveInput(input), state);
 			},
@@ -2881,13 +2883,22 @@ var StateManager = class {
 				if (revoked) return;
 				this.handleStateDelta(agent, event, effectiveInput(input), state);
 			},
+			onTextMessageStartEvent: ({ event, input }) => {
+				if (revoked) return;
+				this.handleTextMessageStart(agent, event, effectiveInput(input));
+			},
 			onMessagesSnapshotEvent: ({ event, input, messages }) => {
 				if (revoked) return;
 				this.handleMessagesSnapshot(agent, event, effectiveInput(input), messages);
+				this.pruneRawEvents(agent.agentId, input.threadId, event.messages, effectiveInput(input));
 			},
 			onNewMessage: ({ message, input }) => {
 				if (revoked) return;
 				this.handleNewMessage(agent, message, input ? effectiveInput(input) : void 0);
+			},
+			onMessagesChanged: ({ messages, input }) => {
+				if (revoked) return;
+				if (!input) this.pruneRawEvents(agent.agentId, agent.threadId, messages);
 			}
 		});
 		this.agentSubscriptions.set(agentId, () => {
@@ -2905,6 +2916,7 @@ var StateManager = class {
 			unsubscribe();
 			this.agentSubscriptions.delete(agentId);
 		}
+		this.rawEventByMessage.delete(agentId);
 	}
 	/**
 	* Get state for a specific run
@@ -2922,6 +2934,13 @@ var StateManager = class {
 		return this.messageToRun.get(agentId)?.get(threadId)?.get(messageId);
 	}
 	/**
+	* Get direct text-start metadata associated with a message.
+	*/
+	getRawEventForMessage(agentId, threadId, messageId) {
+		const rawEvent = this.rawEventByMessage.get(agentId)?.get(threadId)?.get(messageId);
+		return rawEvent === void 0 ? void 0 : (0, _ag_ui_client.structuredClone_)(rawEvent);
+	}
+	/**
 	* Get all states for an agent's thread
 	*/
 	getStatesForThread(agentId, threadId) {
@@ -2973,12 +2992,30 @@ var StateManager = class {
 		this.saveState(agent.agentId, threadId, runId, state);
 	}
 	/**
+	* Capture only defined metadata from a normalized direct text-start event.
+	*/
+	handleTextMessageStart(agent, event, input) {
+		if (!agent.agentId) return;
+		const { threadId } = input;
+		if (event.rawEvent === void 0) {
+			const threadEvents = this.rawEventByMessage.get(agent.agentId)?.get(threadId);
+			threadEvents?.delete(event.messageId);
+			if (threadEvents?.size === 0) this.rawEventByMessage.get(agent.agentId)?.delete(threadId);
+			if (this.rawEventByMessage.get(agent.agentId)?.size === 0) this.rawEventByMessage.delete(agent.agentId);
+			return;
+		}
+		if (!this.rawEventByMessage.has(agent.agentId)) this.rawEventByMessage.set(agent.agentId, /* @__PURE__ */ new Map());
+		const agentEvents = this.rawEventByMessage.get(agent.agentId);
+		if (!agentEvents.has(threadId)) agentEvents.set(threadId, /* @__PURE__ */ new Map());
+		agentEvents.get(threadId).set(event.messageId, event.rawEvent);
+	}
+	/**
 	* Handle messages snapshot event
 	*/
 	handleMessagesSnapshot(agent, event, input, messages) {
 		if (!agent.agentId) return;
 		const { threadId, runId } = input;
-		for (const message of event.messages) this.associateMessageWithRun(agent.agentId, threadId, message.id, runId);
+		for (const message of event.messages) if (this.getRunIdForMessage(agent.agentId, threadId, message.id) === void 0) this.associateMessageWithRun(agent.agentId, threadId, message.id, runId);
 	}
 	/**
 	* Handle new message event
@@ -3012,12 +3049,23 @@ var StateManager = class {
 		if (!agentMessages.has(threadId)) agentMessages.set(threadId, /* @__PURE__ */ new Map());
 		agentMessages.get(threadId).set(messageId, runId);
 	}
+	pruneRawEvents(agentId, fallbackThreadId, messages, input) {
+		const threadId = input?.threadId ?? fallbackThreadId;
+		if (!threadId) return;
+		const threadEvents = this.rawEventByMessage.get(agentId)?.get(threadId);
+		if (!threadEvents) return;
+		const messageIds = new Set(messages.map((message) => message.id));
+		for (const messageId of threadEvents.keys()) if (!messageIds.has(messageId)) threadEvents.delete(messageId);
+		if (threadEvents.size === 0) this.rawEventByMessage.get(agentId)?.delete(threadId);
+		if (this.rawEventByMessage.get(agentId)?.size === 0) this.rawEventByMessage.delete(agentId);
+	}
 	/**
 	* Clear all state for an agent
 	*/
 	clearAgentState(agentId) {
 		this.stateByRun.delete(agentId);
 		this.messageToRun.delete(agentId);
+		this.rawEventByMessage.delete(agentId);
 	}
 	/**
 	* Clear all state for a thread
@@ -3025,6 +3073,7 @@ var StateManager = class {
 	clearThreadState(agentId, threadId) {
 		this.stateByRun.get(agentId)?.delete(threadId);
 		this.messageToRun.get(agentId)?.delete(threadId);
+		this.rawEventByMessage.get(agentId)?.delete(threadId);
 	}
 };
 
@@ -4854,6 +4903,9 @@ var CopilotKitCore = class {
 	getRunIdForMessage(agentId, threadId, messageId) {
 		return this.stateManager.getRunIdForMessage(agentId, threadId, messageId);
 	}
+	getRawEventForMessage(agentId, threadId, messageId) {
+		return this.stateManager.getRawEventForMessage(agentId, threadId, messageId);
+	}
 	getRunIdsForThread(agentId, threadId) {
 		return this.stateManager.getRunIdsForThread(agentId, threadId);
 	}
@@ -5138,6 +5190,16 @@ function optionalString(value) {
 	return typeof value === "string" && value.length > 0 ? value : void 0;
 }
 /**
+* REST `/threads` is agent-scoped. The Phoenix `user_meta` channel is not.
+* Drop live upserts that belong to a different agent, or one Inspector
+* `all-agents` flatten shows the same thread twice until a hard refresh.
+*/
+function threadMetadataTargetsAgent(payload, agentId) {
+	if (!agentId || payload.operation === "deleted") return true;
+	const threadAgentId = optionalString(payload.thread.agentId);
+	return !threadAgentId || threadAgentId === agentId;
+}
+/**
 * Converts gateway run-activity payloads into the internal TypeScript shape.
 */
 function normalizeThreadRunActivityNotification(payload) {
@@ -5563,7 +5625,7 @@ function createThreadStore(environment) {
 					}))), ɵ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$)));
 				});
 			}))),
-			createEffect((actions$, state$) => actions$.pipe(ofType(threadSocketEvents.metadataReceived), (0, rxjs_operators.withLatestFrom)(state$), (0, rxjs_operators.filter)(([action, state]) => action.sessionId === state.sessionId), (0, rxjs_operators.map)(([action, state]) => {
+			createEffect((actions$, state$) => actions$.pipe(ofType(threadSocketEvents.metadataReceived), (0, rxjs_operators.withLatestFrom)(state$), (0, rxjs_operators.filter)(([action, state]) => action.sessionId === state.sessionId && threadMetadataTargetsAgent(action.payload, state.context?.agentId)), (0, rxjs_operators.map)(([action, state]) => {
 				if (action.payload.operation === "deleted") return threadDomainEvents.threadDeleted({
 					sessionId: action.sessionId,
 					threadId: action.payload.deleted.id