Unit 4 · Connect AI steps into a workflow

Turn a workflow into a reusable tool

You can expose a flow as a tool, trace it, wait on an owned continuation, and optimize its components. The shared optimizer tunes the flow without a separate flow-only API.

8 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