sofos 0.4.2

An interactive AI coding agent for your terminal
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

//! Short preview string for the session index: the first non-empty user
//! message — skipping injected `[SYSTEM:` preambles — trimmed to the
//! per-entry display budget used as each saved-session row label.

use crate::api::Message;
use crate::session::history::HistoryManager;

pub(super) const MAX_PREVIEW_LENGTH: usize = 120;

impl HistoryManager {
    pub(super) fn extract_preview(messages: &[Message]) -> String {
        for message in messages {
            if message.role == "user" {
                let text = match &message.content {
                    crate::api::MessageContent::Text { content } => content,
                    crate::api::MessageContent::Blocks { content } => content
                        .iter()
                        .find_map(|block| {
                            if let crate::api::MessageContentBlock::Text { text, .. } = block {
                                Some(text.as_str())
                            } else {
                                None
                            }
                        })
                        .unwrap_or(""),
                };

                let preview = text.trim();
                // Skip injected `[SYSTEM:` preambles; title by the first real message.
                if preview.is_empty() || preview.starts_with(crate::config::SYSTEM_MESSAGE_PREFIX) {
                    continue;
                }

                return if preview.chars().count() > MAX_PREVIEW_LENGTH {
                    let truncate_at = preview
                        .char_indices()
                        .nth(MAX_PREVIEW_LENGTH)
                        .map(|(idx, _)| idx)
                        .unwrap_or(preview.len());
                    format!("{}...", &preview[..truncate_at])
                } else {
                    preview.to_string()
                };
            }
        }
        "Empty session".to_string()
    }
}