oxe-cc 1.2.1 → 1.4.0

package/packages/runtime/src/scheduler/agent-registry.ts

@@ -1,132 +1,132 @@
-import type { TaskExecutor } from './scheduler';
-import type { WorkspaceManager } from '../workspace/workspace-manager';
-import type { AgentRole, AgentActionLog } from './agent-roles';
-
-export type AgentStatus = 'idle' | 'running' | 'paused' | 'failed' | 'timeout';
-
-export interface AgentHeartbeat {
-  agent_id: string;
-  last_seen: string;
-  current_task: string | null;
-  status: AgentStatus;
-}
-
-export interface RegisteredAgent {
-  id: string;
-  executor: TaskExecutor;
-  workspaceManager: WorkspaceManager;
-  assignedTaskIds: string[];
-  heartbeat: AgentHeartbeat;
-  role?: AgentRole;
-  actionLog: AgentActionLog[];
-}
-
-export class AgentRegistry {
-  private agents = new Map<string, RegisteredAgent>();
-  private readonly heartbeatTimeoutMs: number;
-
-  constructor(heartbeatTimeoutMs = 30_000) {
-    this.heartbeatTimeoutMs = heartbeatTimeoutMs;
-  }
-
-  register(
-    id: string,
-    executor: TaskExecutor,
-    workspaceManager: WorkspaceManager,
-    assignedTaskIds: string[] = [],
-    role?: AgentRole
-  ): RegisteredAgent {
-    if (this.agents.has(id)) throw new Error(`Agent "${id}" is already registered`);
-    const agent: RegisteredAgent = {
-      id,
-      executor,
-      workspaceManager,
-      assignedTaskIds,
-      heartbeat: {
-        agent_id: id,
-        last_seen: new Date().toISOString(),
-        current_task: null,
-        status: 'idle',
-      },
-      role,
-      actionLog: [],
-    };
-    this.agents.set(id, agent);
-    return agent;
-  }
-
-  unregister(id: string): void {
-    this.agents.delete(id);
-  }
-
-  beat(id: string, currentTask: string | null = null): void {
-    const agent = this.agents.get(id);
-    if (!agent) return;
-    agent.heartbeat.last_seen = new Date().toISOString();
-    agent.heartbeat.current_task = currentTask;
-    agent.heartbeat.status = currentTask ? 'running' : 'idle';
-  }
-
-  setStatus(id: string, status: AgentStatus): void {
-    const agent = this.agents.get(id);
-    if (agent) agent.heartbeat.status = status;
-  }
-
-  isAlive(id: string): boolean {
-    const agent = this.agents.get(id);
-    if (!agent) return false;
-    const elapsed = Date.now() - new Date(agent.heartbeat.last_seen).getTime();
-    return elapsed < this.heartbeatTimeoutMs;
-  }
-
-  /** Returns agents that haven't sent a heartbeat within the timeout window */
-  timedOut(): RegisteredAgent[] {
-    return [...this.agents.values()].filter((a) => !this.isAlive(a.id));
-  }
-
-  liveAgents(): RegisteredAgent[] {
-    return [...this.agents.values()].filter((a) => this.isAlive(a.id));
-  }
-
-  get(id: string): RegisteredAgent | null {
-    return this.agents.get(id) ?? null;
-  }
-
-  list(): RegisteredAgent[] {
-    return [...this.agents.values()];
-  }
-
-  /**
-   * Reassign orphaned tasks from timed-out agents to a fallback agent.
-   * Returns the list of task IDs that were reassigned.
-   */
-  failover(fallbackAgentId: string): string[] {
-    const fallback = this.agents.get(fallbackAgentId);
-    if (!fallback) throw new Error(`Fallback agent "${fallbackAgentId}" not found`);
-
-    const orphaned: string[] = [];
-    for (const agent of this.timedOut()) {
-      orphaned.push(...agent.assignedTaskIds);
-      agent.assignedTaskIds = [];
-      agent.heartbeat.status = 'failed';
-    }
-
-    fallback.assignedTaskIds = [...fallback.assignedTaskIds, ...orphaned];
-    return orphaned;
-  }
-
-  /** Return all agents assigned to a given role */
-  getByRole(role: AgentRole): RegisteredAgent[] {
-    return [...this.agents.values()].filter((a) => a.role === role);
-  }
-
-  /** Append an action log entry for a registered agent (no-op if unknown) */
-  logAction(agentId: string, log: AgentActionLog): void {
-    const agent = this.agents.get(agentId);
-    if (agent) agent.actionLog.push(log);
-  }
-
-  clear(): void {
-    this.agents.clear();
-  }
-}
+import type { TaskExecutor } from './scheduler';
+import type { WorkspaceManager } from '../workspace/workspace-manager';
+import type { AgentRole, AgentActionLog } from './agent-roles';
+
+export type AgentStatus = 'idle' | 'running' | 'paused' | 'failed' | 'timeout';
+
+export interface AgentHeartbeat {
+  agent_id: string;
+  last_seen: string;
+  current_task: string | null;
+  status: AgentStatus;
+}
+
+export interface RegisteredAgent {
+  id: string;
+  executor: TaskExecutor;
+  workspaceManager: WorkspaceManager;
+  assignedTaskIds: string[];
+  heartbeat: AgentHeartbeat;
+  role?: AgentRole;
+  actionLog: AgentActionLog[];
+}
+
+export class AgentRegistry {
+  private agents = new Map<string, RegisteredAgent>();
+  private readonly heartbeatTimeoutMs: number;
+
+  constructor(heartbeatTimeoutMs = 30_000) {
+    this.heartbeatTimeoutMs = heartbeatTimeoutMs;
+  }
+
+  register(
+    id: string,
+    executor: TaskExecutor,
+    workspaceManager: WorkspaceManager,
+    assignedTaskIds: string[] = [],
+    role?: AgentRole
+  ): RegisteredAgent {
+    if (this.agents.has(id)) throw new Error(`Agent "${id}" is already registered`);
+    const agent: RegisteredAgent = {
+      id,
+      executor,
+      workspaceManager,
+      assignedTaskIds,
+      heartbeat: {
+        agent_id: id,
+        last_seen: new Date().toISOString(),
+        current_task: null,
+        status: 'idle',
+      },
+      role,
+      actionLog: [],
+    };
+    this.agents.set(id, agent);
+    return agent;
+  }
+
+  unregister(id: string): void {
+    this.agents.delete(id);
+  }
+
+  beat(id: string, currentTask: string | null = null): void {
+    const agent = this.agents.get(id);
+    if (!agent) return;
+    agent.heartbeat.last_seen = new Date().toISOString();
+    agent.heartbeat.current_task = currentTask;
+    agent.heartbeat.status = currentTask ? 'running' : 'idle';
+  }
+
+  setStatus(id: string, status: AgentStatus): void {
+    const agent = this.agents.get(id);
+    if (agent) agent.heartbeat.status = status;
+  }
+
+  isAlive(id: string): boolean {
+    const agent = this.agents.get(id);
+    if (!agent) return false;
+    const elapsed = Date.now() - new Date(agent.heartbeat.last_seen).getTime();
+    return elapsed < this.heartbeatTimeoutMs;
+  }
+
+  /** Returns agents that haven't sent a heartbeat within the timeout window */
+  timedOut(): RegisteredAgent[] {
+    return [...this.agents.values()].filter((a) => !this.isAlive(a.id));
+  }
+
+  liveAgents(): RegisteredAgent[] {
+    return [...this.agents.values()].filter((a) => this.isAlive(a.id));
+  }
+
+  get(id: string): RegisteredAgent | null {
+    return this.agents.get(id) ?? null;
+  }
+
+  list(): RegisteredAgent[] {
+    return [...this.agents.values()];
+  }
+
+  /**
+   * Reassign orphaned tasks from timed-out agents to a fallback agent.
+   * Returns the list of task IDs that were reassigned.
+   */
+  failover(fallbackAgentId: string): string[] {
+    const fallback = this.agents.get(fallbackAgentId);
+    if (!fallback) throw new Error(`Fallback agent "${fallbackAgentId}" not found`);
+
+    const orphaned: string[] = [];
+    for (const agent of this.timedOut()) {
+      orphaned.push(...agent.assignedTaskIds);
+      agent.assignedTaskIds = [];
+      agent.heartbeat.status = 'failed';
+    }
+
+    fallback.assignedTaskIds = [...fallback.assignedTaskIds, ...orphaned];
+    return orphaned;
+  }
+
+  /** Return all agents assigned to a given role */
+  getByRole(role: AgentRole): RegisteredAgent[] {
+    return [...this.agents.values()].filter((a) => a.role === role);
+  }
+
+  /** Append an action log entry for a registered agent (no-op if unknown) */
+  logAction(agentId: string, log: AgentActionLog): void {
+    const agent = this.agents.get(agentId);
+    if (agent) agent.actionLog.push(log);
+  }
+
+  clear(): void {
+    this.agents.clear();
+  }
+}

