@markusylisiurunen/tau 0.1.47 → 0.1.49

package/README.md

@@ -57,6 +57,17 @@ npm run build
 npm start
 ```
 
+## theme preview
+
+run a model-free UI preview for theme iteration:
+
+```sh
+tau --theme-preview
+```
+
+theme preview renders a fixed set of UI fixtures and disables model calls so you can tweak colors and spacing
+without asking the model for visible content.
+
 ## risk levels
 
 tau uses risk levels to control what the model can do. this lets you stay in control while working alongside AI.
@@ -75,6 +86,8 @@ or change it during a session with `/risk:restricted`, `/risk:read-only`, or `/r
 
 the default is read-only because it lets the model investigate your code and answer questions without risk of unintended changes. bump it to read-write when you're ready to let the model make edits.
 
+custom personas (loaded from disk) only allow `read-only` and `read-write` risk levels. if you try to use `restricted` with a custom persona, tau will keep the risk level at `read-only`.
+
 ## personas
 
 tau comes with several built-in personas across different models:
@@ -158,6 +171,7 @@ tau supports slash commands for common actions:
 | `/new`                    | clear the session and start fresh              |
 | `/copy`                   | copy the last assistant message                |
 | `/copy:code`              | copy just the code blocks                      |
+| `/export:html`            | export chat history to html                    |
 | `/reload`                 | reload personas, prompts, and skills from disk |
 | `/compact:only-summary`   | compress history and continue with a summary   |
 | `/compact:with-last-turn` | compress history but keep the last exchange    |
