@mastra/code-sdk 1.15.0-alpha.9 → 1.15.0

package/dist/index.js

@@ -21,7 +21,7 @@ import { setAuthStorage as setAuthStorage$1 } from "./providers/claude-max.js";
 import { setAuthStorage as setAuthStorage$2 } from "./providers/github-copilot.js";
 import { setAuthStorage as setAuthStorage$3 } from "./providers/kimi-coding.js";
 import { setAuthStorage as setAuthStorage$4 } from "./providers/xai.js";
-import { createMastraCodeGateway, getDynamicModel, getGoalJudgeModel, resolveModel } from "./agents/model.js";
+import { canUseProvider, createMastraCodeGateway, getDynamicModel, getGoalJudgeModel, resolveModel } from "./agents/model.js";
 import { getDynamicMemory, hasSubconsciousTools } from "./agents/memory.js";
 import { buildMode } from "./agents/modes/build.js";
 import { fastMode } from "./agents/modes/explore.js";
@@ -87,7 +87,6 @@ import { DEFAULT_GOAL_JUDGE_PROMPT } from "@mastra/core/tools";
 import { GithubSignals } from "@mastra/github-signals";
 import { LibSQLStore, LibSQLVector } from "@mastra/libsql";
 import { MastraPlatformExporter, MastraStorageExporter, Observability, SensitiveDataFilter } from "@mastra/observability";
-import { PostgresStore } from "@mastra/pg";
 //#region src/index.ts
 const CODE_AGENT_ID = "code-agent";
 const MASTRACODE_TRANSIENT_CONNECTION_MAX_RETRIES = 10;
