ai-agents-memory 1.0.0-rc.14

Memory implementations for AI Agents framework
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

//! Memory implementations for AI Agents framework

mod compacting;
mod context;
mod events;
mod in_memory;
mod summarizer;
mod token_budget;

use async_trait::async_trait;

pub use ai_agents_core::MemorySnapshot;
pub use compacting::{CompactingMemory, CompactingMemoryConfig, CompressionEvent};
pub use context::{CompressResult, ConversationContext, estimate_message_tokens, estimate_tokens};
pub use events::{
    EvictionReason, FactExtractedEvent, MemoryBudgetEvent, MemoryCompressEvent, MemoryEvictEvent,
};
pub use in_memory::InMemoryStore;
pub use summarizer::{LLMSummarizer, NoopSummarizer, Summarizer};
pub use token_budget::{MemoryBudgetState, MemoryTokenBudget, OverflowStrategy, TokenAllocation};

/// Extended memory trait that preserves the original interface.
#[async_trait]
pub trait Memory: ai_agents_core::Memory {
    async fn get_context(&self) -> ai_agents_core::Result<ConversationContext> {
        let messages = self.get_messages(None).await?;
        Ok(ConversationContext::with_messages(messages))
    }

    async fn compress(
        &self,
        _summarizer: Option<&dyn Summarizer>,
    ) -> ai_agents_core::Result<CompressResult> {
        Ok(CompressResult::NotNeeded)
    }

    fn needs_compression(&self) -> bool {
        false
    }
}