@copilotkit/core 1.55.3-canary.1776243725 → 1.55.3-canary.1776979102

package/dist/index.cjs

@@ -516,7 +516,6 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 	singleEndpointUrl;
 	runtimeMode;
 	intelligence;
-	_capabilities;
 	delegate;
 	runtimeInfoPromise;
 	constructor(config) {
@@ -533,15 +532,8 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 		this.transport = transport;
 		this.runtimeMode = config.runtimeMode ?? _copilotkit_shared.RUNTIME_MODE_SSE;
 		this.intelligence = config.intelligence;
-		this._capabilities = config.capabilities;
 		if (this.transport === "single") this.singleEndpointUrl = this.runtimeUrl;
 	}
-	get capabilities() {
-		return this._capabilities;
-	}
-	async getCapabilities() {
-		return this._capabilities ?? {};
-	}
 	async detachActiveRun() {
 		if (this.delegate) await this.delegate.detachActiveRun();
 		await super.detachActiveRun();
@@ -669,8 +661,7 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 			credentials: this.credentials,
 			transport: this.transport,
 			runtimeMode: this.runtimeMode,
-			intelligence: this.intelligence,
-			capabilities: this._capabilities
+			intelligence: this.intelligence
 		});
 		cloned.threadId = this.threadId;
 		cloned.setState(this.state);
