ultimate-pi 0.6.1 → 0.8.0

package/.pi/agents/harness/trace-librarian.md

@@ -1,32 +1,39 @@
 ---
 description: Harness trace librarian for run replay, artifact indexing, and forensics summaries.
-tools: read, bash, grep, find, ls
+tools: read, grep, find, ls
+extensions: false
 thinking: medium
 max_turns: 20
+inherit_context: false
 ---
 
 You are the Harness Trace Librarian.
 
 ## Mission
 
-Maintain replayable trace narratives and artifact integrity checks.
+Maintain replayable trace narratives and artifact integrity checks from `HarnessSpawnContext` (`run_dir`, optional `--phase` filter).
 
 ## Process
 
-1. Gather trace and artifact records by run ID and phase.
-2. Index artifacts by run and phase using stable, machine-readable references.
+1. Gather trace and artifact records from `.pi/harness/runs/<run_id>/` and spawn context paths.
+2. Index artifacts by phase: `plan`, `execute`, `evaluate`, `adversary`, `merge`.
 3. Surface missing artifacts required by strict pre-PR gates.
 4. Produce concise forensic summaries with evidence pointers and replay instructions.
 
 ## Guardrails
 
-- Do not overthink straightforward indexing tasks; prioritize completeness and consistency.
-- Only report artifacts relevant to the requested run/phases.
-- Never speculate about missing artifacts without checking canonical run locations.
-- Keep references stable and machine-readable.
+- Read-only — no mutations.
+- Only report artifacts for the requested run/phases.
+- Never speculate without checking canonical run locations.
 
 ## Output
 
-- Timeline summary.
-- Artifact manifest and integrity gaps.
-- Replay instructions.
+```json
+{
+  "trace_completeness": "complete",
+  "timeline_summary": "…",
+  "artifact_index": {},
+  "missing_artifacts": [],
+  "next_command_hint": "/harness-review"
+}
+```

package/.pi/agents/pi-pi/ext-expert.md

@@ -19,7 +19,7 @@ You are an extensions expert for the Pi coding agent. You know EVERYTHING about
 - Flags via `pi.registerFlag()`
 - State management via tool result details and `pi.appendEntry()`
 - Custom rendering via renderCall/renderResult
-- Available imports: `@mariozechner/pi-coding-agent`, `@sinclair/typebox`, `@mariozechner/pi-ai` (StringEnum), `@mariozechner/pi-tui`
+- Available imports: `@earendil-works/pi-coding-agent`, `@sinclair/typebox`, `@earendil-works/pi-ai` (StringEnum), `@earendil-works/pi-tui`
 - System prompt override via before_agent_start
 - Context manipulation via context event
 - Tool blocking and result modification

package/.pi/agents/pi-pi/keybinding-expert.md

@@ -87,7 +87,7 @@ This is CRITICAL for building extensions that work on macOS:
 - When a reserved action is remapped away from a key, that key becomes available for extensions
 - The conflict check uses EFFECTIVE keybindings (after user remaps), not defaults
 
-### Key Helper (from @mariozechner/pi-tui)
+### Key Helper (from @earendil-works/pi-tui)
 
 - `Key.ctrl("x")` → `"ctrl+x"`
 - `Key.shift("tab")` → `"shift+tab"`

package/.pi/agents/pi-pi/tui-expert.md

@@ -19,7 +19,7 @@ You are a TUI (Terminal User Interface) expert for the Pi coding agent. You know
 - wantsKeyRelease? — for Kitty protocol key release events
 - invalidate() — clear cached render state
 
-### Built-in Components (from @mariozechner/pi-tui)
+### Built-in Components (from @earendil-works/pi-tui)
 
 - Text: multi-line text with word wrapping, paddingX, paddingY, background function
 - Box: container with padding and background color
@@ -30,7 +30,7 @@ You are a TUI (Terminal User Interface) expert for the Pi coding agent. You know
 - SelectList: selection dialog with theme, onSelect/onCancel
 - SettingsList: toggle settings with theme
 
-### From @mariozechner/pi-coding-agent
+### From @earendil-works/pi-coding-agent
 
 - DynamicBorder: border with color function — ALWAYS type the param: `(s: string) => theme.fg("accent", s)`
 - BorderedLoader: spinner with abort support
@@ -93,7 +93,7 @@ Then read the fetched file. Also search the local codebase for existing TUI comp
 ## How to Respond
 
 - Provide COMPLETE, WORKING component code
