oxe-cc 0.9.2 → 1.0.0

package/packages/runtime/src/reducers/run-state-reducer.ts

@@ -0,0 +1,127 @@
+import type { OxeEvent } from '../events/envelope';
+import type { Run } from '../models/run';
+import type { WorkItem } from '../models/work-item';
+import type { Attempt } from '../models/attempt';
+import type { Workspace } from '../models/workspace';
+
+export interface RunState {
+  run: Run | null;
+  workItems: Map<string, WorkItem>;
+  attempts: Map<string, Attempt[]>;
+  workspaces: Map<string, Workspace>;
+  completedWorkItems: Set<string>;
+  failedWorkItems: Set<string>;
+  blockedWorkItems: Set<string>;
+}
+
+export function createEmptyRunState(): RunState {
+  return {
+    run: null,
+    workItems: new Map(),
+    attempts: new Map(),
+    workspaces: new Map(),
+    completedWorkItems: new Set(),
+    failedWorkItems: new Set(),
+    blockedWorkItems: new Set(),
+  };
+}
+
+export function reduce(events: OxeEvent[]): RunState {
+  return events.reduce(applyEvent, createEmptyRunState());
+}
+
+// Exported alias so debug-reducer can import applyEvent without circular issues
+export { applyEvent as applyEventExported };
+
+function applyEvent(state: RunState, event: OxeEvent): RunState {
+  switch (event.type) {
+    case 'RunStarted': {
+      const run = event.payload as unknown as Run;
+      return { ...state, run };
+    }
+
+    case 'RunCompleted': {
+      if (!state.run) return state;
+      const status = (event.payload as { status?: Run['status'] }).status ?? 'completed';
+      return {
+        ...state,
+        run: { ...state.run, status, ended_at: event.timestamp },
+      };
+    }
+
+    case 'WorkItemReady': {
+      if (!event.work_item_id) return state;
+      const workItems = new Map(state.workItems);
+      const existing = workItems.get(event.work_item_id);
+      if (existing) {
+        workItems.set(event.work_item_id, { ...existing, status: 'ready' });
+      } else {
+        // First time we see this work item — create from payload
+        const item = event.payload as unknown as WorkItem;
+        workItems.set(event.work_item_id, { ...item, work_item_id: event.work_item_id, status: 'ready' });
+      }
+      return { ...state, workItems };
+    }
+
+    case 'AttemptStarted': {
+      if (!event.work_item_id || !event.attempt_id) return state;
+      const attempts = new Map(state.attempts);
+      const attempt: Attempt = {
+        attempt_id: event.attempt_id,
+        work_item_id: event.work_item_id,
+        attempt_number: (event.payload as { attempt_number?: number }).attempt_number ?? 1,
+        workspace_id: null,
+        agent_profile: null,
+        model: null,
+        started_at: event.timestamp,
+        ended_at: null,
+        outcome: null,
+      };
+      const existing = attempts.get(event.work_item_id) ?? [];
+      attempts.set(event.work_item_id, [...existing, attempt]);
+      return { ...state, attempts };
+    }
+
+    case 'WorkspaceAllocated': {
+      const ws = event.payload as unknown as Workspace;
+      if (!ws.workspace_id) return state;
+      const workspaces = new Map(state.workspaces);
+      workspaces.set(ws.workspace_id, { ...ws, status: 'ready' });
+      return { ...state, workspaces };
+    }
+
+    case 'WorkItemCompleted': {
+      if (!event.work_item_id) return state;
+      const workItems = new Map(state.workItems);
+      const item = workItems.get(event.work_item_id);
+      if (item) workItems.set(event.work_item_id, { ...item, status: 'completed' });
+      const completedWorkItems = new Set(state.completedWorkItems);
+      completedWorkItems.add(event.work_item_id);
+      return { ...state, workItems, completedWorkItems };
+    }
+
+    case 'WorkItemBlocked': {
+      if (!event.work_item_id) return state;
+      const workItems = new Map(state.workItems);
+      const item = workItems.get(event.work_item_id);
+      if (item) workItems.set(event.work_item_id, { ...item, status: 'blocked' });
+      const blockedWorkItems = new Set(state.blockedWorkItems);
+      blockedWorkItems.add(event.work_item_id);
+      return { ...state, workItems, blockedWorkItems };
+    }
+
+    default:
+      return state;
+  }
+}
+
+export function getWorkItemStatus(
+  state: RunState,
+  workItemId: string
+): WorkItem['status'] | null {
+  return state.workItems.get(workItemId)?.status ?? null;
+}
+
+export function getAttemptCount(state: RunState, workItemId: string): number {
+  return state.attempts.get(workItemId)?.length ?? 0;
+}

