@anvia/studio 1.0.6 → 1.0.9

package/dist/index.js

@@ -6480,10 +6480,11 @@ var InMemoryStudioStore = class {
   kind = "memory";
   compaction = {
     snapshot: ({ scope }) => {
-      const messages = this.sessions.get(scope.sessionId)?.messages ?? [];
+      const session = this.sessions.get(scope.sessionId);
+      const messages = session?.messages ?? [];
       return Promise.resolve({
-        revision: memoryRevision(messages),
-        messages: cloneMessages(messages)
+        revision: String(session?.storeRevision ?? 0),
+        messages: projectedMessages(messages, session?.compactionState)
       });
     },
     replacePrefix: (options) => this.replaceCompactionPrefix(options)
@@ -6504,6 +6505,7 @@ var InMemoryStudioStore = class {
       updatedAt: now,
       messageCount: 0,
       messages: [],
+      storeRevision: 0,
       runs: [],
       logs: []
     };
@@ -6536,6 +6538,7 @@ var InMemoryStudioStore = class {
     const session = this.sessions.get(input.scope.sessionId);
     if (session !== void 0) {
       session.messages.push(...cloneMessages(input.messages));
+      session.storeRevision += 1;
       session.messageCount = session.messages.length;
       session.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
     }
@@ -6545,6 +6548,8 @@ var InMemoryStudioStore = class {
     const session = this.sessions.get(scope.sessionId);
     if (session !== void 0) {
       session.messages = [];
+      delete session.compactionState;
+      session.storeRevision += 1;
       session.runs = [];
       session.messageCount = 0;
       session.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
@@ -6565,11 +6570,20 @@ var InMemoryStudioStore = class {
       throw new RangeError("messageCount must be a positive integer.");
     }
     const session = this.sessions.get(input.scope.sessionId);
-    if (session === void 0 || memoryRevision(session.messages) !== input.revision || input.messageCount > session.messages.length) {
+    if (session === void 0 || String(session.storeRevision) !== input.revision) {
       return Promise.resolve({ status: "conflict" });
     }
-    session.messages.splice(0, input.messageCount, structuredClone(input.replacement));
-    session.messageCount = session.messages.length;
+    const physicalPrefixCount = input.messageCount - (session.compactionState === void 0 ? 0 : 1);
+    const activeMessageCount = session.messages.length - (session.compactionState?.summarizedThroughPosition ?? -1) - 1;
+    if (physicalPrefixCount < 0 || physicalPrefixCount > activeMessageCount) {
+      return Promise.resolve({ status: "conflict" });
+    }
+    session.compactionState = {
+      generation: (session.compactionState?.generation ?? 0) + 1,
+      summary: structuredClone(input.replacement),
+      summarizedThroughPosition: (session.compactionState?.summarizedThroughPosition ?? -1) + physicalPrefixCount
+    };
+    session.storeRevision += 1;
     session.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
     return Promise.resolve({ status: "committed" });
   }
@@ -6713,8 +6727,12 @@ function studioRunId(scope) {
   const value = scope.metadata?.studioRunId;
   return typeof value === "string" && value.length > 0 ? value : void 0;
 }
-function memoryRevision(messages) {
-  return JSON.stringify(messages);
+function projectedMessages(messages, state) {
+  if (state === void 0) return cloneMessages(messages);
+  return [
+    structuredClone(state.summary),
+    ...cloneMessages(messages.slice(state.summarizedThroughPosition + 1))
+  ];
 }
 function serializeJsonError(error) {
   if (error instanceof Error) {
@@ -7496,6 +7514,50 @@ import { mkdirSync } from "fs";
 import { createRequire } from "module";
 import { dirname, resolve } from "path";
 import { isJsonValue as isJsonValue3 } from "@anvia/core/completion";
+
+// src/storage/compaction-state.ts
+import { parseMessage } from "@anvia/core/completion";
+import { isMemoryCompactionMessage } from "@anvia/core/memory";
+function parseStudioCompactionState(value) {
+  if (value === null) return void 0;
+  const parsed = JSON.parse(value);
+  if (typeof parsed !== "object" || parsed === null) {
+    throw new Error("Stored Studio memory compaction state is invalid.");
+  }
+  const record = parsed;
+  let summary;
+  try {
+    summary = parseMessage(record.summary);
+  } catch (cause) {
+    throw new Error("Stored Studio memory compaction state summary is invalid.", { cause });
+  }
+  if (record.version !== 1 || !Number.isSafeInteger(record.generation) || record.generation < 1 || !isMemoryCompactionMessage(summary) || !Number.isSafeInteger(record.summarizedThroughPosition) || record.summarizedThroughPosition < 0) {
+    throw new Error("Stored Studio memory compaction state is invalid.");
+  }
+  return {
+    version: 1,
+    generation: record.generation,
+    summary,
+    summarizedThroughPosition: record.summarizedThroughPosition
+  };
+}
+function projectStudioCompactionMessages(rows, state) {
+  if (state === void 0) return rows.map((row) => row.message);
+  return [state.summary, ...activeStudioCompactionMessages(rows, state).map((row) => row.message)];
+}
+function activeStudioCompactionMessages(rows, state) {
+  if (state === void 0) return rows;
+  const boundaryIndex = rows.findIndex((row) => row.position === state.summarizedThroughPosition);
+  if (boundaryIndex === -1) {
+    throw new Error("Stored Studio memory compaction state boundary is invalid.");
+  }
+  return rows.slice(boundaryIndex + 1);
+}
+function studioCompactionRevision(rows, state) {
+  return JSON.stringify([state?.generation ?? 0, rows]);
+}
+
+// src/storage/sqlite-store.ts
 var DatabaseSync;
 function createSqliteSessionStore(options = {}) {
   return new SqliteSessionStore(options.path ?? ":memory:");
@@ -7508,8 +7570,13 @@ var SqliteSessionStore = class {
   kind = "sqlite";
   compaction = {
     snapshot: ({ scope }) => {
-      const messages = this.listSessionMessages(scope.sessionId);
-      return Promise.resolve({ revision: memoryRevision2(messages), messages });
+      const session = this.getCompactionSessionRow(scope.sessionId);
+      const state = parseStudioCompactionState(session?.compaction_state_json ?? null);
+      const rows = this.listCompactionMessages(scope.sessionId, state);
+      return Promise.resolve({
+        revision: studioCompactionRevision(rows, state),
+        messages: projectStudioCompactionMessages(rows, state)
+      });
     },
     replacePrefix: (options) => this.replaceCompactionPrefix(options)
   };
@@ -7628,7 +7695,8 @@ var SqliteSessionStore = class {
       db.exec("BEGIN IMMEDIATE");
       db.prepare(
         `UPDATE anvia_studio_sessions
-         SET updated_at = $updatedAt
+         SET compaction_state_json = NULL,
+             updated_at = $updatedAt
          WHERE id = $id`
       ).run({
         $id: scope.sessionId,
@@ -7668,26 +7736,41 @@ var SqliteSessionStore = class {
     const db = this.database();
     try {
       db.exec("BEGIN IMMEDIATE");
-      const messages = this.listSessionMessages(input.scope.sessionId);
-      if (memoryRevision2(messages) !== input.revision || input.messageCount > messages.length || this.getSessionRow(input.scope.sessionId) === void 0) {
+      const session = this.getCompactionSessionRow(input.scope.sessionId);
+      const state = parseStudioCompactionState(session?.compaction_state_json ?? null);
+      const rows = this.listCompactionMessages(input.scope.sessionId, state);
+      if (session === void 0 || studioCompactionRevision(rows, state) !== input.revision) {
         db.exec("ROLLBACK");
         return Promise.resolve({ status: "conflict" });
       }
-      db.prepare("DELETE FROM anvia_studio_session_messages WHERE session_id = $id").run({
-        $id: input.scope.sessionId
-      });
+      const physicalPrefixCount = input.messageCount - (state === void 0 ? 0 : 1);
+      const activeRows = activeStudioCompactionMessages(rows, state);
+      if (physicalPrefixCount < 0 || physicalPrefixCount > activeRows.length) {
+        db.exec("ROLLBACK");
+        return Promise.resolve({ status: "conflict" });
+      }
+      const summarizedThroughPosition = physicalPrefixCount === 0 ? state?.summarizedThroughPosition : activeRows[physicalPrefixCount - 1]?.position;
+      if (summarizedThroughPosition === void 0) {
+        db.exec("ROLLBACK");
+        return Promise.resolve({ status: "conflict" });
+      }
+      const nextState = {
+        version: 1,
+        generation: (state?.generation ?? 0) + 1,
+        summary: input.replacement,
+        summarizedThroughPosition
+      };
       const updatedAt = (/* @__PURE__ */ new Date()).toISOString();
-      this.insertMessages(
-        input.scope.sessionId,
-        [input.replacement, ...messages.slice(input.messageCount)],
-        0,
-        updatedAt
-      );
       db.prepare(
         `UPDATE anvia_studio_sessions
-         SET updated_at = $updatedAt
+         SET compaction_state_json = $compactionState,
+             updated_at = $updatedAt
          WHERE id = $id`
-      ).run({ $id: input.scope.sessionId, $updatedAt: updatedAt });
+      ).run({
+        $id: input.scope.sessionId,
+        $compactionState: JSON.stringify(nextState),
+        $updatedAt: updatedAt
+      });
       db.exec("COMMIT");
       return Promise.resolve({ status: "committed" });
     } catch (error) {
@@ -8185,6 +8268,7 @@ var SqliteSessionStore = class {
         agent_id TEXT NOT NULL,
         title TEXT,
         metadata_json TEXT,
+        compaction_state_json TEXT,
         created_at TEXT NOT NULL,
         updated_at TEXT NOT NULL
       ) STRICT;
@@ -8290,6 +8374,7 @@ var SqliteSessionStore = class {
       CREATE INDEX IF NOT EXISTS anvia_studio_traces_session_started_idx
         ON anvia_studio_traces(session_id, started_at DESC);
     `);
+    ensureSessionCompactionStateColumn(db);
     ensureMessageMetadataColumn(db);
     this.db = db;
     return db;
@@ -8301,6 +8386,13 @@ var SqliteSessionStore = class {
          WHERE id = $id`
     ).get({ $id: id });
   }
+  getCompactionSessionRow(id) {
+    return this.database().prepare(
+      `SELECT id, compaction_state_json
+         FROM anvia_studio_sessions
+         WHERE id = $id`
+    ).get({ $id: id });
+  }
   getSessionRun(sessionId, runId) {
     return this.database().prepare(
       `SELECT run_id, session_id, status, title, transcript_json, error_json, created_at, updated_at
@@ -8333,13 +8425,22 @@ var SqliteSessionStore = class {
     return row.next_sequence;
   }
   listSessionMessages(sessionId) {
+    return this.listSessionMessageEntries(sessionId).map((entry) => entry.message);
+  }
+  listCompactionMessages(sessionId, state) {
+    return this.listSessionMessageEntries(sessionId, state?.summarizedThroughPosition);
+  }
+  listSessionMessageEntries(sessionId, minimumIndex) {
     const db = this.database();
+    const boundaryClause = minimumIndex === void 0 ? "" : "AND message_index >= $minimumIndex";
+    const parameters = minimumIndex === void 0 ? { $sessionId: sessionId } : { $sessionId: sessionId, $minimumIndex: minimumIndex };
     const messageRows = db.prepare(
       `SELECT session_id, message_index, role, message_id, metadata_json, created_at
          FROM anvia_studio_session_messages
          WHERE session_id = $sessionId
+           ${boundaryClause}
          ORDER BY message_index ASC`
-    ).all({ $sessionId: sessionId });
+    ).all(parameters);
     if (messageRows.length === 0) {
       return [];
     }
@@ -8347,17 +8448,19 @@ var SqliteSessionStore = class {
       `SELECT session_id, message_index, part_index, type, part_json
          FROM anvia_studio_session_message_parts
          WHERE session_id = $sessionId
+           ${boundaryClause}
          ORDER BY message_index ASC, part_index ASC`
-    ).all({ $sessionId: sessionId });
+    ).all(parameters);
     const partsByMessage = /* @__PURE__ */ new Map();
     for (const partRow of partRows) {
       const parts = partsByMessage.get(partRow.message_index) ?? [];
       parts.push(partRow);
       partsByMessage.set(partRow.message_index, parts);
     }
-    return messageRows.map(
-      (row) => messageFromRows(row, partsByMessage.get(row.message_index) ?? [])
-    );
+    return messageRows.map((row) => ({
+      position: row.message_index,
+      message: messageFromRows(row, partsByMessage.get(row.message_index) ?? [])
+    }));
   }
   nextMessageIndex(sessionId) {
     const row = this.database().prepare(
@@ -8561,6 +8664,12 @@ function ensureMessageMetadataColumn(db) {
     db.exec("ALTER TABLE anvia_studio_session_messages ADD COLUMN metadata_json TEXT");
   }
 }
+function ensureSessionCompactionStateColumn(db) {
+  const columns = db.prepare("PRAGMA table_info('anvia_studio_sessions')").all();
+  if (!columns.some((column) => column.name === "compaction_state_json")) {
+    db.exec("ALTER TABLE anvia_studio_sessions ADD COLUMN compaction_state_json TEXT");
+  }
+}
 function loadDatabaseSync() {
   if (DatabaseSync !== void 0) {
     return DatabaseSync;
@@ -8624,9 +8733,6 @@ function studioRunId2(scope) {
   const value = scope.metadata?.studioRunId;
   return typeof value === "string" && value.length > 0 ? value : void 0;
 }
-function memoryRevision2(messages) {
-  return JSON.stringify(messages);
-}
 function serializeJsonError2(error) {
   if (error instanceof Error) {
     return {