@oh-my-pi/pi-coding-agent 13.14.2 → 13.15.3

package/src/autoresearch/types.ts

@@ -0,0 +1,218 @@
+import type { AgentToolResult } from "@oh-my-pi/pi-agent-core";
+import type { ExtensionAPI, ExtensionContext } from "../extensibility/extensions";
+import type { SessionEntry } from "../session/session-manager";
+import type { TruncationResult } from "../session/streaming-output";
+
+export type MetricDirection = "lower" | "higher";
+export type ExperimentStatus = "keep" | "discard" | "crash" | "checks_failed";
+
+export type ASIValue = string | number | boolean | null | ASIValue[] | { [key: string]: ASIValue };
+
+export interface ASIData {
+	[key: string]: ASIValue;
+}
+
+export interface NumericMetricMap {
+	[key: string]: number;
+}
+
+export interface MetricDef {
+	name: string;
+	unit: string;
+}
+
+export interface AutoresearchBenchmarkContract {
+	command: string | null;
+	primaryMetric: string | null;
+	metricUnit: string;
+	direction: MetricDirection | null;
+	secondaryMetrics: string[];
+}
+
+export interface AutoresearchContract {
+	benchmark: AutoresearchBenchmarkContract;
+	scopePaths: string[];
+	offLimits: string[];
+	constraints: string[];
+}
+
+export interface ExperimentResult {
+	runNumber: number | null;
+	commit: string;
+	metric: number;
+	metrics: NumericMetricMap;
+	status: ExperimentStatus;
+	description: string;
+	timestamp: number;
+	segment: number;
+	confidence: number | null;
+	asi?: ASIData;
+}
+
+export interface ExperimentState {
+	results: ExperimentResult[];
+	bestMetric: number | null;
+	bestDirection: MetricDirection;
+	metricName: string;
+	metricUnit: string;
+	secondaryMetrics: MetricDef[];
+	name: string | null;
+	currentSegment: number;
+	maxExperiments: number | null;
+	confidence: number | null;
+	benchmarkCommand: string | null;
+	scopePaths: string[];
+	offLimits: string[];
+	constraints: string[];
+	segmentFingerprint: string | null;
+}
+
+export interface RunExperimentProgressDetails {
+	phase: "running";
+	elapsed: string;
+	truncation?: TruncationResult;
+	fullOutputPath?: string;
+	runDirectory?: string;
+}
+
+export interface RunDetails {
+	runNumber: number;
+	runDirectory: string;
+	benchmarkLogPath: string;
+	checksLogPath?: string;
+	command: string;
+	exitCode: number | null;
+	durationSeconds: number;
+	passed: boolean;
+	crashed: boolean;
+	timedOut: boolean;
+	tailOutput: string;
+	checksPass: boolean | null;
+	checksTimedOut: boolean;
+	checksOutput: string;
+	checksDuration: number;
+	parsedMetrics: NumericMetricMap | null;
+	parsedPrimary: number | null;
+	parsedAsi: ASIData | null;
+	metricName: string;
+	metricUnit: string;
+	truncation?: TruncationResult;
+	fullOutputPath?: string;
+}
+
+export interface LogDetails {
+	experiment: ExperimentResult;
+	state: ExperimentState;
+	wallClockSeconds: number | null;
+}
+
+export interface ChecksResult {
+	pass: boolean;
+	output: string;
+	duration: number;
+}
+
+export interface PendingRunSummary {
+	checksDurationSeconds: number | null;
+	checksPass: boolean | null;
+	checksTimedOut: boolean;
+	command: string;
+	durationSeconds: number | null;
+	parsedAsi: ASIData | null;
+	parsedMetrics: NumericMetricMap | null;
+	parsedPrimary: number | null;
+	passed: boolean;
+	runDirectory: string;
+	runNumber: number;
+}
+
+export interface RunningExperiment {
+	startedAt: number;
+	command: string;
+	runDirectory: string;
+	runNumber: number;
+}
+
+export interface AutoresearchRuntime {
+	autoresearchMode: boolean;
+	autoResumeArmed: boolean;
+	dashboardExpanded: boolean;
+	lastAutoResumePendingRunNumber: number | null;
+	lastRunChecks: ChecksResult | null;
+	lastRunDuration: number | null;
+	lastRunAsi: ASIData | null;
+	lastRunArtifactDir: string | null;
+	lastRunNumber: number | null;
+	lastRunSummary: PendingRunSummary | null;
+	runningExperiment: RunningExperiment | null;
+	state: ExperimentState;
+	goal: string | null;
+}
+
+export interface AutoresearchConfig {
+	maxIterations?: number;
+	workingDir?: string;
+}
+
+export interface AutoresearchJsonConfigEntry {
+	type: "config";
+	name?: string;
+	metricName?: string;
+	metricUnit?: string;
+	bestDirection?: MetricDirection;
+	benchmarkCommand?: string;
+	secondaryMetrics?: string[];
+	scopePaths?: string[];
+	offLimits?: string[];
+	constraints?: string[];
+	segmentFingerprint?: string;
+}
+
+export interface AutoresearchJsonRunEntry {
+	run?: number;
+	commit?: string;
+	metric?: number;
+	metrics?: NumericMetricMap;
+	status?: ExperimentStatus;
+	description?: string;
+	timestamp?: number;
+	confidence?: number | null;
+	asi?: ASIData;
+}
+
+export interface ReconstructedExperimentData {
+	hasLog: boolean;
+	state: ExperimentState;
+}
+
+export interface AutoresearchControlEntryData {
+	mode: "on" | "off" | "clear";
+	goal?: string;
+}
+
+export interface ReconstructedControlState {
+	autoresearchMode: boolean;
+	goal: string | null;
+	lastMode: AutoresearchControlEntryData["mode"] | null;
+}
+
+export interface RuntimeStore {
+	clear(sessionKey: string): void;
+	ensure(sessionKey: string): AutoresearchRuntime;
+}
+
+export interface DashboardController {
+	clear(ctx: ExtensionContext): void;
+	requestRender(): void;
+	showOverlay(ctx: ExtensionContext, runtime: AutoresearchRuntime): Promise<void>;
+	updateWidget(ctx: ExtensionContext, runtime: AutoresearchRuntime): void;
+}
+
+export interface AutoresearchToolFactoryOptions {
+	dashboard: DashboardController;
+	getRuntime(ctx: ExtensionContext): AutoresearchRuntime;
+	pi: ExtensionAPI;
+}
+
+export type AutoresearchToolResult<TDetails> = AgentToolResult<TDetails>;
+export type SessionEntries = SessionEntry[];

