@mndrk/agx 2.3.3 → 2.4.1

package/cloud-runtime/standalone/apps/local/lib/cli-runner.ts

@@ -228,11 +228,13 @@ const createCodexJsonParser = (
 };
 
 const createThoughtFilterParser = (
-  inner: StreamParser
+  inner: StreamParser,
+  onThought?: (content: string) => void
 ): StreamParser => {
   let buffer = "";
   let inThought = false;
   let thoughtStarted = false;
+  let thoughtContent = "";
 
   const THINK_START_PATTERNS = [
     /^Thinking\.\.\./,
@@ -260,6 +262,12 @@ const createThoughtFilterParser = (
           }
         }
         if (endIdx !== -1) {
+          // Capture thought content before the end marker
+          thoughtContent += buffer.slice(0, endIdx);
+          if (onThought && thoughtContent.trim()) {
+            onThought(thoughtContent.trim());
+          }
+          thoughtContent = "";
           // Skip everything up to and including the end marker
           buffer = buffer.slice(endIdx + endLen);
           inThought = false;
@@ -267,7 +275,9 @@ const createThoughtFilterParser = (
           buffer = buffer.replace(/^\s*\n*/, "");
           continue;
         }
-        // Haven't found end yet — keep buffering (don't emit)
+        // Haven't found end yet — accumulate thought content, keep buffering
+        thoughtContent += buffer;
+        buffer = "";
         return;
       }
 
@@ -281,6 +291,7 @@ const createThoughtFilterParser = (
           buffer = buffer.slice(match.index + match[0].length);
           inThought = true;
           thoughtStarted = true;
+          thoughtContent = "";
           break;
         }
       }
