takomi 2.1.2 → 2.1.3

package/.pi/extensions/takomi-subagents/live-updates.ts

@@ -1,83 +1,83 @@
-import type { TakomiThinkingLevel } from "../../../src/pi-takomi-core";
-import type { TakomiSubagentRuntimeEvent } from "../takomi-runtime/subagent-types";
-import type { TakomiDispatchResult } from "./dispatch";
-
-type ToolUpdate = (partial: {
-  content: Array<{ type: "text"; text: string }>;
-  details: Record<string, unknown>;
-}) => void;
-
-type LiveTask = {
-  agent: string;
-  task: string;
-  workflow?: string;
-  model?: string;
-  thinking?: TakomiThinkingLevel;
-  conversationId?: string;
-};
-
-function appendLine(value: string, line: string): string {
-  const next = [value, line.trim()].filter(Boolean).join("\n");
-  return next.split(/\r?\n/).slice(-12).join("\n");
-}
-
-export function createTakomiLiveUpdateBridge(
-  tasks: LiveTask[],
-  mode: "single" | "parallel" | "chain",
-  agentScope: string,
-  onUpdate: ToolUpdate | undefined,
-) {
-  const results: TakomiDispatchResult[] = tasks.map((task, index) => ({
-    agent: task.agent,
-    task: task.task,
-    workflow: task.workflow,
-    model: task.model,
-    thinking: task.thinking,
-    conversationId: task.conversationId ?? `pending-${index + 1}`,
-    code: -1,
-    output: "Queued.",
-    stderr: "",
-    preflight: "",
-  }));
-
-  const emit = () => {
-    const done = results.filter((result) => result.code === 0).length;
-    const running = results.filter((result) => result.code === -1).length;
-    onUpdate?.({
-      content: [{ type: "text", text: `Takomi ${mode}: ${done}/${results.length} done, ${running} running` }],
-      details: { results: [...results], mode, agentScope },
-    });
-  };
-
-  return {
-    results,
-    event(index: number, event: TakomiSubagentRuntimeEvent): void {
-      const current = results[index];
-      if (!current) return;
-      if (event.type === "start") {
-        current.conversationId = event.state.conversationId ?? current.conversationId;
-        current.thinking = event.state.thinking ?? current.thinking;
-        current.output = event.state.summary ?? "Starting.";
-      } else if (event.type === "update") {
-        current.model = event.patch.model ?? current.model;
-        current.thinking = event.patch.thinking ?? current.thinking;
-        current.output = event.patch.outputText ?? event.patch.summary ?? current.output;
-        if (event.patch.logs?.length) current.output = appendLine(current.output, event.patch.logs.join("\n"));
-      } else if (event.type === "appendLog") {
-        current.output = appendLine(current.output, event.chunk);
-      } else if (event.type === "complete") {
-        current.code = 0;
-        current.output = event.patch?.outputText ?? event.patch?.summary ?? current.output;
-      } else if (event.type === "block") {
-        current.code = 1;
-        current.output = event.patch?.outputText ?? event.patch?.summary ?? current.output;
-        if (event.patch?.logs?.length) current.stderr = event.patch.logs.join("\n");
-      }
-      emit();
-    },
-    finish(index: number, result: TakomiDispatchResult): void {
-      results[index] = result;
-      emit();
-    },
-  };
-}
+import type { TakomiThinkingLevel } from "../../../src/pi-takomi-core";
+import type { TakomiSubagentRuntimeEvent } from "../takomi-runtime/subagent-types";
+import type { TakomiDispatchResult } from "./dispatch";
+
+type ToolUpdate = (partial: {
+  content: Array<{ type: "text"; text: string }>;
+  details: Record<string, unknown>;
+}) => void;
+
+type LiveTask = {
+  agent: string;
+  task: string;
+  workflow?: string;
+  model?: string;
+  thinking?: TakomiThinkingLevel;
+  conversationId?: string;
+};
+
+function appendLine(value: string, line: string): string {
+  const next = [value, line.trim()].filter(Boolean).join("\n");
+  return next.split(/\r?\n/).slice(-12).join("\n");
+}
+
+export function createTakomiLiveUpdateBridge(
+  tasks: LiveTask[],
+  mode: "single" | "parallel" | "chain",
+  agentScope: string,
+  onUpdate: ToolUpdate | undefined,
+) {
+  const results: TakomiDispatchResult[] = tasks.map((task, index) => ({
+    agent: task.agent,
+    task: task.task,
+    workflow: task.workflow,
+    model: task.model,
+    thinking: task.thinking,
+    conversationId: task.conversationId ?? `pending-${index + 1}`,
+    code: -1,
+    output: "Queued.",
+    stderr: "",
+    preflight: "",
+  }));
+
+  const emit = () => {
+    const done = results.filter((result) => result.code === 0).length;
+    const running = results.filter((result) => result.code === -1).length;
+    onUpdate?.({
+      content: [{ type: "text", text: `Takomi ${mode}: ${done}/${results.length} done, ${running} running` }],
+      details: { results: [...results], mode, agentScope },
+    });
+  };
+
+  return {
+    results,
+    event(index: number, event: TakomiSubagentRuntimeEvent): void {
+      const current = results[index];
+      if (!current) return;
+      if (event.type === "start") {
+        current.conversationId = event.state.conversationId ?? current.conversationId;
+        current.thinking = event.state.thinking ?? current.thinking;
+        current.output = event.state.summary ?? "Starting.";
+      } else if (event.type === "update") {
+        current.model = event.patch.model ?? current.model;
+        current.thinking = event.patch.thinking ?? current.thinking;
+        current.output = event.patch.outputText ?? event.patch.summary ?? current.output;
+        if (event.patch.logs?.length) current.output = appendLine(current.output, event.patch.logs.join("\n"));
+      } else if (event.type === "appendLog") {
+        current.output = appendLine(current.output, event.chunk);
+      } else if (event.type === "complete") {
+        current.code = 0;
+        current.output = event.patch?.outputText ?? event.patch?.summary ?? current.output;
+      } else if (event.type === "block") {
+        current.code = 1;
+        current.output = event.patch?.outputText ?? event.patch?.summary ?? current.output;
+        if (event.patch?.logs?.length) current.stderr = event.patch.logs.join("\n");
+      }
+      emit();
+    },
+    finish(index: number, result: TakomiDispatchResult): void {
+      results[index] = result;
+      emit();
+    },
+  };
+}

