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

Worked example

See the idea in context

.execute('research', s => ({ topic: s.topic })).map(s => ({ ...s, wordLimit: 300 })).execute('write', s => ({ research: s.researchResult, wordLimit: s.wordLimit }));

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