@mastra/code-sdk 1.8.4-alpha.6 → 1.9.0-alpha.10

package/dist/index.js

@@ -43,9 +43,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";
@@ -56,10 +63,11 @@ import { runMC } from "./headless/run-mc.js";
 import { formatHuman, formatJsonl, renderJsonResult, renderTextResult } from "./headless/format.js";
 import { hasHeadlessFlag, runMCCli } from "./headless/cli.js";
 import "./headless/index.js";
-import { TaskSignalProvider } from "@mastra/core/signals";
-import { createHash } from "crypto";
-import { hostname } from "os";
 import path from "path";
+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";
@@ -67,6 +75,7 @@ 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 { RequestContext } from "@mastra/core/request-context";
+import { TaskSignalProvider } from "@mastra/core/signals";
 import { InMemoryHarness, MastraCompositeStore } from "@mastra/core/storage";
 import { DEFAULT_GOAL_JUDGE_PROMPT } from "@mastra/core/tools";
 import { DuckDBStore } from "@mastra/duckdb";
@@ -224,7 +233,7 @@ async function createMastraCodeAgentController(config) {
 	let activeSession;
 	let pluginRuntimeController;
 	if (configDir !== ".mastracode") validateConfigDirName(configDir);
-	try {
+	if (!config?.disableEnvFile) try {
 		process.loadEnvFile(path.join(cwd, ".env"));
 	} catch {}
 	const authStorage = createAuthStorage();
@@ -276,12 +285,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 +316,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 +491,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,
@@ -559,7 +624,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: {
@@ -867,6 +932,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.
 		*