Workflow Config

import { tool } from "ai";
import { openai } from "@ai-sdk/openai";
import { z } from "zod";
import {
  createToolRegistry,
  defineWorkflow,
  model,
  skill,
} from "@little-stack/workflow";

// ── Model slots ─────────────────────────────────────────────────────────────
// Slot id auto-derived from model name. Description auto-filled from the
// bundled registry if the model is known; supply your own to override.
const fastClassifier = model(openai("gpt-4o-mini"), {
  description: "Fast classifier for short text outputs.",
});

const longContextSummarizer = model(openai("gpt-4o-mini"), {
  description: "Long-document summarizer.",
});
// → slot ids: "gpt-4o-mini" (first), "gpt-4o-mini-tiger" (second, deterministic animal suffix)

// ── Workflow definition ──────────────────────────────────────────────────────
const candidateReview = defineWorkflow({
  // Identity
  id: "candidate.review",
  description: "Review one candidate against a job description.",

  // I/O schemas
  input: z.object({
    jobDescription: z.string(),
    candidate: z.object({
      id: z.string(),
      name: z.string(),
      resume: z.string(),
    }),
  }),
  output: z.object({
    passed: z.boolean(),
    reasoning: z.string(),
  }),

  // Execution model options — the planner references these by slot id in LWIR steps.
  models: [fastClassifier, longContextSummarizer],

  // Planner — required; harness defaults to the SDK workflowHarness
  planner: {
    model: openai("gpt-5"),          // AI SDK model (SDK auto-wraps as a private ModelSlot)
    system: "You design candidate-review workflows. Test code in scratch before committing.",
    skills: [
      skill("./skills/writing-loop-lwir/"),
      skill("./skills/output-schema-validation.md"),
    ],
    // tools: optional — extra tools beyond what the SDK injects (bash, scratch, memory read)
  },

  // Worker — optional; defaults to the SDK workflowHarness.
  // worker: { harness: customWorkerHarness, skills: [...] },

  // Tool names the LWIR can reference; the run supplies them (runWorkflow(..., { tools })).
  // Pass AI SDK tools keyed by name instead to declare and supply them in one place.
  tools: ["fetchCandidates", "scoreCandidate"],

  // Memory mounts — optional; these are the defaults.
  memory: {
    workflow: "rw",   // planner has read-write access to /mnt/memory/workflow/
    org: "ro",        // planner has read-only access to /mnt/memory/org/
  },
});

type PlannerConfig = {
  readonly model: AiSdkModel | ModelSlot;    // raw AI SDK model OK; SDK auto-wraps
  readonly harness?: Harness;                // defaults to workflowHarness
  readonly system?: string;                  // role-specific system prompt; merged with SDK addendum
  readonly skills?: readonly Skill[];        // workflow-declared skills
  readonly skillMaxRisk?: SkillRiskLevel;    // max accepted remote-skill risk for this role
  readonly skillOidcToken?: SkillOidcToken;  // token for the Skills API risk audit
  readonly tools?: ToolSet;                  // extra tools beyond SDK-injected (bash, scratch, memory read)
  readonly supervise?: (state: SuperviseOuterLoopState) => Promise<SuperviseDecision>; // outer loop
};

type WorkerConfig = {
  readonly harness?: Harness;                // defaults to workflowHarness
  readonly skills?: readonly Skill[];        // skills available to AI worker sessions
  readonly tools?: ToolSet;                  // extra tools merged with the workflow's tools
};

type MemoryConfig = {
  readonly workflow?: "rw" | "ro" | "none";    // default: "rw"
  readonly org?: "rw" | "ro" | "none";         // default: "ro"
  readonly attach?: ReadonlyArray<{ id: string; mode: "ro" }>;
};

const a = model(openai("gpt-4o-mini"), { description: "Classifier." });
// → slot id "gpt-4o-mini"

const b = model(openai("gpt-4o-mini"), { description: "Summarizer." });
// → slot id "gpt-4o-mini-tiger" (deterministic from description hash)

import { localWorld, runWorkflow } from "@little-stack/workflow";

const result = await runWorkflow(candidateReview, { jobDescription, candidate }, {
  world: localWorld({ dataDir: ".little-workflow" }),
  tools,
  timeout: "10m",
});

planner: {
  skills: [
    skill("./skills/writing-loop-lwir/"),       // directory form (SKILL.md + body)
    skill("./skills/output-schema-validation.md"),  // single-file shorthand
  ],
},