@quatcode/agent 0.2.9 → 0.2.11

package/dist/tui.js

@@ -1,7 +1,7 @@
 import path from "node:path";
 import { homedir } from "node:os";
 import { readFile, stat } from "node:fs/promises";
-import { Session, ToolRegistry, addUsage, emptyUsage, estimateCost, runAgent, } from "@quatcode/core";
+import { Session, ToolRegistry, addUsage, compactMessages, contextWindowFor, emptyUsage, estimateCost, runAgent, } from "@quatcode/core";
 import { analyzeCommand, createProjectTools, McpManager } from "@quatcode/tools";
 import { buildProjectContext, createBrainTools, scanProject } from "@quatcode/brain";
 import { Keyboard } from "./keyboard.js";
@@ -14,7 +14,7 @@ import { renderMarkdown, wrapSegments } from "./markdown.js";
 import { ADD_PROVIDER, allProviders, buildProvider, defaultModelFor, effortsForModel, hasKey, keyEnvFor, probeProviderModels, providerEnvKey, providerHasModel, providerMeta, providerSlug, registerCustomProvider, removeCustomProvider, resolveProviderForModel, } from "./models.js";
 import { ASCII_MODE, activeThemeId, bg, bold, buildColorRemap, clipAnsi, fg, getThemes, glyphs, mix, padEndTo, palette, pulse, remapColors, setTheme, shimmer, stringWidth, stripAnsi, tone, toolGlyph, } from "./theme.js";
 import { listSessions, listSessionsForDir, loadConfig, makeSessionId, saveConfig, saveSession, deleteSession } from "./store.js";
-import { configFile, conversationKey, deleteConversation, displayTitle, latestPerConversation, messagesForStorage, sameDir, sortConversations, updateConversation } from "./store.js";
+import { appendAudit, auditInput, configFile, conversationKey, deleteConversation, displayTitle, latestPerConversation, messagesForStorage, sameDir, sortConversations, updateConversation } from "./store.js";
 import { MCP_USAGE, isMcpUrl, mcpInstructionsSection, parseMcpAdd } from "./mcp.js";
 import { AUTO, PERSONA_SETTINGS, personaSection, personaSummary, } from "./persona.js";
 import { copyToClipboard, readClipboardImage, readClipboardText, } from "./clipboard.js";
@@ -35,6 +35,7 @@ const COMMAND_INFO = [
     { name: "/timeline", desc: "jump to a sent message" },
     { name: "/undo", desc: "take back the last message" },
     { name: "/verbose", desc: "toggle full tool output" },
+    { name: "/compact", desc: "summarize older history to free context" },
     { name: "/settings", desc: "tune the agent's persona: character, tone, verbosity…" },
     { name: "/skills", args: "[add|remove|list]", desc: "manage agent skills" },
     { name: "/mcp", args: "[list|add|remove]", desc: "manage MCP servers" },
@@ -110,6 +111,8 @@ class TuiApp {
     permissionInner = 0;
     permissionBorder = palette.steel;
     lastVisibleStart = 0;
+    lastContextTokens = 0;
+    lastCachedTokens = 0;
     lastInputRender = null;
     lastInputOffset = 1;
     lastHeaderRows = 2;
@@ -1982,21 +1985,50 @@ class TuiApp {
                     this.append([
                         `  ${tone.amber(glyphs.diamondOpen)} ${tone.mist(`// ${req.tool} blocked in PLAN mode (Tab switches to BUILD)`)}`,
                     ]);
+                    this.audit(req, "denied", "plan");
                     return { allowed: false };
                 }
                 if (this.agentMode === "plan" && READ_ONLY_TOOLS.has(req.tool))
                     return { allowed: true };
-                if (this.gateMode === "always")
+                const analysis = req.tool === "run_command" ? this.commandAnalysis(req) : null;
+                const dangerous = analysis?.dangerous === true;
+                if (this.opts.allowAll) {
+                    this.audit(req, "allowed", "allow-all", analysis);
                     return { allowed: true };
-                const choice = await this.permissionCard(req);
-                if (choice === "always") {
-                    this.gateMode = "always";
+                }
+                // "always" never covers dangerous commands — those are confirmed each time.
+                if (this.gateMode === "always" && !dangerous) {
+                    this.audit(req, "allowed", "always", analysis);
                     return { allowed: true };
                 }
-                return { allowed: choice === "allow" };
+                const choice = await this.permissionCard(req);
+                const granted = choice === "allow" || choice === "always";
+                if (choice === "always" && !dangerous)
+                    this.gateMode = "always";
+                this.audit(req, granted ? "allowed" : "denied", choice, analysis);
+                return { allowed: granted };
             },
         };
     }
