vv-agent 0.6.0

VectorVein agent runtime, SDK, CLI, tools, and workspace backends
Documentation 
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

use super::{prompts, MemoryManager};
use crate::types::{Message, MessageRole};

impl MemoryManager {
    pub(super) fn maybe_append_memory_warning(
        &self,
        messages: &[Message],
        message_length: u64,
    ) -> (Vec<Message>, bool) {
        if !self.config.include_memory_warning || self.autocompact_threshold() == 0 {
            return (messages.to_vec(), false);
        }
        if message_length < self.warning_threshold() {
            return (messages.to_vec(), false);
        }
        let warning_text = self.memory_warning_text();
        if messages.iter().rev().take(10).any(|message| {
            message.role == MessageRole::User && message.content.contains(&warning_text)
        }) {
            return (messages.to_vec(), false);
        }
        let mut warned = messages.to_vec();
        warned.push(Message::user(warning_text));
        (warned, true)
    }

    fn memory_warning_text(&self) -> String {
        prompts::memory_warning_text(
            &self.config.language,
            self.config.warning_threshold_percentage,
        )
    }
}