feat(memory): implement FactStore SQLite persistence

Add `facts` table to schema with columns for id, agent_id, content,
category, confidence, source_session, and created_at. Implement
store_facts() and get_top_facts() on MemoryStore using upsert-by-id
and confidence-desc ordering. Facts extracted from conversations are
now durable across sessions.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
iven
2026-04-02 01:26:57 +08:00
parent 1bf0d3a73d
commit 08268b32b8
2 changed files with 88 additions and 0 deletions

View File

@@ -71,4 +71,19 @@ CREATE INDEX IF NOT EXISTS idx_kv_agent ON kv_store(agent_id);
CREATE INDEX IF NOT EXISTS idx_hand_runs_hand ON hand_runs(hand_name);
CREATE INDEX IF NOT EXISTS idx_hand_runs_status ON hand_runs(status);
CREATE INDEX IF NOT EXISTS idx_hand_runs_created ON hand_runs(created_at);
-- Structured facts table (extracted from conversations)
CREATE TABLE IF NOT EXISTS facts (
id TEXT PRIMARY KEY,
agent_id TEXT NOT NULL,
content TEXT NOT NULL,
category TEXT NOT NULL,
confidence REAL NOT NULL,
source_session TEXT,
created_at INTEGER NOT NULL,
FOREIGN KEY (agent_id) REFERENCES agents(id) ON DELETE CASCADE
);
CREATE INDEX IF NOT EXISTS idx_facts_agent ON facts(agent_id);
CREATE INDEX IF NOT EXISTS idx_facts_category ON facts(agent_id, category);
CREATE INDEX IF NOT EXISTS idx_facts_confidence ON facts(agent_id, confidence DESC);
"#;

View File

@@ -482,6 +482,76 @@ impl MemoryStore {
Ok(count as u32)
}
// === Fact CRUD ===
/// Store extracted facts for an agent (upsert by id).
pub async fn store_facts(&self, agent_id: &str, facts: &[crate::fact::Fact]) -> Result<()> {
for fact in facts {
let category_str = serde_json::to_string(&fact.category)
.map_err(|e| ZclawError::StorageError(e.to_string()))?;
// Trim the JSON quotes from serialized enum variant
let category_clean = category_str.trim_matches('"');
sqlx::query(
r#"
INSERT INTO facts (id, agent_id, content, category, confidence, source_session, created_at)
VALUES (?, ?, ?, ?, ?, ?, ?)
ON CONFLICT(id) DO UPDATE SET
content = excluded.content,
category = excluded.category,
confidence = excluded.confidence,
source_session = excluded.source_session
"#,
)
.bind(&fact.id)
.bind(agent_id)
.bind(&fact.content)
.bind(category_clean)
.bind(fact.confidence)
.bind(&fact.source)
.bind(fact.created_at as i64)
.execute(&self.pool)
.await
.map_err(|e| ZclawError::StorageError(e.to_string()))?;
}
Ok(())
}
/// Get top facts for an agent, ordered by confidence descending.
pub async fn get_top_facts(&self, agent_id: &str, limit: usize) -> Result<Vec<crate::fact::Fact>> {
let rows = sqlx::query_as::<_, (String, String, String, f64, Option<String>, i64)>(
r#"
SELECT id, content, category, confidence, source_session, created_at
FROM facts
WHERE agent_id = ?
ORDER BY confidence DESC
LIMIT ?
"#,
)
.bind(agent_id)
.bind(limit as i64)
.fetch_all(&self.pool)
.await
.map_err(|e| ZclawError::StorageError(e.to_string()))?;
let mut facts = Vec::with_capacity(rows.len());
for (id, content, category_str, confidence, source, created_at) in rows {
let category: crate::fact::FactCategory = serde_json::from_value(
serde_json::Value::String(category_str)
).map_err(|e| ZclawError::StorageError(format!("Invalid category: {}", e)))?;
facts.push(crate::fact::Fact {
id,
content,
category,
confidence,
created_at: created_at as u64,
source,
});
}
Ok(facts)
}
fn row_to_hand_run(
row: (String, String, String, String, String, Option<String>, Option<String>, Option<i64>, String, Option<String>, Option<String>),
) -> Result<HandRun> {
@@ -527,10 +597,13 @@ mod tests {
description: None,
model: ModelConfig::default(),
system_prompt: None,
soul: None,
capabilities: vec![],
tools: vec![],
max_tokens: None,
temperature: None,
workspace: None,
compaction_threshold: None,
enabled: true,
}
}