@copilotkit/runtime 1.72.0 → 1.73.0

package/dist/agent/index.cjs

@@ -4,6 +4,8 @@ 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");
@@ -59,6 +61,38 @@ function resolveModel(spec, apiKey) {
 	}
 }
 /**
+* Which vendor a model specifier will actually reach, as a closed vocabulary.
+*
+* `resolveModel` above is the only place this runtime builds a provider, so it
+* is the only place that knows the endpoint. Once built, the endpoint is gone:
+* an AI SDK model reports `provider: "openai.responses"` whether it points at
+* api.openai.com, Azure, OpenRouter or a laptop, and its base URL survives
+* only inside a closure that the public `LanguageModelV3` type does not
+* expose. Azure's own migration guide tells customers to use that same OpenAI
+* client, so the case we are blindest to is the common one.
+*
+* A caller who hands us an already-built LanguageModel gets `unknown`. Nobody
+* can recover the host from it, and saying so is more useful than guessing
+* `openai` from a provider label that means only "speaks the OpenAI wire".
+*
+* The branches below mirror `resolveModel`'s switch and must change with it.
+* `model-host-class.test.ts` walks every provider that switch accepts and
+* fails if one lands here as `unknown`.
+*/
+function classifyModelSpec(spec) {
+	if (typeof spec !== "string") return "unknown";
+	switch (spec.replace("/", ":").trim().split(":")[0]?.toLowerCase()) {
+		case "openai": return (0, _copilotkit_shared.classifyModelHost)(process.env.OPENAI_BASE_URL, "openai");
+		case "anthropic": return (0, _copilotkit_shared.classifyModelHost)(process.env.ANTHROPIC_BASE_URL, "anthropic");
+		case "google":
+		case "gemini":
+		case "google-gemini": return (0, _copilotkit_shared.classifyModelHost)(process.env.GOOGLE_GENERATIVE_AI_BASE_URL, "google");
+		case "minimax": return (0, _copilotkit_shared.classifyModelHost)(process.env.MINIMAX_BASE_URL || "https://api.minimax.io/v1", "minimax");
+		case "vertex": return "vertex";
+		default: return "unknown";
+	}
+}
+/**
 * Thrown by `AgentFactoryContext.interrupt()` on a fresh (non-resume) run to
 * pause the factory. Caught in `runFactory` and translated into a RUN_FINISHED
 * event carrying `outcome:{type:"interrupt",interrupts}`. Not a real error.
@@ -388,6 +422,8 @@ 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);
 	}
 	/**
 	* Check if a property can be overridden by forwardedProps
@@ -551,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)({
@@ -584,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
@@ -595,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) {
@@ -602,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
@@ -893,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(() => {});
 			};
 		});
@@ -936,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 = [];
@@ -954,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;
@@ -1007,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;
 	}
@@ -1038,6 +1111,7 @@ var BasicAgent = class extends BuiltInAgent {
 exports.BasicAgent = BasicAgent;
 exports.BuiltInAgent = BuiltInAgent;
 exports.InterruptSignal = InterruptSignal;
+exports.classifyModelSpec = classifyModelSpec;
 exports.convertJsonSchemaToZodSchema = convertJsonSchemaToZodSchema;
 exports.convertMessagesToVercelAISDKMessages = convertMessagesToVercelAISDKMessages;
 exports.convertToolDefinitionsToVercelAITools = convertToolDefinitionsToVercelAITools;