-- Include all imports from @mariozechner/pi-tui and @mariozechner/pi-coding-agent
+- Include all imports from @earendil-works/pi-tui and @earendil-works/pi-coding-agent
 - Show the ctx.ui.custom() wrapper for interactive components
 - Handle invalidation properly for theme changes
 - Include keyboard input handling where relevant

package/.pi/extensions/00-ultimate-pi-system-prompt.ts

@@ -0,0 +1,194 @@
+/**
+ * System prompt for ultimate-pi end users:
+ * 1. Workspace override: `cwd/.pi/system.md` (lowercase)
+ * 2. Package default: `<ultimate-pi>/.pi/SYSTEM.md` (via package root resolution)
+ *
+ * Does not copy or seed workspace files. Uses `before_agent_start` →
+ * `systemPrompt` replacement (runs early via `00-` prefix so harness extensions
+ * can still append).
+ */
+
+import { existsSync, readFileSync, statSync } from "node:fs";
+import { join } from "node:path";
+import type {
+	BuildSystemPromptOptions,
+	ExtensionAPI,
+} from "@earendil-works/pi-coding-agent";
+import { formatSkillsForPrompt } from "@earendil-works/pi-coding-agent";
+import { resolveHarnessAsset } from "./lib/harness-paths.js";
+
+// @ts-expect-error pi extensions run as ESM
+const MODULE_URL = import.meta.url;
+
+/** Workspace override path (lowercase only — not Pi's SYSTEM.md discovery). */
+const WORKSPACE_SYSTEM_MD = join(".pi", "system.md");
+
+/** Mirror Pi `buildSystemPrompt` customPrompt branch (see system-prompt.js). */
+function buildFromCustomPrompt(
+	customPrompt: string,
+	options: BuildSystemPromptOptions,
+): string {
+	const {
+		appendSystemPrompt,
+		cwd,
+		contextFiles: providedContextFiles,
+		skills: providedSkills,
+		selectedTools,
+	} = options;
+	const promptCwd = cwd.replace(/\\/g, "/");
+	const now = new Date();
+	const year = now.getFullYear();
+	const month = String(now.getMonth() + 1).padStart(2, "0");
+	const day = String(now.getDate()).padStart(2, "0");
+	const date = `${year}-${month}-${day}`;
+	const appendSection = appendSystemPrompt ? `\n\n${appendSystemPrompt}` : "";
+	const contextFiles = providedContextFiles ?? [];
+	const skills = providedSkills ?? [];
+	let prompt = customPrompt;
+	if (appendSection) {
+		prompt += appendSection;
+	}
+	if (contextFiles.length > 0) {
+		prompt += "\n\n# Project Context\n\n";
+		prompt += "Project-specific instructions and guidelines:\n\n";
+		for (const { path: filePath, content } of contextFiles) {
+			prompt += `## ${filePath}\n\n${content}\n\n`;
+		}
+	}
+	const customPromptHasRead = !selectedTools || selectedTools.includes("read");
+	if (customPromptHasRead && skills.length > 0) {
+		prompt += formatSkillsForPrompt(skills);
+	}
+	prompt += `\nCurrent date: ${date}`;
+	prompt += `\nCurrent working directory: ${promptCwd}`;
+	return prompt;
+}
+
+function isDisabled(): boolean {
+	const raw = process.env.ULTIMATE_PI_SYSTEM_PROMPT?.trim().toLowerCase();
+	return raw === "0" || raw === "false" || raw === "off" || raw === "no";
+}
+
+function workspaceSystemPromptPath(cwd: string): string {
+	return join(cwd, WORKSPACE_SYSTEM_MD);
+}
+
+function packageSystemPromptPath(): string {
+	return resolveHarnessAsset(MODULE_URL, ".pi", "SYSTEM.md");
+}
+
+type PromptSource = "workspace" | "package";
+
+function resolveSystemPromptPath(cwd: string): {
+	path: string;
+	source: PromptSource;
+} | null {
+	const workspacePath = workspaceSystemPromptPath(cwd);
+	if (existsSync(workspacePath)) {
+		return { path: workspacePath, source: "workspace" };
+	}
+	const packagePath = packageSystemPromptPath();
+	if (existsSync(packagePath)) {
+		return { path: packagePath, source: "package" };
+	}
+	return null;
+}
+
+interface PromptCache {
+	path: string;
+	source: PromptSource;
+	mtimeMs: number;
+	content: string;
+}
+
+function readPromptFile(path: string): string | null {
+	try {
+		const content = readFileSync(path, "utf-8").trim();
+		return content.length > 0 ? content : null;
+	} catch {
+		return null;
+	}
+}
+
+export default function ultimatePiSystemPrompt(pi: ExtensionAPI) {
+	if (isDisabled()) {
+		return;
+	}
+
+	let cache: PromptCache | null = null;
+	let warnedMissing = false;
+
+	const loadSystemPrompt = (
+		cwd: string,
+	): { content: string; path: string; source: PromptSource } | null => {
+		const resolved = resolveSystemPromptPath(cwd);
+		if (!resolved) {
+			return null;
+		}
+		try {
+			const { mtimeMs } = statSync(resolved.path);
+			if (
+				cache &&
+				cache.path === resolved.path &&
+				cache.source === resolved.source &&
+				cache.mtimeMs === mtimeMs
+			) {
+				return {
+					content: cache.content,
+					path: cache.path,
+					source: cache.source,
+				};
+			}
+			const content = readPromptFile(resolved.path);
+			if (!content) {
+				return null;
+			}
+			cache = {
+				path: resolved.path,
+				source: resolved.source,
+				mtimeMs,
+				content,
+			};
+			return { content, path: resolved.path, source: resolved.source };
+		} catch {
+			return null;
+		}
+	};
+
+	const invalidateCache = () => {
+		cache = null;
+		warnedMissing = false;
+	};
+
+	pi.on("session_start", () => {
+		invalidateCache();
+	});
+
+	pi.on("before_agent_start", async (event, ctx) => {
+		const cwd = ctx.cwd ?? process.cwd();
+		const loaded = loadSystemPrompt(cwd);
+
+		if (!loaded) {
+			if (!warnedMissing) {
+				const workspacePath = workspaceSystemPromptPath(cwd);
+				const pkgPath = packageSystemPromptPath();
+				ctx.ui.notify(
+					`[ultimate-pi] No system prompt found.\n` +
+						`  Workspace override: ${workspacePath}\n` +
+						`  Package default: ${pkgPath}\n` +
+						`Using Pi default system prompt.`,
+					"warning",
+				);
+				warnedMissing = true;
+			}
+			return;
+		}
+
+		return {
+			systemPrompt: buildFromCustomPrompt(
+				loaded.content,
+				event.systemPromptOptions,
+			),
+		};
+	});
+}

