@markusylisiurunen/tau 0.3.36 → 0.3.38

package/dist/tui/session_chat_controller.js

@@ -1,4 +1,3 @@
-import { randomUUID } from "node:crypto";
 import { dirname } from "node:path/posix";
 import { createCommandRegistry, } from "../core/commands/index.js";
 import { getGoogleApiKey } from "../core/config/index.js";
@@ -26,7 +25,6 @@ import { formatDurationMs, formatSessionCost } from "./chat_controller/status_fo
 import { copyTextToClipboard } from "./clipboard.js";
 import { DOUBLE_PRESS_WINDOW_MS } from "./constants.js";
 import { cleanupListenTempFile, createListenTempFilePath, getSpeechToTextApiKey, getSpeechToTextApiKeyErrorMessage, LISTEN_RECORDING_MAX_DURATION_MS, LISTEN_RECORDING_MIN_BYTES, readListenAudio, startListenAudioCapture, transcribeListenAudio, } from "./listen_capture.js";
-import { formatDefaultMemoryModeFilePath, formatMemoryModeUserMessage } from "./memory_mode.js";
 import { createSdkDiffSnapshotDeps, createSdkToolExecutionBackend, } from "./session_tool_execution_backend.js";
 import { runSpeechPlaybackTask } from "./speech_playback.js";
 const CAFFEINATE_COMMAND = "/usr/bin/caffeinate";
