@cuongtran001/kanna 1.2.0 → 1.3.1

package/src/server/claude-loop-orch-commands.ts

@@ -0,0 +1,466 @@
+/**
+ * Standalone loop-orchestration command handlers for AgentCoordinator.
+ *
+ * Extracted from agent.ts so the 11 related private/public methods live in
+ * their own testable module. The coordinator delegates to these functions by
+ * passing an object literal that satisfies `LoopOrchCommandDeps`.
+ *
+ * Side-effect seal: this module contains NO direct IO (no node:fs, no HTTP
+ * calls, no Bun primitives). Every effectful operation is injected through
+ * the deps interface — including `ensureTrackingFile` which is the sole IO
+ * operation used by `setupLoop`.
+ */
+
+import type { TranscriptEntry } from "../shared/types"
+import type { Subagent, AgentProvider } from "../shared/types"
+import { AUTO_CONTINUE_EVENT_VERSION, type AutoContinueEvent } from "./auto-continue/events"
+import { deriveChatSchedules, deriveLoopState, type LoopState } from "./auto-continue/read-model"
+import { timestamped } from "./claude-message-normalizer"
+import { buildTaskNotification } from "./claude-session-config"
+import { validateLoopSetup, reconcileTrackingFile, type LoopSetupInput } from "./loop-template"
+import type { EnsureTrackingFileArgs, EnsureTrackingFileResult } from "./loop-template-io.adapter"
+import { validateOrchRun, toOrchRunDetail, type OrchRunContext } from "./orchestration-input"
+import type { OrchRunDetail, OrchRunInput, OrchRunConfig, OrchTaskSpec, OrchRunSnapshot } from "../shared/orchestration-types"
+import type { WorkerSpawnArgs, WorkerResult } from "./orchestration-queue"
+import type { BackgroundRunOutcome, ProviderRunStart } from "./subagent-orchestrator"
+import type { ClaudeSessionState } from "./claude-session-state"
+import type { SetupLoopHandlerResult } from "./kanna-mcp"
+import { log } from "../shared/log"
+
+// ---------------------------------------------------------------------------
+// Structural sub-interfaces — only the operations this module calls.
+// ---------------------------------------------------------------------------
+
+/** Subset of EventStore used by these handlers. */
+interface LoopOrchCommandStore {
+  getOrchRun(runId: string): OrchRunSnapshot | null
+  getChat(chatId: string): { id: string; projectId: string } | null
+  getProject(projectId: string): { localPath: string; id: string } | null
+  getAutoContinueEvents(chatId: string): AutoContinueEvent[]
+  setSessionTokenForProvider(chatId: string, provider: AgentProvider, token: string | null): Promise<void>
+  appendMessage(chatId: string, entry: TranscriptEntry): Promise<void>
+}
+
+/** Subset of OrchestrationQueue used by these handlers. */
+interface LoopOrchCommandOrchQueue {
+  createRun(config: OrchRunConfig, tasks: OrchTaskSpec[]): Promise<string>
+  cancelRun(runId: string): Promise<void>
+}
+
+// ---------------------------------------------------------------------------
+// Dependency bundle injected by AgentCoordinator
+// ---------------------------------------------------------------------------
+
+export interface LoopOrchCommandDeps {
+  /** EventStore — subset used by these handlers. */
+  store: LoopOrchCommandStore
+
+  /** OrchestrationQueue — subset used by these handlers. */
+  orchestrationQueue: LoopOrchCommandOrchQueue
+
+  /** Live Claude sessions map owned by the coordinator (read-only). */
+  claudeSessions: Pick<Map<string, ClaudeSessionState>, "get">
+
+  /** Active turns map — `.has()` to check existence. */
+  activeTurns: Pick<Map<string, unknown>, "has">
+
+  /** Returns all configured subagents. */
+  getSubagents(): Subagent[]
+
+  /** Returns the current app settings snapshot for subagentRuntime config. */
+  getAppSettingsSnapshot(): {
+    subagentRuntime?: {
+      defaultOrchSubagentId?: string | null
+      defaultLoopSubagentId?: string | null
+    } | null
+  }
+
+  /**
+   * Builds a provider-run start bundle for an orchestration worker subagent.
+   * Delegates to `AgentCoordinator.buildSubagentProviderRunForChat`.
+   */
+  buildSubagentProviderRunForChat(args: {
+    subagent: Subagent
+    chatId: string
+    primer: string | null
+    userInstruction: string | null
+    runId: string
+    abortSignal: AbortSignal
+    depth: number
+    ancestorSubagentIds: string[]
+    parentUserMessageId: string
+    cwdOverride?: string
+  }): ProviderRunStart
+
+  /**
+   * Tears down the given Claude session. No permit semantics here — caller
+   * must ensure the session is in a stable state before calling.
+   */
+  closeClaudeSession(chatId: string, session: ClaudeSessionState): void
+
+  /**
+   * Appends an AutoContinueEvent to the store, notifies the schedule manager,
+   * and emits a state-change event for the chat.
+   */
+  emitAutoContinueEvent(event: AutoContinueEvent): Promise<void>
+
+  /**
+   * IO adapter: create or reconcile the loop tracking file on disk. Injected
+   * so this module stays IO-free. See `loop-template-io.adapter.ts`.
+   */
+  ensureTrackingFile(args: EnsureTrackingFileArgs): Promise<EnsureTrackingFileResult>
+
+  /**
+   * Returns the current armed-loop state for a chat, or null if disarmed.
+   * Injected (rather than calling the module-level `isLoopArmed` fn) so
+   * AgentCoordinator can be monkey-patched in tests via
+   * `coordinator.isLoopArmed = () => ({...})`.
+   */
+  isLoopArmed(chatId: string): LoopState | null
+}
+
+// ---------------------------------------------------------------------------
+// Exported standalone functions
+// ---------------------------------------------------------------------------
+
+// ── Orchestration commands ──────────────────────────────────────────────────
+
+/**
+ * Spawn a single orchestration phase worker. Looks up the run's persisted
+ * config for origin chat + subagent, then delegates to the coordinator's
+ * subagent provider-run machinery. Origin chat + subagent are read from the
+ * persisted run config so this resolves identically on a fresh run and after
+ * a restart.
+ */
+export async function buildOrchWorker(
+  deps: LoopOrchCommandDeps,
+  spawn: WorkerSpawnArgs,
+): Promise<WorkerResult> {
+  const run = deps.store.getOrchRun(spawn.runId)
+  const chatId = run?.config.originChatId
+  const subagentId = run?.config.workerSubagentId
+  if (!chatId || !subagentId) {
+    return { kind: "failed", error: "orchestration run missing originChatId / workerSubagentId" }
+  }
+  const subagent = deps.getSubagents().find((s) => s.id === subagentId)
+  if (!subagent) return { kind: "failed", error: `orchestration worker subagent "${subagentId}" not found` }
+  if (!deps.store.getChat(chatId)) return { kind: "failed", error: `orchestration origin chat ${chatId} not found` }
+
+  const providerRun = deps.buildSubagentProviderRunForChat({
+    subagent,
+    chatId,
+    primer: null,
+    userInstruction: spawn.prompt,
+    runId: `${spawn.runId}:${spawn.workerId}`,
+    abortSignal: spawn.abortSignal,
+    depth: 0,
+    ancestorSubagentIds: [],
+    parentUserMessageId: spawn.runId,
+    cwdOverride: spawn.cwd,
+  })
+  try {
+    if (!(await providerRun.authReady())) {
+      return { kind: "failed", error: "orchestration worker auth not ready" }
+    }
+    const result = await providerRun.start(() => undefined, () => undefined)
+    return { kind: "completed", text: result.text }
+  } catch (err) {
+    if (spawn.abortSignal.aborted) return { kind: "failed", error: "aborted" }
+    return { kind: "failed", error: err instanceof Error ? err.message : String(err) }
+  }
+}
+
+/**
+ * Derive chat/project context for orchestration validation. Returns null when
+ * the chat or project is not found.
+ */
+export function buildOrchRunContext(
+  deps: LoopOrchCommandDeps,
+  chatId: string,
+): OrchRunContext | null {
+  const chat = deps.store.getChat(chatId)
+  if (!chat) return null
+  const project = deps.store.getProject(chat.projectId)
+  if (!project) return null
+  return {
+    chatId,
+    repoRoot: project.localPath,
+    roster: deps.getSubagents().map((s) => ({ id: s.id, name: s.name })),
+    defaultOrchSubagentId: deps.getAppSettingsSnapshot().subagentRuntime?.defaultOrchSubagentId ?? null,
+  }
+}
+
+/**
+ * User-callable entry point (MCP `orch_run` + ws `orch.run`). Validates the
+ * task list into the fixed linear config, then starts the run. Returns the
+ * runId or the flat validation error list — never a partial run.
+ */
+export async function runOrchestration(
+  deps: LoopOrchCommandDeps,
+  chatId: string,
+  input: OrchRunInput,
+): Promise<{ ok: true; runId: string } | { ok: false; errors: string[] }> {
+  const context = buildOrchRunContext(deps, chatId)
+  if (!context) return { ok: false, errors: [`chat ${chatId} not found or has no project`] }
+  const validation = validateOrchRun(input, context)
+  if (!validation.ok) return { ok: false, errors: validation.errors }
+  const runId = await deps.orchestrationQueue.createRun(
+    validation.resolved.config,
+    validation.resolved.tasks,
+  )
+  return { ok: true, runId }
+}
+
+/** Cancel a run (MCP `orch_cancel_run` + ws `orch.cancelRun`). */
+export async function cancelOrchRun(
+  deps: LoopOrchCommandDeps,
+  runId: string,
+): Promise<void> {
+  await deps.orchestrationQueue.cancelRun(runId)
+}
+
+/** Canonical run detail DTO (MCP `orch_run_status` + ws `orch.getRun`). */
+export function getOrchRunDetail(
+  deps: LoopOrchCommandDeps,
+  runId: string,
+): OrchRunDetail | null {
+  const snapshot = deps.store.getOrchRun(runId)
+  return snapshot ? toOrchRunDetail(snapshot) : null
+}
+
+// ── Loop + background delivery ──────────────────────────────────────────────
+
+/**
+ * Wipe the main-agent's Claude session context (the /clear equivalent).
+ *
+ * Three things happen:
+ * - The persisted session token is set to null (so the next spawn starts fresh).
+ * - The live session's `suppressSessionTokenPersist` flag is set to true so the
+ *   in-flight stream cannot overwrite the null we just wrote (avoids a
+ *   121 ms race on setup_loop /clear).
+ * - An idle warm SDK session is torn down so it cannot be reused in-band by
+ *   the next turn (which would make the /clear a no-op).
+ */
+export async function clearClaudeSessionContext(
+  deps: LoopOrchCommandDeps,
+  chatId: string,
+): Promise<void> {
+  await deps.store.setSessionTokenForProvider(chatId, "claude", null)
+  const session = deps.claudeSessions.get(chatId)
+  if (!session) return
+  session.suppressSessionTokenPersist = true
+  if (!deps.activeTurns.has(chatId)) {
+    deps.closeClaudeSession(chatId, session)
+  }
+}
+
+/**
+ * Deliver a finished `run_in_background` subagent's result back into the
+ * main chat as a fresh turn AND clear the main-agent's Claude session so the
+ * next turn starts with a fresh context window. Wired as the orchestrator's
+ * `onBackgroundRunComplete` hook.
+ *
+ * Loop-orchestration invariant: main is stateless-in-context / stateful-in-file.
+ * PROGRESS.md is the durability contract; every delivery re-reads it. Subagent
+ * output is NOT carried forward as prompt content — the subagent is expected
+ * to have written its findings into PROGRESS.md before terminating.
+ *
+ * See adr-20260711-notification-driven-loop-orchestration.
+ */
+export async function deliverSubagentToMain(
+  deps: LoopOrchCommandDeps,
+  chatId: string,
+  runId: string,
+  outcome: BackgroundRunOutcome,
+): Promise<void> {
+  if (!deps.store.getChat(chatId)) return
+
+  // Structured re-entry: the completion is delivered as the same
+  // <task-notification> XML Claude Code's own background agents use
+  // (LocalAgentTask), so the model parses task identity/status with the
+  // format it already knows from native training.
+  //
+  // When a loop is armed, the FULL loop discipline prompt follows the
+  // notification on every wake — not a generic "decide next action" string,
+  // which drifted into self-implementation (the 7.5h marathon-turn bug).
+  // Armed notifications carry NO <result> body: PROGRESS.md is the loop's
+  // only durability contract. Non-loop deliveries include the (truncated)
+  // result since ad-hoc background delegations have no tracking file.
+  const armed = isLoopArmed(deps, chatId)
+  const notification = buildTaskNotification(runId, outcome, { includeResult: !armed })
+  let prompt: string
+  if (armed) {
+    prompt = `${notification}\n\n${armed.prompt}`
+  } else if (outcome.status === "completed") {
+    prompt = `${notification}\n\nYour Claude context has been cleared. Read PROGRESS.md if present, then decide the next action.`
+  } else {
+    prompt = `${notification}\n\nYour Claude context has been cleared. Read PROGRESS.md if present; decide whether to retry, try another approach, or stop.`
+  }
+
+  try {
+    // Wipe the main-agent's Claude session token so the next spawn starts
+    // fresh (the /clear equivalent). Codex path is unaffected.
+    await clearClaudeSessionContext(deps, chatId)
+    await deps.store.appendMessage(chatId, timestamped({ kind: "context_cleared" }))
+
+    const now = Date.now()
+    const scheduleId = crypto.randomUUID()
+    await deps.emitAutoContinueEvent({
+      v: AUTO_CONTINUE_EVENT_VERSION,
+      kind: "auto_continue_accepted",
+      timestamp: now,
+      chatId,
+      scheduleId,
+      scheduledAt: now,
+      tz: "system",
+      source: "subagent_background",
+      resetAt: now,
+      detectedAt: now,
+      prompt,
+    })
+  } catch (err) {
+    log.warn(`[kanna] deliverSubagentToMain failed`, { chatId, runId, err })
+  }
+}
+
+/**
+ * Arm an autonomous loop on the main chat. Validates the loop spec, ensures
+ * the tracking file exists (writes a skeleton if absent), then /clears the
+ * main-agent Claude session and enqueues the templated recurring prompt so
+ * the next turn starts the loop. Backs `mcp__kanna__setup_loop`. See
+ * adr-20260711-setup-loop-template.
+ */
+export async function setupLoop(
+  deps: LoopOrchCommandDeps,
+  args: {
+    chatId: string
+    input: LoopSetupInput
+  },
+): Promise<SetupLoopHandlerResult> {
+  const chat = deps.store.getChat(args.chatId)
+  if (!chat) return { ok: false, errors: [`chat ${args.chatId} not found`] }
+  const project = deps.store.getProject(chat.projectId)
+  if (!project) return { ok: false, errors: [`project ${chat.projectId} not found`] }
+
+  const validation = validateLoopSetup(args.input, project.localPath, {
+    roster: deps.getSubagents().map((s) => ({ id: s.id, name: s.name })),
+    defaultLoopSubagentId: deps.getAppSettingsSnapshot().subagentRuntime?.defaultLoopSubagentId ?? null,
+  })
+  if (!validation.ok) return { ok: false, errors: validation.errors }
+
+  const resolved = validation.resolved
+  let created: boolean
+  let reconciled: boolean
+  let reconcileActions: string[]
+  try {
+    const ensureResult = await deps.ensureTrackingFile({
+      absPath: resolved.trackingFileAbs,
+      skeleton: resolved.skeleton,
+      // Deterministic schema reconcile of an EXISTING tracking file: pure
+      // string transform — server-owned sections rewritten to the inputs,
+      // loop history preserved. No model judgement involved.
+      reconcile: (existing) =>
+        reconcileTrackingFile(existing, {
+          goal: resolved.goal,
+          verifyCommand: resolved.verifyCommand,
+          chunkHint: resolved.chunkHint,
+        }),
+    })
+    created = ensureResult.created
+    reconciled = ensureResult.reconciled
+    reconcileActions = ensureResult.actions
+  } catch (err) {
+    return {
+      ok: false,
+      errors: [`ensureTrackingFile failed: ${err instanceof Error ? err.message : String(err)}`],
+    }
+  }
+
+  try {
+    // Wipe main-agent Claude session so the next turn starts fresh with the
+    // rendered loop prompt. Codex untouched. setup_loop runs from INSIDE a
+    // live turn, so the suppression half of clearClaudeSessionContext is
+    // what keeps the wipe from being overwritten by the in-flight stream.
+    await clearClaudeSessionContext(deps, args.chatId)
+    await deps.store.appendMessage(args.chatId, timestamped({ kind: "context_cleared" }))
+
+    const now = Date.now()
+    // Arm the loop durably: every subsequent background-completion wake
+    // re-injects THIS prompt (not the generic one) and loop turns are
+    // tool-blocked. Superseded by a later setup_loop or cleared by stop_loop
+    // / a real user send. Replays from the auto-continue log on restart.
+    await deps.emitAutoContinueEvent({
+      v: AUTO_CONTINUE_EVENT_VERSION,
+      kind: "loop_armed",
+      timestamp: now,
+      chatId: args.chatId,
+      scheduleId: crypto.randomUUID(),
+      subagentId: resolved.subagentId,
+      prompt: resolved.prompt,
+    })
+
+    const scheduleId = crypto.randomUUID()
+    await deps.emitAutoContinueEvent({
+      v: AUTO_CONTINUE_EVENT_VERSION,
+      kind: "auto_continue_accepted",
+      timestamp: now,
+      chatId: args.chatId,
+      scheduleId,
+      scheduledAt: now,
+      tz: "system",
+      source: "subagent_background",
+      resetAt: now,
+      detectedAt: now,
+      prompt: resolved.prompt,
+    })
+  } catch (err) {
+    return {
+      ok: false,
+      errors: [`enqueue failed: ${err instanceof Error ? err.message : String(err)}`],
+    }
+  }
+
+  return {
+    ok: true,
+    trackingFileRel: resolved.trackingFileRel,
+    created,
+    reconciled,
+    reconcileActions,
+    prompt: resolved.prompt,
+  }
+}
+
+/** Current armed-loop state for a chat, or null. Pure replay of the auto-continue log. */
+export function isLoopArmed(deps: LoopOrchCommandDeps, chatId: string): LoopState | null {
+  return deriveLoopState(deps.store.getAutoContinueEvents(chatId), chatId)
+}
+
+/**
+ * Disarm an armed loop (restores tools + stops prompt re-injection). Backs
+ * the `stop_loop` MCP tool (called by the model on GOAL MET) and the
+ * user-send takeover path. No-op when no loop is armed.
+ */
+export async function stopLoop(
+  deps: LoopOrchCommandDeps,
+  chatId: string,
+  reason: "goal_met" | "user_send" | "chat_deleted",
+): Promise<void> {
+  if (!isLoopArmed(deps, chatId)) return
+  await deps.emitAutoContinueEvent({
+    v: AUTO_CONTINUE_EVENT_VERSION,
+    kind: "loop_disarmed",
+    timestamp: Date.now(),
+    chatId,
+    scheduleId: crypto.randomUUID(),
+    reason,
+  })
+}
+
+/** Returns live schedule IDs (proposed or scheduled) for the given chat. */
+export function listLiveSchedules(deps: LoopOrchCommandDeps, chatId: string): string[] {
+  const { schedules } = deriveChatSchedules(deps.store.getAutoContinueEvents(chatId), chatId)
+  return Object.values(schedules)
+    .filter((s) => s.state === "proposed" || s.state === "scheduled")
+    .map((s) => s.scheduleId)
+    .sort()
+}

