memorix 1.2.0 → 1.2.2

package/src/team/messages.ts

@@ -1,203 +1,203 @@
-/**
- * Message Bus — Agent-to-agent communication
- *
- * In-memory message queue for team coordination.
- * Messages are temporary (not persisted as observations).
- * Supports direct send, broadcast, inbox retrieval, and read tracking.
- */
-
-import { randomUUID } from 'node:crypto';
-import type { AgentRegistry } from './registry.js';
-
-// ─── Types ───────────────────────────────────────────────────────────
-
-export type MessageType = 'request' | 'response' | 'info' | 'announcement' | 'contract' | 'error';
-
-export interface MessageSendInput {
-  from: string;
-  to: string;
-  type: MessageType;
-  content: string;
-}
-
-export interface MessageBroadcastInput {
-  from: string;
-  type: MessageType;
-  content: string;
-}
-
-export interface Message {
-  id: string;
-  from: string;
-  to: string;
-  type: MessageType;
-  content: string;
-  timestamp: Date;
-  read: boolean;
-}
-
-// ─── Message Bus ─────────────────────────────────────────────────────
-
-/** Max messages per inbox. Oldest read messages are evicted first. */
-const MAX_INBOX_SIZE = 200;
-/** Max message content length in bytes */
-const MAX_CONTENT_LENGTH = 10_000;
-
-export class MessageBus {
-  private inboxes = new Map<string, Message[]>();
-  private registry: AgentRegistry;
-
-  constructor(registry: AgentRegistry) {
-    this.registry = registry;
-  }
-
-  /**
-   * Send a message to a specific agent.
-   * Throws if receiver is unknown.
-   */
-  send(input: MessageSendInput): Message {
-    const receiver = this.registry.getAgent(input.to);
-    if (!receiver) {
-      throw new Error(`Unknown receiver agent: ${input.to}`);
-    }
-    if (receiver.status !== 'active') {
-      throw new Error(`Receiver agent is inactive: ${receiver.name}`);
-    }
-
-    const msg: Message = {
-      id: randomUUID(),
-      from: input.from,
-      to: input.to,
-      type: input.type,
-      content: input.content,
-      timestamp: new Date(),
-      read: false,
-    };
-
-    const inbox = this.inboxes.get(input.to) ?? [];
-    inbox.push(msg);
-
-    // Evict oldest read messages if inbox exceeds limit
-    if (inbox.length > MAX_INBOX_SIZE) {
-      const readIndices: number[] = [];
-      for (let i = 0; i < inbox.length; i++) {
-        if (inbox[i].read) readIndices.push(i);
-      }
-      // Remove oldest read messages first
-      const toRemove = inbox.length - MAX_INBOX_SIZE;
-      for (let i = Math.min(toRemove, readIndices.length) - 1; i >= 0; i--) {
-        inbox.splice(readIndices[i], 1);
-      }
-      // If still over limit, remove oldest unread
-      while (inbox.length > MAX_INBOX_SIZE) {
-        inbox.shift();
-      }
-    }
-
-    this.inboxes.set(input.to, inbox);
-    return msg;
-  }
-
-  /**
-   * Broadcast a message to all active agents except the sender.
-   */
-  broadcast(input: MessageBroadcastInput): Message[] {
-    const activeAgents = this.registry.listAgents({ status: 'active' });
-    const messages: Message[] = [];
-
-    for (const agent of activeAgents) {
-      if (agent.id === input.from) continue;
-
-      const msg = this.send({
-        from: input.from,
-        to: agent.id,
-        type: input.type,
-        content: input.content,
-      });
-      messages.push(msg);
-    }
-
-    return messages;
-  }
-
-  /**
-   * Get all messages in an agent's inbox (both read and unread).
-   */
-  getInbox(agentId: string): Message[] {
-    return [...(this.inboxes.get(agentId) ?? [])];
-  }
-
-  /**
-   * Mark specific messages as read.
-   */
-  markRead(agentId: string, messageIds: string[]): number {
-    const inbox = this.inboxes.get(agentId);
-    if (!inbox) return 0;
-
-    const idSet = new Set(messageIds);
-    let count = 0;
-    for (const msg of inbox) {
-      if (idSet.has(msg.id) && !msg.read) {
-        msg.read = true;
-        count++;
-      }
-    }
-    return count;
-  }
-
-  /**
-   * Count unread messages for an agent.
-   */
-  getUnreadCount(agentId: string): number {
-    const inbox = this.inboxes.get(agentId);
-    if (!inbox) return 0;
-    return inbox.filter(m => !m.read).length;
-  }
-
-  /**
-   * Remove all read messages from an agent's inbox.
-   * Returns the number of messages pruned.
-   */
-  pruneRead(agentId: string): number {
-    const inbox = this.inboxes.get(agentId);
-    if (!inbox) return 0;
-    const before = inbox.length;
-    const unread = inbox.filter(m => !m.read);
-    this.inboxes.set(agentId, unread);
-    return before - unread.length;
-  }
-
-  /**
-   * Clear all messages for an agent (used on agent leave).
-   */
-  clearInbox(agentId: string): void {
-    this.inboxes.delete(agentId);
-  }
-
-  /** Max allowed content length */
-  static get MAX_CONTENT_LENGTH() { return MAX_CONTENT_LENGTH; }
-
-  /** Serialize state for file persistence */
-  serialize(): { inboxes: Record<string, unknown[]> } {
-    const inboxes: Record<string, unknown[]> = {};
-    for (const [agentId, msgs] of this.inboxes) {
-      inboxes[agentId] = msgs.map(m => ({
-        ...m,
-        timestamp: m.timestamp.toISOString(),
-      }));
-    }
-    return { inboxes };
-  }
-
-  /** Hydrate state from file persistence */
-  hydrate(data: { inboxes?: Record<string, any[]> }): void {
-    this.inboxes.clear();
-    if (!data?.inboxes) return;
-    for (const [agentId, msgs] of Object.entries(data.inboxes)) {
-      this.inboxes.set(agentId, msgs.map(m => ({
-        ...m,
-        timestamp: new Date(m.timestamp),
-      })));
-    }
-  }
-}
+/**
+ * Message Bus — Agent-to-agent communication
+ *
+ * In-memory message queue for team coordination.
+ * Messages are temporary (not persisted as observations).
+ * Supports direct send, broadcast, inbox retrieval, and read tracking.
+ */
+
+import { randomUUID } from 'node:crypto';
+import type { AgentRegistry } from './registry.js';
+
+// ─── Types ───────────────────────────────────────────────────────────
+
+export type MessageType = 'request' | 'response' | 'info' | 'announcement' | 'contract' | 'error';
+
+export interface MessageSendInput {
+  from: string;
+  to: string;
+  type: MessageType;
+  content: string;
+}
+
+export interface MessageBroadcastInput {
+  from: string;
+  type: MessageType;
+  content: string;
+}
+
+export interface Message {
+  id: string;
+  from: string;
+  to: string;
+  type: MessageType;
+  content: string;
+  timestamp: Date;
+  read: boolean;
+}
+
+// ─── Message Bus ─────────────────────────────────────────────────────
+
+/** Max messages per inbox. Oldest read messages are evicted first. */
+const MAX_INBOX_SIZE = 200;
+/** Max message content length in bytes */
+const MAX_CONTENT_LENGTH = 10_000;
+
+export class MessageBus {
+  private inboxes = new Map<string, Message[]>();
+  private registry: AgentRegistry;
+
+  constructor(registry: AgentRegistry) {
+    this.registry = registry;
+  }
+
+  /**
+   * Send a message to a specific agent.
+   * Throws if receiver is unknown.
+   */
+  send(input: MessageSendInput): Message {
+    const receiver = this.registry.getAgent(input.to);
+    if (!receiver) {
+      throw new Error(`Unknown receiver agent: ${input.to}`);
+    }
+    if (receiver.status !== 'active') {
+      throw new Error(`Receiver agent is inactive: ${receiver.name}`);
+    }
+
+    const msg: Message = {
+      id: randomUUID(),
+      from: input.from,
+      to: input.to,
+      type: input.type,
+      content: input.content,
+      timestamp: new Date(),
+      read: false,
+    };
+
+    const inbox = this.inboxes.get(input.to) ?? [];
+    inbox.push(msg);
+
+    // Evict oldest read messages if inbox exceeds limit
+    if (inbox.length > MAX_INBOX_SIZE) {
+      const readIndices: number[] = [];
+      for (let i = 0; i < inbox.length; i++) {
+        if (inbox[i].read) readIndices.push(i);
+      }
+      // Remove oldest read messages first
+      const toRemove = inbox.length - MAX_INBOX_SIZE;
+      for (let i = Math.min(toRemove, readIndices.length) - 1; i >= 0; i--) {
+        inbox.splice(readIndices[i], 1);
+      }
+      // If still over limit, remove oldest unread
+      while (inbox.length > MAX_INBOX_SIZE) {
+        inbox.shift();
+      }
+    }
+
+    this.inboxes.set(input.to, inbox);
+    return msg;
+  }
+
+  /**
+   * Broadcast a message to all active agents except the sender.
+   */
+  broadcast(input: MessageBroadcastInput): Message[] {
+    const activeAgents = this.registry.listAgents({ status: 'active' });
+    const messages: Message[] = [];
+
+    for (const agent of activeAgents) {
+      if (agent.id === input.from) continue;
+
+      const msg = this.send({
+        from: input.from,
+        to: agent.id,
+        type: input.type,
+        content: input.content,
+      });
+      messages.push(msg);
+    }
+
+    return messages;
+  }
+
+  /**
+   * Get all messages in an agent's inbox (both read and unread).
+   */
+  getInbox(agentId: string): Message[] {
+    return [...(this.inboxes.get(agentId) ?? [])];
+  }
+
+  /**
+   * Mark specific messages as read.
+   */
+  markRead(agentId: string, messageIds: string[]): number {
+    const inbox = this.inboxes.get(agentId);
+    if (!inbox) return 0;
+
+    const idSet = new Set(messageIds);
+    let count = 0;
+    for (const msg of inbox) {
+      if (idSet.has(msg.id) && !msg.read) {
+        msg.read = true;
+        count++;
+      }
+    }
+    return count;
+  }
+
+  /**
+   * Count unread messages for an agent.
+   */
+  getUnreadCount(agentId: string): number {
+    const inbox = this.inboxes.get(agentId);
+    if (!inbox) return 0;
+    return inbox.filter(m => !m.read).length;
+  }
+
+  /**
+   * Remove all read messages from an agent's inbox.
+   * Returns the number of messages pruned.
+   */
+  pruneRead(agentId: string): number {
+    const inbox = this.inboxes.get(agentId);
+    if (!inbox) return 0;
+    const before = inbox.length;
+    const unread = inbox.filter(m => !m.read);
+    this.inboxes.set(agentId, unread);
+    return before - unread.length;
+  }
+
+  /**
+   * Clear all messages for an agent (used on agent leave).
+   */
+  clearInbox(agentId: string): void {
+    this.inboxes.delete(agentId);
+  }
+
+  /** Max allowed content length */
+  static get MAX_CONTENT_LENGTH() { return MAX_CONTENT_LENGTH; }
+
+  /** Serialize state for file persistence */
+  serialize(): { inboxes: Record<string, unknown[]> } {
+    const inboxes: Record<string, unknown[]> = {};
+    for (const [agentId, msgs] of this.inboxes) {
+      inboxes[agentId] = msgs.map(m => ({
+        ...m,
+        timestamp: m.timestamp.toISOString(),
+      }));
+    }
+    return { inboxes };
+  }
+
+  /** Hydrate state from file persistence */
+  hydrate(data: { inboxes?: Record<string, any[]> }): void {
+    this.inboxes.clear();
+    if (!data?.inboxes) return;
+    for (const [agentId, msgs] of Object.entries(data.inboxes)) {
+      this.inboxes.set(agentId, msgs.map(m => ({
+        ...m,
+        timestamp: new Date(m.timestamp),
+      })));
+    }
+  }
+}

