@copilotkit/core 1.68.3 → 1.69.0

package/dist/index.umd.js

@@ -2938,6 +2938,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			this.core = core;
 			_defineProperty(this, "stateByRun", /* @__PURE__ */ new Map());
 			_defineProperty(this, "messageToRun", /* @__PURE__ */ new Map());
+			_defineProperty(this, "rawEventByMessage", /* @__PURE__ */ new Map());
 			_defineProperty(this, "activeRun", /* @__PURE__ */ new Map());
 			_defineProperty(this, "agentSubscriptions", /* @__PURE__ */ new Map());
 			_defineProperty(this, "pendingContinuations", /* @__PURE__ */ new WeakMap());
@@ -2990,16 +2991,16 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				};
 			};
 			const { unsubscribe } = agent.subscribe({
-				onRunStartedEvent: ({ input, state }) => {
+				onRunStartedEvent: ({ event, input, state }) => {
 					if (revoked) return;
 					const pendingForAgent = this.pendingContinuations.get(agent);
 					const internalContinuation = [...pendingForAgent !== null && pendingForAgent !== void 0 ? pendingForAgent : []].find((pending) => pending.expectedInput === input);
 					internalContinuation === null || internalContinuation === void 0 || internalContinuation.cancel();
-					if (runFinished && input.runId === subRunId && !internalContinuation && !isContinuation(input)) subRunId = (0, _ag_ui_client.randomUUID)();
-					else {
-						var _internalContinuation;
-						subRunId = (_internalContinuation = internalContinuation === null || internalContinuation === void 0 ? void 0 : internalContinuation.expectedRunId) !== null && _internalContinuation !== void 0 ? _internalContinuation : input.runId;
-					}
+					if (internalContinuation) {
+						var _ref, _internalContinuation;
+						subRunId = (_ref = (_internalContinuation = internalContinuation.expectedRunId) !== null && _internalContinuation !== void 0 ? _internalContinuation : event.runId) !== null && _ref !== void 0 ? _ref : 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);
 				},
@@ -3021,13 +3022,22 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 					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, () => {
@@ -3045,6 +3055,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 				unsubscribe();
 				this.agentSubscriptions.delete(agentId);
 			}
+			this.rawEventByMessage.delete(agentId);
 		}
 		/**
 		* Get state for a specific run
@@ -3064,6 +3075,14 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			return (_this$messageToRun$ge = this.messageToRun.get(agentId)) === null || _this$messageToRun$ge === void 0 || (_this$messageToRun$ge = _this$messageToRun$ge.get(threadId)) === null || _this$messageToRun$ge === void 0 ? void 0 : _this$messageToRun$ge.get(messageId);
 		}
 		/**
+		* Get direct text-start metadata associated with a message.
+		*/
+		getRawEventForMessage(agentId, threadId, messageId) {
+			var _this$rawEventByMessa;
+			const rawEvent = (_this$rawEventByMessa = this.rawEventByMessage.get(agentId)) === null || _this$rawEventByMessa === void 0 || (_this$rawEventByMessa = _this$rawEventByMessa.get(threadId)) === null || _this$rawEventByMessa === void 0 ? void 0 : _this$rawEventByMessa.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) {
@@ -3117,12 +3136,34 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			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) {
+				var _this$rawEventByMessa2, _this$rawEventByMessa4;
+				const threadEvents = (_this$rawEventByMessa2 = this.rawEventByMessage.get(agent.agentId)) === null || _this$rawEventByMessa2 === void 0 ? void 0 : _this$rawEventByMessa2.get(threadId);
+				threadEvents === null || threadEvents === void 0 || threadEvents.delete(event.messageId);
+				if ((threadEvents === null || threadEvents === void 0 ? void 0 : threadEvents.size) === 0) {
+					var _this$rawEventByMessa3;
+					(_this$rawEventByMessa3 = this.rawEventByMessage.get(agent.agentId)) === null || _this$rawEventByMessa3 === void 0 || _this$rawEventByMessa3.delete(threadId);
+				}
+				if (((_this$rawEventByMessa4 = this.rawEventByMessage.get(agent.agentId)) === null || _this$rawEventByMessa4 === void 0 ? void 0 : _this$rawEventByMessa4.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
@@ -3157,20 +3198,36 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 			if (!agentMessages.has(threadId)) agentMessages.set(threadId, /* @__PURE__ */ new Map());
 			agentMessages.get(threadId).set(messageId, runId);
 		}
+		pruneRawEvents(agentId, fallbackThreadId, messages, input) {
+			var _input$threadId, _this$rawEventByMessa5, _this$rawEventByMessa7;
+			const threadId = (_input$threadId = input === null || input === void 0 ? void 0 : input.threadId) !== null && _input$threadId !== void 0 ? _input$threadId : fallbackThreadId;
+			if (!threadId) return;
+			const threadEvents = (_this$rawEventByMessa5 = this.rawEventByMessage.get(agentId)) === null || _this$rawEventByMessa5 === void 0 ? void 0 : _this$rawEventByMessa5.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) {
+				var _this$rawEventByMessa6;
+				(_this$rawEventByMessa6 = this.rawEventByMessage.get(agentId)) === null || _this$rawEventByMessa6 === void 0 || _this$rawEventByMessa6.delete(threadId);
+			}
+			if (((_this$rawEventByMessa7 = this.rawEventByMessage.get(agentId)) === null || _this$rawEventByMessa7 === void 0 ? void 0 : _this$rawEventByMessa7.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
 		*/
 		clearThreadState(agentId, threadId) {
-			var _this$stateByRun$get4, _this$messageToRun$ge2;
+			var _this$stateByRun$get4, _this$messageToRun$ge2, _this$rawEventByMessa8;
 			(_this$stateByRun$get4 = this.stateByRun.get(agentId)) === null || _this$stateByRun$get4 === void 0 || _this$stateByRun$get4.delete(threadId);
 			(_this$messageToRun$ge2 = this.messageToRun.get(agentId)) === null || _this$messageToRun$ge2 === void 0 || _this$messageToRun$ge2.delete(threadId);
+			(_this$rawEventByMessa8 = this.rawEventByMessage.get(agentId)) === null || _this$rawEventByMessa8 === void 0 || _this$rawEventByMessa8.delete(threadId);
 		}
 	};
 
@@ -5016,6 +5073,9 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		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);
 		}
@@ -5301,6 +5361,16 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 		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) {
@@ -5733,13 +5803,16 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 						}))), ɵ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]) => {
 					var _state$context4;
+					return action.sessionId === state.sessionId && threadMetadataTargetsAgent(action.payload, (_state$context4 = state.context) === null || _state$context4 === void 0 ? void 0 : _state$context4.agentId);
+				}), (0, rxjs_operators.map)(([action, state]) => {
+					var _state$context5;
 					if (action.payload.operation === "deleted") return threadDomainEvents.threadDeleted({
 						sessionId: action.sessionId,
 						threadId: action.payload.deleted.id
 					});
-					if (action.payload.operation === "archived" && !((_state$context4 = state.context) === null || _state$context4 === void 0 ? void 0 : _state$context4.includeArchived)) return threadDomainEvents.threadDeleted({
+					if (action.payload.operation === "archived" && !((_state$context5 = state.context) === null || _state$context5 === void 0 ? void 0 : _state$context5.includeArchived)) return threadDomainEvents.threadDeleted({
 						sessionId: action.sessionId,
 						threadId: action.payload.threadId
 					});