@mastra/code-sdk 0.0.0-aggregate-trace-20261015182851

package/dist/agents/mastracode-gateway.js

@@ -0,0 +1,434 @@
+import { getAppDataDir } from "../utils/project.js";
+import { AuthStorage } from "../auth/storage.js";
+import { createGoogleThinkingMiddleware } from "../providers/google-thinking.js";
+import { MASTRA_GATEWAY_PREFIX, OPENAI_PREFIX, getCredentialProviderId, normalizeAnthropicModelId, remapOpenAIModelForCodexOAuth, stripMastraGatewayPrefix } from "../providers/model-ids.js";
+import { THINKING_LEVEL_TO_REASONING_EFFORT, buildOpenAICodexOAuthFetch, createCodexMiddleware, createReasoningEffortMiddleware, getEffectiveThinkingLevel, openaiCodexProvider } from "../providers/openai-codex.js";
+import { getCustomProviderId, loadSettings } from "../onboarding/settings.js";
+import { resolveCustomProviders } from "./custom-provider-source.js";
+import { buildAnthropicOAuthFetch, claudeCodeMiddleware, createAnthropicThinkingMiddleware, createPromptCacheMiddleware, opencodeClaudeMaxProvider } from "../providers/claude-max.js";
+import { githubCopilotProvider } from "../providers/github-copilot.js";
+import { kimiCodingProvider } from "../providers/kimi-coding.js";
+import { ModelRouterLanguageModelWithThinking, providerThinkingOptions, withThinkingOptionsModel } from "../providers/provider-thinking.js";
+import { xaiProvider } from "../providers/xai.js";
+import { GATEWAY_AUTH_HEADER, MastraGateway, MastraModelGateway, ModelRouterLanguageModel, PROVIDER_REGISTRY } from "@mastra/core/llm";
+import { createOpenAI } from "@ai-sdk/openai";
+import { wrapLanguageModel } from "ai";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
+//#region src/agents/mastracode-gateway.ts
+const MASTRACODE_GATEWAY_ID = "mastracode";
+/**
+* Global file-backed credential store for local (TUI) model resolution.
+*
+* Constructed lazily, NOT at module scope: `AuthStorage`'s default path is
+* resolved from `MASTRA_APP_DATA_DIR` at construction, and this module is
+* routinely imported before test harnesses / embedders point that variable at
+* an isolated directory — an import-time construction would permanently pin
+* the developer's real app-data dir for every credential read.
+*/
+let authStorageSingleton;
+let authStorageSingletonDir;
+function getGlobalAuthStorage() {
+	const dir = getAppDataDir();
+	if (!authStorageSingleton || authStorageSingletonDir !== dir) {
+		authStorageSingleton = new AuthStorage();
+		authStorageSingletonDir = dir;
+	}
+	return authStorageSingleton;
+}
+function reloadAuthStorage() {
+	getGlobalAuthStorage().reload();
+}
+/**
+* Resolve the Anthropic API key.
+* Main slot → dedicated apikey: slot → env var.
+*/
+function getAnthropicApiKey(credentials = getGlobalAuthStorage()) {
+	const storedCred = credentials.get("anthropic");
+	if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return storedCred.key.trim();
+	const dedicatedKey = credentials.getStoredApiKey("anthropic")?.trim();
+	if (dedicatedKey) return dedicatedKey;
+	return credentials.allowEnvironmentFallback === false ? void 0 : process.env.ANTHROPIC_API_KEY?.trim() || void 0;
+}
+/**
+* Resolve the OpenAI API key.
+* Main slot → dedicated apikey: slot → env var.
+*/
+function getOpenAIApiKey(credentials = getGlobalAuthStorage()) {
+	const storedCred = credentials.get("openai-codex");
+	if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return storedCred.key.trim();
+	const dedicatedKey = credentials.getStoredApiKey("openai-codex")?.trim();
+	if (dedicatedKey) return dedicatedKey;
+	return credentials.allowEnvironmentFallback === false ? void 0 : process.env.OPENAI_API_KEY?.trim() || void 0;
+}
+function anthropicApiKeyProvider(modelId, apiKey, headers, thinkingLevel, promptCacheScope) {
+	const anthropic = createAnthropic({
+		apiKey,
+		headers
+	});
+	const thinkingMiddleware = createAnthropicThinkingMiddleware(modelId, thinkingLevel);
+	return wrapLanguageModel({
+		model: anthropic(modelId),
+		middleware: [createPromptCacheMiddleware(promptCacheScope), ...thinkingMiddleware ? [thinkingMiddleware] : []]
+	});
+}
+function openaiApiKeyProvider(modelId, apiKey, headers, thinkingLevel) {
+	const openai = createOpenAI({
+		apiKey,
+		baseURL: process.env.OPENAI_BASE_URL,
+		headers
+	});
+	const middleware = thinkingLevel && thinkingLevel !== "off" ? createReasoningEffortMiddleware("openai", THINKING_LEVEL_TO_REASONING_EFFORT[getEffectiveThinkingLevel(modelId, thinkingLevel)]) : void 0;
+	return wrapLanguageModel({
+		model: openai.responses(modelId),
+		middleware: middleware ? [middleware] : []
+	});
+}
+function getProviderAuthKey(providerId, credentials = getGlobalAuthStorage()) {
+	const authProviderId = getCredentialProviderId(providerId);
+	const storedCred = credentials.get(authProviderId);
+	if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return storedCred.key.trim();
+	const dedicatedKey = credentials.getStoredApiKey(authProviderId)?.trim();
+	if (dedicatedKey) return dedicatedKey;
+	const catalogKey = authProviderId !== providerId ? credentials.getStoredApiKey(providerId)?.trim() : void 0;
+	if (catalogKey) return catalogKey;
+	if (credentials.allowEnvironmentFallback === false) return void 0;
+	return authProviderId === "kimi-for-coding" ? process.env.KIMI_API_KEY?.trim() || void 0 : void 0;
+}
+function resolveAuth(request, mastraGatewayApiKey) {
+	return MastraCodeGateway.resolveProviderAuth(request, mastraGatewayApiKey);
+}
+function getGatewayProviderKey(gatewayId, providerId) {
+	if (gatewayId === "models.dev") return providerId;
+	return providerId === gatewayId ? gatewayId : `${gatewayId}/${providerId}`;
+}
+function parseGatewayRouterId(routerId, gateway) {
+	const [firstPart = "", secondPart = "", ...restParts] = routerId.split("/");
+	const gatewayId = gateway.id;
+	if (firstPart === gatewayId && secondPart) return {
+		gatewayId,
+		providerId: secondPart,
+		modelId: restParts.join("/")
+	};
+	return {
+		gatewayId,
+		providerId: firstPart,
+		modelId: [secondPart, ...restParts].filter(Boolean).join("/")
+	};
+}
+function hasResolvedAuth(auth) {
+	if (!auth) return false;
+	if (auth.apiKey || auth.bearerToken) return true;
+	return auth.headers ? Object.keys(auth.headers).length > 0 : false;
+}
+async function resolveGatewayProviderAuth(gateway, routerId) {
+	if (!gateway.resolveAuth) return void 0;
+	const parsed = parseGatewayRouterId(routerId, gateway);
+	try {
+		const result = await gateway.resolveAuth({
+			gatewayId: parsed.gatewayId,
+			providerId: parsed.providerId,
+			modelId: parsed.modelId,
+			routerId
+		});
+		return hasResolvedAuth(result) ? result : void 0;
+	} catch {
+		return;
+	}
+}
+async function getMastraCodeProviderConfigs(gateway) {
+	const providers = { ...PROVIDER_REGISTRY };
+	try {
+		const gatewayProviders = await gateway.fetchProviders();
+		for (const [providerId, config] of Object.entries(gatewayProviders)) providers[getGatewayProviderKey(gateway.id, providerId)] = {
+			...config,
+			gateway: gateway.id
+		};
+	} catch (error) {
+		console.warn(`Failed to load providers from gateway ${gateway.id}:`, error);
+	}
+	return providers;
+}
+function getApiKeyEnvVar(providerConfig) {
+	const envVars = providerConfig?.apiKeyEnvVar;
+	return Array.isArray(envVars) ? envVars[0] : envVars;
+}
+var MastraCodeGateway = class MastraCodeGateway extends MastraModelGateway {
+	id = MASTRACODE_GATEWAY_ID;
+	name = "MastraCode Gateway";
+	#mastraGateway;
+	#mastraGatewayBaseUrl;
+	#mastraGatewayApiKey;
+	#routeThroughMastraGateway;
+	#thinkingLevel;
+	#anthropicPromptCacheScope;
+	#customProviders;
+	#settingsPath;
+	#credentials;
+	constructor({ mastraGatewayBaseUrl, mastraGatewayApiKey, routeThroughMastraGateway, thinkingLevel, anthropicPromptCacheScope, customProviders, settingsPath, credentialStore }) {
+		super();
+		this.#mastraGateway = new MastraGateway({ baseUrl: mastraGatewayBaseUrl });
+		this.#mastraGatewayBaseUrl = mastraGatewayBaseUrl;
+		this.#mastraGatewayApiKey = mastraGatewayApiKey;
+		this.#routeThroughMastraGateway = routeThroughMastraGateway;
+		this.#thinkingLevel = thinkingLevel;
+		this.#anthropicPromptCacheScope = anthropicPromptCacheScope ?? "conversation";
+		this.#customProviders = customProviders;
+		this.#settingsPath = settingsPath;
+		this.#credentials = credentialStore ?? getGlobalAuthStorage();
+	}
+	static getMastraGatewayApiKey() {
+		return getGlobalAuthStorage().getStoredApiKey("mastra-gateway") ?? process.env["MASTRA_GATEWAY_API_KEY"];
+	}
+	/** @deprecated Renamed to {@link MastraCodeGateway.getMastraGatewayApiKey}. */
+	static getMemoryGatewayApiKey() {
+		return MastraCodeGateway.getMastraGatewayApiKey();
+	}
+	/**
+	* Claim an unprefixed model id (e.g. a bare `anthropic/...`) when this gateway
+	* can authenticate it via OAuth or a stored/env credential. This lets the
+	* shared gateway router select MastraCode for auth resolution, so the status
+	* bar and `/api-keys` reflect OAuth logins instead of only checking env vars.
+	*/
+	handlesModel(modelId) {
+		const parsed = parseGatewayRouterId(modelId, this);
+		if (!parsed.providerId || !parsed.modelId) return false;
+		if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) return true;
+		if (this.#getCustomProviders().find((provider) => parsed.providerId === getCustomProviderId(provider.name))?.apiKey) return true;
+		return hasResolvedAuth(MastraCodeGateway.resolveProviderAuth({
+			gatewayId: this.id,
+			providerId: parsed.providerId,
+			modelId: parsed.modelId,
+			routerId: modelId
+		}, void 0, this.#credentials));
+	}
+	static resolveProviderAuth(request, mastraGatewayApiKey, credentials = getGlobalAuthStorage()) {
+		if (request.gatewayId === "mastra" && mastraGatewayApiKey) return {
+			apiKey: mastraGatewayApiKey,
+			source: "gateway"
+		};
+		const authProviderId = getCredentialProviderId(request.providerId);
+		if (credentials.get(authProviderId)?.type === "oauth") return {
+			bearerToken: "oauth",
+			source: "gateway"
+		};
+		const apiKey = getProviderAuthKey(request.providerId, credentials);
+		return apiKey ? {
+			apiKey,
+			source: "gateway"
+		} : void 0;
+	}
+	static createModelCatalogProvider(gateway) {
+		return async () => {
+			const registry = await getMastraCodeProviderConfigs(gateway);
+			const models = [];
+			for (const [provider, providerConfig] of Object.entries(registry)) {
+				const apiKeyEnvVar = getApiKeyEnvVar(providerConfig);
+				const hasEnvKey = apiKeyEnvVar ? Boolean(process.env[apiKeyEnvVar]) : false;
+				const modelNames = providerConfig.models;
+				if (!Array.isArray(modelNames)) continue;
+				const gatewayAuth = modelNames[0] ? await resolveGatewayProviderAuth(gateway, `${provider}/${modelNames[0]}`) : void 0;
+				for (const modelName of modelNames) {
+					const id = `${provider}/${modelName}`;
+					models.push({
+						id,
+						provider,
+						modelName,
+						hasApiKey: hasEnvKey || Boolean(gatewayAuth),
+						apiKeyEnvVar: apiKeyEnvVar || void 0
+					});
+				}
+			}
+			return models;
+		};
+	}
+	createModelCatalogProvider() {
+		return MastraCodeGateway.createModelCatalogProvider(this);
+	}
+	#getCustomProviders() {
+		return this.#customProviders ?? resolveCustomProviders() ?? loadSettings(this.#settingsPath).customProviders;
+	}
+	async fetchProviders() {
+		this.#credentials.reload();
+		const providers = {};
+		for (const provider of this.#getCustomProviders()) {
+			const models = provider.models ?? [];
+			if (!models.length) continue;
+			providers[getCustomProviderId(provider.name)] = {
+				name: provider.name,
+				url: provider.url,
+				apiKeyEnvVar: "",
+				apiKeyHeader: "Authorization",
+				gateway: this.id,
+				models
+			};
+		}
+		return providers;
+	}
+	buildUrl(modelId) {
+		return this.#routeThroughMastraGateway ? this.#mastraGateway.buildUrl(modelId) : modelId;
+	}
+	async getApiKey(modelId) {
+		const providerId = stripMastraGatewayPrefix(modelId).split("/", 1)[0];
+		if (this.#routeThroughMastraGateway) return this.#mastraGatewayApiKey ?? "";
+		return providerId ? getProviderAuthKey(providerId, this.#credentials) ?? "" : "";
+	}
+	resolveAuth(request) {
+		if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) return {
+			apiKey: this.#mastraGatewayApiKey,
+			source: "gateway"
+		};
+		const customProvider = this.#getCustomProviders().find((provider) => request.providerId === getCustomProviderId(provider.name));
+		if (customProvider?.apiKey) return {
+			apiKey: customProvider.apiKey,
+			source: "gateway"
+		};
+		return MastraCodeGateway.resolveProviderAuth(request, void 0, this.#credentials);
+	}
+	resolveLanguageModel(args) {
+		const customProvider = this.#getCustomProviders().find((provider) => args.providerId === getCustomProviderId(provider.name));
+		if (customProvider) {
+			const provider = createOpenAICompatible({
+				name: args.providerId,
+				baseURL: customProvider.url,
+				apiKey: args.apiKey,
+				headers: args.headers
+			});
+			const reasoningEffort = this.#thinkingLevel ? THINKING_LEVEL_TO_REASONING_EFFORT[this.#thinkingLevel] : void 0;
+			const middleware = createReasoningEffortMiddleware(args.providerId, reasoningEffort);
+			if (!middleware) return provider.chatModel(args.modelId);
+			return wrapLanguageModel({
+				model: provider.chatModel(args.modelId),
+				middleware: [middleware]
+			});
+		}
+		if (args.providerId === "github-copilot") return githubCopilotProvider(args.modelId, {
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+		if (args.providerId === "moonshotai") {
+			const apiKey = args.apiKey?.trim() || process.env.MOONSHOT_API_KEY?.trim() || process.env.MOONSHOT_AI_API_KEY?.trim();
+			if (!apiKey) throw new Error("Need MOONSHOT_API_KEY");
+			return createAnthropic({
+				apiKey,
+				baseURL: "https://api.moonshot.ai/anthropic/v1",
+				name: "moonshotai.anthropicv1",
+				headers: args.headers
+			})(args.modelId);
+		}
+		if (args.providerId === "anthropic") return this.#resolveAnthropicModel(args);
+		if (args.providerId === "openai") {
+			const openaiModel = this.#resolveOpenAIModel(args);
+			if (openaiModel) return openaiModel;
+		}
+		if (args.providerId === "xai" && this.#credentials.get("xai")?.type === "oauth") return withThinkingOptionsModel(xaiProvider(args.modelId, {
+			headers: args.headers,
+			authStorage: this.#credentials
+		}), providerThinkingOptions(`xai/${args.modelId}`, this.#thinkingLevel));
+		if (this.#routeThroughMastraGateway) return this.#mastraGateway.resolveLanguageModel(args);
+		if (args.providerId === "google") return this.#resolveGoogleModel(args);
+		if (args.providerId === "kimi-code-plan-cn" || args.providerId === "kimi-for-coding") return kimiCodingProvider(args.modelId, {
+			apiKey: args.apiKey,
+			headers: args.headers,
+			credentialStore: this.#credentials
+		});
+		const routedModelId = `${args.providerId}/${args.modelId}`;
+		return new ModelRouterLanguageModelWithThinking({
+			id: routedModelId,
+			apiKey: args.apiKey,
+			headers: args.headers
+		}, providerThinkingOptions(routedModelId, this.#thinkingLevel));
+	}
+	#resolveAnthropicModel(args) {
+		const bareModelId = normalizeAnthropicModelId(args.modelId);
+		const storedCred = this.#credentials.get("anthropic");
+		const thinkingMiddleware = createAnthropicThinkingMiddleware(bareModelId, this.#thinkingLevel);
+		if (this.#routeThroughMastraGateway) {
+			if (storedCred?.type === "oauth") return wrapLanguageModel({
+				model: createAnthropic({
+					apiKey: "oauth-gateway-placeholder",
+					baseURL: `${this.#mastraGatewayBaseUrl}/v1`,
+					headers: {
+						[GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,
+						...args.headers
+					},
+					fetch: buildAnthropicOAuthFetch({ authStorage: this.#credentials })
+				})(bareModelId),
+				middleware: [
+					claudeCodeMiddleware,
+					createPromptCacheMiddleware(this.#anthropicPromptCacheScope),
+					...thinkingMiddleware ? [thinkingMiddleware] : []
+				]
+			});
+			return this.#mastraGateway.resolveLanguageModel({
+				...args,
+				modelId: bareModelId
+			});
+		}
+		if (storedCred?.type === "oauth") return opencodeClaudeMaxProvider(bareModelId, {
+			headers: args.headers,
+			authStorage: this.#credentials,
+			thinkingLevel: this.#thinkingLevel,
+			promptCacheScope: this.#anthropicPromptCacheScope
+		});
+		if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return anthropicApiKeyProvider(bareModelId, storedCred.key.trim(), args.headers, this.#thinkingLevel, this.#anthropicPromptCacheScope);
+		const apiKey = getAnthropicApiKey(this.#credentials);
+		if (apiKey) return anthropicApiKeyProvider(bareModelId, apiKey, args.headers, this.#thinkingLevel, this.#anthropicPromptCacheScope);
+		return opencodeClaudeMaxProvider(bareModelId, {
+			headers: args.headers,
+			authStorage: this.#credentials,
+			thinkingLevel: this.#thinkingLevel,
+			promptCacheScope: this.#anthropicPromptCacheScope
+		});
+	}
+	#resolveOpenAIModel(args) {
+		const storedCred = this.#credentials.get("openai-codex");
+		if (this.#routeThroughMastraGateway) {
+			if (storedCred?.type === "oauth") {
+				const resolvedBareModelId = remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`).substring(OPENAI_PREFIX.length);
+				const reasoningEffort = THINKING_LEVEL_TO_REASONING_EFFORT[getEffectiveThinkingLevel(args.modelId, this.#thinkingLevel ?? "medium")];
+				const middleware = createCodexMiddleware(reasoningEffort);
+				return wrapLanguageModel({
+					model: createOpenAI({
+						apiKey: "oauth-gateway-placeholder",
+						baseURL: `${this.#mastraGatewayBaseUrl}/v1`,
+						headers: {
+							[GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,
+							...args.headers
+						},
+						fetch: buildOpenAICodexOAuthFetch({
+							authStorage: this.#credentials,
+							rewriteUrl: false
+						})
+					}).responses(resolvedBareModelId),
+					middleware: [middleware]
+				});
+			}
+			return this.#mastraGateway.resolveLanguageModel(args);
+		}
+		if (storedCred?.type === "oauth") return openaiCodexProvider(remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`).substring(OPENAI_PREFIX.length), {
+			thinkingLevel: this.#thinkingLevel && getEffectiveThinkingLevel(args.modelId, this.#thinkingLevel),
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+		const apiKey = getOpenAIApiKey(this.#credentials);
+		if (apiKey) return openaiApiKeyProvider(args.modelId, apiKey, args.headers, this.#thinkingLevel);
+	}
+	#resolveGoogleModel(args) {
+		const routerModel = new ModelRouterLanguageModel({
+			id: `${args.providerId}/${args.modelId}`,
+			apiKey: args.apiKey,
+			headers: args.headers
+		});
+		const thinkingMiddleware = createGoogleThinkingMiddleware(args.modelId, this.#thinkingLevel);
+		if (!thinkingMiddleware) return routerModel;
+		return wrapLanguageModel({
+			model: routerModel,
+			middleware: [thinkingMiddleware]
+		});
+	}
+};
+//#endregion
+export { MASTRACODE_GATEWAY_ID, MASTRA_GATEWAY_PREFIX, MastraCodeGateway, OPENAI_PREFIX, getAnthropicApiKey, getGlobalAuthStorage, getOpenAIApiKey, reloadAuthStorage, remapOpenAIModelForCodexOAuth, resolveAuth, stripMastraGatewayPrefix };
+
+//# sourceMappingURL=mastracode-gateway.js.map

package/dist/agents/mastracode-gateway.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastracode-gateway.js","names":["#mastraGateway","#mastraGatewayBaseUrl","#mastraGatewayApiKey","#routeThroughMastraGateway","#thinkingLevel","#anthropicPromptCacheScope","#customProviders","#settingsPath","#credentials","#getCustomProviders","#resolveAnthropicModel","#resolveOpenAIModel","#resolveGoogleModel"],"sources":["../../src/agents/mastracode-gateway.ts"],"sourcesContent":["import { createAnthropic } from '@ai-sdk/anthropic';\nimport { createOpenAI } from '@ai-sdk/openai';\nimport { createOpenAICompatible } from '@ai-sdk/openai-compatible';\nimport type { CustomAvailableModel, CustomModelCatalogProvider } from '@mastra/core/agent-controller';\nimport {\n  GATEWAY_AUTH_HEADER,\n  MastraGateway,\n  MastraModelGateway,\n  ModelRouterLanguageModel,\n  PROVIDER_REGISTRY,\n} from '@mastra/core/llm';\nimport type {\n  GatewayAuthRequest,\n  GatewayAuthResult,\n  GatewayLanguageModel,\n  MastraModelGatewayInterface,\n  ProviderConfig,\n} from '@mastra/core/llm';\nimport { wrapLanguageModel } from 'ai';\nimport { AuthStorage } from '../auth/storage.js';\nimport type { CredentialStore } from '../auth/types.js';\nimport { getCustomProviderId, loadSettings, MASTRA_GATEWAY_PROVIDER } from '../onboarding/settings.js';\nimport type { AnthropicPromptCacheScope } from '../providers/anthropic-prompt-cache.js';\nimport {\n  buildAnthropicOAuthFetch,\n  claudeCodeMiddleware,\n  createAnthropicThinkingMiddleware,\n  createPromptCacheMiddleware,\n  opencodeClaudeMaxProvider,\n} from '../providers/claude-max.js';\nimport { githubCopilotProvider } from '../providers/github-copilot.js';\nimport { createGoogleThinkingMiddleware } from '../providers/google-thinking.js';\nimport { kimiCodingProvider } from '../providers/kimi-coding.js';\nimport {\n  getCredentialProviderId,\n  KIMI_CODING_CREDENTIAL_ID,\n  KIMI_CODING_PROVIDER_ID,\n  normalizeAnthropicModelId,\n  OPENAI_PREFIX,\n  remapOpenAIModelForCodexOAuth,\n  stripMastraGatewayPrefix,\n} from '../providers/model-ids.js';\nimport {\n  buildOpenAICodexOAuthFetch,\n  createCodexMiddleware,\n  createReasoningEffortMiddleware,\n  getEffectiveThinkingLevel,\n  openaiCodexProvider,\n  THINKING_LEVEL_TO_REASONING_EFFORT,\n} from '../providers/openai-codex.js';\nimport type { ThinkingLevel } from '../providers/openai-codex.js';\nimport {\n  ModelRouterLanguageModelWithThinking,\n  providerThinkingOptions,\n  withThinkingOptionsModel,\n} from '../providers/provider-thinking.js';\nimport { xaiProvider } from '../providers/xai.js';\nimport { getAppDataDir } from '../utils/project.js';\nimport { resolveCustomProviders } from './custom-provider-source.js';\n\nexport {\n  MASTRA_GATEWAY_PREFIX,\n  OPENAI_PREFIX,\n  remapOpenAIModelForCodexOAuth,\n  stripMastraGatewayPrefix,\n} from '../providers/model-ids.js';\n\nexport const MASTRACODE_GATEWAY_ID = 'mastracode';\n\ntype ModelRequestHeaders = Record<string, string>;\n\nexport type MastraCodeCustomProvider = { name: string; url: string; apiKey?: string; models?: string[] };\n\nexport type MastraCodeGatewayOptions = {\n  mastraGatewayBaseUrl: string;\n  mastraGatewayApiKey?: string;\n  routeThroughMastraGateway: boolean;\n  thinkingLevel?: ThinkingLevel;\n  /** Anthropic prompt-cache breakpoints to write. Defaults to `conversation`. */\n  anthropicPromptCacheScope?: AnthropicPromptCacheScope;\n  customProviders?: MastraCodeCustomProvider[];\n  settingsPath?: string;\n  /**\n   * Per-tenant credential source for this gateway instance. Defaults to the\n   * server-global file-backed `AuthStorage`; deployed web passes the calling\n   * tenant's store so model calls use the caller's own credentials.\n   */\n  credentialStore?: CredentialStore;\n};\n\n/**\n * Global file-backed credential store for local (TUI) model resolution.\n *\n * Constructed lazily, NOT at module scope: `AuthStorage`'s default path is\n * resolved from `MASTRA_APP_DATA_DIR` at construction, and this module is\n * routinely imported before test harnesses / embedders point that variable at\n * an isolated directory — an import-time construction would permanently pin\n * the developer's real app-data dir for every credential read.\n */\nlet authStorageSingleton: AuthStorage | undefined;\nlet authStorageSingletonDir: string | undefined;\n\nexport function getGlobalAuthStorage(): AuthStorage {\n  // Keyed by the resolved app-data dir: in-process harnesses re-point\n  // MASTRA_APP_DATA_DIR per run inside one long-lived worker, so a store\n  // pinned to a previous dir must be discarded, not reused.\n  const dir = getAppDataDir();\n  if (!authStorageSingleton || authStorageSingletonDir !== dir) {\n    authStorageSingleton = new AuthStorage();\n    authStorageSingletonDir = dir;\n  }\n  return authStorageSingleton;\n}\n\nexport function reloadAuthStorage() {\n  getGlobalAuthStorage().reload();\n}\n\n/**\n * Resolve the Anthropic API key.\n * Main slot → dedicated apikey: slot → env var.\n */\nexport function getAnthropicApiKey(credentials: CredentialStore = getGlobalAuthStorage()): string | undefined {\n  const storedCred = credentials.get('anthropic');\n  if (storedCred?.type === 'api_key' && storedCred.key.trim().length > 0) {\n    return storedCred.key.trim();\n  }\n  const dedicatedKey = credentials.getStoredApiKey('anthropic')?.trim();\n  if (dedicatedKey) return dedicatedKey;\n  return credentials.allowEnvironmentFallback === false\n    ? undefined\n    : process.env.ANTHROPIC_API_KEY?.trim() || undefined;\n}\n\n/**\n * Resolve the OpenAI API key.\n * Main slot → dedicated apikey: slot → env var.\n */\nexport function getOpenAIApiKey(credentials: CredentialStore = getGlobalAuthStorage()): string | undefined {\n  const storedCred = credentials.get('openai-codex');\n  if (storedCred?.type === 'api_key' && storedCred.key.trim().length > 0) {\n    return storedCred.key.trim();\n  }\n  const dedicatedKey = credentials.getStoredApiKey('openai-codex')?.trim();\n  if (dedicatedKey) return dedicatedKey;\n  return credentials.allowEnvironmentFallback === false ? undefined : process.env.OPENAI_API_KEY?.trim() || undefined;\n}\n\nfunction anthropicApiKeyProvider(\n  modelId: string,\n  apiKey: string,\n  headers: ModelRequestHeaders | undefined,\n  thinkingLevel: ThinkingLevel | undefined,\n  promptCacheScope: AnthropicPromptCacheScope,\n) {\n  const anthropic = createAnthropic({ apiKey, headers });\n  const thinkingMiddleware = createAnthropicThinkingMiddleware(modelId, thinkingLevel);\n  return wrapLanguageModel({\n    model: anthropic(modelId),\n    middleware: [createPromptCacheMiddleware(promptCacheScope), ...(thinkingMiddleware ? [thinkingMiddleware] : [])],\n  });\n}\n\nfunction openaiApiKeyProvider(\n  modelId: string,\n  apiKey: string,\n  headers?: ModelRequestHeaders,\n  thinkingLevel?: ThinkingLevel,\n) {\n  const openai = createOpenAI({ apiKey, baseURL: process.env.OPENAI_BASE_URL, headers });\n  const middleware =\n    thinkingLevel && thinkingLevel !== 'off'\n      ? createReasoningEffortMiddleware(\n          'openai',\n          THINKING_LEVEL_TO_REASONING_EFFORT[getEffectiveThinkingLevel(modelId, thinkingLevel)],\n        )\n      : undefined;\n  return wrapLanguageModel({\n    model: openai.responses(modelId),\n    middleware: middleware ? [middleware] : [],\n  });\n}\n\nfunction getProviderAuthKey(\n  providerId: string,\n  credentials: CredentialStore = getGlobalAuthStorage(),\n): string | undefined {\n  const authProviderId = getCredentialProviderId(providerId);\n  const storedCred = credentials.get(authProviderId);\n  if (storedCred?.type === 'api_key' && storedCred.key.trim().length > 0) {\n    return storedCred.key.trim();\n  }\n  const dedicatedKey = credentials.getStoredApiKey(authProviderId)?.trim();\n  if (dedicatedKey) return dedicatedKey;\n  // `/api-keys` stores keys under the catalog provider id (`kimi-code-plan-cn`),\n  // which can differ from the credential id.\n  const catalogKey = authProviderId !== providerId ? credentials.getStoredApiKey(providerId)?.trim() : undefined;\n  if (catalogKey) return catalogKey;\n  if (credentials.allowEnvironmentFallback === false) return undefined;\n  return authProviderId === KIMI_CODING_CREDENTIAL_ID ? process.env.KIMI_API_KEY?.trim() || undefined : undefined;\n}\n\nexport function resolveAuth(request: GatewayAuthRequest, mastraGatewayApiKey?: string): GatewayAuthResult | undefined {\n  return MastraCodeGateway.resolveProviderAuth(request, mastraGatewayApiKey);\n}\n\nfunction getGatewayProviderKey(gatewayId: string, providerId: string): string {\n  if (gatewayId === 'models.dev') return providerId;\n  return providerId === gatewayId ? gatewayId : `${gatewayId}/${providerId}`;\n}\n\nfunction parseGatewayRouterId(\n  routerId: string,\n  gateway: MastraModelGatewayInterface,\n): { gatewayId: string; providerId: string; modelId: string } {\n  const [firstPart = '', secondPart = '', ...restParts] = routerId.split('/');\n  const gatewayId = gateway.id;\n\n  if (firstPart === gatewayId && secondPart) {\n    return {\n      gatewayId,\n      providerId: secondPart,\n      modelId: restParts.join('/'),\n    };\n  }\n\n  return {\n    gatewayId,\n    providerId: firstPart,\n    modelId: [secondPart, ...restParts].filter(Boolean).join('/'),\n  };\n}\n\nfunction hasResolvedAuth(auth: GatewayAuthResult | undefined): boolean {\n  if (!auth) return false;\n  if (auth.apiKey || auth.bearerToken) return true;\n  return auth.headers ? Object.keys(auth.headers).length > 0 : false;\n}\n\nasync function resolveGatewayProviderAuth(\n  gateway: MastraModelGatewayInterface,\n  routerId: string,\n): Promise<GatewayAuthResult | undefined> {\n  if (!gateway.resolveAuth) return undefined;\n\n  const parsed = parseGatewayRouterId(routerId, gateway);\n  try {\n    const result = await gateway.resolveAuth({\n      gatewayId: parsed.gatewayId,\n      providerId: parsed.providerId,\n      modelId: parsed.modelId,\n      routerId,\n    });\n    return hasResolvedAuth(result) ? result : undefined;\n  } catch {\n    return undefined;\n  }\n}\n\nasync function getMastraCodeProviderConfigs(\n  gateway: MastraModelGatewayInterface,\n): Promise<Record<string, ProviderConfig>> {\n  const providers: Record<string, ProviderConfig> = { ...(PROVIDER_REGISTRY as Record<string, ProviderConfig>) };\n\n  try {\n    const gatewayProviders = await gateway.fetchProviders();\n    for (const [providerId, config] of Object.entries(gatewayProviders)) {\n      providers[getGatewayProviderKey(gateway.id, providerId)] = {\n        ...config,\n        gateway: gateway.id,\n      };\n    }\n  } catch (error) {\n    console.warn(`Failed to load providers from gateway ${gateway.id}:`, error);\n  }\n\n  return providers;\n}\n\nfunction getApiKeyEnvVar(providerConfig: Pick<ProviderConfig, 'apiKeyEnvVar'> | undefined): string | undefined {\n  const envVars = providerConfig?.apiKeyEnvVar;\n  return Array.isArray(envVars) ? envVars[0] : envVars;\n}\n\nexport class MastraCodeGateway extends MastraModelGateway {\n  readonly id = MASTRACODE_GATEWAY_ID;\n  readonly name = 'MastraCode Gateway';\n\n  readonly #mastraGateway: MastraGateway;\n  readonly #mastraGatewayBaseUrl: string;\n  readonly #mastraGatewayApiKey?: string;\n  readonly #routeThroughMastraGateway: boolean;\n  readonly #thinkingLevel?: ThinkingLevel;\n  readonly #anthropicPromptCacheScope: AnthropicPromptCacheScope;\n  readonly #customProviders?: MastraCodeCustomProvider[];\n  readonly #settingsPath?: string;\n  readonly #credentials: CredentialStore;\n\n  constructor({\n    mastraGatewayBaseUrl,\n    mastraGatewayApiKey,\n    routeThroughMastraGateway,\n    thinkingLevel,\n    anthropicPromptCacheScope,\n    customProviders,\n    settingsPath,\n    credentialStore,\n  }: MastraCodeGatewayOptions) {\n    super();\n    this.#mastraGateway = new MastraGateway({ baseUrl: mastraGatewayBaseUrl });\n    this.#mastraGatewayBaseUrl = mastraGatewayBaseUrl;\n    this.#mastraGatewayApiKey = mastraGatewayApiKey;\n    this.#routeThroughMastraGateway = routeThroughMastraGateway;\n    this.#thinkingLevel = thinkingLevel;\n    this.#anthropicPromptCacheScope = anthropicPromptCacheScope ?? 'conversation';\n    this.#customProviders = customProviders;\n    this.#settingsPath = settingsPath;\n    this.#credentials = credentialStore ?? getGlobalAuthStorage();\n  }\n\n  static getMastraGatewayApiKey(): string | undefined {\n    return getGlobalAuthStorage().getStoredApiKey(MASTRA_GATEWAY_PROVIDER) ?? process.env['MASTRA_GATEWAY_API_KEY'];\n  }\n\n  /** @deprecated Renamed to {@link MastraCodeGateway.getMastraGatewayApiKey}. */\n  static getMemoryGatewayApiKey(): string | undefined {\n    return MastraCodeGateway.getMastraGatewayApiKey();\n  }\n\n  /**\n   * Claim an unprefixed model id (e.g. a bare `anthropic/...`) when this gateway\n   * can authenticate it via OAuth or a stored/env credential. This lets the\n   * shared gateway router select MastraCode for auth resolution, so the status\n   * bar and `/api-keys` reflect OAuth logins instead of only checking env vars.\n   */\n  handlesModel(modelId: string): boolean {\n    const parsed = parseGatewayRouterId(modelId, this);\n    if (!parsed.providerId || !parsed.modelId) return false;\n    if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) return true;\n    const customProvider = this.#getCustomProviders().find(\n      provider => parsed.providerId === getCustomProviderId(provider.name),\n    );\n    if (customProvider?.apiKey) return true;\n    return hasResolvedAuth(\n      MastraCodeGateway.resolveProviderAuth(\n        {\n          gatewayId: this.id,\n          providerId: parsed.providerId,\n          modelId: parsed.modelId,\n          routerId: modelId,\n        },\n        undefined,\n        this.#credentials,\n      ),\n    );\n  }\n\n  static resolveProviderAuth(\n    request: GatewayAuthRequest,\n    mastraGatewayApiKey?: string,\n    credentials: CredentialStore = getGlobalAuthStorage(),\n  ): GatewayAuthResult | undefined {\n    if (request.gatewayId === 'mastra' && mastraGatewayApiKey) {\n      return { apiKey: mastraGatewayApiKey, source: 'gateway' };\n    }\n\n    const authProviderId = getCredentialProviderId(request.providerId);\n    const storedCred = credentials.get(authProviderId);\n    if (storedCred?.type === 'oauth') {\n      return { bearerToken: 'oauth', source: 'gateway' };\n    }\n\n    const apiKey = getProviderAuthKey(request.providerId, credentials);\n    return apiKey ? { apiKey, source: 'gateway' } : undefined;\n  }\n\n  static createModelCatalogProvider(gateway: MastraModelGatewayInterface): CustomModelCatalogProvider {\n    return async () => {\n      const registry = await getMastraCodeProviderConfigs(gateway);\n      const models: CustomAvailableModel[] = [];\n\n      for (const [provider, providerConfig] of Object.entries(registry)) {\n        const apiKeyEnvVar = getApiKeyEnvVar(providerConfig);\n        const hasEnvKey = apiKeyEnvVar ? Boolean(process.env[apiKeyEnvVar]) : false;\n        const modelNames = providerConfig.models;\n        if (!Array.isArray(modelNames)) continue;\n\n        const gatewayAuth = modelNames[0]\n          ? await resolveGatewayProviderAuth(gateway, `${provider}/${modelNames[0]}`)\n          : undefined;\n\n        for (const modelName of modelNames) {\n          const id = `${provider}/${modelName}`;\n          models.push({\n            id,\n            provider,\n            modelName,\n            hasApiKey: hasEnvKey || Boolean(gatewayAuth),\n            apiKeyEnvVar: apiKeyEnvVar || undefined,\n          });\n        }\n      }\n\n      return models;\n    };\n  }\n\n  createModelCatalogProvider(): CustomModelCatalogProvider {\n    return MastraCodeGateway.createModelCatalogProvider(this);\n  }\n\n  #getCustomProviders(): MastraCodeCustomProvider[] {\n    // Explicit constructor list wins; then a registered source (deployed web,\n    // DB-backed — authoritative, so settings.json is never consulted); then\n    // the local file-backed settings.\n    return this.#customProviders ?? resolveCustomProviders() ?? loadSettings(this.#settingsPath).customProviders;\n  }\n\n  async fetchProviders(): Promise<Record<string, ProviderConfig>> {\n    // Catalog auth (`handlesModel`, `resolveAuth`) reads this store, while\n    // `/api-keys` and logins write through other AuthStorage instances. Pick\n    // up their changes on every catalog refresh.\n    this.#credentials.reload();\n    const providers: Record<string, ProviderConfig> = {};\n    for (const provider of this.#getCustomProviders()) {\n      const models = provider.models ?? [];\n      if (!models.length) continue;\n      providers[getCustomProviderId(provider.name)] = {\n        name: provider.name,\n        url: provider.url,\n        apiKeyEnvVar: '',\n        apiKeyHeader: 'Authorization',\n        gateway: this.id,\n        models,\n      };\n    }\n\n    // Subscription providers are not listed here: catalog ids from this gateway\n    // are namespaced (`mastracode/<provider>/...`). Kimi augments the models.dev\n    // `kimi-code-plan-cn` provider (claimed via `handlesModel`), and Copilot,\n    // which models.dev excludes, has its own `github-copilot` gateway.\n    return providers;\n  }\n\n  buildUrl(modelId: string): string | undefined | Promise<string | undefined> {\n    return this.#routeThroughMastraGateway ? this.#mastraGateway.buildUrl(modelId) : modelId;\n  }\n\n  async getApiKey(modelId: string): Promise<string> {\n    const providerId = stripMastraGatewayPrefix(modelId).split('/', 1)[0];\n    if (this.#routeThroughMastraGateway) return this.#mastraGatewayApiKey ?? '';\n    return providerId ? (getProviderAuthKey(providerId, this.#credentials) ?? '') : '';\n  }\n\n  resolveAuth(request: GatewayAuthRequest): GatewayAuthResult | undefined {\n    if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) {\n      return { apiKey: this.#mastraGatewayApiKey, source: 'gateway' };\n    }\n\n    const customProvider = this.#getCustomProviders().find(\n      provider => request.providerId === getCustomProviderId(provider.name),\n    );\n    if (customProvider?.apiKey) {\n      return { apiKey: customProvider.apiKey, source: 'gateway' };\n    }\n\n    return MastraCodeGateway.resolveProviderAuth(request, undefined, this.#credentials);\n  }\n\n  resolveLanguageModel(args: {\n    modelId: string;\n    providerId: string;\n    apiKey: string;\n    headers?: Record<string, string>;\n    transport?: any;\n    responsesWebSocket?: any;\n  }): GatewayLanguageModel {\n    const customProvider = this.#getCustomProviders().find(\n      provider => args.providerId === getCustomProviderId(provider.name),\n    );\n    if (customProvider) {\n      const provider = createOpenAICompatible({\n        name: args.providerId,\n        baseURL: customProvider.url,\n        apiKey: args.apiKey,\n        headers: args.headers,\n      });\n      const reasoningEffort = this.#thinkingLevel ? THINKING_LEVEL_TO_REASONING_EFFORT[this.#thinkingLevel] : undefined;\n      const middleware = createReasoningEffortMiddleware(args.providerId, reasoningEffort);\n      if (!middleware) {\n        return provider.chatModel(args.modelId);\n      }\n      return wrapLanguageModel({\n        model: provider.chatModel(args.modelId),\n        middleware: [middleware],\n      });\n    }\n\n    if (args.providerId === 'github-copilot') {\n      return githubCopilotProvider(args.modelId, {\n        headers: args.headers,\n        authStorage: this.#credentials,\n      }) as unknown as GatewayLanguageModel;\n    }\n\n    if (args.providerId === 'moonshotai') {\n      // Prefer the gateway-resolved key (stored credentials), then the canonical\n      // registry env var. Keep the legacy typo name as a fallback for anyone who\n      // set it because of the previous error message.\n      const apiKey =\n        args.apiKey?.trim() || process.env.MOONSHOT_API_KEY?.trim() || process.env.MOONSHOT_AI_API_KEY?.trim();\n      if (!apiKey) {\n        throw new Error('Need MOONSHOT_API_KEY');\n      }\n      return createAnthropic({\n        apiKey,\n        baseURL: 'https://api.moonshot.ai/anthropic/v1',\n        name: 'moonshotai.anthropicv1',\n        headers: args.headers,\n      })(args.modelId);\n    }\n\n    if (args.providerId === 'anthropic') {\n      return this.#resolveAnthropicModel(args);\n    }\n\n    if (args.providerId === 'openai') {\n      const openaiModel = this.#resolveOpenAIModel(args);\n      if (openaiModel) return openaiModel;\n    }\n\n    if (args.providerId === 'xai' && this.#credentials.get('xai')?.type === 'oauth') {\n      return withThinkingOptionsModel(\n        xaiProvider(args.modelId, { headers: args.headers, authStorage: this.#credentials }),\n        providerThinkingOptions(`xai/${args.modelId}`, this.#thinkingLevel),\n      );\n    }\n\n    if (this.#routeThroughMastraGateway) return this.#mastraGateway.resolveLanguageModel(args);\n\n    if (args.providerId === 'google') {\n      return this.#resolveGoogleModel(args);\n    }\n\n    if (args.providerId === KIMI_CODING_PROVIDER_ID || args.providerId === KIMI_CODING_CREDENTIAL_ID) {\n      return kimiCodingProvider(args.modelId, {\n        apiKey: args.apiKey,\n        headers: args.headers,\n        credentialStore: this.#credentials,\n      }) as unknown as GatewayLanguageModel;\n    }\n\n    const routedModelId: `${string}/${string}` = `${args.providerId}/${args.modelId}`;\n    return new ModelRouterLanguageModelWithThinking(\n      { id: routedModelId, apiKey: args.apiKey, headers: args.headers },\n      providerThinkingOptions(routedModelId, this.#thinkingLevel),\n    ) as unknown as GatewayLanguageModel;\n  }\n\n  #resolveAnthropicModel(args: {\n    modelId: string;\n    providerId: string;\n    apiKey: string;\n    headers?: Record<string, string>;\n    transport?: any;\n    responsesWebSocket?: any;\n  }): GatewayLanguageModel {\n    const bareModelId = normalizeAnthropicModelId(args.modelId);\n    const storedCred = this.#credentials.get('anthropic');\n    const thinkingMiddleware = createAnthropicThinkingMiddleware(bareModelId, this.#thinkingLevel);\n\n    if (this.#routeThroughMastraGateway) {\n      if (storedCred?.type === 'oauth') {\n        const anthropic = createAnthropic({\n          apiKey: 'oauth-gateway-placeholder',\n          baseURL: `${this.#mastraGatewayBaseUrl}/v1`,\n          headers: {\n            [GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,\n            ...args.headers,\n          },\n          fetch: buildAnthropicOAuthFetch({ authStorage: this.#credentials }),\n        });\n\n        return wrapLanguageModel({\n          model: anthropic(bareModelId),\n          middleware: [\n            claudeCodeMiddleware,\n            createPromptCacheMiddleware(this.#anthropicPromptCacheScope),\n            ...(thinkingMiddleware ? [thinkingMiddleware] : []),\n          ],\n        });\n      }\n\n      return this.#mastraGateway.resolveLanguageModel({ ...args, modelId: bareModelId });\n    }\n\n    if (storedCred?.type === 'oauth') {\n      return opencodeClaudeMaxProvider(bareModelId, {\n        headers: args.headers,\n        authStorage: this.#credentials,\n        thinkingLevel: this.#thinkingLevel,\n        promptCacheScope: this.#anthropicPromptCacheScope,\n      }) as unknown as GatewayLanguageModel;\n    }\n\n    if (storedCred?.type === 'api_key' && storedCred.key.trim().length > 0) {\n      return anthropicApiKeyProvider(\n        bareModelId,\n        storedCred.key.trim(),\n        args.headers,\n        this.#thinkingLevel,\n        this.#anthropicPromptCacheScope,\n      );\n    }\n\n    const apiKey = getAnthropicApiKey(this.#credentials);\n    if (apiKey) {\n      return anthropicApiKeyProvider(\n        bareModelId,\n        apiKey,\n        args.headers,\n        this.#thinkingLevel,\n        this.#anthropicPromptCacheScope,\n      );\n    }\n\n    // No stored credentials: use the OAuth-backed provider so the first request can trigger login.\n    return opencodeClaudeMaxProvider(bareModelId, {\n      headers: args.headers,\n      authStorage: this.#credentials,\n      thinkingLevel: this.#thinkingLevel,\n      promptCacheScope: this.#anthropicPromptCacheScope,\n    }) as unknown as GatewayLanguageModel;\n  }\n\n  #resolveOpenAIModel(args: {\n    modelId: string;\n    providerId: string;\n    apiKey: string;\n    headers?: Record<string, string>;\n    transport?: any;\n    responsesWebSocket?: any;\n  }): GatewayLanguageModel | undefined {\n    const storedCred = this.#credentials.get('openai-codex');\n\n    if (this.#routeThroughMastraGateway) {\n      if (storedCred?.type === 'oauth') {\n        const resolvedModelId = remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`);\n        const resolvedBareModelId = resolvedModelId.substring(OPENAI_PREFIX.length);\n        const effectiveLevel = getEffectiveThinkingLevel(args.modelId, this.#thinkingLevel ?? 'medium');\n        const reasoningEffort = THINKING_LEVEL_TO_REASONING_EFFORT[effectiveLevel];\n        const middleware = createCodexMiddleware(reasoningEffort);\n        const openai = createOpenAI({\n          apiKey: 'oauth-gateway-placeholder',\n          baseURL: `${this.#mastraGatewayBaseUrl}/v1`,\n          headers: {\n            [GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,\n            ...args.headers,\n          },\n          fetch: buildOpenAICodexOAuthFetch({ authStorage: this.#credentials, rewriteUrl: false }),\n        });\n\n        return wrapLanguageModel({\n          model: openai.responses(resolvedBareModelId),\n          middleware: [middleware],\n        });\n      }\n\n      return this.#mastraGateway.resolveLanguageModel(args);\n    }\n\n    if (storedCred?.type === 'oauth') {\n      const resolvedModelId = remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`);\n      return openaiCodexProvider(resolvedModelId.substring(OPENAI_PREFIX.length), {\n        thinkingLevel: this.#thinkingLevel && getEffectiveThinkingLevel(args.modelId, this.#thinkingLevel),\n        headers: args.headers,\n        authStorage: this.#credentials,\n      }) as unknown as GatewayLanguageModel;\n    }\n\n    const apiKey = getOpenAIApiKey(this.#credentials);\n    if (apiKey) {\n      return openaiApiKeyProvider(args.modelId, apiKey, args.headers, this.#thinkingLevel);\n    }\n\n    return undefined;\n  }\n\n  #resolveGoogleModel(args: {\n    modelId: string;\n    providerId: string;\n    apiKey: string;\n    headers?: Record<string, string>;\n    transport?: any;\n    responsesWebSocket?: any;\n  }): GatewayLanguageModel {\n    const routerModel = new ModelRouterLanguageModel({\n      id: `${args.providerId}/${args.modelId}`,\n      apiKey: args.apiKey,\n      headers: args.headers,\n    }) as unknown as GatewayLanguageModel;\n    const thinkingMiddleware = createGoogleThinkingMiddleware(args.modelId, this.#thinkingLevel);\n    if (!thinkingMiddleware) return routerModel;\n\n    return wrapLanguageModel({\n      model: routerModel as any,\n      middleware: [thinkingMiddleware],\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAmEA,MAAa,wBAAwB;;;;;;;;;;AAgCrC,IAAI;AACJ,IAAI;AAEJ,SAAgB,uBAAoC;CAIlD,MAAM,MAAM,cAAc;CAC1B,IAAI,CAAC,wBAAwB,4BAA4B,KAAK;EAC5D,uBAAuB,IAAI,YAAY;EACvC,0BAA0B;CAC5B;CACA,OAAO;AACT;AAEA,SAAgB,oBAAoB;CAClC,qBAAqB,CAAC,CAAC,OAAO;AAChC;;;;;AAMA,SAAgB,mBAAmB,cAA+B,qBAAqB,GAAuB;CAC5G,MAAM,aAAa,YAAY,IAAI,WAAW;CAC9C,IAAI,YAAY,SAAS,aAAa,WAAW,IAAI,KAAK,CAAC,CAAC,SAAS,GACnE,OAAO,WAAW,IAAI,KAAK;CAE7B,MAAM,eAAe,YAAY,gBAAgB,WAAW,CAAC,EAAE,KAAK;CACpE,IAAI,cAAc,OAAO;CACzB,OAAO,YAAY,6BAA6B,QAC5C,KAAA,IACA,QAAQ,IAAI,mBAAmB,KAAK,KAAK,KAAA;AAC/C;;;;;AAMA,SAAgB,gBAAgB,cAA+B,qBAAqB,GAAuB;CACzG,MAAM,aAAa,YAAY,IAAI,cAAc;CACjD,IAAI,YAAY,SAAS,aAAa,WAAW,IAAI,KAAK,CAAC,CAAC,SAAS,GACnE,OAAO,WAAW,IAAI,KAAK;CAE7B,MAAM,eAAe,YAAY,gBAAgB,cAAc,CAAC,EAAE,KAAK;CACvE,IAAI,cAAc,OAAO;CACzB,OAAO,YAAY,6BAA6B,QAAQ,KAAA,IAAY,QAAQ,IAAI,gBAAgB,KAAK,KAAK,KAAA;AAC5G;AAEA,SAAS,wBACP,SACA,QACA,SACA,eACA,kBACA;CACA,MAAM,YAAY,gBAAgB;EAAE;EAAQ;CAAQ,CAAC;CACrD,MAAM,qBAAqB,kCAAkC,SAAS,aAAa;CACnF,OAAO,kBAAkB;EACvB,OAAO,UAAU,OAAO;EACxB,YAAY,CAAC,4BAA4B,gBAAgB,GAAG,GAAI,qBAAqB,CAAC,kBAAkB,IAAI,CAAC,CAAE;CACjH,CAAC;AACH;AAEA,SAAS,qBACP,SACA,QACA,SACA,eACA;CACA,MAAM,SAAS,aAAa;EAAE;EAAQ,SAAS,QAAQ,IAAI;EAAiB;CAAQ,CAAC;CACrF,MAAM,aACJ,iBAAiB,kBAAkB,QAC/B,gCACE,UACA,mCAAmC,0BAA0B,SAAS,aAAa,EACrF,IACA,KAAA;CACN,OAAO,kBAAkB;EACvB,OAAO,OAAO,UAAU,OAAO;EAC/B,YAAY,aAAa,CAAC,UAAU,IAAI,CAAC;CAC3C,CAAC;AACH;AAEA,SAAS,mBACP,YACA,cAA+B,qBAAqB,GAChC;CACpB,MAAM,iBAAiB,wBAAwB,UAAU;CACzD,MAAM,aAAa,YAAY,IAAI,cAAc;CACjD,IAAI,YAAY,SAAS,aAAa,WAAW,IAAI,KAAK,CAAC,CAAC,SAAS,GACnE,OAAO,WAAW,IAAI,KAAK;CAE7B,MAAM,eAAe,YAAY,gBAAgB,cAAc,CAAC,EAAE,KAAK;CACvE,IAAI,cAAc,OAAO;CAGzB,MAAM,aAAa,mBAAmB,aAAa,YAAY,gBAAgB,UAAU,CAAC,EAAE,KAAK,IAAI,KAAA;CACrG,IAAI,YAAY,OAAO;CACvB,IAAI,YAAY,6BAA6B,OAAO,OAAO,KAAA;CAC3D,OAAO,mBAAA,oBAA+C,QAAQ,IAAI,cAAc,KAAK,KAAK,KAAA,IAAY,KAAA;AACxG;AAEA,SAAgB,YAAY,SAA6B,qBAA6D;CACpH,OAAO,kBAAkB,oBAAoB,SAAS,mBAAmB;AAC3E;AAEA,SAAS,sBAAsB,WAAmB,YAA4B;CAC5E,IAAI,cAAc,cAAc,OAAO;CACvC,OAAO,eAAe,YAAY,YAAY,GAAG,UAAU,GAAG;AAChE;AAEA,SAAS,qBACP,UACA,SAC4D;CAC5D,MAAM,CAAC,YAAY,IAAI,aAAa,IAAI,GAAG,aAAa,SAAS,MAAM,GAAG;CAC1E,MAAM,YAAY,QAAQ;CAE1B,IAAI,cAAc,aAAa,YAC7B,OAAO;EACL;EACA,YAAY;EACZ,SAAS,UAAU,KAAK,GAAG;CAC7B;CAGF,OAAO;EACL;EACA,YAAY;EACZ,SAAS,CAAC,YAAY,GAAG,SAAS,CAAC,CAAC,OAAO,OAAO,CAAC,CAAC,KAAK,GAAG;CAC9D;AACF;AAEA,SAAS,gBAAgB,MAA8C;CACrE,IAAI,CAAC,MAAM,OAAO;CAClB,IAAI,KAAK,UAAU,KAAK,aAAa,OAAO;CAC5C,OAAO,KAAK,UAAU,OAAO,KAAK,KAAK,OAAO,CAAC,CAAC,SAAS,IAAI;AAC/D;AAEA,eAAe,2BACb,SACA,UACwC;CACxC,IAAI,CAAC,QAAQ,aAAa,OAAO,KAAA;CAEjC,MAAM,SAAS,qBAAqB,UAAU,OAAO;CACrD,IAAI;EACF,MAAM,SAAS,MAAM,QAAQ,YAAY;GACvC,WAAW,OAAO;GAClB,YAAY,OAAO;GACnB,SAAS,OAAO;GAChB;EACF,CAAC;EACD,OAAO,gBAAgB,MAAM,IAAI,SAAS,KAAA;CAC5C,QAAQ;EACN;CACF;AACF;AAEA,eAAe,6BACb,SACyC;CACzC,MAAM,YAA4C,EAAE,GAAI,kBAAqD;CAE7G,IAAI;EACF,MAAM,mBAAmB,MAAM,QAAQ,eAAe;EACtD,KAAK,MAAM,CAAC,YAAY,WAAW,OAAO,QAAQ,gBAAgB,GAChE,UAAU,sBAAsB,QAAQ,IAAI,UAAU,KAAK;GACzD,GAAG;GACH,SAAS,QAAQ;EACnB;CAEJ,SAAS,OAAO;EACd,QAAQ,KAAK,yCAAyC,QAAQ,GAAG,IAAI,KAAK;CAC5E;CAEA,OAAO;AACT;AAEA,SAAS,gBAAgB,gBAAsF;CAC7G,MAAM,UAAU,gBAAgB;CAChC,OAAO,MAAM,QAAQ,OAAO,IAAI,QAAQ,KAAK;AAC/C;AAEA,IAAa,oBAAb,MAAa,0BAA0B,mBAAmB;CACxD,KAAc;CACd,OAAgB;CAEhB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CAEA,YAAY,EACV,sBACA,qBACA,2BACA,eACA,2BACA,iBACA,cACA,mBAC2B;EAC3B,MAAM;EACN,KAAKA,iBAAiB,IAAI,cAAc,EAAE,SAAS,qBAAqB,CAAC;EACzE,KAAKC,wBAAwB;EAC7B,KAAKC,uBAAuB;EAC5B,KAAKC,6BAA6B;EAClC,KAAKC,iBAAiB;EACtB,KAAKC,6BAA6B,6BAA6B;EAC/D,KAAKC,mBAAmB;EACxB,KAAKC,gBAAgB;EACrB,KAAKC,eAAe,mBAAmB,qBAAqB;CAC9D;CAEA,OAAO,yBAA6C;EAClD,OAAO,qBAAqB,CAAC,CAAC,gBAAA,gBAAuC,KAAK,QAAQ,IAAI;CACxF;;CAGA,OAAO,yBAA6C;EAClD,OAAO,kBAAkB,uBAAuB;CAClD;;;;;;;CAQA,aAAa,SAA0B;EACrC,MAAM,SAAS,qBAAqB,SAAS,IAAI;EACjD,IAAI,CAAC,OAAO,cAAc,CAAC,OAAO,SAAS,OAAO;EAClD,IAAI,KAAKL,8BAA8B,KAAKD,sBAAsB,OAAO;EAIzE,IAHuB,KAAKO,oBAAoB,CAAC,CAAC,MAChD,aAAY,OAAO,eAAe,oBAAoB,SAAS,IAAI,CAEpD,CAAC,EAAE,QAAQ,OAAO;EACnC,OAAO,gBACL,kBAAkB,oBAChB;GACE,WAAW,KAAK;GAChB,YAAY,OAAO;GACnB,SAAS,OAAO;GAChB,UAAU;EACZ,GACA,KAAA,GACA,KAAKD,YACP,CACF;CACF;CAEA,OAAO,oBACL,SACA,qBACA,cAA+B,qBAAqB,GACrB;EAC/B,IAAI,QAAQ,cAAc,YAAY,qBACpC,OAAO;GAAE,QAAQ;GAAqB,QAAQ;EAAU;EAG1D,MAAM,iBAAiB,wBAAwB,QAAQ,UAAU;EAEjE,IADmB,YAAY,IAAI,cACtB,CAAC,EAAE,SAAS,SACvB,OAAO;GAAE,aAAa;GAAS,QAAQ;EAAU;EAGnD,MAAM,SAAS,mBAAmB,QAAQ,YAAY,WAAW;EACjE,OAAO,SAAS;GAAE;GAAQ,QAAQ;EAAU,IAAI,KAAA;CAClD;CAEA,OAAO,2BAA2B,SAAkE;EAClG,OAAO,YAAY;GACjB,MAAM,WAAW,MAAM,6BAA6B,OAAO;GAC3D,MAAM,SAAiC,CAAC;GAExC,KAAK,MAAM,CAAC,UAAU,mBAAmB,OAAO,QAAQ,QAAQ,GAAG;IACjE,MAAM,eAAe,gBAAgB,cAAc;IACnD,MAAM,YAAY,eAAe,QAAQ,QAAQ,IAAI,aAAa,IAAI;IACtE,MAAM,aAAa,eAAe;IAClC,IAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;IAEhC,MAAM,cAAc,WAAW,KAC3B,MAAM,2BAA2B,SAAS,GAAG,SAAS,GAAG,WAAW,IAAI,IACxE,KAAA;IAEJ,KAAK,MAAM,aAAa,YAAY;KAClC,MAAM,KAAK,GAAG,SAAS,GAAG;KAC1B,OAAO,KAAK;MACV;MACA;MACA;MACA,WAAW,aAAa,QAAQ,WAAW;MAC3C,cAAc,gBAAgB,KAAA;KAChC,CAAC;IACH;GACF;GAEA,OAAO;EACT;CACF;CAEA,6BAAyD;EACvD,OAAO,kBAAkB,2BAA2B,IAAI;CAC1D;CAEA,sBAAkD;EAIhD,OAAO,KAAKF,oBAAoB,uBAAuB,KAAK,aAAa,KAAKC,aAAa,CAAC,CAAC;CAC/F;CAEA,MAAM,iBAA0D;EAI9D,KAAKC,aAAa,OAAO;EACzB,MAAM,YAA4C,CAAC;EACnD,KAAK,MAAM,YAAY,KAAKC,oBAAoB,GAAG;GACjD,MAAM,SAAS,SAAS,UAAU,CAAC;GACnC,IAAI,CAAC,OAAO,QAAQ;GACpB,UAAU,oBAAoB,SAAS,IAAI,KAAK;IAC9C,MAAM,SAAS;IACf,KAAK,SAAS;IACd,cAAc;IACd,cAAc;IACd,SAAS,KAAK;IACd;GACF;EACF;EAMA,OAAO;CACT;CAEA,SAAS,SAAmE;EAC1E,OAAO,KAAKN,6BAA6B,KAAKH,eAAe,SAAS,OAAO,IAAI;CACnF;CAEA,MAAM,UAAU,SAAkC;EAChD,MAAM,aAAa,yBAAyB,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;EACnE,IAAI,KAAKG,4BAA4B,OAAO,KAAKD,wBAAwB;EACzE,OAAO,aAAc,mBAAmB,YAAY,KAAKM,YAAY,KAAK,KAAM;CAClF;CAEA,YAAY,SAA4D;EACtE,IAAI,KAAKL,8BAA8B,KAAKD,sBAC1C,OAAO;GAAE,QAAQ,KAAKA;GAAsB,QAAQ;EAAU;EAGhE,MAAM,iBAAiB,KAAKO,oBAAoB,CAAC,CAAC,MAChD,aAAY,QAAQ,eAAe,oBAAoB,SAAS,IAAI,CACtE;EACA,IAAI,gBAAgB,QAClB,OAAO;GAAE,QAAQ,eAAe;GAAQ,QAAQ;EAAU;EAG5D,OAAO,kBAAkB,oBAAoB,SAAS,KAAA,GAAW,KAAKD,YAAY;CACpF;CAEA,qBAAqB,MAOI;EACvB,MAAM,iBAAiB,KAAKC,oBAAoB,CAAC,CAAC,MAChD,aAAY,KAAK,eAAe,oBAAoB,SAAS,IAAI,CACnE;EACA,IAAI,gBAAgB;GAClB,MAAM,WAAW,uBAAuB;IACtC,MAAM,KAAK;IACX,SAAS,eAAe;IACxB,QAAQ,KAAK;IACb,SAAS,KAAK;GAChB,CAAC;GACD,MAAM,kBAAkB,KAAKL,iBAAiB,mCAAmC,KAAKA,kBAAkB,KAAA;GACxG,MAAM,aAAa,gCAAgC,KAAK,YAAY,eAAe;GACnF,IAAI,CAAC,YACH,OAAO,SAAS,UAAU,KAAK,OAAO;GAExC,OAAO,kBAAkB;IACvB,OAAO,SAAS,UAAU,KAAK,OAAO;IACtC,YAAY,CAAC,UAAU;GACzB,CAAC;EACH;EAEA,IAAI,KAAK,eAAe,kBACtB,OAAO,sBAAsB,KAAK,SAAS;GACzC,SAAS,KAAK;GACd,aAAa,KAAKI;EACpB,CAAC;EAGH,IAAI,KAAK,eAAe,cAAc;GAIpC,MAAM,SACJ,KAAK,QAAQ,KAAK,KAAK,QAAQ,IAAI,kBAAkB,KAAK,KAAK,QAAQ,IAAI,qBAAqB,KAAK;GACvG,IAAI,CAAC,QACH,MAAM,IAAI,MAAM,uBAAuB;GAEzC,OAAO,gBAAgB;IACrB;IACA,SAAS;IACT,MAAM;IACN,SAAS,KAAK;GAChB,CAAC,CAAC,CAAC,KAAK,OAAO;EACjB;EAEA,IAAI,KAAK,eAAe,aACtB,OAAO,KAAKE,uBAAuB,IAAI;EAGzC,IAAI,KAAK,eAAe,UAAU;GAChC,MAAM,cAAc,KAAKC,oBAAoB,IAAI;GACjD,IAAI,aAAa,OAAO;EAC1B;EAEA,IAAI,KAAK,eAAe,SAAS,KAAKH,aAAa,IAAI,KAAK,CAAC,EAAE,SAAS,SACtE,OAAO,yBACL,YAAY,KAAK,SAAS;GAAE,SAAS,KAAK;GAAS,aAAa,KAAKA;EAAa,CAAC,GACnF,wBAAwB,OAAO,KAAK,WAAW,KAAKJ,cAAc,CACpE;EAGF,IAAI,KAAKD,4BAA4B,OAAO,KAAKH,eAAe,qBAAqB,IAAI;EAEzF,IAAI,KAAK,eAAe,UACtB,OAAO,KAAKY,oBAAoB,IAAI;EAGtC,IAAI,KAAK,eAAA,uBAA0C,KAAK,eAAA,mBACtD,OAAO,mBAAmB,KAAK,SAAS;GACtC,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,iBAAiB,KAAKJ;EACxB,CAAC;EAGH,MAAM,gBAAuC,GAAG,KAAK,WAAW,GAAG,KAAK;EACxE,OAAO,IAAI,qCACT;GAAE,IAAI;GAAe,QAAQ,KAAK;GAAQ,SAAS,KAAK;EAAQ,GAChE,wBAAwB,eAAe,KAAKJ,cAAc,CAC5D;CACF;CAEA,uBAAuB,MAOE;EACvB,MAAM,cAAc,0BAA0B,KAAK,OAAO;EAC1D,MAAM,aAAa,KAAKI,aAAa,IAAI,WAAW;EACpD,MAAM,qBAAqB,kCAAkC,aAAa,KAAKJ,cAAc;EAE7F,IAAI,KAAKD,4BAA4B;GACnC,IAAI,YAAY,SAAS,SAWvB,OAAO,kBAAkB;IACvB,OAXgB,gBAAgB;KAChC,QAAQ;KACR,SAAS,GAAG,KAAKF,sBAAsB;KACvC,SAAS;OACN,sBAAsB,UAAU,KAAK;MACtC,GAAG,KAAK;KACV;KACA,OAAO,yBAAyB,EAAE,aAAa,KAAKO,aAAa,CAAC;IACpE,CAGiB,CAAC,CAAC,WAAW;IAC5B,YAAY;KACV;KACA,4BAA4B,KAAKH,0BAA0B;KAC3D,GAAI,qBAAqB,CAAC,kBAAkB,IAAI,CAAC;IACnD;GACF,CAAC;GAGH,OAAO,KAAKL,eAAe,qBAAqB;IAAE,GAAG;IAAM,SAAS;GAAY,CAAC;EACnF;EAEA,IAAI,YAAY,SAAS,SACvB,OAAO,0BAA0B,aAAa;GAC5C,SAAS,KAAK;GACd,aAAa,KAAKQ;GAClB,eAAe,KAAKJ;GACpB,kBAAkB,KAAKC;EACzB,CAAC;EAGH,IAAI,YAAY,SAAS,aAAa,WAAW,IAAI,KAAK,CAAC,CAAC,SAAS,GACnE,OAAO,wBACL,aACA,WAAW,IAAI,KAAK,GACpB,KAAK,SACL,KAAKD,gBACL,KAAKC,0BACP;EAGF,MAAM,SAAS,mBAAmB,KAAKG,YAAY;EACnD,IAAI,QACF,OAAO,wBACL,aACA,QACA,KAAK,SACL,KAAKJ,gBACL,KAAKC,0BACP;EAIF,OAAO,0BAA0B,aAAa;GAC5C,SAAS,KAAK;GACd,aAAa,KAAKG;GAClB,eAAe,KAAKJ;GACpB,kBAAkB,KAAKC;EACzB,CAAC;CACH;CAEA,oBAAoB,MAOiB;EACnC,MAAM,aAAa,KAAKG,aAAa,IAAI,cAAc;EAEvD,IAAI,KAAKL,4BAA4B;GACnC,IAAI,YAAY,SAAS,SAAS;IAEhC,MAAM,sBADkB,8BAA8B,UAAU,KAAK,SAC3B,CAAC,CAAC,UAAU,cAAc,MAAM;IAE1E,MAAM,kBAAkB,mCADD,0BAA0B,KAAK,SAAS,KAAKC,kBAAkB,QACd;IACxE,MAAM,aAAa,sBAAsB,eAAe;IAWxD,OAAO,kBAAkB;KACvB,OAXa,aAAa;MAC1B,QAAQ;MACR,SAAS,GAAG,KAAKH,sBAAsB;MACvC,SAAS;QACN,sBAAsB,UAAU,KAAK;OACtC,GAAG,KAAK;MACV;MACA,OAAO,2BAA2B;OAAE,aAAa,KAAKO;OAAc,YAAY;MAAM,CAAC;KACzF,CAGc,CAAC,CAAC,UAAU,mBAAmB;KAC3C,YAAY,CAAC,UAAU;IACzB,CAAC;GACH;GAEA,OAAO,KAAKR,eAAe,qBAAqB,IAAI;EACtD;EAEA,IAAI,YAAY,SAAS,SAEvB,OAAO,oBADiB,8BAA8B,UAAU,KAAK,SAC5B,CAAC,CAAC,UAAU,cAAc,MAAM,GAAG;GAC1E,eAAe,KAAKI,kBAAkB,0BAA0B,KAAK,SAAS,KAAKA,cAAc;GACjG,SAAS,KAAK;GACd,aAAa,KAAKI;EACpB,CAAC;EAGH,MAAM,SAAS,gBAAgB,KAAKA,YAAY;EAChD,IAAI,QACF,OAAO,qBAAqB,KAAK,SAAS,QAAQ,KAAK,SAAS,KAAKJ,cAAc;CAIvF;CAEA,oBAAoB,MAOK;EACvB,MAAM,cAAc,IAAI,yBAAyB;GAC/C,IAAI,GAAG,KAAK,WAAW,GAAG,KAAK;GAC/B,QAAQ,KAAK;GACb,SAAS,KAAK;EAChB,CAAC;EACD,MAAM,qBAAqB,+BAA+B,KAAK,SAAS,KAAKA,cAAc;EAC3F,IAAI,CAAC,oBAAoB,OAAO;EAEhC,OAAO,kBAAkB;GACvB,OAAO;GACP,YAAY,CAAC,kBAAkB;EACjC,CAAC;CACH;AACF"}

package/dist/agents/memory.d.ts

@@ -0,0 +1,39 @@
+import type { Knowledge } from '@mastra/core/knowledge';
+import type { RequestContext } from '@mastra/core/request-context';
+import type { MastraCompositeStore } from '@mastra/core/storage';
+import type { MastraVector } from '@mastra/core/vector';
+import { Memory } from '@mastra/memory';
+import { LOCAL_KNOWLEDGE_ORG_ID } from '../knowledge-scope.js';
+import type { MastraCodeState } from '../schema.js';
+export { LOCAL_KNOWLEDGE_ORG_ID };
+/**
+ * Whether the experimental subconscious (knowledge graph + reminder sidekick)
+ * is switched on for this process: it needs a vector store and the opt-in flag.
+ */
+export declare function isSubconsciousEnabled(vector: MastraVector | undefined): boolean;
+/**
+ * Whether the subconscious tools (`knowledge_*`, `ask_memory`) are registered
+ * for a given session. Beyond the process-level switch, a Factory-owned
+ * session that cannot resolve its org refuses the subconscious entirely (see
+ * `getDynamicMemory`, which applies the same two checks inline because it also
+ * needs the resolved identity). The system prompt's tool guidance calls here so
+ * it never advertises tools that `getDynamicMemory` did not register.
+ */
+export declare function hasSubconsciousTools(vector: MastraVector | undefined, state: MastraCodeState | undefined): boolean;
+/**
+ * Dynamic memory factory function.
+ * Reads OM thresholds from controller state via requestContext.
+ * Model functions also read from requestContext (no mutable bridge needed).
+ */
+export declare function getDynamicMemory(storage: MastraCompositeStore, vector?: MastraVector, settingsPath?: string, options?: {
+    disableSettingsOmSeed?: boolean;
+}, knowledge?: Knowledge): (({ requestContext }: {
+    requestContext: RequestContext;
+}) => Memory) & {
+    /**
+     * Wait for background memory work (buffered observations, reflections, indexing) in every
+     * instance this factory created. Await before closing storage.
+     */
+    settled(): Promise<void>;
+};
+//# sourceMappingURL=memory.d.ts.map

package/dist/agents/memory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../src/agents/memory.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AACnE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD,OAAO,EAAE,MAAM,EAAgB,MAAM,gBAAgB,CAAC;AAEtD,OAAO,EAAE,sBAAsB,EAAiC,MAAM,uBAAuB,CAAC;AAI9F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAsLpD,OAAO,EAAE,sBAAsB,EAAE,CAAC;AA8BlC;;;GAGG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,YAAY,GAAG,SAAS,GAAG,OAAO,CAE/E;AAED;;;;;;;GAOG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,YAAY,GAAG,SAAS,EAAE,KAAK,EAAE,eAAe,GAAG,SAAS,GAAG,OAAO,CAElH;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,oBAAoB,EAC7B,MAAM,CAAC,EAAE,YAAY,EACrB,YAAY,CAAC,EAAE,MAAM,EACrB,OAAO,CAAC,EAAE;IAAE,qBAAqB,CAAC,EAAE,OAAO,CAAA;CAAE,EAC7C,SAAS,CAAC,EAAE,SAAS,yBAsBsB;IAAE,cAAc,EAAE,cAAc,CAAA;CAAE;IAmJ3E;;;OAGG;eACc,OAAO,CAAC,IAAI,CAAC;EAKjC"}