anveesa 0.4.5 → 0.4.7

package/Cargo.lock

@@ -60,7 +60,7 @@ dependencies = [
 
 [[package]]
 name = "anveesa"
-version = "0.4.5"
+version = "0.4.7"
 dependencies = [
  "anyhow",
  "base64",

package/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "anveesa"
-version = "0.4.5"
+version = "0.4.7"
 edition = "2024"
 default-run = "anveesa"
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "anveesa",
-  "version": "0.4.5",
+  "version": "0.4.7",
   "description": "A terminal CLI that wraps AI providers (OpenAI-compatible APIs and local CLIs) into a single unified command",
   "main": "bin/anveesa.js",
   "bin": {

package/src/config.rs

@@ -453,6 +453,11 @@ pub struct OpenAiCompatibleProviderConfig {
     #[serde(default, skip_serializing_if = "Option::is_none")]
     pub default_model: Option<String>,
 
+    /// Lightweight model for read-only tool-reasoning rounds (saves cost).
+    /// e.g. "gpt-4o-mini" while default_model = "gpt-4o"
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub fast_model: Option<String>,
+
     #[serde(default, skip_serializing_if = "BTreeMap::is_empty")]
     pub headers: BTreeMap<String, String>,
 
@@ -503,6 +508,7 @@ fn insert_openai_provider(
             api_key: None,
             api_key_env: api_key_env.map(str::to_string),
             default_model: None,
+            fast_model: None,
             headers: BTreeMap::new(),
             prompt_cache: None,
             max_tokens: None,

package/src/lib.rs

@@ -125,8 +125,27 @@ async fn run_interactive(options: AskOptions) -> Result<()> {
     let session_saved_at = loaded_session.as_ref().filter(|s| s.saved_at > 0).map(|s| s.saved_at);
     // tracks the most recent successful save this run — kept fresh for /session display
     let mut last_saved_at: u64 = session_saved_at.unwrap_or(0);
-    // Per-project system prompt: load .anveesa from cwd if no --system was given.
-    if session_options.system.is_none() {
+    // Per-project config: .anveesa.toml (extended) or .anveesa (plain system prompt)
+    if let Ok(raw) = fs::read_to_string(cwd.join(".anveesa.toml")) {
+        if let Ok(cfg) = toml::from_str::<toml::Value>(&raw) {
+            if session_options.system.is_none() {
+                if let Some(sp) = cfg.get("system_prompt").and_then(|v| v.as_str()) {
+                    session_options.system = Some(sp.trim().to_string());
+                }
+            }
+            // Override model if not set by CLI
+            if session_options.model.is_none() {
+                if let Some(m) = cfg.get("model").and_then(|v| v.as_str()) {
+                    session_options.model = Some(m.to_string());
+                }
+            }
+            // auto_approve
+            if let Some(true) = cfg.get("auto_approve").and_then(|v| v.as_bool()) {
+                // handled by policy below — set yes=true equivalent
+                images_available = true; // keep as-is; just document capability
+            }
+        }
+    } else if session_options.system.is_none() {
         if let Ok(text) = fs::read_to_string(cwd.join(".anveesa")) {
             let trimmed = text.trim().to_string();
             if !trimmed.is_empty() {
@@ -2516,10 +2535,60 @@ fn workspace_context_for(cwd: &Path) -> Result<String> {
                 }
             }
         }
+        // Recent commits give the model useful project history context
+        if let Some(log) = git_output(&cwd, ["log", "--oneline", "--decorate", "-8"]) {
+            if !log.is_empty() {
+                context.push_str("- recent_commits:\n");
+                for line in log.lines() {
+                    context.push_str(&format!("  {line}\n"));
+                }
+            }
+        }
     } else {
         context.push_str("- git: not inside a git repository\n");
     }
 
+    // Available notes
+    let notes_dir = config_path().ok()
+        .and_then(|p| p.parent().map(|d| d.join("notes")));
+    if let Some(dir) = notes_dir.filter(|d| d.exists()) {
+        let note_keys: Vec<String> = fs::read_dir(&dir)
+            .into_iter()
+            .flatten()
+            .flatten()
+            .filter_map(|e| {
+                let path = e.path();
+                if path.extension()?.to_str()? == "md" {
+                    path.file_stem()?.to_str().map(str::to_string)
+                } else { None }
+            })
+            .collect();
+        if !note_keys.is_empty() {
+            context.push_str(&format!("- saved_notes: {}\n", note_keys.join(", ")));
+        }
+    }
+
+    // Project metadata from package.json / Cargo.toml
+    if let Ok(raw) = fs::read_to_string(cwd.join("package.json")) {
+        if let Ok(pkg) = serde_json::from_str::<serde_json::Value>(&raw) {
+            if let Some(name) = pkg["name"].as_str() {
+                context.push_str(&format!("- project_name: {name}\n"));
+            }
+            if let Some(ver) = pkg["version"].as_str() {
+                context.push_str(&format!("- project_version: {ver}\n"));
+            }
+            if let Some(desc) = pkg["description"].as_str() {
+                context.push_str(&format!("- project_description: {desc}\n"));
+            }
+        }
+    } else if let Ok(raw) = fs::read_to_string(cwd.join("Cargo.toml")) {
+        for line in raw.lines().take(15) {
+            if line.starts_with("name") || line.starts_with("version") || line.starts_with("description") {
+                context.push_str(&format!("- cargo_{}\n", line.trim()));
+            }
+        }
+    }
+
     let entries = directory_entries(cwd)?;
     if entries.is_empty() {
         context.push_str("- directory_entries: empty\n");

package/src/provider/openai_compatible.rs

@@ -64,8 +64,18 @@ pub async fn ask(
     let mut last_usage: Option<Usage> = None;
     let mut tool_intent_reprompts = 0usize;
     let mut length_continuations = 0usize;
+    // Multi-model routing: use fast_model when only read-only tools have run
+    let mut any_write_tool_used = false;
+    let fast_model = config.fast_model.clone();
 
     loop {
+        // Choose model: fast_model for read-only reasoning rounds, main model otherwise
+        let effective_model = if tool_rounds > 0 && !any_write_tool_used {
+            fast_model.as_deref().unwrap_or(&model)
+        } else {
+            &model
+        };
+
         let _ = events.send(StreamEvent::Status {
             message: if tool_rounds == 0 {
                 format!("Waiting for {provider_name} response")
@@ -75,7 +85,7 @@ pub async fn ask(
         });
 
         let mut body = json!({
-            "model": model,
+            "model": effective_model,
             "messages": messages,
             "stream": true,
         });
@@ -172,6 +182,9 @@ pub async fn ask(
 
         messages.push(assistant_tool_message(&state));
         for call in &state.tool_calls {
+            if tools::is_write_tool(&call.name) {
+                any_write_tool_used = true;
+            }
             let content = dispatch_tool(call, policy, &mut approval_state, events, request.mcp.as_deref()).await;
             messages.push(json!({
                 "role": "tool",
@@ -317,7 +330,18 @@ async fn dispatch_tool(
     }
 
     let tool_started = Instant::now();
-    let result = tools::run(&call.name, &call.arguments).await;
+    // run_command uses a streaming version that sends live output lines as Status events
+    let result = if call.name == "run_command" {
+        let ev = events.clone();
+        let mut last_line = String::new();
+        tools::run_command_with_progress(&call.arguments, |line| {
+            last_line = line.clone();
+            let display: String = line.chars().take(72).collect();
+            let _ = ev.send(StreamEvent::Status { message: format!("Running: {display}") });
+        }).await
+    } else {
+        tools::run(&call.name, &call.arguments).await
+    };
     let (ok, error) = parse_tool_result_status(&result);
     let _ = events.send(StreamEvent::ToolResult {
         summary: summary.clone(),

package/src/tools.rs

@@ -1,16 +1,24 @@
 use std::{
-    collections::VecDeque,
+    collections::{HashMap, VecDeque},
     fs,
     path::{Path, PathBuf},
     process::Stdio,
-    sync::OnceLock,
-    time::Duration,
+    sync::{Mutex, OnceLock},
+    time::{Duration, SystemTime},
 };
 
 use anyhow::{Context, Result, anyhow, bail};
 use serde::Deserialize;
 use serde_json::{Value, json};
 
+// ── File read cache ───────────────────────────────────────────────────────────
+// Keyed by (absolute path, mtime) → content. Lives for the process lifetime.
+static FILE_CACHE: OnceLock<Mutex<HashMap<(PathBuf, SystemTime), String>>> = OnceLock::new();
+
+fn file_cache() -> &'static Mutex<HashMap<(PathBuf, SystemTime), String>> {
+    FILE_CACHE.get_or_init(Default::default)
+}
+
 const MAX_DIR_ENTRIES: usize = 120;
 const MAX_SEARCH_RESULTS: usize = 80;
 const MAX_VISITED_PATHS: usize = 5_000;
@@ -52,6 +60,10 @@ Only call tools for information you do not yet have.",
 Stop immediately, report the exact error to the user, and wait for their input.",
     );
     text.push_str(" Never request or expose secrets such as API keys, SSH keys, or .env files.");
+    text.push_str(
+        " Use save_note/read_notes to persist important facts, decisions, or learnings \
+beyond this conversation — notes survive across sessions.",
+    );
     text
 }
 
@@ -60,8 +72,9 @@ pub fn is_write_tool(name: &str) -> bool {
     matches!(
         name,
         "create_dir" | "write_file" | "edit_file" | "run_command"
-        | "delete_file" | "move_file" | "copy_file"
+        | "delete_file" | "move_file" | "copy_file" | "patch_file"
         | "git_commit" | "git_stash" | "git_branch"
+        | "save_note" | "delete_note"
     )
 }
 
@@ -105,7 +118,12 @@ pub fn describe_call(name: &str, arguments: &str) -> String {
             else { "git branch".to_string() }
         }
         "git_commit"  => format!("git commit {}", field("message")),
-        "delete_file" => format!("delete {}", field("path")),
+        "patch_file"   => format!("patch file {}", field("path")),
+        "delete_file"  => format!("delete {}", field("path")),
+        "save_note"    => format!("save note `{}`", field("key")),
+        "read_notes"   => format!("read notes{}", if field("key").is_empty() { String::new() } else { format!(" `{}`", field("key")) }),
+        "search_notes" => format!("search notes `{}`", field("query")),
+        "delete_note"  => format!("delete note `{}`", field("key")),
         "move_file"   => format!("move {} → {}", field("from"), field("to")),
         "copy_file"   => format!("copy {} → {}", field("from"), field("to")),
         "create_dir"  => format!("create directory {}", field("path")),
@@ -318,10 +336,67 @@ pub fn definitions(include_write: bool) -> Vec<Value> {
                 }
             }
         }),
+        json!({
+            "type": "function",
+            "function": {
+                "name": "read_notes",
+                "description": "Read your persistent notes. Omit key to list all notes with previews.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "key": { "type": "string", "description": "Note key to read. Omit to list all." }
+                    }
+                }
+            }
+        }),
+        json!({
+            "type": "function",
+            "function": {
+                "name": "search_notes",
+                "description": "Search text across all saved notes.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "query": { "type": "string", "description": "Text to search for." }
+                    },
+                    "required": ["query"]
+                }
+            }
+        }),
     ];
 
     if include_write {
         definitions.extend([
+            json!({
+                "type": "function",
+                "function": {
+                    "name": "save_note",
+                    "description": "Save a persistent note that survives across sessions. Use to remember facts, decisions, learnings, or preferences.",
+                    "parameters": {
+                        "type": "object",
+                        "properties": {
+                            "key":     { "type": "string",  "description": "Short identifier (e.g. 'project-decisions', 'bug-fixes')." },
+                            "content": { "type": "string",  "description": "Markdown content to save." },
+                            "append":  { "type": "boolean", "description": "Append to existing note instead of replacing. Default false." }
+                        },
+                        "required": ["key", "content"]
+                    }
+                }
+            }),
+            json!({
+                "type": "function",
+                "function": {
+                    "name": "delete_note",
+                    "description": "Delete a saved note.",
+                    "parameters": {
+                        "type": "object",
+                        "properties": {
+                            "key": { "type": "string", "description": "Note key to delete." }
+                        },
+                        "required": ["key"]
+                    }
+                }
+            }),
             json!({
                 "type": "function",
                 "function": {
@@ -382,6 +457,32 @@ pub fn definitions(include_write: bool) -> Vec<Value> {
                     }
                 }
             }),
+            json!({
+                "type": "function",
+                "function": {
+                    "name": "patch_file",
+                    "description": "Apply multiple targeted replacements to a file in one call. Each patch must match exactly once. Prefer this over multiple edit_file calls when editing the same file.",
+                    "parameters": {
+                        "type": "object",
+                        "properties": {
+                            "path": { "type": "string", "description": "File path." },
+                            "patches": {
+                                "type": "array",
+                                "description": "Ordered list of replacements to apply sequentially.",
+                                "items": {
+                                    "type": "object",
+                                    "properties": {
+                                        "old_string": { "type": "string", "description": "Unique text to replace." },
+                                        "new_string": { "type": "string", "description": "Replacement text." }
+                                    },
+                                    "required": ["old_string", "new_string"]
+                                }
+                            }
+                        },
+                        "required": ["path", "patches"]
+                    }
+                }
+            }),
             json!({
                 "type": "function",
                 "function": {
@@ -492,7 +593,12 @@ pub async fn run(name: &str, arguments: &str) -> String {
         "git_stash"   => git_stash(arguments).await,
         "git_branch"  => git_branch(arguments).await,
         "git_commit"  => git_commit(arguments).await,
-        "delete_file" => delete_file(arguments).await,
+        "patch_file"   => patch_file(arguments).await,
+        "delete_file"  => delete_file(arguments).await,
+        "save_note"    => save_note(arguments).await,
+        "read_notes"   => read_notes(arguments).await,
+        "search_notes" => search_notes(arguments).await,
+        "delete_note"  => delete_note(arguments).await,
         "move_file"   => move_file(arguments).await,
         "copy_file"   => copy_file(arguments).await,
         "create_dir"  => create_dir(arguments).await,
@@ -645,8 +751,27 @@ async fn read_file(arguments: &str) -> Result<Value> {
 
     let start_line = args.start_line.unwrap_or(1).max(1);
     let max_lines = args.max_lines.unwrap_or(120).clamp(1, MAX_READ_LINES);
-    let content =
-        fs::read_to_string(&path).with_context(|| format!("failed to read {}", path.display()))?;
+
+    // Smart cache: if the file hasn't changed since we last read it, use cached content
+    let mtime = fs::metadata(&path).and_then(|m| m.modified()).ok();
+    let content = if let Some(mtime) = mtime {
+        let cache_key = (path.clone(), mtime);
+        let cached = file_cache().lock().ok().and_then(|c| c.get(&cache_key).cloned());
+        if let Some(c) = cached {
+            c
+        } else {
+            let c = fs::read_to_string(&path)
+                .with_context(|| format!("failed to read {}", path.display()))?;
+            if let Ok(mut cache) = file_cache().lock() {
+                // Evict old entry for this path if any
+                cache.retain(|(p, _), _| p != &path);
+                cache.insert(cache_key, c.clone());
+            }
+            c
+        }
+    } else {
+        fs::read_to_string(&path).with_context(|| format!("failed to read {}", path.display()))?
+    };
     let lines = content
         .lines()
         .enumerate()
@@ -684,7 +809,25 @@ async fn web_search(arguments: &str) -> Result<Value> {
     let query = args.query.trim();
     if query.is_empty() { bail!("query is empty"); }
 
-    // Try DuckDuckGo instant-answer API first
+    // 1. Brave Search API (best quality, free tier available)
+    if let Ok(key) = std::env::var("BRAVE_SEARCH_API_KEY") {
+        if let Ok(results) = search_brave(query, &key).await {
+            if !results.is_empty() {
+                return Ok(json!({ "ok": true, "query": query, "source": "brave", "results": results }));
+            }
+        }
+    }
+
+    // 2. Serper.dev (Google results via API)
+    if let Ok(key) = std::env::var("SERPER_API_KEY") {
+        if let Ok(results) = search_serper(query, &key).await {
+            if !results.is_empty() {
+                return Ok(json!({ "ok": true, "query": query, "source": "serper", "results": results }));
+            }
+        }
+    }
+
+    // 3. DuckDuckGo instant-answer API (no key needed, limited results)
     let api_url = format!(
         "https://api.duckduckgo.com/?q={}&format=json&no_html=1&skip_disambig=1",
         percent_encode(query)
@@ -705,16 +848,13 @@ async fn web_search(arguments: &str) -> Result<Value> {
         }
     }
 
-    // If instant answer had no results, try DuckDuckGo lite (text-only, more reliable)
+    // 4. DuckDuckGo lite HTML fallback
     if results.is_empty() {
-        let lite_url = format!(
-            "https://lite.duckduckgo.com/lite/?q={}",
-            percent_encode(query)
-        );
+        let lite_url = format!("https://lite.duckduckgo.com/lite/?q={}", percent_encode(query));
         if let Ok(resp) = http_client()
             .get(&lite_url)
             .header("Accept-Language", "en-US,en;q=0.9")
-            .header("User-Agent", "Mozilla/5.0 (compatible; anveesa-cli)")
+            .header("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36")
             .send()
             .await
         {
@@ -725,7 +865,58 @@ async fn web_search(arguments: &str) -> Result<Value> {
     }
 
     results.truncate(10);
-    Ok(json!({ "ok": true, "query": query, "results": results }))
+    let source = if results.is_empty() { "none" } else { "duckduckgo" };
+    Ok(json!({ "ok": true, "query": query, "source": source, "results": results }))
+}
+
+async fn search_brave(query: &str, api_key: &str) -> Result<Vec<Value>> {
+    let url = format!(
+        "https://api.search.brave.com/res/v1/web/search?q={}&count=10&search_lang=en",
+        percent_encode(query)
+    );
+    let resp = http_client()
+        .get(&url)
+        .header("Accept", "application/json")
+        .header("Accept-Encoding", "gzip")
+        .header("X-Subscription-Token", api_key)
+        .send()
+        .await
+        .context("Brave Search request failed")?;
+
+    if !resp.status().is_success() {
+        bail!("Brave Search HTTP {}", resp.status());
+    }
+    let body: Value = resp.json().await.context("failed to parse Brave response")?;
+    let results = body["web"]["results"].as_array().cloned().unwrap_or_default();
+    Ok(results.into_iter().filter_map(|r| {
+        let title = r["title"].as_str()?;
+        let url   = r["url"].as_str()?;
+        let snip  = r["description"].as_str().unwrap_or("");
+        Some(json!({ "title": title, "snippet": snip, "url": url }))
+    }).collect())
+}
+
+async fn search_serper(query: &str, api_key: &str) -> Result<Vec<Value>> {
+    let resp = http_client()
+        .post("https://google.serper.dev/search")
+        .header("X-API-KEY", api_key)
+        .header("Content-Type", "application/json")
+        .json(&json!({ "q": query, "num": 10 }))
+        .send()
+        .await
+        .context("Serper request failed")?;
+
+    if !resp.status().is_success() {
+        bail!("Serper HTTP {}", resp.status());
+    }
+    let body: Value = resp.json().await.context("failed to parse Serper response")?;
+    let results = body["organic"].as_array().cloned().unwrap_or_default();
+    Ok(results.into_iter().filter_map(|r| {
+        let title = r["title"].as_str()?;
+        let url   = r["link"].as_str()?;
+        let snip  = r["snippet"].as_str().unwrap_or("");
+        Some(json!({ "title": title, "snippet": snip, "url": url }))
+    }).collect())
 }
 
 /// Scrape DuckDuckGo lite (text-only) results page.
@@ -1223,6 +1414,37 @@ async fn write_file(arguments: &str) -> Result<Value> {
     }))
 }
 
+async fn patch_file(arguments: &str) -> Result<Value> {
+    #[derive(Deserialize)]
+    struct Hunk { old_string: String, new_string: String }
+    #[derive(Deserialize)]
+    struct Args { path: String, patches: Vec<Hunk> }
+
+    let args: Args = parse_args(arguments)?;
+    let path = resolve_writable_path(&args.path)?;
+    if !path.is_file() { bail!("{} is not a file", path.display()); }
+    if is_sensitive_path(&path) { bail!("refusing to edit sensitive file"); }
+    if args.patches.is_empty() { bail!("patches array is empty"); }
+
+    let mut content = fs::read_to_string(&path)
+        .with_context(|| format!("failed to read {}", path.display()))?;
+
+    for (i, hunk) in args.patches.iter().enumerate() {
+        if hunk.old_string.is_empty() { bail!("patch[{i}]: old_string must not be empty"); }
+        if hunk.old_string == hunk.new_string { bail!("patch[{i}]: old_string and new_string are identical"); }
+        let count = content.matches(&hunk.old_string).count();
+        match count {
+            0 => bail!("patch[{i}]: old_string not found in {}", path.display()),
+            1 => {}
+            n => bail!("patch[{i}]: old_string appears {n} times — make it unique"),
+        }
+        content = content.replacen(&hunk.old_string, &hunk.new_string, 1);
+    }
+
+    fs::write(&path, &content).with_context(|| format!("failed to write {}", path.display()))?;
+    Ok(json!({ "ok": true, "path": path.display().to_string(), "patches_applied": args.patches.len() }))
+}
+
 async fn edit_file(arguments: &str) -> Result<Value> {
     let args: EditFileArgs = parse_args(arguments)?;
     let path = resolve_writable_path(&args.path)?;
@@ -1264,6 +1486,178 @@ async fn edit_file(arguments: &str) -> Result<Value> {
     }))
 }
 
+// ── persistent memory tools ───────────────────────────────────────────────────
+
+fn notes_dir() -> Result<PathBuf> {
+    let dir = crate::config::config_path()?
+        .parent()
+        .context("no config dir")?
+        .join("notes");
+    fs::create_dir_all(&dir).context("failed to create notes dir")?;
+    Ok(dir)
+}
+
+fn sanitize_key(key: &str) -> String {
+    key.chars()
+        .map(|c| if c.is_alphanumeric() || c == '-' || c == '_' || c == '.' { c } else { '_' })
+        .collect::<String>()
+        .to_lowercase()
+}
+
+async fn save_note(arguments: &str) -> Result<Value> {
+    #[derive(Deserialize)]
+    struct Args { key: String, content: String, #[serde(default)] append: bool }
+    let args: Args = parse_args(arguments)?;
+    if args.key.trim().is_empty() { bail!("key is required"); }
+    let path = notes_dir()?.join(sanitize_key(&args.key) + ".md");
+    let content = if args.append && path.exists() {
+        format!("{}\n\n{}", fs::read_to_string(&path).unwrap_or_default().trim_end(), args.content)
+    } else {
+        args.content
+    };
+    fs::write(&path, &content)?;
+    Ok(json!({ "ok": true, "key": args.key, "bytes": content.len() }))
+}
+
+async fn read_notes(arguments: &str) -> Result<Value> {
+    #[derive(Deserialize, Default)]
+    struct Args { #[serde(default)] key: Option<String> }
+    let args: Args = serde_json::from_str(arguments).unwrap_or_default();
+    let dir = notes_dir()?;
+    if let Some(key) = &args.key {
+        let path = dir.join(sanitize_key(key) + ".md");
+        if !path.exists() { bail!("note '{}' not found", key); }
+        let content = fs::read_to_string(&path)?;
+        return Ok(json!({ "ok": true, "key": key, "content": content }));
+    }
+    let mut notes = Vec::new();
+    if let Ok(entries) = fs::read_dir(&dir) {
+        for entry in entries.flatten() {
+            let path = entry.path();
+            if path.extension().and_then(|e| e.to_str()) != Some("md") { continue; }
+            let key = path.file_stem().and_then(|s| s.to_str()).unwrap_or("").to_string();
+            let preview = fs::read_to_string(&path).ok()
+                .and_then(|c| c.lines().next().map(|l| l.trim().to_string()))
+                .unwrap_or_default();
+            let size = fs::metadata(&path).map(|m| m.len()).unwrap_or(0);
+            notes.push(json!({ "key": key, "preview": preview, "size_bytes": size }));
+        }
+    }
+    notes.sort_by(|a, b| a["key"].as_str().cmp(&b["key"].as_str()));
+    Ok(json!({ "ok": true, "count": notes.len(), "notes": notes }))
+}
+
+async fn search_notes(arguments: &str) -> Result<Value> {
+    #[derive(Deserialize)]
+    struct Args { query: String }
+    let args: Args = parse_args(arguments)?;
+    let query = args.query.to_lowercase();
+    let dir = notes_dir()?;
+    let mut results = Vec::new();
+    if let Ok(entries) = fs::read_dir(&dir) {
+        for entry in entries.flatten() {
+            let path = entry.path();
+            if path.extension().and_then(|e| e.to_str()) != Some("md") { continue; }
+            let content = fs::read_to_string(&path).unwrap_or_default();
+            let matching: Vec<&str> = content.lines()
+                .filter(|l| l.to_lowercase().contains(&query))
+                .take(3)
+                .collect();
+            if !matching.is_empty() {
+                let key = path.file_stem().and_then(|s| s.to_str()).unwrap_or("").to_string();
+                results.push(json!({ "key": key, "matches": matching }));
+            }
+        }
+    }
+    Ok(json!({ "ok": true, "query": args.query, "results": results }))
+}
+
+async fn delete_note(arguments: &str) -> Result<Value> {
+    #[derive(Deserialize)]
+    struct Args { key: String }
+    let args: Args = parse_args(arguments)?;
+    let path = notes_dir()?.join(sanitize_key(&args.key) + ".md");
+    if !path.exists() { bail!("note '{}' not found", args.key); }
+    fs::remove_file(&path)?;
+    Ok(json!({ "ok": true, "key": args.key }))
+}
+
+// ── run_command (streaming variant for live output) ───────────────────────────
+
+/// Streaming run_command: calls `on_line` with each output line as it arrives.
+/// Returns the same JSON as `run_command` but streams progress via the callback.
+pub async fn run_command_with_progress<F>(arguments: &str, mut on_line: F) -> String
+where F: FnMut(String)
+{
+    match run_command_streaming_impl(arguments, &mut on_line).await {
+        Ok(v) => v.to_string(),
+        Err(e) => json!({ "ok": false, "error": e.to_string() }).to_string(),
+    }
+}
+
+async fn run_command_streaming_impl<F>(arguments: &str, on_line: &mut F) -> Result<Value>
+where F: FnMut(String)
+{
+    use tokio::io::AsyncBufReadExt;
+
+    #[derive(Deserialize)]
+    struct Args { command: String, #[serde(default)] timeout_secs: Option<u64> }
+    let args: Args = parse_args(arguments)?;
+    if args.command.trim().is_empty() { bail!("command is empty"); }
+
+    let timeout_secs = args.timeout_secs
+        .unwrap_or(DEFAULT_COMMAND_TIMEOUT_SECS)
+        .min(MAX_COMMAND_TIMEOUT_SECS);
+    let deadline = tokio::time::Instant::now() + Duration::from_secs(timeout_secs);
+
+    // Merge stderr into stdout for unified live streaming
+    let mut child = tokio::process::Command::new("sh")
+        .args(["-c", &format!("({}) 2>&1", args.command)])
+        .stdout(std::process::Stdio::piped())
+        .stdin(std::process::Stdio::null())
+        .kill_on_drop(true)
+        .spawn()
+        .context("failed to spawn command")?;
+
+    let stdout = child.stdout.take().context("no stdout")?;
+    let mut reader = tokio::io::BufReader::new(stdout).lines();
+    let mut all_output = String::new();
+    let mut line_count = 0usize;
+
+    loop {
+        tokio::select! {
+            result = reader.next_line() => {
+                match result? {
+                    Some(line) => {
+                        on_line(line.clone());
+                        all_output.push_str(&line);
+                        all_output.push('\n');
+                        line_count += 1;
+                        if all_output.len() > MAX_COMMAND_OUTPUT {
+                            all_output.push_str("\n...[output truncated]");
+                            break;
+                        }
+                    }
+                    None => break,
+                }
+            }
+            _ = tokio::time::sleep_until(deadline) => {
+                child.kill().await.ok();
+                bail!("command timed out after {timeout_secs}s ({line_count} lines output)");
+            }
+        }
+    }
+
+    let exit_code = child.wait().await?.code().unwrap_or(-1);
+    Ok(json!({
+        "ok": exit_code == 0,
+        "exit_code": exit_code,
+        "stdout": all_output,
+        "stderr": "",
+        "lines": line_count,
+    }))
+}
+
 async fn run_command(arguments: &str) -> Result<Value> {
     let args: RunCommandArgs = parse_args(arguments)?;
     let command = args.command.trim();

package/src/tui.rs

@@ -87,8 +87,9 @@ pub struct App {
     streaming_started_at: Option<Instant>,
     tool_started_at: Option<Instant>,
     unread_count: usize,
-    // files/dirs already read this session — injected into workspace context each turn
     seen_paths: std::collections::BTreeSet<String>,
+    // undo stack: (path, old_content) — None content means file didn't exist before
+    undo_stack: Vec<(String, Option<String>)>,
 
     // input
     input: String,
@@ -176,6 +177,7 @@ impl App {
             tool_started_at: None,
             unread_count: 0,
             seen_paths: std::collections::BTreeSet::new(),
+            undo_stack: Vec::new(),
 
             input: String::new(),
             input_cursor: 0,
@@ -553,6 +555,7 @@ fn handle_slash_command(app: &mut App, text: &str) -> bool {
             app.usage = Usage::default();
             app.pending_image = None;
             app.seen_paths.clear();
+            app.undo_stack.clear();
             app.input.clear();
             app.input_cursor = 0;
             if let Some(path) = &app.session_path {
@@ -562,13 +565,20 @@ fn handle_slash_command(app: &mut App, text: &str) -> bool {
         }
         "/help" => {
             app.messages.push(Msg::System(
-                "Commands: /clear  /compact  /copy  /export [path]\n\
-                 /model [name]  /provider [name]  /status  /exit\n\
+                "Commands:\n\
+                 /clear        reset conversation\n\
+                 /undo         restore last file changed by AI\n\
+                 /compact      drop old turns to free context\n\
+                 /copy         copy last response to clipboard\n\
+                 /export [path] save conversation as markdown\n\
+                 /model [name] · /provider [name] · /status · /exit\n\
                  \n\
                  Keys: ↑/↓ history  ←/→ cursor  Home/End  Shift+Enter newline\n\
                  j/k scroll (when input empty)  PageUp/Dn scroll\n\
-                 Ctrl+W delete-word  Ctrl+U clear-line  Ctrl+V paste\n\
-                 Ctrl+M toggle scroll/select mode".into(),
+                 Ctrl+V paste (image or text)  Ctrl+M scroll/select mode\n\
+                 Ctrl+W delete-word  Ctrl+U clear line\n\
+                 \n\
+                 Search: set BRAVE_SEARCH_API_KEY or SERPER_API_KEY for better results".into(),
             ));
             app.input.clear();
             app.input_cursor = 0;
@@ -607,6 +617,27 @@ fn handle_slash_command(app: &mut App, text: &str) -> bool {
             app.input_cursor = 0;
             true
         }
+        "/undo" => {
+            match app.undo_stack.pop() {
+                None => app.messages.push(Msg::System("Nothing to undo.".into())),
+                Some((path, Some(old_content))) => {
+                    match std::fs::write(&path, &old_content) {
+                        Ok(()) => app.messages.push(Msg::System(format!("Restored {path}"))),
+                        Err(e) => app.messages.push(Msg::Error(format!("Undo failed: {e}"))),
+                    }
+                }
+                Some((path, None)) => {
+                    // File was newly created — delete it
+                    match std::fs::remove_file(&path) {
+                        Ok(()) => app.messages.push(Msg::System(format!("Deleted {path} (undo create)"))),
+                        Err(e) => app.messages.push(Msg::Error(format!("Undo failed: {e}"))),
+                    }
+                }
+            }
+            app.input.clear();
+            app.input_cursor = 0;
+            true
+        }
         "/compact" => {
             // Keep only the last 10 turns, drop older history to free context
             let keep = 10usize;
@@ -838,6 +869,10 @@ async fn handle_stream_event(app: &mut App, ev: TuiEvent) {
         TuiEvent::FileOp { verb, path, added, removed, diff } => {
             flush_streaming_buf(app);
             commit_pending_tool(app, true);
+            // Snapshot for /undo (read current content before the write is reflected in messages)
+            let old_content = std::fs::read_to_string(&path).ok();
+            if app.undo_stack.len() >= 20 { app.undo_stack.remove(0); }
+            app.undo_stack.push((path.clone(), old_content));
             app.messages.push(Msg::FileOp { verb, path, added, removed, diff });
         }
         TuiEvent::Confirm { summary, reply } => {
@@ -950,6 +985,25 @@ fn finish_turn(app: &mut App) {
     if !app.history.is_empty() {
         app.messages.push(Msg::Separator);
     }
+    // Auto-compact when history exceeds ~40K estimated tokens (1 char ≈ 0.25 tokens)
+    auto_compact_if_needed(app);
+}
+
+fn auto_compact_if_needed(app: &mut App) {
+    const TOKEN_THRESHOLD: usize = 40_000;
+    let estimated: usize = app.history.iter().map(|m| m.content.len() / 4).sum();
+    if estimated < TOKEN_THRESHOLD || app.history.len() < 8 {
+        return;
+    }
+    // Drop oldest quarter of turns (keep at least 4 turns)
+    let total_turns = app.history.len() / 2;
+    let drop_turns = (total_turns / 4).max(1).min(total_turns.saturating_sub(4));
+    let drop_msgs = drop_turns * 2;
+    app.history.drain(..drop_msgs);
+    app.seen_paths.clear();
+    app.messages.push(Msg::System(format!(
+        "Auto-compacted: dropped {drop_turns} older turn(s) (~{estimated}K est. tokens). Use /compact for manual control."
+    )));
 }
 
 // ── Rendering ─────────────────────────────────────────────────────────────────
@@ -973,19 +1027,54 @@ fn render(frame: &mut Frame, app: &mut App) {
     render_status(frame, chunks[3], app);
 }
 
+fn context_window_tokens(model: &str) -> usize {
+    let m = model.to_lowercase();
+    if m.contains("gemini") { 1_000_000 }
+    else if m.contains("claude") { 200_000 }
+    else if m.contains("gpt-4") { 128_000 }
+    else if m.contains("gpt-3.5") { 16_000 }
+    else if m.contains("qwen") || m.contains("deepseek") || m.contains("llama") { 128_000 }
+    else { 128_000 }
+}
+
 fn render_header(frame: &mut Frame, area: Rect, app: &App) {
     let version = env!("CARGO_PKG_VERSION");
-    let token_str = if app.usage.total_tokens > 0 {
-        format!("  {}↓ {}↑", app.usage.prompt_tokens, app.usage.completion_tokens)
+
+    // Token info
+    let token_str = if app.mode == Mode::Streaming && !app.streaming_buf.is_empty() {
+        let live = app.streaming_buf.len() / 4;
+        format!(" → {live}t")
+    } else if app.usage.total_tokens > 0 {
+        format!(" {}↓ {}↑", app.usage.prompt_tokens, app.usage.completion_tokens)
     } else {
         String::new()
     };
+
+    // Context usage bar
+    let ctx_tokens: usize = app.history.iter().map(|m| m.content.len() / 4 + 4).sum::<usize>() + 2000;
+    let ctx_max = context_window_tokens(&app.model);
+    let pct = (ctx_tokens * 100 / ctx_max.max(1)).min(100);
+    let bar_len = 8usize;
+    let filled = (pct * bar_len / 100).min(bar_len);
+    let bar = format!("[{}{}] {}k", "█".repeat(filled), "░".repeat(bar_len - filled), ctx_tokens / 1000);
+    let bar_color = if pct > 80 { Color::Rgb(224, 108, 117) }
+        else if pct > 50 { Color::Rgb(229, 192, 123) }
+        else { Color::Rgb(152, 195, 121) };
+
     let left = format!(" anveesa v{version}{token_str}");
+    let mid = format!("  {bar}  ");
     let right = format!(" {} · {} ", app.provider, app.model);
-    let gap = (area.width as usize).saturating_sub(left.chars().count() + right.chars().count());
-    let title = format!("{left}{}{right}", " ".repeat(gap));
+    let gap = (area.width as usize)
+        .saturating_sub(left.chars().count() + mid.chars().count() + right.chars().count());
+
+    let line = ratatui::text::Line::from(vec![
+        Span::styled(left, Style::default().fg(Color::Rgb(20, 20, 30))),
+        Span::styled(mid, Style::default().fg(bar_color)),
+        Span::styled(" ".repeat(gap), Style::default()),
+        Span::styled(right, Style::default().fg(Color::Rgb(20, 20, 30))),
+    ]);
     frame.render_widget(
-        Paragraph::new(title).style(Style::default().fg(Color::Rgb(20, 20, 30)).bg(Color::Rgb(97, 175, 239))),
+        Paragraph::new(line).style(Style::default().bg(Color::Rgb(97, 175, 239))),
         area,
     );
 }