@vzlentin/harness 0.3.1 → 0.4.0

package/pi/model-sync.ts

@@ -1,102 +0,0 @@
-/**
- * The model and the thinking level of a mirror session. The agent in the host is the truth: the Pi session shows the
- * agent's model and level, so Pi's footer and `/model` show them, and a change that the user makes in Pi is a request
- * to the host. When the host refuses it, Pi goes back to the agent's model and level.
- */
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import type { Mirror, MirrorConfigure } from "../dist/mirror.js";
-import type { ChildInfo } from "../dist/protocol.js";
-
-/** What the sync uses of Pi's extension API. */
-export type PiModels = Pick<ExtensionAPI, "setModel" | "getThinkingLevel" | "setThinkingLevel">;
-
-/** What the sync uses of the context of the Pi session. */
-export type PiSession = Pick<ExtensionContext, "model"> & {
-	readonly modelRegistry: Pick<ExtensionContext["modelRegistry"], "find">;
-};
-
-/** What the sync uses of the mirror: the agent, and the requests that change it. */
-export type MirroredAgent = Pick<Mirror, "child" | "configure">;
-
-export class ModelSync {
-	// After a session switch, `#pi` and `#ctx` are stale and throw, so nothing uses them after `close()`.
-	readonly #pi: PiModels;
-	readonly #ctx: PiSession;
-	readonly #mirror: MirroredAgent;
-	readonly #notify: (cause: unknown) => void;
-	/** True while Pi sets the agent's model: `setModel` is async, and the agent changes often while it works. */
-	#settingModel = false;
-	#closed = false;
-
-	constructor(pi: PiModels, ctx: PiSession, mirror: MirroredAgent, notify: (cause: unknown) => void) {
-		this.#pi = pi;
-		this.#ctx = ctx;
-		this.#mirror = mirror;
-		this.#notify = notify;
-	}
-
-	/** Gives the Pi session the model and the thinking level of `child`, the agent as the host gives it. */
-	fromAgent(child: ChildInfo): void {
-		if (this.#closed) return;
-		const model = this.#ctx.modelRegistry.find(child.provider, child.model);
-
-		if (model !== undefined && !this.#piHasModel(child) && !this.#settingModel) {
-			this.#settingModel = true;
-			// Pi sets its own thinking level when it takes a model, so the agent's level follows once Pi has the model.
-			void this.#pi.setModel(model).finally(() => {
-				this.#settingModel = false;
-				const current = this.#mirror.child;
-
-				if (current !== undefined) this.#thinkingFromAgent(current);
-			});
-		}
-
-		this.#thinkingFromAgent(child);
-	}
-
-	/** Asks the host for the model that the user selected in Pi. */
-	piSelected(provider: string, model: string): void {
-		const child = this.#mirror.child;
-
-		if (child === undefined || (child.provider === provider && child.model === model)) return;
-		this.#configure({ provider, model });
-	}
-
-	/**
-	 * Asks the host for the thinking level that the user cycled to in Pi, once Pi has the agent's model. Pi also changes
-	 * its level when it takes a model, so only the cycle key calls this, not each change of Pi's level.
-	 */
-	piCycled(): void {
-		const child = this.#mirror.child;
-
-		if (this.#closed || child === undefined || !this.#piHasModel(child)) return;
-		const thinking = this.#pi.getThinkingLevel();
-
-		if (child.thinking !== thinking) this.#configure({ thinking });
-	}
-
-	close(): void {
-		this.#closed = true;
-	}
-
-	#configure(change: MirrorConfigure): void {
-		this.#mirror.configure(change).catch((cause: unknown) => {
-			this.#notify(cause);
-			const child = this.#mirror.child;
-
-			if (child !== undefined) this.fromAgent(child);
-		});
-	}
-
-	#thinkingFromAgent(child: ChildInfo): void {
-		if (!this.#closed && this.#piHasModel(child) && this.#pi.getThinkingLevel() !== child.thinking) {
-			this.#pi.setThinkingLevel(child.thinking);
-		}
-	}
-
-	#piHasModel(child: ChildInfo): boolean {
-		const model = this.#ctx.model;
-
-		return model?.provider === child.provider && model.id === child.model;
-	}
-}

package/pi/package.json

@@ -1,15 +0,0 @@
-{
-	"private": true,
-	"description": "The Pi extension of harness: the subagent tool, the /subagent command, the subagents widget and the /harness mirror sessions",
-	"type": "module",
-	"pi": {
-		"extensions": [
-			"./index.ts"
-		]
-	},
-	"peerDependencies": {
-		"@earendil-works/pi-ai": "*",
-		"@earendil-works/pi-coding-agent": "*",
-		"@earendil-works/pi-tui": "*"
-	}
-}

package/pi/program-status.ts

@@ -1,66 +0,0 @@
-/**
- * The state of the agent of a mirror session for the terminal, with the Program Status Protocol (OSC 7501): `working`
- * while it works, `error` when its last request failed, and `idle` in all other cases.
- */
-import type { ProgramStatus, Terminal } from "@earendil-works/pi-tui";
-import { type Mirror, targetTitle } from "../dist/mirror.js";
-import { childTitle } from "../dist/protocol.js";
-
-/** The program name of each report. */
-const APP = "harness";
-
-type Mirrored = Pick<Mirror, "target" | "status" | "child" | "view" | "busy">;
-
-/** The title of the agent of `mirror`. */
-function title({ target, child }: Pick<Mirrored, "target" | "child">): string {
-	return child === undefined ? targetTitle(target) : childTitle(child);
-}
-
-/** The first line of the error of the last failed request, or `Error` when there is none. */
-function errorLine({ view }: Pick<Mirrored, "view">): string {
-	const failed = view?.entries.findLast((entry) => {
-		const message = entry.model?.[0];
-
-		return entry.kind === "pi.assistant" && message?.role === "assistant" && message.stopReason === "error";
-	})?.model?.[0];
-
-	const text = failed?.role === "assistant" ? failed.errorMessage : undefined;
-
-	return text?.split(/\r?\n/, 1)[0]?.trim() || "Error";
-}
-
-/** The state of the agent of `mirror`. */
-function programStatus(mirror: Mirrored): ProgramStatus {
-	if (mirror.busy) return { state: "working", app: APP, message: title(mirror) };
-
-	if (mirror.status.kind === "open" && mirror.child?.state === "error") {
-		return { state: "error", app: APP, message: errorLine(mirror) };
-	}
-
-	return { state: "idle", app: APP, message: title(mirror) };
-}
-
-export class ProgramStatusReport {
-	readonly #terminal: Pick<Terminal, "setProgramStatus">;
-	/** The state and the message of the last report. */
-	#last: string | undefined;
-
-	constructor(terminal: Pick<Terminal, "setProgramStatus">) {
-		this.#terminal = terminal;
-	}
-
-	/** Reports the state of the agent of `mirror` when its state or its message changed since the last report. */
-	update(mirror: Mirrored): void {
-		const report = programStatus(mirror);
-		const key = `${report.state}\n${report.message}`;
-
-		if (key === this.#last) return;
-		this.#last = key;
-		this.#terminal.setProgramStatus(report);
-	}
-
-	/** Reports `idle` one more time, when Pi leaves the session of `mirror`. */
-	leave(mirror: Pick<Mirrored, "target" | "child">): void {
-		this.#terminal.setProgramStatus({ state: "idle", app: APP, message: title(mirror) });
-	}
-}

package/pi/usage-line.ts

@@ -1,59 +0,0 @@
-/** 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`;
-}