oris-runtime 0.15.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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use std::sync::Arc;

use async_trait::async_trait;
use oris_execution_runtime::{
    ExecutionCheckpointView, ExecutionGraphBridge, ExecutionGraphBridgeError, ExecutionInvokeView,
    ExecutionStateView,
};
use serde_json::Value;

use crate::graph::{Command, CompiledGraph, MessagesState, RunnableConfig, StateOrCommand};
use crate::schemas::messages::Message;

pub(crate) struct CompiledGraphExecutionBridge {
    compiled: Arc<CompiledGraph<MessagesState>>,
}

impl CompiledGraphExecutionBridge {
    pub(crate) fn new(compiled: Arc<CompiledGraph<MessagesState>>) -> Self {
        Self { compiled }
    }
}

#[async_trait]
impl ExecutionGraphBridge for CompiledGraphExecutionBridge {
    async fn run(
        &self,
        thread_id: &str,
        input: &str,
    ) -> Result<ExecutionInvokeView, ExecutionGraphBridgeError> {
        let initial = MessagesState::with_messages(vec![Message::new_human_message(input)]);
        let config = RunnableConfig::with_thread_id(thread_id);
        let result = self
            .compiled
            .invoke_with_config_interrupt(StateOrCommand::State(initial), &config)
            .await
            .map_err(|e| ExecutionGraphBridgeError::internal(e.to_string()))?;
        Ok(ExecutionInvokeView {
            interrupts: result
                .interrupt
                .unwrap_or_default()
                .into_iter()
                .map(|interrupt| interrupt.value)
                .collect(),
        })
    }

    async fn resume(
        &self,
        thread_id: &str,
        checkpoint_id: Option<&str>,
        value: Value,
    ) -> Result<ExecutionInvokeView, ExecutionGraphBridgeError> {
        let config = checkpoint_config(thread_id, checkpoint_id);
        let result = self
            .compiled
            .invoke_with_config_interrupt(StateOrCommand::Command(Command::resume(value)), &config)
            .await
            .map_err(|e| ExecutionGraphBridgeError::internal(e.to_string()))?;
        Ok(ExecutionInvokeView {
            interrupts: result
                .interrupt
                .unwrap_or_default()
                .into_iter()
                .map(|interrupt| interrupt.value)
                .collect(),
        })
    }

    async fn replay(
        &self,
        thread_id: &str,
        checkpoint_id: Option<&str>,
    ) -> Result<(), ExecutionGraphBridgeError> {
        let config = checkpoint_config(thread_id, checkpoint_id);
        self.compiled
            .invoke_with_config(None, &config)
            .await
            .map_err(|e| ExecutionGraphBridgeError::internal(e.to_string()))?;
        Ok(())
    }

    async fn snapshot(
        &self,
        thread_id: &str,
        checkpoint_id: Option<&str>,
    ) -> Result<ExecutionStateView, ExecutionGraphBridgeError> {
        let config = checkpoint_config(thread_id, checkpoint_id);
        let snapshot = self
            .compiled
            .get_state(&config)
            .await
            .map_err(map_snapshot_error)?;
        let values = serde_json::to_value(&snapshot.values).map_err(|e| {
            ExecutionGraphBridgeError::internal(format!("serialize state failed: {}", e))
        })?;
        Ok(ExecutionStateView {
            checkpoint_id: snapshot.checkpoint_id().cloned(),
            created_at: snapshot.created_at,
            values,
        })
    }

    async fn history(
        &self,
        thread_id: &str,
    ) -> Result<Vec<ExecutionCheckpointView>, ExecutionGraphBridgeError> {
        let config = RunnableConfig::with_thread_id(thread_id);
        let history = self
            .compiled
            .get_state_history(&config)
            .await
            .map_err(|e| ExecutionGraphBridgeError::internal(e.to_string()))?;
        Ok(history
            .into_iter()
            .map(|snapshot| ExecutionCheckpointView {
                checkpoint_id: snapshot.checkpoint_id().cloned(),
                created_at: snapshot.created_at,
            })
            .collect())
    }
}

fn checkpoint_config(thread_id: &str, checkpoint_id: Option<&str>) -> RunnableConfig {
    if let Some(checkpoint_id) = checkpoint_id {
        RunnableConfig::with_checkpoint(thread_id, checkpoint_id)
    } else {
        RunnableConfig::with_thread_id(thread_id)
    }
}

fn map_snapshot_error(error: crate::graph::GraphError) -> ExecutionGraphBridgeError {
    let message = error.to_string();
    if message.contains("No state found") || message.contains("Checkpoint not found") {
        ExecutionGraphBridgeError::not_found(message)
    } else {
        ExecutionGraphBridgeError::internal(message)
    }
}