@copilotkit/react-core 1.57.0 → 1.57.1

package/dist/index.umd.js

@@ -203,6 +203,206 @@ react_markdown = __toESM(react_markdown);
 		return (0, react.useContext)(CopilotChatConfiguration);
 	};
 
+//#endregion
+//#region src/v2/lib/react-core.ts
+	var CopilotKitCoreReact = class extends _copilotkit_core.CopilotKitCore {
+		constructor(config) {
+			var _config$renderToolCal, _config$renderCustomM, _config$renderActivit;
+			super(config);
+			this._renderToolCalls = [];
+			this._hookRenderToolCalls = /* @__PURE__ */ new Map();
+			this._cachedMergedRenderToolCalls = null;
+			this._renderCustomMessages = [];
+			this._renderActivityMessages = [];
+			this._interruptElement = null;
+			this._renderToolCalls = (_config$renderToolCal = config.renderToolCalls) !== null && _config$renderToolCal !== void 0 ? _config$renderToolCal : [];
+			this._renderCustomMessages = (_config$renderCustomM = config.renderCustomMessages) !== null && _config$renderCustomM !== void 0 ? _config$renderCustomM : [];
+			this._renderActivityMessages = (_config$renderActivit = config.renderActivityMessages) !== null && _config$renderActivit !== void 0 ? _config$renderActivit : [];
+		}
+		get renderCustomMessages() {
+			return this._renderCustomMessages;
+		}
+		get renderActivityMessages() {
+			return this._renderActivityMessages;
+		}
+		get renderToolCalls() {
+			if (this._hookRenderToolCalls.size === 0) return this._renderToolCalls;
+			if (this._cachedMergedRenderToolCalls) return this._cachedMergedRenderToolCalls;
+			const merged = /* @__PURE__ */ new Map();
+			for (const rc of this._renderToolCalls) {
+				var _rc$agentId;
+				merged.set(`${(_rc$agentId = rc.agentId) !== null && _rc$agentId !== void 0 ? _rc$agentId : ""}:${rc.name}`, rc);
+			}
+			for (const [key, rc] of this._hookRenderToolCalls) merged.set(key, rc);
+			this._cachedMergedRenderToolCalls = Array.from(merged.values());
+			return this._cachedMergedRenderToolCalls;
+		}
+		setRenderActivityMessages(renderers) {
+			this._renderActivityMessages = renderers;
+		}
+		setRenderCustomMessages(renderers) {
+			this._renderCustomMessages = renderers;
+		}
+		setRenderToolCalls(renderToolCalls) {
+			this._renderToolCalls = renderToolCalls;
+			this._cachedMergedRenderToolCalls = null;
+			this._notifyRenderToolCallsChanged();
+		}
+		addHookRenderToolCall(entry) {
+			var _entry$agentId;
+			const key = `${(_entry$agentId = entry.agentId) !== null && _entry$agentId !== void 0 ? _entry$agentId : ""}:${entry.name}`;
+			this._hookRenderToolCalls.set(key, entry);
+			this._cachedMergedRenderToolCalls = null;
+			this._notifyRenderToolCallsChanged();
+		}
+		removeHookRenderToolCall(name, agentId) {
+			const key = `${agentId !== null && agentId !== void 0 ? agentId : ""}:${name}`;
+			if (this._hookRenderToolCalls.delete(key)) {
+				this._cachedMergedRenderToolCalls = null;
+				this._notifyRenderToolCallsChanged();
+			}
+		}
+		_notifyRenderToolCallsChanged() {
+			this.notifySubscribers((subscriber) => {
+				const reactSubscriber = subscriber;
+				if (reactSubscriber.onRenderToolCallsChanged) reactSubscriber.onRenderToolCallsChanged({
+					copilotkit: this,
+					renderToolCalls: this.renderToolCalls
+				});
+			}, "Subscriber onRenderToolCallsChanged error:");
+		}
+		get interruptElement() {
+			return this._interruptElement;
+		}
+		setInterruptElement(element) {
+			this._interruptElement = element;
+			this.notifySubscribers((subscriber) => {
+				var _reactSubscriber$onIn;
+				const reactSubscriber = subscriber;
+				(_reactSubscriber$onIn = reactSubscriber.onInterruptElementChanged) === null || _reactSubscriber$onIn === void 0 || _reactSubscriber$onIn.call(reactSubscriber, {
+					copilotkit: this,
+					interruptElement: this._interruptElement
+				});
+			}, "Subscriber onInterruptElementChanged error:");
+		}
+		subscribe(subscriber) {
+			return super.subscribe(subscriber);
+		}
+		/**
+		* Wait for pending React state updates before the follow-up agent run.
+		*
+		* When a frontend tool handler calls setState(), React 18 batches the update
+		* and schedules a commit via its internal scheduler (MessageChannel). The
+		* useAgentContext hook registers context via useLayoutEffect, which runs
+		* synchronously after React commits that batch.
+		*
+		* Awaiting a zero-delay timeout yields to the macrotask queue. React's
+		* MessageChannel task runs first, committing the pending state and running
+		* useLayoutEffect (which updates the context store). The follow-up runAgent
+		* call then reads fresh context.
+		*/
+		async waitForPendingFrameworkUpdates() {
+			await new Promise((resolve) => setTimeout(resolve, 0));
+		}
+	};
+
+//#endregion
+//#region src/v2/context.ts
+	const CopilotKitContext = (0, react.createContext)(null);
+	const useCopilotKit = () => {
+		const context = (0, react.useContext)(CopilotKitContext);
+		const [, forceUpdate] = (0, react.useReducer)((x) => x + 1, 0);
+		if (!context) throw new Error("useCopilotKit must be used within CopilotKitProvider");
+		(0, react.useEffect)(() => {
+			const subscription = context.copilotkit.subscribe({ onRuntimeConnectionStatusChanged: () => {
+				forceUpdate();
+			} });
+			return () => {
+				subscription.unsubscribe();
+			};
+		}, []);
+		return context;
+	};
+	const LicenseContext = (0, react.createContext)({
+		status: null,
+		license: null,
+		checkFeature: () => true,
+		getLimit: () => null
+	});
+
+//#endregion
+//#region src/v2/hooks/use-render-tool-call.tsx
+/**
+	* Memoized component that renders a single tool call.
+	* This prevents unnecessary re-renders when parent components update
+	* but the tool call data hasn't changed.
+	*/
+	const ToolCallRenderer = react.default.memo(function ToolCallRenderer({ toolCall, toolMessage, RenderComponent, isExecuting }) {
+		const args = (0, react.useMemo)(() => (0, _copilotkit_shared.partialJSONParse)(toolCall.function.arguments), [toolCall.function.arguments]);
+		const toolName = toolCall.function.name;
+		if (toolMessage) return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RenderComponent, {
+			name: toolName,
+			toolCallId: toolCall.id,
+			args,
+			status: _copilotkit_core.ToolCallStatus.Complete,
+			result: toolMessage.content
+		});
+		else if (isExecuting) return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RenderComponent, {
+			name: toolName,
+			toolCallId: toolCall.id,
+			args,
+			status: _copilotkit_core.ToolCallStatus.Executing,
+			result: void 0
+		});
+		else return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RenderComponent, {
+			name: toolName,
+			toolCallId: toolCall.id,
+			args,
+			status: _copilotkit_core.ToolCallStatus.InProgress,
+			result: void 0
+		});
+	}, (prevProps, nextProps) => {
+		var _prevProps$toolMessag, _nextProps$toolMessag;
+		if (prevProps.toolCall.id !== nextProps.toolCall.id) return false;
+		if (prevProps.toolCall.function.name !== nextProps.toolCall.function.name) return false;
+		if (prevProps.toolCall.function.arguments !== nextProps.toolCall.function.arguments) return false;
+		if (((_prevProps$toolMessag = prevProps.toolMessage) === null || _prevProps$toolMessag === void 0 ? void 0 : _prevProps$toolMessag.content) !== ((_nextProps$toolMessag = nextProps.toolMessage) === null || _nextProps$toolMessag === void 0 ? void 0 : _nextProps$toolMessag.content)) return false;
+		if (prevProps.isExecuting !== nextProps.isExecuting) return false;
+		if (prevProps.RenderComponent !== nextProps.RenderComponent) return false;
+		return true;
+	});
+	/**
+	* Hook that returns a function to render tool calls based on the render functions
+	* defined in CopilotKitProvider.
+	*
+	* @returns A function that takes a tool call and optional tool message and returns the rendered component
+	*/
+	function useRenderToolCall$1() {
+		var _config$agentId;
+		const { copilotkit, executingToolCallIds } = useCopilotKit();
+		const config = useCopilotChatConfiguration();
+		const agentId = (_config$agentId = config === null || config === void 0 ? void 0 : config.agentId) !== null && _config$agentId !== void 0 ? _config$agentId : _copilotkit_shared.DEFAULT_AGENT_ID;
+		const renderToolCalls = (0, react.useSyncExternalStore)((callback) => {
+			return copilotkit.subscribe({ onRenderToolCallsChanged: callback }).unsubscribe;
+		}, () => copilotkit.renderToolCalls, () => copilotkit.renderToolCalls);
+		return (0, react.useCallback)(({ toolCall, toolMessage }) => {
+			const exactMatches = renderToolCalls.filter((rc) => rc.name === toolCall.function.name);
+			const renderConfig = exactMatches.find((rc) => rc.agentId === agentId) || exactMatches.find((rc) => !rc.agentId) || exactMatches[0] || renderToolCalls.find((rc) => rc.name === "*");
+			if (!renderConfig) return null;
+			const RenderComponent = renderConfig.render;
+			return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(ToolCallRenderer, {
+				toolCall,
+				toolMessage,
+				RenderComponent,
+				isExecuting: executingToolCallIds.has(toolCall.id)
+			}, toolCall.id);
+		}, [
+			renderToolCalls,
+			executingToolCallIds,
+			agentId
+		]);
+	}
+
 //#endregion
 //#region src/v2/components/CopilotKitInspector.tsx
 	const CopilotKitInspector = ({ core, ...rest }) => {
@@ -1835,109 +2035,6 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 		return null;
 	}
 
