echo_state 0.1.4

State management for echo-agent framework (memory, compression, audit)
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

//! Memory system
//!
//! Layered architecture, each with distinct responsibilities:
//!
//! | Layer | Implementation | Scope |
//! |------|------|--------|
//! | Short-term context | [`compression::ContextManager`] | Within a single `execute()` call |
//! | Thread state | [`Checkpointer`] / [`FileCheckpointer`] | Cross-process recovery of the same thread |
//! | Conversation history | [`ConversationStore`] / `SqliteConversationStore` | Transcript projection, history browsing, multi-user isolation |
//! | Long-term memory | [`Store`] / [`FileStore`] / `SqliteStore` | Cross-session, cross-user sharing |
//!
//! **Note**: Trait definitions and data types now live in `echo_core::memory`.
//! They are re-exported here for backward compatibility.

pub mod checkpointer;
pub mod conversation;
pub mod embedder;
pub mod embedding_store;
pub mod snapshot;
#[cfg(feature = "sqlite")]
pub mod sqlite_conversation;
#[cfg(feature = "sqlite")]
pub mod sqlite_store;
pub mod store;

// Re-export traits and data types from echo-core (backward compatibility)
pub use echo_core::memory::checkpointer::{Checkpoint, Checkpointer, ThreadState};
pub use echo_core::memory::conversation::{
    Conversation, ConversationFilter, ConversationMeta, ConversationStore, NewConversation,
    StoredMessage,
};
pub use echo_core::memory::embedder::Embedder;
pub use echo_core::memory::store::{SearchMode, SearchQuery, Store, StoreItem};

// Re-export concrete implementations from sub-modules
pub use checkpointer::{FileCheckpointer, InMemoryCheckpointer};
pub use conversation::{project_message, project_messages};
pub use embedder::HttpEmbedder;
pub use embedding_store::EmbeddingStore;
pub use snapshot::{SnapshotManager, SnapshotPolicy, StateSnapshot};
#[cfg(feature = "sqlite")]
pub use sqlite_conversation::SqliteConversationStore;
#[cfg(feature = "sqlite")]
pub use sqlite_store::SqliteStore;
pub use store::{FileStore, InMemoryStore};

// Legacy alias
pub use Checkpointer as ThreadStore;

#[cfg(test)]
pub use test_utils::MockEmbedder;

/// Test embedder (visible only in tests)
#[cfg(test)]
mod test_utils {
    use echo_core::error::Result;
    use echo_core::memory::embedder::Embedder;
    use futures::future::BoxFuture;

    pub struct MockEmbedder {
        dimension: usize,
    }

    impl MockEmbedder {
        pub fn new(dimension: usize) -> Self {
            assert!(dimension > 0);
            Self { dimension }
        }
    }

    impl Embedder for MockEmbedder {
        fn embed<'a>(&'a self, text: &'a str) -> BoxFuture<'a, Result<Vec<f32>>> {
            Box::pin(async move {
                let mut vec = vec![0.0f32; self.dimension];
                for (i, b) in text.bytes().enumerate() {
                    vec[i % self.dimension] += b as f32;
                }
                let norm: f32 = vec.iter().map(|x| x * x).sum::<f32>().sqrt();
                if norm > 0.0 {
                    for v in &mut vec {
                        *v /= norm;
                    }
                }
                Ok(vec)
            })
        }
    }
}