@@ -348,6 +359,7 @@ function providerNativeCommand({
     case "claude": {
       const args = [
         "-p",
+        prompt,
         "--verbose",
         "--output-format",
         "stream-json",
@@ -355,7 +367,6 @@ function providerNativeCommand({
       ];
       if (model) args.push("--model", model);
       if (systemPrompt) args.push("--system-prompt", systemPrompt);
-      args.push(prompt);
       return { command: "claude", args, parser: "claude-stream-json" };
     }
     case "gemini":
@@ -386,6 +397,7 @@ function providerNativeCommand({
       const zaiApiKey = process.env.ZAI_API_KEY?.trim();
       const zaiArgs = [
         "-p",
+        prompt,
         "--verbose",
         "--output-format",
         "stream-json",
@@ -393,7 +405,6 @@ function providerNativeCommand({
       ];
       if (model) zaiArgs.push("--model", model);
       if (systemPrompt) zaiArgs.push("--system-prompt", systemPrompt);
-      zaiArgs.push(prompt);
       return {
         command: "claude",
         args: zaiArgs,
@@ -409,6 +420,22 @@ function providerNativeCommand({
   }
 }
 
+function agxStreamingPassthrough(provider: ChatProvider): {
+  args: string[];
+  parser: CliAttempt["parser"];
+} {
+  switch (provider) {
+    case "claude":
+      return { args: ["--output-format", "stream-json", "--include-partial-messages"], parser: "claude-stream-json" };
+    case "gemini":
+      return { args: ["-o", "stream-json"], parser: "gemini-stream-json" };
+    case "codex":
+      return { args: ["--json"], parser: "codex-json" };
+    default:
+      return { args: [], parser: "raw" };
+  }
+}
+
 function agxCommandForProvider({
   provider,
   model,
@@ -419,17 +446,15 @@ function agxCommandForProvider({
   model: string | null;
   prompt: string;
   passthroughArgs?: string[];
-}): {
-  command: string;
-  args: string[];
-  parser: "raw";
-} {
+}): CliAttempt {
   const args = [provider, "-y", "--print", "--prompt", prompt];
   if (model) args.push("--model", model);
-  if (passthroughArgs && passthroughArgs.length > 0) {
-    args.push("--", ...passthroughArgs);
+  const streaming = agxStreamingPassthrough(provider);
+  const allPassthrough = [...(passthroughArgs || []), ...streaming.args];
+  if (allPassthrough.length > 0) {
+    args.push("--", ...allPassthrough);
   }
-  return { command: "agx", args, parser: "raw" };
+  return { command: "agx", args, parser: streaming.parser };
 }
 
 function resolveBundledCliPath(): string | null {
@@ -549,20 +574,18 @@ function bundledAgxCommandForProvider({
   prompt: string;
   cliPath?: string | null;
   passthroughArgs?: string[];
-}): {
-  command: string;
-  args: string[];
-  parser: "raw";
-} | null {
+}): CliAttempt | null {
   const resolvedCliPath = cliPath === undefined ? resolveBundledCliPath() : cliPath;
   if (!resolvedCliPath) return null;
 
   const args = [resolvedCliPath, provider, "-y", "--print", "--prompt", prompt];
   if (model) args.push("--model", model);
-  if (passthroughArgs && passthroughArgs.length > 0) {
-    args.push("--", ...passthroughArgs);
+  const streaming = agxStreamingPassthrough(provider);
+  const allPassthrough = [...(passthroughArgs || []), ...streaming.args];
+  if (allPassthrough.length > 0) {
+    args.push("--", ...allPassthrough);
   }
-  return { command: process.execPath, args, parser: "raw" };
+  return { command: process.execPath, args, parser: streaming.parser };
 }
 
 // Core runner
@@ -713,6 +736,7 @@ export async function runCliResponse({
   passthroughArgs,
   signal,
   onDelta,
+  onThought,
   onLog,
   onSpawn,
 }: {
@@ -726,6 +750,7 @@ export async function runCliResponse({
   passthroughArgs?: string[];
   signal?: AbortSignal;
   onDelta: (chunk: string) => void;
+  onThought?: (content: string) => void;
   onLog?: (stream: "stdout" | "stderr", line: string) => void;
   onSpawn?: (pid: number) => void;
 }): Promise<void> {
@@ -780,7 +805,7 @@ export async function runCliResponse({
             ? createGeminiStreamJsonParser(wrappedOnDelta)
             : createRawParser(wrappedOnDelta);
     const parser = attempt.filterThoughts
-      ? createThoughtFilterParser(baseParser)
+      ? createThoughtFilterParser(baseParser, onThought)
       : baseParser;
 
     try {

package/cloud-runtime/standalone/apps/local/lib/constants/timing.ts

@@ -8,6 +8,7 @@ export const UI_POLL_CHAT_CHECK_MS = 10_000;
 export const UI_POLL_CHAT_ALT_MS = 15_000;
 export const UI_POLL_OFFLINE_CHECK_MS = 30_000;
 export const UI_RECONNECT_DELAY_MS = 2_000;
+export const UI_POLL_UPDATE_CHECK_MS = 30 * 60 * 1000;
 
 // ── Backend / Infrastructure ────────────────────────────
 export const DB_BUSY_TIMEOUT_MS = 5_000;

package/cloud-runtime/standalone/apps/local/lib/history-store.ts

@@ -1207,6 +1207,20 @@ export async function sweepStaleChatRuns(input?: {
       params.push(normalizedThreadId);
     }
 
+    // Skip runs that still have live agent process heartbeats.
+    // If any agent_processes row has recent last_activity and is in an
+    // active state, the run is still alive even though chat_runs.updated_at
+    // hasn't been bumped (streaming deltas update agent_processes, not chat_runs).
+    clauses.push(`id NOT IN (
+      SELECT cr.id FROM chat_runs cr
+      JOIN agent_processes ap ON ap.workspace_id = cr.thread_id
+      WHERE cr.status = 'running'
+        AND cr.updated_at <= ?
+        AND ap.state IN ('running', 'spawning')
+        AND ap.last_activity > ?
+    )`);
+    params.push(cutoff, cutoff);
+
     const result = db
       .prepare(
         `UPDATE chat_runs

package/cloud-runtime/standalone/apps/local/lib/linear-run-status.ts

@@ -1,4 +1,6 @@
-import type { LinearRun } from "@/hooks/useLinearRuns";
+import type { TrackerRunRecord } from "@/lib/tracker/tracker-run-store";
+
+type LinearRun = TrackerRunRecord;
 
 export type RunDisplayTone = LinearRun["status"] | "ready";
 

package/cloud-runtime/standalone/apps/local/lib/{objective-linear-issues.ts → objective-tracker-issues.ts}

@@ -1,3 +1,7 @@
+// Tracker-agnostic objective issue helpers.
+// Inlined from the deleted objective-linear-issues.ts with canonical
+// tracker-agnostic names.
+
 import "server-only";
 
 import {
@@ -6,7 +10,9 @@ import {
   type LinearIssueSummary,
 } from "@/lib/linear-issues";
 
-const OBJECTIVE_LINEAR_TERMINAL_STATUSES = new Set([
+export type ObjectiveTrackerIssueSummary = LinearIssueSummary;
+
+const OBJECTIVE_TRACKER_TERMINAL_STATUSES = new Set([
   "done",
   "cancelled",
   "canceled",
@@ -17,11 +23,11 @@ export function matchesObjectiveLabel(label: string, objectiveKey: string): bool
   return label.trim().toLowerCase() === objectiveKey.trim().toLowerCase();
 }
 
-export function isObjectiveLinearTerminalStatus(status: string | null | undefined): boolean {
-  return OBJECTIVE_LINEAR_TERMINAL_STATUSES.has(String(status ?? "").trim().toLowerCase());
+export function isObjectiveTrackerTerminalStatus(status: string | null | undefined): boolean {
+  return OBJECTIVE_TRACKER_TERMINAL_STATUSES.has(String(status ?? "").trim().toLowerCase());
 }
 
-export function filterObjectiveLinearIssuesForAction(
+export function filterObjectiveTrackerIssuesForAction(
   issues: LinearIssueSummary[],
   blockedIssueIds: Iterable<string> = []
 ): LinearIssueSummary[] {
@@ -30,11 +36,11 @@ export function filterObjectiveLinearIssuesForAction(
   );
 
   return issues.filter(
-    (issue) => !isObjectiveLinearTerminalStatus(issue.status) && !blocked.has(issue.id)
+    (issue) => !isObjectiveTrackerTerminalStatus(issue.status) && !blocked.has(issue.id)
   );
 }
 
-export async function listObjectiveLinearIssues(input: {
+export async function listObjectiveTrackerIssues(input: {
   projectId: string;
   objectiveKey: string;
   projectSlug?: string | null;

package/cloud-runtime/standalone/apps/local/lib/orchestrator/chat-processor.ts

@@ -3,12 +3,14 @@ import { createMultiplexedStream } from "@/lib/stream-multiplexer";
 import { loadDbParticipants } from "@/lib/agent-participants";
 import { killByThread, killByWorkspace } from "@/lib/agent-process-registry";
 import type { ChatRunPayload, ChatRunJobData } from "@/lib/orchestrator/chat-types";
+import type { ChatEvent } from "@/lib/types";
 import {
   appendChatRunStepActivity,
   completeChatRunStepActivity,
   loadChatRunActivity,
   updateChatRunActivity,
 } from "@/lib/orchestrator/chat-activities";
+import { getChatEventBus } from "@/lib/chat-event-bus";
 import { writeDebugLog } from "@/lib/debug-log";
 
 function isChatRunPayload(value: unknown): value is ChatRunPayload {
@@ -75,6 +77,8 @@ async function handleStart(job: Job<ChatRunJobData>): Promise<void> {
     lastError: null,
   });
 
+  const eventBus = getChatEventBus();
+
   try {
     writeDebugLog("chat-processor.start.running", {
       chatRunId: chatRun.id,
@@ -121,10 +125,39 @@ async function handleStart(job: Job<ChatRunJobData>): Promise<void> {
       chatRunId: chatRun.id,
       participantCount: participants.length,
     });
+    let sseBuffer = "";
     while (true) {
-      const { done } = await reader.read();
+      const { done, value } = await reader.read();
       if (done) break;
+      if (value) {
+        sseBuffer += new TextDecoder().decode(value);
+        const lines = sseBuffer.split("\n");
+        // Keep the last incomplete line in the buffer
+        sseBuffer = lines.pop() || "";
+        for (const line of lines) {
+          const trimmed = line.trim();
+          if (!trimmed.startsWith("data: ")) continue;
+          try {
+            const event: ChatEvent = JSON.parse(trimmed.slice(6));
+            if (event.type !== "log") {
+              eventBus.publish(chatRun.id, event);
+            }
+          } catch {
+            // malformed SSE line — ignore
+          }
+        }
+      }
+    }
+    // Process any remaining buffer
+    if (sseBuffer.trim().startsWith("data: ")) {
+      try {
+        const event: ChatEvent = JSON.parse(sseBuffer.trim().slice(6));
+        if (event.type !== "log") {
+          eventBus.publish(chatRun.id, event);
+        }
+      } catch { /* ignore */ }
     }
+    eventBus.complete(chatRun.id);
     writeDebugLog("chat-processor.stream.complete", {
       chatRunId: chatRun.id,
     });
@@ -152,6 +185,7 @@ async function handleStart(job: Job<ChatRunJobData>): Promise<void> {
     });
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
+    eventBus.complete(chatRun.id);
     writeDebugLog("chat-processor.error", {
       chatRunId: chatRun.id,
       error,

package/cloud-runtime/standalone/apps/local/lib/sqlite-query-adapter.ts

@@ -462,6 +462,38 @@ function runMigrations(db: DatabaseSync): void {
     }
   }
 
+  // ── Task Groups ────────────────────────────────────────────────────────────
+  db.exec(`
+    CREATE TABLE IF NOT EXISTS task_groups (
+      id TEXT NOT NULL PRIMARY KEY DEFAULT (lower(hex(randomblob(16)))),
+      project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+      name TEXT NOT NULL DEFAULT 'Untitled',
+      position INTEGER NOT NULL DEFAULT 0,
+      collapsed INTEGER NOT NULL DEFAULT 0,
+      created_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ','now')),
+      updated_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ','now'))
+    );
+    CREATE INDEX IF NOT EXISTS idx_task_groups_project_id ON task_groups (project_id);
+
+    CREATE TRIGGER IF NOT EXISTS task_groups_updated_at
+      AFTER UPDATE ON task_groups
+      FOR EACH ROW
+      BEGIN
+        UPDATE task_groups SET updated_at = strftime('%Y-%m-%dT%H:%M:%fZ','now') WHERE rowid = NEW.rowid;
+      END;
+  `);
+
+  // Join table for group ↔ item membership (works with any external ID, e.g. Linear issues)
+  db.exec(`
+    CREATE TABLE IF NOT EXISTS task_group_items (
+      group_id TEXT NOT NULL REFERENCES task_groups(id) ON DELETE CASCADE,
+      item_id  TEXT NOT NULL,
+      position INTEGER NOT NULL DEFAULT 0,
+      PRIMARY KEY (group_id, item_id)
+    );
+    CREATE INDEX IF NOT EXISTS idx_task_group_items_item ON task_group_items (item_id);
+  `);
+
   backfillAgentsFromFilesystem(db);
   backfillAgentSkillsFromLegacyParticipants(db);
   ensureRuntimeArtifactsForDbAgents(db);

package/cloud-runtime/standalone/apps/local/lib/stream-multiplexer.ts

@@ -724,6 +724,7 @@ async function runAgent(
 }> {
   let fullResponse = "";
 
+  write({ type: "participant-start", participantId: p.id });
   write({ type: "participant-thinking", participantId: p.id });
 
   // Resolve identity from the canonical DB-backed fields only.
@@ -842,6 +843,10 @@ Resolved memory entries: ${executionProvenance.memory.map((entry) => `${entry.so
       onDelta: (delta) => {
         fullResponse += delta;
         updateProcess(workspaceId, p.id, { lastActivity: Date.now() });
+        write({ type: "text-delta", participantId: p.id, delta });
+      },
+      onThought: (content) => {
+        write({ type: "participant-thought", participantId: p.id, content });
       },
     });
     updateProcess(workspaceId, p.id, { state: "done", lastActivity: Date.now() });

package/cloud-runtime/standalone/apps/local/lib/task-groups-db.ts

@@ -0,0 +1,130 @@
+import type { DatabaseSync } from "node:sqlite";
+
+export interface TaskGroup {
+  id: string;
+  project_id: string;
+  name: string;
+  position: number;
+  collapsed: number;
+  created_at: string;
+  updated_at: string;
+}
+
+export interface TaskGroupWithTaskIds extends TaskGroup {
+  task_ids: string[];
+}
+
+export function createTaskGroup(
+  db: DatabaseSync,
+  opts: { projectId: string; name: string; taskIds?: string[] },
+): TaskGroupWithTaskIds {
+  const id = generateId();
+  const { projectId, name, taskIds = [] } = opts;
+
+  const maxPos = db
+    .prepare("SELECT COALESCE(MAX(position), -1) AS max_pos FROM task_groups WHERE project_id = ?")
+    .get(projectId) as { max_pos: number };
+
+  db.prepare(
+    "INSERT INTO task_groups (id, project_id, name, position) VALUES (?, ?, ?, ?)",
+  ).run(id, projectId, name, maxPos.max_pos + 1);
+
+  if (taskIds.length > 0) {
+    const stmt = db.prepare(
+      "INSERT OR IGNORE INTO task_group_items (group_id, item_id, position) VALUES (?, ?, ?)",
+    );
+    taskIds.forEach((taskId, idx) => {
+      stmt.run(id, taskId, idx);
+    });
+  }
+
+  const group = db.prepare("SELECT * FROM task_groups WHERE id = ?").get(id) as unknown as TaskGroup;
+  return { ...group, task_ids: taskIds };
+}
+
+export function listTaskGroups(
+  db: DatabaseSync,
+  projectId: string,
+): TaskGroupWithTaskIds[] {
+  const groups = db
+    .prepare("SELECT * FROM task_groups WHERE project_id = ? ORDER BY position ASC")
+    .all(projectId) as unknown as TaskGroup[];
+
+  return groups.map((group) => {
+    const items = db
+      .prepare("SELECT item_id FROM task_group_items WHERE group_id = ? ORDER BY position ASC")
+      .all(group.id) as unknown as { item_id: string }[];
+    return { ...group, task_ids: items.map((t) => t.item_id) };
+  });
+}
+
+export function updateTaskGroup(
+  db: DatabaseSync,
+  groupId: string,
+  updates: Partial<Pick<TaskGroup, "name" | "position" | "collapsed">>,
+): TaskGroup {
+  const setClauses: string[] = [];
+  const values: (string | number)[] = [];
+
+  if (updates.name !== undefined) {
+    setClauses.push("name = ?");
+    values.push(updates.name);
+  }
+  if (updates.position !== undefined) {
+    setClauses.push("position = ?");
+    values.push(updates.position);
+  }
+  if (updates.collapsed !== undefined) {
+    setClauses.push("collapsed = ?");
+    values.push(updates.collapsed);
+  }
+
+  if (setClauses.length > 0) {
+    values.push(groupId);
+    db.prepare(`UPDATE task_groups SET ${setClauses.join(", ")} WHERE id = ?`).run(
+      ...values,
+    );
+  }
+
+  return db.prepare("SELECT * FROM task_groups WHERE id = ?").get(groupId) as unknown as TaskGroup;
+}
+
+export function deleteTaskGroup(db: DatabaseSync, groupId: string): void {
+  db.prepare("DELETE FROM task_group_items WHERE group_id = ?").run(groupId);
+  db.prepare("DELETE FROM task_groups WHERE id = ?").run(groupId);
+}
+
+export function assignTasksToGroup(
+  db: DatabaseSync,
+  groupId: string,
+  taskIds: string[],
+): void {
+  const maxPos = db
+    .prepare(
+      "SELECT COALESCE(MAX(position), -1) AS max_pos FROM task_group_items WHERE group_id = ?",
+    )
+    .get(groupId) as { max_pos: number };
+
+  const stmt = db.prepare(
+    "INSERT OR IGNORE INTO task_group_items (group_id, item_id, position) VALUES (?, ?, ?)",
+  );
+  taskIds.forEach((taskId, idx) => {
+    stmt.run(groupId, taskId, maxPos.max_pos + 1 + idx);
+  });
+}
+
+export function removeTaskFromGroup(
+  db: DatabaseSync,
+  groupId: string,
+  taskId: string,
+): void {
+  db.prepare(
+    "DELETE FROM task_group_items WHERE group_id = ? AND item_id = ?",
+  ).run(groupId, taskId);
+}
+
+function generateId(): string {
+  const bytes = new Uint8Array(16);
+  globalThis.crypto.getRandomValues(bytes);
+  return Array.from(bytes, (b) => b.toString(16).padStart(2, "0")).join("");
+}

package/cloud-runtime/standalone/apps/local/lib/tiptap/composer-mentions.ts

@@ -152,10 +152,10 @@ export const ThreadMention = Node.create({
 });
 
 /**
- * Linear issue mention pill — renders @ENG-123.
+ * Tracker item mention pill — renders @ENG-123.
  */
-export const LinearIssueMention = Node.create({
-  name: "linearIssueMention",
+export const TrackerItemMention = Node.create({
+  name: "trackerItemMention",
   group: "inline",
   inline: true,
   atom: true,
@@ -171,14 +171,14 @@ export const LinearIssueMention = Node.create({
   },
 
   parseHTML() {
-    return [{ tag: 'span[data-linear-issue-mention]' }];
+    return [{ tag: 'span[data-tracker-item-mention]' }];
   },
 
   renderHTML({ node, HTMLAttributes }) {
     return [
       "span",
       mergeAttributes(HTMLAttributes, {
-        "data-linear-issue-mention": "",
+        "data-tracker-item-mention": "",
         "data-id": node.attrs.id,
         "data-identifier": node.attrs.identifier,
         class: "composer-pill composer-pill--linear-issue",

package/cloud-runtime/standalone/apps/local/lib/tiptap/editor-utils.ts

@@ -98,7 +98,7 @@ function getAtomDisplayText(node: { type: { name: string }; attrs: Record<string
       const shortId = (node.attrs.threadId ?? "").slice(0, 8);
       return `@#${shortId}`;
     }
-    case "linearIssueMention": {
+    case "trackerItemMention": {
       const identifier = String(node.attrs.identifier ?? "").trim();
       return identifier ? `@${identifier}` : "";
     }

package/cloud-runtime/standalone/apps/local/lib/tiptap/serialize-composer.ts

@@ -7,7 +7,7 @@ import type { JSONContent } from "@tiptap/core";
  * - projectMention → @~project:slug
  * - fileMention → @/path or @~/path
  * - threadMention → @#shortId
- * - linearIssueMention → @ABC-123
+ * - trackerItemMention → @ABC-123
  * - text nodes → verbatim text
  * - paragraphs separated by \n
  */
@@ -71,7 +71,7 @@ function serializeParagraph(node: JSONContent): string {
         break;
       }
 
-      case "linearIssueMention": {
+      case "trackerItemMention": {
         const attrs = child.attrs ?? {};
         const identifier = String(attrs.identifier ?? "").trim();
         if (identifier) {