Struct Persistence 

pub struct Persistence { /* private fields */ }

Implementations

impl Persistence

pub fn new<P>(db_path: P) -> Result<Persistence, Error>

where P: AsRef<Path>,

Create or open the database at the provided path and run migrations.

pub fn with_instance_id<P>( db_path: P, instance_id: String, ) -> Result<Persistence, Error>

where P: AsRef<Path>,

Create with a specific instance_id

pub fn instance_id(&self) -> &str

Get the instance ID for this persistence instance

pub fn checkpoint(&self) -> Result<(), Error>

Checkpoint the database to ensure all WAL data is written to the main database file. Call this before shutdown to ensure clean database state.

pub fn new_default() -> Result<Persistence, Error>

Creates or opens the default database at ~/.spec-ai/agent_data.duckdb

pub fn conn(&self) -> MutexGuard<'_, Connection>

Get access to the pooled database connection. Returns a MutexGuard that provides exclusive access to the connection.

pub fn insert_message( &self, session_id: &str, role: MessageRole, content: &str, ) -> Result<i64, Error>

pub fn list_messages( &self, session_id: &str, limit: i64, ) -> Result<Vec<Message>, Error>

pub fn get_message(&self, message_id: i64) -> Result<Option<Message>, Error>

pub fn prune_messages( &self, session_id: &str, keep_latest: i64, ) -> Result<u64, Error>

Simple pruning by keeping only the most recent keep_latest messages.

pub fn insert_memory_vector( &self, session_id: &str, message_id: Option<i64>, embedding: &[f32], ) -> Result<i64, Error>

pub fn recall_top_k( &self, session_id: &str, query_embedding: &[f32], k: usize, ) -> Result<Vec<(MemoryVector, f32)>, Error>

pub fn list_sessions(&self) -> Result<Vec<String>, Error>

List known session IDs ordered by most recent activity

pub fn log_tool( &self, session_id: &str, agent_name: &str, run_id: &str, tool_name: &str, arguments: &Value, result: &Value, success: bool, error: Option<&str>, ) -> Result<i64, Error>

pub fn policy_upsert(&self, key: &str, value: &Value) -> Result<(), Error>

pub fn policy_get(&self, key: &str) -> Result<Option<PolicyEntry>, Error>

impl Persistence

pub fn insert_graph_node( &self, session_id: &str, node_type: NodeType, label: &str, properties: &Value, embedding_id: Option<i64>, ) -> Result<i64, Error>

pub fn get_graph_node(&self, node_id: i64) -> Result<Option<GraphNode>, Error>

pub fn list_graph_nodes( &self, session_id: &str, node_type: Option<NodeType>, limit: Option<i64>, ) -> Result<Vec<GraphNode>, Error>

pub fn count_graph_nodes(&self, session_id: &str) -> Result<i64, Error>

pub fn update_graph_node( &self, node_id: i64, properties: &Value, ) -> Result<(), Error>

pub fn delete_graph_node(&self, node_id: i64) -> Result<(), Error>

pub fn insert_graph_edge( &self, session_id: &str, source_id: i64, target_id: i64, edge_type: EdgeType, predicate: Option<&str>, properties: Option<&Value>, weight: f32, ) -> Result<i64, Error>

pub fn get_graph_edge(&self, edge_id: i64) -> Result<Option<GraphEdge>, Error>

pub fn list_graph_edges( &self, session_id: &str, source_id: Option<i64>, target_id: Option<i64>, ) -> Result<Vec<GraphEdge>, Error>

pub fn count_graph_edges(&self, session_id: &str) -> Result<i64, Error>

pub fn delete_graph_edge(&self, edge_id: i64) -> Result<(), Error>

pub fn find_shortest_path( &self, session_id: &str, source_id: i64, target_id: i64, max_hops: Option<usize>, ) -> Result<Option<GraphPath>, Error>

pub fn traverse_neighbors( &self, session_id: &str, node_id: i64, direction: TraversalDirection, depth: usize, ) -> Result<Vec<GraphNode>, Error>

pub fn insert_transcription( &self, session_id: &str, chunk_id: i64, text: &str, timestamp: DateTime<Utc>, ) -> Result<i64, Error>

pub fn update_transcription_embedding( &self, transcription_id: i64, embedding_id: i64, ) -> Result<(), Error>

pub fn list_transcriptions( &self, session_id: &str, limit: Option<i64>, ) -> Result<Vec<(i64, i64, String, DateTime<Utc>)>, Error>

pub fn get_full_transcription(&self, session_id: &str) -> Result<String, Error>

pub fn delete_transcriptions(&self, session_id: &str) -> Result<(), Error>

