blazen 0.1.143

Blazen - A Rust-native AI workflow engine with event-driven orchestration
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
68
69
70
71

//! Basic workflow example: a simple 3-step sequential pipeline.
//!
//! Demonstrates:
//! - Defining custom events with `#[derive(Event)]`
//! - Writing step functions with `#[step]`
//! - Building and running a workflow with `WorkflowBuilder`
//! - Awaiting the final result
//!
//! Run with: `cargo run -p blazen --example basic_workflow`

// Steps must be async for the #[step] macro even when they don't await.
#![allow(clippy::unused_async)]

use blazen::prelude::*;

// ---------------------------------------------------------------------------
// Custom events
// ---------------------------------------------------------------------------

/// Intermediate event carrying a parsed name.
#[derive(Debug, Clone, Serialize, Deserialize, Event)]
struct GreetEvent {
    name: String,
}

// ---------------------------------------------------------------------------
// Steps
// ---------------------------------------------------------------------------

/// Parse the raw JSON input into a typed `GreetEvent`.
#[step]
async fn parse_input(event: StartEvent, _ctx: Context) -> Result<GreetEvent, WorkflowError> {
    let name = event.data["name"].as_str().unwrap_or("World").to_string();
    println!("[parse_input] Extracted name: {name}");
    Ok(GreetEvent { name })
}

/// Produce the final greeting and terminate the workflow.
#[step]
async fn greet(event: GreetEvent, _ctx: Context) -> Result<StopEvent, WorkflowError> {
    let greeting = format!("Hello, {}!", event.name);
    println!("[greet] {greeting}");
    Ok(StopEvent {
        result: serde_json::json!({ "greeting": greeting }),
    })
}

// ---------------------------------------------------------------------------
// Main
// ---------------------------------------------------------------------------

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    // Build the workflow from step registrations.
    let workflow = WorkflowBuilder::new("greeter")
        .step(parse_input_registration())
        .step(greet_registration())
        .build()?;

    // Run the workflow with a JSON payload.
    let handler = workflow.run(serde_json::json!({ "name": "Zach" })).await?;

    // Await the final result.
    let result = handler.result().await?.event;

    if let Some(stop) = result.downcast_ref::<StopEvent>() {
        println!("\nFinal result: {}", stop.result);
    }

    Ok(())
}