@copilotkit/react-core 1.55.0-next.7 → 1.55.0-next.9

package/dist/index.umd.js

@@ -1327,6 +1327,154 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 		]);
 	}
 
+//#endregion
+//#region src/v2/hooks/use-agent.tsx
+	let UseAgentUpdate = /* @__PURE__ */ function(UseAgentUpdate) {
+		UseAgentUpdate["OnMessagesChanged"] = "OnMessagesChanged";
+		UseAgentUpdate["OnStateChanged"] = "OnStateChanged";
+		UseAgentUpdate["OnRunStatusChanged"] = "OnRunStatusChanged";
+		return UseAgentUpdate;
+	}({});
+	const ALL_UPDATES = [
+		UseAgentUpdate.OnMessagesChanged,
+		UseAgentUpdate.OnStateChanged,
+		UseAgentUpdate.OnRunStatusChanged
+	];
+	/**
+	* Clone a registry agent for per-thread isolation.
+	* Copies agent configuration (transport, headers, etc.) but resets conversation
+	* state (messages, threadId, state) so each thread starts fresh.
+	*/
+	function cloneForThread(source, threadId, headers) {
+		const clone = source.clone();
+		if (clone === source) throw new Error(`useAgent: ${source.constructor.name}.clone() returned the same instance. clone() must return a new, independent object.`);
+		clone.threadId = threadId;
+		clone.setMessages([]);
+		clone.setState({});
+		if (clone instanceof _ag_ui_client.HttpAgent) clone.headers = { ...headers };
+		return clone;
+	}
+	/**
+	* Module-level WeakMap: registryAgent → (threadId → clone).
+	* Shared across all useAgent() calls so that every component using the same
+	* (agentId, threadId) pair receives the same agent instance. Using WeakMap
+	* ensures the clone map is garbage-collected when the registry agent is
+	* replaced (e.g. after reconnect or hot-reload).
+	*/
+	const globalThreadCloneMap = /* @__PURE__ */ new WeakMap();
+	/**
+	* Look up an existing per-thread clone without creating one.
+	* Returns undefined when no clone has been created yet for this pair.
+	*/
+	function getThreadClone(registryAgent, threadId) {
+		var _globalThreadCloneMap;
+		if (!registryAgent || !threadId) return void 0;
+		return (_globalThreadCloneMap = globalThreadCloneMap.get(registryAgent)) === null || _globalThreadCloneMap === void 0 ? void 0 : _globalThreadCloneMap.get(threadId);
+	}
+	function getOrCreateThreadClone(existing, threadId, headers) {
+		let byThread = globalThreadCloneMap.get(existing);
+		if (!byThread) {
+			byThread = /* @__PURE__ */ new Map();
+			globalThreadCloneMap.set(existing, byThread);
+		}
+		const cached = byThread.get(threadId);
+		if (cached) return cached;
+		const clone = cloneForThread(existing, threadId, headers);
+		byThread.set(threadId, clone);
+		return clone;
+	}
+	function useAgent({ agentId, threadId, updates } = {}) {
+		var _agentId, _threadId;
+		(_agentId = agentId) !== null && _agentId !== void 0 || (agentId = _copilotkit_shared.DEFAULT_AGENT_ID);
+		const { copilotkit } = useCopilotKit();
+		const chatConfig = useCopilotChatConfiguration();
+		(_threadId = threadId) !== null && _threadId !== void 0 || (threadId = chatConfig === null || chatConfig === void 0 ? void 0 : chatConfig.threadId);
+		const [, forceUpdate] = (0, react.useReducer)((x) => x + 1, 0);
+		const updateFlags = (0, react.useMemo)(() => updates !== null && updates !== void 0 ? updates : ALL_UPDATES, [JSON.stringify(updates)]);
+		const provisionalAgentCache = (0, react.useRef)(/* @__PURE__ */ new Map());
+		const agent = (0, react.useMemo)(() => {
+			var _copilotkit$agents;
+			const cacheKey = threadId ? `${agentId}:${threadId}` : agentId;
+			const existing = copilotkit.getAgent(agentId);
+			if (existing) {
+				provisionalAgentCache.current.delete(cacheKey);
+				provisionalAgentCache.current.delete(agentId);
+				if (!threadId) return existing;
+				return getOrCreateThreadClone(existing, threadId, copilotkit.headers);
+			}
+			const isRuntimeConfigured = copilotkit.runtimeUrl !== void 0;
+			const status = copilotkit.runtimeConnectionStatus;
+			if (isRuntimeConfigured && (status === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Disconnected || status === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Connecting)) {
+				const cached = provisionalAgentCache.current.get(cacheKey);
+				if (cached) {
+					cached.headers = { ...copilotkit.headers };
+					return cached;
+				}
+				const provisional = new _copilotkit_core.ProxiedCopilotRuntimeAgent({
+					runtimeUrl: copilotkit.runtimeUrl,
+					agentId,
+					transport: copilotkit.runtimeTransport,
+					runtimeMode: "pending"
+				});
+				provisional.headers = { ...copilotkit.headers };
+				if (threadId) provisional.threadId = threadId;
+				provisionalAgentCache.current.set(cacheKey, provisional);
+				return provisional;
+			}
+			if (isRuntimeConfigured && status === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Error) {
+				const cached = provisionalAgentCache.current.get(cacheKey);
+				if (cached) {
+					cached.headers = { ...copilotkit.headers };
+					return cached;
+				}
+				const provisional = new _copilotkit_core.ProxiedCopilotRuntimeAgent({
+					runtimeUrl: copilotkit.runtimeUrl,
+					agentId,
+					transport: copilotkit.runtimeTransport,
+					runtimeMode: "pending"
+				});
+				provisional.headers = { ...copilotkit.headers };
+				if (threadId) provisional.threadId = threadId;
+				provisionalAgentCache.current.set(cacheKey, provisional);
+				return provisional;
+			}
+			const knownAgents = Object.keys((_copilotkit$agents = copilotkit.agents) !== null && _copilotkit$agents !== void 0 ? _copilotkit$agents : {});
+			const runtimePart = isRuntimeConfigured ? `runtimeUrl=${copilotkit.runtimeUrl}` : "no runtimeUrl";
+			throw new Error(`useAgent: Agent '${agentId}' not found after runtime sync (${runtimePart}). ` + (knownAgents.length ? `Known agents: [${knownAgents.join(", ")}]` : "No agents registered.") + " Verify your runtime /info and/or agents__unsafe_dev_only.");
+		}, [
+			agentId,
+			threadId,
+			copilotkit.agents,
+			copilotkit.runtimeConnectionStatus,
+			copilotkit.runtimeUrl,
+			copilotkit.runtimeTransport,
+			JSON.stringify(copilotkit.headers)
+		]);
+		(0, react.useEffect)(() => {
+			if (updateFlags.length === 0) return;
+			const handlers = {};
+			if (updateFlags.includes(UseAgentUpdate.OnMessagesChanged)) handlers.onMessagesChanged = () => {
+				forceUpdate();
+			};
+			if (updateFlags.includes(UseAgentUpdate.OnStateChanged)) handlers.onStateChanged = forceUpdate;
+			if (updateFlags.includes(UseAgentUpdate.OnRunStatusChanged)) {
+				handlers.onRunInitialized = forceUpdate;
+				handlers.onRunFinalized = forceUpdate;
+				handlers.onRunFailed = forceUpdate;
+			}
+			const subscription = agent.subscribe(handlers);
+			return () => subscription.unsubscribe();
+		}, [
+			agent,
+			forceUpdate,
+			JSON.stringify(updateFlags)
+		]);
+		(0, react.useEffect)(() => {
+			if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...copilotkit.headers };
+		}, [agent, JSON.stringify(copilotkit.headers)]);
+		return { agent };
+	}
+
 //#endregion
 //#region src/v2/hooks/use-render-custom-messages.tsx
 	function useRenderCustomMessages() {
@@ -1340,12 +1488,13 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 			return aHasAgent ? -1 : 1;
 		});
 		return function(params) {
-			var _copilotkit$getRunIdF;
+			var _copilotkit$getRunIdF, _getThreadClone;
 			if (!customMessageRenderers.length) return null;
 			const { message, position } = params;
 			const resolvedRunId = (_copilotkit$getRunIdF = copilotkit.getRunIdForMessage(agentId, threadId, message.id)) !== null && _copilotkit$getRunIdF !== void 0 ? _copilotkit$getRunIdF : copilotkit.getRunIdsForThread(agentId, threadId).slice(-1)[0];
 			const runId = resolvedRunId !== null && resolvedRunId !== void 0 ? resolvedRunId : `missing-run-id:${message.id}`;
-			const agent = copilotkit.getAgent(agentId);
+			const registryAgent = copilotkit.getAgent(agentId);
+			const agent = (_getThreadClone = getThreadClone(registryAgent, threadId)) !== null && _getThreadClone !== void 0 ? _getThreadClone : registryAgent;
 			if (!agent) throw new Error("Agent not found");
 			const messagesIdsInRun = resolvedRunId ? agent.messages.filter((msg) => copilotkit.getRunIdForMessage(agentId, threadId, msg.id) === resolvedRunId).map((msg) => msg.id) : [message.id];
 			const rawMessageIndex = agent.messages.findIndex((msg) => msg.id === message.id);
@@ -1485,94 +1634,6 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 		]);
 	}
 
