@mastra/code-sdk 1.9.0-alpha.9 → 1.10.0-alpha.0

package/dist/index.js

@@ -5,6 +5,7 @@ import { AuthStorage } from "./auth/storage.js";
 import { setAuthStorage } from "./providers/openai-codex.js";
 import { getAvailableModePacks, getAvailableOmPacks, selectPreferredOMPack } from "./onboarding/packs.js";
 import { MASTRA_GATEWAY_PROVIDER, OBSERVABILITY_AUTH_PREFIX, THREAD_ACTIVE_MODEL_PACK_ID_KEY, loadSettings, resolveModelDefaults, resolveOmRoleModel, saveSettings } from "./onboarding/settings.js";
+import { resolveExperimentalAgent, validateExperimentalAgent, wrapExperimentalAgent } from "./experimental-agent.js";
 import { createSessionThreadAdvertisement } from "./agent-connections/session-advertisement.js";
 import { AgentConnectionsSignalProvider } from "./agent-connections/signal-provider.js";
 import { createBackgroundCompletionEvents } from "./agents/background-completion-events.js";
@@ -43,9 +44,16 @@ import { getToolCategory } from "./permissions.js";
 import { PluginManager } from "./plugins/manager.js";
 import { PluginSignalLane } from "./plugins/signal-lane.js";
 import { PlanRejectionAbortProcessor } from "./processors/plan-rejection-abort.js";
+import { assembleSchedulePrompt } from "./schedules/prompt.js";
+import { ThreadScheduler, scheduleSignalAttributes, shortScheduleId } from "./schedules/scheduler.js";
+import { runScript } from "./schedules/run-script.js";
+import { createScheduleTools } from "./schedules/tools.js";
+import "./schedules/index.js";
 import { stateSchema } from "./schema.js";
 import { mastraBrand } from "./theme-palette.js";
+import { DiscardingScoresStorage } from "./utils/discarding-scores-storage.js";
 import { syncGateways } from "./utils/gateway-sync.js";
+import { registerSessionAndWaitForMaintenance, unregisterSession } from "./utils/maintenance-lock.js";
 import { createSignalsPubSub } from "./utils/signals-pubsub.js";
 import { DEFAULT_RETENTION, createStorageMaintenance, resolveLocalDbFiles } from "./utils/storage-maintenance.js";
 import { createStorage, createVectorStore } from "./utils/storage-factory.js";
@@ -57,14 +65,16 @@ import { formatHuman, formatJsonl, renderJsonResult, renderTextResult } from "./
 import { hasHeadlessFlag, runMCCli } from "./headless/cli.js";
 import "./headless/index.js";
 import path from "path";
-import { hostname } from "os";
+import { statSync } from "fs";
+import { homedir, hostname } from "os";
 import { createHash } from "crypto";
+import { readFile } from "fs/promises";
 import { AgentController } from "@mastra/core/agent-controller";
 import { createCodingAgent } from "@mastra/core/coding-agent";
-import { PROVIDER_REGISTRY } from "@mastra/core/llm";
+import { PROVIDER_REGISTRY, findGatewayForModel, getGatewayId } from "@mastra/core/llm";
 import { Mastra } from "@mastra/core/mastra";
 import { defaultNotificationDeliveryDecision } from "@mastra/core/notifications";
-import { AgentsMDInjector, CyberRefusalHandler, PrefillErrorHandler, ProviderHistoryCompat, StreamErrorRetryProcessor, createBackgroundWorkSignalProcessor, isBadRequestError } from "@mastra/core/processors";
+import { AgentsMDInjector, CyberRefusalHandler, ProviderHistoryCompat, StreamErrorRetryProcessor, createBackgroundWorkSignalProcessor, isBadRequestError } from "@mastra/core/processors";
 import { RequestContext } from "@mastra/core/request-context";
 import { TaskSignalProvider } from "@mastra/core/signals";
 import { InMemoryHarness, MastraCompositeStore } from "@mastra/core/storage";
@@ -217,6 +227,15 @@ function resolveInjectedStorageBackend(storage, configuredBackend) {
 	if (storage instanceof PostgresStore || hasAncestorClassNamed(storage, "PostgresStore")) return "pg";
 	throw new Error("storageBackend is required when injecting a custom storage instance.");
 }