-//#endregion
-//#region src/v2/lib/react-core.ts
-	var CopilotKitCoreReact = class extends _copilotkit_core.CopilotKitCore {
-		constructor(config) {
-			var _config$renderToolCal, _config$renderCustomM, _config$renderActivit;
-			super(config);
-			this._renderToolCalls = [];
-			this._hookRenderToolCalls = /* @__PURE__ */ new Map();
-			this._cachedMergedRenderToolCalls = null;
-			this._renderCustomMessages = [];
-			this._renderActivityMessages = [];
-			this._interruptElement = null;
-			this._renderToolCalls = (_config$renderToolCal = config.renderToolCalls) !== null && _config$renderToolCal !== void 0 ? _config$renderToolCal : [];
-			this._renderCustomMessages = (_config$renderCustomM = config.renderCustomMessages) !== null && _config$renderCustomM !== void 0 ? _config$renderCustomM : [];
-			this._renderActivityMessages = (_config$renderActivit = config.renderActivityMessages) !== null && _config$renderActivit !== void 0 ? _config$renderActivit : [];
-		}
-		get renderCustomMessages() {
-			return this._renderCustomMessages;
-		}
-		get renderActivityMessages() {
-			return this._renderActivityMessages;
-		}
-		get renderToolCalls() {
-			if (this._hookRenderToolCalls.size === 0) return this._renderToolCalls;
-			if (this._cachedMergedRenderToolCalls) return this._cachedMergedRenderToolCalls;
-			const merged = /* @__PURE__ */ new Map();
-			for (const rc of this._renderToolCalls) {
-				var _rc$agentId;
-				merged.set(`${(_rc$agentId = rc.agentId) !== null && _rc$agentId !== void 0 ? _rc$agentId : ""}:${rc.name}`, rc);
-			}
-			for (const [key, rc] of this._hookRenderToolCalls) merged.set(key, rc);
-			this._cachedMergedRenderToolCalls = Array.from(merged.values());
-			return this._cachedMergedRenderToolCalls;
-		}
-		setRenderActivityMessages(renderers) {
-			this._renderActivityMessages = renderers;
-		}
-		setRenderCustomMessages(renderers) {
-			this._renderCustomMessages = renderers;
-		}
-		setRenderToolCalls(renderToolCalls) {
-			this._renderToolCalls = renderToolCalls;
-			this._cachedMergedRenderToolCalls = null;
-			this._notifyRenderToolCallsChanged();
-		}
-		addHookRenderToolCall(entry) {
-			var _entry$agentId;
-			const key = `${(_entry$agentId = entry.agentId) !== null && _entry$agentId !== void 0 ? _entry$agentId : ""}:${entry.name}`;
-			this._hookRenderToolCalls.set(key, entry);
-			this._cachedMergedRenderToolCalls = null;
-			this._notifyRenderToolCallsChanged();
-		}
-		removeHookRenderToolCall(name, agentId) {
-			const key = `${agentId !== null && agentId !== void 0 ? agentId : ""}:${name}`;
-			if (this._hookRenderToolCalls.delete(key)) {
-				this._cachedMergedRenderToolCalls = null;
-				this._notifyRenderToolCallsChanged();
-			}
-		}
-		_notifyRenderToolCallsChanged() {
-			this.notifySubscribers((subscriber) => {
-				const reactSubscriber = subscriber;
-				if (reactSubscriber.onRenderToolCallsChanged) reactSubscriber.onRenderToolCallsChanged({
-					copilotkit: this,
-					renderToolCalls: this.renderToolCalls
-				});
-			}, "Subscriber onRenderToolCallsChanged error:");
-		}
-		get interruptElement() {
-			return this._interruptElement;
-		}
-		setInterruptElement(element) {
-			this._interruptElement = element;
-			this.notifySubscribers((subscriber) => {
-				var _reactSubscriber$onIn;
-				const reactSubscriber = subscriber;
-				(_reactSubscriber$onIn = reactSubscriber.onInterruptElementChanged) === null || _reactSubscriber$onIn === void 0 || _reactSubscriber$onIn.call(reactSubscriber, {
-					copilotkit: this,
-					interruptElement: this._interruptElement
-				});
-			}, "Subscriber onInterruptElementChanged error:");
-		}
-		subscribe(subscriber) {
-			return super.subscribe(subscriber);
-		}
-		/**
-		* Wait for pending React state updates before the follow-up agent run.
-		*
-		* When a frontend tool handler calls setState(), React 18 batches the update
-		* and schedules a commit via its internal scheduler (MessageChannel). The
-		* useAgentContext hook registers context via useLayoutEffect, which runs
-		* synchronously after React commits that batch.
-		*
-		* Awaiting a zero-delay timeout yields to the macrotask queue. React's
-		* MessageChannel task runs first, committing the pending state and running
-		* useLayoutEffect (which updates the context store). The follow-up runAgent
-		* call then reads fresh context.
-		*/
-		async waitForPendingFrameworkUpdates() {
-			await new Promise((resolve) => setTimeout(resolve, 0));
-		}
-	};
-
 //#endregion
 //#region src/v2/providers/CopilotKitProvider.tsx
 	const HEADER_NAME = "X-CopilotCloud-Public-Api-Key";
