@vzlentin/harness 0.2.3 → 0.3.0

package/pi/usage-line.ts

@@ -0,0 +1,59 @@
+/** The usage line of a mirror session, above the editor, as Pi's footer shows the usage of its own session. */
+import type { ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
+import * as coding from "@earendil-works/pi-coding-agent";
+import type { ConversationView } from "@earendil-works/pi-durable";
+import { type ChildInfo, childTitle } from "../dist/protocol.js";
+import { lastAnswerUsage, usageTotals } from "../dist/view-state.js";
+
+/**
+ * The title of the agent, the tokens and the cost of its model requests and tool results, and the context of its last
+ * answer that did not abort or fail, against the context window of its model, as Pi's footer shows them for its own
+ * session.
+ */
+export function usageLine(
+	theme: Theme,
+	models: ExtensionContext["modelRegistry"],
+	child: ChildInfo,
+	view: ConversationView,
+): string {
+	const totals = usageTotals(view);
+	const last = lastAnswerUsage(view);
+	const model = models.find(child.provider, child.model);
+	const subscription = model !== undefined && models.isUsingOAuth(model);
+	const usage: string[] = [];
+
+	if (totals.input) usage.push(`↑${formatTokens(totals.input)}`);
+
+	if (totals.output) usage.push(`↓${formatTokens(totals.output)}`);
+
+	if (totals.cacheRead) usage.push(`R${formatTokens(totals.cacheRead)}`);
+
+	if (totals.cacheWrite) usage.push(`W${formatTokens(totals.cacheWrite)}`);
+
+	if (totals.cost || subscription) usage.push(`$${totals.cost.toFixed(3)}${subscription ? " (sub)" : ""}`);
+	const parts = [childTitle(child)];
+
+	if (usage.length > 0) parts.push(usage.join(" "));
+	const contextWindow = model?.contextWindow ?? 0;
+
+	if (contextWindow > 0) {
+		const percent = last === undefined ? 0 : (coding.calculateContextTokens(last) / contextWindow) * 100;
+		const context = `${percent.toFixed(1)}%/${formatTokens(contextWindow)}`;
+		parts.push(percent > 90 ? theme.fg("error", context) : percent > 70 ? theme.fg("warning", context) : context);
+	}
+
+	return theme.fg("dim", parts.join(" • "));
+}
+
+/** A token count as Pi's footer shows it: `950`, `1.2k`, `45k`, `1.5M`. */
+function formatTokens(count: number): string {
+	if (count < 1000) return count.toString();
+
+	if (count < 10_000) return `${(count / 1000).toFixed(1)}k`;
+
+	if (count < 1_000_000) return `${Math.round(count / 1000)}k`;
+
+	if (count < 10_000_000) return `${(count / 1_000_000).toFixed(1)}M`;
+
+	return `${Math.round(count / 1_000_000)}M`;
+}

package/dist/chat.d.ts

@@ -1,48 +0,0 @@
-/**
- * A durable conversation drawn with Pi's own message and tool components.
- *
- * Adapted from `ChatView` in mavam/pi-agents, src/ui/attach.ts. MIT License, Copyright (c) 2026 Matthias Vallentin.
- */
-import type { ToolResultMessage } from "@earendil-works/pi-ai";
-import type { ConversationView } from "@earendil-works/pi-durable";
-import { Container, type TUI } from "@earendil-works/pi-tui";
-/**
- * A tool result as Pi shows its own. pi-durable adds the diagnostics of a call for the model as a `<harness>` block, such
- * as `Command exited with code 3`, and the card shows them as plain lines.
- */
-export declare function displayedResult(result: ToolResultMessage): ToolResultMessage;
-/** The transcript of one conversation view, updated in place as the view changes. */
-export declare class ChatView {
-    readonly transcript: Container;
-    /** The newest card for each call ID: providers can use a call ID again in a later turn. */
-    private readonly tools;
-    /** Every card shown, also the older ones whose call ID a later turn used again. */
-    private readonly cards;
-    /** The cards that have no final result yet. */
-    private readonly running;
-    /** The call IDs whose cards the streamed answer made; its entry takes them over. */
-    private readonly streamingCalls;
-    private readonly renderers;
-    private renderedEntryIds;
-    private streaming;
-    /** Ctrl+O shows tool output in full, as in Pi. */
-    private expanded;
-    private readonly ui;
-    private readonly cwd;
-    constructor(ui: TUI, cwd: string);
-    apply(view: ConversationView): void;
-    /** Shows tool output in full, or collapses it again. */
-    toggleExpanded(): void;
-    /** Finishes the running cards: a running bash card keeps a timer until it gets a final result. */
-    stopTimers(): void;
-    private syncTranscript;
-    private rebuild;
-    private addEntry;
-    private syncStreaming;
-    private renderer;
-    /** The newest card of a call, or a new card when the call has none. */
-    private card;
-    /** A new card of a call, also for a call ID that an earlier turn used. */
-    private newCard;
-    private addText;
-}

package/dist/chat.js

@@ -1,208 +0,0 @@
-import { AssistantMessageComponent, createBashToolDefinition, createEditToolDefinition, createFindToolDefinition, createGrepToolDefinition, createLsToolDefinition, createReadToolDefinition, createWriteToolDefinition, getMarkdownTheme, ToolExecutionComponent, UserMessageComponent, } from "@earendil-works/pi-coding-agent";
-import { Container, Spacer, Text } from "@earendil-works/pi-tui";
-import { color } from "./theme.js";
-import { liveOf, userText } from "./view-state.js";
-/** Pi's renderers of the tools that children have under the same names. */
-const RENDERERS = new Map([
-    ["read", createReadToolDefinition],
-    ["bash", createBashToolDefinition],
-    ["edit", createEditToolDefinition],
-    ["write", createWriteToolDefinition],
-    ["grep", createGrepToolDefinition],
-    ["find", createFindToolDefinition],
-    ["ls", createLsToolDefinition],
-]);
-const DIAGNOSTICS = /^<harness>\n([\s\S]*)\n<\/harness>$/;
-/**
- * A tool result as Pi shows its own. pi-durable adds the diagnostics of a call for the model as a `<harness>` block, such
- * as `Command exited with code 3`, and the card shows them as plain lines.
- */
-export function displayedResult(result) {
-    let changed = false;
-    const content = result.content.map((block) => {
-        const match = block.type === "text" ? DIAGNOSTICS.exec(block.text) : null;
-        if (!match)
-            return block;
-        changed = true;
-        const lines = (match[1] ?? "").split("\n").map((line) => line.replace(/^\[(error|warning|info)\] /, ""));
-        return { type: "text", text: `\n${lines.join("\n")}` };
-    });
-    return changed ? { ...result, content } : result;
-}
-/** The transcript of one conversation view, updated in place as the view changes. */
-export class ChatView {
-    transcript = new Container();
-    /** The newest card for each call ID: providers can use a call ID again in a later turn. */
-    tools = new Map();
-    /** Every card shown, also the older ones whose call ID a later turn used again. */
-    cards = [];
-    /** The cards that have no final result yet. */
-    running = new Set();
-    /** The call IDs whose cards the streamed answer made; its entry takes them over. */
-    streamingCalls = new Set();
-    renderers = new Map();
-    renderedEntryIds = [];
-    streaming;
-    /** Ctrl+O shows tool output in full, as in Pi. */
-    expanded = false;
-    ui;
-    cwd;
-    constructor(ui, cwd) {
-        this.ui = ui;
-        this.cwd = cwd;
-    }
-    apply(view) {
-        const live = liveOf(view);
-        this.syncTranscript(view.entries);
-        const message = live.generation?.message;
-        // A streamed answer without its entry was dropped, for example by a retry.
-        if (message === undefined && this.streaming !== undefined)
-            this.rebuild(view.entries);
-        if (message !== undefined)
-            this.syncStreaming(message);
-        for (const slot of live.tools ?? []) {
-            if (slot.status === "pending")
-                continue;
-            const component = this.card(slot.name, slot.callId);
-            component.setArgsComplete();
-            if (slot.status !== "running")
-                continue;
-            component.markExecutionStarted();
-            if (slot.output !== undefined || slot.details !== undefined) {
-                component.updateResult({ content: [{ type: "text", text: slot.output ?? "" }], details: slot.details, isError: false }, true);
-            }
-        }
-        this.transcript.invalidate();
-    }
-    /** Shows tool output in full, or collapses it again. */
-    toggleExpanded() {
-        this.expanded = !this.expanded;
-        for (const card of this.cards)
-            card.setExpanded(this.expanded);
-        this.transcript.invalidate();
-    }
-    /** Finishes the running cards: a running bash card keeps a timer until it gets a final result. */
-    stopTimers() {
-        for (const component of this.running)
-            component.updateResult({ content: [], isError: false }, false);
-        this.running.clear();
-    }
-    syncTranscript(entries) {
-        // A compaction or a reset replaces the head of the active transcript.
-        if (this.renderedEntryIds.some((id, index) => entries[index]?.id !== id))
-            this.rebuild(entries);
-        for (const entry of entries.slice(this.renderedEntryIds.length)) {
-            this.addEntry(entry);
-            this.renderedEntryIds.push(entry.id);
-        }
-    }
-    rebuild(entries) {
-        this.stopTimers();
-        this.transcript.clear();
-        this.cards.length = 0;
-        this.tools.clear();
-        this.streamingCalls.clear();
-        this.renderedEntryIds = [];
-        this.streaming = undefined;
-        for (const entry of entries) {
-            this.addEntry(entry);
-            this.renderedEntryIds.push(entry.id);
-        }
-    }
-    addEntry(entry) {
-        const message = entry.model?.[0];
-        if (entry.kind === "pi.user" && message?.role === "user") {
-            this.transcript.addChild(new Spacer(1));
-            this.transcript.addChild(new UserMessageComponent(userText(message.content), getMarkdownTheme()));
-        }
-        else if (entry.kind === "pi.assistant" && message?.role === "assistant") {
-            const component = this.streaming ?? new AssistantMessageComponent(undefined, false, getMarkdownTheme());
-            if (this.streaming === undefined)
-                this.transcript.addChild(component);
-            this.streaming = undefined;
-            component.updateContent(message, false);
-            // Only an answer that asks for tools runs its calls; an aborted, failed or cut answer never does.
-            const ran = message.stopReason === "toolUse";
-            for (const content of message.content) {
-                if (content.type !== "toolCall")
-                    continue;
-                const streamed = this.streamingCalls.has(content.id);
-                if (!ran && !streamed)
-                    continue;
-                const card = streamed
-                    ? this.card(content.name, content.id, content.arguments)
-                    : this.newCard(content.name, content.id, content.arguments);
-                card.setArgsComplete();
-                if (!ran) {
-                    const text = "Not run: the answer was interrupted.";
-                    card.updateResult({ content: [{ type: "text", text }], isError: true }, false);
-                    this.running.delete(card);
-                }
-            }
-            this.streamingCalls.clear();
-        }
-        else if (entry.kind === "pi.tool-result" && message?.role === "toolResult") {
-            const card = this.card(message.toolName, message.toolCallId);
-            card.updateResult(displayedResult(message));
-            this.running.delete(card);
-        }
-        else if (entry.kind === "pi.compaction") {
-            this.addText(color("muted", "[compaction]"));
-            if (message?.role === "user")
-                this.addText(userText(message.content));
-        }
-        else if (entry.kind === "pi.reset") {
-            this.addText(color("muted", "[new context]"));
-        }
-    }
-    syncStreaming(message) {
-        if (this.streaming === undefined) {
-            this.streaming = new AssistantMessageComponent(undefined, false, getMarkdownTheme());
-            this.transcript.addChild(this.streaming);
-        }
-        this.streaming.updateContent(message, true);
-        for (const content of message.content) {
-            if (content.type !== "toolCall")
-                continue;
-            if (this.streamingCalls.has(content.id))
-                this.card(content.name, content.id, content.arguments);
-            else
-                this.newCard(content.name, content.id, content.arguments);
-            this.streamingCalls.add(content.id);
-        }
-    }
-    renderer(toolName) {
-        let definition = this.renderers.get(toolName);
-        if (definition === undefined) {
-            const create = RENDERERS.get(toolName);
-            if (create === undefined)
-                return undefined;
-            definition = create(this.cwd);
-            this.renderers.set(toolName, definition);
-        }
-        return definition;
-    }
-    /** The newest card of a call, or a new card when the call has none. */
-    card(toolName, toolCallId, args) {
-        const existing = this.tools.get(toolCallId);
-        if (existing === undefined)
-            return this.newCard(toolName, toolCallId, args ?? {});
-        if (args !== undefined)
-            existing.updateArgs(args);
-        return existing;
-    }
-    /** A new card of a call, also for a call ID that an earlier turn used. */
-    newCard(toolName, toolCallId, args) {
-        const component = new ToolExecutionComponent(toolName, toolCallId, args, {}, this.renderer(toolName), this.ui, this.cwd);
-        component.setExpanded(this.expanded);
-        this.transcript.addChild(component);
-        this.cards.push(component);
-        this.running.add(component);
-        this.tools.set(toolCallId, component);
-        return component;
-    }
-    addText(text) {
-        this.transcript.addChild(new Spacer(1));
-        this.transcript.addChild(new Text(text, 1, 0));
-    }
-}

package/dist/chrome.d.ts

@@ -1,18 +0,0 @@
-import { type ConversationView } from "@earendil-works/pi-durable";
-import { type Component } from "@earendil-works/pi-tui";
-import type { ChildInfo } from "./protocol.ts";
-/** One line, cut to the width, with `hints` at the right end: the first hints go when they do not all fit. */
-export declare class Line implements Component {
-    text: string;
-    hints: readonly string[];
-    style: (text: string) => string;
-    render(width: number): string[];
-    invalidate(): void;
-}
-/** `harness │ <title> │ <cwd> │ <state>`. */
-export declare function headerText(child: ChildInfo, title: string): string;
-/**
- * `<provider>/<model> · <thinking> │ <tokens> tok $<cost> │ <used>%/<window> ctx`. The context part shows when
- * `contextWindow` is known and the child has answered once.
- */
-export declare function footerText(child: ChildInfo, view: ConversationView, contextWindow: number | undefined): string;

package/dist/chrome.js

@@ -1,74 +0,0 @@
-import { homedir } from "node:os";
-import { calculateContextTokens } from "@earendil-works/pi-coding-agent";
-import { UsageDoc } from "@earendil-works/pi-durable";
-import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
-import { bold, color } from "./theme.js";
-import { docOf } from "./view-state.js";
-/** One line, cut to the width, with `hints` at the right end: the first hints go when they do not all fit. */
-export class Line {
-    text = "";
-    hints = [];
-    style = (text) => text;
-    render(width) {
-        for (let first = 0; first < this.hints.length; first++) {
-            const right = `${this.hints.slice(first).join(" · ")} `;
-            const gap = width - visibleWidth(this.text) - visibleWidth(right);
-            if (gap >= 2)
-                return [`${this.text}${" ".repeat(gap)}${this.style(right)}`];
-        }
-        return [truncateToWidth(this.text, width)];
-    }
-    invalidate() { }
-}
-/** `harness │ <title> │ <cwd> │ <state>`. */
-export function headerText(child, title) {
-    const state = child.state === "busy"
-        ? color("warning", "◉ busy")
-        : child.state === "idle"
-            ? color("success", "● idle")
-            : color("error", "⊗ error");
-    return ` ${[bold(color("accent", "harness")), title, color("muted", tildePath(child.cwd)), state].join(color("borderMuted", "  │  "))}`;
-}
-/**
- * `<provider>/<model> · <thinking> │ <tokens> tok $<cost> │ <used>%/<window> ctx`. The context part shows when
- * `contextWindow` is known and the child has answered once.
- */
-export function footerText(child, view, contextWindow) {
-    const usage = totalUsage(docOf(view, UsageDoc));
-    const used = lastUsage(view.entries);
-    const context = contextWindow === undefined || contextWindow === 0 || used === undefined
-        ? []
-        : [`${((100 * used) / contextWindow).toFixed(1)}%/${compactCount(contextWindow)} ctx`];
-    return ` ${color("muted", [
-        `${child.provider}/${child.model} · ${child.thinking}`,
-        `${compactCount(usage.tokens)} tok $${usage.cost.toFixed(3)}`,
-        ...context,
-    ].join(" │ "))}`;
-}
-/** The tokens and the cost of every model request of the conversation. */
-function totalUsage(usage) {
-    let tokens = 0;
-    let cost = 0;
-    for (const model of Object.values(usage?.models ?? {})) {
-        tokens += model.totalTokens;
-        cost += model.cost.total;
-    }
-    return { tokens, cost };
-}
-function compactCount(count) {
-    return Intl.NumberFormat("en", { notation: "compact", maximumFractionDigits: 1 }).format(count);
-}
-function tildePath(path) {
-    const home = homedir();
-    return home !== "" && (path === home || path.startsWith(`${home}/`)) ? `~${path.slice(home.length)}` : path;
-}
-/** The prompt and output tokens of the last answer, which fill the context window of the next request. */
-function lastUsage(entries) {
-    for (const entry of entries.toReversed()) {
-        const message = entry.model?.[0];
-        if (entry.kind !== "pi.assistant" || message?.role !== "assistant")
-            continue;
-        return calculateContextTokens(message.usage);
-    }
-    return undefined;
-}

package/dist/picker.d.ts

@@ -1,22 +0,0 @@
-import { type Component } from "@earendil-works/pi-tui";
-import type { ModelInfo } from "./protocol.ts";
-/**
- * A filter over a list of models, fuzzy as Pi's model selector: typing narrows the list and keeps the selected model
- * when it still matches, ↑↓ move, Enter picks, Esc closes.
- */
-export declare class ModelPicker implements Component {
-    private readonly input;
-    private readonly items;
-    private readonly modelOf;
-    private readonly onSelect;
-    private readonly onCancel;
-    private readonly border;
-    private list;
-    focused: boolean;
-    constructor(models: readonly ModelInfo[], current: ModelInfo | undefined, border: (text: string) => string, onSelect: (model: ModelInfo) => void, onCancel: () => void);
-    /** A frame with full-width lines, so the transcript does not show through. */
-    render(width: number): string[];
-    handleInput(data: string): void;
-    invalidate(): void;
-    private listOf;
-}

package/dist/picker.js

@@ -1,70 +0,0 @@
-import { getSelectListTheme } from "@earendil-works/pi-coding-agent";
-import { fuzzyFilter, Input, matchesKey, SelectList, truncateToWidth, visibleWidth, } from "@earendil-works/pi-tui";
-/** The keys that the list of the model picker gets; the filter gets the others. */
-const LIST_KEYS = ["up", "down", "enter", "escape", "pageUp", "pageDown"];
-/**
- * A filter over a list of models, fuzzy as Pi's model selector: typing narrows the list and keeps the selected model
- * when it still matches, ↑↓ move, Enter picks, Esc closes.
- */
-export class ModelPicker {
-    input = new Input();
-    items;
-    modelOf = new Map();
-    onSelect;
-    onCancel;
-    border;
-    list;
-    focused = true;
-    constructor(models, current, border, onSelect, onCancel) {
-        this.items = models.map((model) => {
-            const item = {
-                value: `${model.provider}/${model.id}`,
-                label: `${model.provider}/${model.id}`,
-                description: model.name,
-            };
-            this.modelOf.set(item, model);
-            return item;
-        });
-        this.border = border;
-        this.onSelect = onSelect;
-        this.onCancel = onCancel;
-        this.input.focused = true;
-        this.list = this.listOf(this.items);
-        this.list.setSelectedIndex(current === undefined ? 0 : models.indexOf(current));
-    }
-    /** A frame with full-width lines, so the transcript does not show through. */
-    render(width) {
-        const inner = Math.max(1, width - 2);
-        const body = [" Model", ...this.input.render(inner), ...this.list.render(inner)].map((line) => {
-            const cut = truncateToWidth(line, inner);
-            return `${this.border("│")}${cut}${" ".repeat(Math.max(0, inner - visibleWidth(cut)))}${this.border("│")}`;
-        });
-        return [this.border(`╭${"─".repeat(inner)}╮`), ...body, this.border(`╰${"─".repeat(inner)}╯`)];
-    }
-    handleInput(data) {
-        if (LIST_KEYS.some((key) => matchesKey(data, key))) {
-            this.list.handleInput(data);
-            return;
-        }
-        this.input.handleInput(data);
-        const selected = this.list.getSelectedItem();
-        const query = this.input.getValue();
-        const items = fuzzyFilter(this.items, query, (item) => `${item.label} ${item.description ?? ""}`);
-        this.list = this.listOf(items);
-        this.list.setSelectedIndex(selected === null ? 0 : items.indexOf(selected));
-    }
-    invalidate() {
-        this.input.invalidate();
-        this.list.invalidate();
-    }
-    listOf(items) {
-        const list = new SelectList(items, 10, getSelectListTheme());
-        list.onSelect = (item) => {
-            const model = this.modelOf.get(item);
-            if (model !== undefined)
-                this.onSelect(model);
-        };
-        list.onCancel = this.onCancel;
-        return list;
-    }
-}

package/dist/theme.d.ts

@@ -1,3 +0,0 @@
-import type { Theme } from "@earendil-works/pi-coding-agent";
-export declare function color(name: Parameters<Theme["fg"]>[0], text: string): string;
-export declare function bold(text: string): string;

package/dist/theme.js

@@ -1,19 +0,0 @@
-var __rewriteRelativeImportExtension = (this && this.__rewriteRelativeImportExtension) || function (path, preserveJsx) {
-    if (typeof path === "string" && /^\.\.?\//.test(path)) {
-        return path.replace(/\.(tsx)$|((?:\.d)?)((?:\.[^./]+?)?)\.([cm]?)ts$/i, function (m, tsx, d, ext, cm) {
-            return tsx ? preserveJsx ? ".jsx" : ".js" : d && (!ext || !cm) ? m : (d + ext + "." + cm.toLowerCase() + "js");
-        });
-    }
-    return path;
-};
-// Pi gives its theme object only to extensions. Pi's cards color with the theme that `initTheme()` sets in this module,
-// so the viewer loads the same module by its path to color its own lines the same way. The path is not in the `exports`
-// of pi-coding-agent, so it holds only for the pinned version 1.3.0.
-// SAFETY: this module of pi-coding-agent 1.3.0, the version that package.json pins, exports the `theme` of Pi's cards.
-const piTheme = (await import(__rewriteRelativeImportExtension(new URL("./modes/interactive/theme/theme.js", import.meta.resolve("@earendil-works/pi-coding-agent")).href)));
-export function color(name, text) {
-    return piTheme.theme.fg(name, text);
-}
-export function bold(text) {
-    return piTheme.theme.bold(text);
-}

package/dist/viewer.d.ts

@@ -1,20 +0,0 @@
-import { type Terminal } from "@earendil-works/pi-tui";
-declare module "@earendil-works/pi-tui" {
-    interface Keybindings {
-        "app.scroll.lineUp": true;
-        "app.scroll.lineDown": true;
-        "app.scroll.pageUp": true;
-        "app.scroll.pageDown": true;
-    }
-}
-export type AttachOptions = {
-    readonly terminal: Terminal;
-    readonly env: NodeJS.ProcessEnv;
-};
-/**
- * Shows the conversation of the child `handle`, or of the project agent for `agent`, in `terminal` and sends
- * what the user types to it. It starts the host when none runs, and rejects before the terminal starts when the handle
- * is unknown. It resolves with the exit code: 0 after Ctrl+C or when the child is stopped, 1 when the host closes the
- * connection. The child keeps running.
- */
-export declare function attach(handle: string, options: AttachOptions): Promise<number>;