mastracode 0.31.1-alpha.0 → 0.31.1-alpha.2

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # mastracode
 
+## 0.31.1-alpha.2
+
+### Patch Changes
+
+- Fixed a stuck ask_user question box after the underlying run failed: the prompt is now dismissed automatically instead of accepting an answer that could never be delivered. ([#19444](https://github.com/mastra-ai/mastra/pull/19444))
+
+- Updated dependencies [[`8b20926`](https://github.com/mastra-ai/mastra/commit/8b20926cd59e2ba3d66458e062fa0e6e2ada3e68), [`f9717e4`](https://github.com/mastra-ai/mastra/commit/f9717e4a381500042d088577347a787b0ec8caff), [`74faf8b`](https://github.com/mastra-ai/mastra/commit/74faf8bd9c1018f2492653c06b1e25fc8300e9e6), [`1fadac4`](https://github.com/mastra-ai/mastra/commit/1fadac44537caeefe81f9f775ae2f2f3d94e9069), [`970c032`](https://github.com/mastra-ai/mastra/commit/970c032502751ee5dd4d0b603331d9838cb538fc), [`792ec9a`](https://github.com/mastra-ai/mastra/commit/792ec9a0869bab8274cf5e0ed2840738737a1607), [`712b864`](https://github.com/mastra-ai/mastra/commit/712b864aa1ed12b14c54390ec17b69de163c37f7), [`8f7a5de`](https://github.com/mastra-ai/mastra/commit/8f7a5dedc246cdc938bb65516703cf9b27b03756), [`c0bec73`](https://github.com/mastra-ai/mastra/commit/c0bec732c93d1a22ae5e51ed66cf8cacca8bd6a6)]:
+  - @mastra/core@1.52.0-alpha.2
+  - @mastra/code-sdk@0.2.0-alpha.2
+  - @mastra/mcp@1.15.0-alpha.0
+
+## 0.31.1-alpha.1
+
+### Patch Changes
+
+- Updated dependencies:
+  - @mastra/core@1.51.1-alpha.1
+  - @mastra/code-sdk@0.1.1-alpha.1
+
 ## 0.31.1-alpha.0
 
 ### Patch Changes

package/dist/{chunk-DOIRTZT6.js → chunk-EDKUI55Y.js}

@@ -6,9 +6,11 @@ import { loadSettings, MASTRA_GATEWAY_PROVIDER, resolveThreadActiveModelPackId,
 import { fetchLatestVersion, isNewerVersion, detectPackageManager, fetchChangelog, runUpdate, getInstallCommand } from '@mastra/code-sdk/utils/update-check';
 import { visibleWidth, Box, Text, Spacer, Input, Container, fuzzyFilter, getKeybindings, Markdown, ProcessTerminal, TUI, truncateToWidth, Editor, matchesKey, CombinedAutocompleteProvider, SelectList, wrapTextWithAnsi, SettingsList, parseKey } from '@earendil-works/pi-tui';
 import { processSlashCommand } from '@mastra/code-sdk/utils/slash-command-processor';
+import { createGoalReminderSignal } from '@mastra/code-sdk/goal-signal';
 import { mastraBrand } from '@mastra/code-sdk/theme-palette';
 export { mastraBrand } from '@mastra/code-sdk/theme-palette';
 import chalk15 from 'chalk';
+import { GoalManager, DEFAULT_MAX_TURNS } from '@mastra/code-sdk/goal-manager';
 import { resolveSTTModel, sttModelsForProvider, sttProviders, defaultModelForProvider, DEFAULT_STT_MODEL } from '@mastra/code-sdk/voice/stt-registry';
 import { Readable } from 'stream';
 import { DeepgramVoice } from '@mastra/voice-deepgram';
@@ -818,276 +820,6 @@ var ModelSelectorComponent = class extends Box {
     return this.searchInput;
   }
 };
-var DEFAULT_MAX_TURNS = 50;
-var THREAD_GOAL_KEY = "goal";
-var GoalManager = class {
-  /** Synchronous in-memory view of the active objective record (source of truth is ThreadState). */
-  record = null;
-  /** Display-only active-timer accounting (not persisted to the objective record). */
-  activeStartedAt = null;
-  activeDurationMs = 0;
-  persistGoalOnNextThreadCreate = false;
-  // ---------------------------------------------------------------------------
-  // Synchronous TUI surface
-  // ---------------------------------------------------------------------------
-  getGoal() {
-    if (!this.record) return null;
-    const { judgeModelId, maxTurns } = this.effectiveSettings(this.record);
-    return {
-      id: this.record.id,
-      objective: this.record.objective,
-      status: this.record.status,
-      turnsUsed: this.record.runsUsed,
-      maxTurns,
-      judgeModelId,
-      startedAt: new Date(this.record.startedAt).toISOString(),
-      activeStartedAt: this.activeStartedAt ?? void 0,
-      activeDurationMs: this.activeDurationMs
-    };
-  }
-  isActive() {
-    return this.record?.status === "active";
-  }
-  persistOnNextThreadCreate() {
-    this.persistGoalOnNextThreadCreate = true;
-  }
-  consumePersistOnNextThreadCreate() {
-    if (!this.persistGoalOnNextThreadCreate) return false;
-    this.persistGoalOnNextThreadCreate = false;
-    return true;
-  }
-  startActiveTimer() {
-    if (this.record?.status === "active" && !this.activeStartedAt) {
-      this.activeStartedAt = (/* @__PURE__ */ new Date()).toISOString();
-    }
-  }
-  stopActiveTimer() {
-    if (!this.activeStartedAt) return;
-    const startedMs = Date.parse(this.activeStartedAt);
-    if (Number.isFinite(startedMs)) {
-      this.activeDurationMs += Math.max(0, Date.now() - startedMs);
-    }
-    this.activeStartedAt = null;
-  }
-  /** Reset active-timer accounting to zero (e.g. for an untriggered plan goal). */
-  resetActiveTimer() {
-    this.activeStartedAt = null;
-    this.activeDurationMs = 0;
-  }
-  // ---------------------------------------------------------------------------
-  // Objective lifecycle (ThreadState-backed via the agent)
-  // ---------------------------------------------------------------------------
-  /**
-   * Set a new objective. Persists to ThreadState via `agent.setObjective` and
-   * updates the in-memory view. Only the provided settings are persisted into
-   * the record; unset ones fall back to the agent's `goal` config at read time.
-   */
-  async setGoal(state, objective, judgeModelId, maxTurns = DEFAULT_MAX_TURNS) {
-    const threadId = state.session.thread.getId();
-    const agent = this.getAgent(state);
-    const now = Date.now();
-    const id = randomUUID();
-    if (agent && threadId) {
-      const persisted = await agent.setObjective(objective, {
-        id,
-        threadId,
-        resourceId: state.session.identity.getResourceId(),
-        ...judgeModelId ? { judgeModelId } : {},
-        maxRuns: maxTurns
-      });
-      this.record = persisted ? { ...persisted, id: persisted.id ?? id } : this.localRecord(objective, judgeModelId, maxTurns, now, id);
-    } else {
-      this.record = this.localRecord(objective, judgeModelId, maxTurns, now, id);
-    }
-    this.activeStartedAt = new Date(now).toISOString();
-    this.activeDurationMs = 0;
-    return this.getGoal();
-  }
-  /**
-   * Update the judge model / max-runs defaults. Persists into the active record
-   * (so the override is remembered in thread state) when a goal is set.
-   */
-  async updateJudgeDefaults(state, judgeModelId, maxTurns) {
-    if (!this.record) return null;
-    const threadId = state.session.thread.getId();
-    const agent = this.getAgent(state);
-    if (agent && threadId) {
-      const updated = await agent.updateObjectiveOptions({
-        threadId,
-        ...judgeModelId ? { judgeModelId } : {},
-        maxRuns: maxTurns
-      });
-      if (updated) this.record = { ...updated, id: this.record.id };
-    } else {
-      this.record = {
-        ...this.record,
-        ...judgeModelId ? { judgeModelId } : {},
-        maxRuns: maxTurns,
-        updatedAt: Date.now()
-      };
-    }
-    return this.getGoal();
-  }
-  pause(reason) {
-    if (this.record && this.record.status === "active") {
-      this.stopActiveTimer();
-      this.record = { ...this.record, status: "paused", pausedReason: reason, updatedAt: Date.now() };
-    }
-    return this.getGoal();
-  }
-  resume() {
-    if (this.record && this.record.status === "paused") {
-      this.record = { ...this.record, status: "active", pausedReason: void 0, updatedAt: Date.now() };
-      this.startActiveTimer();
-    }
-    return this.getGoal();
-  }
-  markDone() {
-    if (this.record) {
-      this.stopActiveTimer();
-      this.record = { ...this.record, status: "done", updatedAt: Date.now() };
-    }
-  }
-  clear() {
-    this.record = null;
-    this.activeStartedAt = null;
-    this.activeDurationMs = 0;
-    this.persistGoalOnNextThreadCreate = false;
-  }
-  /**
-   * Sync the latest objective record from ThreadState into the in-memory view.
-   * Called from the `goal` stream-chunk handler after each evaluation.
-   */
-  applyEvaluation(update) {
-    if (!this.record) return null;
-    this.record = { ...this.record, runsUsed: update.runsUsed, status: update.status, updatedAt: Date.now() };
-    if (update.status !== "active") this.stopActiveTimer();
-    return this.getGoal();
-  }
-  // ---------------------------------------------------------------------------
-  // Persistence
-  // ---------------------------------------------------------------------------
-  /**
-   * Persist the active objective to ThreadState via the agent. The objective
-   * record is the source of truth; the legacy thread-metadata key is cleared so
-   * stale state from older sessions does not resurface.
-   */
-  async saveToThread(state) {
-    const threadId = state.session.thread.getId();
-    const agent = this.getAgent(state);
-    try {
-      if (agent && threadId) {
-        if (this.record) {
-          const updated = await agent.updateObjectiveOptions({
-            threadId,
-            status: this.record.status,
-            ...this.record.pausedReason ? { pausedReason: this.record.pausedReason } : {},
-            ...this.record.judgeModelId ? { judgeModelId: this.record.judgeModelId } : {},
-            ...this.record.maxRuns !== void 0 ? { maxRuns: this.record.maxRuns } : {}
-          });
-          if (!updated) {
-            const desiredStatus = this.record.status;
-            await agent.setObjective(this.record.objective, {
-              id: this.record.id,
-              threadId,
-              resourceId: state.session.identity.getResourceId(),
-              ...this.record.judgeModelId ? { judgeModelId: this.record.judgeModelId } : {},
-              ...this.record.maxRuns !== void 0 ? { maxRuns: this.record.maxRuns } : {}
-            });
-            if (desiredStatus !== "active") {
-              await agent.updateObjectiveOptions({
-                threadId,
-                status: desiredStatus,
-                ...this.record.pausedReason ? { pausedReason: this.record.pausedReason } : {}
-              });
-            }
-          }
-        } else {
-          await agent.clearObjective({ threadId });
-        }
-      }
-      await state.session.thread.setSetting({ key: THREAD_GOAL_KEY, value: void 0 });
-    } catch {
-    }
-  }
-  /**
-   * Load the objective from ThreadState (called on thread switch). Falls back to
-   * the legacy thread-metadata goal for threads created before this migration.
-   */
-  async loadFromThread(state) {
-    this.persistGoalOnNextThreadCreate = false;
-    this.activeStartedAt = null;
-    this.activeDurationMs = 0;
-    const threadId = state.session.thread.getId();
-    const agent = this.getAgent(state);
-    if (agent && threadId) {
-      try {
-        const record = await agent.getObjective({ threadId });
-        if (record) {
-          this.record = { ...record, id: record.id ?? randomUUID() };
-          return;
-        }
-      } catch {
-      }
-    }
-    this.record = null;
-  }
-  /**
-   * Legacy entry point retained for thread-switch call sites that only have the
-   * thread metadata available. Hydrates from a previously-persisted GoalState.
-   */
-  loadFromThreadMetadata(metadata) {
-    const saved = metadata?.[THREAD_GOAL_KEY];
-    this.persistGoalOnNextThreadCreate = false;
-    this.activeStartedAt = null;
-    this.activeDurationMs = 0;
-    if (saved && saved.objective && saved.status) {
-      this.record = {
-        objective: saved.objective,
-        status: saved.status,
-        runsUsed: saved.turnsUsed ?? 0,
-        maxRuns: saved.maxTurns ?? DEFAULT_MAX_TURNS,
-        judgeModelId: saved.judgeModelId ?? "",
-        startedAt: saved.startedAt ? Date.parse(saved.startedAt) || Date.now() : Date.now(),
-        updatedAt: Date.now(),
-        id: saved.id ?? randomUUID()
-      };
-      this.activeDurationMs = saved.activeDurationMs ?? 0;
-    } else {
-      this.record = null;
-    }
-  }
-  // ---------------------------------------------------------------------------
-  // Private
-  // ---------------------------------------------------------------------------
-  getAgent(state) {
-    try {
-      return state.controller.getCurrentAgent(state.session);
-    } catch {
-      return void 0;
-    }
-  }
-  /** Resolve effective judge model + max runs (record value → settings default). */
-  effectiveSettings(record) {
-    const settings = loadSettings();
-    return {
-      judgeModelId: record.judgeModelId ?? settings.models.goalJudgeModel ?? "",
-      maxTurns: record.maxRuns ?? settings.models.goalMaxTurns ?? DEFAULT_MAX_TURNS
-    };
-  }
-  localRecord(objective, judgeModelId, maxTurns, now, id) {
-    return {
-      objective,
-      status: "active",
-      runsUsed: 0,
-      maxRuns: maxTurns,
-      ...judgeModelId ? { judgeModelId } : {},
-      startedAt: now,
-      updatedAt: now,
-      id
-    };
-  }
-};
 function parseSizeValue(value, total) {
   if (typeof value === "number") return value;
   const match = value?.match(/^(\d+(?:\.\d+)?)%$/);
@@ -1527,18 +1259,6 @@ async function startGoal(ctx, objective, judgeModelId, maxTurns, options = {}) {
     ctx.showError(`Goal paused \u2014 failed to start: ${err instanceof Error ? err.message : String(err)}`);
   }
 }
-function createGoalReminderSignal(goal) {
-  return {
-    type: "system-reminder",
-    contents: goal.objective,
-    attributes: { type: "goal" },
-    metadata: {
-      goalId: goal.id,
-      maxTurns: goal.maxTurns,
-      judgeModelId: goal.judgeModelId
-    }
-  };
-}
 
 // src/tui/components/help-overlay.ts
 function getCommands(modes) {
@@ -3854,6 +3574,17 @@ var AskQuestionInlineComponent = class _AskQuestionInlineComponent extends Conta
     this.borderedBox.setCancelled();
     this.onCancel?.();
   }
+  /**
+   * Retract the question without invoking any submit/cancel callback. Used when
+   * the underlying tool suspension was cancelled (the suspended run died, e.g.
+   * its snapshot failed to persist), so an answer could never be delivered.
+   */
+  dismiss() {
+    if (this.answered) return;
+    this.answered = true;
+    this.focused = false;
+    this.borderedBox.setCancelled();
+  }
   handleInput(data) {
     if (this.answered) return;
     if (this.selectList) {
@@ -16372,10 +16103,39 @@ function processNextInlineQuestion(state) {
 async function handleAskQuestion(ctx, toolCallId, question, options, selectionMode) {
   const { state } = ctx;
   return new Promise((resolve) => {
+    let cancelled = false;
+    let retract = () => {
+    };
+    const unsubscribeCancel = state.session.subscribe?.((event) => {
+      if (event.type !== "tool_suspension_cancelled" || event.toolCallId !== toolCallId) return;
+      if (cancelled) return;
+      cancelled = true;
+      unsubscribeCancel();
+      retract();
+      resolve();
+    }) ?? (() => {
+    });
+    const finish = () => {
+      unsubscribeCancel();
+      resolve();
+    };
     if (state.options.inlineQuestions) {
       const askUserComponent = state.pendingAskUserComponents?.get(toolCallId) ?? state.lastAskUserComponent;
       state.pendingAskUserComponents?.delete(toolCallId);
+      let shownComponent = askUserComponent;
+      retract = () => {
+        shownComponent?.dismiss();
+        if (state.activeInlineQuestion === shownComponent) {
+          state.activeInlineQuestion = void 0;
+          processNextInlineQuestion(state);
+        }
+        state.ui.requestRender();
+      };
       const activate = () => {
+        if (cancelled) {
+          processNextInlineQuestion(state);
+          return;
+        }
         try {
           let questionComponent;
           if (askUserComponent) {
@@ -16388,19 +16148,19 @@ async function handleAskQuestion(ctx, toolCallId, question, options, selectionMo
               onSubmit: (answer) => {
                 state.activeInlineQuestion = void 0;
                 state.session.respondToToolSuspension({ toolCallId, resumeData: answer });
-                resolve();
+                finish();
                 processNextInlineQuestion(state);
               },
               onSubmitMulti: (answers) => {
                 state.activeInlineQuestion = void 0;
                 state.session.respondToToolSuspension({ toolCallId, resumeData: answers });
-                resolve();
+                finish();
                 processNextInlineQuestion(state);
               },
               onCancel: () => {
                 state.activeInlineQuestion = void 0;
                 state.session.respondToToolSuspension({ toolCallId, resumeData: "(skipped)" });
-                resolve();
+                finish();
                 processNextInlineQuestion(state);
               }
             });
@@ -16415,19 +16175,19 @@ async function handleAskQuestion(ctx, toolCallId, question, options, selectionMo
                 onSubmit: (answer) => {
                   state.activeInlineQuestion = void 0;
                   state.session.respondToToolSuspension({ toolCallId, resumeData: answer });
-                  resolve();
+                  finish();
                   processNextInlineQuestion(state);
                 },
                 onSubmitMulti: (answers) => {
                   state.activeInlineQuestion = void 0;
                   state.session.respondToToolSuspension({ toolCallId, resumeData: answers });
-                  resolve();
+                  finish();
                   processNextInlineQuestion(state);
                 },
                 onCancel: () => {
                   state.activeInlineQuestion = void 0;
                   state.session.respondToToolSuspension({ toolCallId, resumeData: "(skipped)" });
-                  resolve();
+                  finish();
                   processNextInlineQuestion(state);
                 }
               },
@@ -16435,6 +16195,7 @@ async function handleAskQuestion(ctx, toolCallId, question, options, selectionMo
             );
             state.chatContainer.addChild(questionComponent);
           }
+          shownComponent = questionComponent;
           state.activeInlineQuestion = questionComponent;
           state.ui.requestRender();
           state.chatContainer.invalidate();
@@ -16442,7 +16203,7 @@ async function handleAskQuestion(ctx, toolCallId, question, options, selectionMo
         } catch {
           state.activeInlineQuestion = void 0;
           state.session.respondToToolSuspension({ toolCallId, resumeData: "(skipped)" });
-          resolve();
+          finish();
           processNextInlineQuestion(state);
         }
       };
@@ -16452,6 +16213,10 @@ async function handleAskQuestion(ctx, toolCallId, question, options, selectionMo
         activate();
       }
     } else {
+      retract = () => {
+        state.ui.hideOverlay();
+        state.ui.requestRender();
+      };
       const dialog = new AskQuestionDialogComponent({
         question,
         options,
@@ -16461,17 +16226,17 @@ async function handleAskQuestion(ctx, toolCallId, question, options, selectionMo
         onSubmit: (answer) => {
           state.ui.hideOverlay();
           state.session.respondToToolSuspension({ toolCallId, resumeData: answer });
-          resolve();
+          finish();
         },
         onSubmitMulti: (answers) => {
           state.ui.hideOverlay();
           state.session.respondToToolSuspension({ toolCallId, resumeData: answers });
-          resolve();
+          finish();
         },
         onCancel: () => {
           state.ui.hideOverlay();
           state.session.respondToToolSuspension({ toolCallId, resumeData: "(skipped)" });
-          resolve();
+          finish();
         }
       });
       showModalOverlay(state.ui, dialog, { widthPercent: 0.7 });
@@ -21791,5 +21556,5 @@ Would you like to update now?`;
 };
 
 export { AssistantMessageComponent, LoginDialogComponent, LoginSelectorComponent, MastraTUI, ModelSelectorComponent, OMProgressComponent, ToolExecutionComponentEnhanced, UserMessageComponent, applyThemeMode, createTUIState, detectTerminalTheme, formatOMStatus, getEditorTheme, getMarkdownTheme, getThemeMode, mastra, restoreTerminalForeground, theme };
-//# sourceMappingURL=chunk-DOIRTZT6.js.map
-//# sourceMappingURL=chunk-DOIRTZT6.js.map
+//# sourceMappingURL=chunk-EDKUI55Y.js.map
+//# sourceMappingURL=chunk-EDKUI55Y.js.map