oris-runtime 0.61.0

An agentic workflow runtime and programmable AI execution system in Rust: stateful graphs, agents, tools, and multi-step execution.
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

use std::sync::Arc;

use async_trait::async_trait;

use crate::graph::state::State;

use super::{error::PersistenceError, snapshot::StateSnapshot};

/// Trait for checkpoint savers
///
/// Checkpointers save and retrieve state snapshots for graph execution.
/// This allows for persistence, replay, and fault tolerance.
#[async_trait]
pub trait Checkpointer<S: State>: Send + Sync {
    /// Save a checkpoint
    ///
    /// Returns the checkpoint_id of the saved checkpoint.
    async fn put(
        &self,
        thread_id: &str,
        checkpoint: &StateSnapshot<S>,
    ) -> Result<String, PersistenceError>;

    /// Get a checkpoint
    ///
    /// If checkpoint_id is None, returns the latest checkpoint for the thread.
    async fn get(
        &self,
        thread_id: &str,
        checkpoint_id: Option<&str>,
    ) -> Result<Option<StateSnapshot<S>>, PersistenceError>;

    /// List checkpoints for a thread
    ///
    /// Returns checkpoints in chronological order (oldest first).
    /// If limit is specified, returns only the most recent N checkpoints.
    async fn list(
        &self,
        thread_id: &str,
        limit: Option<usize>,
    ) -> Result<Vec<StateSnapshot<S>>, PersistenceError>;
}

/// Type alias for a boxed checkpointer
pub type CheckpointerBox<S> = Arc<dyn Checkpointer<S>>;