pi-agent-fleet 0.2.0

package/src/types.ts

@@ -0,0 +1,109 @@
+export type OutputKind = "markdown" | "file-exists" | "verdict" | "json" | "yaml";
+
+export interface ContractOutput {
+  path: string;
+  kind: OutputKind;
+  required: boolean;
+}
+
+export type WorkerType = "research" | "code-run" | "reviewer" | "write" | "read-only";
+
+export type GateKind = "reviewer" | "none";
+export type Verdict = "lgtm" | "iterate" | "escalate";
+
+export interface LoopConfig {
+  gate: GateKind;
+  max_iterations: number;
+  lgtm_count: number;
+}
+
+export interface WorkerSpec {
+  id: string;
+  type: WorkerType;
+  task: string;
+  model?: string;
+  depends_on: string[];
+  outputs: ContractOutput[];
+  iterate?: boolean;
+  worktree?: boolean;
+}
+
+export interface FleetConfig {
+  max_concurrent: number;
+  model: string;
+  warn_cost_usd?: number;
+  loop?: LoopConfig;
+}
+
+export interface FleetSpec {
+  fleet_name: string;
+  type: "dag";
+  config: FleetConfig;
+  workers: WorkerSpec[];
+}
+
+export type NodeStatus =
+  | "pending" | "ready" | "running"
+  | "completed" | "failed" | "contract_failed" | "killed" | "blocked";
+
+export const TERMINAL_NODE_STATUSES: ReadonlySet<NodeStatus> = new Set([
+  "completed", "failed", "contract_failed", "killed", "blocked",
+]);
+
+export type FleetStatus = "planned" | "running" | "paused" | "completed" | "failed" | "killed";
+
+export interface ContractCheck {
+  path: string;
+  kind: OutputKind;
+  required: boolean;
+  ok: boolean;
+  error?: string;
+}
+
+export interface ContractResult {
+  ok: boolean;
+  checks: ContractCheck[];
+  verdict?: Verdict;
+  verdict_body?: string;
+}
+
+export interface NodeState {
+  status: NodeStatus;
+  started_at?: string;
+  ended_at?: string;
+  turns: number;
+  tokens: number;
+  cost_usd_estimate: number;
+  contract_result?: ContractResult;
+  produced_outputs: string[];
+  status_note?: string;
+}
+
+export interface IterationSnapshot {
+  n: number;
+  verdict: Verdict | null;
+  verdict_body: string | null;
+  started_at: string;
+  ended_at: string;
+  nodes: Record<string, NodeState>;
+}
+
+export interface FleetState {
+  fleet_name: string;
+  status: FleetStatus;
+  created_at: string;
+  cost_usd_estimate: number;
+  nodes: Record<string, NodeState>;
+  iteration: number;
+  lgtm_streak: number;
+  paused: boolean;
+  iterations: IterationSnapshot[];
+}
+
+export const WORKER_TYPE_TOOLS: Record<WorkerType, string[]> = {
+  "research": ["read", "grep", "find", "ls", "write", "web_search", "fetch_content"],
+  "code-run": ["read", "bash", "edit", "write", "grep", "find", "ls"],
+  "reviewer": ["read", "write", "grep", "find", "ls"],
+  "write": ["read", "write", "grep", "find", "ls"],
+  "read-only": ["read", "grep", "find", "ls"],
+};

package/src/ui.ts

@@ -0,0 +1,30 @@
+import type { FleetSpec, FleetState, NodeStatus } from "./types.js";
+
+const ICON: Record<NodeStatus, string> = {
+  completed: "✓", failed: "✗", contract_failed: "✗",
+  running: "⠹", blocked: "⊘", killed: "⊘", pending: "○", ready: "○",
+};
+
+export function buildWidgetLines(spec: FleetSpec, state: FleetState): string[] {
+  const done = spec.workers.filter((w) => state.nodes[w.id].status === "completed").length;
+  const loop = spec.config.loop;
+  let header: string;
+  if (loop) {
+    const lgtmCount = loop.lgtm_count ?? 1;
+    const lastVerdict = state.iterations.length > 0 ? state.iterations[state.iterations.length - 1].verdict : null;
+    const streakSegment = loop.gate === "reviewer" ? ` · streak ${state.lgtm_streak}/${lgtmCount}` : "";
+    header = `● fleet: ${spec.fleet_name} · iteration ${state.iteration}/${loop.max_iterations} · last verdict: ${lastVerdict ?? "—"}${streakSegment} (${done}/${spec.workers.length} done · $${state.cost_usd_estimate.toFixed(2)})`;
+  } else {
+    header = `● fleet: ${spec.fleet_name}  (${done}/${spec.workers.length} done · $${state.cost_usd_estimate.toFixed(2)})`;
+  }
+  const lines = [header];
+  spec.workers.forEach((w, i) => {
+    const n = state.nodes[w.id];
+    const branch = i === spec.workers.length - 1 ? "└─" : "├─";
+    const detail = n.status === "running" ? ` · ${n.turns} turns · ${(n.tokens / 1000).toFixed(1)}k tok` : "";
+    const note = n.status_note ? ` · ${n.status_note}` : "";
+    const model = w.model ?? spec.config.model;
+    lines.push(`${branch} ${ICON[n.status]} ${w.id} (${model})${detail}${note}`);
+  });
+  return lines;
+}

package/src/viz.ts

@@ -0,0 +1,36 @@
+import { topoLayers } from "./dag.js";
+import type { FleetSpec, FleetState, NodeStatus } from "./types.js";
+
+const ICON: Record<NodeStatus, string> = {
+  completed: "✓", failed: "✗", contract_failed: "✗",
+  running: "◌", blocked: "⊘", killed: "⊘",
+  pending: "○", ready: "○",
+};
+
+export function renderDag(spec: FleetSpec, state?: FleetState): string {
+  const layers = topoLayers(spec);
+  const modelOf = (id: string): string => {
+    const w = spec.workers.find((x) => x.id === id);
+    return w?.model ?? spec.config.model;
+  };
+  const label = (id: string): string => {
+    const st = state?.nodes[id]?.status;
+    const base = `${id} (${modelOf(id)})`;
+    return st ? `${ICON[st]} ${base}` : base;
+  };
+  const lines: string[] = [];
+  const header = layers.map((l, i) => `layer ${i}: ${l.map(label).join("  ")}`).join("\n");
+  lines.push(header, "", "edges:");
+  for (const w of spec.workers) {
+    for (const d of w.depends_on) lines.push(`  ${d} --▶ ${w.id}`);
+  }
+  if (spec.workers.every((w) => w.depends_on.length === 0)) lines.push("  (none — all parallel)");
+  return lines.join("\n");
+}
+
+export function dagNeedsFileFallback(spec: FleetSpec, termWidth: number): boolean {
+  if (spec.workers.length > 15) return true;
+  const layers = topoLayers(spec);
+  const widest = Math.max(...layers.map((l) => l.join("  ").length + 9));
+  return widest > termWidth;
+}