@copilotkit/core 1.56.5 → 1.57.0

package/dist/index.cjs

@@ -2265,6 +2265,61 @@ var StateManager = class {
 	}
 };
 
+//#endregion
+//#region src/core/thread-store-registry.ts
+var ThreadStoreRegistry = class {
+	_stores = {};
+	_snapshot = null;
+	constructor(core) {
+		this.core = core;
+	}
+	register(agentId, store) {
+		if (agentId in this._stores) {
+			const prevStore = this._stores[agentId];
+			delete this._stores[agentId];
+			this._snapshot = null;
+			this.notifyUnregistered(agentId, prevStore).catch((err) => {
+				console.error("ThreadStoreRegistry notifyUnregistered failed:", err);
+			});
+		}
+		this._stores[agentId] = store;
+		this._snapshot = null;
+		this.notifyRegistered(agentId, store).catch((err) => {
+			console.error("ThreadStoreRegistry notifyRegistered failed:", err);
+		});
+	}
+	unregister(agentId) {
+		if (!(agentId in this._stores)) return;
+		const prevStore = this._stores[agentId];
+		delete this._stores[agentId];
+		this._snapshot = null;
+		this.notifyUnregistered(agentId, prevStore).catch((err) => {
+			console.error("ThreadStoreRegistry notifyUnregistered failed:", err);
+		});
+	}
+	get(agentId) {
+		return this._stores[agentId];
+	}
+	getAll() {
+		if (this._snapshot === null) this._snapshot = Object.freeze({ ...this._stores });
+		return this._snapshot;
+	}
+	async notifyRegistered(agentId, store) {
+		await this.core.notifySubscribers((subscriber) => subscriber.onThreadStoreRegistered?.({
+			copilotkit: this.core,
+			agentId,
+			store
+		}), "Subscriber onThreadStoreRegistered error:");
+	}
+	async notifyUnregistered(agentId, prevStore) {
+		await this.core.notifySubscribers((subscriber) => subscriber.onThreadStoreUnregistered?.({
+			copilotkit: this.core,
+			agentId,
+			prevStore
+		}), "Subscriber onThreadStoreUnregistered error:");
+	}
+};
+
 //#endregion
 //#region src/core/core.ts
 let CopilotKitCoreErrorCode = /* @__PURE__ */ function(CopilotKitCoreErrorCode) {
@@ -2333,6 +2388,14 @@ var CopilotKitCore = class {
 	suggestionEngine;
 	runHandler;
 	stateManager;
+	threadStoreRegistry;
+	/**
+	* Tracks the agent IDs from the most recent `onAgentsChanged` notification.
+	* Used to gate thread-store auto-unregister so the FIRST empty-agents
+	* notification (before published agents are merged in) does not rip out a
+	* store that was registered prior to that initial notification.
+	*/
+	previousAgentIds = /* @__PURE__ */ new Set();
 	constructor({ runtimeUrl, runtimeTransport = "auto", headers = {}, credentials, properties = {}, agents__unsafe_dev_only = {}, tools = [], suggestionsConfig = [], debug }) {
 		this._headers = headers;
 		this._credentials = credentials;
@@ -2343,16 +2406,21 @@ var CopilotKitCore = class {
 		this.suggestionEngine = new SuggestionEngine(this);
 		this.runHandler = new RunHandler(this);
 		this.stateManager = new StateManager(this);
+		this.threadStoreRegistry = new ThreadStoreRegistry(this);
 		this.agentRegistry.initialize(agents__unsafe_dev_only);
 		this.runHandler.initialize(tools);
 		this.suggestionEngine.initialize(suggestionsConfig);
 		this.stateManager.initialize();
 		this.agentRegistry.setRuntimeTransport(runtimeTransport);
 		this.agentRegistry.setRuntimeUrl(runtimeUrl);
+		this.previousAgentIds = new Set(Object.keys(agents__unsafe_dev_only));
 		this.subscribe({ onAgentsChanged: ({ agents }) => {
 			Object.values(agents).forEach((agent) => {
 				if (agent.agentId) this.stateManager.subscribeToAgent(agent);
 			});
+			const currentAgentIds = new Set(Object.keys(agents));
+			for (const agentId of Object.keys(this.threadStoreRegistry.getAll())) if (this.previousAgentIds.has(agentId) && !currentAgentIds.has(agentId)) this.threadStoreRegistry.unregister(agentId);
+			this.previousAgentIds = currentAgentIds;
 		} });
 	}
 	/**
@@ -2527,6 +2595,21 @@ var CopilotKitCore = class {
 		this.contextStore.removeContext(id);
 	}
 	/**
+	* Thread store registry (delegated to ThreadStoreRegistry)
+	*/
+	registerThreadStore(agentId, store) {
+		this.threadStoreRegistry.register(agentId, store);
+	}
+	unregisterThreadStore(agentId) {
+		this.threadStoreRegistry.unregister(agentId);
+	}
+	getThreadStore(agentId) {
+		return this.threadStoreRegistry.get(agentId);
+	}
+	getThreadStores() {
+		return this.threadStoreRegistry.getAll();
+	}
+	/**
 	* Suggestions management (delegated to SuggestionEngine)
 	*/
 	addSuggestionsConfig(config) {
@@ -3611,6 +3694,11 @@ function createThreadStore(environment) {
 		setContext(context) {
 			store.dispatch(threadAdapterEvents.contextChanged({ context }));
 		},
+		refresh() {
+			const { sessionId, context } = store.getState();
+			if (!context) return;
+			store.dispatch(threadRestEvents.listRequested({ sessionId }));
+		},
 		fetchNextPage() {
 			store.dispatch(threadAdapterEvents.fetchNextPageRequested());
 		},
@@ -3645,6 +3733,13 @@ const ɵselectThreadsIsLoading = selectThreadsIsLoading;
 const ɵselectThreadsError = selectThreadsError;
 const ɵselectHasNextPage = selectHasNextPage;
 const ɵselectIsFetchingNextPage = selectIsFetchingNextPage;
+/**
+* Number of consecutive WebSocket connection failures after which the
+* threads channel tears itself down rather than retrying indefinitely.
+* Exposed for tests so they can assert teardown semantics without
+* hardcoding the threshold separately from production.
+*/
+const ɵMAX_SOCKET_RETRIES = MAX_SOCKET_RETRIES;
 
 //#endregion
 exports.AgentRegistry = AgentRegistry;
@@ -3672,6 +3767,7 @@ exports.on = on;
 exports.parseToolArguments = parseToolArguments;
 exports.props = props;
 exports.select = select;
+exports.ɵMAX_SOCKET_RETRIES = ɵMAX_SOCKET_RETRIES;
 exports.ɵcreateThreadStore = createThreadStore;
 exports.ɵjoinPhoenixChannel$ = ɵjoinPhoenixChannel$;
 exports.ɵobservePhoenixEvent$ = ɵobservePhoenixEvent$;