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

step = ax('request:string -> route:class "support, sales, engineering"')
program = (
    flow({"id": "supportRouter"})
    .execute("classify", step)
    .returns({"route": "classify"})
)

Run itIn your own project

pip install axllm

from axllm import ai, ax

llm = ai('openai', api_key=os.environ['OPENAI_API_KEY'])
classify = ax('review:string -> sentiment:class "positive, negative, neutral"')

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 -- python src/examples/python/flows/branch-flow.py

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