@copilotkit/react-core 1.64.1 → 1.64.2-canary.1785346263

package/dist/v2/headless.cjs

@@ -293,7 +293,10 @@ const ALL_UPDATES = [
 	UseAgentUpdate.OnStateChanged,
 	UseAgentUpdate.OnRunStatusChanged
 ];
-function useAgent({ agentId, updates, throttleMs } = {}) {
+function useAgent({ agentId, threadId, runtimeAgentId, updates, throttleMs } = {}) {
+	if (threadId != null && runtimeAgentId == null) throw new Error(`useAgent: \`threadId\` requires \`runtimeAgentId\`. A threadId is written onto a single agent, but an agent resolved by agentId alone is shared, so scoping a thread to it would clobber other useAgent callers. Pass a distinct local \`agentId\` and the runtime agent to route to, e.g. useAgent({ agentId: "chat-1", runtimeAgentId: "${agentId ?? "default"}", threadId }).`);
+	if (runtimeAgentId != null && threadId == null) throw new Error(`useAgent: \`runtimeAgentId\` requires \`threadId\`. A proxied agent exists to scope a thread to a private instance; without a threadId it behaves like the shared agent while adding a registration and a local agentId to keep unique. Either pass the thread, e.g. useAgent({ agentId: "${agentId ?? "chat-1"}", runtimeAgentId: "${runtimeAgentId}", threadId }), or bind to the agent directly with useAgent({ agentId: "${runtimeAgentId}" }).`);
+	if (runtimeAgentId != null && agentId == null) throw new Error(`useAgent: \`runtimeAgentId\` requires an explicit \`agentId\`. The proxied agent is registered under \`agentId\`, and the usual fallbacks (chat configuration, then "${_copilotkit_shared.DEFAULT_AGENT_ID}") name agents that already exist — registering over one throws or shadows it. Pick a local id for this hook, e.g. useAgent({ agentId: "chat-1", runtimeAgentId: "${runtimeAgentId}", threadId }).`);
 	const chatConfig = useCopilotChatConfiguration();
 	const resolvedAgentId = agentId ?? chatConfig?.agentId ?? _copilotkit_shared.DEFAULT_AGENT_ID;
 	const { copilotkit } = (0, _copilotkit_react_core_v2_context.useCopilotKit)();
@@ -301,7 +304,58 @@ function useAgent({ agentId, updates, throttleMs } = {}) {
 	const [, forceUpdate] = (0, react.useReducer)((x) => x + 1, 0);
 	const updateFlags = (0, react.useMemo)(() => updates ?? ALL_UPDATES, [JSON.stringify(updates)]);
 	const provisionalAgentCache = (0, react.useRef)(/* @__PURE__ */ new Map());
+	const [registeredProxyAgent, setRegisteredProxyAgent] = (0, react.useState)(null);
+	(0, react.useEffect)(() => {
+		if (runtimeAgentId == null) {
+			setRegisteredProxyAgent(null);
+			return;
+		}
+		const { agent: proxy, unregister } = copilotkit.registerProxiedAgent({
+			agentId: resolvedAgentId,
+			runtimeAgentId
+		});
+		provisionalAgentCache.current.delete(resolvedAgentId);
+		setRegisteredProxyAgent(proxy);
+		return () => {
+			unregister();
+			setRegisteredProxyAgent(null);
+		};
+	}, [
+		copilotkit,
+		resolvedAgentId,
+		runtimeAgentId
+	]);
 	const { agent, isReady } = (0, react.useMemo)(() => {
+		if (runtimeAgentId != null) {
+			if (registeredProxyAgent) {
+				provisionalAgentCache.current.delete(resolvedAgentId);
+				return {
+					agent: registeredProxyAgent,
+					isReady: true
+				};
+			}
+			const cached = provisionalAgentCache.current.get(resolvedAgentId);
+			if (cached) {
+				copilotkit.applyHeadersToAgent(cached);
+				return {
+					agent: cached,
+					isReady: false
+				};
+			}
+			const provisional = new _copilotkit_core.ProxiedCopilotRuntimeAgent({
+				runtimeUrl: copilotkit.runtimeUrl,
+				agentId: resolvedAgentId,
+				runtimeAgentId,
+				transport: copilotkit.runtimeTransport,
+				runtimeMode: "pending"
+			});
+			copilotkit.applyHeadersToAgent(provisional);
+			provisionalAgentCache.current.set(resolvedAgentId, provisional);
+			return {
+				agent: provisional,
+				isReady: false
+			};
+		}
 		const existing = copilotkit.getAgent(resolvedAgentId);
 		if (existing) {
 			provisionalAgentCache.current.delete(resolvedAgentId);
@@ -357,6 +411,8 @@ function useAgent({ agentId, updates, throttleMs } = {}) {
 		throw new Error(`useAgent: Agent '${resolvedAgentId}' 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.");
 	}, [
 		resolvedAgentId,
+		runtimeAgentId,
+		registeredProxyAgent,
 		copilotkit.agents,
 		copilotkit.runtimeConnectionStatus,
 		copilotkit.runtimeUrl,
@@ -409,14 +465,11 @@ function useAgent({ agentId, updates, throttleMs } = {}) {
 	]);
 	const configThreadId = chatConfig?.threadId;
 	const configHasExplicitThreadId = chatConfig?.hasExplicitThreadId;
+	const resolvedThreadId = threadId ?? (configHasExplicitThreadId ? configThreadId : void 0);
 	(0, react.useEffect)(() => {
-		if (!configHasExplicitThreadId || !configThreadId) return;
-		agent.threadId = configThreadId;
-	}, [
-		agent,
-		configThreadId,
-		configHasExplicitThreadId
-	]);
+		if (!resolvedThreadId) return;
+		agent.threadId = resolvedThreadId;
+	}, [agent, resolvedThreadId]);
 	return {
 		agent,
 		isReady