ds4-context-engine 0.1.0

package/src/pi-adapter/compaction-adapter.ts

@@ -0,0 +1,150 @@
+import {
+  convertToLlm,
+  serializeConversation,
+  sessionEntryToContextMessages,
+  type SessionBeforeCompactEvent,
+  type SessionEntry,
+} from "@earendil-works/pi-coding-agent";
+import {
+  parseDs4CompactionDetails,
+  type EmbeddedSummaryNode,
+} from "ds4-context-core/compaction/compaction-record";
+import { computeSummarySourceHash } from "ds4-context-core/compaction/summary-contract";
+import { sha256 } from "ds4-context-core/shared/hash";
+import { stableStringify } from "ds4-context-core/shared/stable-json";
+
+interface SourceCandidate {
+  entryId: string;
+  fingerprint: string;
+  used: boolean;
+}
+
+export interface PreparedCompactionSource {
+  messages: SessionBeforeCompactEvent["preparation"]["messagesToSummarize"];
+  conversationText: string;
+  sourceText: string;
+  sourceEntryIds: string[];
+  sourceHash: string;
+  segmentReadFiles: string[];
+  segmentModifiedFiles: string[];
+  readFiles: string[];
+  modifiedFiles: string[];
+  previousSummary?: string;
+  previousNode?: EmbeddedSummaryNode;
+}
+
+function fingerprint(message: unknown): string {
+  return sha256(stableStringify(message));
+}
+
+function candidates(entries: readonly SessionEntry[]): SourceCandidate[] {
+  return entries.flatMap((entry) => sessionEntryToContextMessages(entry).map((message) => ({
+    entryId: entry.id,
+    fingerprint: fingerprint(message),
+    used: false,
+  })));
+}
+
+function mapSourceEntryIds(messages: readonly unknown[], entries: readonly SessionEntry[]): string[] {
+  const available = candidates(entries);
+  const ids: string[] = [];
+
+  for (const message of messages) {
+    const match = available.find((candidate) => !candidate.used && candidate.fingerprint === fingerprint(message));
+    if (!match) throw new Error("Compaction source message has no exact canonical Pi session entry");
+    match.used = true;
+    if (!ids.includes(match.entryId)) ids.push(match.entryId);
+  }
+
+  return ids;
+}
+
+function previousFiles(entries: readonly SessionEntry[]): { read: string[]; modified: string[] } {
+  for (let index = entries.length - 1; index >= 0; index--) {
+    const entry = entries[index];
+    if (entry?.type !== "compaction") continue;
+    const details = parseDs4CompactionDetails(entry.details);
+    if (details) return { read: details.readFiles, modified: details.modifiedFiles };
+  }
+  return { read: [], modified: [] };
+}
+
+function sortedUnique(values: Iterable<string>): string[] {
+  return [...new Set(values)].sort((left, right) => left.localeCompare(right));
+}
+
+export function findActiveBranchSummary(
+  entries: readonly SessionEntry[],
+  previousSummary?: string,
+): EmbeddedSummaryNode | undefined {
+  if (!previousSummary) return undefined;
+  for (let index = entries.length - 1; index >= 0; index--) {
+    const entry = entries[index];
+    if (entry?.type !== "compaction" || entry.summary !== previousSummary) continue;
+    const details = parseDs4CompactionDetails(entry.details);
+    if (!details) return undefined;
+    const metadata = details.ds4ContextEngine;
+    return {
+      id: metadata.summaryId,
+      kind: metadata.summaryKind,
+      content: entry.summary,
+      sourceHash: metadata.sourceHash,
+      sourceEntryIds: [...metadata.sourceEntryIds],
+      childSummaryIds: [...metadata.childSummaryIds],
+      graphLevel: metadata.graphLevel,
+      createdAt: metadata.generatedAt,
+      validationStatus: metadata.validationStatus,
+      validationIssueCodes: [...metadata.validationIssueCodes],
+      provider: metadata.provider,
+      model: metadata.model,
+    };
+  }
+  return undefined;
+}
+
+export function prepareCompactionSource(event: SessionBeforeCompactEvent): PreparedCompactionSource {
+  const messages = [
+    ...event.preparation.messagesToSummarize,
+    ...event.preparation.turnPrefixMessages,
+  ];
+  if (messages.length === 0) throw new Error("Pi compaction preparation contains no source messages");
+
+  const sourceEntryIds = mapSourceEntryIds(messages, event.branchEntries);
+  const priorFiles = previousFiles(event.branchEntries);
+  const segmentReadFiles = sortedUnique(event.preparation.fileOps.read);
+  const segmentModifiedFiles = sortedUnique([
+    ...event.preparation.fileOps.written,
+    ...event.preparation.fileOps.edited,
+  ]);
+  const readFiles = sortedUnique([...priorFiles.read, ...segmentReadFiles]);
+  const modifiedFiles = sortedUnique([...priorFiles.modified, ...segmentModifiedFiles]);
+  const conversationText = serializeConversation(convertToLlm(messages));
+  const canonicalSourceText = stableStringify(messages);
+  const previousSummary = event.preparation.previousSummary;
+  const previousNode = findActiveBranchSummary(event.branchEntries, previousSummary);
+  const sourceText = [
+    canonicalSourceText,
+    conversationText,
+    segmentReadFiles.join("\n"),
+    segmentModifiedFiles.join("\n"),
+  ].filter(Boolean).join("\n\n");
+
+  return {
+    messages,
+    conversationText,
+    sourceText,
+    sourceEntryIds,
+    sourceHash: computeSummarySourceHash({
+      conversationText: canonicalSourceText,
+      sourceEntryIds,
+      readFiles: segmentReadFiles,
+      modifiedFiles: segmentModifiedFiles,
+    }),
+    segmentReadFiles,
+    segmentModifiedFiles,
+    readFiles,
+    modifiedFiles,
+    ...(previousSummary ? { previousSummary } : {}),
+    ...(previousNode ? { previousNode } : {}),
+  };
+}