@mastra/factory 0.21.0-alpha.2 → 0.21.0-alpha.4

package/dist/routes/config.js

@@ -1,14 +1,14 @@
 import { WEB_OAUTH_FLOW_KINDS, getAuthProviderId, listTenantCredentialsForRequest, resolveCredentialContext, tenantOrgId } from "./provider-credentials.js";
+import "../session/memory-settings-hydration.js";
 import { factoryMemorySettingsUserId } from "../storage/domains/memory-settings/base.js";
-import { applyStoredMemorySettings } from "../session/memory-settings-hydration.js";
-import { seedPersonalOmDefaults } from "./om-seed.js";
 import { Route } from "./route.js";
 import { registerApiRoute } from "@mastra/core/server";
 import { z } from "zod";
-import { DEFAULT_OM_MODEL_ID } from "@mastra/code-sdk/constants";
-import { resolveProviderOMDefault } from "@mastra/code-sdk/onboarding/packs";
-import { THINKING_LEVEL_VALUES, getCustomProviderId, isThinkingLevelSetting, loadSettings, saveSettings } from "@mastra/code-sdk/onboarding/settings";
+import { MastraCodeGateway } from "@mastra/code-sdk/agents/mastracode-gateway";
+import { resolveAutoOMModelId } from "@mastra/code-sdk/onboarding/packs";
+import { MASTRA_GATEWAY_PROVIDER, THINKING_LEVEL_VALUES, getCustomProviderId, isThinkingLevelSetting, loadSettings, saveSettings } from "@mastra/code-sdk/onboarding/settings";
 import { AMAZON_BEDROCK_GATEWAY_ID } from "@mastra/code-sdk/providers/amazon-bedrock-gateway";
+import { MASTRA_GATEWAY_PREFIX } from "@mastra/code-sdk/providers/model-ids";
 import { getModelReasoningOptions } from "@mastra/core/llm";
 //#region src/routes/config.ts
 /** Widen a route-local Hono context to the plain `Context` the auth helpers take. */
@@ -217,21 +217,26 @@ async function resolveModelDefaultsContext({ c, auth, modelDefaults }) {
 		message: "Default model storage is unavailable — the app database is not configured or failed to start."
 	}, 503) };
 }
