@copilotkit/core 1.61.1 → 1.61.2

package/dist/index.cjs

@@ -152,12 +152,6 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 	activeChannel = null;
 	canonicalRunId = null;
 	sharedState;
-	/**
-	* Resolves when the active run's pipeline finalizes; `undefined` between runs.
-	* Set/cleared inside {@link connectAgent}'s observable lifecycle. Declared
-	* here so {@link RunCompletionAware} consumers can read it without a cast.
-	*/
-	activeRunCompletionPromise;
 	constructor(config, sharedState = { lastSeenEventIds: /* @__PURE__ */ new Map() }) {
 		super();
 		this.config = config;
@@ -203,7 +197,7 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 			const previousMessageIds = new Set(this.messages.map((m) => m.id));
 			const subscribers = [
 				{ onRunFinishedEvent: (event) => {
-					result = event.result;
+					if (event.outcome === "success") result = event.result;
 				} },
 				...this.subscribers,
 				subscriber ?? {}
@@ -211,10 +205,10 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 			await this.onInitialize(input, subscribers);
 			self.activeRunDetach$ = new rxjs.Subject();
 			let resolveCompletion;
-			this.activeRunCompletionPromise = new Promise((resolve) => {
+			self.activeRunCompletionPromise = new Promise((resolve) => {
 				resolveCompletion = resolve;
 			});
-			const source$ = (0, rxjs.defer)(() => this.connect(input)).pipe((0, _ag_ui_client.transformChunks)(this.debug), (0, rxjs_operators.takeUntil)(self.activeRunDetach$));
+			const source$ = (0, rxjs.defer)(() => this.connect(input)).pipe((0, _ag_ui_client.transformChunks)(this.debugLogger), (0, rxjs_operators.takeUntil)(self.activeRunDetach$));
 			const applied$ = this.apply(input, source$, subscribers);
 			await (0, rxjs.lastValueFrom)(this.processApplyEvents(input, applied$, subscribers).pipe((0, rxjs_operators.catchError)((error) => {
 				this.isRunning = false;
@@ -224,7 +218,7 @@ var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAg
 				this.onFinalize(input, subscribers);
 				resolveCompletion?.();
 				resolveCompletion = void 0;
-				this.activeRunCompletionPromise = void 0;
+				self.activeRunCompletionPromise = void 0;
 				self.activeRunDetach$ = void 0;
 			})), { defaultValue: void 0 });
 			const newMessages = (0, _ag_ui_client.structuredClone_)(this.messages).filter((m) => !previousMessageIds.has(m.id));
@@ -726,8 +720,9 @@ var ProxiedCopilotRuntimeAgent = class ProxiedCopilotRuntimeAgent extends _ag_ui
 		cloned.setState(this.state);
 		cloned.setMessages(this.messages);
 		if (this.delegate) {
-			cloned.delegate = this.delegate.clone();
-			cloned.syncDelegate(cloned.delegate);
+			const clonedDelegate = this.delegate.clone();
+			cloned.delegate = clonedDelegate;
+			cloned.syncDelegate(clonedDelegate);
 		}
 		return cloned;
 	}
@@ -882,6 +877,17 @@ var AgentRegistry = class {
 	_openGenerativeUIEnabled = false;
 	_licenseStatus;
 	_telemetryDisabled = false;
+	/**
+	* The headers each HttpAgent was constructed with, captured on the first
+	* `applyHeadersToAgent` call for that agent (which, for agents the registry
+	* owns, happens at registration before any core headers are applied). Core
+	* headers are merged ON TOP of this baseline so that headers configured
+	* directly on an agent (e.g. an `Authorization` for a self-hosted backend)
+	* survive registration instead of being silently replaced. The baseline is
+	* captured once and never re-captured, so a later direct mutation of
+	* `agent.headers` is not folded into it. See #5635.
+	*/
+	agentOwnHeaders = /* @__PURE__ */ new WeakMap();
 	constructor(core) {
 		this.core = core;
 	}
@@ -1054,10 +1060,19 @@ var AgentRegistry = class {
 		console.warn(`Agent ${id} not found`);
 	}
 	/**
-	* Apply current headers to an agent
+	* Apply current core headers to an agent, merged ON TOP of the agent's own
+	* construction-time headers (the per-agent baseline in `agentOwnHeaders`).
+	* Core wins on a key conflict. Non-`HttpAgent` agents are left untouched
+	* because only `HttpAgent` carries a `headers` field. See #5635.
 	*/
 	applyHeadersToAgent(agent) {
-		if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...this.core.headers };
+		if (agent instanceof _ag_ui_client.HttpAgent) {
+			if (!this.agentOwnHeaders.has(agent)) this.agentOwnHeaders.set(agent, { ...agent.headers });
+			agent.headers = {
+				...this.agentOwnHeaders.get(agent),
+				...this.core.headers
+			};
+		}
 	}
 	/**
 	* Apply current headers to all agents
@@ -1729,7 +1744,7 @@ var RunHandler = class {
 				const proxied = agent;
 				if (incomingThreadId && typeof proxied.clearReplayCursor === "function") proxied.clearReplayCursor(incomingThreadId);
 			}
-			if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...this._internal.headers };
+			this._internal.applyHeadersToAgent(agent);
 			const runAgentResult = await agent.connectAgent({
 				forwardedProps: this._internal.properties,
 				tools: this.buildFrontendTools(agent.agentId),
@@ -1750,15 +1765,18 @@ var RunHandler = class {
 					context
 				});
 			}
-			return { newMessages: [] };
+			return {
+				result: void 0,
+				newMessages: []
+			};
 		}
 	}
 	/**
 	* Run an agent
 	*/
-	async runAgent({ agent, forwardedProps }) {
+	async runAgent({ agent, forwardedProps, resume }) {
 		if (agent.agentId) this._internal.suggestionEngine.clearSuggestions(agent.agentId);
-		if (agent instanceof _ag_ui_client.HttpAgent) agent.headers = { ...this._internal.headers };
+		this._internal.applyHeadersToAgent(agent);
 		if (agent.detachActiveRun) await agent.detachActiveRun();
 		const isTopLevel = this._runDepth === 0;
 		let originalAbortRun;
@@ -1778,6 +1796,7 @@ var RunHandler = class {
 					...this._internal.properties,
 					...forwardedProps
 				},
+				...resume !== void 0 ? { resume } : {},
 				tools: this.buildFrontendTools(agent.agentId),
 				context: this._internal.getContextForAgent(agent.agentId)
 			}, this.createAgentErrorSubscriber(agent));
@@ -1794,7 +1813,10 @@ var RunHandler = class {
 				code: CopilotKitCoreErrorCode.AGENT_RUN_FAILED,
 				context
 			});
-			return { newMessages: [] };
+			return {
+				result: void 0,
+				newMessages: []
+			};
 		} finally {
 			this._runDepth--;
 			if (isTopLevel && originalAbortRun) agent.abortRun = originalAbortRun;
@@ -2066,18 +2088,19 @@ var RunHandler = class {
 			throw error;
 		}
 		const toolCallId = (0, _copilotkit_shared.randomUUID)();
+		const assistantToolCall = {
+			id: toolCallId,
+			type: "function",
+			function: {
+				name,
+				arguments: JSON.stringify(parameters)
+			}
+		};
 		const assistantMessage = {
 			id: (0, _copilotkit_shared.randomUUID)(),
 			role: "assistant",
 			content: "",
-			toolCalls: [{
-				id: toolCallId,
-				type: "function",
-				function: {
-					name,
-					arguments: JSON.stringify(parameters)
-				}
-			}]
+			toolCalls: [assistantToolCall]
 		};
 		agent.messages.push(assistantMessage);
 		let handlerResult = {
@@ -2087,7 +2110,7 @@ var RunHandler = class {
 		};
 		if (tool.handler) handlerResult = await this.executeToolHandler({
 			tool,
-			toolCall: assistantMessage.toolCalls[0],
+			toolCall: assistantToolCall,
 			agent,
 			agentId: resolvedAgentId,
 			handlerArgs: parameters,
@@ -2173,9 +2196,9 @@ const EMPTY_TOOL_SCHEMA = {
 */
 function createToolSchema(tool) {
 	if (!tool.parameters) return { ...EMPTY_TOOL_SCHEMA };
-	const rawSchema = (0, _copilotkit_shared.schemaToJsonSchema)(tool.parameters, { zodToJsonSchema: zod_to_json_schema.zodToJsonSchema });
+	const rawSchema = (0, _copilotkit_shared.schemaToJsonSchema)(tool.parameters, { zodToJsonSchema: (schema, options) => (0, zod_to_json_schema.zodToJsonSchema)(schema, options) });
 	if (!rawSchema || typeof rawSchema !== "object") return { ...EMPTY_TOOL_SCHEMA };
-	const { $schema, ...schema } = rawSchema;
+	const { $schema: _$schema, ...schema } = rawSchema;
 	if (typeof schema.type !== "string") schema.type = "object";
 	if (typeof schema.properties !== "object" || schema.properties === null) schema.properties = {};
 	stripAdditionalProperties(schema);
@@ -2762,9 +2785,19 @@ var CopilotKitCore = class {
 	* });
 	* ```
 	*
-	* The resulting header set is also re-applied to every registered
-	* `HttpAgent`-derived agent (replacing its `headers`) and `onHeadersChanged`
-	* subscribers are notified.
+	* The resulting header set is also re-applied to every agent in the registry
+	* and `onHeadersChanged` subscribers are notified. These headers are merged
+	* ON TOP of the headers each `HttpAgent` was constructed with, so per-agent
+	* headers (e.g. an `Authorization` for a self-hosted backend) are preserved;
+	* on a key conflict the core-level value wins.
+	*
+	* Because the agent's construction-time headers form the merge baseline, this
+	* method can override a per-agent header but cannot REMOVE one: clearing a
+	* core key here only drops the core-level override, after which the agent's
+	* own value (if any) re-surfaces. To change a header an agent was constructed
+	* with, set it at the provider/core level instead of on the agent, or update
+	* it on the agent directly. The clear-on-logout pattern above is for
+	* core-level headers.
 	*/
 	setHeaders(headers) {
 		this._headers = normalizeHeaders(headers);
@@ -2823,6 +2856,23 @@ var CopilotKitCore = class {
 		return this.agentRegistry.getAgent(id);
 	}
 	/**
+	* Re-apply the current headers to a single agent (delegated to
+	* AgentRegistry). Core headers are merged on top of the agent's own
+	* construction-time headers rather than replacing them, so headers
+	* configured directly on an `HttpAgent` (e.g. an `Authorization` for a
+	* self-hosted backend) survive header updates instead of being silently
+	* dropped (see #5635). On a key conflict the core-level value wins.
+	*
+	* The merge baseline is the agent's headers as captured the first time the
+	* agent is applied (at registration), so the way to change headers
+	* afterwards is `setHeaders` (which re-applies to every agent), not mutating
+	* `agent.headers` directly — a direct mutation is overwritten on the next
+	* re-apply.
+	*/
+	applyHeadersToAgent(agent) {
+		this.agentRegistry.applyHeadersToAgent(agent);
+	}
+	/**
 	* Context management (delegated to ContextStore).
 	* Pass `agentIds` to restrict the entry to runs of specific agents;
 	* omit it for context every agent should receive.
@@ -3548,6 +3598,7 @@ const threadAdapterEvents = createActionGroup("Thread Adapter", {
 	fetchNextPageRequested: empty(),
 	renameRequested: props(),
 	archiveRequested: props(),
+	unarchiveRequested: props(),
 	deleteRequested: props()
 });
 const threadRestEvents = createActionGroup("Thread REST", {
@@ -3690,13 +3741,16 @@ function createThreadRequestId() {
 	threadRequestId += 1;
 	return `thread-request-${threadRequestId}`;
 }
+function threadFromFetch(input, init) {
+	return (0, rxjs_fetch.fromFetch)(input, init);
+}
 function createThreadFetchObservable(environment, context, sessionId) {
 	return (0, rxjs.defer)(() => {
 		const params = { agentId: context.agentId };
 		if (context.includeArchived) params.includeArchived = "true";
 		if (context.limit != null) params.limit = String(context.limit);
 		const qs = new URLSearchParams(params);
-		return (0, rxjs_fetch.fromFetch)(`${context.runtimeUrl}/threads?${qs.toString()}`, {
+		return threadFromFetch(`${context.runtimeUrl}/threads?${qs.toString()}`, {
 			selector: (response) => {
 				if (!response.ok) throw new Error(`Failed to fetch threads: ${response.status}`);
 				return response.json();
@@ -3724,7 +3778,7 @@ function createThreadFetchObservable(environment, context, sessionId) {
 }
 function createThreadMetadataCredentialsObservable(environment, context, sessionId) {
 	return (0, rxjs.defer)(() => {
-		return (0, rxjs_fetch.fromFetch)(`${context.runtimeUrl}${THREAD_SUBSCRIBE_PATH}`, {
+		return threadFromFetch(`${context.runtimeUrl}${THREAD_SUBSCRIBE_PATH}`, {
 			selector: async (response) => {
 				if (!response.ok) throw new Error(`Failed to fetch thread metadata credentials: ${response.status}`);
 				return response.json();
@@ -3757,7 +3811,7 @@ function createThreadMetadataCredentialsObservable(environment, context, session
 }
 function createThreadMutationObservable(environment, context, request) {
 	return (0, rxjs.defer)(() => {
-		return (0, rxjs_fetch.fromFetch)(`${context.runtimeUrl}${request.path}`, {
+		return threadFromFetch(`${context.runtimeUrl}${request.path}`, {
 			selector: async (response) => {
 				if (!response.ok) throw new Error(`Request failed: ${response.status}`);
 				return null;
@@ -3856,7 +3910,7 @@ function createThreadStore(environment) {
 				};
 				if (context.includeArchived) params.includeArchived = "true";
 				if (context.limit != null) params.limit = String(context.limit);
-				return (0, rxjs_fetch.fromFetch)(`${context.runtimeUrl}/threads?${new URLSearchParams(params).toString()}`, {
+				return threadFromFetch(`${context.runtimeUrl}/threads?${new URLSearchParams(params).toString()}`, {
 					selector: (response) => {
 						if (!response.ok) throw new Error(`Failed to fetch next page: ${response.status}`);
 						return response.json();
@@ -3878,7 +3932,7 @@ function createThreadStore(environment) {
 					error: error instanceof Error ? error : new Error(String(error))
 				}))), (0, rxjs_operators.takeUntil)(actions$.pipe(ofType(threadAdapterEvents.contextChanged, threadAdapterEvents.stopped))));
 			}))),
-			createEffect((actions$, state$) => actions$.pipe(ofType(threadAdapterEvents.renameRequested, threadAdapterEvents.archiveRequested, threadAdapterEvents.deleteRequested), (0, rxjs_operators.withLatestFrom)(state$), (0, rxjs_operators.mergeMap)(([action, state]) => {
+			createEffect((actions$, state$) => actions$.pipe(ofType(threadAdapterEvents.renameRequested, threadAdapterEvents.archiveRequested, threadAdapterEvents.unarchiveRequested, threadAdapterEvents.deleteRequested), (0, rxjs_operators.withLatestFrom)(state$), (0, rxjs_operators.mergeMap)(([action, state]) => {
 				const context = state.context;
 				if (!context?.runtimeUrl) {
 					const requestId = action.requestId;
@@ -3904,6 +3958,15 @@ function createThreadStore(environment) {
 					path: `/threads/${encodeURIComponent(action.threadId)}/archive`,
 					body: commonBody
 				});
+				if (threadAdapterEvents.unarchiveRequested.match(action)) return createThreadMutationObservable(environment, context, {
+					requestId: action.requestId,
+					method: "PATCH",
+					path: `/threads/${encodeURIComponent(action.threadId)}`,
+					body: {
+						...commonBody,
+						archived: false
+					}
+				});
 				return createThreadMutationObservable(environment, context, {
 					requestId: action.requestId,
 					method: "DELETE",
@@ -3958,6 +4021,12 @@ function createThreadStore(environment) {
 				threadId
 			}));
 		},
+		unarchiveThread(threadId) {
+			return trackMutation(threadAdapterEvents.unarchiveRequested({
+				requestId: createThreadRequestId(),
+				threadId
+			}));
+		},
 		deleteThread(threadId) {
 			return trackMutation(threadAdapterEvents.deleteRequested({
 				requestId: createThreadRequestId(),