@mastra/code-sdk 1.15.0-alpha.1 → 1.15.0-alpha.11

package/dist/index.js

@@ -11,7 +11,7 @@ import { createSessionThreadAdvertisement } from "./agent-connections/session-ad
 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, withCredentialTenantPreparation } from "./agents/credential-resolver.js";
+import { hasCredentialStoreProvider, setCredentialLogger, withCredentialTenantPreparation } from "./agents/credential-resolver.js";
 import { getDynamicWorkspace, getGoalJudgeTools } from "./agents/workspace.js";
 import { createGitRefReminderReader, instructionSourcePaths, loadSessionInstructions, sessionInstructionRoots } from "./agents/prompts/agent-instructions.js";
 import { getDynamicInstructions } from "./agents/instructions.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";
@@ -54,12 +54,14 @@ 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 { openLocalObservabilityStore } from "./utils/local-observability.js";
 import { registerSessionAndWaitForMaintenance, unregisterSession } from "./utils/maintenance-lock.js";
 import { createSignalsPubSub } from "./utils/signals-pubsub.js";
 import { createResourceNotificationDispatcher, shouldHoldNotificationDelivery } from "./utils/notification-dispatch.js";
 import { DEFAULT_RETENTION, createStorageMaintenance, resolveLocalDbFiles } from "./utils/storage-maintenance.js";
 import { createStorage, createVectorStore } from "./utils/storage-factory.js";
 import { acquireThreadLock, releaseThreadLock } from "./utils/thread-lock.js";
+import { FOREIGN_THREAD, findWakeSession, sessionLeaseScopes } from "./utils/wake-session.js";
 import { registerWorkflowBuilderPrimitives } from "./workflows/register-primitives.js";
 import { autoApprovePolicy, denyPolicy, permissionModeToPolicy } from "./headless/policy.js";
 import { runMC } from "./headless/run-mc.js";
@@ -82,7 +84,6 @@ import { RequestContext } from "@mastra/core/request-context";
 import { TaskSignalProvider } from "@mastra/core/signals";
 import { MastraCompositeStore } from "@mastra/core/storage";
 import { DEFAULT_GOAL_JUDGE_PROMPT } from "@mastra/core/tools";
-import { DuckDBStore } from "@mastra/duckdb";
 import { GithubSignals } from "@mastra/github-signals";
 import { LibSQLStore, LibSQLVector } from "@mastra/libsql";
 import { MastraPlatformExporter, MastraStorageExporter, Observability, SensitiveDataFilter } from "@mastra/observability";