-function readOMConfig(session) {
-	const observeAttachments = (session.state.get() ?? {}).observeAttachments;
+function modelProvider(modelId) {
+	return modelId.replace(/^mastracode\//, "").replace(/^mastra\//, "").split("/", 1)[0] ?? "";
+}
+function roleConfig(model, effectiveModelId, autoModelSource, availableProviders) {
 	return {
-		observerModelId: session.om.observer.modelId() ?? "",
-		reflectorModelId: session.om.reflector.modelId() ?? "",
-		observationThreshold: session.om.observer.threshold() ?? 3e4,
-		reflectionThreshold: session.om.reflector.threshold() ?? 4e4,
-		observeAttachments: observeAttachments === true || observeAttachments === false ? observeAttachments : "auto"
+		model,
+		effectiveModelId,
+		effectiveModelSource: model === "auto" ? autoModelSource : "explicit",
+		providerStatus: availableProviders.has(modelProvider(effectiveModelId)) || effectiveModelId.startsWith(MASTRA_GATEWAY_PREFIX) && availableProviders.has(MASTRA_GATEWAY_PROVIDER) ? "available" : "unavailable"
 	};
 }
-function readStoredOMConfig(record, fallbackOmModelId) {
-	const fallback = fallbackOmModelId ?? DEFAULT_OM_MODEL_ID;
+function readStoredOMConfig(record, currentModelId, autoModelSource, availableProviders) {
+	const autoModelId = resolveAutoOMModelId(currentModelId);
+	const observerModelId = record?.observerModelId ?? autoModelId;
+	const reflectorModelId = record?.reflectorModelId ?? autoModelId;
 	return {
-		observerModelId: record?.observerModelId ?? fallback,
-		reflectorModelId: record?.reflectorModelId ?? fallback,
+		observer: roleConfig(record?.observerModelId ?? "auto", observerModelId, autoModelSource, availableProviders),
+		reflector: roleConfig(record?.reflectorModelId ?? "auto", reflectorModelId, autoModelSource, availableProviders),
+		observerModelId,
+		reflectorModelId,
 		observationThreshold: record?.observationThreshold ?? 3e4,
 		reflectionThreshold: record?.reflectionThreshold ?? 4e4,
 		observeAttachments: record?.observeAttachments ?? "auto"
@@ -279,12 +284,11 @@ async function resolveMemorySettingsContext({ c, auth, memorySettings, factoryPr
 	}, 503) };
 }
 /** Persist an OM knob change to the caller's memory-settings row. */
-async function persistMemorySettings(context, patch, fillIfUnset) {
+async function persistMemorySettings(context, patch) {
 	await context.storage.patch({
 		orgId: context.orgId,
 		userId: context.userId,
-		patch,
-		fillIfUnset
+		patch
 	});
 }
 /**
@@ -310,12 +314,34 @@ var ConfigRoutes = class extends Route {
 		const { controller, authStorage, auth, deploymentProviders } = options;
 		const onCredentialsChanged = options.onCredentialsChanged ?? (() => {});
 		const onCustomProvidersChanged = options.onCustomProvidersChanged ?? (() => {});
+		const resolveOMResponseContext = async (c) => {
+			const tenantCredentials = await listTenantCredentialsForRequest({
+				c: loose(c),
+				auth,
+				credentials: options.modelCredentials
+			});
+			const access = await buildProviderAccess({
+				controller,
+				authStorage: tenantCredentials ? void 0 : authStorage,
+				tenantCredentials,
+				deploymentProviders
+			});
+			const availableProviders = new Set(Object.entries(access).flatMap(([providerId, configured]) => configured ? [providerId] : []));
+			if (MastraCodeGateway.getMastraGatewayApiKey()) availableProviders.add(MASTRA_GATEWAY_PROVIDER);
+			if (options.customProviders) {
+				const context = await resolveCustomProvidersContext({
+					c: loose(c),
+					auth,
+					customProviders: options.customProviders
+				});
+				if (!("response" in context)) for (const provider of await context.storage.list({ orgId: context.orgId })) availableProviders.add(provider.providerId);
+			}
+			return availableProviders;
+		};
 		const factoryOmFallback = async (factoryProjectId) => {
-			if (!factoryProjectId || !options.factoryProjects) return void 0;
 			try {
-				const defaultModelId = (await options.factoryProjects.getById({ id: factoryProjectId }))?.defaultModelId ?? void 0;
-				const provider = defaultModelId?.split("/")[0];
-				return provider ? resolveProviderOMDefault(provider, defaultModelId).modelId : void 0;
+				const defaultModelId = factoryProjectId ? await options.factoryProjects?.getById({ id: factoryProjectId }).then((project) => project?.defaultModelId ?? void 0) : controller.listModes?.().find((mode) => mode.defaultModelId)?.defaultModelId;
+				return defaultModelId ? resolveAutoOMModelId(defaultModelId) : void 0;
 			} catch {
 				return;
 			}
@@ -389,11 +415,6 @@ var ConfigRoutes = class extends Route {
 								key
 							});
 							onCredentialsChanged(tenant);
-							await seedPersonalOmDefaults({
-								memorySettings: options.memorySettings,
-								tenant,
-								provider
-							});
 							const records = await ctx.storage.listCredentials(ctx.orgId, ctx.userId);
 							const providers = await listProviders({
 								controller,
@@ -741,59 +762,6 @@ var ConfigRoutes = class extends Route {
 					}
 				}
 			}),
-			registerApiRoute("/web/config/om/provider-defaults", {
-				method: "POST",
-				requiresAuth: false,
-				handler: async (c) => {
-					let body;
-					try {
-						body = await c.req.json();
-					} catch {
-						return c.json({ error: "Invalid JSON body" }, 400);
-					}
-					const providerId = typeof body.providerId === "string" ? body.providerId.trim() : "";
-					const factoryModelId = typeof body.factoryModelId === "string" ? body.factoryModelId.trim() : "";
-					const factoryProjectId = typeof body.factoryId === "string" && body.factoryId ? body.factoryId : void 0;
-					if (!providerId) return c.json({ error: "Missing required field: providerId" }, 400);
-					const context = await resolveMemorySettingsContext({
-						c: loose(c),
-						auth,
-						memorySettings: options.memorySettings,
-						factoryProjectId,
-						factoryProjects: options.factoryProjects
-					});
-					if ("response" in context) return context.response;
-					try {
-						const tenantCredentials = await listTenantCredentialsForRequest({
-							c: loose(c),
-							auth,
-							credentials: options.modelCredentials
-						});
-						if (!(await buildProviderAccess({
-							controller,
-							authStorage: tenantCredentials ? void 0 : authStorage,
-							tenantCredentials,
-							deploymentProviders
-						}))[providerId]) return c.json({ error: `Provider "${providerId}" is not configured` }, 400);
-						const modelId = resolveProviderOMDefault(providerId, factoryModelId).modelId;
-						const record = await context.storage.patch({
-							orgId: context.orgId,
-							userId: context.userId,
-							patch: {},
-							fillIfUnset: {
-								observerModelId: modelId,
-								reflectorModelId: modelId
-							}
-						});
-						return c.json({
-							ok: true,
-							config: readStoredOMConfig(record)
-						});
-					} catch (error) {
-						return c.json({ error: error instanceof Error ? error.message : String(error) }, 500);
-					}
-				}
-			}),
 			registerApiRoute("/web/config/om", {
 				method: "GET",
 				requiresAuth: false,
@@ -815,11 +783,9 @@ var ConfigRoutes = class extends Route {
 							userId: context.userId
 						});
 						const fallback = await factoryOmFallback(factoryProjectId);
-						if (!resourceId) return c.json({ config: readStoredOMConfig(record, fallback) });
-						const session = await controller.getSessionByResource?.(resourceId, scope);
-						if (!session) return c.json({ config: readStoredOMConfig(record, fallback) });
-						await applyStoredMemorySettings(session, record, fallback);
-						return c.json({ config: readOMConfig(session) });
+						const availableProviders = await resolveOMResponseContext(loose(c));
+						const sessionModelId = (resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0)?.model.get();
+						return c.json({ config: readStoredOMConfig(record, sessionModelId || fallback, sessionModelId ? "live-session" : "configured-default", availableProviders) });
 					} catch (error) {
 						return c.json({ error: error instanceof Error ? error.message : String(error) }, 500);
 					}
@@ -840,8 +806,9 @@ var ConfigRoutes = class extends Route {
 					const resourceId = typeof body.resourceId === "string" ? body.resourceId : "";
 					const scope = typeof body.scope === "string" && body.scope ? body.scope : void 0;
 					const factoryProjectId = typeof body.factoryId === "string" && body.factoryId ? body.factoryId : void 0;
-					const modelId = typeof body.modelId === "string" ? body.modelId.trim() : "";
-					if (!modelId) return c.json({ error: "Missing required field: modelId" }, 400);
+					const model = typeof body.modelId === "string" ? body.modelId.trim() : "";
+					if (!model) return c.json({ error: "Missing required field: modelId" }, 400);
+					const wantsAuto = model === "auto";
 					const context = await resolveMemorySettingsContext({
 						c: loose(c),
 						auth,
@@ -850,16 +817,15 @@ var ConfigRoutes = class extends Route {
 						factoryProjects: options.factoryProjects
 					});
 					if ("response" in context) return context.response;
+					const availableProviders = await resolveOMResponseContext(loose(c));
 					try {
-						const session = resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0;
-						const otherRoleCurrentModelId = (session ? role === "observer" ? session.om.reflector : session.om.observer : void 0)?.modelId() ?? null;
-						await session?.om[role].switchModel({ modelId });
-						const otherKey = role === "observer" ? "reflectorModelId" : "observerModelId";
-						await persistMemorySettings(context, { [role === "observer" ? "observerModelId" : "reflectorModelId"]: modelId }, otherRoleCurrentModelId ? { [otherKey]: otherRoleCurrentModelId } : void 0);
-						const config = session ? readOMConfig(session) : readStoredOMConfig(await context.storage.get({
+						await persistMemorySettings(context, { [role === "observer" ? "observerModelId" : "reflectorModelId"]: wantsAuto ? null : model });
+						const record = await context.storage.get({
 							orgId: context.orgId,
 							userId: context.userId
-						}), await factoryOmFallback(factoryProjectId));
+						});
+						const sessionModelId = (resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0)?.model.get();
+						const config = readStoredOMConfig(record, sessionModelId || await factoryOmFallback(factoryProjectId), sessionModelId ? "live-session" : "configured-default", availableProviders);
 						return c.json({
 							ok: true,
 							config
@@ -893,30 +859,18 @@ var ConfigRoutes = class extends Route {
 						factoryProjects: options.factoryProjects
 					});
 					if ("response" in context) return context.response;
+					const availableProviders = await resolveOMResponseContext(loose(c));
 					try {
-						const session = resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0;
-						if (observation !== void 0 && session) {
-							await session.state.set({ observationThreshold: observation });
-							await session.thread.setSetting({
-								key: "observationThreshold",
-								value: observation
-							});
-						}
-						if (reflection !== void 0 && session) {
-							await session.state.set({ reflectionThreshold: reflection });
-							await session.thread.setSetting({
-								key: "reflectionThreshold",
-								value: reflection
-							});
-						}
 						await persistMemorySettings(context, {
 							...observation !== void 0 ? { observationThreshold: observation } : {},
 							...reflection !== void 0 ? { reflectionThreshold: reflection } : {}
 						});
-						const config = session ? readOMConfig(session) : readStoredOMConfig(await context.storage.get({
+						const record = await context.storage.get({
 							orgId: context.orgId,
 							userId: context.userId
-						}), await factoryOmFallback(factoryProjectId));
+						});
+						const sessionModelId = (resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0)?.model.get();
+						const config = readStoredOMConfig(record, sessionModelId || await factoryOmFallback(factoryProjectId), sessionModelId ? "live-session" : "configured-default", availableProviders);
 						return c.json({
 							ok: true,
 							config
@@ -950,20 +904,15 @@ var ConfigRoutes = class extends Route {
 						factoryProjects: options.factoryProjects
 					});
 					if ("response" in context) return context.response;
+					const availableProviders = await resolveOMResponseContext(loose(c));
 					try {
-						const session = resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0;
-						if (session) {
-							await session.state.set({ observeAttachments: value });
-							await session.thread.setSetting({
-								key: "observeAttachments",
-								value
-							});
-						}
 						await persistMemorySettings(context, { observeAttachments: value });
-						const config = session ? readOMConfig(session) : readStoredOMConfig(await context.storage.get({
+						const record = await context.storage.get({
 							orgId: context.orgId,
 							userId: context.userId
-						}), await factoryOmFallback(factoryProjectId));
+						});
+						const sessionModelId = (resourceId ? await controller.getSessionByResource?.(resourceId, scope) : void 0)?.model.get();
+						const config = readStoredOMConfig(record, sessionModelId || await factoryOmFallback(factoryProjectId), sessionModelId ? "live-session" : "configured-default", availableProviders);
 						return c.json({
 							ok: true,
 							config
@@ -977,6 +926,6 @@ var ConfigRoutes = class extends Route {
 	}
 };
 //#endregion
-export { ConfigRoutes, DEPLOYMENT_MODEL_PROVIDERS, buildProviderAccess, isDeploymentModelProvider, listProviders, readOMConfig, resolveDeploymentModelProviders };
+export { ConfigRoutes, DEPLOYMENT_MODEL_PROVIDERS, buildProviderAccess, isDeploymentModelProvider, listProviders, resolveDeploymentModelProviders };
 
 //# sourceMappingURL=config.js.map