package/.pi/extensions/budget-guard.ts

@@ -7,7 +7,8 @@
 
 import { appendFile, mkdir, readFile } from "node:fs/promises";
 import { join } from "node:path";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { getRunIdFromSession } from "../lib/harness-run-context.js";
 
 type HarnessPhase = "plan" | "execute" | "evaluate" | "adversary" | "merge";
 
@@ -129,8 +130,15 @@ function getPolicyContext(ctx: {
 	return { phase: null, budgetBypass: false };
 }
 
-function getRunId(ctx: { sessionManager: { getSessionId(): string } }): string {
-	return ctx.sessionManager.getSessionId();
+function getRunId(ctx: {
+	sessionManager: { getEntries(): unknown[]; getSessionId(): string };
+}): string {
+	return (
+		getRunIdFromSession(
+			ctx.sessionManager.getEntries(),
+			ctx.sessionManager.getSessionId(),
+		) ?? ctx.sessionManager.getSessionId()
+	);
 }
 
 async function readDebateCapsFromSchema(): Promise<{

package/.pi/extensions/custom-footer.ts

@@ -9,9 +9,14 @@ import type {
 	ExtensionAPI,
 	ExtensionContext,
 	ThemeColor,
-} from "@mariozechner/pi-coding-agent";
-import type { TUI } from "@mariozechner/pi-tui";
-import { Box, Text, truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+} from "@earendil-works/pi-coding-agent";
+import type { TUI } from "@earendil-works/pi-tui";
+import {
+	Box,
+	Text,
+	truncateToWidth,
+	visibleWidth,
+} from "@earendil-works/pi-tui";
 
 // ── router decision reader ──────────────────────────────────────────
 

package/.pi/extensions/custom-header.ts

@@ -6,8 +6,8 @@
  * doubling vertical resolution in the same terminal footprint.
  */
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { truncateToWidth } from "@mariozechner/pi-tui";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { truncateToWidth } from "@earendil-works/pi-tui";
 import * as JimpModule from "jimp";
 import { resolveHarnessAsset } from "./lib/harness-paths.js";
 

package/.pi/extensions/debate-orchestrator.ts

@@ -16,7 +16,8 @@
 
 import { appendFile, mkdir, readFile, writeFile } from "node:fs/promises";
 import { join } from "node:path";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { getRunIdFromSession } from "../lib/harness-run-context.js";
 
 type DebateParticipant =
 	| "EvaluatorAgent"
@@ -92,8 +93,15 @@ async function ensureDebatesDir(): Promise<void> {
 	await mkdir(DEBATES_DIR, { recursive: true });
 }
 
-function getRunId(ctx: { sessionManager: { getSessionId(): string } }): string {
-	return ctx.sessionManager.getSessionId();
+function getRunId(ctx: {
+	sessionManager: { getEntries(): unknown[]; getSessionId(): string };
+}): string {
+	return (
+		getRunIdFromSession(
+			ctx.sessionManager.getEntries(),
+			ctx.sessionManager.getSessionId(),
+		) ?? ctx.sessionManager.getSessionId()
+	);
 }
 
 async function readRoundCapsFromSchema(): Promise<{

package/.pi/extensions/dotenv-loader.ts

@@ -13,7 +13,7 @@
 
 import { existsSync, readFileSync } from "node:fs";
 import { resolve } from "node:path";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 
 // ── .env parser ──────────────────────────────────────────────────
 

package/.pi/extensions/drift-monitor.ts

@@ -4,7 +4,7 @@
  * Emits harness-drift-report custom entries for harness-telemetry + observation bus.
  */
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 
 type HarnessPhase = "plan" | "execute" | "evaluate" | "adversary" | "merge";
 

package/.pi/extensions/harness-ask-user.ts

@@ -3,7 +3,7 @@
  * Design references: pi-ask-user, @pi-unipi/ask-user, rpiv-ask-user-question (not vendored).
  */
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { runAskDialog } from "./lib/ask-user/dialog.js";
 import { runAskFallback } from "./lib/ask-user/fallback.js";
 import { renderAskCall, renderAskResult } from "./lib/ask-user/render.js";

package/.pi/extensions/harness-live-widget.ts

@@ -1,7 +1,7 @@
 import type {
 	ExtensionAPI,
 	ExtensionContext,
-} from "@mariozechner/pi-coding-agent";
+} from "@earendil-works/pi-coding-agent";
 import {
 	type HarnessUiState,
 	HarnessUiStateStore,
@@ -285,9 +285,16 @@ class HarnessWidgetComponent {
 		const toolDisplay = this.inFlight.lastToolName
 			? `${this.inFlight.toolCount}:${this.inFlight.lastToolName}`
 			: String(this.inFlight.toolCount);
-		const traceDisplay = this.state.traceRunId ?? "n/a";
+		const nextDisplay =
+			this.state.nextRecommendedCommand != null
+				? this.state.nextRecommendedCommand.length > 36
+					? `${this.state.nextRecommendedCommand.slice(0, 33)}...`
+					: this.state.nextRecommendedCommand
+				: null;
 		const row3Left = `${planFlag} ${reviewFlag} ${budgetFlag} ${testsFlag}`;
-		const row3Right = `${theme.fg("dim", "inFlight:")}${theme.fg("accent", toolDisplay)} ${theme.fg("dim", "trace:")}${theme.fg("muted", traceDisplay)}`;
+		const row3Right = nextDisplay
+			? `${theme.fg("dim", "inFlight:")}${theme.fg("accent", toolDisplay)} ${theme.fg("dim", "next:")}${theme.fg("accent", nextDisplay)}`
+			: `${theme.fg("dim", "inFlight:")}${theme.fg("accent", toolDisplay)}`;
 		const row3 = composeZones(row3Left, row3Right, rowWidth);
 
 		const lines: string[] = [truncateToWidth(row1, rowWidth)];
@@ -353,7 +360,7 @@ export default function harnessLiveWidget(pi: ExtensionAPI) {
 			policyDecision: state.policyDecision,
 			consensusDelta: state.consensusDelta,
 			severity: state.severity,
-			traceRunId: state.traceRunId,
+			nextRecommendedCommand: state.nextRecommendedCommand,
 			inFlight,
 		});
 	}