05cb353f7f
- 22 AI providers (OpenRouter, Anthropic, OpenAI, Mistral, etc.) - 7 channels (CLI, Telegram, Discord, Slack, iMessage, Matrix, Webhook) - 5-step onboarding wizard with Project Context personalization - OpenClaw-aligned system prompt (SOUL.md, IDENTITY.md, USER.md, AGENTS.md, etc.) - SQLite memory backend with auto-save - Skills system with on-demand loading - Security: autonomy levels, command allowlists, cost limits - 532 tests passing, 0 clippy warnings
68 lines
2 KiB
Rust
68 lines
2 KiB
Rust
use async_trait::async_trait;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
/// A single memory entry
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
pub struct MemoryEntry {
|
|
pub id: String,
|
|
pub key: String,
|
|
pub content: String,
|
|
pub category: MemoryCategory,
|
|
pub timestamp: String,
|
|
pub session_id: Option<String>,
|
|
pub score: Option<f64>,
|
|
}
|
|
|
|
/// Memory categories for organization
|
|
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
|
|
#[serde(rename_all = "snake_case")]
|
|
pub enum MemoryCategory {
|
|
/// Long-term facts, preferences, decisions
|
|
Core,
|
|
/// Daily session logs
|
|
Daily,
|
|
/// Conversation context
|
|
Conversation,
|
|
/// User-defined custom category
|
|
Custom(String),
|
|
}
|
|
|
|
impl std::fmt::Display for MemoryCategory {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
Self::Core => write!(f, "core"),
|
|
Self::Daily => write!(f, "daily"),
|
|
Self::Conversation => write!(f, "conversation"),
|
|
Self::Custom(name) => write!(f, "{name}"),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Core memory trait — implement for any persistence backend
|
|
#[async_trait]
|
|
pub trait Memory: Send + Sync {
|
|
/// Backend name
|
|
fn name(&self) -> &str;
|
|
|
|
/// Store a memory entry
|
|
async fn store(&self, key: &str, content: &str, category: MemoryCategory)
|
|
-> anyhow::Result<()>;
|
|
|
|
/// Recall memories matching a query (keyword search)
|
|
async fn recall(&self, query: &str, limit: usize) -> anyhow::Result<Vec<MemoryEntry>>;
|
|
|
|
/// Get a specific memory by key
|
|
async fn get(&self, key: &str) -> anyhow::Result<Option<MemoryEntry>>;
|
|
|
|
/// List all memory keys, optionally filtered by category
|
|
async fn list(&self, category: Option<&MemoryCategory>) -> anyhow::Result<Vec<MemoryEntry>>;
|
|
|
|
/// Remove a memory by key
|
|
async fn forget(&self, key: &str) -> anyhow::Result<bool>;
|
|
|
|
/// Count total memories
|
|
async fn count(&self) -> anyhow::Result<usize>;
|
|
|
|
/// Health check
|
|
async fn health_check(&self) -> bool;
|
|
}
|