@oh-my-pi/pi-coding-agent 15.0.2 → 15.1.1

package/src/session/compaction/errors.ts

@@ -1,31 +0,0 @@
-/**
- * Compaction error types.
- *
- * `CompactionCancelledError` is the canonical signal raised when a compaction
- * is explicitly aborted — operator Esc, extension hook returning `cancel`,
- * programmatic `session.abortCompaction()` call, or any other deliberate
- * abort source. Downstream callers (e.g. `executeCompaction`) discriminate
- * cancellation from other failures via `instanceof CompactionCancelledError`
- * rather than introspecting error messages or `name` fields — the typed
- * sentinel makes classification source-agnostic and refactor-stable.
- */
-
-export class CompactionCancelledError extends Error {
-	readonly name = "CompactionCancelledError" as const;
-
-	constructor(message = "Compaction cancelled") {
-		super(message);
-	}
-}
-
-/**
- * Outcome of a compaction attempt, surfaced by `CommandController.executeCompaction`
- * so callers (e.g. the plan-mode approval flow) can distinguish a deliberate abort
- * from an unrelated failure.
- *
- *   "ok"        — compaction completed; transcript was summarized.
- *   "cancelled" — `CompactionCancelledError` was raised. Operator Esc, extension
- *                 hook, programmatic abort — all source-agnostic.
- *   "failed"    — any other rejection from `session.compact()`.
- */
-export type CompactionOutcome = "ok" | "cancelled" | "failed";

package/src/session/compaction/index.ts

@@ -1,8 +0,0 @@
-/**
- * Compaction and summarization utilities.
- */
-
-export * from "./branch-summarization";
-export * from "./compaction";
-export * from "./errors";
-export * from "./utils";

package/src/session/compaction/pruning.ts

@@ -1,91 +0,0 @@
-/**
- * Tool output pruning utilities for compaction.
- */
-import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { ToolResultMessage } from "@oh-my-pi/pi-ai";
-import type { SessionEntry, SessionMessageEntry } from "../session-manager";
-import { estimateTokens } from "./compaction";
-
-export interface PruneConfig {
-	/** Keep the most recent tool output tokens intact. */
-	protectTokens: number;
-	/** Only prune if total savings meets this threshold. */
-	minimumSavings: number;
-	/** Tool names that should never be pruned. */
-	protectedTools: string[];
-}
-
-export const DEFAULT_PRUNE_CONFIG: PruneConfig = {
-	protectTokens: 40_000,
-	minimumSavings: 20_000,
-	protectedTools: ["skill", "read"],
-};
-
-export interface PruneResult {
-	prunedCount: number;
-	tokensSaved: number;
-}
-
-function createPrunedNotice(tokens: number): string {
-	return `[Output truncated - ${tokens} tokens]`;
-}
-
-function getToolResultMessage(entry: SessionEntry): ToolResultMessage | undefined {
-	if (entry.type !== "message") return undefined;
-	const message = entry.message as AgentMessage;
-	if (message.role !== "toolResult") return undefined;
-	return message as ToolResultMessage;
-}
-
-function estimatePrunedSavings(tokens: number): number {
-	const noticeTokens = Math.ceil(createPrunedNotice(tokens).length / 4);
-	return Math.max(0, tokens - noticeTokens);
-}
-
-export function pruneToolOutputs(entries: SessionEntry[], config: PruneConfig = DEFAULT_PRUNE_CONFIG): PruneResult {
-	let accumulatedTokens = 0;
-	let tokensSaved = 0;
-	let prunedCount = 0;
-
-	const candidates: Array<{ entry: SessionMessageEntry; tokens: number }> = [];
-
-	for (let i = entries.length - 1; i >= 0; i--) {
-		const entry = entries[i];
-		const message = getToolResultMessage(entry);
-		if (!message) continue;
-
-		const tokens = estimateTokens(message as AgentMessage);
-		const isProtected = config.protectedTools.includes(message.toolName);
-
-		if (message.prunedAt !== undefined) {
-			accumulatedTokens += tokens;
-			continue;
-		}
-
-		if (accumulatedTokens < config.protectTokens || isProtected) {
-			accumulatedTokens += tokens;
-			continue;
-		}
-
-		candidates.push({ entry: entry as SessionMessageEntry, tokens });
-		accumulatedTokens += tokens;
-	}
-
-	for (const candidate of candidates) {
-		tokensSaved += estimatePrunedSavings(candidate.tokens);
-	}
-
-	if (tokensSaved < config.minimumSavings || candidates.length === 0) {
-		return { prunedCount: 0, tokensSaved: 0 };
-	}
-
-	const prunedAt = Date.now();
-	for (const candidate of candidates) {
-		const message = candidate.entry.message as ToolResultMessage;
-		message.content = [{ type: "text", text: createPrunedNotice(candidate.tokens) }];
-		message.prunedAt = prunedAt;
-		prunedCount++;
-	}
-
-	return { prunedCount, tokensSaved };
-}