@@ -246,6 +247,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 {
@@ -322,11 +327,7 @@ async function createMastraCodeAgentController(config) {
 	let observabilityDomain;
 	let observabilityWarning;
 	if (globalSettings.observability.localTracing) try {
-		const observabilityDuckDB = new DuckDBStore({
-			id: "mastra-code-observability",
-			path: getObservabilityDatabasePath()
-		});
-		await observabilityDuckDB.db.getConnection();
+		const { store: observabilityDuckDB } = await openLocalObservabilityStore(getObservabilityDatabasePath());
 		observabilityDomain = observabilityDuckDB.observability;
 	} catch (err) {
 		const message = err instanceof Error ? err.message : String(err);
@@ -415,16 +416,35 @@ async function createMastraCodeAgentController(config) {
 	const efficiencyScorer = createEfficiencyScorer();
 	const ownsNotificationDispatch = !configuredPubSub;
 	const dispatchLeasePubSub = ownsNotificationDispatch && !ownSignalsPubSub && process.platform !== "win32" ? createSignalsPubSub(project.resourceId) : void 0;
+	const notificationDispatchLeases = ownsNotificationDispatch ? (ownSignalsPubSub ?? dispatchLeasePubSub)?.getLeaseProvider() : void 0;
 	const notificationDispatcher = createResourceNotificationDispatcher({
 		getMastra: () => controller.getMastra(),
-		getResourceIds: () => [...liveSessions].map((session) => session.identity.getResourceId()),
-		leases: ownsNotificationDispatch ? (ownSignalsPubSub ?? dispatchLeasePubSub)?.getLeaseProvider() : void 0,
+		getSessionScopes: async () => (await Promise.all([...liveSessions].map(async (session) => {
+			const resourceId = session.identity.getResourceId();
+			return (await sessionLeaseScopes(session)).map((scope) => ({
+				resourceId,
+				scope
+			}));
+		}))).flat(),
+		leases: notificationDispatchLeases,
 		owner: `${ownerId}:${process.pid}:${randomUUID()}`,
 		onError: (error) => console.warn("Notification dispatch failed:", error)
 	});
-	const getWakeStreamOptions = async ({ resourceId, threadId }) => {
-		const session = await controller.getSessionByResource(resourceId);
-		if (!session) return void 0;
+	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,
+			threadId
+		});
+		if (!wake || wake === FOREIGN_THREAD) return void 0;
+		const { session } = wake;
 		const targetThread = await session.thread.getById({ threadId });
 		const metadata = targetThread?.resourceId === resourceId ? targetThread.metadata : void 0;
 		const modes = controller.listModes();
@@ -493,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(),
@@ -546,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",
@@ -557,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})`);
@@ -692,10 +718,20 @@ async function createMastraCodeAgentController(config) {
 		notifications: { deliveryPolicy: { decide: async (input) => {
 			const decision = defaultNotificationDeliveryDecision(input);
 			if (!input.record.resourceId) return decision;
-			const streamOptions = await getWakeStreamOptions({
+			const target = {
 				resourceId: input.record.resourceId,
 				threadId: input.record.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
+			};
+			const streamOptions = owner ? await getWakeStreamOptions({
+				resourceId: input.record.resourceId,
+				threadId: input.record.threadId,
+				owner
+			}) : void 0;
 			const withOptions = streamOptions ? {
 				...decision,
 				streamOptions
@@ -703,7 +739,8 @@ async function createMastraCodeAgentController(config) {
 			if (!ownsNotificationDispatch) return withOptions;
 			return shouldHoldNotificationDelivery({
 				resourceId: input.record.resourceId,
-				ownedHere: streamOptions !== void 0,
+				scope: owner?.scope,
+				ownedHere: owner !== void 0,
 				dispatcher: notificationDispatcher
 			}) ? {
 				...withOptions,
@@ -749,11 +786,17 @@ async function createMastraCodeAgentController(config) {
 			const configured = config.inputProcessors({ requestContext });
 			return configured instanceof Promise ? configured.then(resolveProcessors) : resolveProcessors(configured);
 		},
-		outputProcessors: () => [
-			new CyberRefusalHandler(),
-			...readPluginProcessors().output.map((entry) => entry.value),
-			...pluginSignalLane?.getOutputProcessors() ?? []
-		],
+		outputProcessors: ({ requestContext }) => {
+			const resolveProcessors = (configured) => [
+				new CyberRefusalHandler(),
+				...configured,
+				...readPluginProcessors().output.map((entry) => entry.value),
+				...pluginSignalLane?.getOutputProcessors() ?? []
+			];
+			if (typeof config?.outputProcessors !== "function") return resolveProcessors(config?.outputProcessors ?? []);
+			const configured = config.outputProcessors({ requestContext });
+			return configured instanceof Promise ? configured.then(resolveProcessors) : resolveProcessors(configured);
+		},
 		errorProcessors: [
 			new ProviderHistoryCompat(),
 			new CyberRefusalHandler(),
@@ -903,7 +946,7 @@ async function createMastraCodeAgentController(config) {
 		...backgroundToolsEnabled ? { backgroundTasks: {
 			enabled: true,
 			recoverStaleTasksOnStart: false,
-			...createBackgroundCompletionCallbacks(() => controller, backgroundCompletionEvents)
+			...createBackgroundCompletionCallbacks(() => sessionRouter, backgroundCompletionEvents)
 		} } : {},
 		observability,
 		memory,
@@ -914,7 +957,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,
@@ -1281,6 +1324,7 @@ async function prepareAgentControllerMount(config) {
 	const finalize = async () => {
 		await controller.init();
 		validateExperimentalAgent(codeAgent, controller.getMastra());
+		setCredentialLogger(controller.getMastra()?.getLogger());
 		if (weOwnTheMastra) {
 			const mastra = controller.getMastra();
 			if (mastra) await registerWorkflowBuilderPrimitives(mastra, {