@maxiaochao/pi-toolkit 0.10.21 → 0.10.22

package/packages/herdr-subagents/src/runner.ts

@@ -1,234 +0,0 @@
-import { randomUUID } from "node:crypto";
-import { mkdir, writeFile } from "node:fs/promises";
-import { join } from "node:path";
-import { CliHerdrAutomation, type HerdrAutomation } from "./herdr.ts";
-import { ensureRuntimeIgnored } from "./history.ts";
-import { childEnvironment, liveAgentName, promptLiveAgent } from "./live-agent.ts";
-import { writeJsonAtomic } from "./state.ts";
-import { cleanupRunTab } from "./tab-cleanup.ts";
-import type { OwnerIdentity } from "./ownership.ts";
-import { ownerRecord } from "./ownership.ts";
-import { applyReport, applyTaskStatus, settleAttemptFailure, type PersistedDocument, type PersistedReport, type RunRecord, type TaskRecord } from "./records.ts";
-import { reportProtocolPrompt } from "./protocol.ts";
-import { applyRunStatus } from "./run-status.ts";
-import { isAbortError } from "./errors.ts";
-
-export type HerdrSubagentsDocument = PersistedDocument;
-export type HerdrSubagentsReport = PersistedReport;
-
-export interface HerdrSubagentsOptions {
-	agent?: string;
-	label?: string;
-	model?: string;
-	thinking?: string;
-	tools?: string[];
-	skills?: string[];
-	access?: "read" | "write";
-	systemPrompt?: string;
-	cwd?: string;
-	signal?: AbortSignal;
-	herdr?: HerdrAutomation;
-	owner?: OwnerIdentity;
-	onUpdate?: (result: HerdrSubagentsResult) => void;
-	onCleanupError?: (message: string) => void;
-	stalledWarningSeconds?: number;
-	onStalled?: (message: string) => void;
-}
-
-export interface HerdrSubagentsResult {
-	ok: boolean;
-	status: "starting" | "running" | "blocked" | "completed" | "failed" | "cancelled";
-	summary: string;
-	documents: HerdrSubagentsDocument[];
-	output: string;
-	errorMessage?: string;
-	runId?: string;
-	taskId?: string;
-	tabId?: string;
-	paneId?: string;
-	sessionFile?: string;
-	recordDirectory?: string;
-	model?: string;
-	cwd: string;
-	durationMs: number;
-}
-
-function compactText(report: HerdrSubagentsReport): string {
-	const lines = [report.summary];
-	if (report.documents.length > 0) {
-		lines.push("", "Documents:", ...report.documents.map((document) => `- ${document.description}: ${document.path}`));
-	}
-	if (report.error) lines.push("", `Error: ${report.error}`);
-	return lines.join("\n");
-}
-
-function failure(cwd: string, startedAt: number, errorMessage: string, fields: Partial<HerdrSubagentsResult> = {}): HerdrSubagentsResult {
-	return {
-		ok: false,
-		status: "failed",
-		summary: "Herdr Subagents failed.",
-		documents: [],
-		output: `Herdr Subagents failed.\nError: ${errorMessage}`,
-		errorMessage,
-		cwd,
-		durationMs: Date.now() - startedAt,
-		...fields,
-	};
-}
-
-async function writeJson(path: string, value: unknown): Promise<void> {
-	await writeJsonAtomic(path, value);
-}
-
-export async function runHerdrSubagents(prompt: string, options: HerdrSubagentsOptions = {}): Promise<HerdrSubagentsResult> {
-	const startedAt = Date.now();
-	const cwd = options.cwd ?? process.cwd();
-	if (process.env.HERDR_ENV !== "1" || !process.env.HERDR_WORKSPACE_ID) {
-		return failure(cwd, startedAt, "Herdr Subagents must run from a Pi session inside Herdr.");
-	}
-
-	const runId = randomUUID();
-	const taskId = randomUUID();
-	const agent = options.agent?.trim() || "worker";
-	const label = options.label?.trim() || agent;
-	const runDirectory = join(cwd, ".pi", "herdr-subagents", "runs", `${new Date().toISOString().replace(/[-:.]/g, "").replace("Z", "Z-")}${runId.slice(0, 8)}`);
-	const taskDirectory = join(runDirectory, "tasks", `01-${taskId.slice(0, 8)}`);
-	const sessionFile = join(taskDirectory, "session.jsonl");
-	const systemPromptFile = join(taskDirectory, "system-prompt.md");
-	const reportFile = join(taskDirectory, "report.json");
-	const runFile = join(runDirectory, "run.json");
-	const taskFile = join(taskDirectory, "task.json");
-	const startedAtIso = new Date(startedAt).toISOString();
-	const systemPrompt = reportProtocolPrompt(options.systemPrompt ?? "You are a focused worker subagent.");
-	const runRecord: RunRecord = {
-		id: runId,
-		label,
-		status: "starting",
-		cwd,
-		...ownerRecord(options.owner),
-		stalledWarningSeconds: options.stalledWarningSeconds,
-		taskIds: [taskId],
-		startedAt: startedAtIso,
-	};
-	const taskRecord: TaskRecord = {
-		id: taskId,
-		runId,
-		order: 1,
-		name: label,
-		status: "starting",
-		prompt,
-		agent,
-		...(options.model ? { model: options.model } : {}),
-		...(options.thinking ? { thinking: options.thinking } : {}),
-		...(options.tools ? { tools: options.tools } : {}),
-		...(options.skills ? { skills: options.skills } : {}),
-		...(options.access ? { access: options.access } : {}),
-		systemPrompt,
-		sessionFile,
-		startedAt: startedAtIso,
-	};
-	await ensureRuntimeIgnored(cwd);
-	await mkdir(taskDirectory, { recursive: true, mode: 0o700 });
-	await writeFile(systemPromptFile, systemPrompt, { encoding: "utf8", mode: 0o600 });
-	await Promise.all([writeJson(runFile, runRecord), writeJson(taskFile, taskRecord)]);
-
-	const herdr = options.herdr ?? new CliHerdrAutomation();
-	let tabId: string | undefined;
-	let paneId: string | undefined;
-	let result: HerdrSubagentsResult | undefined;
-	const common = (): Partial<HerdrSubagentsResult> => ({
-		runId,
-		taskId,
-		...(tabId ? { tabId } : {}),
-		...(paneId ? { paneId } : {}),
-		sessionFile,
-		recordDirectory: taskDirectory,
-		...(options.model ? { model: options.model } : {}),
-	});
-	const update = (status: HerdrSubagentsResult["status"], summary: string) => {
-		options.onUpdate?.({
-			ok: false,
-			status,
-			summary,
-			documents: [],
-			output: summary,
-			cwd,
-			durationMs: Date.now() - startedAt,
-			...common(),
-		});
-	};
-	update("starting", "Creating Herdr tab...");
-
-	let attempt: { status: "blocked" | "completed" | "failed" | "cancelled"; report: PersistedReport; error?: unknown };
-	try {
-		const tab = await herdr.createTab({
-			workspaceId: process.env.HERDR_WORKSPACE_ID,
-			cwd,
-			label: `SA · ${label}`,
-			env: childEnvironment(taskDirectory),
-			focus: false,
-			signal: options.signal,
-		});
-		tabId = tab.tabId;
-		paneId = tab.paneId;
-		const paneLabel = `01 · ${label}`;
-		await herdr.renamePane(paneId, paneLabel, options.signal);
-		applyRunStatus(runRecord, "running");
-		Object.assign(runRecord, { tabId, paneId });
-		applyTaskStatus(taskRecord, "running");
-		Object.assign(taskRecord, { tabId, paneId, paneLabel });
-		await Promise.all([writeJson(runFile, runRecord), writeJson(taskFile, taskRecord)]);
-
-		const name = liveAgentName(taskId);
-		taskRecord.agentName = name;
-		await writeJson(taskFile, taskRecord);
-		update("starting", "Starting child Pi in Herdr...");
-		update("running", "Child Pi is working...");
-		const prompted = await promptLiveAgent({ herdr, task: { ...taskRecord, id: taskId, sessionFile }, paneId, prompt, systemPromptFile, agentName: name, signal: options.signal, stalledWarningMs: options.stalledWarningSeconds ? options.stalledWarningSeconds * 1000 : undefined, onStalled: () => options.onStalled?.(`${label} appears stalled; it is still running.`) });
-		taskRecord.currentAttemptId = prompted.attemptId;
-		const report = prompted.report;
-		attempt = { status: applyReport(taskRecord, report), report };
-	} catch (error) {
-		const settled = await settleAttemptFailure(taskRecord, reportFile, {
-			cancelled: isAbortError(error, options.signal),
-			error: error instanceof Error ? error.message : String(error),
-			name: "Herdr Subagents",
-			summary: isAbortError(error, options.signal) ? "Herdr Subagents was cancelled." : "Herdr Subagents failed.",
-		});
-		attempt = { ...settled, error };
-	}
-	try {
-		if (!attempt.error) {
-			const { report, status: settledStatus } = attempt;
-			const completedAt = new Date().toISOString();
-			applyRunStatus(runRecord, settledStatus, { at: completedAt, error: report.error, question: report.question });
-			await Promise.all([writeJson(runFile, runRecord), writeJson(taskFile, taskRecord)]);
-			result = {
-				ok: report.status === "completed",
-				status: settledStatus,
-				summary: report.summary,
-				documents: report.documents,
-				output: compactText(report),
-				...(report.error || report.question ? { errorMessage: report.error ?? report.question } : {}),
-				cwd,
-				durationMs: Date.now() - startedAt,
-				...common(),
-			};
-		} else {
-			const errorMessage = attempt.error instanceof Error ? attempt.error.message : String(attempt.error);
-			const cancelled = attempt.status === "cancelled";
-			const completedAt = new Date().toISOString();
-			applyRunStatus(runRecord, cancelled ? "cancelled" : "failed", { at: completedAt, error: errorMessage });
-			await Promise.all([writeJson(runFile, runRecord), writeJson(taskFile, taskRecord)]);
-			result = cancelled
-				? { ...failure(cwd, startedAt, errorMessage, common()), status: "cancelled", summary: "Herdr Subagents cancelled.", output: `Herdr Subagents cancelled.\nError: ${errorMessage}` }
-				: failure(cwd, startedAt, errorMessage, common());
-		}
-	} finally {
-		if (tabId && result?.status !== "blocked") {
-			await cleanupRunTab({ herdr, tabId, runFile, runRecord, signal: options.signal, onError: options.onCleanupError });
-		}
-	}
-
-	return result ?? failure(cwd, startedAt, "Herdr Subagents ended without a result.", common());
-}