@mastra/code-sdk 1.7.3-alpha.0 → 1.8.0-alpha.11

package/dist/index.js

@@ -3,9 +3,12 @@ import { detectProject, getObservabilityDatabasePath, getResourceIdOverride, get
 import { isKimiCodingDeviceId } from "./auth/providers/kimi-coding.js";
 import { AuthStorage } from "./auth/storage.js";
 import { setAuthStorage } from "./providers/openai-codex.js";
-import { MASTRA_GATEWAY_PROVIDER, OBSERVABILITY_AUTH_PREFIX, loadSettings, resolveModelDefaults, resolveOmRoleModel, saveSettings } from "./onboarding/settings.js";
-import { createThreadOwnershipManager } from "./agent-connections/ownership.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 { createSessionThreadAdvertisement } from "./agent-connections/session-advertisement.js";
 import { AgentConnectionsSignalProvider } from "./agent-connections/signal-provider.js";
+import { createBackgroundCompletionEvents } from "./agents/background-completion-events.js";
+import { createBackgroundCompletionCallbacks } from "./agents/background-completion.js";
 import { hasCredentialStoreProvider } from "./agents/credential-resolver.js";
 import { getDynamicWorkspace, getGoalJudgeTools } from "./agents/workspace.js";
 import { createGitRefInstructionReader, createGitRefReminderReader, getStaticallyLoadedInstructionPaths } from "./agents/prompts/agent-instructions.js";
@@ -26,6 +29,7 @@ import { exploreSubagent } from "./agents/subagents/explore.js";
 import { planSubagent } from "./agents/subagents/plan.js";
 import { attachOMThreadStatePersistence, restoreOMThreadStateForCurrentThread } from "./agents/thread-caveman-state.js";
 import { createDynamicTools, createToolHooks } from "./agents/tools.js";
+import { AccountRotationProcessor, AccountStartNoticeProcessor, PACK_FALLBACK_STATE_KEY } from "./auth/account-rotation-processor.js";
 import { createOutcomeScorer } from "./evals/scorers/outcome.js";
 import { createEfficiencyScorer } from "./evals/scorers/efficiency.js";
 import "./evals/scorers/index.js";
@@ -35,7 +39,6 @@ import { KnowledgeInspectorError, createKnowledgeInspector } from "./knowledge-i
 import { createMcpManager } from "./mcp/manager.js";
 import "./mcp/index.js";
 import { hasExplicitOMConfiguration } from "./onboarding/om-settings.js";
-import { getAvailableModePacks, getAvailableOmPacks, selectPreferredOMPack } from "./onboarding/packs.js";
 import { getToolCategory } from "./permissions.js";
 import { PluginManager } from "./plugins/manager.js";
 import { PluginSignalLane } from "./plugins/signal-lane.js";
@@ -62,7 +65,7 @@ import { createCodingAgent } from "@mastra/core/coding-agent";
 import { PROVIDER_REGISTRY } from "@mastra/core/llm";
 import { Mastra } from "@mastra/core/mastra";
 import { defaultNotificationDeliveryDecision } from "@mastra/core/notifications";
-import { AgentsMDInjector, PrefillErrorHandler, ProviderHistoryCompat, StreamErrorRetryProcessor, isBadRequestError } from "@mastra/core/processors";
+import { AgentsMDInjector, PrefillErrorHandler, ProviderHistoryCompat, StreamErrorRetryProcessor, createBackgroundWorkSignalProcessor, isBadRequestError } from "@mastra/core/processors";
 import { RequestContext } from "@mastra/core/request-context";
 import { InMemoryHarness, MastraCompositeStore } from "@mastra/core/storage";
 import { DEFAULT_GOAL_JUDGE_PROMPT } from "@mastra/core/tools";
@@ -74,15 +77,12 @@ import { PostgresStore } from "@mastra/pg";
 //#region src/index.ts
 const CODE_AGENT_ID = "code-agent";
 const MASTRACODE_TRANSIENT_CONNECTION_MAX_RETRIES = 10;
+const MASTRACODE_MAX_PROCESSOR_RETRIES = 64;
 const MASTRACODE_TRANSIENT_CONNECTION_RETRY_INITIAL_DELAY_MS = 500;
 const MASTRACODE_TRANSIENT_CONNECTION_RETRY_MAX_DELAY_MS = 3e4;
 const TRANSIENT_CONNECTION_ERROR_CODES = /* @__PURE__ */ new Set(["ECONNRESET", "EPIPE"]);
 const TRANSIENT_CONNECTION_MESSAGE_PATTERN = /econnreset|socket hang up|write epipe|other side closed/i;
-const TRANSIENT_SERVER_ERROR_STATUSES = /* @__PURE__ */ new Set([
-	500,
-	502,
-	503
-]);
+const isTransientServerStatus = (status) => status >= 500 && status < 600;
 const TRANSIENT_SERVER_ERROR_MESSAGE_PATTERN = /internal server|server error|api may be experiencing issues/i;
 /**
 * Matcher for transient connection failures. Cause-chain traversal is handled
@@ -107,7 +107,7 @@ function isTransientConnectionError(error) {
 function isTransientServerError(error) {
 	if (!error) return false;
 	const errorObj = typeof error === "object" ? error : void 0;
-	if (typeof errorObj?.status === "number" && TRANSIENT_SERVER_ERROR_STATUSES.has(errorObj.status) || typeof errorObj?.statusCode === "number" && TRANSIENT_SERVER_ERROR_STATUSES.has(errorObj.statusCode)) return true;
+	if (typeof errorObj?.status === "number" && isTransientServerStatus(errorObj.status) || typeof errorObj?.statusCode === "number" && isTransientServerStatus(errorObj.statusCode)) return true;
 	const message = error instanceof Error ? error.message : void 0;
 	return typeof message === "string" && TRANSIENT_SERVER_ERROR_MESSAGE_PATTERN.test(message);
 }
@@ -221,6 +221,7 @@ async function createMastraCodeAgentController(config) {
 	} catch {}
 	const authStorage = createAuthStorage();
 	const globalSettings = loadSettings(config?.settingsPath);
+	const backgroundToolsEnabled = globalSettings.backgroundTools?.enabled ?? false;
 	const storedGatewayKey = authStorage.getStoredApiKey(MASTRA_GATEWAY_PROVIDER);
 	const storedGatewayUrl = globalSettings.memoryGateway?.baseUrl;
 	if (storedGatewayKey) process.env["MASTRA_GATEWAY_API_KEY"] ??= storedGatewayKey;
@@ -327,7 +328,7 @@ async function createMastraCodeAgentController(config) {
 		localDbFiles: storageConfig ? resolveLocalDbFiles(storageConfig, storageResult.backend) : [],
 		closeVector: vector instanceof LibSQLVector ? () => vector.close() : void 0
 	});
-	const memory = config?.memory === false ? void 0 : config?.memory ?? getDynamicMemory(storage, vector);
+	const memory = config?.memory === false ? void 0 : config?.memory ?? getDynamicMemory(storage, vector, config?.settingsPath);
 	const hasSubconscious = config?.memory === void 0 ? (state) => hasSubconsciousTools(vector, state) : false;
 	const mcpManager = config?.disableMcp ? void 0 : createMcpManager(project.rootPath, configDir, config?.mcpServers, globalSettings.mcp);
 	const hookManager = config?.disableHooks ? void 0 : new HookManager(project.rootPath, "session-init", configDir, homeDir, project.isWorktree ? {
@@ -342,7 +343,12 @@ async function createMastraCodeAgentController(config) {
 	});
 	pluginManager?.setRuntime({
 		getController: () => pluginRuntimeController,
-		getActiveSession: () => activeSession
+		getActiveSession: () => activeSession,
+		getStorage: () => ({
+			storage,
+			storageBackend: storageResult.backend,
+			vector
+		})
 	});
 	const loadedPlugins = pluginManager ? await pluginManager.reload() : [];
 	const pluginTools = pluginManager?.getPluginTools() ?? {};
@@ -351,15 +357,63 @@ async function createMastraCodeAgentController(config) {
 	const getNotificationStreamOptions = async ({ resourceId, threadId }) => {
 		const session = await controller.getSessionByResource(resourceId) ?? activeSession;
 		if (!session) return void 0;
-		const modeId = session.mode.get();
-		const defaultModeModelId = controller.listModes().find((mode) => mode.id === modeId)?.defaultModelId;
-		const modelId = session.model.get() || activeSession?.model.get() || defaultModeModelId || "";
+		const targetThread = await session.thread.getById({ threadId });
+		const metadata = targetThread?.resourceId === resourceId ? targetThread.metadata : void 0;
+		const modes = controller.listModes();
+		const savedModeId = metadata?.currentModeId;
+		const defaultMode = modes.find((mode) => mode.default) ?? modes[0];
+		const modeId = typeof savedModeId === "string" && modes.some((mode) => mode.id === savedModeId) ? savedModeId : defaultMode?.id ?? session.mode.get();
+		const savedModeModelId = metadata?.[`modeModelId_${modeId}`];
+		const legacyModelId = metadata?.currentModelId;
+		const defaultModeModelId = modes.find((mode) => mode.id === modeId)?.defaultModelId;
+		const modelId = (typeof savedModeModelId === "string" ? savedModeModelId : void 0) ?? (typeof legacyModelId === "string" ? legacyModelId : void 0) ?? defaultModeModelId ?? session.model.get() ?? "";
+		const baseState = { ...session.state.get() };
+		delete baseState.activeModelPackId;
+		delete baseState.mastracodePendingPackFallback;
+		const persistedSandboxPaths = metadata?.sandboxAllowedPaths;
+		baseState.sandboxAllowedPaths = Array.isArray(persistedSandboxPaths) && persistedSandboxPaths.every((path) => typeof path === "string") ? persistedSandboxPaths : [];
+		const persistedStateKeys = [
+			"thinkingLevel",
+			"notifications",
+			THREAD_ACTIVE_MODEL_PACK_ID_KEY,
+			PACK_FALLBACK_STATE_KEY
+		];
+		for (const key of persistedStateKeys) {
+			const value = metadata?.[key];
+			if (value !== void 0) baseState[key] = value;
+		}
+		let notificationState = baseState;
+		const getNotificationState = () => ({ ...notificationState });
+		const setNotificationState = async (updates) => {
+			notificationState = {
+				...notificationState,
+				...updates
+			};
+			if (session.thread.getId() === threadId) await session.state.set(updates);
+		};
+		const updateNotificationState = async (updater) => {
+			const update = await updater(getNotificationState());
+			if (update.updates) await setNotificationState(update.updates);
+			for (const event of update.events ?? []) if (session.thread.getId() === threadId) session.emit(event);
+			return update.result;
+		};
 		const requestContext = new RequestContext();
 		const agentControllerContext = {
 			controllerId: controller.id,
-			state: session.state.get(),
-			getState: () => session.state.get(),
-			setState: (updates) => session.state.set(updates),
+			state: getNotificationState(),
+			getState: getNotificationState,
+			setState: setNotificationState,
+			updateState: updateNotificationState,
+			getThreadSetting: (key) => session.thread.getSettingOn({
+				threadId,
+				key
+			}),
+			setThreadSetting: (setting) => session.thread.setSettingOn({
+				threadId,
+				key: setting.key,
+				value: setting.value
+			}),
+			isThreadActive: () => session.thread.getId() === threadId,
 			threadId,
 			resourceId,
 			session: {
@@ -368,13 +422,23 @@ async function createMastraCodeAgentController(config) {
 				modeId,
 				modelId,
 				state: {
-					get: () => session.state.get(),
-					set: (updates) => session.state.set(updates),
-					update: (updater) => session.state.update(updater)
+					get: getNotificationState,
+					set: setNotificationState,
+					update: updateNotificationState
 				}
 			},
 			workspace: session.getWorkspace(),
-			getSubagentModelId: (params) => session.subagents.model.get(params ?? {})
+			emitEvent: (event) => {
+				if (session.thread.getId() === threadId) session.emit(event);
+			},
+			getSubagentModelId: (params) => {
+				const agentType = params?.agentType;
+				const perType = agentType ? metadata?.[`subagentModelId_${agentType}`] : void 0;
+				const global = metadata?.subagentModelId;
+				if (typeof perType === "string") return perType;
+				if (typeof global === "string") return global;
+				return session.thread.getId() === threadId ? session.subagents.model.get(params ?? {}) : null;
+			}
 		};
 		requestContext.set("controller", agentControllerContext);
 		return {
@@ -385,7 +449,7 @@ async function createMastraCodeAgentController(config) {
 			requestContext,
 			maxSteps: 1e3,
 			savePerStep: false,
-			requireToolApproval: session.state.get().yolo !== true,
+			requireToolApproval: notificationState.yolo !== true,
 			modelSettings: { temperature: 1 }
 		};
 	};
@@ -398,6 +462,7 @@ async function createMastraCodeAgentController(config) {
 	const mastraCodeInputProcessors = [
 		...config?.inputProcessors ?? [],
 		new PlanRejectionAbortProcessor(),
+		...backgroundToolsEnabled ? [createBackgroundWorkSignalProcessor()] : [],
 		new AgentsMDInjector({
 			isEnabled: ({ requestContext }) => {
 				const state = getInjectorSessionState(requestContext);
@@ -462,7 +527,6 @@ async function createMastraCodeAgentController(config) {
 				hasSubagents: subagents.length > 0
 			});
 		},
-		maxProcessorRetries: MASTRACODE_TRANSIENT_CONNECTION_MAX_RETRIES,
 		model: (ctx) => getDynamicModel(ctx, config?.settingsPath),
 		notifications: { deliveryPolicy: { decide: async (input) => {
 			const decision = defaultNotificationDeliveryDecision(input);
@@ -476,7 +540,7 @@ async function createMastraCodeAgentController(config) {
 				streamOptions
 			} : decision;
 		} } },
-		tools: createDynamicTools(mcpManager, config?.extraTools, config?.disabledTools, storage, pluginTools),
+		tools: createDynamicTools(mcpManager, config?.extraTools, config?.disabledTools, storage, pluginTools, backgroundToolsEnabled),
 		hooks: createToolHooks(hookManager, config?.postToolObserver),
 		scorers: {
 			outcome: {
@@ -505,6 +569,10 @@ async function createMastraCodeAgentController(config) {
 		},
 		inputProcessors: () => [
 			...mastraCodeInputProcessors,
+			new AccountStartNoticeProcessor({
+				credentialStore: authStorage,
+				settingsPath: config?.settingsPath
+			}),
 			...readPluginProcessors().input.map((entry) => entry.value),
 			...pluginSignalLane?.getInputProcessors() ?? []
 		],
@@ -530,8 +598,14 @@ async function createMastraCodeAgentController(config) {
 					onRetry: ({ error, retryCount, delayMs, requestContext }) => emitTransientRetry(error, retryCount, delayMs, requestContext)
 				}
 			] }),
-			new PrefillErrorHandler()
-		]
+			new PrefillErrorHandler(),
+			new AccountRotationProcessor({
+				credentialStore: authStorage,
+				maxProcessorRetries: MASTRACODE_MAX_PROCESSOR_RETRIES,
+				settingsPath: config?.settingsPath
+			})
+		],
+		maxProcessorRetries: MASTRACODE_MAX_PROCESSOR_RETRIES
 	});
 	const defaultModes = [
 		{
@@ -633,10 +707,17 @@ async function createMastraCodeAgentController(config) {
 	for (const [key, modelId] of Object.entries(globalSettings.models.subagentModels)) if (key === "default" || key === "_default") globalInitialState.subagentModelId = modelId;
 	else globalInitialState[`subagentModelId_${key}`] = modelId;
 	const typedStateSchema = stateSchema;
-	const controller = new AgentController({
+	const backgroundCompletionEvents = backgroundToolsEnabled ? createBackgroundCompletionEvents() : void 0;
+	let controller;
+	controller = new AgentController({
 		id: "mastra-code",
 		resourceId: project.resourceId,
 		storage,
+		...backgroundToolsEnabled ? { backgroundTasks: {
+			enabled: true,
+			recoverStaleTasksOnStart: false,
+			...createBackgroundCompletionCallbacks(() => controller, backgroundCompletionEvents)
+		} } : {},
 		observability,
 		memory,
 		pubsub: signalsPubSub,
@@ -644,7 +725,10 @@ async function createMastraCodeAgentController(config) {
 		agent: codeAgent,
 		subagents,
 		gateways: [amazonBedrockGateway, mastraCodeGateway],
-		workspace: config?.workspace ?? ((args) => getDynamicWorkspace(args)),
+		workspace: config?.workspace ?? ((args) => getDynamicWorkspace({
+			...args,
+			backgroundToolsEnabled
+		})),
 		browser: config?.browser,
 		idGenerator: config?.idGenerator,
 		toolCategoryResolver: getToolCategory,
@@ -680,32 +764,14 @@ async function createMastraCodeAgentController(config) {
 	const sessionPeerCleanup = /* @__PURE__ */ new WeakMap();
 	if (useCrossAgentSignals) {
 		controller.onSessionCreated(async (session) => {
-			const threadOwnership = createThreadOwnershipManager((threadId) => controller.getCurrentAgent(session).claimThreadOwnership({
-				threadId,
-				resourceId: session.identity.getResourceId(),
-				streamOptions: () => session.machinery.buildStreamOptions({}),
-				peer: {
-					label: `${project.name} (${threadId})`,
-					title: project.name
-				}
-			}));
-			const claimThreadOwnership = async (threadId) => {
-				try {
-					await threadOwnership.claim(threadId);
-				} catch (error) {
-					console.error(`Failed to claim cross-agent thread ownership for ${threadId}`, error);
-				}
-			};
-			const unsubscribeSession = session.subscribe((event) => {
-				if (event.type === "thread_changed") claimThreadOwnership(event.threadId);
-				else if (event.type === "thread_created") claimThreadOwnership(event.thread.id);
-			});
-			sessionPeerCleanup.set(session, () => {
-				unsubscribeSession();
-				threadOwnership.close();
+			const advertisement = createSessionThreadAdvertisement({
+				session,
+				controller,
+				projectName: project.name
 			});
+			sessionPeerCleanup.set(session, () => advertisement.close());
 			const initialThreadId = session.thread.getId();
-			if (initialThreadId) await Promise.race([claimThreadOwnership(initialThreadId), new Promise((resolve) => {
+			if (initialThreadId) await Promise.race([advertisement.claim(initialThreadId), new Promise((resolve) => {
 				setTimeout(resolve, 5e3).unref?.();
 			})]);
 		}, { blocking: true });
@@ -743,6 +809,8 @@ async function createMastraCodeAgentController(config) {
 		builtinOmPacks,
 		effectiveDefaults,
 		githubSignals,
+		backgroundToolsEnabled,
+		backgroundCompletionEvents,
 		sessionId,
 		ownerId,
 		projectPath: project.rootPath,
@@ -916,7 +984,7 @@ async function mountAgentControllerOnMastra(config) {
 */
 async function prepareAgentControllerMount(config) {
 	const base = await createMastraCodeAgentController(config);
-	const { controller, storage, authStorage, projectPath, codeAgent, mcpManager } = base;
+	const { controller, storage, authStorage, projectPath, codeAgent, mcpManager, backgroundToolsEnabled, backgroundCompletionEvents } = base;
 	const controllerId = config?.controllerId ?? controller.id;
 	const apiRoutes = config?.buildApiRoutes?.({
 		controller,
@@ -934,6 +1002,11 @@ async function prepareAgentControllerMount(config) {
 	const mastraArgs = {
 		agentControllers: { [controllerId]: controller },
 		storage,
+		...backgroundToolsEnabled ? { backgroundTasks: {
+			enabled: true,
+			recoverStaleTasksOnStart: false,
+			...createBackgroundCompletionCallbacks(() => controller, backgroundCompletionEvents)
+		} } : {},
 		...base.signalsPubSub ? { pubsub: base.signalsPubSub } : {},
 		...Object.keys(serverConfig).length ? { server: serverConfig } : {}
 	};