pi-antiloop 1.0.0 → 1.1.0

package/src/types.ts

@@ -0,0 +1,84 @@
+// antiloop — shared types. Zero runtime cost.
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+export interface AntiloopConfig {
+	enabled: boolean;
+	warningThreshold: number;
+	forceBreakThreshold: number;
+	abortThreshold: number;
+	similarityThreshold: number;
+	/**
+	 * How close tool-call arguments must be (0..1) to count as the SAME call.
+	 * High by default: long bash commands share scaffolding (env setup, flags,
+	 * paths) even when they are different operations — a parameter sweep or a
+	 * retry after a fix is NOT a loop. Only near-identical repeats qualify.
+	 */
+	toolSimilarityThreshold: number;
+	/**
+	 * How many PRIOR occurrences of a near-identical tool-call set must exist
+	 * in the window before a tool loop is flagged. 2 = the same call seen 3x.
+	 */
+	minToolRepeatCount: number;
+	/**
+	 * Result-aware veto: when both runs have a captured result, the normalized
+	 * result tails must be at least this similar for the pair to count as a
+	 * loop. Same command + different outcome = progress, not a loop.
+	 */
+	resultSimilarityThreshold: number;
+	detectToolLoops: boolean;
+	detectThinkingLoops: boolean;
+	detectTextLoops: boolean;
+	notifyOnDetection: boolean;
+	maxHistoryEntries: number;
+	detectionWindow: number;
+}
+
+export type LoopKind = "text" | "tool" | "thinking" | "structural";
+
+export interface LoopDetection {
+	type: LoopKind;
+	similarity: number;
+	messageIndices: number[];
+	description: string;
+	timestamp: number;
+}
+
+export interface TrackedToolCall {
+	name: string;
+	args: string;
+	/** toolCallId — used to attach the execution result at turn_end. */
+	id?: string;
+	/**
+	 * Normalized tail of the tool result ("err|" / "ok|" prefix + last chars).
+	 * Only set when the turn completed and a result was captured. When both
+	 * sides of a comparison have one, a mismatch vetoes the loop.
+	 */
+	result?: string;
+}
+
+export interface TrackedMessage {
+	content: string;
+	thinking?: string;
+	toolCalls?: TrackedToolCall[];
+	timestamp: number;
+	turnIndex: number;
+}
+
+export interface AntiloopState {
+	recentMessages: TrackedMessage[];
+	detections: LoopDetection[];
+	currentLevel: 0 | 1 | 2 | 3;
+	consecutiveDetections: number;
+	inForcedBreak: boolean;
+	totalDetections: number;
+	lastUserMessageTime: number;
+	/** turnIndex of the last tracked message detection already ran on. */
+	lastDetectedTurnIndex: number;
+}
+
+export interface Runtime {
+	config: AntiloopConfig;
+	state: AntiloopState;
+	pendingIntervention: string | null;
+	updateStatus(ctx: ExtensionContext): void;
+}

package/src/ui.ts

@@ -0,0 +1,23 @@
+// antiloop — UI helpers. Lazy-loaded with command handler.
+
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+export function formatDuration(ms: number): string {
+	if (ms < 1000) return `${ms}ms`;
+	if (ms < 60_000) return `${Math.round(ms / 1000)}s`;
+	if (ms < 3_600_000) return `${Math.round(ms / 60_000)}m`;
+	return `${Math.round(ms / 3_600_000)}h`;
+}
+
+// Labeled select: returns the matched `value` from `items`, or undefined if cancelled.
+export async function selectFrom<T>(
+	ctx: ExtensionContext,
+	title: string,
+	items: Array<{ value: T; label: string; description?: string }>,
+): Promise<T | undefined> {
+	const strings = items.map((it) => (it.description ? `${it.label} — ${it.description}` : it.label));
+	const picked = await ctx.ui.select(title, strings);
+	if (picked === undefined) return undefined;
+	const idx = strings.indexOf(picked);
+	return idx >= 0 ? items[idx].value : undefined;
+}