-//#endregion
-//#region src/v2/hooks/use-agent.tsx
-	let UseAgentUpdate = /* @__PURE__ */ function(UseAgentUpdate) {
-		UseAgentUpdate["OnMessagesChanged"] = "OnMessagesChanged";
-		UseAgentUpdate["OnStateChanged"] = "OnStateChanged";
-		UseAgentUpdate["OnRunStatusChanged"] = "OnRunStatusChanged";
-		return UseAgentUpdate;
-	}({});
-	const ALL_UPDATES = [
-		UseAgentUpdate.OnMessagesChanged,
-		UseAgentUpdate.OnStateChanged,
-		UseAgentUpdate.OnRunStatusChanged
-	];
-	function useAgent({ agentId, updates } = {}) {
-		var _agentId;
-		(_agentId = agentId) !== null && _agentId !== void 0 || (agentId = _copilotkit_shared.DEFAULT_AGENT_ID);
-		const { copilotkit } = useCopilotKit();
-		const [, forceUpdate] = (0, react.useReducer)((x) => x + 1, 0);
-		const updateFlags = (0, react.useMemo)(() => updates !== null && updates !== void 0 ? updates : ALL_UPDATES, [JSON.stringify(updates)]);
-		const provisionalAgentCache = (0, react.useRef)(/* @__PURE__ */ new Map());
-		const agent = (0, react.useMemo)(() => {
-			var _copilotkit$agents;
-			const existing = copilotkit.getAgent(agentId);
-			if (existing) {
-				provisionalAgentCache.current.delete(agentId);
-				return existing;
-			}
-			const isRuntimeConfigured = copilotkit.runtimeUrl !== void 0;
-			const status = copilotkit.runtimeConnectionStatus;
-			if (isRuntimeConfigured && (status === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Disconnected || status === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Connecting)) {
-				const cached = provisionalAgentCache.current.get(agentId);
-				if (cached) {
-					cached.headers = { ...copilotkit.headers };
-					return cached;
-				}
-				const provisional = new _copilotkit_core.ProxiedCopilotRuntimeAgent({
-					runtimeUrl: copilotkit.runtimeUrl,
-					agentId,
-					transport: copilotkit.runtimeTransport,
-					runtimeMode: "pending"
-				});
-				provisional.headers = { ...copilotkit.headers };
-				provisionalAgentCache.current.set(agentId, provisional);
-				return provisional;
-			}
-			if (isRuntimeConfigured && status === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Error) {
-				const provisional = new _copilotkit_core.ProxiedCopilotRuntimeAgent({
-					runtimeUrl: copilotkit.runtimeUrl,
-					agentId,
-					transport: copilotkit.runtimeTransport,
-					runtimeMode: "pending"
-				});
-				provisional.headers = { ...copilotkit.headers };
-				return provisional;
-			}
-			const knownAgents = Object.keys((_copilotkit$agents = copilotkit.agents) !== null && _copilotkit$agents !== void 0 ? _copilotkit$agents : {});
-			const runtimePart = isRuntimeConfigured ? `runtimeUrl=${copilotkit.runtimeUrl}` : "no runtimeUrl";
-			throw new Error(`useAgent: Agent '${agentId}' not found after runtime sync (${runtimePart}). ` + (knownAgents.length ? `Known agents: [${knownAgents.join(", ")}]` : "No agents registered.") + " Verify your runtime /info and/or agents__unsafe_dev_only.");
-		}, [
-			agentId,
-			copilotkit.agents,
-			copilotkit.runtimeConnectionStatus,
-			copilotkit.runtimeUrl,
-			copilotkit.runtimeTransport,
-			JSON.stringify(copilotkit.headers)
-		]);
-		(0, react.useEffect)(() => {
-			if (updateFlags.length === 0) return;
-			const handlers = {};
-			if (updateFlags.includes(UseAgentUpdate.OnMessagesChanged)) handlers.onMessagesChanged = () => {
-				forceUpdate();
-			};
-			if (updateFlags.includes(UseAgentUpdate.OnStateChanged)) handlers.onStateChanged = forceUpdate;
-			if (updateFlags.includes(UseAgentUpdate.OnRunStatusChanged)) {
-				handlers.onRunInitialized = forceUpdate;
-				handlers.onRunFinalized = forceUpdate;
-				handlers.onRunFailed = forceUpdate;
-			}
-			const subscription = agent.subscribe(handlers);
-			return () => subscription.unsubscribe();
-		}, [
-			agent,
-			forceUpdate,
-			JSON.stringify(updateFlags)
-		]);
-		return { agent };
-	}
-
 //#endregion
 //#region src/v2/hooks/use-suggestions.tsx
 	function useSuggestions({ agentId } = {}) {
@@ -3967,7 +4028,7 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 				children: [
 					/* @__PURE__ */ (0, react_jsx_runtime.jsx)(CopilotListeners, {}),
 					/* @__PURE__ */ (0, react_jsx_runtime.jsx)(CopilotKitErrorBridge, {}),
-					/* @__PURE__ */ (0, react_jsx_runtime.jsxs)(CoAgentStateRendersProvider, { children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)(MessagesTapProvider, { children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(CopilotMessages, { children: [memoizedChildren, /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RegisteredActionsRenderer, {})] }) }), bannerError && showDevConsole && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(UsageBanner, {
+					/* @__PURE__ */ (0, react_jsx_runtime.jsxs)(CoAgentStateRendersProvider, { children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)(MessagesTapProvider, { children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(CopilotMessages, { children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)(react.default.Fragment, { children: memoizedChildren }, "children"), /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RegisteredActionsRenderer, {}, "actions")] }) }), bannerError && showDevConsole && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(UsageBanner, {
 						severity: bannerError.severity,
 						message: bannerError.message,
 						onClose: () => setBannerError(null),
@@ -4037,7 +4098,11 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 		const existingConfig = useCopilotChatConfiguration();
 		const [agentAvailable, setAgentAvailable] = (0, react.useState)(false);
 		const resolvedAgentId = (_existingConfig$agent = existingConfig === null || existingConfig === void 0 ? void 0 : existingConfig.agentId) !== null && _existingConfig$agent !== void 0 ? _existingConfig$agent : "default";
-		const { agent } = useAgent({ agentId: resolvedAgentId });
+		const { agent } = useAgent({
+			agentId: resolvedAgentId,
+			threadId: existingConfig === null || existingConfig === void 0 ? void 0 : existingConfig.threadId
+		});
+		const lastConnectedAgentRef = (0, react.useRef)(null);
 		(0, react.useEffect)(() => {
 			let detached = false;
 			const connectAbortController = new AbortController();
@@ -4052,11 +4117,12 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 					if (error instanceof _ag_ui_client.AGUIConnectNotImplementedError) {} else console.error("CopilotChat: connectAgent failed", error);
 				}
 			};
-			if (agent && (existingConfig === null || existingConfig === void 0 ? void 0 : existingConfig.threadId) && agent.threadId !== existingConfig.threadId && copilotkit.runtimeConnectionStatus === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Connected) {
-				agent.threadId = existingConfig.threadId;
+			if (agent && agent !== lastConnectedAgentRef.current && copilotkit.runtimeConnectionStatus === _copilotkit_core.CopilotKitCoreRuntimeConnectionStatus.Connected) {
+				lastConnectedAgentRef.current = agent;
 				connect(agent);
 			}
 			return () => {
+				lastConnectedAgentRef.current = null;
 				detached = true;
 				connectAbortController.abort();
 				agent === null || agent === void 0 || agent.detachActiveRun();