@mastra/memory 1.38.0 → 1.39.0-alpha.0

package/dist/{src-CTyZJ6A-.js → src-D9Uu93Xu.js}

@@ -26517,7 +26517,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.
 */
@@ -26526,7 +26527,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];
@@ -31207,6 +31208,7 @@ var ObservationalMemory = class ObservationalMemory {
 	mastra;
 	autoModels;
 	resolveAutoModelHook;
+	canUseProvider;
 	memory;
 	/**
 	* Track message IDs observed during this instance's lifetime.
@@ -31305,6 +31307,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;
@@ -31504,11 +31507,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;
@@ -31575,7 +31585,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;
@@ -35562,6 +35572,7 @@ ${workingMemory}`;
 			shareTokenBudget: omConfig.shareTokenBudget,
 			model: omConfig.model,
 			autoModels: omConfig.autoModels,
+			canUseProvider: omConfig.canUseProvider,
 			resolveModel: omConfig.resolveModel,
 			mastra: this._mastraInstance,
 			onIndexObservations,
@@ -36717,4 +36728,4 @@ Notes:
 //#endregion
 export { renderSubconsciousActivity as $, buildObserverSystemPrompt as A, OBSERVATION_CONTEXT_INSTRUCTIONS as B, WorkingMemoryExtractor as C, TokenCounter as D, summarizeConversation as E, parseObserverOutput as F, Subconscious as G, OBSERVATION_CONTEXT_PROMPT_THREAD as H, injectAnchorIds as I, SubconsciousRemindExtractor as J, KnowledgeSemanticIndexCoordinator as K, parseAnchorId as L, formatMessagesForObserver as M, hasCurrentTaskSection as N, OBSERVER_SYSTEM_PROMPT as O, optimizeObservationsForContext as P, publishSubconsciousActivity as Q, stripEphemeralAnchorIds as R, wrapInObservationGroup as S, SUMMARIZE_THREAD_DEFAULTS as T, OBSERVATION_CONTINUATION_HINT as U, OBSERVATION_CONTEXT_PROMPT as V, getObservationContextPrompt as W, SUBCONSCIOUS_ACTIVITY_STATE_ID as X, ModelByInputTokens as Y, buildSubconsciousActivitySnapshot as Z, deriveObservationGroupProvenance as _, extractWorkingMemoryContent as a, renderObservationGroupsForReflection as b, getObservationsAsOf as c, AUTO_MODEL_BY_PROVIDER as d, Extractor as et, resolveAutoModelId as f, combineObservationGroupRanges as g, WorkingMemoryStateProcessor as h, WorkingMemory as i, extractCurrentTask as j, buildObserverPrompt as k, ObservationalMemoryProcessor as l, WORKING_MEMORY_STATE_PROCESSOR_ID as m, MessageHistory$1 as n, extractWorkingMemoryTags as o, WORKING_MEMORY_STATE_ID as p, StaleKnowledgeSemanticIndexError as q, SemanticRecall as r, removeWorkingMemoryTags as s, Memory as t, ObservationalMemory as u, parseObservationGroups as v, deepMergeWorkingMemory as w, stripObservationGroups as x, reconcileObservationGroupsFromReflection as y, OBSERVATIONAL_MEMORY_DEFAULTS as z };
 
-//# sourceMappingURL=src-CTyZJ6A-.js.map
+//# sourceMappingURL=src-D9Uu93Xu.js.map