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. go academy academy/topics/flow-composition website/content-src/academy/course.mjs academy Pass useful state from step to step
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 := 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

Source-backed follow-up