mastracode 0.21.2-alpha.4 → 0.21.2-alpha.6

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # mastracode
 
+## 0.21.2-alpha.6
+
+### Patch Changes
+
+- Fixed TUI crash on narrow terminals when prompt dialogs render lines wider than terminal width ([#17431](https://github.com/mastra-ai/mastra/pull/17431))
+
+- Updated dependencies [[`b0771a4`](https://github.com/mastra-ai/mastra/commit/b0771a48b46e46d270fca208a587922f3b7104a8), [`19a8658`](https://github.com/mastra-ai/mastra/commit/19a86589c788ef48bb6c1b0612cc82a201857379), [`a659a77`](https://github.com/mastra-ai/mastra/commit/a659a779bdebe3a52a518c56d2260592d0240fe0), [`3332be9`](https://github.com/mastra-ai/mastra/commit/3332be9701ecd77aba840959d9a1d1ce7aef02d3)]:
+  - @mastra/agent-browser@0.3.0-alpha.1
+  - @mastra/core@1.38.0-alpha.6
+
+## 0.21.2-alpha.5
+
+### Patch Changes
+
+- Updated dependencies [[`a18775a`](https://github.com/mastra-ai/mastra/commit/a18775a693172546ee2378d39b67d4e32895b251), [`1baf2d1`](https://github.com/mastra-ai/mastra/commit/1baf2d152c6881338ff8f114633d5316fe13dd15), [`8260167`](https://github.com/mastra-ai/mastra/commit/8260167431f98400f3acef4bbb7bd6027efd7a4b)]:
+  - @mastra/core@1.38.0-alpha.5
+  - @mastra/duckdb@1.4.1-alpha.0
+
 ## 0.21.2-alpha.4
 
 ### Patch Changes

package/dist/HarnessCompat.d.ts

@@ -0,0 +1,41 @@
+import type { Agent } from '@mastra/core/agent';
+import { Harness as HarnessLegacy } from '@mastra/core/harness';
+import type { HarnessConfig, HarnessMode as HarnessModeLegacy, HarnessThread } from '@mastra/core/harness';
+import type { Session, HarnessMode, Harness } from '@mastra/core/harness/v1';
+type CloneSessionOptions = {
+    sessionId?: string;
+    threadId?: string;
+    resourceId?: string;
+    parentSessionId?: string;
+    origin?: 'top-level' | 'subagent-tool';
+    modeId?: string;
+    modelId?: string;
+};
+export declare function v1ModeToLegacy<TState = {}>(mode: HarnessMode, agent: Agent): HarnessModeLegacy<TState>;
+export declare class HarnessCompat<TState = {}> extends HarnessLegacy<TState> {
+    #private;
+    constructor(args: HarnessConfig<TState>, harnessV1: Harness<HarnessMode[]>);
+    switchThread({ threadId }: {
+        threadId: string;
+    }): Promise<void>;
+    listThreads(options?: {
+        allResources?: boolean;
+        includeForkedSubagents?: boolean;
+    }): Promise<HarnessThread[]>;
+    cloneSession(opts?: CloneSessionOptions): Promise<Session>;
+    cloneThread({ sourceThreadId, title, resourceId, }?: {
+        sourceThreadId?: string;
+        title?: string;
+        resourceId?: string;
+    }): Promise<HarnessThread>;
+    getCurrentMode(): HarnessModeLegacy<TState>;
+    /**
+     * Switch to a different mode.
+     * Aborts any in-progress generation and switches to the mode's default model.
+     */
+    switchMode({ modeId }: {
+        modeId: string;
+    }): Promise<void>;
+}
+export {};
+//# sourceMappingURL=HarnessCompat.d.ts.map

package/dist/HarnessCompat.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HarnessCompat.d.ts","sourceRoot":"","sources":["../src/HarnessCompat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,OAAO,IAAI,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,IAAI,iBAAiB,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC3G,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAE7E,KAAK,mBAAmB,GAAG;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,WAAW,GAAG,eAAe,CAAC;IACvC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAIF,wBAAgB,cAAc,CAAC,MAAM,GAAG,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAUtG;AAED,qBAAa,aAAa,CAAC,MAAM,GAAG,EAAE,CAAE,SAAQ,aAAa,CAAC,MAAM,CAAC;;gBAIvD,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;IAMpE,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB/D,WAAW,CAAC,OAAO,CAAC,EAAE;QAAE,YAAY,CAAC,EAAE,OAAO,CAAC;QAAC,sBAAsB,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IA4C7G,YAAY,CAAC,IAAI,GAAE,mBAAwB,GAAG,OAAO,CAAC,OAAO,CAAC;IAS9D,WAAW,CAAC,EAChB,cAAc,EACd,KAAK,EACL,UAAU,GACX,GAAE;QACD,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,UAAU,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,aAAa,CAAC;IAiC/B,cAAc,IAAI,iBAAiB,CAAC,MAAM,CAAC;IAc3C;;;OAGG;IACG,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAchE"}

package/dist/{chunk-2TQ5QO5H.cjs → chunk-A4TDK7QP.cjs}

@@ -3,9 +3,11 @@
 var chunkDJBUYADN_cjs = require('./chunk-DJBUYADN.cjs');
 var chunkJ3ISOP5J_cjs = require('./chunk-J3ISOP5J.cjs');
 var chunkAWYGW5ZV_cjs = require('./chunk-AWYGW5ZV.cjs');
+var crypto = require('crypto');
+var os = require('os');
 var path = require('path');
 var agent = require('@mastra/core/agent');
-var harness = require('@mastra/core/harness');
+var v1 = require('@mastra/core/harness/v1');
 var llm = require('@mastra/core/llm');
 var processors = require('@mastra/core/processors');
 var storage = require('@mastra/core/storage');
@@ -15,7 +17,6 @@ var child_process = require('child_process');
 var tools = require('@mastra/core/tools');
 var tavily = require('@mastra/tavily');
 var tokenx = require('tokenx');
-var os = require('os');
 var workspace = require('@mastra/core/workspace');
 var zod = require('zod');
 var fs = require('fs');
@@ -25,6 +26,7 @@ var memory = require('@mastra/memory');
 var anthropic = require('@ai-sdk/anthropic');
 var openai = require('@ai-sdk/openai');
 var evals = require('@mastra/core/evals');
+var harness = require('@mastra/core/harness');
 var mcp = require('@mastra/mcp');
 var promises = require('fs/promises');
 var events = require('@mastra/core/events');
@@ -50,8 +52,8 @@ function _interopNamespace(e) {
   return Object.freeze(n);
 }
 
-var path__namespace = /*#__PURE__*/_interopNamespace(path);
 var os__namespace = /*#__PURE__*/_interopNamespace(os);
+var path__namespace = /*#__PURE__*/_interopNamespace(path);
 var fs__namespace = /*#__PURE__*/_interopNamespace(fs);
 
 var COMMON_BINARIES = [
@@ -599,7 +601,7 @@ function getDynamicWorkspace({ requestContext, mastra: mastra2 }) {
     return existing;
   }
   const userLsp = chunkDJBUYADN_cjs.loadSettings().lsp ?? {};
-  const mcModulePath = path.join(path.dirname(url.fileURLToPath((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-2TQ5QO5H.cjs', document.baseURI).href)))), "..");
+  const mcModulePath = path.join(path.dirname(url.fileURLToPath((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-A4TDK7QP.cjs', document.baseURI).href)))), "..");
   const lspConfig = {
     ...userLsp,
     packageRunner: userLsp.packageRunner || detectPackageRunner(projectPath),
@@ -1916,6 +1918,134 @@ function createEfficiencyScorer() {
     return parts.join("\n");
   });
 }
+function v1ModeToLegacy(mode, agent) {
+  const meta = mode.metadata ?? {};
+  return {
+    id: mode.id,
+    name: mode.description,
+    default: meta.default === true,
+    defaultModelId: mode.defaultModelId,
+    color: typeof meta.color === "string" ? meta.color : void 0,
+    agent
+  };
+}
+var HarnessCompat = class extends harness.Harness {
+  #session;
+  #harnessV1;
+  constructor(args, harnessV1) {
+    super(args);
+    this.#harnessV1 = harnessV1;
+  }
+  async switchThread({ threadId }) {
+    const modes = this.listModes();
+    this.#session = await this.#harnessV1.session({
+      threadId,
+      resourceId: this.getResourceId()
+    });
+    const defaultModelId = modes.find((mode) => mode.id === this.#session.getMode().id)?.defaultModelId;
+    if (defaultModelId) {
+      this.#session.setModelId(defaultModelId);
+    }
+    await super.switchThread({ threadId });
+  }
+  async listThreads(options) {
+    const [sessions, legacyThreads] = await Promise.all([this.#harnessV1.listSessions(), super.listThreads(options)]);
+    const resourceId = this.getResourceId();
+    const sessionThreads = sessions.filter((session) => options?.allResources || session.resourceId === resourceId).map((session) => {
+      const legacyThread = legacyThreads.find(
+        (thread) => thread.id === session.threadId && (!thread.resourceId || thread.resourceId === session.resourceId)
+      );
+      const metadata = legacyThread?.metadata;
+      if (!options?.includeForkedSubagents && metadata?.forkedSubagent === true) {
+        return void 0;
+      }
+      return {
+        id: session.threadId,
+        resourceId: session.resourceId,
+        createdAt: session.createdAt,
+        updatedAt: session.lastActivityAt,
+        metadata: {
+          ...metadata,
+          sessionId: session.id,
+          modeId: session.modeId,
+          modelId: session.modelId,
+          parentSessionId: session.parentSessionId,
+          origin: session.origin
+        }
+      };
+    }).filter((thread) => thread !== void 0);
+    const sessionKeys = new Set(sessionThreads.map((thread) => `${thread.resourceId}:${thread.id}`));
+    const sessionThreadIds = new Set(sessionThreads.map((thread) => thread.id));
+    return [
+      ...sessionThreads,
+      ...legacyThreads.filter(
+        (thread) => !sessionKeys.has(`${thread.resourceId}:${thread.id}`) && !(!thread.resourceId && sessionThreadIds.has(thread.id))
+      )
+    ];
+  }
+  async cloneSession(opts = {}) {
+    const session = this.#session;
+    if (!session) {
+      throw new Error("No active session to clone");
+    }
+    return this.#harnessV1.cloneSession(session, opts);
+  }
+  async cloneThread({
+    sourceThreadId,
+    title,
+    resourceId
+  } = {}) {
+    const sourceId = sourceThreadId ?? this.getCurrentThreadId();
+    if (!sourceId) {
+      throw new Error("No source thread to clone");
+    }
+    const sourceResourceId = resourceId ?? this.getResourceId();
+    const currentSession = this.#session;
+    const sourceSession = currentSession?.threadId === sourceId && currentSession.resourceId === sourceResourceId ? currentSession : await this.#harnessV1.session({
+      threadId: sourceId,
+      resourceId: sourceResourceId
+    });
+    this.#session = await this.#harnessV1.cloneSession(sourceSession, { title });
+    const thread = await this.#session.getThread();
+    if (!thread) {
+      throw new Error("Failed to load cloned thread");
+    }
+    return {
+      id: thread.id,
+      resourceId: thread.resourceId,
+      title: title ?? thread.title ?? "Cloned Thread",
+      createdAt: thread.createdAt,
+      updatedAt: thread.updatedAt,
+      metadata: thread.metadata
+    };
+  }
+  getCurrentMode() {
+    if (!this.#session) {
+      return super.getCurrentMode();
+    }
+    const mode = this.#session.getMode();
+    const mastra2 = this.getMastra();
+    if (!mastra2) {
+      throw new Error("HarnessCompat requires an initialized Mastra instance");
+    }
+    return v1ModeToLegacy(mode, mastra2.getAgentById(mode.agentId));
+  }
+  /**
+   * Switch to a different mode.
+   * Aborts any in-progress generation and switches to the mode's default model.
+   */
+  async switchMode({ modeId }) {
+    const mode = this.#harnessV1.getMode(modeId);
+    if (!mode) {
+      throw new Error(`Mode not found: ${modeId}`);
+    }
+    if (!this.#session) {
+      throw new Error("No active session to switch mode");
+    }
+    this.#session.setMode(mode);
+    await super.switchMode({ modeId });
+  }
+};
 var VALID_EVENTS = [
   "PreToolUse",
   "PostToolUse",
@@ -3027,6 +3157,37 @@ var PROVIDER_TO_OAUTH_ID = {
   openai: "openai-codex",
   "github-copilot": "github-copilot"
 };
+var CODE_AGENT_ID = "code-agent";
+function hash(value) {
+  return crypto.createHash("sha256").update(value).digest("hex").slice(0, 32);
+}
+function legacyModeToV1(mode, fallbackAgent) {
+  const agent = typeof mode.agent === "function" ? mode.agent({}) : mode.agent;
+  return {
+    id: mode.id,
+    agentId: (agent ?? fallbackAgent).id,
+    defaultModelId: mode.defaultModelId ?? "openai/gpt-5.5",
+    description: mode.name,
+    ...mode.id === "plan" ? { transitionsTo: "build" } : {},
+    metadata: {
+      color: mode.color,
+      default: mode.default,
+      name: mode.name
+    }
+  };
+}
+function applyEffectiveDefaultsToV1Modes(modes, effectiveDefaults) {
+  return modes.map((mode) => {
+    const savedModel = effectiveDefaults[mode.id];
+    if (!savedModel) {
+      return mode;
+    }
+    return {
+      ...mode,
+      defaultModelId: savedModel
+    };
+  });
+}
 function createAuthStorage() {
   const authStorage = new chunkJ3ISOP5J_cjs.AuthStorage();
   chunkDJBUYADN_cjs.setAuthStorage(authStorage);
@@ -3126,11 +3287,13 @@ async function createMastraCode(config) {
       }
     }
   }
+  const harnessStorage = new storage.InMemoryHarness();
   const storage$1 = new storage.MastraCompositeStore({
     id: "mastra-code-storage",
     default: storageResult.storage,
     domains: {
-      ...observabilityDomain ? { observability: observabilityDomain } : {}
+      ...observabilityDomain ? { observability: observabilityDomain } : {},
+      harness: harnessStorage
     }
   });
   const observability$1 = new observability.Observability({
@@ -3190,7 +3353,7 @@ async function createMastraCode(config) {
   const outcomeScorer = createOutcomeScorer();
   const efficiencyScorer = createEfficiencyScorer();
   const codeAgent = new agent.Agent({
-    id: "code-agent",
+    id: CODE_AGENT_ID,
     name: "Code Agent",
     instructions: getDynamicInstructions,
     model: chunkDJBUYADN_cjs.getDynamicModel,
@@ -3218,28 +3381,35 @@ async function createMastraCode(config) {
     errorProcessors: [new processors.StreamErrorRetryProcessor(), new processors.PrefillErrorHandler(), new processors.ProviderHistoryCompat()]
   });
   const defaultSubagents = [exploreSubagent, planSubagent, executeSubagent];
-  const defaultModes = [
+  const defaultModesV1 = [
     {
       id: "build",
-      name: "Build",
-      default: true,
-      defaultModelId: "anthropic/claude-opus-4-6",
-      color: chunkDJBUYADN_cjs.mastra.green,
-      agent: codeAgent
+      agentId: CODE_AGENT_ID,
+      description: "Build",
+      defaultModelId: "anthropic/claude-opus-4-7",
+      metadata: {
+        color: chunkDJBUYADN_cjs.mastra.green,
+        default: true
+      }
     },
     {
       id: "plan",
-      name: "Plan",
-      defaultModelId: "openai/gpt-5.2-codex",
-      color: chunkDJBUYADN_cjs.mastra.purple,
-      agent: codeAgent
+      agentId: CODE_AGENT_ID,
+      description: "Plan",
+      transitionsTo: "build",
+      defaultModelId: "openai/gpt-5.5",
+      metadata: {
+        color: chunkDJBUYADN_cjs.mastra.purple
+      }
     },
     {
       id: "fast",
-      name: "Fast",
+      agentId: CODE_AGENT_ID,
+      description: "Fast",
       defaultModelId: "cerebras/zai-glm-4.7",
-      color: chunkDJBUYADN_cjs.mastra.orange,
-      agent: codeAgent
+      metadata: {
+        color: chunkDJBUYADN_cjs.mastra.orange
+      }
     }
   ];
   const defaultHeartbeatHandlers = [
@@ -3286,10 +3456,15 @@ async function createMastraCode(config) {
   const effectiveReflectionThreshold = globalSettings.models.omReflectionThreshold ?? void 0;
   const effectiveCavemanObservations = globalSettings.models.omCavemanObservations ?? void 0;
   const effectiveObserveAttachments = globalSettings.models.omObserveAttachments ?? "auto";
-  const modes = (config?.modes ?? defaultModes).map((mode) => {
-    const savedModel = effectiveDefaults[mode.id];
-    return savedModel ? { ...mode, defaultModelId: savedModel } : mode;
-  });
+  const modesV1 = applyEffectiveDefaultsToV1Modes(
+    config?.modes ? config.modes.map((mode) => legacyModeToV1(mode, codeAgent)) : defaultModesV1,
+    effectiveDefaults
+  );
+  const defaultModeId = modesV1.find((mode) => mode.metadata?.default === true)?.id ?? modesV1.find((mode) => mode.id === "plan")?.id ?? modesV1[0]?.id;
+  if (!defaultModeId) {
+    throw new Error("MastraCode requires at least one mode");
+  }
+  const modes = modesV1.map((mode) => v1ModeToLegacy(mode, codeAgent));
   const subagentModeMap = { explore: "fast", plan: "plan", execute: "build" };
   const subagents = (config?.subagents ?? defaultSubagents).map((sa) => {
     const modeId = subagentModeMap[sa.id];
@@ -3344,114 +3519,152 @@ async function createMastraCode(config) {
       globalInitialState[`subagentModelId_${key}`] = modelId;
     }
   }
-  const typedStateSchema = stateSchema;
-  const harness$1 = new harness.Harness({
-    id: "mastra-code",
-    resourceId: project.resourceId,
-    storage: storage$1,
-    observability: observability$1,
+  const { threads } = await (await storage$1.getStore("memory"))?.listThreads({
+    perPage: false,
+    filter: {
+      resourceId: project.resourceId
+    }
+  }) ?? { threads: [] };
+  const ownerId = `mastracode-${hash(`${os.hostname()}\0${project.rootPath}`)}`;
+  await Promise.all(
+    threads.map((thread) => {
+      const sessionHash = hash(`${thread.resourceId}\0${thread.id}`);
+      const meta = thread.metadata;
+      const modeId = typeof meta?.currentModeId === "string" ? meta.currentModeId : defaultModeId;
+      const mode = modesV1.find((mode2) => mode2.id === modeId) ?? modesV1.find((mode2) => mode2.id === defaultModeId);
+      const modelId = typeof meta?.currentModelId === "string" ? meta.currentModelId : mode.defaultModelId;
+      return harnessStorage.saveSession({
+        id: `sess-${sessionHash}`,
+        ownerId,
+        resourceId: thread.resourceId,
+        threadId: thread.id,
+        modeId: mode.id,
+        modelId,
+        origin: "top-level",
+        createdAt: thread.createdAt,
+        lastActivityAt: thread.updatedAt
+      });
+    })
+  );
+  const harnessV1 = new v1.Harness({
+    ownerId,
+    agents: { [CODE_AGENT_ID]: codeAgent },
     memory,
-    pubsub: signalsPubSub,
-    stateSchema: typedStateSchema,
-    subagents,
-    resolveModel: (modelId) => chunkDJBUYADN_cjs.resolveModel(modelId),
-    toolCategoryResolver: chunkAWYGW5ZV_cjs.getToolCategory,
-    initialState: {
-      projectPath: project.rootPath,
-      projectName: project.name,
-      gitBranch: project.gitBranch,
-      yolo: true,
-      ...globalInitialState,
-      ...config?.initialState,
-      // configDir must always win over initialState spreads to stay in sync
-      // with MCP/hooks/storage which were already initialized with this value.
-      configDir
-    },
-    workspace: config?.workspace ?? getDynamicWorkspace,
-    browser: config?.browser,
-    modes,
-    heartbeatHandlers: config?.heartbeatHandlers ?? defaultHeartbeatHandlers,
-    modelAuthChecker: (provider) => {
-      const gatewayKey = authStorage.getStoredApiKey(chunkDJBUYADN_cjs.MEMORY_GATEWAY_PROVIDER) ?? process.env["MASTRA_GATEWAY_API_KEY"];
-      if (gatewayKey) {
-        const providerConfig = gatewayRegistry.getProviders()[provider];
-        if (providerConfig?.gateway === "mastra") return true;
-      }
-      const oauthId = PROVIDER_TO_OAUTH_ID[provider];
-      if (oauthId && authStorage.isLoggedIn(oauthId)) {
-        return true;
-      }
-      if (authStorage.hasStoredApiKey(provider)) {
-        return true;
-      }
-      if (provider === "anthropic") {
-        const cred = authStorage.get("anthropic");
-        if (cred?.type === "api_key" && cred.key.trim().length > 0) {
+    modes: modesV1,
+    defaultModeId,
+    storage: harnessStorage
+  });
+  const typedStateSchema = stateSchema;
+  const harness = new HarnessCompat(
+    {
+      id: "mastra-code",
+      resourceId: project.resourceId,
+      storage: storage$1,
+      observability: observability$1,
+      memory,
+      pubsub: signalsPubSub,
+      stateSchema: typedStateSchema,
+      subagents,
+      resolveModel: (modelId) => chunkDJBUYADN_cjs.resolveModel(modelId),
+      toolCategoryResolver: chunkAWYGW5ZV_cjs.getToolCategory,
+      initialState: {
+        projectPath: project.rootPath,
+        projectName: project.name,
+        gitBranch: project.gitBranch,
+        yolo: true,
+        ...globalInitialState,
+        ...config?.initialState,
+        // configDir must always win over initialState spreads to stay in sync
+        // with MCP/hooks/storage which were already initialized with this value.
+        configDir
+      },
+      workspace: config?.workspace ?? getDynamicWorkspace,
+      browser: config?.browser,
+      modes,
+      heartbeatHandlers: config?.heartbeatHandlers ?? defaultHeartbeatHandlers,
+      modelAuthChecker: (provider) => {
+        const gatewayKey = authStorage.getStoredApiKey(chunkDJBUYADN_cjs.MEMORY_GATEWAY_PROVIDER) ?? process.env["MASTRA_GATEWAY_API_KEY"];
+        if (gatewayKey) {
+          const providerConfig = gatewayRegistry.getProviders()[provider];
+          if (providerConfig?.gateway === "mastra") return true;
+        }
+        const oauthId = PROVIDER_TO_OAUTH_ID[provider];
+        if (oauthId && authStorage.isLoggedIn(oauthId)) {
           return true;
         }
-      }
-      if (provider === "openai") {
-        const cred = authStorage.get("openai-codex");
-        if (cred?.type === "api_key" && cred.key.trim().length > 0) {
+        if (authStorage.hasStoredApiKey(provider)) {
           return true;
         }
-      }
-      const customProvider = chunkDJBUYADN_cjs.loadSettings().customProviders.find((entry) => {
-        return provider === chunkDJBUYADN_cjs.getCustomProviderId(entry.name);
-      });
-      if (customProvider) {
-        return true;
-      }
-      return void 0;
-    },
-    modelUseCountProvider: () => chunkDJBUYADN_cjs.loadSettings().modelUseCounts,
-    modelUseCountTracker: (modelId) => {
-      try {
+        if (provider === "anthropic") {
+          const cred = authStorage.get("anthropic");
+          if (cred?.type === "api_key" && cred.key.trim().length > 0) {
+            return true;
+          }
+        }
+        if (provider === "openai") {
+          const cred = authStorage.get("openai-codex");
+          if (cred?.type === "api_key" && cred.key.trim().length > 0) {
+            return true;
+          }
+        }
+        const customProvider = chunkDJBUYADN_cjs.loadSettings().customProviders.find((entry) => {
+          return provider === chunkDJBUYADN_cjs.getCustomProviderId(entry.name);
+        });
+        if (customProvider) {
+          return true;
+        }
+        return void 0;
+      },
+      modelUseCountProvider: () => chunkDJBUYADN_cjs.loadSettings().modelUseCounts,
+      modelUseCountTracker: (modelId) => {
+        try {
+          const settings = chunkDJBUYADN_cjs.loadSettings();
+          settings.modelUseCounts[modelId] = (settings.modelUseCounts[modelId] ?? 0) + 1;
+          chunkDJBUYADN_cjs.saveSettings(settings);
+        } catch (error) {
+          console.error("Failed to persist model usage count", error);
+        }
+      },
+      customModelCatalogProvider: async () => {
         const settings = chunkDJBUYADN_cjs.loadSettings();
-        settings.modelUseCounts[modelId] = (settings.modelUseCounts[modelId] ?? 0) + 1;
-        chunkDJBUYADN_cjs.saveSettings(settings);
-      } catch (error) {
-        console.error("Failed to persist model usage count", error);
-      }
-    },
-    customModelCatalogProvider: async () => {
-      const settings = chunkDJBUYADN_cjs.loadSettings();
-      const customModels = [];
-      for (const provider of settings.customProviders) {
-        const providerId = chunkDJBUYADN_cjs.getCustomProviderId(provider.name);
-        for (const modelName of provider.models) {
-          customModels.push({
-            id: chunkDJBUYADN_cjs.toCustomProviderModelId(provider.name, modelName),
-            provider: providerId,
-            modelName,
-            hasApiKey: true,
-            apiKeyEnvVar: void 0
-          });
+        const customModels = [];
+        for (const provider of settings.customProviders) {
+          const providerId = chunkDJBUYADN_cjs.getCustomProviderId(provider.name);
+          for (const modelName of provider.models) {
+            customModels.push({
+              id: chunkDJBUYADN_cjs.toCustomProviderModelId(provider.name, modelName),
+              provider: providerId,
+              modelName,
+              hasApiKey: true,
+              apiKeyEnvVar: void 0
+            });
+          }
         }
-      }
-      try {
-        const copilotModels = await chunkDJBUYADN_cjs.getCopilotModelCatalog({ authStorage });
-        for (const m of copilotModels) {
-          customModels.push({
-            id: `github-copilot/${m.id}`,
-            provider: "github-copilot",
-            modelName: m.id,
-            hasApiKey: true,
-            apiKeyEnvVar: void 0
-          });
+        try {
+          const copilotModels = await chunkDJBUYADN_cjs.getCopilotModelCatalog({ authStorage });
+          for (const m of copilotModels) {
+            customModels.push({
+              id: `github-copilot/${m.id}`,
+              provider: "github-copilot",
+              modelName: m.id,
+              hasApiKey: true,
+              apiKeyEnvVar: void 0
+            });
+          }
+        } catch (error) {
+          console.warn("Failed to load GitHub Copilot model catalog:", error);
         }
-      } catch (error) {
-        console.warn("Failed to load GitHub Copilot model catalog:", error);
+        return customModels;
+      },
+      threadLock: crossProcessPubSub ? void 0 : {
+        acquire: chunkDJBUYADN_cjs.acquireThreadLock,
+        release: chunkDJBUYADN_cjs.releaseThreadLock
       }
-      return customModels;
     },
-    threadLock: crossProcessPubSub ? void 0 : {
-      acquire: chunkDJBUYADN_cjs.acquireThreadLock,
-      release: chunkDJBUYADN_cjs.releaseThreadLock
-    }
-  });
+    harnessV1
+  );
   if (hookManager) {
-    harness$1.subscribe((event) => {
+    harness.subscribe((event) => {
       if (event.type === "thread_changed") {
         hookManager.setSessionId(event.threadId);
       } else if (event.type === "thread_created") {
@@ -3459,12 +3672,12 @@ async function createMastraCode(config) {
       }
     });
   }
-  const omThreadStateHarness = harness$1;
+  const omThreadStateHarness = harness;
   attachOMThreadStatePersistence(omThreadStateHarness);
   await restoreOMThreadStateForCurrentThread(omThreadStateHarness).catch(() => {
   });
   return {
-    harness: harness$1,
+    harness,
     mcpManager,
     hookManager,
     signalsPubSub,
@@ -3480,5 +3693,5 @@ async function createMastraCode(config) {
 
 exports.createAuthStorage = createAuthStorage;
 exports.createMastraCode = createMastraCode;
-//# sourceMappingURL=chunk-2TQ5QO5H.cjs.map
-//# sourceMappingURL=chunk-2TQ5QO5H.cjs.map
+//# sourceMappingURL=chunk-A4TDK7QP.cjs.map
+//# sourceMappingURL=chunk-A4TDK7QP.cjs.map