@mastra/server 1.78.0 → 1.78.1-alpha.0

package/dist/{dist-CPHBJa2G.js → dist-CdMzvOLk.js}

@@ -21,8 +21,8 @@ import { InMemoryStore, MAX_KNOWLEDGE_NODE_DESCRIPTION_LENGTH, MAX_KNOWLEDGE_REC
 import { coreFeatures } from "@mastra/core/features";
 import { EntityType, SpanType, createObservabilityContext, getOrCreateSpan } from "@mastra/core/observability";
 import { MessageHistory, TokenLimiterProcessor } from "@mastra/core/processors";
-import { createStep, createWorkflow } from "@mastra/core/workflows";
 import { appendFileSync, existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { createStep, createWorkflow } from "@mastra/core/workflows";
 import { tmpdir } from "os";
 import { basename, dirname, extname, isAbsolute, join, relative, resolve } from "path";
 import { createHash } from "crypto";
@@ -4637,7 +4637,7 @@ function splitLines(text) {
 	return result;
 }
 //#endregion
-//#region ../memory/dist/src-CTyZJ6A-.js
+//#region ../memory/dist/src-D9Uu93Xu.js
 var __defProp$3 = Object.defineProperty;
 var __exportAll = (all, no_symbols) => {
 	let target = {};
@@ -30989,7 +30989,8 @@ function hasGoogleGenerativeAIKey() {
 /**
 * Resolve the concrete model ID `'auto'` selects for a main model ID.
 *
-* Order: Gemini when `GOOGLE_GENERATIVE_AI_API_KEY` is set; else the main provider's low-cost
+* Order: Gemini when Google is usable (`canUseProvider('google')`, or `GOOGLE_GENERATIVE_AI_API_KEY`
+* when no check is given); else the main provider's low-cost
 * model (keeping a `mastra/` gateway prefix); else the main model ID itself. Returns `undefined`
 * only when there is no Google key and no main model ID.
 */
@@ -30998,7 +30999,7 @@ function resolveAutoModelId(mainModelId, options) {
 		...AUTO_MODEL_BY_PROVIDER,
 		...options?.autoModels
 	};
-	if (hasGoogleGenerativeAIKey() && models.google) return models.google;
+	if ((options?.canUseProvider ? options.canUseProvider("google") : hasGoogleGenerativeAIKey()) && models.google) return models.google;
 	if (!mainModelId) return;
 	const prefix = mainModelId.startsWith(ROUTE_PREFIX) ? ROUTE_PREFIX : "";
 	const provider = mainModelId.slice(prefix.length).split("/", 1)[0];
@@ -35635,6 +35636,7 @@ var ObservationalMemory = class ObservationalMemory {
 	mastra;
 	autoModels;
 	resolveAutoModelHook;
+	canUseProvider;
 	memory;
 	/**
 	* Track message IDs observed during this instance's lifetime.
@@ -35733,6 +35735,7 @@ var ObservationalMemory = class ObservationalMemory {
 		this.hooks = config.hooks;
 		this.autoModels = config.autoModels;
 		this.resolveAutoModelHook = config.resolveModel;
+		this.canUseProvider = config.canUseProvider;
 		this.hookExecution = config.hookExecution ?? "non-blocking";
 		this.mastra = config.mastra;
 		this.memory = config.memory;
@@ -35932,11 +35935,18 @@ var ObservationalMemory = class ObservationalMemory {
 			model: await this.resolveAutoModel(options)
 		};
 	}
+	autoModelOptions(requestContext) {
+		const canUseProvider = this.canUseProvider;
+		return {
+			autoModels: this.autoModels,
+			canUseProvider: canUseProvider ? (providerId) => canUseProvider(providerId, { requestContext }) : void 0
+		};
+	}
 	async resolveAutoModel(options) {
 		let mainModel = options?.currentModel?.model;
 		if (!mainModel && options?.mainAgent) mainModel = await options.mainAgent.getModel({ requestContext: options.requestContext });
 		const mainModelId = getModelId(mainModel);
-		const pick = resolveAutoModelId(mainModelId, { autoModels: this.autoModels }) ?? (mainModel ? void 0 : OBSERVATIONAL_MEMORY_DEFAULTS.observation.model);
+		const pick = resolveAutoModelId(mainModelId, this.autoModelOptions(options?.requestContext)) ?? (mainModel ? void 0 : OBSERVATIONAL_MEMORY_DEFAULTS.observation.model);
 		if (mainModel && (!pick || pick === mainModelId)) return mainModel;
 		const modelId = pick;
 		return this.resolveAutoModelHook ? await this.resolveAutoModelHook(modelId, { requestContext: options?.requestContext }) : modelId;
@@ -36003,7 +36013,7 @@ var ObservationalMemory = class ObservationalMemory {
 	async getCompressionStartLevel(requestContext) {
 		try {
 			const reflectionModel = await this.resolveModelFunction(this.reflectionConfig.model, requestContext);
-			if ((reflectionModel === "auto" ? resolveAutoModelId(void 0, { autoModels: this.autoModels }) ?? "" : (await this.resolveModelContext(reflectionModel, requestContext))?.modelId ?? "").includes("gemini-2.5-flash")) return 2;
+			if ((reflectionModel === "auto" ? resolveAutoModelId(void 0, this.autoModelOptions(requestContext)) ?? "" : (await this.resolveModelContext(reflectionModel, requestContext))?.modelId ?? "").includes("gemini-2.5-flash")) return 2;
 			return 1;
 		} catch {
 			return 1;
@@ -39963,6 +39973,7 @@ ${workingMemory}`;
 			shareTokenBudget: omConfig.shareTokenBudget,
 			model: omConfig.model,
 			autoModels: omConfig.autoModels,
+			canUseProvider: omConfig.canUseProvider,
 			resolveModel: omConfig.resolveModel,
 			mastra: this._mastraInstance,
 			onIndexObservations,
@@ -51896,4 +51907,4 @@ const agentBuilderWorkflows = {
 //#endregion
 export { agentBuilderWorkflows };
 
-//# sourceMappingURL=dist-CPHBJa2G.js.map
+//# sourceMappingURL=dist-CdMzvOLk.js.map