@@ -1953,11 +2050,6 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 - Minimal transitions (150ms) for hover/focus states only. No decorative animations.
 - Keep the UI focused and dense — avoid excessive padding. Use compact spacing (8–12px gaps, 10–14px padding in controls).`;
 	const GENERATE_SANDBOXED_UI_DESCRIPTION = "Generate sandboxed UI. IMPORTANT: The generated code runs in a sandboxed iframe WITHOUT same-origin access. Do NOT use localStorage, sessionStorage, document.cookie, IndexedDB, or fetch/XMLHttpRequest to same-origin URLs. To communicate with the host application, use Websandbox.connection.remote.<functionName>(args) which returns a Promise.\n\nYou CAN use external libraries from CDNs by including <script> or <link> tags in the HTML <head> (e.g., Chart.js, D3, Three.js, x-data-spreadsheet, etc.). CDN resources load normally inside the sandbox.\n\nPARAMETER ORDER IS CRITICAL — generate parameters in exactly this order:\n1. initialHeight + placeholderMessages (shown to user while generating)\n2. css (all styles FIRST — the user sees a placeholder until CSS is complete)\n3. html (streams in live — the user watches the UI build as HTML is generated)\n4. jsFunctions (reusable helper functions)\n5. jsExpressions (applied one-by-one — the user sees each expression take effect)";
-	const CopilotKitContext = (0, react.createContext)({
-		copilotkit: null,
-		executingToolCallIds: /* @__PURE__ */ new Set()
-	});
-	const LicenseContext = (0, react.createContext)((0, _copilotkit_shared.createLicenseContextValue)(null));
 	function useStableArrayProp(prop, warningMessage, isMeaningfulChange) {
 		const empty = (0, react.useMemo)(() => [], []);
 		const value = prop !== null && prop !== void 0 ? prop : empty;
@@ -2167,399 +2259,145 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 						next.add(toolCallId);
 						return next;
 					});
-				},
-				onToolExecutionEnd: ({ toolCallId }) => {
-					setExecutingToolCallIds((prev) => {
-						if (!prev.has(toolCallId)) return prev;
-						const next = new Set(prev);
-						next.delete(toolCallId);
-						return next;
-					});
-				}
-			});
-			return () => {
-				subscription.unsubscribe();
-			};
-		}, [copilotkit]);
-		const onErrorRef = (0, react.useRef)(onError);
-		(0, react.useEffect)(() => {
-			onErrorRef.current = onError;
-		}, [onError]);
-		(0, react.useEffect)(() => {
-			if (!onErrorRef.current) return;
-			const subscription = copilotkit.subscribe({ onError: (event) => {
-				var _onErrorRef$current;
-				(_onErrorRef$current = onErrorRef.current) === null || _onErrorRef$current === void 0 || _onErrorRef$current.call(onErrorRef, {
-					error: event.error,
-					code: event.code,
-					context: event.context
-				});
-			} });
-			return () => {
-				subscription.unsubscribe();
-			};
-		}, [copilotkit]);
-		(0, react.useEffect)(() => {
-			copilotkit.setRuntimeUrl(chatApiEndpoint);
-			copilotkit.setRuntimeTransport(useSingleEndpoint === true ? "single" : useSingleEndpoint === false ? "rest" : "auto");
-			copilotkit.setHeaders(mergedHeaders);
-			copilotkit.setCredentials(credentials);
-			copilotkit.setProperties(properties);
-			copilotkit.setAgents__unsafe_dev_only(mergedAgents);
-			copilotkit.setDebug(debug);
-		}, [
-			copilotkit,
-			chatApiEndpoint,
-			mergedHeaders,
-			credentials,
-			properties,
-			mergedAgents,
-			useSingleEndpoint,
-			debug
-		]);
-		const didMountRef = (0, react.useRef)(false);
-		(0, react.useEffect)(() => {
-			if (!didMountRef.current) return;
-			copilotkit.setTools(allTools);
-		}, [copilotkit, allTools]);
-		(0, react.useEffect)(() => {
-			if (!didMountRef.current) return;
-			copilotkit.setRenderToolCalls(allRenderToolCalls);
-		}, [copilotkit, allRenderToolCalls]);
-		(0, react.useEffect)(() => {
-			if (!didMountRef.current) return;
-			copilotkit.setRenderActivityMessages(allActivityRenderers);
-		}, [copilotkit, allActivityRenderers]);
-		(0, react.useEffect)(() => {
-			if (!didMountRef.current) return;
-			copilotkit.setRenderCustomMessages(renderCustomMessagesList);
-		}, [copilotkit, renderCustomMessagesList]);
-		(0, react.useEffect)(() => {
-			didMountRef.current = true;
-		}, []);
-		(0, react.useEffect)(() => {
-			copilotkit.setDefaultThrottleMs(defaultThrottleMs);
-		}, [copilotkit, defaultThrottleMs]);
-		const designSkill = (_openGenerativeUI$des = openGenerativeUI === null || openGenerativeUI === void 0 ? void 0 : openGenerativeUI.designSkill) !== null && _openGenerativeUI$des !== void 0 ? _openGenerativeUI$des : DEFAULT_DESIGN_SKILL;
-		(0, react.useLayoutEffect)(() => {
-			if (!copilotkit || !openGenUIActive) return;
-			const id = copilotkit.addContext({
-				description: "Design guidelines for the generateSandboxedUi tool. Follow these when building UI.",
-				value: designSkill
-			});
-			return () => {
-				copilotkit.removeContext(id);
-			};
-		}, [
-			copilotkit,
-			designSkill,
-			openGenUIActive
-		]);
-		const sandboxFunctionsDescriptors = (0, react.useMemo)(() => {
-			if (sandboxFunctionsList.length === 0) return null;
-			return JSON.stringify(sandboxFunctionsList.map((fn) => ({
-				name: fn.name,
-				description: fn.description,
-				parameters: (0, _copilotkit_shared.schemaToJsonSchema)(fn.parameters, { zodToJsonSchema: zod_to_json_schema.zodToJsonSchema })
-			})));
-		}, [sandboxFunctionsList]);
-		(0, react.useLayoutEffect)(() => {
-			if (!copilotkit || !sandboxFunctionsDescriptors || !openGenUIActive) return;
-			const id = copilotkit.addContext({
-				description: "Sandbox functions available in generated sandboxed UI code. Call via: await Websandbox.connection.remote.<functionName>(args)",
-				value: sandboxFunctionsDescriptors
-			});
-			return () => {
-				copilotkit.removeContext(id);
-			};
-		}, [
-			copilotkit,
-			sandboxFunctionsDescriptors,
-			openGenUIActive
-		]);
-		const contextValue = (0, react.useMemo)(() => ({
-			copilotkit,
-			executingToolCallIds
-		}), [copilotkit, executingToolCallIds]);
-		const licenseContextValue = (0, react.useMemo)(() => (0, _copilotkit_shared.createLicenseContextValue)(null), []);
-		return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(SandboxFunctionsContext.Provider, {
-			value: sandboxFunctionsList,
-			children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)(CopilotKitContext.Provider, {
-				value: contextValue,
-				children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(LicenseContext.Provider, {
-					value: licenseContextValue,
-					children: [
-						runtimeA2UIEnabled && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(A2UIBuiltInToolCallRenderer, {}),
-						runtimeA2UIEnabled && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(A2UICatalogContext, {
-							catalog: a2ui === null || a2ui === void 0 ? void 0 : a2ui.catalog,
-							includeSchema: a2ui === null || a2ui === void 0 ? void 0 : a2ui.includeSchema
-						}),
-						children,
-						shouldRenderInspector ? /* @__PURE__ */ (0, react_jsx_runtime.jsx)(CopilotKitInspector, {
-							core: copilotkit,
-							defaultAnchor: inspectorDefaultAnchor
-						}) : null,
-						runtimeLicenseStatus === "none" && !resolvedPublicKey && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "no_license" }),
-						runtimeLicenseStatus === "expired" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "expired" }),
-						runtimeLicenseStatus === "invalid" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "invalid" }),
-						runtimeLicenseStatus === "expiring" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "expiring" })
-					]
-				})
-			})
-		});
-	};
-	const useCopilotKit = () => {
-		const context = (0, react.useContext)(CopilotKitContext);
-		const [, forceUpdate] = (0, react.useReducer)((x) => x + 1, 0);
-		if (!context) throw new Error("useCopilotKit must be used within CopilotKitProvider");
-		(0, react.useEffect)(() => {
-			const subscription = context.copilotkit.subscribe({ onRuntimeConnectionStatusChanged: () => {
-				forceUpdate();
-			} });
-			return () => {
-				subscription.unsubscribe();
-			};
-		}, []);
-		return context;
-	};
-
-//#endregion
-//#region src/v2/hooks/use-render-tool-call.tsx
-/**
-	* Memoized component that renders a single tool call.
-	* This prevents unnecessary re-renders when parent components update
-	* but the tool call data hasn't changed.
-	*/
-	const ToolCallRenderer = react.default.memo(function ToolCallRenderer({ toolCall, toolMessage, RenderComponent, isExecuting }) {
-		const args = (0, react.useMemo)(() => (0, _copilotkit_shared.partialJSONParse)(toolCall.function.arguments), [toolCall.function.arguments]);
-		const toolName = toolCall.function.name;
-		if (toolMessage) return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RenderComponent, {
-			name: toolName,
-			toolCallId: toolCall.id,
-			args,
-			status: _copilotkit_core.ToolCallStatus.Complete,
-			result: toolMessage.content
-		});
-		else if (isExecuting) return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RenderComponent, {
-			name: toolName,
-			toolCallId: toolCall.id,
-			args,
-			status: _copilotkit_core.ToolCallStatus.Executing,
-			result: void 0
-		});
-		else return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(RenderComponent, {
-			name: toolName,
-			toolCallId: toolCall.id,
-			args,
-			status: _copilotkit_core.ToolCallStatus.InProgress,
-			result: void 0
-		});
-	}, (prevProps, nextProps) => {
-		var _prevProps$toolMessag, _nextProps$toolMessag;
-		if (prevProps.toolCall.id !== nextProps.toolCall.id) return false;
-		if (prevProps.toolCall.function.name !== nextProps.toolCall.function.name) return false;
-		if (prevProps.toolCall.function.arguments !== nextProps.toolCall.function.arguments) return false;
-		if (((_prevProps$toolMessag = prevProps.toolMessage) === null || _prevProps$toolMessag === void 0 ? void 0 : _prevProps$toolMessag.content) !== ((_nextProps$toolMessag = nextProps.toolMessage) === null || _nextProps$toolMessag === void 0 ? void 0 : _nextProps$toolMessag.content)) return false;
-		if (prevProps.isExecuting !== nextProps.isExecuting) return false;
-		if (prevProps.RenderComponent !== nextProps.RenderComponent) return false;
-		return true;
-	});
-	/**
-	* Hook that returns a function to render tool calls based on the render functions
-	* defined in CopilotKitProvider.
-	*
-	* @returns A function that takes a tool call and optional tool message and returns the rendered component
-	*/
-	function useRenderToolCall$1() {
-		var _config$agentId;
-		const { copilotkit, executingToolCallIds } = useCopilotKit();
-		const config = useCopilotChatConfiguration();
-		const agentId = (_config$agentId = config === null || config === void 0 ? void 0 : config.agentId) !== null && _config$agentId !== void 0 ? _config$agentId : _copilotkit_shared.DEFAULT_AGENT_ID;
-		const renderToolCalls = (0, react.useSyncExternalStore)((callback) => {
-			return copilotkit.subscribe({ onRenderToolCallsChanged: callback }).unsubscribe;
-		}, () => copilotkit.renderToolCalls, () => copilotkit.renderToolCalls);
-		return (0, react.useCallback)(({ toolCall, toolMessage }) => {
-			const exactMatches = renderToolCalls.filter((rc) => rc.name === toolCall.function.name);
-			const renderConfig = exactMatches.find((rc) => rc.agentId === agentId) || exactMatches.find((rc) => !rc.agentId) || exactMatches[0] || renderToolCalls.find((rc) => rc.name === "*");
-			if (!renderConfig) return null;
-			const RenderComponent = renderConfig.render;
-			return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(ToolCallRenderer, {
-				toolCall,
-				toolMessage,
-				RenderComponent,
-				isExecuting: executingToolCallIds.has(toolCall.id)
-			}, toolCall.id);
-		}, [
-			renderToolCalls,
-			executingToolCallIds,
-			agentId
-		]);
-	}
-
-//#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, throttleMs } = {}) {
-		var _agentId, _threadId;
-		(_agentId = agentId) !== null && _agentId !== void 0 || (agentId = _copilotkit_shared.DEFAULT_AGENT_ID);
-		const { copilotkit } = useCopilotKit();
-		const providerThrottleMs = copilotkit.defaultThrottleMs;
-		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;
-			let active = true;
-			const handlers = {};
-			let batchScheduled = false;
-			const batchedForceUpdate = () => {
-				if (!active) return;
-				if (!batchScheduled) {
-					batchScheduled = true;
-					queueMicrotask(() => {
-						batchScheduled = false;
-						if (active) forceUpdate();
-					});
+				},
+				onToolExecutionEnd: ({ toolCallId }) => {
+					setExecutingToolCallIds((prev) => {
+						if (!prev.has(toolCallId)) return prev;
+						const next = new Set(prev);
+						next.delete(toolCallId);
+						return next;
+					});
 				}
+			});
+			return () => {
+				subscription.unsubscribe();
 			};
-			if (updateFlags.includes(UseAgentUpdate.OnMessagesChanged)) handlers.onMessagesChanged = forceUpdate;
-			if (updateFlags.includes(UseAgentUpdate.OnStateChanged)) handlers.onStateChanged = batchedForceUpdate;
-			if (updateFlags.includes(UseAgentUpdate.OnRunStatusChanged)) {
-				handlers.onRunInitialized = batchedForceUpdate;
-				handlers.onRunFinalized = batchedForceUpdate;
-				handlers.onRunFailed = batchedForceUpdate;
-				handlers.onRunErrorEvent = batchedForceUpdate;
-			}
-			const subscription = copilotkit.subscribeToAgentWithOptions(agent, handlers, { throttleMs });
+		}, [copilotkit]);
+		const onErrorRef = (0, react.useRef)(onError);
+		(0, react.useEffect)(() => {
+			onErrorRef.current = onError;
+		}, [onError]);
+		(0, react.useEffect)(() => {
+			const subscription = copilotkit.subscribe({ onError: (event) => {
+				if (onErrorRef.current) onErrorRef.current(event);
+				else {
+					var _event$context;
+					console.error(`[CopilotKit] Error (${event.code}):`, event.error, (_event$context = event.context) !== null && _event$context !== void 0 ? _event$context : {});
+				}
+			} });
 			return () => {
-				active = false;
 				subscription.unsubscribe();
 			};
+		}, [copilotkit]);
+		(0, react.useEffect)(() => {
+			copilotkit.setRuntimeUrl(chatApiEndpoint);
+			copilotkit.setRuntimeTransport(useSingleEndpoint === true ? "single" : useSingleEndpoint === false ? "rest" : "auto");
+			copilotkit.setHeaders(mergedHeaders);
+			copilotkit.setCredentials(credentials);
+			copilotkit.setProperties(properties);
+			copilotkit.setAgents__unsafe_dev_only(mergedAgents);
+			copilotkit.setDebug(debug);
 		}, [
-			agent,
-			forceUpdate,
-			throttleMs,
-			providerThrottleMs,
-			updateFlags
+			copilotkit,
+			chatApiEndpoint,
+			mergedHeaders,
+			credentials,
+			properties,
+			mergedAgents,
+			useSingleEndpoint,
+			debug
 		]);
+		const didMountRef = (0, react.useRef)(false);
 		(0, react.useEffect)(() => {
-			if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...copilotkit.headers };
-		}, [agent, JSON.stringify(copilotkit.headers)]);
-		return { agent };
-	}
+			if (!didMountRef.current) return;
+			copilotkit.setTools(allTools);
+		}, [copilotkit, allTools]);
+		(0, react.useEffect)(() => {
+			if (!didMountRef.current) return;
+			copilotkit.setRenderToolCalls(allRenderToolCalls);
+		}, [copilotkit, allRenderToolCalls]);
+		(0, react.useEffect)(() => {
+			if (!didMountRef.current) return;
+			copilotkit.setRenderActivityMessages(allActivityRenderers);
+		}, [copilotkit, allActivityRenderers]);
+		(0, react.useEffect)(() => {
+			if (!didMountRef.current) return;
+			copilotkit.setRenderCustomMessages(renderCustomMessagesList);
+		}, [copilotkit, renderCustomMessagesList]);
+		(0, react.useEffect)(() => {
+			didMountRef.current = true;
+		}, []);
+		(0, react.useEffect)(() => {
+			copilotkit.setDefaultThrottleMs(defaultThrottleMs);
+		}, [copilotkit, defaultThrottleMs]);
+		const designSkill = (_openGenerativeUI$des = openGenerativeUI === null || openGenerativeUI === void 0 ? void 0 : openGenerativeUI.designSkill) !== null && _openGenerativeUI$des !== void 0 ? _openGenerativeUI$des : DEFAULT_DESIGN_SKILL;
+		(0, react.useLayoutEffect)(() => {
+			if (!copilotkit || !openGenUIActive) return;
+			const id = copilotkit.addContext({
+				description: "Design guidelines for the generateSandboxedUi tool. Follow these when building UI.",
+				value: designSkill
+			});
+			return () => {
+				copilotkit.removeContext(id);
+			};
+		}, [
+			copilotkit,
+			designSkill,
+			openGenUIActive
+		]);
+		const sandboxFunctionsDescriptors = (0, react.useMemo)(() => {
+			if (sandboxFunctionsList.length === 0) return null;
+			return JSON.stringify(sandboxFunctionsList.map((fn) => ({
+				name: fn.name,
+				description: fn.description,
+				parameters: (0, _copilotkit_shared.schemaToJsonSchema)(fn.parameters, { zodToJsonSchema: zod_to_json_schema.zodToJsonSchema })
+			})));
+		}, [sandboxFunctionsList]);
+		(0, react.useLayoutEffect)(() => {
+			if (!copilotkit || !sandboxFunctionsDescriptors || !openGenUIActive) return;
+			const id = copilotkit.addContext({
+				description: "Sandbox functions available in generated sandboxed UI code. Call via: await Websandbox.connection.remote.<functionName>(args)",
+				value: sandboxFunctionsDescriptors
+			});
+			return () => {
+				copilotkit.removeContext(id);
+			};
+		}, [
+			copilotkit,
+			sandboxFunctionsDescriptors,
+			openGenUIActive
+		]);
+		const contextValue = (0, react.useMemo)(() => ({
+			copilotkit,
+			executingToolCallIds
+		}), [copilotkit, executingToolCallIds]);
+		const licenseContextValue = (0, react.useMemo)(() => (0, _copilotkit_shared.createLicenseContextValue)(null), []);
+		return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(SandboxFunctionsContext.Provider, {
+			value: sandboxFunctionsList,
+			children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)(CopilotKitContext.Provider, {
+				value: contextValue,
+				children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(LicenseContext.Provider, {
+					value: licenseContextValue,
+					children: [
+						runtimeA2UIEnabled && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(A2UIBuiltInToolCallRenderer, {}),
+						runtimeA2UIEnabled && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(A2UICatalogContext, {
+							catalog: a2ui === null || a2ui === void 0 ? void 0 : a2ui.catalog,
+							includeSchema: a2ui === null || a2ui === void 0 ? void 0 : a2ui.includeSchema
+						}),
+						children,
+						shouldRenderInspector ? /* @__PURE__ */ (0, react_jsx_runtime.jsx)(CopilotKitInspector, {
+							core: copilotkit,
+							defaultAnchor: inspectorDefaultAnchor
+						}) : null,
+						runtimeLicenseStatus === "none" && !resolvedPublicKey && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "no_license" }),
+						runtimeLicenseStatus === "expired" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "expired" }),
+						runtimeLicenseStatus === "invalid" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "invalid" }),
+						runtimeLicenseStatus === "expiring" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(LicenseWarningBanner, { type: "expiring" })
+					]
+				})
+			})
+		});
+	};
 
 //#endregion
 //#region src/v2/hooks/use-render-custom-messages.tsx
@@ -2574,13 +2412,12 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 			return aHasAgent ? -1 : 1;
 		});
 		return function(params) {
-			var _copilotkit$getRunIdF, _getThreadClone;
+			var _copilotkit$getRunIdF;
 			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 registryAgent = copilotkit.getAgent(agentId);
-			const agent = (_getThreadClone = getThreadClone(registryAgent, threadId)) !== null && _getThreadClone !== void 0 ? _getThreadClone : registryAgent;
+			const agent = copilotkit.getAgent(agentId);
 			if (!agent) return null;
 			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);
@@ -2637,8 +2474,7 @@ window.parent.postMessage({jsonrpc:"2.0",method:"ui/notifications/sandbox-proxy-
 			tool.name,
 			tool.available,
 			copilotkit,
-			extraDeps.length,
-			...extraDeps
+			JSON.stringify(extraDeps)
 		]);
 	}
 
@@ -2708,6 +2544,120 @@ 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, throttleMs } = {}) {
+		var _agentId;
+		(_agentId = agentId) !== null && _agentId !== void 0 || (agentId = _copilotkit_shared.DEFAULT_AGENT_ID);
+		const { copilotkit } = useCopilotKit();
+		const providerThrottleMs = copilotkit.defaultThrottleMs;
+		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 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;
+			}
+			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;
+			let active = true;
+			const handlers = {};
+			let batchScheduled = false;
+			const batchedForceUpdate = () => {
+				if (!active) return;
+				if (!batchScheduled) {
+					batchScheduled = true;
+					queueMicrotask(() => {
+						batchScheduled = false;
+						if (active) forceUpdate();
+					});
+				}
+			};
+			if (updateFlags.includes(UseAgentUpdate.OnMessagesChanged)) handlers.onMessagesChanged = batchedForceUpdate;
+			if (updateFlags.includes(UseAgentUpdate.OnStateChanged)) handlers.onStateChanged = batchedForceUpdate;
+			if (updateFlags.includes(UseAgentUpdate.OnRunStatusChanged)) {
+				handlers.onRunInitialized = batchedForceUpdate;
+				handlers.onRunFinalized = batchedForceUpdate;
+				handlers.onRunFailed = batchedForceUpdate;
+				handlers.onRunErrorEvent = batchedForceUpdate;
+			}
+			const subscription = copilotkit.subscribeToAgentWithOptions(agent, handlers, { throttleMs });
+			return () => {
+				active = false;
+				subscription.unsubscribe();
+			};
+		}, [
+			agent,
+			forceUpdate,
+			throttleMs,
+			providerThrottleMs,
+			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-suggestions.tsx
 	function useSuggestions({ agentId } = {}) {
@@ -5299,10 +5249,7 @@ 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,
-			threadId: existingConfig === null || existingConfig === void 0 ? void 0 : existingConfig.threadId
-		});
+		const { agent } = useAgent({ agentId: resolvedAgentId });
 		const lastConnectedAgentRef = (0, react.useRef)(null);
 		(0, react.useEffect)(() => {
 			let detached = false;