potato-agent 0.26.0

Potato brands
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

pub mod in_memory;
pub mod windowed;

pub use in_memory::InMemoryMemory;
pub use windowed::WindowedMemory;

use potato_type::prompt::MessageNum;
use serde::{Deserialize, Serialize};
use std::fmt::Debug;

/// A single conversation turn (user + assistant message pair).
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MemoryTurn {
    pub user: MessageNum,
    pub assistant: MessageNum,
}

/// Trait for agent conversation memory.
pub trait Memory: Send + Sync + Debug {
    fn push_turn(&mut self, turn: MemoryTurn);
    /// Returns messages in chronological order (oldest first).
    fn messages(&self) -> Vec<MessageNum>;
    fn clear(&mut self);
    fn len(&self) -> usize;
    fn is_empty(&self) -> bool {
        self.len() == 0
    }
    /// Enables downcasting to concrete types (e.g., `PersistentMemory`).
    /// Override in concrete types to support `downcast_mut`.
    fn as_any_mut(&mut self) -> Option<&mut dyn std::any::Any> {
        None
    }
}