Cookbook: Multi-Agent Orchestration

import { AllternitClient } from '@allternit/sdk';
import { ToolRegistry } from '@allternit/sdk/ai-runtime/tools';
import { NativeToolBelt } from '@allternit/sdk/ai-runtime/tools/search';

const client = new AllternitClient({
  apiKey: process.env.ALLTERNIT_API_KEY!,
});

const registry = new ToolRegistry();
new NativeToolBelt(registry, {
  provider: 'tavily',
  apiKeys: { tavily: process.env.TAVILY_API_KEY! },
  workspaceRoot: process.cwd(),
});

const researcher = await client.agents.create({
  name: 'researcher',
  runtime: 'gizzi-pro',
  skills: ['web_search', 'web_fetch'],
  systemPrompt: `You are a research agent. Given a topic, search the web for
    relevant information and produce a structured summary with sources.`,
});

const writer = await client.agents.create({
  name: 'writer',
  runtime: 'gizzi-pro',
  skills: ['str_replace_editor'],
  systemPrompt: `You are a writing agent. Given research notes, write a clear,
    well-structured document. Save it to output.md.`,
});

const reviewer = await client.agents.create({
  name: 'reviewer',
  runtime: 'gizzi-pro',
  skills: ['str_replace_editor'],
  systemPrompt: `You are a review agent. Read a document and provide specific,
    actionable feedback. If the document is ready, say APPROVED.`,
});

// Step 1: Research
const research = await client.agents.run(researcher.id, {
  input: 'Research the benefits of multi-agent systems in software engineering',
});
const researchNotes = research.output;

// Step 2: Write
const draft = await client.agents.run(writer.id, {
  input: `Write a document based on these research notes:\n\n${researchNotes}`,
});

// Step 3: Review (loop until approved)
let feedback = '';
let iterations = 0;
const MAX_ITERATIONS = 3;

while (iterations < MAX_ITERATIONS) {
  const review = await client.agents.run(reviewer.id, {
    input: `Review this document and provide feedback:\n\n${draft.output}\n\nPrevious feedback: ${feedback}`,
  });

  if (review.output.includes('APPROVED')) {
    console.log('Document approved!');
    break;
  }

  feedback = review.output;

  // Revise
  const revision = await client.agents.run(writer.id, {
    input: `Revise the document based on this feedback:\n\nFeedback: ${feedback}\n\nCurrent draft: ${draft.output}`,
  });
  draft.output = revision.output;
  iterations++;
}