@copilotkit/runtime 1.72.1-canary.sp-longthread-perf → 1.73.1

package/dist/agent/index.cjs

@@ -4,18 +4,20 @@ const require_state_delta = require('./state-delta.cjs');
 const require_usage = require('./converters/usage.cjs');
 const require_aisdk = require('./converters/aisdk.cjs');
 const require_tanstack = require('./converters/tanstack.cjs');
+const require_registry = require('../v2/runtime/intelligence-platform/skill-registry/registry.cjs');
+const require_learned_skills = require('./learned-skills.cjs');
 let _ai_sdk_openai = require("@ai-sdk/openai");
 let _copilotkit_shared = require("@copilotkit/shared");
 let _ai_sdk_google = require("@ai-sdk/google");
 let rxjs = require("rxjs");
 let _ag_ui_client = require("@ag-ui/client");
-let zod = require("zod");
 let _ag_ui_a2ui_middleware = require("@ag-ui/a2ui-middleware");
 let _cfworker_json_schema = require("@cfworker/json-schema");
 let ai = require("ai");
 let _ai_sdk_mcp = require("@ai-sdk/mcp");
 let _ai_sdk_anthropic = require("@ai-sdk/anthropic");
 let _ai_sdk_google_vertex = require("@ai-sdk/google-vertex");
+let zod = require("zod");
 let _modelcontextprotocol_sdk_client_streamableHttp_js = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
 
 //#region src/agent/index.ts
@@ -420,6 +422,7 @@ var BuiltInAgent = class BuiltInAgent extends _ag_ui_client.AbstractAgent {
 	constructor(config) {
 		super();
 		this.config = config;
+		this.skillRegistry = config.learnedSkills === void 0 ? void 0 : new require_registry.SkillRegistry(config.learnedSkills);
 		this.modelHostClass = isFactoryConfig(config) ? "unknown" : classifyModelSpec(config.model);
 	}
 	/**
@@ -584,6 +587,15 @@ This is state from the application that you can edit by calling AGUISendStateSna
 					subscriber.next(reasoningEnd);
 				};
 				try {
+					const learnedSkills = await require_learned_skills.prepareLearnedSkills(this.skillRegistry, abortController.signal);
+					abortController.signal.throwIfAborted();
+					if (learnedSkills.catalog) {
+						messages.unshift({
+							role: "system",
+							content: learnedSkills.catalog
+						});
+						if (config.maxSteps === void 0) streamTextParams.stopWhen = (0, ai.stepCountIs)(10);
+					}
 					streamTextParams.tools = {
 						...streamTextParams.tools,
 						AGUISendStateSnapshot: (0, ai.tool)({
@@ -617,6 +629,7 @@ This is state from the application that you can edit by calling AGUISendStateSna
 					};
 					if (config.mcpClients && config.mcpClients.length > 0) for (const client of config.mcpClients) {
 						const mcpTools = await client.tools();
+						abortController.signal.throwIfAborted();
 						streamTextParams.tools = {
 							...streamTextParams.tools,
 							...mcpTools
@@ -628,6 +641,7 @@ This is state from the application that you can edit by calling AGUISendStateSna
 						if (serverConfig.type === "http") transport = new _modelcontextprotocol_sdk_client_streamableHttp_js.StreamableHTTPClientTransport(new URL(serverConfig.url), serverConfig.options);
 						else if (serverConfig.type === "sse") {
 							const { SSEClientTransport } = await import("@modelcontextprotocol/sdk/client/sse.js");
+							abortController.signal.throwIfAborted();
 							transport = new SSEClientTransport(new URL(serverConfig.url), serverConfig.headers ? { requestInit: { headers: serverConfig.headers } } : void 0);
 						}
 						if (transport) {
@@ -635,21 +649,36 @@ This is state from the application that you can edit by calling AGUISendStateSna
 							try {
 								mcpClient = await (0, _ai_sdk_mcp.createMCPClient)({ transport });
 							} catch (err) {
+								abortController.signal.throwIfAborted();
 								console.error(`[CopilotKit] MCP server ${serverConfig.url} failed to connect — skipping it for this run:`, err);
 								continue;
 							}
+							if (abortController.signal.aborted) {
+								await mcpClient.close();
+								abortController.signal.throwIfAborted();
+							}
 							mcpClients.push(mcpClient);
 							try {
 								const mcpTools = await mcpClient.tools();
+								abortController.signal.throwIfAborted();
 								streamTextParams.tools = {
 									...streamTextParams.tools,
 									...mcpTools
 								};
 							} catch (err) {
+								abortController.signal.throwIfAborted();
 								console.error(`[CopilotKit] MCP server ${serverConfig.url} tools() failed — skipping its tools for this run:`, err);
 							}
 						}
 					}
+					if (this.skillRegistry) {
+						require_learned_skills.assertNoSkillToolConflicts(streamTextParams.tools ?? {});
+						streamTextParams.tools = {
+							...streamTextParams.tools,
+							...learnedSkills.tools
+						};
+					}
+					abortController.signal.throwIfAborted();
 					const response = (0, ai.streamText)({
 						...streamTextParams,
 						abortSignal: abortController.signal
@@ -926,11 +955,13 @@ This is state from the application that you can edit by calling AGUISendStateSna
 						subscriber.error(error);
 					}
 				} finally {
-					this.abortController = void 0;
+					if (this.abortController === abortController) this.abortController = void 0;
 					await Promise.all(mcpClients.map((client) => client.close()));
 				}
 			})();
 			return () => {
+				abortController.abort();
+				if (this.abortController === abortController) this.abortController = void 0;
 				Promise.all(mcpClients.map((client) => client.close())).catch(() => {});
 			};
 		});
@@ -969,13 +1000,16 @@ This is state from the application that you can edit by calling AGUISendStateSna
 				...input,
 				messages: [...input.messages, ...resumeToolMessages]
 			} : input;
-			const factoryCtx = {
-				...ctx,
-				input: factoryInput
-			};
 			(async () => {
 				const runFinishedDetails = {};
 				try {
+					const learnedSkills = await require_learned_skills.prepareLearnedSkills(this.skillRegistry, controller.signal);
+					controller.signal.throwIfAborted();
+					const factoryCtx = {
+						...ctx,
+						input: factoryInput,
+						learnedSkills
+					};
 					let events;
 					let customRunFinishedEvent;
 					const pendingInterrupts = [];
@@ -987,7 +1021,7 @@ This is state from the application that you can edit by calling AGUISendStateSna
 							events = require_tanstack.convertTanStackStream(await config.factory(factoryCtx), controller.signal, pendingInterrupts, input.state, runFinishedDetails);
 							break;
 						case "custom":
-							events = await config.factory(ctx);
+							events = await config.factory(factoryCtx);
 							break;
 						default: {
 							const _exhaustive = config;
@@ -1040,16 +1074,22 @@ This is state from the application that you can edit by calling AGUISendStateSna
 						subscriber.error(error);
 					}
 				} finally {
-					this.abortController = void 0;
+					if (this.abortController === controller) this.abortController = void 0;
 				}
 			})();
 			return () => {
 				controller.abort();
+				if (this.abortController === controller) this.abortController = void 0;
 			};
 		});
 	}
 	clone() {
-		const cloned = new BuiltInAgent(this.config);
+		const cloned = new BuiltInAgent({
+			...this.config,
+			learnedSkills: void 0
+		});
+		cloned.config = this.config;
+		cloned.skillRegistry = this.skillRegistry;
 		cloned.middlewares = [...this.middlewares];
 		return cloned;
 	}