sage-cli 0.13.57

Command-line interface for Sage Agent
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

//! Message display component

use rnk::prelude::*;
use sage_core::ui::bridge::state::{Message, Role, UiMessageContent};

use crate::ui::rnk_app::formatting::{truncate_to_width, wrap_text_with_prefix};
use crate::ui::rnk_app::theme::TerminalTheme;

use super::tool_call::render_tool_call;

// Alias rnk's Box to avoid conflict with std::boxed::Box
use rnk::prelude::Box as RnkBox;

fn role_color(role: &Role, theme: &TerminalTheme) -> Color {
    match role {
        Role::User => theme.accent_user,
        Role::Assistant => theme.accent_assistant,
        Role::System => theme.accent_system,
    }
}

fn content_line(color: Color, text: String, text_color: Color, is_first: bool) -> Element {
    RnkBox::new()
        .flex_direction(FlexDirection::Row)
        .child(
            Text::new(if is_first { "● " } else { "  " })
                .color(color)
                .into_element(),
        )
        .child(Text::new(text).color(text_color).into_element())
        .into_element()
}

/// Format a message for printing via rnk::println
pub fn format_message(msg: &Message, theme: &TerminalTheme) -> Element {
    let term_width = crossterm::terminal::size()
        .map(|(w, _)| w as usize)
        .unwrap_or(80);
    let color = role_color(&msg.role, theme);

    match &msg.content {
        UiMessageContent::Text(text) => {
            let mut container = RnkBox::new().flex_direction(FlexDirection::Column);

            match msg.role {
                Role::User | Role::Assistant => {
                    let mut is_first = true;
                    for paragraph in text.split('\n') {
                        let wrapped =
                            wrap_text_with_prefix("", paragraph, term_width.saturating_sub(4));
                        for line in wrapped {
                            container = container.child(content_line(
                                color,
                                line,
                                theme.text_primary,
                                is_first,
                            ));
                            is_first = false;
                        }
                    }
                }
                Role::System => {
                    let sys_text = truncate_to_width(text, term_width.saturating_sub(4));
                    container =
                        container.child(content_line(color, sys_text, theme.text_muted, true));
                }
            }

            container.into_element()
        }
        UiMessageContent::Thinking(text) => {
            let preview: String = text.lines().take(2).collect::<Vec<_>>().join(" ");
            RnkBox::new()
                .flex_direction(FlexDirection::Row)
                .child(Text::new("💭 ").color(theme.text_subtle).into_element())
                .child(
                    Text::new(format!(
                        "{}…",
                        truncate_to_width(&preview, term_width.saturating_sub(6))
                    ))
                    .color(theme.text_subtle)
                    .into_element(),
                )
                .into_element()
        }
        UiMessageContent::ToolCall {
            tool_name,
            params,
            result,
        } => render_tool_call(tool_name, params, result.as_ref(), theme),
    }
}