Struct AgentDB 

pub struct AgentDB { /* private fields */ }

Main AgentDB interface providing vector storage, reflexion, causal, and skill management

Implementations

impl AgentDB

pub async fn new(config: AgentDBConfig) -> Result<Self, AgentDBError>

Creates a new AgentDB instance with the given configuration

pub async fn vector_store( &self, embedding: Embedding, metadata: Value, ) -> Result<VectorId, AgentDBError>

Stores a vector embedding with associated metadata

pub async fn vector_search( &self, query: &Embedding, k: usize, ) -> Result<Vec<VectorResult>, AgentDBError>

Searches for the k most similar vectors using HNSW index

pub async fn vector_get( &self, id: &str, ) -> Result<(Embedding, Value), AgentDBError>

Retrieves a specific vector by ID

pub async fn vector_delete(&self, id: &str) -> Result<(), AgentDBError>

Deletes a vector by ID

pub async fn reflexion_store( &self, episode: ReflexionEpisode, ) -> Result<ReflexionId, AgentDBError>

Stores a reflexion episode for learning from experience

pub async fn reflexion_retrieve( &self, task: &str, limit: usize, ) -> Result<Vec<ReflexionEpisode>, AgentDBError>

Retrieves similar reflexion episodes for a given task

pub async fn reflexion_by_session( &self, session_id: &str, ) -> Result<Vec<ReflexionEpisode>, AgentDBError>

Retrieves reflexion episodes by session ID

pub async fn reflexion_analyze( &self, task_prefix: &str, ) -> Result<ReflexionStats, AgentDBError>

Analyzes reflexion episodes to calculate success metrics

pub async fn causal_add_edge( &self, edge: CausalEdge, ) -> Result<(), AgentDBError>

Adds or updates a causal edge

pub async fn causal_query_effects( &self, cause: &str, ) -> Result<Vec<CausalEdge>, AgentDBError>

Queries effects caused by a specific cause

pub async fn causal_query_causes( &self, effect: &str, ) -> Result<Vec<CausalEdge>, AgentDBError>

Queries causes that lead to a specific effect

pub async fn causal_find_path( &self, start: &str, end: &str, max_depth: usize, ) -> Result<Vec<Vec<String>>, AgentDBError>

Finds causal paths between a cause and effect

pub async fn skill_create(&self, skill: Skill) -> Result<SkillId, AgentDBError>

Creates a new skill with embedding

pub async fn skill_search( &self, query: &str, limit: usize, ) -> Result<Vec<Skill>, AgentDBError>

Searches for skills using semantic similarity

pub async fn skill_search_by_embedding( &self, query_embedding: &Embedding, limit: usize, ) -> Result<Vec<(Skill, f64)>, AgentDBError>

Searches for skills using embedding similarity

pub async fn skill_update_stats( &self, skill_id: &str, success: bool, ) -> Result<(), AgentDBError>

Updates skill usage statistics

pub async fn skill_get(&self, skill_id: &str) -> Result<Skill, AgentDBError>

Gets a skill by ID

pub async fn stats(&self) -> AgentDBStats

Returns statistics about the database

pub async fn clear(&self) -> Result<(), AgentDBError>

Clears all data from the database