Unit 4 · Connect AI steps into a workflow

Pass useful state from step to step

You make dependencies visible by passing state between sequential nodes. Use map() for ordinary transformations that do not need another model call.

6 focused minutesNot started

Unit example (nearest native match)

See the idea in context

let mermaid = r#"flowchart TD
  %%ax classify: request:string -> route:class "support, sales"
  classify{route}"#;
let program = flow(mermaid);
println!("{}", program);

Run itIn your own project

cargo add axllm

use axllm::{ai, ax, AxResult};
use serde_json::json;

fn main() -> AxResult<()> {
    let mut llm = ai("openai", json!({"apiKey": std::env::var("OPENAI_API_KEY")?}))?;
    let mut classify = ax("review:string -> sentiment:class \"positive, negative, neutral\"")?;
    let result = classify.forward(&mut llm, json!({
        "review": "Useful and boring in the best way."
    }))?;

    println!("sentiment: {}", result["sentiment"].as_str().unwrap_or_default());
    Ok(())
}

Set OPENAI_APIKEY in your environment before running provider-backed code.

In the ax repo

From a clone of the ax repo:

npm run example -- rust src/examples/rust/flows/branch_flow.rs

Active practice

Show that you can use it

Answer 2 in a row to learn this · attempt 1

Keep exploring

See the idea in context

Show that you can use it

Source-backed follow-up