package/packages/runtime/src/scheduler/agent-roles.ts

@@ -1,109 +1,109 @@
-import type { TaskResult } from './scheduler';
-
-export type AgentRole = 'planner' | 'executor' | 'reviewer' | 'verifier';
-
-export interface AgentBudget {
-  max_tokens: number;
-  max_time_ms: number;
-  max_retries: number;
-  consumed_tokens: number;
-  consumed_time_ms: number;
-  consumed_retries: number;
-}
-
-export interface CooperativeHandoff {
-  handoff_id: string;
-  from_agent_id: string;
-  to_agent_id: string;
-  from_role: AgentRole;
-  to_role: AgentRole;
-  work_item_id: string;
-  context_pack_ref: string | null;
-  transferred_at: string;
-}
-
-export interface AgentActionLog {
-  agent_id: string;
-  role: AgentRole;
-  work_item_id: string;
-  action: 'execute' | 'verify' | 'review' | 'plan';
-  result: 'success' | 'failure';
-  duration_ms: number;
-  timestamp: string;
-}
-
-export function createBudget(
-  opts: Partial<Omit<AgentBudget, 'consumed_tokens' | 'consumed_time_ms' | 'consumed_retries'>> = {}
-): AgentBudget {
-  return {
-    max_tokens: opts.max_tokens ?? Infinity,
-    max_time_ms: opts.max_time_ms ?? Infinity,
-    max_retries: opts.max_retries ?? Infinity,
-    consumed_tokens: 0,
-    consumed_time_ms: 0,
-    consumed_retries: 0,
-  };
-}
-
-export function consumeBudget(
-  budget: AgentBudget,
-  delta: { tokens?: number; time_ms?: number; retries?: number }
-): AgentBudget {
-  return {
-    ...budget,
-    consumed_tokens: budget.consumed_tokens + (delta.tokens ?? 0),
-    consumed_time_ms: budget.consumed_time_ms + (delta.time_ms ?? 0),
-    consumed_retries: budget.consumed_retries + (delta.retries ?? 0),
-  };
-}
-
-export function isBudgetExhausted(budget: AgentBudget): boolean {
-  return (
-    budget.consumed_tokens >= budget.max_tokens ||
-    budget.consumed_time_ms >= budget.max_time_ms ||
-    budget.consumed_retries >= budget.max_retries
-  );
-}
-
-export function buildHandoff(opts: {
-  from_agent_id: string;
-  to_agent_id: string;
-  from_role: AgentRole;
-  to_role: AgentRole;
-  work_item_id: string;
-  context_pack_ref?: string | null;
-}): CooperativeHandoff {
-  return {
-    handoff_id: `hoff-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 6)}`,
-    from_agent_id: opts.from_agent_id,
-    to_agent_id: opts.to_agent_id,
-    from_role: opts.from_role,
-    to_role: opts.to_role,
-    work_item_id: opts.work_item_id,
-    context_pack_ref: opts.context_pack_ref ?? null,
-    transferred_at: new Date().toISOString(),
-  };
-}
-
-// ─── ArbitrationEngine ────────────────────────────────────────────────────────
-
-export class ArbitrationEngine {
-  /**
-   * Choose the best result from multiple competing agents.
-   * Prefers success; among successes prefers more evidence; falls back to first.
-   */
-  arbitrate(results: Array<{ agent_id: string; result: TaskResult }>): TaskResult {
-    if (results.length === 0) {
-      return { success: false, failure_class: null, evidence: [], output: 'no results to arbitrate' };
-    }
-
-    const successes = results.filter((r) => r.result.success);
-    if (successes.length === 0) {
-      return results[0].result;
-    }
-
-    // Among successes, prefer the one with most evidence
-    successes.sort((a, b) => b.result.evidence.length - a.result.evidence.length);
-    return successes[0].result;
-  }
-}
+import type { TaskResult } from './scheduler';
+
+export type AgentRole = 'planner' | 'executor' | 'reviewer' | 'verifier';
+
+export interface AgentBudget {
+  max_tokens: number;
+  max_time_ms: number;
+  max_retries: number;
+  consumed_tokens: number;
+  consumed_time_ms: number;
+  consumed_retries: number;
+}
+
+export interface CooperativeHandoff {
+  handoff_id: string;
+  from_agent_id: string;
+  to_agent_id: string;
+  from_role: AgentRole;
+  to_role: AgentRole;
+  work_item_id: string;
+  context_pack_ref: string | null;
+  transferred_at: string;
+}
+
+export interface AgentActionLog {
+  agent_id: string;
+  role: AgentRole;
+  work_item_id: string;
+  action: 'execute' | 'verify' | 'review' | 'plan';
+  result: 'success' | 'failure';
+  duration_ms: number;
+  timestamp: string;
+}
+
+export function createBudget(
+  opts: Partial<Omit<AgentBudget, 'consumed_tokens' | 'consumed_time_ms' | 'consumed_retries'>> = {}
+): AgentBudget {
+  return {
+    max_tokens: opts.max_tokens ?? Infinity,
+    max_time_ms: opts.max_time_ms ?? Infinity,
+    max_retries: opts.max_retries ?? Infinity,
+    consumed_tokens: 0,
+    consumed_time_ms: 0,
+    consumed_retries: 0,
+  };
+}
+
+export function consumeBudget(
+  budget: AgentBudget,
+  delta: { tokens?: number; time_ms?: number; retries?: number }
+): AgentBudget {
+  return {
+    ...budget,
+    consumed_tokens: budget.consumed_tokens + (delta.tokens ?? 0),
+    consumed_time_ms: budget.consumed_time_ms + (delta.time_ms ?? 0),
+    consumed_retries: budget.consumed_retries + (delta.retries ?? 0),
+  };
+}
+
+export function isBudgetExhausted(budget: AgentBudget): boolean {
+  return (
+    budget.consumed_tokens >= budget.max_tokens ||
+    budget.consumed_time_ms >= budget.max_time_ms ||
+    budget.consumed_retries >= budget.max_retries
+  );
+}
+
+export function buildHandoff(opts: {
+  from_agent_id: string;
+  to_agent_id: string;
+  from_role: AgentRole;
+  to_role: AgentRole;
+  work_item_id: string;
+  context_pack_ref?: string | null;
+}): CooperativeHandoff {
+  return {
+    handoff_id: `hoff-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 6)}`,
+    from_agent_id: opts.from_agent_id,
+    to_agent_id: opts.to_agent_id,
+    from_role: opts.from_role,
+    to_role: opts.to_role,
+    work_item_id: opts.work_item_id,
+    context_pack_ref: opts.context_pack_ref ?? null,
+    transferred_at: new Date().toISOString(),
+  };
+}
+
+// ─── ArbitrationEngine ────────────────────────────────────────────────────────
+
+export class ArbitrationEngine {
+  /**
+   * Choose the best result from multiple competing agents.
+   * Prefers success; among successes prefers more evidence; falls back to first.
+   */
+  arbitrate(results: Array<{ agent_id: string; result: TaskResult }>): TaskResult {
+    if (results.length === 0) {
+      return { success: false, failure_class: null, evidence: [], output: 'no results to arbitrate' };
+    }
+
+    const successes = results.filter((r) => r.result.success);
+    if (successes.length === 0) {
+      return results[0].result;
+    }
+
+    // Among successes, prefer the one with most evidence
+    successes.sort((a, b) => b.result.evidence.length - a.result.evidence.length);
+    return successes[0].result;
+  }
+}

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

@@ -1,4 +1,4 @@
-export * from './scheduler';
-export * from './run-journal';
-export * from './agent-registry';
-export * from './agent-roles';
+export * from './scheduler';
+export * from './run-journal';
+export * from './agent-registry';
+export * from './agent-roles';