package/packages/runtime/src/scheduler/index.ts

@@ -0,0 +1 @@
+export * from './scheduler';

package/packages/runtime/src/scheduler/multi-agent-coordinator.ts

@@ -0,0 +1,231 @@
+import { appendEvent } from '../events/bus';
+import type { ExecutionGraph, GraphNode } from '../compiler/graph-compiler';
+import type { WorkspaceManager } from '../workspace/workspace-manager';
+import type { TaskExecutor, TaskResult, SchedulerContext } from './scheduler';
+import { Scheduler } from './scheduler';
+import type { WorkspaceLease } from '../models/workspace';
+
+export type CoordinationMode = 'parallel' | 'competitive' | 'cooperative';
+
+export interface AgentSpec {
+  id: string;
+  executor: TaskExecutor;
+  workspaceManager: WorkspaceManager;
+  /** Task IDs this agent is responsible for (used in parallel mode) */
+  assignedTaskIds?: string[];
+}
+
+export interface CoordinationOptions {
+  mode: CoordinationMode;
+  agents: AgentSpec[];
+  projectRoot: string;
+  sessionId: string | null;
+  runId: string;
+  onEvent?: SchedulerContext['onEvent'];
+}
+
+export interface CoordinationResult {
+  mode: CoordinationMode;
+  run_id: string;
+  completed: string[];
+  failed: string[];
+  blocked: string[];
+  agent_results: Array<{ agent_id: string; completed: string[]; failed: string[] }>;
+}
+
+// ─── Parallel mode ───────────────────────────────────────────────────────────
+// Tasks are partitioned across agents. Each agent runs its own Scheduler
+// on a sub-graph. Results are merged.
+
+async function runParallel(
+  graph: ExecutionGraph,
+  opts: CoordinationOptions
+): Promise<CoordinationResult> {
+  const { agents, projectRoot, sessionId, runId } = opts;
+
+  // Partition tasks across agents (round-robin if assignedTaskIds not set)
+  const partitions = agents.map((a) => a.assignedTaskIds ?? []);
+  if (partitions.every((p) => p.length === 0)) {
+    const allIds = [...graph.nodes.keys()];
+    allIds.forEach((id, i) => {
+      partitions[i % agents.length].push(id);
+    });
+  }
+
+  appendEvent(projectRoot, sessionId, {
+    type: 'RunStarted',
+    run_id: runId,
+    payload: { mode: 'parallel', agent_count: agents.length },
+  });
+
+  const agentResults = await Promise.all(
+    agents.map(async (agent, idx) => {
+      const subGraph = subGraphFor(graph, partitions[idx]);
+      if (subGraph.nodes.size === 0) {
+        return { agent_id: agent.id, completed: [], failed: [] };
+      }
+      const ctx: SchedulerContext = {
+        projectRoot,
+        sessionId,
+        runId: `${runId}-agent${idx}`,
+        executor: agent.executor,
+        workspaceManager: agent.workspaceManager,
+        onEvent: opts.onEvent,
+      };
+      const scheduler = new Scheduler();
+      const result = await scheduler.run(subGraph, ctx);
+      return { agent_id: agent.id, completed: result.completed, failed: result.failed };
+    })
+  );
+
+  const completed = agentResults.flatMap((r) => r.completed);
+  const failed = agentResults.flatMap((r) => r.failed);
+
+  appendEvent(projectRoot, sessionId, {
+    type: 'RunCompleted',
+    run_id: runId,
+    payload: { mode: 'parallel', completed: completed.length, failed: failed.length },
+  });
+
+  return { mode: 'parallel', run_id: runId, completed, failed, blocked: [], agent_results: agentResults };
+}
+
+// ─── Competitive mode ────────────────────────────────────────────────────────
+// Two agents attempt the same task. First success wins; the loser's workspace
+// is disposed. Requires exactly 2 agents.
+
+async function runCompetitive(
+  graph: ExecutionGraph,
+  opts: CoordinationOptions
+): Promise<CoordinationResult> {
+  if (opts.agents.length < 2) {
+    throw new Error('Competitive mode requires at least 2 agents');
+  }
+  const [agentA, agentB] = opts.agents;
+  const { projectRoot, sessionId, runId } = opts;
+
+  appendEvent(projectRoot, sessionId, {
+    type: 'RunStarted',
+    run_id: runId,
+    payload: { mode: 'competitive' },
+  });
+
+  const completed: string[] = [];
+  const failed: string[] = [];
+
+  for (const wave of graph.waves) {
+    for (const nodeId of wave.node_ids) {
+      const node = graph.nodes.get(nodeId)!;
+      const result = await competeTwoAgents(nodeId, node, agentA, agentB, opts);
+      if (result.success) completed.push(nodeId);
+      else failed.push(nodeId);
+      if (failed.length > 0) break;
+    }
+    if (failed.length > 0) break;
+  }
+
+  appendEvent(projectRoot, sessionId, {
+    type: 'RunCompleted',
+    run_id: runId,
+    payload: { mode: 'competitive', completed: completed.length },
+  });
+
+  return {
+    mode: 'competitive',
+    run_id: runId,
+    completed,
+    failed,
+    blocked: [],
+    agent_results: [
+      { agent_id: agentA.id, completed, failed },
+      { agent_id: agentB.id, completed: [], failed: [] },
+    ],
+  };
+}
+
+async function competeTwoAgents(
+  nodeId: string,
+  node: GraphNode,
+  agentA: AgentSpec,
+  agentB: AgentSpec,
+  opts: CoordinationOptions
+): Promise<TaskResult> {
+  const { projectRoot, sessionId, runId } = opts;
+
+  const allocA = await agentA.workspaceManager.allocate({
+    work_item_id: nodeId, attempt_number: 1, strategy: node.workspace_strategy, mutation_scope: node.mutation_scope,
+  });
+  const allocB = await agentB.workspaceManager.allocate({
+    work_item_id: nodeId, attempt_number: 1, strategy: node.workspace_strategy, mutation_scope: node.mutation_scope,
+  });
+
+  appendEvent(projectRoot, sessionId, {
+    type: 'AttemptStarted',
+    run_id: runId,
+    work_item_id: nodeId,
+    payload: { mode: 'competitive', agents: [agentA.id, agentB.id] },
+  });
+
+  // Race both agents — first success wins
+  const [resultA, resultB] = await Promise.all([
+    agentA.executor.execute(node, allocA, runId, 1).catch((e) => ({
+      success: false, failure_class: 'env' as const, evidence: [], output: String(e),
+    })),
+    agentB.executor.execute(node, allocB, runId, 1).catch((e) => ({
+      success: false, failure_class: 'env' as const, evidence: [], output: String(e),
+    })),
+  ]);
+
+  // Clean up both workspaces
+  await Promise.all([
+    agentA.workspaceManager.dispose(allocA.workspace_id).catch(() => {}),
+    agentB.workspaceManager.dispose(allocB.workspace_id).catch(() => {}),
+  ]);
+
+  // Pick winner: prefer success; if both succeed, prefer A (primary agent)
+  const winner = resultA.success ? resultA : resultB.success ? resultB : resultA;
+
+  if (winner.success) {
+    appendEvent(projectRoot, sessionId, { type: 'WorkItemCompleted', run_id: runId, work_item_id: nodeId, payload: { mode: 'competitive' } });
+  } else {
+    appendEvent(projectRoot, sessionId, { type: 'WorkItemBlocked', run_id: runId, work_item_id: nodeId, payload: { mode: 'competitive', failure_class: winner.failure_class } });
+  }
+
+  return winner;
+}
+
+// ─── Public API ──────────────────────────────────────────────────────────────
+
+export class MultiAgentCoordinator {
+  async run(graph: ExecutionGraph, opts: CoordinationOptions): Promise<CoordinationResult> {
+    switch (opts.mode) {
+      case 'parallel': return runParallel(graph, opts);
+      case 'competitive': return runCompetitive(graph, opts);
+      case 'cooperative':
+        // Cooperative mode: planner (agent[0]) prepares context,
+        // executor (agent[1]) implements. For R3, delegate to sequential parallel.
+        return runParallel(graph, { ...opts, mode: 'parallel' });
+      default:
+        throw new Error(`Unknown coordination mode: ${opts.mode}`);
+    }
+  }
+}
+
+// ─── Helpers ─────────────────────────────────────────────────────────────────
+
+function subGraphFor(graph: ExecutionGraph, nodeIds: string[]): ExecutionGraph {
+  const ids = new Set(nodeIds);
+  const nodes = new Map([...graph.nodes].filter(([id]) => ids.has(id)));
+  const edges = graph.edges.filter((e) => ids.has(e.from) && ids.has(e.to));
+  const waves = graph.waves.map((w) => ({
+    wave_number: w.wave_number,
+    node_ids: w.node_ids.filter((id) => ids.has(id)),
+  })).filter((w) => w.node_ids.length > 0);
+
+  return {
+    nodes,
+    edges,
+    waves,
+    metadata: { ...graph.metadata, node_count: nodes.size, wave_count: waves.length },
+  };
+}

