mastracode 0.41.0-alpha.8 → 0.41.0

package/dist/{tui-V-RCfOzz.cjs → tui-D5UsAQOX.cjs}

@@ -35,6 +35,7 @@ let _mastra_code_sdk_onboarding_index = require("@mastra/code-sdk/onboarding/ind
 let _mastra_code_sdk_utils_update_check = require("@mastra/code-sdk/utils/update-check");
 let _earendil_works_pi_tui = require("@earendil-works/pi-tui");
 let _mastra_core_signals = require("@mastra/core/signals");
+let zod = require("zod");
 let _mastra_code_sdk_utils_slash_command_processor = require("@mastra/code-sdk/utils/slash-command-processor");
 let _mastra_code_sdk_goal_signal = require("@mastra/code-sdk/goal-signal");
 let _mastra_code_sdk_goal_manager = require("@mastra/code-sdk/goal-manager");
@@ -666,6 +667,20 @@ async function detectTerminalTheme() {
 	return { mode: "dark" };
 }
 //#endregion
+//#region src/tui/background-tool-result.ts
+const backgroundTaskMetadataSchema = zod.z.object({ mastra: zod.z.object({ backgroundTask: zod.z.object({
+	taskId: zod.z.string().min(1),
+	status: zod.z.enum([
+		"running",
+		"completed",
+		"failed"
+	])
+}) }) });
+function getBackgroundToolMetadata(providerMetadata) {
+	const parsed = backgroundTaskMetadataSchema.safeParse(providerMetadata);
+	return parsed.success ? parsed.data.mastra.backgroundTask : void 0;
+}
+//#endregion
 //#region src/tui/db-message-parts.ts
 function getParts(message) {
 	const content = message.content;
@@ -762,7 +777,9 @@ function getAssistantRenderParts(message) {
 			case "tool-invocation": {
 				const inv = part.toolInvocation;
 				const hasResult = inv.state === "result" && inv.result !== void 0;
+				const backgroundTask = getBackgroundToolMetadata(part.providerMetadata);
 				out.push({
+					...backgroundTask ? { backgroundTask } : {},
 					kind: "tool",
 					toolCallId: String(inv.toolCallId ?? ""),
 					toolName: String(inv.toolName ?? ""),
@@ -786,7 +803,9 @@ function getAssistantRenderParts(message) {
 				const legacyPart = part;
 				const toolCallId = String(legacyPart.toolCallId ?? "");
 				const call = toolCalls.get(toolCallId);
+				const backgroundTask = getBackgroundToolMetadata(part.providerMetadata);
 				out.push({
+					...backgroundTask ? { backgroundTask } : {},
 					kind: "tool",
 					toolCallId,
 					toolName: String(legacyPart.toolName ?? call?.toolName ?? ""),
@@ -7547,6 +7566,9 @@ var ToolExecutionComponentEnhanced = class extends WidthAwareContainer {
 		this.backgroundTaskId = taskId;
 		this.rebuild();
 	}
+	getBackgroundTaskId() {
+		return this.backgroundTaskId;
+	}
 	cancelBackground() {
 		this.backgroundCancelled = true;
 		this.isPartial = false;
@@ -8368,18 +8390,18 @@ var ToolExecutionComponentEnhanced = class extends WidthAwareContainer {
 			return lines;
 		};
 		if (this.result.isError) {
-			const status = theme.fg("error", " ✗");
+			const status = this.getStatusIndicator(true);
 			const output = this.streamingOutput.trim() || this.getFormattedOutput();
 			renderBorderedShell(status, this.limitQuietShellLines(prepareOutputLines(output)));
 			return;
 		}
 		if (this.getFormattedOutput().match(/Error:|TypeError:|SyntaxError:|ReferenceError:|command not found|fatal:|error:/i)) {
-			const status = theme.fg("error", " ✗");
+			const status = this.getStatusIndicator(true);
 			const output = this.streamingOutput.trim() || this.getFormattedOutput();
 			renderBorderedShell(status, this.limitQuietShellLines(prepareOutputLines(output)));
 			return;
 		}
-		const status = theme.fg("success", " ✓");
+		const status = this.getStatusIndicator(false);
 		const output = this.streamingOutput.trim() || this.getFormattedOutput();
 		renderBorderedShell(status, this.limitQuietShellLines(prepareOutputLines(output)));
 	}
@@ -8416,7 +8438,7 @@ var ToolExecutionComponentEnhanced = class extends WidthAwareContainer {
 			renderBorderedProcess(status, lines);
 			return;
 		}
-		renderBorderedProcess(this.result.isError ? theme.fg("error", " ✗") : theme.fg("success", " ✓"), prepareOutputLines(this.streamingOutput.trim() || this.getFormattedOutput()));
+		renderBorderedProcess(this.getStatusIndicator(this.result.isError), prepareOutputLines(this.streamingOutput.trim() || this.getFormattedOutput()));
 	}
 	renderEditToolEnhanced() {
 		const argsObj = this.args;
@@ -9060,16 +9082,16 @@ var ToolExecutionComponentEnhanced = class extends WidthAwareContainer {
 		}
 		return " " + theme.fg("toolArgs", parts.join(", "));
 	}
-	getBackgroundStatusIndicator() {
+	getBackgroundStatusIndicator(isError = this.isErrorResult()) {
 		if (!this.backgroundTaskId) return "";
 		if (this.backgroundCancelled) return theme.fg("muted", ` ■ background · ${this.backgroundTaskId}`);
 		if (this.isPartial) return theme.fg("warning", ` ◌ background · ${this.backgroundTaskId}`);
-		return this.isErrorResult() ? theme.fg("error", ` ✗ background · ${this.backgroundTaskId}`) : theme.fg("success", ` ✓ background · ${this.backgroundTaskId}`);
+		return isError ? theme.fg("error", ` ✗ background · ${this.backgroundTaskId}`) : theme.fg("success", ` ✓ background · ${this.backgroundTaskId}`);
 	}
-	getStatusIndicator() {
-		const backgroundStatus = this.getBackgroundStatusIndicator();
+	getStatusIndicator(isError = this.isErrorResult()) {
+		const backgroundStatus = this.getBackgroundStatusIndicator(isError);
 		if (backgroundStatus) return backgroundStatus;
-		return this.isPartial ? theme.fg("muted", " ⋯") : this.isErrorResult() ? theme.fg("error", " ✗") : theme.fg("success", " ✓");
+		return this.isPartial ? theme.fg("muted", " ⋯") : isError ? theme.fg("error", " ✗") : theme.fg("success", " ✓");
 	}
 	getDurationSuffix() {
 		if (this.isPartial) return "";
@@ -9333,12 +9355,6 @@ var PendingUserMessageComponent = class extends _earendil_works_pi_tui.Container
 	}
 };
 //#endregion
-//#region src/tui/background-tool-result.ts
-const BACKGROUND_TOOL_TASK_ID_PATTERN = /^Background task started\. Task ID: ([^.\s]+)/;
-function parseBackgroundToolTaskId(result) {
-	return BACKGROUND_TOOL_TASK_ID_PATTERN.exec(result)?.[1];
-}
-//#endregion
 //#region src/tui/components/tool-approval-dialog.ts
 /**
 * Tool approval dialog component.
@@ -9792,12 +9808,6 @@ function ensureSubmitPlanComponent(ctx, toolCallId, args) {
 function isToolResultError(result) {
 	return typeof result === "object" && result !== null && result.isError === true;
 }
-function getBackgroundToolTaskId(result) {
-	return parseBackgroundToolTaskId(formatToolResult(result));
-}
-function isBackgroundToolPlaceholder(result) {
-	return getBackgroundToolTaskId(result) !== void 0;
-}
 function formatToolResult(result) {
 	if (result === null || result === void 0) return "";
 	if (typeof result === "string") return sanitizeAnsiForRendering(result);
@@ -10097,18 +10107,17 @@ function handleToolInputEnd(ctx, toolCallId) {
 	flushLatestParsedToolArgs(ctx, toolCallId);
 	closeToolInputParser(toolCallId);
 }
-function handleToolEnd(ctx, toolCallId, result, isError) {
+function handleToolEnd(ctx, toolCallId, result, isError, providerMetadata) {
 	flushPendingShellOutput(ctx, toolCallId);
 	const { state } = ctx;
+	const background = state.options?.backgroundToolsEnabled ? getBackgroundToolMetadata(providerMetadata) : void 0;
 	const subagentComponent = state.pendingSubagents.get(toolCallId);
 	if (subagentComponent) {
 		const resultText = formatToolResult(result);
 		if (pluginSubagentToolCallIds.has(toolCallId)) {
-			const backgroundTaskId = state.options?.backgroundToolsEnabled && !isError ? getBackgroundToolTaskId(result) : void 0;
-			if (backgroundTaskId) {
-				subagentComponent.setBackgroundTaskId(backgroundTaskId);
-				flushRender(state);
-			} else {
+			if (background) subagentComponent.setBackgroundTaskId(background.taskId);
+			if (background?.status === "running" && !isError) flushRender(state);
+			else {
 				subagentComponent.finish(isError, 0, resultText);
 				state.pendingSubagents.delete(toolCallId);
 				pluginSubagentToolCallIds.delete(toolCallId);
@@ -10127,9 +10136,8 @@ function handleToolEnd(ctx, toolCallId, result, isError) {
 			state.allToolComponents.push(component);
 		}
 		const resultText = formatToolResult(result);
-		const backgroundTaskId = state.options?.backgroundToolsEnabled && !effectiveIsError ? getBackgroundToolTaskId(result) : void 0;
-		const isBackgroundPlaceholder = backgroundTaskId !== void 0;
-		if (backgroundTaskId) component.setBackgroundTaskId?.(backgroundTaskId);
+		const isBackgroundPlaceholder = background?.status === "running" && !effectiveIsError;
+		if (background) component.setBackgroundTaskId?.(background.taskId);
 		const toolResult = {
 			content: [{
 				type: "text",
@@ -10857,7 +10865,7 @@ async function renderExistingMessages(state) {
 			const hasResult = part.hasResult;
 			const resultValue = part.result;
 			const resultIsError = part.isError;
-			const isBackgroundPlaceholder = !!state.options?.backgroundToolsEnabled && hasResult && !resultIsError && isBackgroundToolPlaceholder(resultValue);
+			const isBackgroundPlaceholder = !!state.options?.backgroundToolsEnabled && hasResult && !resultIsError && part.backgroundTask?.status === "running" && !backgroundTasksByToolCallId.has(part.toolCallId);
 			if (toolName === "subagent") {
 				const subArgs = toolArgs;
 				const rawResult = hasResult ? formatToolResult(resultValue) : void 0;
@@ -10905,7 +10913,7 @@ async function renderExistingMessages(state) {
 					colors: pluginRenderConfig.colors,
 					icons: pluginRenderConfig.icons
 				});
-				const backgroundTaskId = (isBackgroundPlaceholder ? getBackgroundToolTaskId(resultValue) : void 0) ?? backgroundTasksByToolCallId.get(part.toolCallId);
+				const backgroundTaskId = (isBackgroundPlaceholder || part.backgroundTask?.status !== "running" ? part.backgroundTask?.taskId : void 0) ?? backgroundTasksByToolCallId.get(part.toolCallId);
 				if (backgroundTaskId) subComponent.setBackgroundTaskId(backgroundTaskId);
 				if (cancelledBackgroundToolCalls.has(part.toolCallId)) subComponent.cancel();
 				else if (isBackgroundPlaceholder) state.pendingSubagents.set(part.toolCallId, subComponent);
@@ -10918,7 +10926,7 @@ async function renderExistingMessages(state) {
 				showImages: false,
 				collapsedByDefault: !state.toolOutputExpanded
 			}, state.ui);
-			const backgroundTaskId = (isBackgroundPlaceholder ? getBackgroundToolTaskId(resultValue) : void 0) ?? backgroundTasksByToolCallId.get(part.toolCallId);
+			const backgroundTaskId = (isBackgroundPlaceholder || part.backgroundTask?.status !== "running" ? part.backgroundTask?.taskId : void 0) ?? backgroundTasksByToolCallId.get(part.toolCallId);
 			if (backgroundTaskId) toolComponent.setBackgroundTaskId(backgroundTaskId);
 			if (hasResult) toolComponent.updateResult({
 				content: [{
@@ -10927,8 +10935,16 @@ async function renderExistingMessages(state) {
 				}],
 				isError: resultIsError
 			}, isBackgroundPlaceholder);
-			if (cancelledBackgroundToolCalls.has(part.toolCallId)) toolComponent.cancelBackground();
-			else if (isBackgroundPlaceholder) state.pendingTools.set(part.toolCallId, toolComponent);
+			if (cancelledBackgroundToolCalls.has(part.toolCallId)) {
+				toolComponent.updateResult({
+					content: [{
+						type: "text",
+						text: "Background execution cancelled."
+					}],
+					isError: true
+				}, true);
+				toolComponent.cancelBackground();
+			} else if (isBackgroundPlaceholder) state.pendingTools.set(part.toolCallId, toolComponent);
 			let replacedWithInline = false;
 			if (isTaskMutationTool(toolName) && hasResult && !resultIsError) {
 				hasReplayedTaskState = true;
@@ -11187,6 +11203,11 @@ async function handleNewCommand(ctx) {
 	state.globalBackgroundNotice.setActivities([]);
 	setCurrentThreadTitle(state, void 0);
 	disposeAssistantRenderState(state);
+	state.gradientAnimator?.stop();
+	state.githubPrGradientAnimator?.stop();
+	state.githubPrPollingActive = false;
+	state.agentRunStartedAt = void 0;
+	state.agentRunLastStreamPartAt = void 0;
 	state.chatContainer.clear();
 	state.pendingTools.clear();
 	state.pendingTaskToolIds?.clear();
@@ -18280,8 +18301,8 @@ function showStatus(ctx) {
 		lines.push("");
 		lines.push(theme.fg("dim", "  Run /observability connect to re-enter credentials."));
 	} else {
-		const envToken = process.env.MASTRA_CLOUD_ACCESS_TOKEN;
-		const envProject = process.env.MASTRA_PROJECT_ID;
+		const envToken = process.env.MASTRACODE_CLOUD_ACCESS_TOKEN?.trim();
+		const envProject = process.env.MASTRACODE_PROJECT_ID?.trim();
 		if (envToken) {
 			lines.push(`  ${theme.fg("success", "●")} Connected ${theme.fg("dim", "(via environment variables)")}`);
 			if (envProject) lines.push(`  Project:    ${envProject}`);
@@ -20559,16 +20580,19 @@ function handleMessageEnd(ctx, message) {
 		});
 		if (stopReason === "aborted" || stopReason === "error") {
 			const abortMessage = errorMessage || "Operation aborted";
-			for (const [, component] of state.pendingTools) component.updateResult({
-				content: [{
-					type: "text",
-					text: abortMessage
-				}],
-				isError: true
-			}, false);
+			for (const [toolCallId, component] of state.pendingTools) {
+				if (component.getBackgroundTaskId?.()) continue;
+				component.updateResult({
+					content: [{
+						type: "text",
+						text: abortMessage
+					}],
+					isError: true
+				}, false);
+				state.pendingTools.delete(toolCallId);
+				state.pendingTaskToolIds?.delete(toolCallId);
+			}
 			reconcileChatBoundarySpacers(state.chatContainer);
-			state.pendingTools.clear();
-			state.pendingTaskToolIds?.clear();
 		}
 		state.assistantRenderRegistry.finalize(message.id);
 		finalizeStreamingAssistant(state);
@@ -23618,6 +23642,7 @@ function applyMessageUpdate(message, update) {
 	};
 }
 async function dispatchEvent(event, ectx, state) {
+	if ("toolCallId" in event && "threadId" in event && event.threadId && (state.pendingNewThread || event.threadId !== state.session.thread.getId())) return;
 	switch (event.type) {
 		case "agent_start":
 			clearToolInputParsers();
@@ -23673,7 +23698,7 @@ async function dispatchEvent(event, ectx, state) {
 		case "tool_start":
 			state.agentRunLastStreamPartAt = Date.now();
 			if (state.options.backgroundToolsEnabled) {
-				const threadId = state.session.thread.getId();
+				const threadId = event.threadId ?? state.session.thread.getId();
 				if (threadId) state.backgroundToolContexts.set(event.toolCallId, {
 					toolName: event.toolName,
 					resourceId: state.session.identity.getResourceId(),
@@ -23716,7 +23741,8 @@ async function dispatchEvent(event, ectx, state) {
 		case "tool_end":
 			state.agentRunLastStreamPartAt = Date.now();
 			if (state.options.backgroundToolsEnabled) {
-				const taskId = getBackgroundToolTaskId(event.result);
+				const background = getBackgroundToolMetadata(event.providerMetadata);
+				const taskId = !event.isError && background?.status === "running" ? background.taskId : void 0;
 				const context = state.backgroundToolContexts.get(event.toolCallId);
 				if (taskId && context) {
 					acceptBackgroundActivity(state.backgroundActivities, taskId, event.toolCallId, context);
@@ -23726,7 +23752,7 @@ async function dispatchEvent(event, ectx, state) {
 				}
 				if (!taskId) state.backgroundToolContexts.delete(event.toolCallId);
 			}
-			handleToolEnd(ectx, event.toolCallId, event.result, event.isError);
+			handleToolEnd(ectx, event.toolCallId, event.result, event.isError, event.providerMetadata);
 			break;
 		case "info":
 			ectx.showInfo(event.message);
@@ -23743,6 +23769,7 @@ async function dispatchEvent(event, ectx, state) {
 		case "thread_changed": {
 			ectx.showInfo(`Switched to thread: ${event.threadId}`);
 			state.latestRequestPromptTokens = void 0;
+			state.backgroundToolContexts?.clear();
 			await state.session.state.set({
 				tasks: [],
 				activePlan: null,
@@ -23777,6 +23804,7 @@ async function dispatchEvent(event, ectx, state) {
 		case "thread_created": {
 			ectx.showInfo(`Created thread: ${event.thread.id}`);
 			state.latestRequestPromptTokens = void 0;
+			state.backgroundToolContexts?.clear();
 			setCurrentThreadTitle(state, event.thread.title);
 			state.activeGithubPrSubscriptions = getGithubPrSubscriptionsFromMetadata(event.thread.metadata);
 			state.githubPrPollingActive = false;
@@ -26978,4 +27006,4 @@ Object.defineProperty(exports, "theme", {
 	}
 });
 
-//# sourceMappingURL=tui-V-RCfOzz.cjs.map
+//# sourceMappingURL=tui-D5UsAQOX.cjs.map