+    commandAnalysis(req) {
+        const command = req.input && typeof req.input === "object" && "command" in req.input
+            ? String(req.input.command)
+            : "";
+        return command ? analyzeCommand(command) : null;
+    }
+    audit(req, decision, via, analysis) {
+        void appendAudit({
+            ts: new Date().toISOString(),
+            dir: this.dir,
+            tool: req.tool,
+            decision,
+            via,
+            danger: analysis
+                ? { dangerous: analysis.dangerous, sensitive: analysis.sensitive, reasons: analysis.reasons }
+                : undefined,
+            input: auditInput(req.input),
+        }).catch(() => undefined);
+    }
     async promptForKey(keyEnv) {
         this.append([`  ${tone.mist(`// ${keyEnv} IS NOT SET`)}`]);
         const result = await this.readLine({
@@ -2055,7 +2087,7 @@ class TuiApp {
         this.undoAfterAbort = false;
         this.mode = "running";
         this.scrollOffset = 0;
-        const controller = new TurnController({ model: this.model, maxSteps: this.opts.maxSteps }, {
+        const controller = new TurnController({ model: this.model, maxSteps: this.opts.maxSteps, contextWindow: this.contextLimit() }, {
             width: () => this.contentWidth() - 2,
             append: (lines) => this.append(lines),
             beginLive: () => this.beginLive(),
@@ -2079,6 +2111,7 @@ class TuiApp {
                 stream: true,
                 signal: this.abort.signal,
                 images,
+                contextLimit: this.contextLimit(),
                 onEvent: controller.handle,
             });
         }
@@ -2117,6 +2150,8 @@ class TuiApp {
         }
         this.totalUsage = addUsage(this.totalUsage, result.usage);
         this.turns += 1;
+        this.lastContextTokens = controller.contextTokens;
+        this.lastCachedTokens = controller.cachedTokens;
         const status = this.abortRequested
             ? "interrupted"
             : result.status === "complete"
@@ -2219,6 +2254,9 @@ class TuiApp {
             case "settings":
                 await this.openSettings();
                 return false;
+            case "compact":
+                await this.compactCommand();
+                return false;
             case "theme":
                 await this.changeTheme(arg);
                 return false;
@@ -2239,6 +2277,50 @@ class TuiApp {
                 return false;
         }
     }
+    contextLimit() {
+        const flag = this.opts.contextLimit;
+        if (typeof flag === "number" && flag > 0)
+            return flag;
+        const env = Number(process.env.QUATCODE_CONTEXT_LIMIT);
+        if (Number.isFinite(env) && env > 0)
+            return env;
+        const configured = this.config.contextLimit;
+        if (typeof configured === "number" && configured > 0)
+            return configured;
+        return contextWindowFor(this.model);
+    }
+    async compactCommand() {
+        const provider = this.ensureProvider();
+        if (!provider) {
+            this.append([
+                `  ${tone.red(glyphs.cross)} ${tone.mist("// COMPACT needs a configured provider")}`,
+            ]);
+            return;
+        }
+        const before = this.session.getMessages().length;
+        this.setStatus({ label: "compacting context" });
+        try {
+            const result = await compactMessages(this.session.getMessages(), {
+                provider,
+                model: this.model,
+            });
+            if (!result) {
+                this.append([`  ${tone.mist("// nothing to compact yet")}`]);
+                return;
+            }
+            this.session.replaceMessages(result.messages);
+            this.append([
+                `  ${tone.accent(glyphs.diamond)} ${tone.mist(`// CONTEXT COMPACTED · ${before} → ${result.messages.length} messages · ~${fmtTokens(result.tokensBefore)} → ~${fmtTokens(result.tokensAfter)} tokens`)}`,
+                "",
+            ]);
+        }
+        catch (err) {
+            this.append([`  ${tone.red(`// compact failed: ${errorMessage(err)}`)}`]);
+        }
+        finally {
+            this.setStatus(null);
+        }
+    }
     async openSettings() {
         const before = JSON.stringify(this.persona);
         for (;;) {
@@ -2967,11 +3049,17 @@ class TuiApp {
     }
     printUsage() {
         const cost = estimateCost(this.totalUsage, this.model, this.providerId);
+        const window = this.contextLimit();
+        const context = this.lastContextTokens > 0
+            ? `${fmtTokens(this.lastContextTokens)} / ${fmtTokens(window)} (${Math.round((this.lastContextTokens / window) * 100)}%)`
+            : "—";
         const rows = [
             ["turns", String(this.turns)],
+            ["context", context],
+            ["cached", this.lastCachedTokens > 0 ? `${fmtTokens(this.lastCachedTokens)} read` : "—"],
             [
                 "tokens",
-                `${fmtTokens(this.totalUsage.inputTokens)}↑ ${fmtTokens(this.totalUsage.outputTokens)}↓`,
+                `${fmtTokens(this.totalUsage.inputTokens)}↑ ${fmtTokens(this.totalUsage.outputTokens)}↓ (session)`,
             ],
             ["cost", fmtCost(cost)],
             ["elapsed", fmtDuration(performance.now() - this.startedAt)],