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

use serde_json::Value;

use crate::graph::state::{State, StateUpdate};

use super::metadata::{DebugInfo, MessageChunk};

/// Stream chunk - represents a single item in a stream
///
/// Different stream modes produce different types of chunks.
#[derive(Clone, Debug)]
pub enum StreamChunk<S: State> {
    /// Full state value (values mode)
    Values { state: S },
    /// State update (updates mode)
    Updates { node: String, update: StateUpdate },
    /// LLM message chunk (messages mode)
    Messages { chunk: MessageChunk },
    /// Custom data (custom mode)
    Custom { node: String, data: Value },
    /// Debug information (debug mode)
    Debug { info: DebugInfo },
}

impl<S: State> StreamChunk<S> {
    /// Get the stream mode for this chunk
    pub fn mode(&self) -> super::mode::StreamMode {
        use super::mode::StreamMode;
        match self {
            Self::Values { .. } => StreamMode::Values,
            Self::Updates { .. } => StreamMode::Updates,
            Self::Messages { .. } => StreamMode::Messages,
            Self::Custom { .. } => StreamMode::Custom,
            Self::Debug { .. } => StreamMode::Debug,
        }
    }
}