@gotgenes/pi-subagents 6.16.3 → 6.17.1

package/src/ui/agent-config-editor.ts

@@ -7,10 +7,10 @@
 
 import { join } from "node:path";
 
-import type { AgentTypeRegistry } from "../agent-types.js";
-import type { AgentConfig } from "../types.js";
-import type { AgentFileOps } from "./agent-file-ops.js";
-import type { MenuUI } from "./agent-menu.js";
+import type { AgentTypeRegistry } from "../agent-types";
+import type { AgentConfig } from "../types";
+import type { AgentFileOps } from "./agent-file-ops";
+import type { MenuUI } from "./agent-menu";
 
 // ---- Factory ----
 

package/src/ui/agent-creation-wizard.ts

@@ -7,11 +7,11 @@
 
 import { join } from "node:path";
 
-import { BUILTIN_TOOL_NAMES } from "../agent-types.js";
-import type { ParentSnapshot } from "../parent-snapshot.js";
-import type { AgentRecord } from "../types.js";
-import type { AgentFileOps } from "./agent-file-ops.js";
-import type { MenuUI } from "./agent-menu.js";
+import { BUILTIN_TOOL_NAMES } from "../agent-types";
+import type { ParentSnapshot } from "../parent-snapshot";
+import type { AgentRecord } from "../types";
+import type { AgentFileOps } from "./agent-file-ops";
+import type { MenuUI } from "./agent-menu";
 
 // ---- Deps interface ----
 

package/src/ui/agent-menu.ts

@@ -1,12 +1,13 @@
-import { AgentTypeRegistry } from "../agent-types.js";
-import type { ModelRegistry } from "../model-resolver.js";
-import type { ParentSnapshot } from "../parent-snapshot.js";
-import type { AgentConfig, AgentRecord } from "../types.js";
-import type { AgentActivityTracker } from "./agent-activity-tracker.js";
-import { createAgentConfigEditor } from "./agent-config-editor.js";
-import { createAgentCreationWizard } from "./agent-creation-wizard.js";
-import type { AgentFileOps } from "./agent-file-ops.js";
-import { formatDuration, getDisplayName } from "./display.js";
+import { wrapTextWithAnsi } from "@earendil-works/pi-tui";
+import { AgentTypeRegistry } from "../agent-types";
+import type { ModelRegistry } from "../model-resolver";
+import type { ParentSnapshot } from "../parent-snapshot";
+import type { AgentConfig, AgentRecord } from "../types";
+import type { AgentActivityTracker } from "./agent-activity-tracker";
+import { createAgentConfigEditor } from "./agent-config-editor";
+import { createAgentCreationWizard } from "./agent-creation-wizard";
+import type { AgentFileOps } from "./agent-file-ops";
+import { formatDuration, getDisplayName } from "./display";
 
 // ---- Deps interface ----
 
