wesichain-graph 0.3.0

Rust-native LLM agents & chains with resumable ReAct workflows
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use serde::{Deserialize, Serialize};
use wesichain_graph::{GraphBuilder, GraphError, GraphState, StateSchema};

#[derive(Clone, Default, Debug, Serialize, Deserialize, PartialEq)]
struct DemoState {
    count: i32,
}

impl StateSchema for DemoState {
    type Update = Self;
    fn apply(_: &Self, update: Self) -> Self {
        update
    }
}

#[tokio::test]
async fn graph_returns_missing_node_error() {
    let graph = GraphBuilder::new().set_entry("missing").build();
    let state = GraphState::new(DemoState { count: 0 });
    let err = graph.invoke_graph(state).await.unwrap_err();
    assert!(matches!(err, GraphError::MissingNode { .. }));
}