@@ -258,6 +272,7 @@ you can also set model parameters via optional frontmatter fields:
       model: claude-haiku-4-5
       reasoning: medium
   ```
+- `tools`: list of tool names to enable for this persona. allowed: `bash`, `write`, `edit`, `read`, `list`, `grep`, `task`, `fork`. if omitted, defaults to `bash`, `write`, `edit` (and `task` when subagents are enabled). risk levels still apply.
 
 use it with `--persona my-assistant` or `/persona:my-assistant`. if a project persona id conflicts with a user or built-in persona, the project persona wins.
 

package/dist/app.js

@@ -1,4 +1,5 @@
-import { homedir } from "node:os";
+import { mkdtemp, writeFile } from "node:fs/promises";
+import { homedir, tmpdir } from "node:os";
 import { join, resolve } from "node:path";
 import { streamSimple } from "@mariozechner/pi-ai";
 import { Spacer, TUI } from "@mariozechner/pi-tui";
@@ -7,6 +8,8 @@ import { copyTextToClipboard } from "./clipboard.js";
 import { buildHelpText, getRiskLevelDescription, parseCommand } from "./commands.js";
 import { getApiKeyForProvider } from "./config.js";
 import { loadAllContent } from "./content_loader.js";
+import { buildExportEntriesFromHistory } from "./export/engine_history.js";
+import { renderExport } from "./export/index.js";
 import { SessionEngine } from "./session/session_engine.js";
 import { formatSubagentsForPrompt } from "./subagents/registry.js";
 import { createAppTerminal } from "./terminal.js";
@@ -30,6 +33,7 @@ import { buildGrepBlockedView, buildGrepFinishedView, buildGrepRunningView, buil
 import { getFileAutocompleteToken, SlashAutocompleteProvider } from "./ui/slash_autocomplete.js";
 import { buildTaskBlockedView, buildTaskFinishedView, buildTaskRunningView, } from "./ui/task_execution.js";
 import { createUiTheme } from "./ui/theme.js";
+import { buildThemePreviewMessages } from "./ui/theme_preview.js";
 import { buildBaseSystemPrompt, buildEnvironmentTag, buildProjectContextBlock, buildSkillsIndexBlock, findAgentsFilesFromCwdToHome, formatRiskLevelChangeNotice, } from "./utils/context.js";
 import { formatHistoryForCompression } from "./utils/fork.js";
 import { formatAdaptiveNumber, formatCwd, formatTokenWindow } from "./utils/format.js";
@@ -82,6 +86,7 @@ export class ChatApp {
     currentTurnStartedAt;
     lastTurnDurationMs = 0;
     turnTimer;
+    themePreview;
     constructor(options) {
         this.personas = options.personas;
         this.prompts = options.prompts ?? [];
@@ -91,15 +96,11 @@ export class ChatApp {
         this.initialUserMessage = options.initialUserMessage;
         this.config = options.config ?? {};
         this.compactToolUi = this.config.toolDisplayMode !== "full";
-        if (options.initialRiskLevel) {
-            this.riskLevel = options.initialRiskLevel;
+        this.themePreview = options.themePreview ?? false;
+        this.showThinking = this.themePreview;
+        if (this.themePreview) {
+            this.queuedUserMessages.push("Queue: adjust muted contrast and preview again", "Queue: verify tool error colors");
         }
-        this.initialRiskLevel = this.riskLevel;
-        this.environmentTag = buildEnvironmentTag({
-            riskLevel: this.initialRiskLevel,
-            cwd: process.cwd(),
-            datetime: new Date().toISOString(),
-        });
         this.agentsFiles = options.withContext
             ? findAgentsFilesFromCwdToHome(process.cwd(), homedir())
             : [];
@@ -112,6 +113,19 @@ export class ChatApp {
                 this.personas.find((p) => p.id.toLowerCase() === options.initialPersonaId.toLowerCase())) ||
                 this.personas[0];
         this.clampPersonaReasoning(this.currentPersona);
+        if (options.initialRiskLevel) {
+            this.riskLevel = options.initialRiskLevel;
+        }
+        const allowedRiskLevels = this.getAllowedRiskLevelsForPersona(this.currentPersona);
+        if (!allowedRiskLevels.includes(this.riskLevel)) {
+            this.riskLevel = allowedRiskLevels[0] ?? "read-only";
+        }
+        this.initialRiskLevel = this.riskLevel;
+        this.environmentTag = buildEnvironmentTag({
+            riskLevel: this.initialRiskLevel,
+            cwd: process.cwd(),
+            datetime: new Date().toISOString(),
+        });
         this.baseSystemPrompt = buildBaseSystemPrompt({
             personaSystemPrompt: this.currentPersona.systemPrompt,
             skillsBlock: this.getSkillsIndexBlockForPersona(this.currentPersona).skillsBlock,
@@ -139,7 +153,7 @@ export class ChatApp {
         });
         this.uiTheme = createUiTheme("ansi");
         this.ui = new TUI(createAppTerminal());
-        this.chatContainer = new ChatContainerComponent(this.uiTheme);
+        this.chatContainer = new ChatContainerComponent(this.uiTheme, this.showThinking);
         this.chatContainer.setCompactToolUi(this.compactToolUi);
         this.footer = new FooterComponent(this.uiTheme, this.ui);
         this.queuedMessages = new QueuedMessagesComponent(this.uiTheme, this.queuedUserMessages);
@@ -153,12 +167,20 @@ export class ChatApp {
         this.ui.addChild(this.queuedMessages);
         this.ui.addChild(this.editor);
         this.ui.addChild(this.footer);
-        this.chatContainer.addMessage({
-            type: "app_intro",
-            appName: "tau",
-            version: APP_VERSION,
-            helpText: buildHelpText(this.agentsFiles, this.skills),
-        });
+        if (this.themePreview) {
+            const messages = buildThemePreviewMessages(this.uiTheme);
+            for (const message of messages) {
+                this.chatContainer.addMessage(message);
+            }
+        }
+        else {
+            this.chatContainer.addMessage({
+                type: "app_intro",
+                appName: "tau",
+                version: APP_VERSION,
+                helpText: buildHelpText(this.agentsFiles, this.skills, this.getAllowedRiskLevelsForPersona(this.currentPersona)),
+            });
+        }
         this.ui.setFocus(this.editor);
         this.updateFooter();
         this.updateEditorBorderColor();
@@ -182,6 +204,9 @@ export class ChatApp {
         };
         this.editor.onAltUp = () => this.popQueuedUserMessageIntoEditor();
         this.editor.beforeSubmit = (text) => {
+            if (this.themePreview) {
+                return false;
+            }
             if (!this.isStreaming)
                 return true;
             const trimmed = text.trimStart();
@@ -206,7 +231,7 @@ export class ChatApp {
         this.editor.setAutocompleteProvider(new SlashAutocompleteProvider(() => this.personas.map((p) => ({ id: p.id, label: p.label })), () => this.prompts.map((t) => ({ id: t.id, label: t.label })), () => this.bashCommands.map((b) => ({
             id: b.id,
             description: b.description,
-        })), () => this.projectFiles, () => this.skills.map((skill) => skill.name)));
+        })), () => this.projectFiles, () => this.skills.map((skill) => skill.name), () => this.getAllowedRiskLevelsForPersona(this.currentPersona)));
         this.editor.onSubmit = (text) => this.handleSubmit(text);
     }
     getEditorTextBeforeCursor() {
@@ -232,6 +257,9 @@ export class ChatApp {
     }
     async start() {
         this.ui.start();
+        if (this.themePreview) {
+            return;
+        }
         if (this.initialUserMessage) {
             await this.sendInitialUserMessage(this.initialUserMessage);
         }
@@ -271,6 +299,14 @@ export class ChatApp {
         this.ui.requestRender();
     }
     updateEditorHeader(cwd = formatCwd(process.cwd()), personaName = this.currentPersona.label || this.currentPersona.id, reasoningLabel = this.currentPersona.settings.reasoning ?? "none") {
+        if (this.themePreview) {
+            const labelStyle = this.uiTheme.palette.muted;
+            this.editor.setHeader("theme preview", "model disabled", {
+                leftStyle: labelStyle,
+                rightStyle: labelStyle,
+            });
+            return;
+        }
         if (this.isBashMode) {
             this.editor.setHeader("bash", "", { leftStyle: this.editor.borderColor });
             return;
@@ -332,7 +368,7 @@ export class ChatApp {
             ? this.getContextWindowForLastTurn(last)
             : this.currentPersona.model.contextWindow;
         const { input, read, write, output } = this.getSessionTotals();
-        const stats = `↑${formatTokenWindow(input)} ↓${formatTokenWindow(output)} cache ${formatTokenWindow(read)}/${formatTokenWindow(write)}`;
+        const stats = `↑${formatTokenWindow(input)} ↓${formatTokenWindow(output)} (cache r${formatTokenWindow(read)} w${formatTokenWindow(write)})`;
         const promptTokensSent = last
             ? (last.usage?.input ?? 0) + (last.usage?.cacheRead ?? 0) + (last.usage?.cacheWrite ?? 0)
             : 0;
@@ -449,12 +485,20 @@ export class ChatApp {
             persona.settings.reasoning = allowed[0];
         }
     }
+    isCustomPersona(persona) {
+        return persona.source !== "builtin";
+    }
+    getAllowedRiskLevelsForPersona(persona) {
+        return this.isCustomPersona(persona)
+            ? ["read-only", "read-write"]
+            : ["restricted", "read-only", "read-write"];
+    }
     // Risk Level Management ---------------------------------------------------------------
     cycleRiskLevel() {
-        const levels = ["restricted", "read-only", "read-write"];
+        const levels = this.getAllowedRiskLevelsForPersona(this.currentPersona);
         const previous = this.riskLevel;
         const index = levels.indexOf(this.riskLevel);
-        const next = levels[(index + 1) % levels.length];
+        const next = levels[(index + 1) % levels.length] ?? levels[0];
         this.riskLevel = next;
         this.engine.setRiskLevel(next);
         this.updateFooter();
@@ -476,6 +520,13 @@ export class ChatApp {
         const next = this.personas[(index + 1) % this.personas.length];
         this.currentPersona = next;
         this.clampPersonaReasoning(this.currentPersona);
+        const allowedRiskLevels = this.getAllowedRiskLevelsForPersona(this.currentPersona);
+        if (!allowedRiskLevels.includes(this.riskLevel)) {
+            this.setRiskLevel(this.riskLevel, {
+                force: true,
+                reason: "restricted risk level is not available for custom personas.",
+            });
+        }
         const skillsContext = this.getSkillsIndexBlockForPersona(this.currentPersona);
         this.baseSystemPrompt = buildBaseSystemPrompt({
             personaSystemPrompt: this.currentPersona.systemPrompt,
@@ -619,6 +670,9 @@ export class ChatApp {
         }
     }
     async handleSubmit(text) {
+        if (this.themePreview) {
+            return;
+        }
         const trimmed = text.trim();
         if (!trimmed)
             return;
@@ -711,6 +765,9 @@ export class ChatApp {
             case "copyCode":
                 await this.copyLastAssistantCodeBlock();
                 break;
+            case "export":
+                await this.exportSessionHtml();
+                break;
             case "new":
                 this.clearSession();
                 break;
@@ -741,7 +798,7 @@ export class ChatApp {
         }
     }
     showHelp() {
-        this.addSystemMessage(buildHelpText(this.agentsFiles, this.skills), "muted");
+        this.addSystemMessage(buildHelpText(this.agentsFiles, this.skills, this.getAllowedRiskLevelsForPersona(this.currentPersona)), "muted");
     }
     async copyLastAssistantMessage() {
         const lastAssistant = this.getLastAssistantMessage();
@@ -782,6 +839,31 @@ export class ChatApp {
             this.addSystemMessage(`clipboard copy failed: ${err.message}`, "error");
         }
     }
+    async exportSessionHtml() {
+        const history = this.engine.history;
+        if (history.length === 0) {
+            this.addSystemMessage("no conversation to export.", "warn");
+            return;
+        }
+        try {
+            const entries = buildExportEntriesFromHistory(history);
+            if (entries.length === 0) {
+                this.addSystemMessage("no conversation to export.", "warn");
+                return;
+            }
+            const html = renderExport("html", entries, {
+                title: "tau chat export",
+                generatedAt: Date.now(),
+            });
+            const dir = await mkdtemp(join(tmpdir(), "tau-export-"));
+            const filePath = join(dir, "index.html");
+            await writeFile(filePath, html, "utf8");
+            this.addSystemMessage(filePath, "muted");
+        }
+        catch (err) {
+            this.addSystemMessage(`export failed: ${err.message}`, "error");
+        }
+    }
     async stashEditorToClipboard() {
         const text = this.editor.getText();
         if (!text.trim()) {
@@ -1016,21 +1098,45 @@ Write plain prose, no formatting. Be thorough enough that the reader can resume
             void this.drainQueuedUserMessages();
         }
     }
-    setRiskLevel(level) {
+    setRiskLevel(level, options) {
+        const allowed = this.getAllowedRiskLevelsForPersona(this.currentPersona);
+        let target = level;
+        let forced = false;
+        if (!allowed.includes(level)) {
+            if (!options?.force) {
+                if (!options?.silent) {
+                    this.addSystemMessage(`risk level '${level}' is not available for the current persona. allowed: ${allowed.join(", ")}.`, "error");
+                }
+                return;
+            }
+            target = allowed[0] ?? "read-only";
+            forced = true;
+        }
         const previous = this.riskLevel;
-        this.riskLevel = level;
-        this.engine.setRiskLevel(level);
+        this.riskLevel = target;
+        this.engine.setRiskLevel(target);
         this.updateFooter();
-        if (previous !== level) {
+        if (previous !== target) {
             const from = this.pendingRiskLevelChange?.from ?? previous;
-            if (from === level) {
+            if (from === target) {
                 this.pendingRiskLevelChange = undefined;
             }
             else {
-                this.pendingRiskLevelChange = { from, to: level };
+                this.pendingRiskLevelChange = { from, to: target };
             }
         }
-        this.addSystemMessage(this.formatRiskLevelNotice(level), "success");
+        if (options?.silent) {
+            return;
+        }
+        if (forced) {
+            const reason = options?.reason ?? `risk level '${level}' is not available for the current persona.`;
+            const msg = previous === target
+                ? `${reason} staying at ${target}.`
+                : `${reason} switched to ${target}.`;
+            this.addSystemMessage(msg, "warn");
+            return;
+        }
+        this.addSystemMessage(this.formatRiskLevelNotice(target), "success");
     }
     switchPersona(id) {
         const persona = this.personas.find((p) => p.id.toLowerCase() === id.toLowerCase());
@@ -1040,6 +1146,13 @@ Write plain prose, no formatting. Be thorough enough that the reader can resume
         }
         this.currentPersona = persona;
         this.clampPersonaReasoning(this.currentPersona);
+        const allowedRiskLevels = this.getAllowedRiskLevelsForPersona(this.currentPersona);
+        if (!allowedRiskLevels.includes(this.riskLevel)) {
+            this.setRiskLevel(this.riskLevel, {
+                force: true,
+                reason: "restricted risk level is not available for custom personas.",
+            });
+        }
         const skillsContext = this.getSkillsIndexBlockForPersona(this.currentPersona);
         this.baseSystemPrompt = buildBaseSystemPrompt({
             personaSystemPrompt: this.currentPersona.systemPrompt,
@@ -1102,6 +1215,13 @@ Write plain prose, no formatting. Be thorough enough that the reader can resume
                 this.clampPersonaReasoning(this.currentPersona);
                 this.addSystemMessage(`previous persona no longer available; switched to ${this.currentPersona.label || this.currentPersona.id}.`, "warn");
             }
+            const allowedRiskLevels = this.getAllowedRiskLevelsForPersona(this.currentPersona);
+            if (!allowedRiskLevels.includes(this.riskLevel)) {
+                this.setRiskLevel(this.riskLevel, {
+                    force: true,
+                    reason: "restricted risk level is not available for custom personas.",
+                });
+            }
             // Rebuild system prompt and update the engine
             const skillsContext = this.getSkillsIndexBlockForPersona(this.currentPersona);
             this.baseSystemPrompt = buildBaseSystemPrompt({