@electric-ax/agents 0.4.12 → 0.4.14

package/dist/index.cjs

@@ -39,6 +39,7 @@ const sqlite_vec = __toESM(require("sqlite-vec"));
 const nanoid = __toESM(require("nanoid"));
 const __mariozechner_pi_ai = __toESM(require("@mariozechner/pi-ai"));
 const __electric_ax_agents_mcp = __toESM(require("@electric-ax/agents-mcp"));
+const undici = __toESM(require("undici"));
 
 //#region src/log.ts
 const LOG_LEVEL = process.env.ELECTRIC_AGENTS_LOG_LEVEL ?? `info`;
@@ -889,6 +890,15 @@ async function fetchAvailableModelIds(provider) {
 function knownModelsForProvider(provider) {
 	return provider === __electric_ax_agents_runtime.MOONSHOT_PROVIDER ? (0, __electric_ax_agents_runtime.getMoonshotModels)() : (0, __mariozechner_pi_ai.getModels)(provider);
 }
+function resolveBuiltinModelContextWindow(modelConfig) {
+	const modelId = String(modelConfig.model);
+	if (modelConfig.provider === __electric_ax_agents_runtime.MOONSHOT_PROVIDER) return (0, __electric_ax_agents_runtime.getMoonshotModel)(modelId)?.contextWindow ?? null;
+	if (!modelConfig.provider) return null;
+	return knownModelsForProvider(modelConfig.provider).find((model) => model.id === modelId)?.contextWindow ?? null;
+}
+function resolveBuiltinModelSourceBudget(modelConfig) {
+	return resolveBuiltinModelContextWindow(modelConfig) ?? 1e5;
+}
 function choiceForKnownModel(provider, model) {
 	return {
 		provider,
@@ -992,6 +1002,8 @@ function modelInputSchemaDefs(catalog) {
 const HORTON_MODEL = `claude-sonnet-4-6`;
 const TITLE_SYSTEM_PROMPT = "You generate concise chat session titles in 3-5 words. Respond with only the title, no quotes, no punctuation, no preamble.";
 const TITLE_USER_PROMPT = (userMessage) => `User request:\n${userMessage}`;
+const TITLE_GENERATION_TIMEOUT_MS = 8e3;
+const HORTON_SKILLS_SLASH_COMMAND_OWNER = `horton:skills`;
 const TITLE_STOP_WORDS = new Set([
 	`a`,
 	`an`,
@@ -1071,6 +1083,17 @@ function createConfiguredTitleCall(catalog, modelConfig, logPrefix) {
 		maxTokens: 64
 	});
 }
+function withTimeout(promise, ms, description) {
+	let timeout;
+	const timeoutPromise = new Promise((_resolve, reject) => {
+		timeout = setTimeout(() => {
+			reject(new Error(`${description} timed out after ${ms}ms`));
+		}, ms);
+	});
+	return Promise.race([promise, timeoutPromise]).finally(() => {
+		if (timeout) clearTimeout(timeout);
+	});
+}
 async function generateTitle(userMessage, llmCall, onFallback) {
 	try {
 		const raw = await llmCall(TITLE_USER_PROMPT(userMessage));
@@ -1200,8 +1223,12 @@ function payloadToTitleText(payload) {
 	if (typeof payload === `string`) return payload;
 	if (payload == null) return ``;
 	if (typeof payload === `object`) {
-		const text = payload.text;
-		return typeof text === `string` ? text : JSON.stringify(payload);
+		const record = payload;
+		const text = record.text;
+		if (typeof text === `string`) return text;
+		const source = record.source;
+		if (typeof source === `string`) return source;
+		return JSON.stringify(payload);
 	}
 	return String(payload);
 }
@@ -1259,10 +1286,13 @@ async function readAgentsMd(sandbox) {
 }
 function createAssistantHandler(options) {
 	const { streamFn, docsSupport, docsSearchTool, skillsRegistry, modelCatalog, docsUrl } = options;
-	const hasSkills = Boolean(skillsRegistry && skillsRegistry.catalog.size > 0);
+	const skillLoader = (0, __electric_ax_agents_runtime.createContextSkillLoader)(skillsRegistry, { slashCommandOwner: HORTON_SKILLS_SLASH_COMMAND_OWNER });
+	const hasSkills = skillLoader.hasSkills;
 	return async function assistantHandler(ctx, wake) {
+		const loadedSkills = await skillLoader.load(ctx);
 		const readSet = new Set();
 		const modelConfig = resolveBuiltinModelConfig(modelCatalog, ctx.args);
+		const sourceBudget = resolveBuiltinModelSourceBudget(modelConfig);
 		const sandboxCwd = ctx.sandbox.workingDirectory;
 		const agentsMd = await readAgentsMd(ctx.sandbox);
 		const tools = [
@@ -1273,7 +1303,7 @@ function createAssistantHandler(options) {
 				modelCatalog,
 				logPrefix: `[horton ${ctx.entityUrl}]`
 			}),
-			...skillsRegistry && skillsRegistry.catalog.size > 0 ? (0, __electric_ax_agents_runtime.createSkillTools)(skillsRegistry, ctx) : [],
+			...loadedSkills.tools,
 			...__electric_ax_agents_mcp.mcp.tools()
 		];
 		const hasEventSourceTools = tools.some((tool) => getToolName(tool) === `list_event_sources`);
@@ -1282,21 +1312,22 @@ function createAssistantHandler(options) {
 			if (!firstUserMessage) return;
 			let title = null;
 			try {
-				const result = await generateTitle(firstUserMessage, createConfiguredTitleCall(modelCatalog, modelConfig, `[horton ${ctx.entityUrl}]`), (reason) => {
+				const result = await generateTitle(firstUserMessage, (prompt) => withTimeout(createConfiguredTitleCall(modelCatalog, modelConfig, `[horton ${ctx.entityUrl}]`)(prompt), TITLE_GENERATION_TIMEOUT_MS, `title generation`), (reason) => {
 					serverLog.warn(`[horton ${ctx.entityUrl}] title generation fell back to local title: ${reason}`);
 				});
 				if (result.length > 0) title = result;
 			} catch (err) {
 				serverLog.warn(`[horton ${ctx.entityUrl}] title generation failed: ${err instanceof Error ? err.message : String(err)}`);
+				title = buildFallbackTitle(firstUserMessage);
 			}
 			if (title !== null) try {
-				await ctx.setTag(`title`, title);
+				await withTimeout(ctx.setTag(`title`, title), TITLE_GENERATION_TIMEOUT_MS, `set title tag`);
 			} catch (err) {
 				serverLog.warn(`[horton ${ctx.entityUrl}] setTag failed: ${err instanceof Error ? err.message : String(err)}`);
 			}
 		})() : Promise.resolve();
 		if (docsSupport) ctx.useContext({
-			sourceBudget: 1e5,
+			sourceBudget,
 			sources: {
 				docs_toc: {
 					content: () => docsSupport.renderCompressedToc(),
@@ -1317,21 +1348,13 @@ function createAssistantHandler(options) {
 					max: 2e4,
 					cache: `stable`
 				} } : {},
-				...skillsRegistry && skillsRegistry.catalog.size > 0 ? { skills_catalog: {
-					content: () => skillsRegistry.renderCatalog(2e3),
-					max: 2e3,
-					cache: `stable`
-				} } : {}
+				...skillsRegistry && skillsRegistry.catalog.size > 0 ? loadedSkills.sources : {}
 			}
 		});
 		else if (skillsRegistry && skillsRegistry.catalog.size > 0) ctx.useContext({
-			sourceBudget: 1e5,
+			sourceBudget,
 			sources: {
-				skills_catalog: {
-					content: () => skillsRegistry.renderCatalog(2e3),
-					max: 2e3,
-					cache: `stable`
-				},
+				...loadedSkills.sources,
 				conversation: {
 					content: () => ctx.timelineMessages(),
 					cache: `volatile`
@@ -1344,7 +1367,7 @@ function createAssistantHandler(options) {
 			}
 		});
 		else if (agentsMd) ctx.useContext({
-			sourceBudget: 1e5,
+			sourceBudget,
 			sources: {
 				conversation: {
 					content: () => ctx.timelineMessages(),
@@ -1401,6 +1424,16 @@ function registerHorton(registry, options) {
 	registry.define(`horton`, {
 		description: `Friendly capable assistant — chat, code, research, dispatch`,
 		creationSchema: hortonCreationSchema,
+		permissionGrants: [{
+			subject_kind: `principal_kind`,
+			subject_value: `user`,
+			permission: `spawn`
+		}, {
+			subject_kind: `principal_kind`,
+			subject_value: `user`,
+			permission: `manage`
+		}],
+		slashCommands: (0, __electric_ax_agents_runtime.buildSkillSlashCommands)(skillsRegistry),
 		handler: assistantHandler
 	});
 	return [`horton`];
@@ -1442,7 +1475,7 @@ function parseWorkerArgs(value) {
 	if (typeof value.reasoningEffort === `string` && REASONING_EFFORT_VALUES.includes(value.reasoningEffort)) args.reasoningEffort = value.reasoningEffort;
 	return args;
 }
-function buildToolsForWorker(tools, sandbox, ctx, readSet) {
+function buildToolsForWorker(tools, sandbox, ctx, readSet, opts) {
 	const out = [];
 	for (const name of tools) switch (name) {
 		case `bash`:
@@ -1461,7 +1494,10 @@ function buildToolsForWorker(tools, sandbox, ctx, readSet) {
 			out.push(__electric_ax_agents_runtime_tools.braveSearchTool);
 			break;
 		case `fetch_url`:
-			out.push((0, __electric_ax_agents_runtime_tools.createFetchUrlTool)(sandbox));
+			out.push((0, __electric_ax_agents_runtime_tools.createFetchUrlTool)(sandbox, {
+				catalog: opts.modelCatalog,
+				modelConfig: opts.modelConfig
+			}));
 			break;
 		case `spawn_worker`:
 			out.push(createSpawnWorkerTool(ctx));
@@ -1572,11 +1608,23 @@ function registerWorker(registry, options) {
 	const { streamFn, modelCatalog } = options;
 	registry.define(`worker`, {
 		description: `Internal — generic worker spawned by other agents. Configure via spawn args (systemPrompt + tools + optional sharedDb).`,
+		permissionGrants: [{
+			subject_kind: `principal_kind`,
+			subject_value: `user`,
+			permission: `spawn`
+		}, {
+			subject_kind: `principal_kind`,
+			subject_value: `user`,
+			permission: `manage`
+		}],
 		async handler(ctx) {
 			const args = parseWorkerArgs(ctx.args);
 			const readSet = new Set();
-			const builtinTools = buildToolsForWorker(args.tools, ctx.sandbox, ctx, readSet);
 			const modelConfig = resolveBuiltinModelConfig(modelCatalog, args);
+			const builtinTools = buildToolsForWorker(args.tools, ctx.sandbox, ctx, readSet, {
+				modelCatalog,
+				modelConfig
+			});
 			const sharedStateTools = [];
 			if (args.sharedDb) {
 				const shared = await ctx.observe((0, __electric_ax_agents_runtime.db)(args.sharedDb.id, args.sharedDb.schema));
@@ -1770,6 +1818,18 @@ function resolveCwd(args, fallback) {
 	return readWorkingDirectoryArg(args) ?? fallback;
 }
 
+//#endregion
+//#region src/durable-streams-cache.ts
+const MEMORY_CACHE_SIZE_BYTES = 100 * 1024 * 1024;
+function installDurableStreamsFetchCache(options = {}) {
+	if (options === false) return;
+	const store = options.store === `sqlite` || options.sqliteLocation ? new undici.cacheStores.SqliteCacheStore({
+		location: options.sqliteLocation,
+		maxCount: options.maxCount
+	}) : new undici.cacheStores.MemoryCacheStore({ maxSize: MEMORY_CACHE_SIZE_BYTES });
+	(0, undici.setGlobalDispatcher)(new undici.Agent().compose(undici.interceptors.cache({ store })));
+}
+
 //#endregion
 //#region src/server.ts
 var BuiltinAgentsServer = class {
@@ -1779,6 +1839,8 @@ var BuiltinAgentsServer = class {
 	mcpToolProviderName = null;
 	mcpApplyInFlight = new Set();
 	mcpStopping = false;
+	mcpExtras = [];
+	mcpLastJsonConfig = null;
 	pullWakeRunner = null;
 	options;
 	constructor(options) {
@@ -1788,8 +1850,70 @@ var BuiltinAgentsServer = class {
 	get mcpRegistry() {
 		return this._mcpRegistry;
 	}
+	/**
+	* Replace the in-memory `extras` list and re-apply the merged config
+	* against the last-known workspace `mcp.json` state. Workspace
+	* `mcp.json` still wins on name collision. No-op once `stop()` has
+	* latched `mcpStopping`.
+	*/
+	async setExtraMcpServers(extras) {
+		if (!this._mcpRegistry || this.mcpStopping) return;
+		this.mcpExtras = extras;
+		await this.applyMerged(this.mcpLastJsonConfig);
+	}
+	async wirePersistence(cfg) {
+		const servers = [];
+		for (const s of cfg.servers) if (s.transport === `http` && s.auth?.mode === `authorizationCode`) {
+			const persist = await (0, __electric_ax_agents_mcp.keychainPersistence)({ server: s.name });
+			servers.push({
+				...s,
+				auth: {
+					...s.auth,
+					...persist
+				}
+			});
+		} else servers.push(s);
+		return {
+			...cfg,
+			servers
+		};
+	}
+	mergeMcp(jsonCfg) {
+		const jsonServers = jsonCfg?.servers ?? [];
+		const jsonNames = new Set(jsonServers.map((s) => s.name));
+		const filteredExtras = this.mcpExtras.filter((s) => !jsonNames.has(s.name));
+		return {
+			servers: [...filteredExtras, ...jsonServers],
+			raw: jsonCfg?.raw
+		};
+	}
+	async runApply(jsonCfg) {
+		if (this.mcpStopping) return;
+		const registry = this._mcpRegistry;
+		if (!registry) return;
+		try {
+			const wired = await this.wirePersistence(this.mergeMcp(jsonCfg));
+			if (this.mcpStopping) return;
+			await registry.applyConfig(wired);
+		} catch (e) {
+			serverLog.error(`[mcp] applyConfig:`, e);
+			try {
+				this.options.onConfigError?.(e);
+			} catch (cbErr) {
+				serverLog.error(`[mcp] onConfigError callback failed:`, cbErr);
+			}
+		}
+	}
+	applyMerged(jsonCfg) {
+		this.mcpLastJsonConfig = jsonCfg;
+		const p = this.runApply(jsonCfg);
+		this.mcpApplyInFlight.add(p);
+		p.finally(() => this.mcpApplyInFlight.delete(p));
+		return p;
+	}
 	async start() {
 		if (this.bootstrap || this.pullWakeRunner) throw new Error(`Builtin agents runtime already started`);
+		installDurableStreamsFetchCache(this.options.durableStreamsFetchCache);
 		const pullWake = this.options.pullWake;
 		if (!pullWake?.runnerId) throw new Error(`Builtin agents require a pull-wake runner id`);
 		try {
@@ -1800,76 +1924,28 @@ var BuiltinAgentsServer = class {
 			});
 			this._mcpRegistry = mcpRegistry;
 			const mcpConfigPath = this.options.loadProjectMcpConfig ? node_path.default.resolve(this.options.workingDirectory ?? process.cwd(), `mcp.json`) : null;
-			const extras = this.options.extraMcpServers ?? [];
-			const wirePersistence = async (cfg) => {
-				const servers = [];
-				for (const s of cfg.servers) if (s.transport === `http` && s.auth?.mode === `authorizationCode`) {
-					const persist = await (0, __electric_ax_agents_mcp.keychainPersistence)({ server: s.name });
-					servers.push({
-						...s,
-						auth: {
-							...s.auth,
-							...persist
-						}
-					});
-				} else servers.push(s);
-				return {
-					...cfg,
-					servers
-				};
-			};
-			const merge = (jsonCfg) => {
-				const jsonServers = jsonCfg?.servers ?? [];
-				const jsonNames = new Set(jsonServers.map((s) => s.name));
-				const filteredExtras = extras.filter((s) => !jsonNames.has(s.name));
-				return {
-					servers: [...filteredExtras, ...jsonServers],
-					raw: jsonCfg?.raw
-				};
-			};
-			const onConfigError = this.options.onConfigError;
-			const runApply = async (jsonCfg) => {
-				if (this.mcpStopping) return;
-				try {
-					const wired = await wirePersistence(merge(jsonCfg));
-					if (this.mcpStopping) return;
-					await mcpRegistry.applyConfig(wired);
-				} catch (e) {
-					serverLog.error(`[mcp] applyConfig:`, e);
-					try {
-						onConfigError?.(e);
-					} catch (cbErr) {
-						serverLog.error(`[mcp] onConfigError callback failed:`, cbErr);
-					}
-				}
-			};
-			const applyMerged = (jsonCfg) => {
-				const p = runApply(jsonCfg);
-				this.mcpApplyInFlight.add(p);
-				p.finally(() => this.mcpApplyInFlight.delete(p));
-				return p;
-			};
+			this.mcpExtras = this.options.extraMcpServers ?? [];
 			if (mcpConfigPath) {
 				try {
 					const cfg = await (0, __electric_ax_agents_mcp.loadConfig)(mcpConfigPath, process.env);
-					applyMerged(cfg);
+					this.applyMerged(cfg);
 				} catch (err) {
 					if (err.code !== `ENOENT`) throw err;
-					if (extras.length === 0) serverLog.info(`[mcp] no ${mcpConfigPath} — starting with no servers`);
-					else serverLog.info(`[mcp] no ${mcpConfigPath} — starting with ${extras.length} server(s) from extras`);
-					applyMerged(null);
+					if (this.mcpExtras.length === 0) serverLog.info(`[mcp] no ${mcpConfigPath} — starting with no servers`);
+					else serverLog.info(`[mcp] no ${mcpConfigPath} — starting with ${this.mcpExtras.length} server(s) from extras`);
+					this.applyMerged(null);
 				}
 				try {
 					this.mcpWatcherCloser = await (0, __electric_ax_agents_mcp.watchConfig)(mcpConfigPath, {
-						onChange: (cfg) => void applyMerged(cfg),
+						onChange: (cfg) => void this.applyMerged(cfg),
 						onError: (e) => serverLog.error(`[mcp] config error:`, e)
 					});
 				} catch (e) {
 					serverLog.error(`[mcp] config watcher failed to start:`, e);
 				}
 			} else {
-				if (extras.length > 0) serverLog.info(`[mcp] starting with ${extras.length} server(s) from extras`);
-				applyMerged(null);
+				if (this.mcpExtras.length > 0) serverLog.info(`[mcp] starting with ${this.mcpExtras.length} server(s) from extras`);
+				this.applyMerged(null);
 			}
 			this.mcpToolProviderName = `mcp`;
 			(0, __electric_ax_agents_runtime.registerToolProvider)({

package/dist/index.d.cts

@@ -36,12 +36,26 @@ declare function createAgentHandler(agentServerUrl: string, workingDirectory?: s
 declare function registerBuiltinAgentTypes(bootstrap: AgentHandlerResult): Promise<void>;
 declare const registerAgentTypes: typeof registerBuiltinAgentTypes;
 
+//#endregion
+//#region src/durable-streams-cache.d.ts
+type DurableStreamsFetchCacheOptions = false | {
+  store?: `memory` | `sqlite`;
+  sqliteLocation?: string;
+  maxCount?: number;
+};
+
 //#endregion
 //#region src/server.d.ts
 interface BuiltinAgentsServerOptions {
   agentServerUrl: string;
   workingDirectory?: string;
   mockStreamFn?: StreamFn;
+  /**
+   * Configure the process-wide HTTP cache used by Undici-backed fetch calls.
+   * Defaults to a 100 MiB in-memory cache. Pass `false` to leave the global
+   * dispatcher unchanged.
+   */
+  durableStreamsFetchCache?: DurableStreamsFetchCacheOptions;
   /** Pull-wake runner configuration for built-in agents. */
   pullWake: {
     runnerId: string;
@@ -92,11 +106,24 @@ declare class BuiltinAgentsServer {
   private mcpToolProviderName;
   private mcpApplyInFlight;
   private mcpStopping;
+  private mcpExtras;
+  private mcpLastJsonConfig;
   private pullWakeRunner;
   readonly options: BuiltinAgentsServerOptions;
   constructor(options: BuiltinAgentsServerOptions);
   /** Embedded MCP registry. `null` until `start()` has run. */
   get mcpRegistry(): Registry | null;
+  /**
+   * Replace the in-memory `extras` list and re-apply the merged config
+   * against the last-known workspace `mcp.json` state. Workspace
+   * `mcp.json` still wins on name collision. No-op once `stop()` has
+   * latched `mcpStopping`.
+   */
+  setExtraMcpServers(extras: ReadonlyArray<McpServerConfig$1>): Promise<void>;
+  private wirePersistence;
+  private mergeMcp;
+  private runApply;
+  private applyMerged;
   start(): Promise<string>;
   stop(): Promise<void>;
   private registerPullWakeRunner;

package/dist/index.d.ts

@@ -36,12 +36,26 @@ declare function createAgentHandler(agentServerUrl: string, workingDirectory?: s
 declare function registerBuiltinAgentTypes(bootstrap: AgentHandlerResult): Promise<void>;
 declare const registerAgentTypes: typeof registerBuiltinAgentTypes;
 
+//#endregion
+//#region src/durable-streams-cache.d.ts
+type DurableStreamsFetchCacheOptions = false | {
+  store?: `memory` | `sqlite`;
+  sqliteLocation?: string;
+  maxCount?: number;
+};
+
 //#endregion
 //#region src/server.d.ts
 interface BuiltinAgentsServerOptions {
   agentServerUrl: string;
   workingDirectory?: string;
   mockStreamFn?: StreamFn;
+  /**
+   * Configure the process-wide HTTP cache used by Undici-backed fetch calls.
+   * Defaults to a 100 MiB in-memory cache. Pass `false` to leave the global
+   * dispatcher unchanged.
+   */
+  durableStreamsFetchCache?: DurableStreamsFetchCacheOptions;
   /** Pull-wake runner configuration for built-in agents. */
   pullWake: {
     runnerId: string;
@@ -92,11 +106,24 @@ declare class BuiltinAgentsServer {
   private mcpToolProviderName;
   private mcpApplyInFlight;
   private mcpStopping;
+  private mcpExtras;
+  private mcpLastJsonConfig;
   private pullWakeRunner;
   readonly options: BuiltinAgentsServerOptions;
   constructor(options: BuiltinAgentsServerOptions);
   /** Embedded MCP registry. `null` until `start()` has run. */
   get mcpRegistry(): Registry | null;
+  /**
+   * Replace the in-memory `extras` list and re-apply the merged config
+   * against the last-known workspace `mcp.json` state. Workspace
+   * `mcp.json` still wins on name collision. No-op once `stop()` has
+   * latched `mcpStopping`.
+   */
+  setExtraMcpServers(extras: ReadonlyArray<McpServerConfig$1>): Promise<void>;
+  private wirePersistence;
+  private mergeMcp;
+  private runApply;
+  private applyMerged;
   start(): Promise<string>;
   stop(): Promise<void>;
   private registerPullWakeRunner;