pub fn get_transcription_by_embedding( &self, embedding_id: i64, ) -> Result<Option<String>, Error>

pub fn upsert_tokenized_file( &self, session_id: &str, path: &str, file_hash: &str, raw_tokens: usize, cleaned_tokens: usize, bytes_captured: usize, truncated: bool, embedding_id: Option<i64>, ) -> Result<i64, Error>

Persist tokenization metadata for a file, replacing any existing entry for the path.

pub fn get_tokenized_file( &self, session_id: &str, path: &str, ) -> Result<Option<TokenizedFileRecord>, Error>

pub fn list_tokenized_files( &self, session_id: &str, ) -> Result<Vec<TokenizedFileRecord>, Error>

pub fn mesh_message_store( &self, message_id: &str, source_instance: &str, target_instance: Option<&str>, message_type: &str, payload: &Value, status: &str, ) -> Result<i64, Error>

Store a mesh message in the database

pub fn mesh_message_exists(&self, message_id: &str) -> Result<bool, Error>

Check if a message with this ID already exists (for duplicate detection)

pub fn mesh_message_update_status( &self, message_id: i64, status: &str, ) -> Result<(), Error>

Update message status (e.g., delivered, failed)

pub fn mesh_message_get_pending( &self, target_instance: &str, ) -> Result<Vec<MeshMessageRecord>, Error>

Get pending messages for a target instance

pub fn mesh_message_get_history( &self, instance_id: Option<&str>, limit: usize, ) -> Result<Vec<MeshMessageRecord>, Error>

Get message history for analytics

pub fn graph_changelog_append( &self, session_id: &str, instance_id: &str, entity_type: &str, entity_id: i64, operation: &str, vector_clock: &str, data: Option<&str>, ) -> Result<i64, Error>

Append an entry to the graph changelog

pub fn graph_changelog_get_since( &self, session_id: &str, since_timestamp: &str, ) -> Result<Vec<ChangelogEntry>, Error>

Get changelog entries since a given timestamp for a session

pub fn graph_changelog_prune(&self, days_to_keep: i64) -> Result<usize, Error>

Prune old changelog entries (keep last N days)

pub fn graph_sync_state_get( &self, instance_id: &str, session_id: &str, graph_name: &str, ) -> Result<Option<String>, Error>

Get the vector clock for an instance/session/graph combination

pub fn graph_sync_state_update( &self, instance_id: &str, session_id: &str, graph_name: &str, vector_clock: &str, ) -> Result<(), Error>

Update the vector clock for an instance/session/graph combination

pub fn graph_set_sync_enabled( &self, session_id: &str, graph_name: &str, enabled: bool, ) -> Result<(), Error>

Enable or disable sync for a graph

pub fn graph_get_sync_enabled( &self, session_id: &str, graph_name: &str, ) -> Result<bool, Error>

Check if sync is enabled for a graph

pub fn graph_list(&self, session_id: &str) -> Result<Vec<String>, Error>

List all graphs for a session

pub fn graph_get_node_with_sync( &self, node_id: i64, ) -> Result<Option<SyncedNodeRecord>, Error>

Get a node with its sync metadata

pub fn graph_list_nodes_with_sync( &self, session_id: &str, sync_enabled_only: bool, include_deleted: bool, ) -> Result<Vec<SyncedNodeRecord>, Error>

Get all synced nodes for a session with optional filters

pub fn graph_get_edge_with_sync( &self, edge_id: i64, ) -> Result<Option<SyncedEdgeRecord>, Error>

Get an edge with its sync metadata

pub fn graph_list_edges_with_sync( &self, session_id: &str, sync_enabled_only: bool, include_deleted: bool, ) -> Result<Vec<SyncedEdgeRecord>, Error>

Get all synced edges for a session with optional filters

pub fn graph_update_node_sync_metadata( &self, node_id: i64, vector_clock: &str, last_modified_by: &str, sync_enabled: bool, ) -> Result<(), Error>

Update a node’s sync metadata

pub fn graph_update_edge_sync_metadata( &self, edge_id: i64, vector_clock: &str, last_modified_by: &str, sync_enabled: bool, ) -> Result<(), Error>

Update an edge’s sync metadata

pub fn graph_mark_node_deleted( &self, node_id: i64, vector_clock: &str, deleted_by: &str, ) -> Result<(), Error>

Mark a node as deleted (tombstone pattern)

pub fn graph_mark_edge_deleted( &self, edge_id: i64, vector_clock: &str, deleted_by: &str, ) -> Result<(), Error>

Mark an edge as deleted (tombstone pattern)

Trait Implementations

impl Clone for Persistence

fn clone(&self) -> Persistence

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.