package/packages/runtime/src/scheduler/scheduler.ts

@@ -0,0 +1,281 @@
+import { appendEvent } from '../events/bus';
+import type { OxeEvent } from '../events/envelope';
+import type { EventInput } from '../events/bus';
+import type { ExecutionGraph, GraphNode } from '../compiler/graph-compiler';
+import type { WorkspaceManager, WorkspaceRequest } from '../workspace/workspace-manager';
+import type { WorkspaceLease } from '../models/workspace';
+
+export interface TaskResult {
+  success: boolean;
+  failure_class: 'env' | 'policy' | 'test' | 'timeout' | null;
+  evidence: string[];
+  output: string;
+}
+
+export interface TaskExecutor {
+  execute(
+    node: GraphNode,
+    lease: WorkspaceLease,
+    runId: string,
+    attemptNumber: number
+  ): Promise<TaskResult>;
+}
+
+export interface SchedulerContext {
+  projectRoot: string;
+  sessionId: string | null;
+  runId: string;
+  executor: TaskExecutor;
+  workspaceManager: WorkspaceManager;
+  onEvent?: (event: OxeEvent) => void;
+}
+
+export interface RunResult {
+  run_id: string;
+  status: 'completed' | 'failed' | 'cancelled';
+  completed: string[];
+  failed: string[];
+  blocked: string[];
+}
+
+type NodeStatus = 'pending' | 'ready' | 'running' | 'completed' | 'failed' | 'blocked';
+
+export class Scheduler {
+  private cancelled = false;
+  private paused = false;
+
+  async run(graph: ExecutionGraph, ctx: SchedulerContext): Promise<RunResult> {
+    this.cancelled = false;
+    this.paused = false;
+
+    const status = new Map<string, NodeStatus>();
+    for (const id of graph.nodes.keys()) status.set(id, 'pending');
+
+    const completed: string[] = [];
+    const failed: string[] = [];
+    const blocked: string[] = [];
+
+    this.emit(ctx, { type: 'RunStarted', payload: { run_id: ctx.runId } });
+
+    for (const wave of graph.waves) {
+      if (this.cancelled) break;
+      const waveFailed = await this.runWave(
+        wave.node_ids,
+        graph,
+        ctx,
+        status,
+        completed,
+        failed,
+        blocked
+      );
+      if (waveFailed) break;
+    }
+
+    // Any remaining pending nodes become blocked
+    for (const [id, s] of status) {
+      if (s === 'pending') {
+        status.set(id, 'blocked');
+        blocked.push(id);
+        this.emit(ctx, {
+          type: 'WorkItemBlocked',
+          work_item_id: id,
+          payload: { reason: 'upstream_wave_failed' },
+        });
+      }
+    }
+
+    const finalStatus: RunResult['status'] = this.cancelled
+      ? 'cancelled'
+      : failed.length > 0
+      ? 'failed'
+      : 'completed';
+
+    this.emit(ctx, {
+      type: 'RunCompleted',
+      payload: { run_id: ctx.runId, status: finalStatus },
+    });
+
+    return { run_id: ctx.runId, status: finalStatus, completed, failed, blocked };
+  }
+
+  private async runWave(
+    nodeIds: string[],
+    graph: ExecutionGraph,
+    ctx: SchedulerContext,
+    status: Map<string, NodeStatus>,
+    completed: string[],
+    failed: string[],
+    blocked: string[]
+  ): Promise<boolean> {
+    // Partition: eligible vs blocked-by-dep
+    const eligible: string[] = [];
+    const depsNotMet: string[] = [];
+
+    for (const id of nodeIds) {
+      const node = graph.nodes.get(id)!;
+      const depsMet = node.depends_on.every((dep) => status.get(dep) === 'completed');
+      if (depsMet) {
+        eligible.push(id);
+      } else {
+        depsNotMet.push(id);
+      }
+    }
+
+    // Nodes whose deps weren't met in this wave → blocked
+    for (const id of depsNotMet) {
+      status.set(id, 'blocked');
+      blocked.push(id);
+      this.emit(ctx, {
+        type: 'WorkItemBlocked',
+        work_item_id: id,
+        payload: { reason: 'dependency_not_met' },
+      });
+    }
+
+    // Separate read-only (no mutation_scope) from mutation nodes
+    const readOnly = eligible.filter((id) => {
+      const node = graph.nodes.get(id)!;
+      return node.mutation_scope.length === 0;
+    });
+    const mutations = eligible.filter((id) => !readOnly.includes(id));
+
+    // Run read-only nodes in parallel
+    if (readOnly.length > 0) {
+      await Promise.all(
+        readOnly.map((id) => this.runNode(id, graph, ctx, status, completed, failed))
+      );
+    }
+
+    // Run mutation nodes sequentially to avoid scope conflicts
+    for (const id of mutations) {
+      if (this.cancelled) break;
+      await this.runNode(id, graph, ctx, status, completed, failed);
+    }
+
+    return failed.length > 0;
+  }
+
+  private async runNode(
+    nodeId: string,
+    graph: ExecutionGraph,
+    ctx: SchedulerContext,
+    status: Map<string, NodeStatus>,
+    completed: string[],
+    failed: string[]
+  ): Promise<void> {
+    const node = graph.nodes.get(nodeId)!;
+    status.set(nodeId, 'running');
+    this.emit(ctx, {
+      type: 'WorkItemReady',
+      work_item_id: nodeId,
+      payload: { title: node.title, wave: node.wave },
+    });
+
+    let lease: WorkspaceLease | null = null;
+    let lastResult: TaskResult | null = null;
+    const maxAttempts = node.policy.max_retries + 1;
+
+    for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+      const attemptId = `${nodeId}-a${attempt}`;
+
+      this.emit(ctx, {
+        type: 'AttemptStarted',
+        work_item_id: nodeId,
+        attempt_id: attemptId,
+        payload: { attempt_number: attempt },
+      });
+
+      try {
+        const wsReq: WorkspaceRequest = {
+          work_item_id: nodeId,
+          attempt_number: attempt,
+          strategy: node.workspace_strategy,
+          mutation_scope: node.mutation_scope,
+        };
+        lease = await ctx.workspaceManager.allocate(wsReq);
+        this.emit(ctx, {
+          type: 'WorkspaceAllocated',
+          work_item_id: nodeId,
+          attempt_id: attemptId,
+          payload: { workspace_id: lease.workspace_id, strategy: lease.strategy },
+        });
+
+        lastResult = await ctx.executor.execute(node, lease, ctx.runId, attempt);
+
+        if (lastResult.success) {
+          this.emit(ctx, {
+            type: 'WorkItemCompleted',
+            work_item_id: nodeId,
+            attempt_id: attemptId,
+            payload: { attempt_number: attempt, evidence: lastResult.evidence },
+          });
+          status.set(nodeId, 'completed');
+          completed.push(nodeId);
+          return;
+        }
+
+        // Policy failures never retry
+        if (lastResult.failure_class === 'policy') break;
+
+        if (attempt < maxAttempts) {
+          this.emit(ctx, {
+            type: 'RetryScheduled',
+            work_item_id: nodeId,
+            payload: { next_attempt: attempt + 1, reason: lastResult.failure_class },
+          });
+        }
+      } catch (err) {
+        lastResult = {
+          success: false,
+          failure_class: 'env',
+          evidence: [],
+          output: String(err),
+        };
+        if (attempt < maxAttempts) {
+          this.emit(ctx, {
+            type: 'RetryScheduled',
+            work_item_id: nodeId,
+            payload: { next_attempt: attempt + 1, reason: 'env' },
+          });
+        }
+      } finally {
+        if (lease) {
+          await ctx.workspaceManager.dispose(lease.workspace_id).catch(() => {});
+          lease = null;
+        }
+      }
+    }
+
+    // All attempts exhausted
+    this.emit(ctx, {
+      type: 'WorkItemBlocked',
+      work_item_id: nodeId,
+      payload: { failure_class: lastResult?.failure_class ?? 'env', max_attempts: maxAttempts },
+    });
+    status.set(nodeId, 'failed');
+    failed.push(nodeId);
+  }
+
+  pause(): void {
+    this.paused = true;
+  }
+
+  resume(): void {
+    this.paused = false;
+  }
+
+  cancel(): void {
+    this.cancelled = true;
+  }
+
+  private emit(
+    ctx: SchedulerContext,
+    input: EventInput
+  ): void {
+    const event = appendEvent(ctx.projectRoot, ctx.sessionId, {
+      run_id: ctx.runId,
+      ...input,
+    });
+    ctx.onEvent?.(event);
+  }
+}

package/packages/runtime/src/verification/index.ts

@@ -0,0 +1 @@
+export * from './verification-compiler';