newt-tui 0.8.4

Newt-Agent TUI surfaces (ratatui): code mode + pilot mode
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

//! #2959: the stateless part of the per-turn system message, shared by the
//! live turn and the `/context request` preview. Pure over borrowed text, so
//! the preview cannot touch session state by going through it.

use newt_core::agentic::request_report::Annotation;

pub(crate) struct TurnSystemParts<'a> {
    pub ws_state: &'a str,
    pub session_cwd: &'a std::path::Path,
    pub runtime_ctx: &'a str,
    pub system: &'a str,
    pub session_controls: &'a str,
    /// Empty adds no bytes.
    pub communication_style: &'a str,
    pub is_clarification_answer: bool,
}

pub(crate) struct TurnSystem {
    pub text: String,
    /// Label and source of each piece, in assembly order, for the request report.
    pub notes: Vec<Annotation>,
}

/// The session's semantic config with the pulled default embedding model
/// adopted (#1279). Shared by the live turn and the preview so both agree on
/// whether `code_search` is advertised.
pub(crate) fn session_semantic_config(cfg: &newt_core::Config) -> newt_core::SemanticConfig {
    let mut semantic = cfg
        .context
        .as_ref()
        .map(|c| c.semantic.clone())
        .unwrap_or_default();
    semantic.embedding_model_path = crate::effective_embedding_model_path(
        semantic.embedding_model_path.take(),
        newt_inference::palette::embed_model_dir_if_present(),
    );
    semantic
}

fn note(label: &str, source: &str, text: &str) -> Annotation {
    Annotation {
        label: label.into(),
        source: source.into(),
        text: text.into(),
    }
}

pub(crate) fn assemble_turn_system(p: &TurnSystemParts<'_>) -> TurnSystem {
    let mut notes = vec![
        note(
            "workspace state",
            "newt-tui workspace_state_block",
            p.ws_state,
        ),
        note(
            "runtime context",
            "newt-tui runtime_context_block",
            p.runtime_ctx,
        ),
        note(
            "session controls",
            "newt-tui session_control_prompt",
            p.session_controls,
        ),
    ];
    let mut text = format!(
        "{}\n\nCurrent command working directory: {}. \
         run_command without cwd starts here. Explicit relative cwd and \
         relative file-tool paths resolve from the workspace root. \
         This does not change granted access.\n\n{}\n\n{}\n\n{}",
        p.ws_state,
        serde_json::json!(p.session_cwd),
        p.runtime_ctx,
        p.system,
        p.session_controls,
    );
    if !p.communication_style.is_empty() {
        notes.push(note(
            "communication style",
            "newt-tui personality prompt_block",
            p.communication_style,
        ));
        text.push_str("\n\n");
        text.push_str(p.communication_style);
    }
    if p.is_clarification_answer {
        text = format!(
            "<clarification_context>\n\
             This accepted operator line is a clarification continuation. The protected prompt card names its objective root; use prompt_read {{\"address\":\"root\"}} to re-read the original objective before acting whenever the answer alone is insufficient.\n\
             </clarification_context>\n\n{text}"
        );
    }
    TurnSystem { text, notes }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn parts<'a>(style: &'a str, clar: bool) -> TurnSystemParts<'a> {
        TurnSystemParts {
            ws_state: "WS",
            session_cwd: std::path::Path::new("/w"),
            runtime_ctx: "RT",
            system: "SYS",
            session_controls: "CTL",
            communication_style: style,
            is_clarification_answer: clar,
        }
    }

    #[test]
    fn pieces_join_in_turn_order_and_style_is_optional() {
        let t = assemble_turn_system(&parts("", false));
        assert!(t
            .text
            .starts_with("WS\n\nCurrent command working directory: \"/w\"."));
        assert!(t.text.ends_with("RT\n\nSYS\n\nCTL"));
        assert_eq!(t.notes.len(), 3);
        let t = assemble_turn_system(&parts("STYLE", false));
        assert!(t.text.ends_with("CTL\n\nSTYLE"));
        assert_eq!(t.notes.last().unwrap().label, "communication style");
    }

    #[test]
    fn clarification_wraps_the_head() {
        let t = assemble_turn_system(&parts("", true));
        assert!(t.text.starts_with("<clarification_context>"));
        assert!(t.text.ends_with("RT\n\nSYS\n\nCTL"));
    }
}