@cuylabs/agent-core 0.8.0 → 0.9.0

package/dist/{chunk-BDBZ3SLK.js → chunk-UHCJEM2E.js}

@@ -2,6 +2,9 @@
 var STORAGE_VERSION = 1;
 
 // src/storage/utils.ts
+function logStorageWarning(message) {
+  console.warn(message);
+}
 function generateEntryId(existingIds) {
   for (let i = 0; i < 100; i++) {
     const id = crypto.randomUUID().slice(0, 8);
@@ -18,7 +21,7 @@ function parseJSONL(content) {
     try {
       entries.push(JSON.parse(trimmed));
     } catch {
-      console.warn("[Storage] Skipping malformed JSONL line");
+      logStorageWarning("[Storage] Skipping malformed JSONL line");
     }
   }
   return entries;
@@ -160,7 +163,6 @@ function getLeafId(entries) {
     children.push(e.id);
     childMap.set(e.parentId, children);
   }
-  const _allIds = new Set(entries.slice(1).map((e) => e.id));
   const hasChildren = /* @__PURE__ */ new Set();
   for (const children of childMap.values()) {
     for (const child of children) {
@@ -204,6 +206,7 @@ function buildMessagesFromEntries(entries, leafId) {
   let skipUntilId;
   for (const entry of path) {
     if (entry.type === "compaction") {
+      messages.length = 0;
       skipUntilId = entry.firstKeptEntryId;
       messages.push({
         id: entry.id,
@@ -615,6 +618,40 @@ var SessionManager = class {
     this.currentLeafId = entry.id;
     return entry.id;
   }
+  async replaceWithCompaction(options) {
+    if (!this.currentSessionId) {
+      throw new Error("No session loaded");
+    }
+    const compactionId = generateEntryId(this.idsCache);
+    this.idsCache.add(compactionId);
+    let parentId = compactionId;
+    const messageEntries = [];
+    for (const message of options.messages) {
+      const entry = createMessageEntry(message, parentId, this.idsCache);
+      messageEntries.push(entry);
+      this.idsCache.add(entry.id);
+      parentId = entry.id;
+    }
+    const compactionEntry = {
+      type: "compaction",
+      id: compactionId,
+      parentId: this.currentLeafId,
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      summary: options.summary,
+      firstKeptEntryId: messageEntries[0]?.id ?? compactionId,
+      tokensBefore: options.tokensBefore,
+      tokensAfter: options.tokensAfter,
+      ...options.readFiles ? { readFiles: options.readFiles } : {},
+      ...options.modifiedFiles ? { modifiedFiles: options.modifiedFiles } : {}
+    };
+    await this.storage.appendBatch(this.currentSessionId, [
+      compactionEntry,
+      ...messageEntries
+    ]);
+    this.entriesCache.push(compactionEntry, ...messageEntries);
+    this.currentLeafId = messageEntries[messageEntries.length - 1]?.id ?? compactionEntry.id;
+    return compactionEntry.id;
+  }
   async branch(fromEntryId, summary) {
     if (!this.currentSessionId) {
       throw new Error("No session loaded");