@@ -35,6 +33,7 @@ export class SessionChatController {
     session;
     createSession;
     targetLabel;
+    configuredClientToolNames;
     config;
     defaultDiffTool;
     diffToolLauncher;
@@ -47,7 +46,6 @@ export class SessionChatController {
     snapshot;
     renderedMessageIds = [];
     hiddenHistoryEntryIds = new Set();
-    clientRenderedUserMessages = new Map();
     observedSessionRevision;
     eventUnsubscribe;
     pendingUserMessagesUnsubscribe;
@@ -62,7 +60,6 @@ export class SessionChatController {
     sessionReplacementInProgress = false;
     isBashMode = false;
     isBashIncognito = false;
-    isMemoryMode = false;
     showThinking = false;
     compactToolUi = true;
     commandHint;
@@ -85,6 +82,7 @@ export class SessionChatController {
         this.createSession = options.createSession;
         this.snapshot = options.snapshot;
         this.targetLabel = options.targetLabel;
+        this.configuredClientToolNames = options.configuredClientToolNames ?? [];
         this.config = options.config ?? {};
         this.defaultDiffTool = options.defaultDiffTool;
         this.diffToolLauncher = options.diffToolLauncher;
@@ -269,20 +267,7 @@ export class SessionChatController {
             }
             return;
         }
-        if (this.isSingleLineInput(text) && trimmed.startsWith("#")) {
-            const request = trimmed.slice(1).trim();
-            if (!request) {
-                this.view.addSystemMessage("memory mode request was empty.", "warn");
-                return;
-            }
-            const agentsFilePath = formatDefaultMemoryModeFilePath(this.snapshot.executionEnvironment.cwd);
-            await this.submitUserText(request, {
-                textForModel: formatMemoryModeUserMessage(agentsFilePath, request),
-                kind: "memory",
-            });
-            return;
-        }
-        await this.submitUserText(trimmed);
+        await this.runSessionTurn(() => this.session.submit(trimmed));
     }
     async onUserInput(text) {
         await this.handleSubmit(text);
@@ -302,21 +287,17 @@ export class SessionChatController {
     handleEditorChange(text) {
         const wasBash = this.isBashMode;
         const wasBashIncognito = this.isBashIncognito;
-        const wasMemory = this.isMemoryMode;
         const previousCommandHint = this.commandHint;
         if (text.trim().length > 0) {
             this.lastEmptySubmitAt = undefined;
         }
         const trimmed = text.trimStart();
-        const isSingleLine = this.isSingleLineInput(text);
         const isIncognito = trimmed.startsWith("!!");
         this.isBashIncognito = isIncognito;
         this.isBashMode = trimmed.startsWith("!") && !isIncognito;
-        this.isMemoryMode = isSingleLine && trimmed.startsWith("#");
         this.commandHint = this.getCommandHintForInput(text);
         if (wasBash !== this.isBashMode ||
             wasBashIncognito !== this.isBashIncognito ||
-            wasMemory !== this.isMemoryMode ||
             previousCommandHint !== this.commandHint) {
             this.refreshStatus();
         }
@@ -357,6 +338,10 @@ export class SessionChatController {
         if (this.snapshot.catalog.skills.length > 0) {
             parts.push(`${this.snapshot.catalog.skills.length} skills`);
         }
+        if (this.configuredClientToolNames.length > 0) {
+            const count = this.configuredClientToolNames.length;
+            parts.push(`${count} client tool${count === 1 ? "" : "s"}`);
+        }
         return parts.join(" · ");
     }
     buildStartupIntroBody() {
@@ -364,7 +349,6 @@ export class SessionChatController {
             "type `/help` for commands and keybindings",
             "mention files with `@`, agents and skills with `@@`",
             "run bash commands with `!` or `!!`",
-            "use `#` to update AGENTS.md",
         ];
         if (this.snapshot.catalog.skills.length > 0) {
             lines.push("", "skills:");
@@ -380,21 +364,15 @@ export class SessionChatController {
                 lines.push(`  ${agentsFile}`);
             }
         }
+        if (this.configuredClientToolNames.length > 0) {
+            lines.push("", "client tools:");
+            for (const clientToolName of this.configuredClientToolNames) {
+                lines.push(`  ${clientToolName}`);
+            }
+        }
         lines.push("", this.formatSessionIdentityText());
         return lines;
     }
-    async submitUserText(text, options = {}) {
-        const historyEntryId = `session-user-${randomUUID()}`;
-        const model = {
-            type: "user",
-            text,
-            ...(options.kind ? { kind: options.kind } : {}),
-        };
-        this.clientRenderedUserMessages.set(historyEntryId, model);
-        this.view.addMessage(model, historyEntryId);
-        this.renderedMessageIds.push(historyEntryId);
-        await this.runSessionTurn(() => this.session.submit(options.textForModel ?? text, { historyEntryId }));
-    }
     async retryTurn() {
         await this.runSessionTurn(() => this.session.retry());
     }
@@ -492,7 +470,7 @@ export class SessionChatController {
             return;
         }
         if (!this.isSessionOperationActive()) {
-            await this.submitUserText(trimmed);
+            await this.runSessionTurn(() => this.session.submit(trimmed));
             return;
         }
         if (this.isBlockingSessionOperationActive()) {
@@ -502,9 +480,7 @@ export class SessionChatController {
         this.submitSteeringText(trimmed);
     }
     submitQueuedText(text) {
-        void this.session
-            .queue(text, { historyEntryId: `session-queue-${randomUUID()}` })
-            .catch((error) => {
+        void this.session.queue(text).catch((error) => {
             if (!this.isPendingMessageCancellation(error)) {
                 this.view.addSystemMessage(`queueing failed: ${formatSessionError(error)}`, "error");
             }
@@ -833,6 +809,12 @@ export class SessionChatController {
         }
         const createInput = {
             executionEnvironment: this.createExecutionEnvironmentInputFromSnapshot(),
+            attributes: {
+                source: "tui",
+                ...(this.snapshot.attributes.repository
+                    ? { repository: this.snapshot.attributes.repository }
+                    : {}),
+            },
             personaId: this.snapshot.settings.personaId,
             ...(this.snapshot.settings.reasoning !== undefined
                 ? { reasoning: this.snapshot.settings.reasoning }
@@ -1052,9 +1034,17 @@ export class SessionChatController {
         this.snapshot = snapshot;
         this.observedSessionRevision = Math.max(this.observedSessionRevision, snapshot.revision);
         this.assistantMessages = [];
-        const messages = getTimelineMessages(snapshot);
-        for (const entry of messages) {
-            this.renderProtocolMessage(entry);
+        for (const item of getRenderableTimelineItems(snapshot)) {
+            if (item.type === "notice") {
+                this.renderedMessageIds.push(this.view.addMessage({
+                    type: "system",
+                    text: item.text,
+                    kind: item.severity === "info" ? "muted" : item.severity,
+                }, item.id));
+            }
+            else {
+                this.renderProtocolMessage(item.message);
+            }
         }
         this.syncSnapshotToolAndAgentUi(snapshot);
     }
@@ -1079,38 +1069,41 @@ export class SessionChatController {
     syncRenderedHistory(snapshot) {
         this.snapshot = snapshot;
         this.observedSessionRevision = Math.max(this.observedSessionRevision, snapshot.revision);
-        const messages = getTimelineMessages(snapshot);
-        const snapshotIds = new Set(messages.map((entry) => entry.id));
+        const items = getRenderableTimelineItems(snapshot);
+        const snapshotIds = new Set(items.map((item) => item.id));
         const staleIds = this.renderedMessageIds.filter((id) => !snapshotIds.has(id));
         if (staleIds.length > 0) {
             this.view.removeMessages(staleIds);
-            for (const id of staleIds) {
-                this.clientRenderedUserMessages.delete(id);
-            }
         }
         const renderedIds = new Set(this.renderedMessageIds);
         this.renderedMessageIds.splice(0, this.renderedMessageIds.length, ...this.renderedMessageIds.filter((id) => snapshotIds.has(id)));
-        for (const entry of messages) {
-            if (this.hiddenHistoryEntryIds.has(entry.id)) {
-                if (!renderedIds.has(entry.id)) {
-                    this.renderedMessageIds.push(entry.id);
+        for (const item of items) {
+            if (item.type === "message" && this.hiddenHistoryEntryIds.has(item.id)) {
+                if (!renderedIds.has(item.id)) {
+                    this.renderedMessageIds.push(item.id);
                 }
                 continue;
             }
-            const model = this.clientRenderedUserMessages.get(entry.id) ?? this.buildProtocolMessageModel(entry);
+            const model = item.type === "notice"
+                ? {
+                    type: "system",
+                    text: item.text,
+                    kind: item.severity === "info" ? "muted" : item.severity,
+                }
+                : this.buildProtocolMessageModel(item.message);
             if (!model) {
                 continue;
             }
-            if (renderedIds.has(entry.id)) {
+            if (renderedIds.has(item.id)) {
                 if (model.type === "assistant") {
-                    this.view.updateAssistantMessage(entry.id, model);
+                    this.view.updateAssistantMessage(item.id, model);
                 }
                 else {
-                    this.view.updateMessage(entry.id, model);
+                    this.view.updateMessage(item.id, model);
                 }
             }
             else {
-                this.renderedMessageIds.push(this.view.addMessage(model, entry.id));
+                this.renderedMessageIds.push(this.view.addMessage(model, item.id));
             }
         }
         this.assistantMessages = this.collectAssistantMessages(snapshot);
@@ -1250,7 +1243,6 @@ export class SessionChatController {
     }
     startLocalUiSession() {
         this.hiddenHistoryEntryIds.clear();
-        this.clientRenderedUserMessages.clear();
         this.renderedMessageIds.splice(0);
         this.view.addMessage({ type: "session_divider", label: "new session" });
     }
@@ -1548,8 +1540,6 @@ export class SessionChatController {
             return "bash_incognito";
         if (this.isBashMode)
             return "bash";
-        if (this.isMemoryMode)
-            return "memory";
         return "normal";
     }
     getSessionOperationStatusHint() {
@@ -1795,15 +1785,6 @@ export class SessionChatController {
         return { bridge, result };
     }
     async handleReturnedDiffReview(review, session) {
-        const model = {
-            type: "user",
-            text: review.review,
-            kind: "review",
-        };
-        this.clientRenderedUserMessages.set(review.historyEntryId, model);
-        if (!this.renderedMessageIds.includes(review.historyEntryId)) {
-            this.renderedMessageIds.push(review.historyEntryId);
-        }
         try {
             const result = await session.record(formatDiffReviewUserMessage(review), {
                 historyEntryId: review.historyEntryId,
@@ -1813,7 +1794,13 @@ export class SessionChatController {
             }
         }
         catch (error) {
-            this.view.addSystemMessage(`failed to add diff review to session: ${error.message}`, "error");
+            const reviewWasCommitted = this.session === session &&
+                this.snapshot.messages.some((entry) => entry.id === review.historyEntryId &&
+                    entry.state === "committed" &&
+                    entry.message.role === "user");
+            if (!reviewWasCommitted) {
+                throw error;
+            }
         }
     }
     startRewindFlow() {
@@ -1923,14 +1910,24 @@ function formatSessionError(error) {
         : undefined;
     return cause && cause !== message ? `${message}: ${cause}` : message;
 }
-function getTimelineMessages(snapshot) {
+function getRenderableTimelineItems(snapshot) {
     const messagesById = new Map(snapshot.messages.map((message) => [message.id, message]));
     return snapshot.timeline.flatMap((item) => {
+        if (item.type === "notice") {
+            return [
+                {
+                    type: "notice",
+                    id: item.id,
+                    severity: item.notice.severity,
+                    text: item.notice.text,
+                },
+            ];
+        }
         if (item.type !== "message") {
             return [];
         }
         const message = messagesById.get(item.messageId);
-        return message ? [message] : [];
+        return message ? [{ type: "message", id: message.id, message }] : [];
     });
 }
 function isMessageInTimeline(snapshot, messageId) {