@@ -246,7 +247,7 @@ export function createAgentsMenuHandler({
     }
 
     const { ConversationViewer, VIEWPORT_HEIGHT_PCT } = await import(
-      "./conversation-viewer.js"
+      "./conversation-viewer"
     );
     const activity = agentActivity.get(record.id);
 
@@ -260,6 +261,7 @@ export function createAgentsMenuHandler({
           theme,
           done,
           registry,
+          wrapText: wrapTextWithAnsi,
         });
       },
       {

package/src/ui/agent-widget.ts

@@ -5,11 +5,11 @@
  * Uses the callback form of setWidget for themed rendering.
  */
 
-import type { AgentManager } from "../agent-manager.js";
-import { AgentTypeRegistry } from "../agent-types.js";
-import type { AgentActivityTracker } from "./agent-activity-tracker.js";
-import { ERROR_STATUSES, type Theme } from "./display.js";
-import { renderWidgetLines } from "./widget-renderer.js";
+import type { AgentManager } from "../agent-manager";
+import { AgentTypeRegistry } from "../agent-types";
+import type { AgentActivityTracker } from "./agent-activity-tracker";
+import { ERROR_STATUSES, type Theme } from "./display";
+import { renderWidgetLines } from "./widget-renderer";
 
 // ---- Types ----
 

package/src/ui/conversation-viewer.ts

@@ -6,13 +6,13 @@
  */
 
 import type { AgentSession } from "@earendil-works/pi-coding-agent";
-import { type Component, matchesKey, type TUI, truncateToWidth, visibleWidth, wrapTextWithAnsi } from "@earendil-works/pi-tui";
-import type { AgentConfigLookup } from "../agent-types.js";
-import { extractText } from "../context.js";
-import type { AgentRecord } from "../types.js";
-import { getLifetimeTotal, getSessionContextPercent } from "../usage.js";
-import type { AgentActivityTracker } from "./agent-activity-tracker.js";
-import { buildInvocationTags, describeActivity, formatDuration, formatSessionTokens, getDisplayName, getPromptModeLabel, type Theme } from "./display.js";
+import { type Component, matchesKey, type TUI, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import type { AgentConfigLookup } from "../agent-types";
+import { extractText } from "../context";
+import type { AgentRecord } from "../types";
+import { getLifetimeTotal, getSessionContextPercent } from "../usage";
+import type { AgentActivityTracker } from "./agent-activity-tracker";
+import { buildInvocationTags, describeActivity, formatDuration, formatSessionTokens, getDisplayName, getPromptModeLabel, type Theme } from "./display";
 
 // ── Local message-shape types ───────────────────────────────────────────────
 // The Pi SDK does not export narrow types for all message content variants.
@@ -59,6 +59,7 @@ export interface ConversationViewerOptions {
   theme: Theme;
   done: (result: undefined) => void;
   registry: AgentConfigLookup;
+  wrapText: (text: string, width: number) => string[];
 }
 
 export class ConversationViewer implements Component {
@@ -75,16 +76,27 @@ export class ConversationViewer implements Component {
   private theme: Theme;
   private done: (result: undefined) => void;
   private registry: AgentConfigLookup;
-
-  constructor(options: ConversationViewerOptions) {
-    this.tui = options.tui;
-    this.session = options.session;
-    this.record = options.record;
-    this.activity = options.activity;
-    this.theme = options.theme;
-    this.done = options.done;
-    this.registry = options.registry;
-    this.unsubscribe = options.session.subscribe(() => {
+  private wrapText: (text: string, width: number) => string[];
+
+  constructor({
+    tui,
+    session,
+    record,
+    activity,
+    theme,
+    done,
+    registry,
+    wrapText,
+  }: ConversationViewerOptions) {
+    this.tui = tui;
+    this.session = session;
+    this.record = record;
+    this.activity = activity;
+    this.theme = theme;
+    this.done = done;
+    this.registry = registry;
+    this.wrapText = wrapText;
+    this.unsubscribe = session.subscribe(() => {
       if (this.closed) return;
       this.tui.requestRender();
     });
@@ -253,7 +265,7 @@ export class ConversationViewer implements Component {
         if (!text.trim()) continue;
         if (needsSeparator) lines.push(th.fg("dim", "───"));
         lines.push(th.fg("accent", "[User]"));
-        for (const line of wrapTextWithAnsi(text.trim(), width)) {
+        for (const line of this.wrapText(text.trim(), width)) {
           lines.push(line);
         }
       } else if (msg.role === "assistant") {
@@ -268,7 +280,7 @@ export class ConversationViewer implements Component {
         if (needsSeparator) lines.push(th.fg("dim", "───"));
         lines.push(th.bold("[Assistant]"));
         if (textParts.length > 0) {
-          for (const line of wrapTextWithAnsi(textParts.join("\n").trim(), width)) {
+          for (const line of this.wrapText(textParts.join("\n").trim(), width)) {
             lines.push(line);
           }
         }
@@ -281,7 +293,7 @@ export class ConversationViewer implements Component {
         if (!truncated.trim()) continue;
         if (needsSeparator) lines.push(th.fg("dim", "───"));
         lines.push(th.fg("dim", "[Result]"));
-        for (const line of wrapTextWithAnsi(truncated.trim(), width)) {
+        for (const line of this.wrapText(truncated.trim(), width)) {
           lines.push(th.fg("dim", line));
         }
       } else if (isBashExecution(msg)) {
@@ -291,7 +303,7 @@ export class ConversationViewer implements Component {
           const out = msg.output.length > 500
             ? msg.output.slice(0, 500) + "... (truncated)"
             : msg.output;
-          for (const line of wrapTextWithAnsi(out.trim(), width)) {
+          for (const line of this.wrapText(out.trim(), width)) {
             lines.push(th.fg("dim", line));
           }
         }

package/src/ui/display.ts

@@ -5,8 +5,8 @@
  * Consumed by the widget, the menu, tool modules, and the notification renderer.
  */
 
-import type { AgentConfigLookup } from "../agent-types.js";
-import type { AgentInvocation, SubagentType } from "../types.js";
+import type { AgentConfigLookup } from "../agent-types";
+import type { AgentInvocation, SubagentType } from "../types";
 
 // ---- Types ----
 

package/src/ui/ui-observer.ts

@@ -6,7 +6,7 @@
  * turn count, lifetime usage).
  */
 
-import type { AgentActivityTracker } from "./agent-activity-tracker.js";
+import type { AgentActivityTracker } from "./agent-activity-tracker";
 
 /** Narrow session interface — only the subscribe method needed by the observer. */
 interface SubscribableSession {

package/src/ui/widget-renderer.ts

@@ -6,10 +6,10 @@
  */
 
 import { truncateToWidth } from "@earendil-works/pi-tui";
-import type { AgentConfigLookup } from "../agent-types.js";
-import type { SubagentType } from "../types.js";
-import type { LifetimeUsage, SessionLike } from "../usage.js";
-import { getLifetimeTotal, getSessionContextPercent } from "../usage.js";
+import type { AgentConfigLookup } from "../agent-types";
+import type { SubagentType } from "../types";
+import type { LifetimeUsage, SessionLike } from "../usage";
+import { getLifetimeTotal, getSessionContextPercent } from "../usage";
 import {
 	describeActivity,
 	formatMs,
@@ -19,7 +19,7 @@ import {
 	getPromptModeLabel,
 	SPINNER,
 	type Theme,
-} from "./display.js";
+} from "./display";
 
 // ── Data interfaces ──────────────────────────────────────────────────────────
 

package/src/worktree-state.ts

@@ -6,7 +6,7 @@
  * cleanupResult is recorded once at completion or error — it is not set at construction.
  */
 
-import type { WorktreeCleanupResult, WorktreeInfo } from "./worktree.js";
+import type { WorktreeCleanupResult, WorktreeInfo } from "./worktree";
 
 export type { WorktreeCleanupResult, WorktreeInfo };
 

package/src/worktree.ts

@@ -11,7 +11,7 @@ import { randomUUID } from "node:crypto";
 import { existsSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
-import { debugLog } from "./debug.js";
+import { debugLog } from "./debug";
 
 export interface WorktreeInfo {
   /** Absolute path to the worktree directory. */

package/vitest.config.ts

@@ -0,0 +1,14 @@
+import path from "node:path";
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+  resolve: {
+    alias: {
+      "#src": path.resolve(import.meta.dirname, "src"),
+      "#test": path.resolve(import.meta.dirname, "test"),
+    },
+  },
+  test: {
+    include: ["test/**/*.test.ts"],
+  },
+});