package/src/cli/args.ts

@@ -7,7 +7,7 @@ import chalk from "chalk";
 import { parseEffort } from "../thinking";
 import { BUILTIN_TOOLS } from "../tools";
 
-export type Mode = "text" | "json" | "rpc";
+export type Mode = "text" | "json" | "rpc" | "acp";
 
 export interface Args {
 	cwd?: string;
@@ -28,6 +28,8 @@ export interface Args {
 	mode?: Mode;
 	noSession?: boolean;
 	sessionDir?: string;
+	providerSessionId?: string;
+	fork?: string;
 	models?: string[];
 	tools?: string[];
 	noTools?: boolean;
@@ -67,7 +69,7 @@ export function parseArgs(args: string[], extensionFlags?: Map<string, { type: "
 			result.allowHome = true;
 		} else if (arg === "--mode" && i + 1 < args.length) {
 			const mode = args[++i];
-			if (mode === "text" || mode === "json" || mode === "rpc") {
+			if (mode === "text" || mode === "json" || mode === "rpc" || mode === "acp") {
 				result.mode = mode;
 			}
 		} else if (arg === "--continue" || arg === "-c") {
@@ -79,6 +81,8 @@ export function parseArgs(args: string[], extensionFlags?: Map<string, { type: "
 			} else {
 				result.resume = true;
 			}
+		} else if (arg === "--fork" && i + 1 < args.length) {
+			result.fork = args[++i];
 		} else if (arg === "--provider" && i + 1 < args.length) {
 			result.provider = args[++i];
 		} else if (arg === "--model" && i + 1 < args.length) {
@@ -95,6 +99,8 @@ export function parseArgs(args: string[], extensionFlags?: Map<string, { type: "
 			result.systemPrompt = args[++i];
 		} else if (arg === "--append-system-prompt" && i + 1 < args.length) {
 			result.appendSystemPrompt = args[++i];
+		} else if (arg === "--provider-session-id" && i + 1 < args.length) {
+			result.providerSessionId = args[++i];
 		} else if (arg === "--no-session") {
 			result.noSession = true;
 		} else if (arg === "--session-dir" && i + 1 < args.length) {

package/src/cli/initial-message.ts

@@ -0,0 +1,58 @@
+import type { ImageContent } from "@oh-my-pi/pi-ai";
+import type { Args } from "./args";
+
+export interface InitialMessageInput {
+	parsed: Args;
+	fileText?: string;
+	fileImages?: ImageContent[];
+	stdinContent?: string;
+}
+
+export interface InitialMessageResult {
+	initialMessage?: string;
+	initialImages?: ImageContent[];
+}
+
+/**
+ * Combine stdin content, @file text, and the first CLI message into a single
+ * initial prompt for non-interactive mode.
+ */
+export function buildInitialMessage({
+	parsed,
+	fileText,
+	fileImages,
+	stdinContent,
+}: InitialMessageInput): InitialMessageResult {
+	const hasInitialContext = stdinContent !== undefined || fileText !== undefined || (fileImages?.length ?? 0) > 0;
+	if (!hasInitialContext) {
+		return {
+			initialImages: undefined,
+		};
+	}
+
+	let body = "";
+	if (fileText !== undefined) {
+		body += fileText;
+	}
+
+	if (parsed.messages.length > 0) {
+		body += parsed.messages[0];
+		parsed.messages.shift();
+	}
+
+	const initialMessage =
+		stdinContent !== undefined
+			? body.length > 0
+				? `${stdinContent}\n${body}`
+				: stdinContent
+			: body.length > 0
+				? body
+				: fileImages && fileImages.length > 0
+					? ""
+					: undefined;
+
+	return {
+		initialMessage,
+		initialImages: fileImages && fileImages.length > 0 ? fileImages : undefined,
+	};
+}