miii-cli 1.0.1 → 1.1.0

package/dist/tui/hooks/useRunLoop.js

@@ -1,4 +1,5 @@
 import { useState, useRef, useCallback, useEffect } from 'react';
+import { readFileSync, writeFileSync, unlinkSync, existsSync } from 'fs';
 import { chat } from '../../llm/stream.js';
 import { tools as staticTools } from '../../tools/index.js';
 import { StreamParser, extractBareToolCall } from '../../parser/stream-parser.js';
@@ -8,6 +9,7 @@ const MAX_TOOL_DEPTH = 6;
 const FILE_EDIT_TOOLS = new Set(['edit_file', 'create_file', 'patch_file', 'delete_file']);
 const SHOW_RESULT_TOOLS = new Set(['run_tests', 'git_commit']);
 const PERMISSION_TOOLS = new Set(['edit_file', 'patch_file', 'delete_file', 'create_file', 'move_file', 'run_command', 'git_commit']);
+const CHECKPOINT_TOOLS = new Set(['edit_file', 'patch_file', 'create_file', 'delete_file']);
 export function useRunLoop(config, currentModelRef, pushHistory, extraTools = [], abortRef) {
     const [status, setStatus] = useState('idle');
     const [tick, setTick] = useState(0);
@@ -15,16 +17,25 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
     const [taskLabel, setTaskLabel] = useState();
     const [permissionRequest, setPermissionRequest] = useState(null);
     const permissionResolveRef = useRef(null);
+    const [compactRequest, setCompactRequest] = useState(null);
+    const compactResolveRef = useRef(null);
+    const checkpointRef = useRef(new Map());
+    const sessionApprovedRef = useRef(new Set());
     const thinkingStartRef = useRef(0);
     const extraToolsRef = useRef(extraTools);
     extraToolsRef.current = extraTools;
     const pushHistoryRef = useRef(pushHistory);
     useEffect(() => { pushHistoryRef.current = pushHistory; }, [pushHistory]);
-    const resolvePermission = useCallback((approved) => {
-        permissionResolveRef.current?.(approved);
+    const resolvePermission = useCallback((result) => {
+        permissionResolveRef.current?.(result);
         permissionResolveRef.current = null;
         setPermissionRequest(null);
     }, []);
+    const resolveCompact = useCallback((approved) => {
+        compactResolveRef.current?.(approved);
+        compactResolveRef.current = null;
+        setCompactRequest(null);
+    }, []);
     useEffect(() => {
         if (status === 'idle')
             return;
@@ -38,9 +49,30 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
             return;
         }
         setStatus('thinking');
-        if (depth === 0)
+        if (depth === 0) {
             thinkingStartRef.current = Date.now();
-        const msgs = shouldCompact(contextMsgs) ? compactContext(contextMsgs, goal) : contextMsgs;
+            checkpointRef.current.clear();
+        }
+        let msgs = contextMsgs;
+        if (shouldCompact(contextMsgs)) {
+            const approved = await new Promise(resolve => {
+                compactResolveRef.current = resolve;
+                setCompactRequest({ messageCount: contextMsgs.length });
+            });
+            if (approved) {
+                printer.systemMsg('compacting context…');
+                msgs = await compactContext(contextMsgs, {
+                    provider: config.provider,
+                    model: currentModelRef.current,
+                    baseUrl: config.baseUrl,
+                    apiKey: config.apiKey,
+                }, goal);
+                printer.systemMsg(`compacted: ${contextMsgs.length} → ${msgs.length} messages`);
+            }
+            else {
+                printer.systemMsg('keeping full context — responses may be slower');
+            }
+        }
         abortRef.current = new AbortController();
         await chat({
             provider: config.provider,
@@ -67,6 +99,8 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
                 if (displayText)
                     printer.assistantMsg(displayText);
                 pushHistoryRef.current({ role: 'assistant', content: fullText });
+                if (pendingTools.length)
+                    printer.planSummary(pendingTools);
                 if (!pendingTools.length) {
                     const hasFencedCode = /```[\w]*\n[\s\S]{50,}?\n```/.test(fullText);
                     if (hasFencedCode && depth < MAX_TOOL_DEPTH - 1) {
@@ -88,20 +122,42 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
                         const tool = allTools.find(t => t.name === tc.name);
                         setCurrentTool(tc.name);
                         if (PERMISSION_TOOLS.has(tc.name)) {
-                            const approved = await new Promise(resolve => {
-                                permissionResolveRef.current = resolve;
-                                setPermissionRequest({ toolName: tc.name, args: tc.args });
-                            });
-                            if (!approved) {
+                            const sessionKey = tc.name;
+                            let decision;
+                            if (sessionApprovedRef.current.has(sessionKey)) {
+                                decision = 'yes';
+                            }
+                            else {
+                                decision = await new Promise(resolve => {
+                                    permissionResolveRef.current = resolve;
+                                    setPermissionRequest({ toolName: tc.name, args: tc.args });
+                                });
+                            }
+                            if (decision === 'session')
+                                sessionApprovedRef.current.add(sessionKey);
+                            if (decision === 'no') {
                                 printer.systemMsg(`denied: ${tc.name}`);
                                 next.push({ role: 'user', content: `Tool ${tc.name} was denied by the user` });
                                 break;
                             }
+                            // Checkpoint: store pre-execution file state
+                            if (CHECKPOINT_TOOLS.has(tc.name)) {
+                                const path = tc.args.path;
+                                if (path && !checkpointRef.current.has(path)) {
+                                    try {
+                                        checkpointRef.current.set(path, readFileSync(path, 'utf-8'));
+                                    }
+                                    catch {
+                                        checkpointRef.current.set(path, null);
+                                    }
+                                }
+                            }
                         }
                         if (tool) {
                             try {
                                 printer.toolCallStart(tc.name, tc.args);
                                 const result = await tool.execute(tc.args);
+                                printer.toolResultSummary(tc.name, tc.args, result);
                                 if (SHOW_RESULT_TOOLS.has(tc.name))
                                     printer.toolMsg(tc.name, result);
                                 next.push({ role: 'user', content: `Tool ${tc.name} result:\n${result}` });
@@ -131,6 +187,7 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
                             printer.toolCallStart('run_tests', {});
                             const testResult = await testTool.execute({});
                             if (testResult && !testResult.startsWith('(no test script') && !testResult.startsWith('(no package.json')) {
+                                printer.toolResultSummary('run_tests', {}, testResult);
                                 printer.toolMsg('run_tests', testResult);
                                 next.push({ role: 'user', content: `Test results after edits:\n${testResult}` });
                             }
@@ -156,11 +213,37 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
     }, [config]);
     const handleAbort = useCallback(() => {
         abortRef.current?.abort();
+        sessionApprovedRef.current.clear();
         if (permissionResolveRef.current) {
-            permissionResolveRef.current(false);
+            permissionResolveRef.current('no');
             permissionResolveRef.current = null;
             setPermissionRequest(null);
         }
+        if (compactResolveRef.current) {
+            compactResolveRef.current(false);
+            compactResolveRef.current = null;
+            setCompactRequest(null);
+        }
+        // Restore checkpointed files
+        if (checkpointRef.current.size > 0) {
+            let restored = 0;
+            for (const [path, content] of checkpointRef.current) {
+                try {
+                    if (content === null) {
+                        if (existsSync(path))
+                            unlinkSync(path);
+                    }
+                    else {
+                        writeFileSync(path, content, 'utf-8');
+                    }
+                    restored++;
+                }
+                catch { }
+            }
+            checkpointRef.current.clear();
+            if (restored > 0)
+                printer.systemMsg(`restored ${restored} file(s) to pre-session state`);
+        }
         setStatus('idle');
     }, []);
     return {
@@ -170,5 +253,6 @@ export function useRunLoop(config, currentModelRef, pushHistory, extraTools = []
         thinkingStartRef,
         runLoop, handleAbort,
         permissionRequest, resolvePermission,
+        compactRequest, resolveCompact,
     };
 }

package/dist/tui/hooks/useSession.js

@@ -6,22 +6,22 @@ import * as printer from '../printer.js';
 import { loadLongMemory, saveLongMemory, mergeFacts, formatMemoryBlock } from '../../memory/store.js';
 import { extractFacts } from '../../memory/extractor.js';
 const SHORT_MEMORY_SIZE = 40;
-function buildSystemPrompt(cwd, facts) {
-    return getSystemPrompt(`\n- CWD: ${cwd}`) + formatMemoryBlock(facts);
+function buildSystemPrompt(cwd, facts, extraTools = []) {
+    return getSystemPrompt(`\n- CWD: ${cwd}`, extraTools) + formatMemoryBlock(facts);
 }
-export function useSession(initialSession, cwd, config) {
+export function useSession(initialSession, cwd, config, extraTools = []) {
     const [sessionName, setSessionName] = useState(initialSession);
     const sessionNameRef = useRef(initialSession);
     const historyRef = useRef([]);
     const saveTimerRef = useRef(null);
     const firstMessageSentRef = useRef(false);
     const longMemoryRef = useRef([]);
-    const systemPromptRef = useRef(buildSystemPrompt(cwd, []));
+    const systemPromptRef = useRef(buildSystemPrompt(cwd, [], extraTools));
     useEffect(() => { sessionNameRef.current = sessionName; }, [sessionName]);
     useEffect(() => {
         const facts = loadLongMemory(initialSession);
         longMemoryRef.current = facts;
-        systemPromptRef.current = buildSystemPrompt(cwd, facts);
+        systemPromptRef.current = buildSystemPrompt(cwd, facts, extraTools);
         if (facts.length)
             printer.systemMsg(`long memory: ${facts.length} facts loaded`);
         const history = loadSession(initialSession);
@@ -51,7 +51,7 @@ export function useSession(initialSession, cwd, config) {
                     return;
                 const updated = mergeFacts(longMemoryRef.current, newFacts);
                 longMemoryRef.current = updated;
-                systemPromptRef.current = buildSystemPrompt(cwd, updated);
+                systemPromptRef.current = buildSystemPrompt(cwd, updated, extraTools);
                 saveLongMemory(sessionNameRef.current, updated);
             });
         }

package/dist/tui/hooks/useSubmit.js

@@ -1,10 +1,11 @@
 import { useCallback, useRef } from 'react';
 import { readFile, guardPath } from '../../files/ops.js';
 import { getSystemPrompt } from '../../tools/index.js';
+import { saveConfig } from '../../config.js';
 import { loadSession, saveSession, listSessions, deleteSession, deleteAllSessions } from '../../sessions.js';
+import { compactContext } from '../../tasks/compactor.js';
 import { runDeepThink } from '../deepThink.js';
 import { buildGitContext, looksCodeRelated } from '../git-context.js';
-import { getTavilyKey, saveTavilyKey } from '../../tavily/client.js';
 import { buildIndex } from '../../index/indexer.js';
 import { indexStats, clearIndex } from '../../index/store.js';
 import { embed } from '../../index/embedder.js';
@@ -34,7 +35,7 @@ export function useSubmit(deps) {
     const depsRef = useRef(deps);
     depsRef.current = deps;
     const handleSubmit = useCallback(async (text) => {
-        const { config, skills, cwd, version, currentModelRef, setCurrentModel, historyRef, sessionNameRef, saveTimerRef, systemPromptRef, abortRef, planningMode, setPlanningMode, runLoop, buildContext, pushHistory, setSessionName, renameFromMessage, openPicker, setStatus, setTaskLabel, setCurrentTool, runRefactor, handleGit, lastGitStatusRef, } = depsRef.current;
+        const { config, skills, cwd, version, currentModelRef, setCurrentModel, historyRef, sessionNameRef, saveTimerRef, systemPromptRef, abortRef, setPlanningMode, runLoop, buildContext, pushHistory, setSessionName, renameFromMessage, setStatus, setTaskLabel, setCurrentTool, runRefactor, handleGit, lastGitStatusRef, mcpTools, setConfig, setConfigOpen, } = depsRef.current;
         const cmd = text.trim();
         if (cmd === '?') {
             printer.systemMsg('shortcuts:\n' +
@@ -56,21 +57,6 @@ export function useSubmit(deps) {
             printer.systemMsg(`miii v${version ?? 'unknown'}`);
             return;
         }
-        if (cmd === '/tavily-key' || cmd.startsWith('/tavily-key ')) {
-            const key = cmd.slice(11).trim();
-            if (!key) {
-                const existing = getTavilyKey();
-                printer.systemMsg(existing ? 'Tavily key set (use /tavily-key <key> to update)' : 'No Tavily key set. Usage: /tavily-key tvly-...');
-                return;
-            }
-            if (!key.startsWith('tvly-')) {
-                printer.systemMsg('Key should start with tvly-. Get yours at https://tavily.com');
-                return;
-            }
-            saveTavilyKey(key);
-            printer.systemMsg('Tavily API key saved to ~/.config/miii/tavily.key (mode 600)');
-            return;
-        }
         if (cmd === '/skills' || cmd.startsWith('/skills ')) {
             const sub = cmd.slice(7).trim();
             if (!sub || sub === 'list') {
@@ -110,6 +96,62 @@ export function useSubmit(deps) {
             printer.systemMsg('usage: /skills install <name> | /skills uninstall <name> | /skills list');
             return;
         }
+        if (cmd === '/config' || cmd.startsWith('/config ')) {
+            const sub = cmd.slice(7).trim();
+            if (!sub) {
+                setConfigOpen(true);
+                return;
+            }
+            if (sub.startsWith('provider ')) {
+                const val = sub.slice(9).trim();
+                const valid = ['ollama', 'anthropic', 'openai-compat'];
+                if (!valid.includes(val)) {
+                    printer.systemMsg(`valid providers: ${valid.join(', ')}`);
+                    return;
+                }
+                setConfig(c => ({ ...c, provider: val }));
+                saveConfig({ provider: val });
+                printer.systemMsg(`provider → ${val}`);
+                return;
+            }
+            if (sub.startsWith('model ')) {
+                const val = sub.slice(6).trim();
+                if (!val) {
+                    printer.systemMsg('usage: /config model <name>');
+                    return;
+                }
+                setConfig(c => ({ ...c, model: val }));
+                saveConfig({ model: val });
+                setCurrentModel(val);
+                currentModelRef.current = val;
+                printer.systemMsg(`model → ${val}`);
+                return;
+            }
+            if (sub.startsWith('key ')) {
+                const val = sub.slice(4).trim();
+                if (!val) {
+                    printer.systemMsg('usage: /config key <api-key>');
+                    return;
+                }
+                setConfig(c => ({ ...c, apiKey: val }));
+                saveConfig({ apiKey: val });
+                printer.systemMsg(`apiKey → ${val.slice(0, 8)}…`);
+                return;
+            }
+            if (sub.startsWith('url ')) {
+                const val = sub.slice(4).trim();
+                if (!val) {
+                    printer.systemMsg('usage: /config url <base-url>');
+                    return;
+                }
+                setConfig(c => ({ ...c, baseUrl: val }));
+                saveConfig({ baseUrl: val });
+                printer.systemMsg(`baseUrl → ${val}`);
+                return;
+            }
+            printer.systemMsg('usage: /config [provider|model|key|url] <value>');
+            return;
+        }
         if (cmd === '/model' || cmd.startsWith('/model ')) {
             const name = cmd.slice(6).trim();
             if (!name) {
@@ -121,8 +163,31 @@ export function useSubmit(deps) {
             printer.systemMsg(`model → ${name}`);
             return;
         }
-        if (cmd === '/models') {
-            await openPicker();
+        if (cmd === '/compact') {
+            const full = buildContext();
+            if (full.length <= 2) {
+                printer.systemMsg('nothing to compact');
+                return;
+            }
+            printer.systemMsg(`compacting ${full.length} messages…`);
+            setStatus('thinking');
+            try {
+                const compacted = await compactContext(full, {
+                    provider: config.provider,
+                    model: currentModelRef.current,
+                    baseUrl: config.baseUrl,
+                    apiKey: config.apiKey,
+                }, undefined);
+                historyRef.current = compacted.filter(m => m.role !== 'system');
+                saveSession(sessionNameRef.current, historyRef.current);
+                printer.systemMsg(`compacted: ${full.length} → ${compacted.length} messages`);
+            }
+            catch (e) {
+                printer.errorMsg(`compact failed: ${e}`);
+            }
+            finally {
+                setStatus('idle');
+            }
             return;
         }
         if (cmd === '/new') {
@@ -135,7 +200,7 @@ export function useSubmit(deps) {
             historyRef.current = [];
             setSessionName(newName);
             setPlanningMode(false);
-            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}`);
+            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}`, mcpTools);
             printer.systemMsg(`new session → ${newName}`);
             return;
         }
@@ -143,7 +208,7 @@ export function useSubmit(deps) {
             historyRef.current = [];
             saveSession(sessionNameRef.current, []);
             setPlanningMode(false);
-            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}`);
+            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}`, mcpTools);
             printer.systemMsg('chat cleared');
             return;
         }
@@ -201,7 +266,7 @@ export function useSubmit(deps) {
         if (cmd === '/plan' || cmd.startsWith('/plan ')) {
             const topic = cmd.slice(5).trim();
             setPlanningMode(true);
-            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}\n- MODE: Planning assistant. Help the user plan step by step. Ask clarifying questions. Suggest concrete next steps. Use plain text only — no markdown, no headers, no bold, no bullets with asterisks, no backtick blocks. Use numbered lists and plain indentation for structure.`);
+            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}\n- MODE: Planning assistant. Help the user plan step by step. Ask clarifying questions. Suggest concrete next steps. Use plain text only — no markdown, no headers, no bold, no bullets with asterisks, no backtick blocks. Use numbered lists and plain indentation for structure.`, mcpTools);
             const msg = topic ? `I want to plan: ${topic}` : 'I want to start planning. Help me think through my goals step by step.';
             printer.userMsg(msg);
             pushHistory({ role: 'user', content: msg });
@@ -210,7 +275,7 @@ export function useSubmit(deps) {
         }
         if (cmd === '/plan:done') {
             setPlanningMode(false);
-            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}`);
+            systemPromptRef.current = getSystemPrompt(`\n- CWD: ${cwd}`, mcpTools);
             printer.systemMsg('planning mode off');
             return;
         }

package/dist/tui/printer.js

@@ -192,12 +192,82 @@ function toolLabel(name, args) {
         }
     }
 }
+export function planSummary(tools) {
+    if (!tools.length)
+        return;
+    const header = gray(`─ plan (${tools.length} action${tools.length === 1 ? '' : 's'})`);
+    write(header + '\n');
+    for (const t of tools) {
+        const dot = DELETE_TOOLS.has(t.name) ? red('◦') : EDIT_TOOLS.has(t.name) ? green('◦') : blue('◦');
+        const label = toolLabel(t.name, t.args);
+        write(`  ${dot} ${gray(label)}\n`);
+    }
+}
 export function toolCallStart(name, args) {
     const dot = DELETE_TOOLS.has(name) ? red('●') : EDIT_TOOLS.has(name) ? green('●') : blue('●');
-    write(`  ${dot} ${toolLabel(name, args)}\n`);
+    write(`\n${dot} ${bold(toolLabel(name, args))}\n`);
+}
+export function toolResultSummary(name, args, result) {
+    const a = args;
+    const lines = result.trim().split('\n').filter(Boolean);
+    let summary = '';
+    switch (name) {
+        case 'edit_file':
+        case 'write_file': {
+            const n = (a.content ?? '').split('\n').length;
+            summary = `Wrote ${n} line${n === 1 ? '' : 's'}`;
+            break;
+        }
+        case 'create_file': {
+            const n = (a.content ?? '').split('\n').length;
+            summary = `Created file · ${n} line${n === 1 ? '' : 's'}`;
+            break;
+        }
+        case 'patch_file':
+            summary = lines[0] ?? 'Applied patch';
+            break;
+        case 'delete_file':
+            summary = 'Deleted';
+            break;
+        case 'move_file':
+            summary = `Moved → ${a.to ?? ''}`;
+            break;
+        case 'read_file': {
+            const n = lines.length;
+            summary = `Read ${n} line${n === 1 ? '' : 's'}`;
+            break;
+        }
+        case 'list_files':
+            summary = `Found ${lines.length} file${lines.length === 1 ? '' : 's'}`;
+            break;
+        case 'run_command':
+        case 'run_tests':
+        case 'git_commit':
+        case 'git_status':
+        case 'git_diff':
+        case 'git_log': {
+            const first = lines[0]?.slice(0, 80) ?? '';
+            const more = lines.length > 1 ? ` (+${lines.length - 1} more)` : '';
+            summary = first + more;
+            break;
+        }
+        case 'web_search':
+            summary = `Found ${lines.length} result${lines.length === 1 ? '' : 's'}`;
+            break;
+        case 'web_extract':
+            summary = `Extracted ${lines.length} line${lines.length === 1 ? '' : 's'}`;
+            break;
+        case 'search_codebase':
+            summary = lines[0]?.slice(0, 80) ?? 'Done';
+            break;
+        default:
+            summary = lines[0]?.slice(0, 80) ?? 'Done';
+    }
+    if (summary)
+        write(gray(`  ${summary}`) + '\n');
 }
 export function toolMsg(name, result) {
-    const preview = result.length > 250 ? result.slice(0, 250) + '…' : result;
+    const preview = result.length > 600 ? result.slice(0, 600) + '…' : result;
     const body = preview.trim()
         ? preview.split('\n').map(l => gray('    ' + l)).join('\n')
         : '';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "miii-cli",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "type": "module",
   "description": "The high-performance local AI coding agent for your terminal. Automate complex workflows with local LLMs.",
   "license": "MIT",