opencrabs 0.5.11

//! `/clear`: start the agent fresh at the current point, at no cost.
//!
//! A failed manual `/compact` on a large context is expensive: the whole
//! post-marker snapshot goes to the summariser, and on failure the walk
//! sends it again to every provider in the chain. When the user only wants
//! to move on there was no LLM-free way to do that; `/new` starts a
//! different session row and takes the visible history and title with it
//! (#1585).
//!
//! Clearing is one user row starting with the compaction-marker prefix. The
//! context loader already cuts at the last such row and the TUI already
//! hides it on reload, so history stays on screen and in the database, the
//! session keeps its id and title, and the next turn loads only the marker
//! body. That body names the session, points the agent at
//! `session_search`, and quotes the last six exchanges so the fresh start
//! keeps a warm tail of the conversation instead of none (#1905).

use uuid::Uuid;

use super::builder::AgentService;
use crate::brain::agent::error::{AgentError, Result};
use crate::services::{MessageService, SessionService};

/// Prefix the context loader and the TUI reload both key on. Must start
/// with the compaction marker text, which is the only thing either checks.
pub(crate) const CLEAR_MARKER_PREFIX: &str = "[CONTEXT COMPACTION: cleared by the user]";

/// What a failed manual compaction tells the user about the way out.
pub(crate) const CLEAR_HINT: &str = "To continue fresh without a summary: cancel if it is \
     still running, then run /clear. The history stays in the session and the agent \
     starts from that point at no cost.";

/// How many user/assistant pairs the marker quotes as context.
const RECENT_PAIRS: usize = 6;

/// Per-message cap for the quoted tail, in characters so accented text
/// never splits a codepoint. Bounds the marker on chatty sessions.
const RECENT_MESSAGE_CAP: usize = 500;

/// Rows the quoted tail never includes: this prefix covers both `/clear`
/// markers and ordinary compaction summaries, neither of which is
/// conversation worth quoting back.
fn is_compaction_row(content: &str) -> bool {
    content.starts_with("[CONTEXT COMPACTION")
}

/// The pre-clear rows worth quoting: non-empty, not themselves markers,
/// newest last, capped at the last six pairs.
fn recent_tail(rows: &[(String, String)]) -> Vec<(String, String)> {
    let relevant: Vec<(String, String)> = rows
        .iter()
        .filter(|(_, content)| !content.trim().is_empty() && !is_compaction_row(content))
        .cloned()
        .collect();
    let start = relevant.len().saturating_sub(RECENT_PAIRS * 2);
    relevant[start..].to_vec()
}

/// One quoted message, trimmed and capped.
fn quoted(content: &str) -> String {
    let trimmed = content.trim();
    if trimmed.chars().count() <= RECENT_MESSAGE_CAP {
        return trimmed.to_string();
    }
    let cut: String = trimmed.chars().take(RECENT_MESSAGE_CAP).collect();
    format!("{cut} [...]")
}

/// The pairs section appended under the instructions, if anything carried.
fn quoted_exchanges(recent: &[(String, String)]) -> Option<String> {
    if recent.is_empty() {
        return None;
    }
    let lines: Vec<String> = recent
        .iter()
        .map(|(role, content)| format!("{role}: {}", quoted(content)))
        .collect();
    Some(lines.join("\n\n"))
}

/// What the agent reads as its whole context after a clear.
pub(crate) fn clear_marker(session_title: Option<&str>, recent: &[(String, String)]) -> String {
    let title = session_title
        .map(str::trim)
        .filter(|t| !t.is_empty())
        .unwrap_or("untitled");
    let exchanges = quoted_exchanges(&recent_tail(recent));
    // The opening sentence must match what follows: the empty-history
    // variant denies any carryover, the tailed one admits the quote.
    let carry = if exchanges.is_some() {
        "Only the last exchanges quoted below carry over; everything older is \
         out of your context."
    } else {
        "Nothing said or done before it is in your context."
    };
    let base = format!(
        "{CLEAR_MARKER_PREFIX}\n\n\
         The user cleared this session's context at this point. {carry} The full history \
         is still stored in this session, titled '{title}'. If you need something from \
         before, use the session_search tool: operation 'tail' with session '{title}' \
         reads the last messages, operation 'search' with a query and session '{title}' \
         finds specific content. Fetch only what the task needs. Continue from the \
         user's next message."
    );
    match exchanges {
        None => base,
        Some(exchanges) => format!(
            "{base}\n\n\
             The last exchanges before the clear, quoted as context only:\n\n\
             {exchanges}"
        ),
    }
}

/// What happened, for the surface that asked.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ClearReceipt {
    pub session_title: Option<String>,
    /// A background summariser was in flight and has been aborted, so its
    /// summary of the old context can never land over the cleared one.
    pub aborted_background_compaction: bool,
}

impl ClearReceipt {
    /// One line for the user, the same on every surface.
    pub fn user_line(&self) -> String {
        let title = self
            .session_title
            .as_deref()
            .map(str::trim)
            .filter(|t| !t.is_empty())
            .map(|t| format!(" (title '{t}')"))
            .unwrap_or_default();
        let aborted = if self.aborted_background_compaction {
            " A background compaction that was running has been cancelled."
        } else {
            ""
        };
        format!(
            "Context cleared. The history above stays in this session{title}; the agent \
             starts fresh from here and can search it with session_search when it needs \
             something. No summariser call was made.{aborted}"
        )
    }
}

impl AgentService {
    /// Cut the agent's context at the current point without a provider call.
    ///
    /// Aborts an in-flight background summariser first, exactly as
    /// `compact_context` does: its result is applied on a later visit and
    /// would otherwise overwrite the cleared context with a summary of the
    /// old one. Then appends the marker row the loader cuts at.
    pub async fn clear_context(&self, session_id: Uuid) -> Result<ClearReceipt> {
        let aborted_background_compaction = match self.take_pending_compaction(session_id) {
            Some(pending) => {
                tracing::info!("/clear: aborting the background compaction in flight");
                pending.abort();
                true
            }
            None => false,
        };

        let session_title = SessionService::new(self.context.clone())
            .get_session(session_id)
            .await
            .map_err(AgentError::db)?
            .and_then(|s| s.title);

        let messages = MessageService::new(self.context.clone());

        // Read before the marker row exists, so the quote is exactly what
        // the user saw vanish. `clear_marker` filters out old marker and
        // summary rows and caps the quote at the last six pairs itself.
        let recent: Vec<(String, String)> = messages
            .list_messages_for_session(session_id)
            .await
            .map_err(AgentError::db)?
            .iter()
            .map(|m| (m.role.clone(), m.content.clone()))
            .collect();

        messages
            .create_message(
                session_id,
                "user".to_string(),
                clear_marker(session_title.as_deref(), &recent),
            )
            .await
            .map_err(AgentError::db)?;

        tracing::info!(
            "/clear: context cut for session {session_id} (title={:?}, aborted_background={})",
            session_title,
            aborted_background_compaction
        );
        Ok(ClearReceipt {
            session_title,
            aborted_background_compaction,
        })
    }
}