Unit 4 · Connect AI steps into a workflow

Branch, loop, and run work in parallel

You keep important branches, loops, and parallel work under application control. The model handles typed steps instead of improvising the orchestration plan.

9 focused minutesNot started

Worked example

See the idea in context

.branch(s => s.score > 0.8, highConfidenceFlow, reviewFlow).derive('items', s => s.documents, summarizeDocument);

Run itIn your own project

npm install @ax-llm/ax

import { ai, ax } from '@ax-llm/ax';

const llm = ai({ name: 'openai', apiKey: process.env.OPENAI_APIKEY! });
const classify = ax('review:string -> sentiment:class "positive, negative, neutral"');

const result = await classify.forward(llm, {
  review: 'Useful and boring in the best way.',
});

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 -- typescript src/examples/typescript/flows/branch-flow.ts

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