@@ -972,7 +963,7 @@ var AgentRegistry = class {
 			const runtimeInfoResponse = await this.fetchRuntimeInfo();
 			const { version, ...runtimeInfo } = runtimeInfoResponse;
 			const credentials = this.core.credentials;
-			this.remoteAgents = Object.fromEntries(Object.entries(runtimeInfo.agents).map(([id, { description, capabilities }]) => {
+			this.remoteAgents = Object.fromEntries(Object.entries(runtimeInfo.agents).map(([id, { description }]) => {
 				const agent = new ProxiedCopilotRuntimeAgent({
 					runtimeUrl: this.runtimeUrl,
 					agentId: id,
@@ -980,8 +971,7 @@ var AgentRegistry = class {
 					transport: this._runtimeTransport,
 					credentials,
 					runtimeMode: runtimeInfoResponse.mode ?? _copilotkit_shared.RUNTIME_MODE_SSE,
-					intelligence: runtimeInfoResponse.intelligence,
-					capabilities
+					intelligence: runtimeInfoResponse.intelligence
 				});
 				this.applyHeadersToAgent(agent);
 				return [id, agent];
@@ -1189,16 +1179,10 @@ var SuggestionEngine = class {
 	/**
 	* Reload suggestions for a specific agent
 	* This triggers generation of new suggestions based on current configs
-	*
-	* @param agentId - The consumer agent ID
-	* @param consumerAgent - Optional: the specific agent instance whose messages should be used
-	*   for availability filtering and context. When running with per-thread clones, the thread
-	*   clone holds the conversation messages; passing it here ensures dynamic suggestions fire
-	*   after the first message even though the registry agent has an empty message list.
 	*/
-	reloadSuggestions(agentId, consumerAgent) {
+	reloadSuggestions(agentId) {
 		this.clearSuggestions(agentId);
-		const agent = consumerAgent ?? this.core.getAgent(agentId);
+		const agent = this.core.getAgent(agentId);
 		if (!agent) return;
 		const messageCount = agent.messages?.length ?? 0;
 		let hasAnySuggestions = false;
@@ -1211,7 +1195,7 @@ var SuggestionEngine = class {
 					hasAnySuggestions = true;
 					this.notifySuggestionsStartedLoading(agentId);
 				}
-				this.generateSuggestions(suggestionId, config, agentId, agent);
+				this.generateSuggestions(suggestionId, config, agentId);
 			} else if (isStaticSuggestionsConfig(config)) this.addStaticSuggestions(suggestionId, config, agentId);
 		}
 	}
@@ -1239,12 +1223,12 @@ var SuggestionEngine = class {
 	/**
 	* Generate suggestions using a provider agent
 	*/
-	async generateSuggestions(suggestionId, config, consumerAgentId, consumerAgent) {
+	async generateSuggestions(suggestionId, config, consumerAgentId) {
 		let agent = void 0;
 		try {
 			const suggestionsProviderAgent = this.core.getAgent(config.providerAgentId ?? "default");
 			if (!suggestionsProviderAgent) throw new Error(`Suggestions provider agent not found: ${config.providerAgentId}`);
-			const suggestionsConsumerAgent = consumerAgent ?? this.core.getAgent(consumerAgentId);
+			const suggestionsConsumerAgent = this.core.getAgent(consumerAgentId);
 			if (!suggestionsConsumerAgent) throw new Error(`Suggestions consumer agent not found: ${consumerAgentId}`);
 			agent = suggestionsProviderAgent.clone();
 			agent.threadId = suggestionId;
@@ -1549,10 +1533,6 @@ var RunHandler = class {
 			agent.setMessages([]);
 			agent.setState({});
 			if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...this._internal.headers };
-			await this._internal.notifySubscribers((subscriber) => subscriber.onAgentRunStarted?.({
-				copilotkit: this.core,
-				agent
-			}), "Subscriber onAgentRunStarted error:");
 			const runAgentResult = await agent.connectAgent({
 				forwardedProps: this._internal.properties,
 				tools: this.buildFrontendTools(agent.agentId),
@@ -1583,11 +1563,6 @@ var RunHandler = class {
 		if (agent.agentId) this._internal.suggestionEngine.clearSuggestions(agent.agentId);
 		if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...this._internal.headers };
 		if (agent.detachActiveRun) await agent.detachActiveRun();
-		this._internal.subscribeAgentToStateManager(agent);
-		await this._internal.notifySubscribers((subscriber) => subscriber.onAgentRunStarted?.({
-			copilotkit: this.core,
-			agent
-		}), "Subscriber onAgentRunStarted error:");
 		const isTopLevel = this._runDepth === 0;
 		let originalAbortRun;
 		if (isTopLevel) {
@@ -1658,7 +1633,7 @@ var RunHandler = class {
 			await this._internal.waitForPendingFrameworkUpdates();
 			return await this.runAgent({ agent });
 		}
-		this._internal.suggestionEngine.reloadSuggestions(agentId, agent);
+		this._internal.suggestionEngine.reloadSuggestions(agentId);
 		return runAgentResult;
 	}
 	/**
@@ -2059,7 +2034,6 @@ function parseToolArguments(rawArgs, toolName) {
 var StateManager = class {
 	stateByRun = /* @__PURE__ */ new Map();
 	messageToRun = /* @__PURE__ */ new Map();
-	activeRun = /* @__PURE__ */ new Map();
 	agentSubscriptions = /* @__PURE__ */ new Map();
 	constructor(core) {
 		this.core = core;
@@ -2069,42 +2043,18 @@ var StateManager = class {
 	*/
 	initialize() {}
 	/**
-	* Subscribe to an agent's events to track state and messages.
-	*
-	* Registry agents (subscribed via `onAgentsChanged`) use the bare `agentId`
-	* key so that `unsubscribeFromAgent(agentId)` can remove them when they
-	* are replaced. Per-thread clones (subscribed via `subscribeAgentToStateManager`)
-	* pass `{ isClone: true }` to use a composite `agentId:threadId` key, keeping
-	* their subscription independent of the registry agent's.
+	* Subscribe to an agent's events to track state and messages
 	*/
-	subscribeToAgent(agent, { isClone = false } = {}) {
+	subscribeToAgent(agent) {
 		if (!agent.agentId) return;
 		const agentId = agent.agentId;
-		const subscriptionKey = isClone && agent.threadId ? `${agentId}:${agent.threadId}` : agentId;
-		const existingUnsubscribe = this.agentSubscriptions.get(subscriptionKey);
-		if (existingUnsubscribe) {
-			existingUnsubscribe();
-			this.agentSubscriptions.delete(subscriptionKey);
-		}
-		let revoked = false;
-		let subRunId;
-		let runFinished = false;
-		const effectiveInput = (input) => ({
-			...input,
-			runId: subRunId ?? input.runId
-		});
+		this.unsubscribeFromAgent(agentId);
 		const { unsubscribe } = agent.subscribe({
-			onRunStartedEvent: ({ input, state }) => {
-				if (revoked) return;
-				if (runFinished && input.runId === subRunId) subRunId = (0, _ag_ui_client.randomUUID)();
-				else subRunId = input.runId;
-				runFinished = false;
-				this.handleRunStarted(agent, effectiveInput(input), state);
+			onRunStartedEvent: ({ event, state }) => {
+				this.handleRunStarted(agent, event, state);
 			},
-			onRunFinishedEvent: ({ input, state }) => {
-				if (revoked) return;
-				runFinished = true;
-				this.handleRunFinished(agent, effectiveInput(input), state);
+			onRunFinishedEvent: ({ event, state }) => {
+				this.handleRunFinished(agent, event, state);
 			},
 			onRunErrorEvent: ({ input, state }) => {
 				if (revoked) return;
@@ -2112,32 +2062,22 @@ var StateManager = class {
 				this.handleRunFinished(agent, effectiveInput(input), state);
 			},
 			onStateSnapshotEvent: ({ event, input, state }) => {
-				if (revoked) return;
-				this.handleStateSnapshot(agent, event, effectiveInput(input), state);
+				this.handleStateSnapshot(agent, event, input, state);
 			},
 			onStateDeltaEvent: ({ event, input, state }) => {
-				if (revoked) return;
-				this.handleStateDelta(agent, event, effectiveInput(input), state);
+				this.handleStateDelta(agent, event, input, state);
 			},
 			onMessagesSnapshotEvent: ({ event, input, messages }) => {
-				if (revoked) return;
-				this.handleMessagesSnapshot(agent, event, effectiveInput(input), messages);
+				this.handleMessagesSnapshot(agent, event, input, messages);
 			},
 			onNewMessage: ({ message, input }) => {
-				if (revoked) return;
-				this.handleNewMessage(agent, message, input ? effectiveInput(input) : void 0);
+				this.handleNewMessage(agent, message, input);
 			}
 		});
-		this.agentSubscriptions.set(subscriptionKey, () => {
-			revoked = true;
-			unsubscribe();
-		});
+		this.agentSubscriptions.set(agentId, unsubscribe);
 	}
 	/**
-	* Unsubscribe a registry agent's subscription (bare `agentId` key).
-	* Per-thread clone subscriptions use composite `agentId:threadId` keys and
-	* are replaced (not removed) by subsequent subscribeToAgent(agent, { isClone: true })
-	* calls for the same (agentId, threadId) pair.
+	* Unsubscribe from an agent's events
 	*/
 	unsubscribeFromAgent(agentId) {
 		const unsubscribe = this.agentSubscriptions.get(agentId);
@@ -2177,20 +2117,18 @@ var StateManager = class {
 	/**
 	* Handle run started event
 	*/
-	handleRunStarted(agent, input, state) {
+	handleRunStarted(agent, event, state) {
 		if (!agent.agentId) return;
-		const { threadId, runId } = input;
-		this.activeRun.set(`${agent.agentId}:${threadId}`, runId);
-		if (state && Object.keys(state).length > 0) this.saveState(agent.agentId, threadId, runId, state);
+		const { threadId, runId } = event;
+		this.saveState(agent.agentId, threadId, runId, state);
 	}
 	/**
 	* Handle run finished event
 	*/
-	handleRunFinished(agent, input, state) {
+	handleRunFinished(agent, event, state) {
 		if (!agent.agentId) return;
-		const { threadId, runId } = input;
-		this.activeRun.delete(`${agent.agentId}:${threadId}`);
-		if (state && Object.keys(state).length > 0) this.saveState(agent.agentId, threadId, runId, state);
+		const { threadId, runId } = event;
+		this.saveState(agent.agentId, threadId, runId, state);
 	}
 	/**
 	* Handle state snapshot event
@@ -2224,13 +2162,7 @@ var StateManager = class {
 	* Handle new message event
 	*/
 	handleNewMessage(agent, message, input) {
-		if (!agent.agentId) return;
-		if (!input) {
-			const threadId = agent.threadId ?? "";
-			const runId = this.activeRun.get(`${agent.agentId}:${threadId}`);
-			if (runId) this.associateMessageWithRun(agent.agentId, threadId, message.id, runId);
-			return;
-		}
+		if (!agent.agentId || !input) return;
 		const { threadId, runId } = input;
 		this.associateMessageWithRun(agent.agentId, threadId, message.id, runId);
 	}
@@ -2560,9 +2492,6 @@ var CopilotKitCore = class {
 	getRunIdsForThread(agentId, threadId) {
 		return this.stateManager.getRunIdsForThread(agentId, threadId);
 	}
-	subscribeAgentToStateManager(agent) {
-		this.stateManager.subscribeToAgent(agent, { isClone: true });
-	}
 	/**
 	* Internal method used by RunHandler to build frontend tools
 	*/