gsd-pi 2.77.0-dev.538325aea → 2.77.0-dev.eaa4973bc

package/packages/gsd-agent-core/src/compaction/index.ts

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

package/packages/gsd-agent-core/src/compaction/utils.ts

@@ -1,312 +0,0 @@
-/**
- * Shared utilities for compaction and branch summarization.
- */
-
-import type { AgentMessage } from "@gsd/pi-agent-core";
-import type { ImageContent, Message, TextContent } from "@gsd/pi-ai";
-// TOOL_RESULT_MAX_CHARS removed from @gsd/pi-coding-agent 0.67.2 public API.
-// Value matches packages/pi-coding-agent/src/core/compaction/utils.ts:89.
-// Phase 09 moves to @gsd/agent-types.
-const TOOL_RESULT_MAX_CHARS = 2000;
-import type { SessionEntry } from "@gsd/agent-types";
-
-function parseEntryTimestamp(timestamp: string | number): number {
-	return typeof timestamp === "number" ? timestamp : new Date(timestamp).getTime();
-}
-
-function createCustomMessage(
-	customType: string,
-	content: string | (TextContent | ImageContent)[],
-	display: boolean,
-	details: unknown | undefined,
-	timestamp: string | number,
-): AgentMessage {
-	return {
-		role: "custom",
-		customType,
-		content,
-		display,
-		details,
-		timestamp: parseEntryTimestamp(timestamp),
-	};
-}
-
-function createBranchSummaryMessage(summary: string, fromId: string, timestamp: string | number): AgentMessage {
-	return {
-		role: "branchSummary",
-		summary,
-		fromId,
-		timestamp: parseEntryTimestamp(timestamp),
-	};
-}
-
-function createCompactionSummaryMessage(
-	summary: string,
-	tokensBefore: number,
-	timestamp: string | number,
-): AgentMessage {
-	return {
-		role: "compactionSummary",
-		summary,
-		tokensBefore,
-		timestamp: parseEntryTimestamp(timestamp),
-	};
-}
-
-// ============================================================================
-// 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.
- */
-export function formatFileOperations(readFiles: string[], modifiedFiles: string[]): string {
-	const sections: string[] = [];
-	if (readFiles.length > 0) {
-		sections.push(`<read-files>\n${readFiles.join("\n")}\n</read-files>`);
-	}
-	if (modifiedFiles.length > 0) {
-		sections.push(`<modified-files>\n${modifiedFiles.join("\n")}\n</modified-files>`);
-	}
-	if (sections.length === 0) return "";
-	return `\n\n${sections.join("\n\n")}`;
-}
-
-// ============================================================================
-// Message Extraction
-// ============================================================================
-
-/**
- * Extract AgentMessage from a session entry.
- *
- * Handles all entry types: message, custom_message, branch_summary, and compaction.
- * Returns undefined for entries that don't contribute to LLM context (e.g., settings changes).
- *
- * @param skipToolResults - If true, skips toolResult messages (used by branch summarization
- *   where tool call context is sufficient). Default false.
- */
-export function getMessageFromEntry(entry: SessionEntry, skipToolResults = false): AgentMessage | undefined {
-	switch (entry.type) {
-		case "message":
-			if (skipToolResults && entry.message.role === "toolResult") return undefined;
-			return entry.message;
-
-		case "custom_message":
-			return createCustomMessage(entry.customType, entry.content, entry.display, entry.details, entry.timestamp);
-
-		case "branch_summary":
-			return createBranchSummaryMessage(entry.summary, entry.fromId, entry.timestamp);
-
-		case "compaction":
-			return createCompactionSummaryMessage(entry.summary, entry.tokensBefore, entry.timestamp);
-
-		case "thinking_level_change":
-		case "model_change":
-		case "custom":
-		case "label":
-		case "session_info":
-			return undefined;
-	}
-}
-
-/**
- * Collect AgentMessages from a range of session entries.
- *
- * @param entries - Session entries array
- * @param startIndex - First index (inclusive)
- * @param endIndex - Last index (exclusive)
- * @param skipToolResults - If true, skips toolResult messages. Default false.
- */
-export function collectMessages(
-	entries: SessionEntry[],
-	startIndex: number,
-	endIndex: number,
-	skipToolResults = false,
-): AgentMessage[] {
-	const result: AgentMessage[] = [];
-	for (let i = startIndex; i < endIndex; i++) {
-		const msg = getMessageFromEntry(entries[i], skipToolResults);
-		if (msg) result.push(msg);
-	}
-	return result;
-}
-
-// ============================================================================
-// Text Content Extraction
-// ============================================================================
-
-/**
- * Extract text from an array of content blocks, filtering to text-type blocks.
- * Replaces the recurring `.filter(c => c.type === "text").map(c => c.text).join(sep)` pattern.
- */
-export function extractTextContent(
-	content: Array<{ type: string; text?: string }>,
-	separator = "\n",
-): string {
-	return content
-		.filter((c): c is { type: "text"; text: string } => c.type === "text")
-		.map((c) => c.text)
-		.join(separator);
-}
-
-// ============================================================================
-// Summarization Message Construction
-// ============================================================================
-
-/**
- * Create a single-message array for summarization prompts.
- * Wraps promptText in the standard `[{ role: "user", content: [{ type: "text", text }], timestamp }]` shape.
- */
-export function createSummarizationMessage(promptText: string): [{ role: "user"; content: [{ type: "text"; text: string }]; timestamp: number }] {
-	return [
-		{
-			role: "user" as const,
-			content: [{ type: "text" as const, text: promptText }],
-			timestamp: Date.now(),
-		},
-	];
-}
-
-// ============================================================================
-// Message Serialization
-// ============================================================================
-
-// TOOL_RESULT_MAX_CHARS imported from @gsd/pi-coding-agent
-
-/**
- * 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.
- *
- * Tool results are truncated to keep the summarization request within
- * reasonable token budgets. Full content is not needed for summarization.
- */
-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 said:** ${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 responded:** ${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 = `You are a context summarization assistant. Your task is to read a conversation between a user and an AI coding assistant, then produce a structured summary following the exact format specified.
-
-Do NOT continue the conversation. Do NOT respond to any questions in the conversation. ONLY output the structured summary.`;