package/src/team/poll.ts

@@ -1,132 +1,132 @@
-/**
- * Team Poll — Situational awareness for agents.
- *
- * Phase 4b: Provides `computeWatermark` (extracted from server.ts session_start)
- * and `computePoll` (full situational awareness in one call).
- *
- * Architectural boundary (B3): These are cross-process consistency mechanisms.
- * They read from SQLite (canonical truth), never from EventBus.
- */
-
-import type { TeamStore, TeamAgentRow, TeamTaskRow, TeamMessageRow } from './team-store.js';
-
-// ── Types ──────────────────────────────────────────────────────────
-
-export interface WatermarkResult {
-  lastSeenGeneration: number;
-  currentGeneration: number;
-  newObservationCount: number;
-}
-
-export interface PollResult {
-  agent: {
-    agentId: string;
-    status: string;
-    lastHeartbeat: number;
-  } | null;
-  watermark: WatermarkResult;
-  inbox: {
-    unreadCount: number;
-    messages: TeamMessageRow[];
-  };
-  tasks: {
-    myInProgress: TeamTaskRow[];
-    availableToClaim: TeamTaskRow[];
-    recentlyCompleted: TeamTaskRow[];
-    recentlyFailed: TeamTaskRow[];
-  };
-  team: {
-    activeAgents: TeamAgentRow[];
-    totalAgents: number;
-  };
-}
-
-// ── computeWatermark ───────────────────────────────────────────────
-
-/**
- * Compute the watermark for an agent: how many new observations
- * have appeared in this project since the agent's last session.
- *
- * Extracted from server.ts session_start to be shared by poll and session_start.
- *
- * @param lastSeenGeneration - Agent's last_seen_obs_generation from TeamStore
- * @param currentGeneration  - Current global generation from observation store
- * @param projectObservations - Pre-filtered observations for this project
- *                              with writeGeneration > lastSeenGeneration
- */
-export function computeWatermark(
-  lastSeenGeneration: number,
-  currentGeneration: number,
-  newObservationCount: number,
-): WatermarkResult {
-  return {
-    lastSeenGeneration,
-    currentGeneration,
-    newObservationCount,
-  };
-}
-
-// ── computePoll ────────────────────────────────────────────────────
-
-/**
- * Compute full situational awareness for an agent in one call.
- * Reads only from SQLite (TeamStore) — never from EventBus.
- *
- * @param teamStore - The TeamStore instance
- * @param projectId - Current project ID
- * @param agentId   - The requesting agent's ID (null if no agent registered)
- * @param watermark - Pre-computed watermark result
- */
-export function computePoll(
-  teamStore: TeamStore,
-  projectId: string,
-  agentId: string | null,
-  watermark: WatermarkResult,
-): PollResult {
-  // Agent info
-  let agent: PollResult['agent'] = null;
-  if (agentId) {
-    const row = teamStore.getAgent(agentId);
-    if (row) {
-      agent = {
-        agentId: row.agent_id,
-        status: row.status,
-        lastHeartbeat: row.last_heartbeat,
-      };
-    }
-  }
-
-  // Inbox
-  const unreadCount = agentId ? teamStore.getUnreadCount(projectId, agentId) : 0;
-  const messages = agentId ? teamStore.getInbox(projectId, agentId) : [];
-
-  // Tasks
-  const allTasks = teamStore.listTasks(projectId);
-  const myInProgress = agentId
-    ? allTasks.filter(t => t.assignee_agent_id === agentId && t.status === 'in_progress')
-    : [];
-
-  // Available to claim: pending, no assignee, all deps met
-  const pendingTasks = allTasks.filter(t => t.status === 'pending' && !t.assignee_agent_id);
-  const availableToClaim = pendingTasks.filter(t => {
-    const unmet = teamStore.getTaskDeps(t.task_id)
-      .map(depId => teamStore.getTask(depId))
-      .filter(dep => dep && dep.status !== 'completed');
-    return unmet.length === 0;
-  });
-
-  const recentlyCompleted = allTasks.filter(t => t.status === 'completed');
-  const recentlyFailed = allTasks.filter(t => t.status === 'failed');
-
-  // Team
-  const allAgents = teamStore.listAgents(projectId);
-  const activeAgents = allAgents.filter(a => a.status === 'active');
-
-  return {
-    agent,
-    watermark,
-    inbox: { unreadCount, messages },
-    tasks: { myInProgress, availableToClaim, recentlyCompleted, recentlyFailed },
-    team: { activeAgents, totalAgents: allAgents.length },
-  };
-}
+/**
+ * Team Poll — Situational awareness for agents.
+ *
+ * Phase 4b: Provides `computeWatermark` (extracted from server.ts session_start)
+ * and `computePoll` (full situational awareness in one call).
+ *
+ * Architectural boundary (B3): These are cross-process consistency mechanisms.
+ * They read from SQLite (canonical truth), never from EventBus.
+ */
+
+import type { TeamStore, TeamAgentRow, TeamTaskRow, TeamMessageRow } from './team-store.js';
+
+// ── Types ──────────────────────────────────────────────────────────
+
+export interface WatermarkResult {
+  lastSeenGeneration: number;
+  currentGeneration: number;
+  newObservationCount: number;
+}
+
+export interface PollResult {
+  agent: {
+    agentId: string;
+    status: string;
+    lastHeartbeat: number;
+  } | null;
+  watermark: WatermarkResult;
+  inbox: {
+    unreadCount: number;
+    messages: TeamMessageRow[];
+  };
+  tasks: {
+    myInProgress: TeamTaskRow[];
+    availableToClaim: TeamTaskRow[];
+    recentlyCompleted: TeamTaskRow[];
+    recentlyFailed: TeamTaskRow[];
+  };
+  team: {
+    activeAgents: TeamAgentRow[];
+    totalAgents: number;
+  };
+}
+
+// ── computeWatermark ───────────────────────────────────────────────
+
+/**
+ * Compute the watermark for an agent: how many new observations
+ * have appeared in this project since the agent's last session.
+ *
+ * Extracted from server.ts session_start to be shared by poll and session_start.
+ *
+ * @param lastSeenGeneration - Agent's last_seen_obs_generation from TeamStore
+ * @param currentGeneration  - Current global generation from observation store
+ * @param projectObservations - Pre-filtered observations for this project
+ *                              with writeGeneration > lastSeenGeneration
+ */
+export function computeWatermark(
+  lastSeenGeneration: number,
+  currentGeneration: number,
+  newObservationCount: number,
+): WatermarkResult {
+  return {
+    lastSeenGeneration,
+    currentGeneration,
+    newObservationCount,
+  };
+}
+
+// ── computePoll ────────────────────────────────────────────────────
+
+/**
+ * Compute full situational awareness for an agent in one call.
+ * Reads only from SQLite (TeamStore) — never from EventBus.
+ *
+ * @param teamStore - The TeamStore instance
+ * @param projectId - Current project ID
+ * @param agentId   - The requesting agent's ID (null if no agent registered)
+ * @param watermark - Pre-computed watermark result
+ */
+export function computePoll(
+  teamStore: TeamStore,
+  projectId: string,
+  agentId: string | null,
+  watermark: WatermarkResult,
+): PollResult {
+  // Agent info
+  let agent: PollResult['agent'] = null;
+  if (agentId) {
+    const row = teamStore.getAgent(agentId);
+    if (row) {
+      agent = {
+        agentId: row.agent_id,
+        status: row.status,
+        lastHeartbeat: row.last_heartbeat,
+      };
+    }
+  }
+
+  // Inbox
+  const unreadCount = agentId ? teamStore.getUnreadCount(projectId, agentId) : 0;
+  const messages = agentId ? teamStore.getInbox(projectId, agentId) : [];
+
+  // Tasks
+  const allTasks = teamStore.listTasks(projectId);
+  const myInProgress = agentId
+    ? allTasks.filter(t => t.assignee_agent_id === agentId && t.status === 'in_progress')
+    : [];
+
+  // Available to claim: pending, no assignee, all deps met
+  const pendingTasks = allTasks.filter(t => t.status === 'pending' && !t.assignee_agent_id);
+  const availableToClaim = pendingTasks.filter(t => {
+    const unmet = teamStore.getTaskDeps(t.task_id)
+      .map(depId => teamStore.getTask(depId))
+      .filter(dep => dep && dep.status !== 'completed');
+    return unmet.length === 0;
+  });
+
+  const recentlyCompleted = allTasks.filter(t => t.status === 'completed');
+  const recentlyFailed = allTasks.filter(t => t.status === 'failed');
+
+  // Team
+  const allAgents = teamStore.listAgents(projectId);
+  const activeAgents = allAgents.filter(a => a.status === 'active');
+
+  return {
+    agent,
+    watermark,
+    inbox: { unreadCount, messages },
+    tasks: { myInProgress, availableToClaim, recentlyCompleted, recentlyFailed },
+    team: { activeAgents, totalAgents: allAgents.length },
+  };
+}