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

step := axllm.NewAx("request:string -> route:class \"support, sales, engineering\"", nil)
program := axllm.NewFlow(map[string]axllm.Value{"id": "supportRouter"}).
    Execute("classify", step, nil).
    Returns(map[string]axllm.Value{"route": "classify"})

Run itIn your own project

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

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

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)

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