package/src/server/claude-message-normalizer.test.ts

@@ -0,0 +1,213 @@
+import { describe, expect, test } from "bun:test"
+import {
+  getClaudeAssistantMessageUsageId,
+  normalizeClaudeStreamMessage,
+  normalizeToolContent,
+  type ClaudeRawSdkMessage,
+} from "./claude-message-normalizer"
+
+// ---------------------------------------------------------------------------
+// getClaudeAssistantMessageUsageId
+// ---------------------------------------------------------------------------
+
+describe("getClaudeAssistantMessageUsageId", () => {
+  test("returns uuid when present on message", () => {
+    const msg: ClaudeRawSdkMessage = { uuid: "msg-uuid-123", type: "assistant" }
+    expect(getClaudeAssistantMessageUsageId(msg)).toBe("msg-uuid-123")
+  })
+
+  test("returns message.id when uuid is absent", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "assistant",
+      message: { id: "inner-id-456", content: [] },
+    }
+    expect(getClaudeAssistantMessageUsageId(msg)).toBe("inner-id-456")
+  })
+
+  test("returns null when neither uuid nor message.id are present", () => {
+    const msg: ClaudeRawSdkMessage = { type: "user" }
+    expect(getClaudeAssistantMessageUsageId(msg)).toBeNull()
+  })
+
+  test("prefers message.id over uuid when both are present", () => {
+    const msg: ClaudeRawSdkMessage = {
+      uuid: "outer",
+      type: "assistant",
+      message: { id: "inner", content: [] },
+    }
+    // message.id is checked first
+    expect(getClaudeAssistantMessageUsageId(msg)).toBe("inner")
+  })
+})
+
+// ---------------------------------------------------------------------------
+// normalizeToolContent
+// ---------------------------------------------------------------------------
+
+describe("normalizeToolContent", () => {
+  test("returns null for null input", () => {
+    expect(normalizeToolContent(null)).toBeNull()
+  })
+
+  test("returns null for undefined input", () => {
+    expect(normalizeToolContent(undefined)).toBeNull()
+  })
+
+  test("returns string unchanged", () => {
+    expect(normalizeToolContent("hello")).toBe("hello")
+  })
+
+  test("returns object unchanged", () => {
+    const obj = { a: 1 }
+    expect(normalizeToolContent(obj)).toEqual({ a: 1 })
+  })
+
+  test("returns array unchanged", () => {
+    const arr = [1, 2, 3]
+    expect(normalizeToolContent(arr)).toEqual([1, 2, 3])
+  })
+})
+
+// ---------------------------------------------------------------------------
+// normalizeClaudeStreamMessage
+// ---------------------------------------------------------------------------
+
+describe("normalizeClaudeStreamMessage", () => {
+  test("returns [] for unrecognized message type", () => {
+    const msg: ClaudeRawSdkMessage = { type: "unknown_type" }
+    expect(normalizeClaudeStreamMessage(msg)).toEqual([])
+  })
+
+  test("normalizes assistant text block to assistant_text entry", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "assistant",
+      message: {
+        role: "assistant",
+        content: [{ type: "text", text: "Hello, world!" }],
+      },
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("assistant_text")
+    expect((entries[0] as { text: string }).text).toBe("Hello, world!")
+  })
+
+  test("normalizes assistant tool_use block to tool_call entry", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "assistant",
+      message: {
+        role: "assistant",
+        content: [
+          { type: "tool_use", name: "Bash", id: "tool-123", input: { command: "ls" } },
+        ],
+      },
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("tool_call")
+    const tc = entries[0] as { tool: { toolName: string; toolId: string } }
+    expect(tc.tool.toolName).toBe("Bash")
+    expect(tc.tool.toolId).toBe("tool-123")
+  })
+
+  test("normalizes user tool_result block to tool_result entry", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "user",
+      message: {
+        role: "user",
+        content: [
+          { type: "tool_result", tool_use_id: "tool-123", content: "output text", is_error: false },
+        ],
+      },
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("tool_result")
+    const tr = entries[0] as { toolId: string; content: string; isError: boolean }
+    expect(tr.toolId).toBe("tool-123")
+    expect(tr.content).toBe("output text")
+    expect(tr.isError).toBe(false)
+  })
+
+  test("drops synthetic non-error placeholder ('No response requested.')", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "assistant",
+      isApiErrorMessage: false,
+      message: {
+        role: "assistant",
+        model: "<synthetic>",
+        content: [{ type: "text", text: "No response requested." }],
+      },
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(0)
+  })
+
+  test("normalizes result message to result entry", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "result",
+      subtype: "success",
+      is_error: false,
+      duration_ms: 1200,
+      result: "all good",
+      total_cost_usd: 0.005,
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("result")
+    const r = entries[0] as { subtype: string; durationMs: number; result: string; costUsd: number }
+    expect(r.subtype).toBe("success")
+    expect(r.durationMs).toBe(1200)
+    expect(r.result).toBe("all good")
+    expect(r.costUsd).toBeCloseTo(0.005)
+  })
+
+  test("normalizes api_error assistant message to api_error entry", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "assistant",
+      isApiErrorMessage: true,
+      apiErrorStatus: 529,
+      message: {
+        role: "assistant",
+        model: "<synthetic>",
+        content: [{ type: "text", text: "API Error: 529 Overloaded" }],
+      },
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("api_error")
+    const ae = entries[0] as { status: number }
+    expect(ae.status).toBe(529)
+  })
+
+  test("normalizes cancelled result to interrupted entry", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "result",
+      subtype: "cancelled",
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("interrupted")
+  })
+
+  test("normalizes system_init message with tools and agents", () => {
+    const msg: ClaudeRawSdkMessage = {
+      type: "system",
+      subtype: "init",
+      model: "claude-opus-4-5",
+      tools: ["Bash", "Read"],
+      agents: ["my-agent"],
+      slash_commands: ["help", "._internal"],
+      mcp_servers: [],
+    }
+    const entries = normalizeClaudeStreamMessage(msg)
+    expect(entries).toHaveLength(1)
+    expect(entries[0].kind).toBe("system_init")
+    const si = entries[0] as { model: string; tools: string[]; agents: string[]; slashCommands: string[] }
+    expect(si.model).toBe("claude-opus-4-5")
+    expect(si.tools).toEqual(["Bash", "Read"])
+    expect(si.agents).toEqual(["my-agent"])
+    // Internal slash commands prefixed with '._' are filtered
+    expect(si.slashCommands).toEqual(["help"])
+  })
+})