package/src/session/compaction/utils.ts

@@ -1,184 +0,0 @@
-/**
- * Shared utilities for compaction and branch summarization.
- */
-import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { Message } from "@oh-my-pi/pi-ai";
-import { prompt } from "@oh-my-pi/pi-utils";
-import fileOperationsTemplate from "../../prompts/system/file-operations.md" with { type: "text" };
-import summarizationSystemPrompt from "../../prompts/system/summarization-system.md" with { type: "text" };
-
-// ============================================================================
-// File Operation Tracking
-// ============================================================================
-
-export interface FileOperations {
-	read: Set<string>;
-	written: Set<string>;
-	edited: Set<string>;
-}
-
-export function createFileOps(): FileOperations {
-	return {
-		read: new Set(),
-		written: new Set(),
-		edited: new Set(),
-	};
-}
-
-/**
- * Extract file operations from tool calls in an assistant message.
- */
-export function extractFileOpsFromMessage(message: AgentMessage, fileOps: FileOperations): void {
-	if (message.role !== "assistant") return;
-	if (!("content" in message) || !Array.isArray(message.content)) return;
-
-	for (const block of message.content) {
-		if (typeof block !== "object" || block === null) continue;
-		if (!("type" in block) || block.type !== "toolCall") continue;
-		if (!("arguments" in block) || !("name" in block)) continue;
-
-		const args = block.arguments as Record<string, unknown> | undefined;
-		if (!args) continue;
-
-		const path = typeof args.path === "string" ? args.path : undefined;
-		if (!path) continue;
-
-		switch (block.name) {
-			case "read":
-				fileOps.read.add(path);
-				break;
-			case "write":
-				fileOps.written.add(path);
-				break;
-			case "edit":
-				fileOps.edited.add(path);
-				break;
-		}
-	}
-}
-
-/**
- * Compute final file lists from file operations.
- * Returns readFiles (files only read, not modified) and modifiedFiles.
- */
-export function computeFileLists(fileOps: FileOperations): { readFiles: string[]; modifiedFiles: string[] } {
-	const modified = new Set([...fileOps.edited, ...fileOps.written]);
-	const readOnly = [...fileOps.read].filter(f => !modified.has(f)).sort();
-	const modifiedFiles = [...modified].sort();
-	return { readFiles: readOnly, modifiedFiles };
-}
-
-/**
- * Format file operations as XML tags for summary.
- */
-const FILE_OPERATION_SUMMARY_LIMIT = 20;
-
-function truncateFileList(files: string[]): string[] {
-	if (files.length <= FILE_OPERATION_SUMMARY_LIMIT) return files;
-	const omitted = files.length - FILE_OPERATION_SUMMARY_LIMIT;
-	return [...files.slice(0, FILE_OPERATION_SUMMARY_LIMIT), `… (${omitted} more files omitted)`];
-}
-
-function stripFileOperationTags(summary: string): string {
-	const withoutReadFiles = summary.replace(/<read-files>[\s\S]*?<\/read-files>\s*/g, "");
-	const withoutModifiedFiles = withoutReadFiles.replace(/<modified-files>[\s\S]*?<\/modified-files>\s*/g, "");
-	return withoutModifiedFiles.trimEnd();
-}
-export function formatFileOperations(readFiles: string[], modifiedFiles: string[]): string {
-	if (readFiles.length === 0 && modifiedFiles.length === 0) return "";
-	return prompt.render(fileOperationsTemplate, {
-		readFiles: truncateFileList(readFiles),
-		modifiedFiles: truncateFileList(modifiedFiles),
-	});
-}
-
-export function upsertFileOperations(summary: string, readFiles: string[], modifiedFiles: string[]): string {
-	const baseSummary = stripFileOperationTags(summary);
-	const fileOperations = formatFileOperations(readFiles, modifiedFiles);
-	if (!fileOperations) return baseSummary;
-	if (!baseSummary) return fileOperations;
-	return `${baseSummary}\n\n${fileOperations}`;
-}
-
-// ============================================================================
-// Message Serialization
-// ============================================================================
-
-/** Maximum characters for a tool result in serialized summaries. */
-const TOOL_RESULT_MAX_CHARS = 2000;
-
-/**
- * Truncate text to a maximum character length for summarization.
- * Keeps the beginning and appends a truncation marker.
- */
-function truncateForSummary(text: string, maxChars: number): string {
-	if (text.length <= maxChars) return text;
-	const truncatedChars = text.length - maxChars;
-	return `${text.slice(0, maxChars)}\n\n[... ${truncatedChars} more characters truncated]`;
-}
-
-/**
- * Serialize LLM messages to text for summarization.
- * This prevents the model from treating it as a conversation to continue.
- * Call convertToLlm() first to handle custom message types.
- */
-export function serializeConversation(messages: Message[]): string {
-	const parts: string[] = [];
-
-	for (const msg of messages) {
-		if (msg.role === "user") {
-			const content =
-				typeof msg.content === "string"
-					? msg.content
-					: msg.content
-							.filter((c): c is { type: "text"; text: string } => c.type === "text")
-							.map(c => c.text)
-							.join("");
-			if (content) parts.push(`[User]: ${content}`);
-		} else if (msg.role === "assistant") {
-			const textParts: string[] = [];
-			const thinkingParts: string[] = [];
-			const toolCalls: string[] = [];
-
-			for (const block of msg.content) {
-				if (block.type === "text") {
-					textParts.push(block.text);
-				} else if (block.type === "thinking") {
-					thinkingParts.push(block.thinking);
-				} else if (block.type === "toolCall") {
-					const args = block.arguments as Record<string, unknown>;
-					const argsStr = Object.entries(args)
-						.map(([k, v]) => `${k}=${JSON.stringify(v)}`)
-						.join(", ");
-					toolCalls.push(`${block.name}(${argsStr})`);
-				}
-			}
-
-			if (thinkingParts.length > 0) {
-				parts.push(`[Assistant thinking]: ${thinkingParts.join("\n")}`);
-			}
-			if (textParts.length > 0) {
-				parts.push(`[Assistant]: ${textParts.join("\n")}`);
-			}
-			if (toolCalls.length > 0) {
-				parts.push(`[Assistant tool calls]: ${toolCalls.join("; ")}`);
-			}
-		} else if (msg.role === "toolResult") {
-			const content = msg.content
-				.filter((c): c is { type: "text"; text: string } => c.type === "text")
-				.map(c => c.text)
-				.join("");
-			if (content) {
-				parts.push(`[Tool result]: ${truncateForSummary(content, TOOL_RESULT_MAX_CHARS)}`);
-			}
-		}
-	}
-
-	return parts.join("\n\n");
-}
-
-// ============================================================================
-// Summarization System Prompt
-// ============================================================================
-
-export const SUMMARIZATION_SYSTEM_PROMPT = prompt.render(summarizationSystemPrompt);