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

AxGen step = Ax.ax("request:string -> route:class \"support, sales, engineering\"");
AxFlow program = Ax.flow(Map.of("id", "supportRouter"))
    .execute("classify", step)
    .returns(Map.of("route", "classify"));

Run itIn your own project

// Gradle (build.gradle):
implementation 'dev.axllm:ax:22.0.4'
// Maven (pom.xml):
<dependency>
  <groupId>dev.axllm</groupId>
  <artifactId>ax</artifactId>
  <version>22.0.4</version>
</dependency>

import dev.axllm.ax.Ax;

var llm = Ax.ai("openai", Map.of("apiKey", System.getenv("OPENAI_API_KEY")));
var classify = Ax.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 -- java src/examples/java/flows/BranchFlowExample.java

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