@@ -228,7 +227,7 @@ function hasAncestorClassNamed(value, className) {
 function resolveInjectedStorageBackend(storage, configuredBackend) {
 	if (configuredBackend) return configuredBackend;
 	if (storage instanceof LibSQLStore || hasAncestorClassNamed(storage, "LibSQLStore")) return "libsql";
-	if (storage instanceof PostgresStore || hasAncestorClassNamed(storage, "PostgresStore")) return "pg";
+	if (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`. */
@@ -247,6 +246,10 @@ async function createMastraCodeAgentController(config) {
 	const modeless = config?.modes === false;
 	let activeSession;
 	const liveSessions = /* @__PURE__ */ new Set();
+	const sessionRouter = {
+		getSession: (sessionId) => controller.getSession(sessionId),
+		listLiveSessions: () => liveSessions
+	};
 	let pluginRuntimeController;
 	if (configDir !== ".mastracode") validateConfigDirName(configDir);
 	if (!config?.disableEnvFile) try {
@@ -280,8 +283,9 @@ async function createMastraCodeAgentController(config) {
 		});
 	}
 	const mgApiKey = process.env["MASTRA_GATEWAY_API_KEY"] ?? storedGatewayKey;
+	const mastraGatewayBaseUrl = (process.env["MASTRA_GATEWAY_URL"] ?? storedGatewayUrl ?? "https://gateway-api.mastra.ai").replace(/\/+$/, "").replace(/\/v1$/, "");
 	const mastraCodeGateway = createMastraCodeGateway({
-		mastraGatewayBaseUrl: (process.env["MASTRA_GATEWAY_URL"] ?? storedGatewayUrl ?? "https://gateway-api.mastra.ai").replace(/\/+$/, "").replace(/\/v1$/, ""),
+		mastraGatewayBaseUrl,
 		mastraGatewayApiKey: mgApiKey,
 		routeThroughMastraGateway: false,
 		settingsPath: config?.settingsPath
@@ -426,6 +430,14 @@ async function createMastraCodeAgentController(config) {
 		owner: `${ownerId}:${process.pid}:${randomUUID()}`,
 		onError: (error) => console.warn("Notification dispatch failed:", error)
 	});
+	const resolveWakeOwner = async (target, leases) => {
+		const owner = await findWakeSession(controller, liveSessions, target, leases);
+		if (owner || !config?.resumeWakeSession) return owner;
+		await config.resumeWakeSession(target);
+		const resumed = await findWakeSession(controller, liveSessions, target, leases);
+		if (!resumed) throw new Error(`No session owns resource ${target.resourceId} to run its wake on thread ${target.threadId}.`);
+		return resumed;
+	};
 	const getWakeStreamOptions = async ({ resourceId, threadId, owner }) => {
 		const wake = owner ?? await findWakeSession(controller, liveSessions, {
 			resourceId,
@@ -501,6 +513,7 @@ async function createMastraCodeAgentController(config) {
 			isThreadActive: () => session.thread.getId() === threadId,
 			threadId,
 			resourceId,
+			sessionId: session.identity.getId(),
 			session: {
 				id: session.identity.getId(),
 				ownerId: session.identity.getOwnerId(),
@@ -554,7 +567,11 @@ async function createMastraCodeAgentController(config) {
 				resourceId: schedule.resourceId,
 				threadId: schedule.threadId
 			};
-			const streamOptions = await getWakeStreamOptions(target);
+			const owner = await resolveWakeOwner(target);
+			const streamOptions = owner && owner !== FOREIGN_THREAD ? await getWakeStreamOptions({
+				...target,
+				owner
+			}) : void 0;
 			const accepted = await codeAgent.sendSignal({
 				type: "user",
 				tagName: "user",
@@ -565,7 +582,8 @@ async function createMastraCodeAgentController(config) {
 				ifActive: { behavior: "deliver" },
 				ifIdle: {
 					behavior: "wake",
-					...streamOptions ? { streamOptions } : {}
+					...streamOptions ? { streamOptions } : {},
+					...owner === FOREIGN_THREAD ? { requireClaimedOwner: true } : {}
 				}
 			}).accepted;
 			if (accepted.action === "blocked" || accepted.action === "discard") throw new Error(`the agent did not accept the prompt (${accepted.action})`);
@@ -628,12 +646,13 @@ async function createMastraCodeAgentController(config) {
 				const projectPath = state?.projectPath ?? project.rootPath;
 				const untrusted = state?.untrustedCheckout === true && typeof state?.baseRef === "string";
 				const roots = sessionInstructionRoots(projectPath, state?.instructionRoots);
-				return instructionSourcePaths(loadSessionInstructions({
+				const sources = loadSessionInstructions({
 					workingDir: projectPath,
 					instructionRoots: state?.instructionRoots,
 					untrustedCheckout: untrusted,
 					baseRef: untrusted ? state?.baseRef : void 0
-				}), roots);
+				});
+				return instructionSourcePaths(sources, roots);
 			}
 		})
 	];
@@ -675,15 +694,16 @@ async function createMastraCodeAgentController(config) {
 			return NO_PLUGIN_PROCESSORS;
 		}
 	};
-	const codeAgent = wrapExperimentalAgent(createCodingAgent({
+	const baseCodeAgent = createCodingAgent({
 		id: CODE_AGENT_ID,
 		name: "Code Agent",
 		workspace: void 0,
 		instructions: async ({ requestContext }) => {
 			const configured = config?.hostInstructions;
+			const hostInstructions = typeof configured === "function" ? await configured({ requestContext }) : configured;
 			return getDynamicInstructions({
 				requestContext,
-				hostInstructions: typeof configured === "function" ? await configured({ requestContext }) : configured,
+				hostInstructions,
 				coAuthor: config?.coAuthor,
 				hasSubconscious,
 				hasSubagents: subagents.length > 0,
@@ -691,7 +711,8 @@ async function createMastraCodeAgentController(config) {
 			});
 		},
 		model: withCredentialTenantPreparation(async (ctx) => {
-			const hasSessionLevel = isThinkingLevelSetting(ctx.requestContext.get("controller")?.state?.thinkingLevel);
+			const controller = ctx.requestContext.get("controller");
+			const hasSessionLevel = isThinkingLevelSetting(controller?.state?.thinkingLevel);
 			return getDynamicModel(ctx, config?.settingsPath, {
 				modeless,
 				defaultThinkingLevel: hasSessionLevel ? void 0 : await config?.resolveDefaultThinkingLevel?.(ctx)
@@ -704,12 +725,7 @@ async function createMastraCodeAgentController(config) {
 				resourceId: input.record.resourceId,
 				threadId: input.record.threadId
 			};
-			let owner = await findWakeSession(controller, liveSessions, target, notificationDispatchLeases);
-			if (!owner && config?.resumeWakeSession && (decision.action === "deliver" || decision.action === "summarize")) {
-				await config.resumeWakeSession(target);
-				owner = await findWakeSession(controller, liveSessions, target, notificationDispatchLeases);
-				if (!owner) throw new Error(`No session owns resource ${target.resourceId} to run its wake on thread ${target.threadId}.`);
-			}
+			const owner = decision.action === "deliver" || decision.action === "summarize" ? await resolveWakeOwner(target, notificationDispatchLeases) : await findWakeSession(controller, liveSessions, target, notificationDispatchLeases);
 			if (owner === FOREIGN_THREAD) return {
 				...decision,
 				hold: true
@@ -815,7 +831,8 @@ async function createMastraCodeAgentController(config) {
 			})
 		],
 		maxProcessorRetries: MASTRACODE_MAX_PROCESSOR_RETRIES
-	}), experimentalAgent);
+	});
+	const codeAgent = wrapExperimentalAgent(baseCodeAgent, experimentalAgent);
 	const planMode = createPlanMode(!disabledBuiltinTools.has("submit_plan"));
 	const defaultModes = [
 		{
@@ -842,7 +859,7 @@ async function createMastraCodeAgentController(config) {
 	];
 	const intervalHandlers = config?.intervalHandlers ?? [{
 		id: "gateway-sync",
-		intervalMs: 300 * 1e3,
+		intervalMs: 3e5,
 		immediate: false,
 		handler: () => syncGateways()
 	}];
@@ -933,7 +950,7 @@ async function createMastraCodeAgentController(config) {
 		...backgroundToolsEnabled ? { backgroundTasks: {
 			enabled: true,
 			recoverStaleTasksOnStart: false,
-			...createBackgroundCompletionCallbacks(() => controller, backgroundCompletionEvents)
+			...createBackgroundCompletionCallbacks(() => sessionRouter, backgroundCompletionEvents)
 		} } : {},
 		observability,
 		memory,
@@ -944,7 +961,7 @@ async function createMastraCodeAgentController(config) {
 			defaultObserverModelId: DEFAULT_OM_MODEL_ID,
 			reflectorModel: "auto",
 			defaultReflectorModelId: DEFAULT_OM_MODEL_ID,
-			resolveAutoModelId: ({ currentModelId }) => resolveAutoOMModelId(currentModelId)
+			resolveAutoModelId: ({ currentModelId }) => resolveAutoOMModelId(currentModelId, canUseProvider)
 		},
 		agent: codeAgent,
 		subagents,