sein 0.1.1

Graph experiment
Documentation 
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

use std::time::Duration;
use anyhow::Result;
use async_trait::async_trait;
use sein::{AgentEngine, NodeTrait};

// Define application-specific node types
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
enum AppNode {
    Thinking,
    GreetTool,
    Exit,
}

struct CounterContext {
    current: u32,
    max: u32,
}

struct ThinkingNode;

#[async_trait]
impl NodeTrait<CounterContext> for ThinkingNode {
    async fn enter(&self, ctx: &mut CounterContext) -> Result<()> {
        tokio::time::sleep(Duration::from_millis(500)).await;
        println!("Thinking... current count: {}/{}", ctx.current, ctx.max);
        Ok(())
    }
}

struct GreetToolNode;

#[async_trait]
impl NodeTrait<CounterContext> for GreetToolNode {
    async fn enter(&self, ctx: &mut CounterContext) -> Result<()> {
        tokio::time::sleep(Duration::from_secs(1)).await;
        ctx.current += 1;
        println!("Hello! This is greeting number {}", ctx.current);
        Ok(())
    }
}

struct ExitNode;

#[async_trait]
impl NodeTrait<CounterContext> for ExitNode {
    async fn enter(&self, _ctx: &mut CounterContext) -> Result<()> {
        println!("Done! Exiting...");
        Ok(())
    }
}

#[tokio::main]
async fn main() -> Result<()> {
    let mut engine = AgentEngine::builder(CounterContext { current: 0, max: 3 })
        .with_node(AppNode::Thinking, ThinkingNode)
        .with_node(AppNode::GreetTool, GreetToolNode)
        .with_node(AppNode::Exit, ExitNode)
        .with_transition(AppNode::Thinking, AppNode::GreetTool, |ctx| ctx.current < ctx.max)?
        .with_transition(AppNode::Thinking, AppNode::Exit, |ctx| ctx.current >= ctx.max)?
        .with_transition(AppNode::GreetTool, AppNode::Thinking, |_| true)?
        .with_start_node(AppNode::Thinking)
        .build()?;

    engine.run().await?;

    Ok(())
}