vtcode 0.98.6

A Rust-based terminal coding agent with modular architecture supporting multiple LLM providers
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

use anyhow::Result;
use vtcode_core::config::loader::VTCodeConfig;
use vtcode_core::utils::ansi::{AnsiRenderer, MessageStyle};
use vtcode_tui::ui::interactive_list::SelectionEntry;

use super::common::{prompt_with_placeholder, run_selection};

pub(crate) fn resolve_initial_persistent_memory_enabled(config: &VTCodeConfig) -> bool {
    config.agent.persistent_memory.enabled
}

pub(crate) fn prompt_persistent_memory(
    renderer: &mut AnsiRenderer,
    default_enabled: bool,
) -> Result<bool> {
    renderer.line(
        MessageStyle::Status,
        "Persistent memory controls whether VT Code keeps durable repository notes between sessions.",
    )?;
    renderer.line(
        MessageStyle::Info,
        "Leave it off if you want a lighter first-run setup. You can enable it later in `vtcode.toml`.",
    )?;

    match select_persistent_memory_with_ratatui(default_enabled) {
        Ok(enabled) => Ok(enabled),
        Err(error) => {
            renderer.line(
                MessageStyle::Info,
                &format!("Falling back to manual input ({error})."),
            )?;
            prompt_persistent_memory_text(renderer, default_enabled)
        }
    }
}

fn persistent_memory_entries() -> [(bool, SelectionEntry); 2] {
    [
        (
            false,
            SelectionEntry::new(
                "Off (recommended)".to_string(),
                Some("Do not write durable repository memory yet.".to_string()),
            ),
        ),
        (
            true,
            SelectionEntry::new(
                "On".to_string(),
                Some("Keep opt-in repository memory summaries for future sessions.".to_string()),
            ),
        ),
    ]
}

fn select_persistent_memory_with_ratatui(default_enabled: bool) -> Result<bool> {
    let entries = persistent_memory_entries();
    let default_index = usize::from(default_enabled);
    let selection_entries: Vec<SelectionEntry> = entries
        .iter()
        .map(|(_enabled, entry)| entry.clone())
        .collect();
    let instructions = if default_enabled {
        "Default: On. Use ↑/↓ or j/k to choose, Enter to confirm, Esc to keep the default."
    } else {
        "Default: Off. Use ↑/↓ or j/k to choose, Enter to confirm, Esc to keep the default."
    };
    let selected_index = run_selection(
        "Persistent memory",
        instructions,
        &selection_entries,
        default_index,
    )?;
    Ok(entries[selected_index].0)
}

fn prompt_persistent_memory_text(
    renderer: &mut AnsiRenderer,
    default_enabled: bool,
) -> Result<bool> {
    let entries = persistent_memory_entries();
    for (index, (enabled, _entry)) in entries.iter().enumerate() {
        renderer.line(
            MessageStyle::Info,
            &format!(
                "  {}) {}",
                index + 1,
                if *enabled { "On" } else { "Off (recommended)" }
            ),
        )?;
    }

    loop {
        let default_label = if default_enabled { "on" } else { "off" };
        let input = prompt_with_placeholder(&format!("Persistent memory [{}]", default_label))?;
        let trimmed = input.trim();
        if trimmed.is_empty() {
            return Ok(default_enabled);
        }

        match trimmed.to_ascii_lowercase().as_str() {
            "1" | "off" | "disable" | "disabled" | "no" | "n" => return Ok(false),
            "2" | "on" | "enable" | "enabled" | "yes" | "y" => return Ok(true),
            _ => renderer.line(
                MessageStyle::Error,
                "Please choose Off or On for persistent memory.",
            )?,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use vtcode_core::config::loader::VTCodeConfig;

    #[test]
    fn persistent_memory_entries_are_unnumbered() {
        let entries = persistent_memory_entries();

        assert_eq!(entries[0].1.title, "Off (recommended)");
        assert_eq!(entries[1].1.title, "On");
    }

    #[test]
    fn resolve_initial_persistent_memory_enabled_reads_config() {
        let mut config = VTCodeConfig::default();
        config.agent.persistent_memory.enabled = true;

        assert!(resolve_initial_persistent_memory_enabled(&config));
    }
}