+/** Whether the model router would resolve `modelId` through a gateway outside `sdkGatewayIds`. */
+function routesToOtherGateway(modelId, gateways, sdkGatewayIds) {
+	try {
+		return !sdkGatewayIds.has(getGatewayId(findGatewayForModel(modelId, gateways)));
+	} catch (error) {
+		if (error?.id === "MODEL_ROUTER_NO_GATEWAY_FOUND") return false;
+		throw error;
+	}
+}
 async function createMastraCodeAgentController(config) {
 	const cwd = config?.cwd ?? process.cwd();
 	const homeDir = config?.homeDir ?? config?.initialState?.homeDir;
@@ -229,6 +248,7 @@ async function createMastraCodeAgentController(config) {
 	} catch {}
 	const authStorage = createAuthStorage();
 	const globalSettings = loadSettings(config?.settingsPath);
+	const experimentalAgent = resolveExperimentalAgent(globalSettings);
 	const backgroundToolsEnabled = globalSettings.backgroundTools?.enabled ?? false;
 	const storedGatewayKey = authStorage.getStoredApiKey(MASTRA_GATEWAY_PROVIDER);
 	const storedGatewayUrl = globalSettings.memoryGateway?.baseUrl;
@@ -276,12 +296,21 @@ async function createMastraCodeAgentController(config) {
 	const crossProcessPubSub = config?.crossProcessPubSub ?? (!configuredPubSub && useUnixSocketPubSub);
 	if (crossProcessPubSub && !signalsPubSub) throw new Error("crossProcessPubSub requires a pubsub instance");
 	const useCrossAgentSignals = config?.crossAgentSignals ?? globalSettings.signals?.experimentalCrossAgentSignals ?? false;
+	const useScheduleTools = config?.scheduleTools ?? globalSettings.signals?.experimentalScheduleTools ?? false;
 	const injectedStorage = isInjectedStorageInstance(config?.storage) ? config.storage : void 0;
 	const storageConfig = injectedStorage ? void 0 : config?.storage ?? getStorageConfig(project.rootPath, globalSettings.storage, configDir);
-	const storageResult = injectedStorage ? {
+	if (!injectedStorage) await registerSessionAndWaitForMaintenance({ onWait: (pid) => console.error(`Waiting for storage maintenance (mastracode prune${pid === -1 ? "" : `, PID ${pid}`}) to finish. Press Ctrl+C to give up.`) });
+	let storageResult;
+	if (injectedStorage) storageResult = {
 		storage: injectedStorage,
 		backend: resolveInjectedStorageBackend(injectedStorage, config?.storageBackend)
-	} : await createStorage(storageConfig);
+	};
+	else try {
+		storageResult = await createStorage(storageConfig);
+	} catch (error) {
+		unregisterSession();
+		throw error;
+	}
 	const storageWarning = storageResult.warning;
 	let observabilityDomain;
 	let observabilityWarning;
@@ -298,12 +327,14 @@ async function createMastraCodeAgentController(config) {
 		else observabilityWarning = `Observability unavailable — DuckDB initialization failed: ${message}`;
 	}
 	const harnessStorage = new InMemoryHarness();
+	const scoresStorage = new DiscardingScoresStorage();
 	const storage = new MastraCompositeStore({
 		id: "mastra-code-storage",
 		default: storageResult.storage,
 		domains: {
 			observability: observabilityDomain ?? false,
-			harness: harnessStorage
+			harness: harnessStorage,
+			scores: scoresStorage
 		}
 	});
 	const observability = new Observability({ configs: { default: {
@@ -471,6 +502,51 @@ async function createMastraCodeAgentController(config) {
 			modelSettings: { temperature: 1 }
 		};
 	};
+	const threadScheduler = new ThreadScheduler({
+		assemblePrompt: (schedule) => assembleSchedulePrompt(schedule, {
+			cwd: project.rootPath,
+			runScript,
+			readFile: (absPath) => readFile(absPath, "utf8")
+		}),
+		deliver: async (schedule, assembled) => {
+			const target = {
+				resourceId: schedule.resourceId,
+				threadId: schedule.threadId
+			};
+			const streamOptions = await getWakeStreamOptions(target);
+			const accepted = await codeAgent.sendSignal({
+				type: "user",
+				tagName: "user",
+				contents: assembled.prompt,
+				attributes: scheduleSignalAttributes(schedule, assembled)
+			}, {
+				...target,
+				ifActive: { behavior: "deliver" },
+				ifIdle: {
+					behavior: "wake",
+					...streamOptions ? { streamOptions } : {}
+				}
+			}).accepted;
+			if (accepted.action === "blocked" || accepted.action === "discard") throw new Error(`the agent did not accept the prompt (${accepted.action})`);
+		},
+		onError: (error, schedule) => {
+			console.warn(`Schedule ${shortScheduleId(schedule.id)} failed to fire:`, error);
+		}
+	});
+	const scheduleTools = useScheduleTools ? createScheduleTools({
+		scheduler: threadScheduler,
+		fileOptions: () => ({
+			cwd: project.rootPath,
+			homeDir: homedir(),
+			fileExists: (absPath) => {
+				try {
+					return statSync(absPath).isFile();
+				} catch {
+					return false;
+				}
+			}
+		})
+	}) : void 0;
 	const githubSignals = globalSettings.signals?.experimentalGithubSignals && !config?.disableGithubSignals ? new GithubSignals({
 		cwd: project.rootPath,
 		pollIntervalMs: globalSettings.signals.githubPollIntervalMs,
@@ -497,8 +573,7 @@ async function createMastraCodeAgentController(config) {
 				const projectPath = state?.projectPath ?? project.rootPath;
 				return getStaticallyLoadedInstructionPaths(projectPath, void 0, state?.untrustedCheckout === true && typeof state?.baseRef === "string" ? createGitRefInstructionReader(projectPath, state.baseRef) : void 0);
 			}
-		}),
-		new ProviderHistoryCompat()
+		})
 	];
 	const taskSignalProvider = new TaskSignalProvider();
 	const agentConnectionsSignalProvider = useCrossAgentSignals ? new AgentConnectionsSignalProvider(config?.agentConnections) : void 0;
@@ -532,7 +607,7 @@ async function createMastraCodeAgentController(config) {
 			return NO_PLUGIN_PROCESSORS;
 		}
 	};
-	const codeAgent = createCodingAgent({
+	const codeAgent = wrapExperimentalAgent(createCodingAgent({
 		id: CODE_AGENT_ID,
 		name: "Code Agent",
 		workspace: void 0,
@@ -559,7 +634,7 @@ async function createMastraCodeAgentController(config) {
 				streamOptions
 			} : decision;
 		} } },
-		tools: createDynamicTools(mcpManager, config?.extraTools, config?.disabledTools, storage, pluginTools, backgroundToolsEnabled),
+		tools: createDynamicTools(mcpManager, config?.extraTools, config?.disabledTools, storage, pluginTools, backgroundToolsEnabled, scheduleTools),
 		hooks: createToolHooks(hookManager, config?.postToolObserver),
 		scorers: {
 			outcome: {
@@ -622,7 +697,6 @@ async function createMastraCodeAgentController(config) {
 					onRetry: ({ error, retryCount, delayMs, requestContext }) => emitTransientRetry(error, retryCount, delayMs, requestContext)
 				}
 			] }),
-			new PrefillErrorHandler(),
 			new AccountRotationProcessor({
 				credentialStore: authStorage,
 				maxProcessorRetries: MASTRACODE_MAX_PROCESSOR_RETRIES,
@@ -630,7 +704,7 @@ async function createMastraCodeAgentController(config) {
 			})
 		],
 		maxProcessorRetries: MASTRACODE_MAX_PROCESSOR_RETRIES
-	});
+	}), experimentalAgent);
 	const defaultModes = [
 		{
 			...buildMode,
@@ -748,6 +822,11 @@ async function createMastraCodeAgentController(config) {
 		stateSchema: typedStateSchema,
 		agent: codeAgent,
 		subagents,
+		resolveSubagentModel: (modelId, { requestContext }) => routesToOtherGateway(modelId, Object.values(controller.getMastra()?.listGateways() ?? {}), /* @__PURE__ */ new Set([
+			mastraCodeGateway.id,
+			amazonBedrockGateway.id,
+			"models.dev"
+		])) ? modelId : resolveModel(modelId, { requestContext }),
 		gateways: [amazonBedrockGateway, mastraCodeGateway],
 		workspace: config?.workspace ?? ((args) => getDynamicWorkspace({
 			...args,
@@ -867,6 +946,8 @@ async function createMastraCodeAgentController(config) {
 			unsubscribePluginReload = void 0;
 			pluginSignalLane?.stopAll();
 		},
+		/** Process-local `/schedules` scheduler. Call `stop()` on shutdown. */
+		threadScheduler,
 		/**
 		* Hands Mastra to the statically configured input processors.
 		*
@@ -941,6 +1022,7 @@ async function bootLocalAgentController(config) {
 	const base = await createMastraCodeAgentController(config);
 	const { controller, sessionId, ownerId, projectPath, codeAgent, mcpManager } = base;
 	await controller.init();
+	validateExperimentalAgent(codeAgent, controller.getMastra());
 	const mastra = controller.getMastra();
 	if (mastra) await registerWorkflowBuilderPrimitives(mastra, {
 		projectPath,
@@ -1036,6 +1118,7 @@ async function prepareAgentControllerMount(config) {
 	};
 	const finalize = async () => {
 		await controller.init();
+		validateExperimentalAgent(codeAgent, controller.getMastra());
 		if (weOwnTheMastra) {
 			const mastra = controller.getMastra();
 			if (mastra) await registerWorkflowBuilderPrimitives(mastra, {