rexis-rag 0.1.0

Rexis RAG - High-performance Retrieval-Augmented Generation framework with memory-first agents, vector search, and LLM integration
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
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

//! Conversation memory management (legacy in-memory implementation)

use rexis_llm::{ChatMessage, MessageRole};

/// Manages conversation history for the agent
#[derive(Debug, Clone)]
pub struct ConversationMemory {
    messages: Vec<ChatMessage>,
    max_length: usize,
}

impl ConversationMemory {
    /// Create a new conversation memory
    pub fn new() -> Self {
        Self {
            messages: Vec::new(),
            max_length: 50,
        }
    }

    /// Create with custom max length
    pub fn with_max_length(max_length: usize) -> Self {
        Self {
            messages: Vec::new(),
            max_length,
        }
    }

    /// Add a message to history
    pub fn add_message(&mut self, message: ChatMessage) {
        self.messages.push(message);

        // Trim if exceeds max length (keep system message)
        if self.messages.len() > self.max_length {
            let system_msg = self.messages.first().cloned();
            self.messages
                .drain(1..self.messages.len() - self.max_length + 1);
            if let Some(sys_msg) = system_msg {
                if matches!(sys_msg.role, MessageRole::System) {
                    // System message was removed, restore it
                    if !matches!(self.messages.first().unwrap().role, MessageRole::System) {
                        self.messages.insert(0, sys_msg);
                    }
                }
            }
        }
    }

    /// Get all messages
    pub fn get_messages(&self) -> &[ChatMessage] {
        &self.messages
    }

    /// Get messages as owned Vec
    pub fn to_messages(&self) -> Vec<ChatMessage> {
        self.messages.clone()
    }

    /// Clear all messages except system
    pub fn clear(&mut self) {
        let system_msg = self
            .messages
            .iter()
            .find(|m| matches!(m.role, MessageRole::System))
            .cloned();

        self.messages.clear();

        if let Some(msg) = system_msg {
            self.messages.push(msg);
        }
    }

    /// Get number of messages
    pub fn len(&self) -> usize {
        self.messages.len()
    }

    /// Check if empty
    pub fn is_empty(&self) -> bool {
        self.messages.is_empty()
    }
}

impl Default for ConversationMemory {
    fn default() -> Self {
        Self::new()
    }
}