package/.pi/extensions/takomi-subagents/native-render.ts

@@ -1,174 +1,174 @@
-import { renderSubagentResult, syncResultAnimation } from "pi-subagents/src/tui/render";
-import type { AgentToolResult } from "@mariozechner/pi-agent-core";
-import type { Details, SingleResult, AgentProgress } from "pi-subagents/src/shared/types";
-import type { Theme } from "@mariozechner/pi-coding-agent";
-import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
-import type { TakomiDispatchResult } from "./dispatch";
-import type { TakomiSubagentToolParams } from "./tool-runner";
-
-type ToolResult = {
-  content?: Array<{ type: string; text?: string }>;
-  details?: {
-    results?: TakomiDispatchResult[];
-    mode?: "single" | "parallel" | "chain";
-    agentScope?: string;
-    plan?: unknown;
-  };
-  isError?: boolean;
-};
-
-function taskList(params: TakomiSubagentToolParams): Array<{ agent: string; task: string }> {
-  if (params.chain?.length) return params.chain;
-  if (params.tasks?.length) return params.tasks;
-  if (params.agent || params.task) return [{ agent: params.agent ?? "...", task: params.task ?? "..." }];
-  return [];
-}
-
-export function renderTakomiSubagentCall(params: TakomiSubagentToolParams, theme: Theme) {
-  const tasks = taskList(params);
-  const mode = params.chain?.length ? "chain" : params.tasks?.length ? "parallel" : "single";
-  if (tasks.length === 1) {
-    return new Text(
-      `${theme.fg("toolTitle", theme.bold("takomi_subagent "))}${theme.fg("accent", tasks[0]?.agent || "?")}`,
-      0,
-      0,
-    );
-  }
-  return new Text(
-    `${theme.fg("toolTitle", theme.bold("takomi_subagent "))}${mode} (${tasks.length})`,
-    0,
-    0,
-  );
-}
-
-function parseTakomiOutput(outputText: string) {
-  const rawLines = outputText.split(/\r?\n/);
-  const textLines: string[] = [];
-  const recentTools: Array<{ tool: string; args: string; endMs: number }> = [];
-  let currentTool: string | undefined;
-  let currentToolArgs: string | undefined;
-
-  for (const line of rawLines) {
-    if (!line.trim()) continue;
-
-    // 1. Check for tool lifecycle markers
-    if (line.startsWith("Tool start: ")) {
-      currentTool = line.replace("Tool start: ", "").trim();
-      continue;
-    }
-    if (line.startsWith("Tool complete: ") || line.startsWith("Tool failed: ")) {
-      const toolName = line.replace(/Tool (complete|failed): /, "").trim();
-      recentTools.push({ tool: toolName, args: "", endMs: Date.now() });
-      if (currentTool === toolName) {
-        currentTool = undefined;
-        currentToolArgs = undefined;
-      }
-      continue;
-    }
-
-    // 2. Handle JSON blobs (tool calls)
-    const jsonStartIdx = line.indexOf('{"');
-    if (jsonStartIdx !== -1) {
-      const beforeJson = line.substring(0, jsonStartIdx).trim();
-      if (beforeJson) textLines.push(beforeJson);
-
-      const jsonPart = line.substring(jsonStartIdx);
-      try {
-        const parsed = JSON.parse(jsonPart);
-        const toolName = parsed.tool || (parsed.command ? parsed.command.split(" ")[0] : undefined);
-        if (toolName) {
-          const args = parsed.args ? JSON.stringify(parsed.args) : (parsed.command || "");
-          currentTool = toolName;
-          currentToolArgs = args;
-        }
-      } catch (e) {
-        if (jsonPart.trim().length > 1) {
-          textLines.push(jsonPart);
-        }
-      }
-      continue;
-    }
-
-    // 3. Skip lone brackets
-    const trimmed = line.trim();
-    if (trimmed === "{" || trimmed === "}") continue;
-
-    // 4. Everything else is text (thoughts)
-    textLines.push(line);
-  }
-
-  return { textLines, recentTools, currentTool, currentToolArgs };
-}
-
-export function renderTakomiSubagentResult(result: ToolResult, options: { expanded?: boolean; isPartial?: boolean }, theme: Theme, context: any) {
-  const details = result.details;
-  const results = details?.results ?? [];
-
-  if (results.length === 0) {
-    const text = result.content?.find((item) => item.type === "text")?.text ?? "(no output)";
-    return new Text(text, 0, 0);
-  }
-
-  const mappedResults: SingleResult[] = results.map((r, i) => {
-    const isRunning = r.code === -1;
-    const outputText = r.output || r.stderr || "";
-    const { textLines, recentTools, currentTool, currentToolArgs } = parseTakomiOutput(outputText);
-    
-    const recentOutput = textLines.slice(-5);
-
-    const progress: AgentProgress | undefined = isRunning ? {
-      index: i,
-      agent: r.agent,
-      status: "running",
-      task: r.task || "",
-      lastActivityAt: Date.now(),
-      currentTool,
-      currentToolArgs,
-      currentToolStartedAt: currentTool ? Date.now() : undefined,
-      recentTools,
-      recentOutput,
-      toolCount: recentTools.length + (currentTool ? 1 : 0),
-      tokens: 0,
-      durationMs: 0,
-    } : {
-      index: i,
-      agent: r.agent,
-      status: "completed", // Takomi results in mappedResults are usually final if code !== -1
-      task: r.task || "",
-      lastActivityAt: Date.now(),
-      recentTools,
-      recentOutput: [],
-      toolCount: recentTools.length,
-      tokens: 0,
-      durationMs: 0,
-    };
-
-    return {
-      agent: r.agent,
-      task: r.task || "",
-      exitCode: isRunning ? 0 : r.code,
-      model: r.model,
-      usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, turns: 0 },
-      progress,
-      truncation: isRunning ? undefined : {
-        text: textLines.join("\n").replace(/\n/g, "  \n"),
-        truncated: false
-      }
-    };
-  });
-
-  const mappedDetails: Details = {
-    mode: details?.mode ?? "single",
-    results: mappedResults,
-    chainAgents: details?.mode === "chain" ? results.map(r => r.agent) : undefined,
-  };
-
-  const agentToolResult: AgentToolResult<Details> = {
-    content: result.content as any || [],
-    details: mappedDetails,
-  };
-
-  syncResultAnimation(agentToolResult, context);
-  return renderSubagentResult(agentToolResult, { expanded: options.expanded ?? false }, theme);
-}
-
+import { renderSubagentResult, syncResultAnimation } from "pi-subagents/src/tui/render";
+import type { AgentToolResult } from "@mariozechner/pi-agent-core";
+import type { Details, SingleResult, AgentProgress } from "pi-subagents/src/shared/types";
+import type { Theme } from "@mariozechner/pi-coding-agent";
+import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
+import type { TakomiDispatchResult } from "./dispatch";
+import type { TakomiSubagentToolParams } from "./tool-runner";
+
+type ToolResult = {
+  content?: Array<{ type: string; text?: string }>;
+  details?: {
+    results?: TakomiDispatchResult[];
+    mode?: "single" | "parallel" | "chain";
+    agentScope?: string;
+    plan?: unknown;
+  };
+  isError?: boolean;
+};
+
+function taskList(params: TakomiSubagentToolParams): Array<{ agent: string; task: string }> {
+  if (params.chain?.length) return params.chain;
+  if (params.tasks?.length) return params.tasks;
+  if (params.agent || params.task) return [{ agent: params.agent ?? "...", task: params.task ?? "..." }];
+  return [];
+}
+
+export function renderTakomiSubagentCall(params: TakomiSubagentToolParams, theme: Theme) {
+  const tasks = taskList(params);
+  const mode = params.chain?.length ? "chain" : params.tasks?.length ? "parallel" : "single";
+  if (tasks.length === 1) {
+    return new Text(
+      `${theme.fg("toolTitle", theme.bold("takomi_subagent "))}${theme.fg("accent", tasks[0]?.agent || "?")}`,
+      0,
+      0,
+    );
+  }
+  return new Text(
+    `${theme.fg("toolTitle", theme.bold("takomi_subagent "))}${mode} (${tasks.length})`,
+    0,
+    0,
+  );
+}
+
+function parseTakomiOutput(outputText: string) {
+  const rawLines = outputText.split(/\r?\n/);
+  const textLines: string[] = [];
+  const recentTools: Array<{ tool: string; args: string; endMs: number }> = [];
+  let currentTool: string | undefined;
+  let currentToolArgs: string | undefined;
+
+  for (const line of rawLines) {
+    if (!line.trim()) continue;
+
+    // 1. Check for tool lifecycle markers
+    if (line.startsWith("Tool start: ")) {
+      currentTool = line.replace("Tool start: ", "").trim();
+      continue;
+    }
+    if (line.startsWith("Tool complete: ") || line.startsWith("Tool failed: ")) {
+      const toolName = line.replace(/Tool (complete|failed): /, "").trim();
+      recentTools.push({ tool: toolName, args: "", endMs: Date.now() });
+      if (currentTool === toolName) {
+        currentTool = undefined;
+        currentToolArgs = undefined;
+      }
+      continue;
+    }
+
+    // 2. Handle JSON blobs (tool calls)
+    const jsonStartIdx = line.indexOf('{"');
+    if (jsonStartIdx !== -1) {
+      const beforeJson = line.substring(0, jsonStartIdx).trim();
+      if (beforeJson) textLines.push(beforeJson);
+
+      const jsonPart = line.substring(jsonStartIdx);
+      try {
+        const parsed = JSON.parse(jsonPart);
+        const toolName = parsed.tool || (parsed.command ? parsed.command.split(" ")[0] : undefined);
+        if (toolName) {
+          const args = parsed.args ? JSON.stringify(parsed.args) : (parsed.command || "");
+          currentTool = toolName;
+          currentToolArgs = args;
+        }
+      } catch (e) {
+        if (jsonPart.trim().length > 1) {
+          textLines.push(jsonPart);
+        }
+      }
+      continue;
+    }
+
+    // 3. Skip lone brackets
+    const trimmed = line.trim();
+    if (trimmed === "{" || trimmed === "}") continue;
+
+    // 4. Everything else is text (thoughts)
+    textLines.push(line);
+  }
+
+  return { textLines, recentTools, currentTool, currentToolArgs };
+}
+
+export function renderTakomiSubagentResult(result: ToolResult, options: { expanded?: boolean; isPartial?: boolean }, theme: Theme, context: any) {
+  const details = result.details;
+  const results = details?.results ?? [];
+
+  if (results.length === 0) {
+    const text = result.content?.find((item) => item.type === "text")?.text ?? "(no output)";
+    return new Text(text, 0, 0);
+  }
+
+  const mappedResults: SingleResult[] = results.map((r, i) => {
+    const isRunning = r.code === -1;
+    const outputText = r.output || r.stderr || "";
+    const { textLines, recentTools, currentTool, currentToolArgs } = parseTakomiOutput(outputText);
+    
+    const recentOutput = textLines.slice(-5);
+
+    const progress: AgentProgress | undefined = isRunning ? {
+      index: i,
+      agent: r.agent,
+      status: "running",
+      task: r.task || "",
+      lastActivityAt: Date.now(),
+      currentTool,
+      currentToolArgs,
+      currentToolStartedAt: currentTool ? Date.now() : undefined,
+      recentTools,
+      recentOutput,
+      toolCount: recentTools.length + (currentTool ? 1 : 0),
+      tokens: 0,
+      durationMs: 0,
+    } : {
+      index: i,
+      agent: r.agent,
+      status: "completed", // Takomi results in mappedResults are usually final if code !== -1
+      task: r.task || "",
+      lastActivityAt: Date.now(),
+      recentTools,
+      recentOutput: [],
+      toolCount: recentTools.length,
+      tokens: 0,
+      durationMs: 0,
+    };
+
+    return {
+      agent: r.agent,
+      task: r.task || "",
+      exitCode: isRunning ? 0 : r.code,
+      model: r.model,
+      usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, turns: 0 },
+      progress,
+      truncation: isRunning ? undefined : {
+        text: textLines.join("\n").replace(/\n/g, "  \n"),
+        truncated: false
+      }
+    };
+  });
+
+  const mappedDetails: Details = {
+    mode: details?.mode ?? "single",
+    results: mappedResults,
+    chainAgents: details?.mode === "chain" ? results.map(r => r.agent) : undefined,
+  };
+
+  const agentToolResult: AgentToolResult<Details> = {
+    content: result.content as any || [],
+    details: mappedDetails,
+  };
+
+  syncResultAnimation(agentToolResult, context);
+  return renderSubagentResult(agentToolResult, { expanded: options.expanded ?? false }, theme);
+}
+