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

mermaid := `flowchart TD
  %%ax classify: request:string -> route:class "support, sales"
  classify{route}`
program := axllm.NewFlow(mermaid)
fmt.Println(program.String())

Run itIn your own project

go get github.com/ax-llm/ax/packages/go

package main

import (
    "context"
    "fmt"
    "os"

    axllm "github.com/ax-llm/ax/packages/go"
)

func main() {
    client := axllm.NewAI("openai", map[string]axllm.Value{"apiKey": os.Getenv("OPENAI_API_KEY")})
    classify := axllm.NewAx("review:string -> sentiment:class \"positive, negative, neutral\"", nil)
    result, err := classify.Forward(context.Background(), client, map[string]axllm.Value{
        "review": "Useful and boring in the best way."
    }, nil)
    if err != nil {
        panic(err)
    }

    fmt.Println("sentiment:", result.(map[string]axllm.Value)["sentiment"])
}

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 -- go src/examples/go/flows/branch_flow.go

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