flavor-code 1.3.16 → 1.3.17

package/dist/desktop/main.js

@@ -5507,9 +5507,8 @@ var ContextManager = class _ContextManager {
     const currentStableHash = currentStable === void 0 ? this.#epoch.stableSourceHash : hashMessages(currentStable);
     if (currentStable !== void 0 && currentStableHash !== this.#epoch.stableSourceHash) {
       const value = currentStable.map((message2) => message2.content).join("\n\n");
-      const previous = this.#epoch.sources["system-baseline"] ?? this.#epoch.stableMessages.map((message2) => message2.content).join("\n\n");
       if (this.#epoch.sources["system-baseline"] !== value) {
-        updates.push(contextUpdateMessage("system-baseline", deltaSourceValue(previous, value)));
+        updates.push(contextUpdateMessage("system-baseline", value));
         this.#epoch.sources["system-baseline"] = value;
       }
     } else if (currentStable !== void 0 && this.#epoch.sources["system-baseline"] !== void 0) {
@@ -5530,8 +5529,6 @@ var ContextManager = class _ContextManager {
     }
     if (updates.length === 0) return false;
     this.#messages.push(...updates);
-    this.#lastRecordedInputTokens = void 0;
-    this.#estimatedTokensAtLastRecordedUsage = void 0;
     return true;
   }
   snapshot() {
@@ -5591,10 +5588,7 @@ var ContextManager = class _ContextManager {
   async prepareForModelCall(signal = new AbortController().signal) {
     signal.throwIfAborted();
     this.refreshContextSources();
-    if (this.#dropStaleContextUpdates()) {
-      this.#lastRecordedInputTokens = void 0;
-      this.#estimatedTokensAtLastRecordedUsage = void 0;
-    }
+    this.#dropStaleContextUpdates();
     if (!this.needsCompaction()) return false;
     const originalMessages = this.#messages;
     const originalRecordedUsage = this.#lastRecordedInputTokens;
@@ -5607,8 +5601,6 @@ var ContextManager = class _ContextManager {
     const microcompact = microcompactMessages(this.#messages, this.#compaction.microcompactKeepRecentToolResults);
     if (microcompact.changed) {
       this.#messages = microcompact.messages;
-      this.#lastRecordedInputTokens = void 0;
-      this.#estimatedTokensAtLastRecordedUsage = void 0;
     }
     if (!this.needsCompaction()) {
       if (microcompact.changed) {
@@ -6572,7 +6564,7 @@ function memoryPressureError(heapUsed, heapLimit) {
   const mb = (bytes) => Math.round(bytes / 1048576);
   return {
     code: "memory_pressure",
-    message: `Heap usage reached ${Math.round(heapUsed / heapLimit * 100)}% of the V8 limit (${mb(heapUsed)}MB of ${mb(heapLimit)}MB); stopping to avoid a crash. The session is saved and will resume after restart.`
+    message: `Heap usage reached ${Math.round(heapUsed / heapLimit * 100)}% of the V8 limit (${mb(heapUsed)}MB of ${mb(heapLimit)}MB); stopping the current turn to avoid a crash. The saved session will reopen after restart.`
   };
 }
 function currentMemoryPressure() {
@@ -6692,6 +6684,11 @@ var AgentLoop = class {
         yield { type: "error", error: normalizeProviderError(error) };
         return;
       }
+      const preparedPressure = currentMemoryPressure();
+      if (preparedPressure !== void 0) {
+        yield { type: "error", error: preparedPressure };
+        return;
+      }
       let assistantText = "";
       const collectedToolCalls = [];
       const collectedThinking = [];
@@ -6715,6 +6712,11 @@ var AgentLoop = class {
           tools: [...this.#options.tools],
           ...request.signal === void 0 ? {} : { signal: request.signal }
         };
+        const requestPressure = currentMemoryPressure();
+        if (requestPressure !== void 0) {
+          yield { type: "error", error: requestPressure };
+          return;
+        }
         let before;
         try {
           before = await this.#options.hooks.emit({
@@ -15501,6 +15503,8 @@ function appendThinkingText(previous, delta) {
 }
 
 // src/ui/transcript.ts
+var MAX_TRANSCRIPT_TURNS = 200;
+var MAX_TRANSCRIPT_CHARS = 2e6;
 function createTranscriptState() {
   return { completed: [], nextId: 1 };
 }
@@ -15797,7 +15801,7 @@ function hydrateHistory(messages, compact) {
     }
   }
   if (turn !== void 0) completed.push(turn);
-  return { completed, nextId: completed.length + 1 };
+  return { completed: boundedCompleted(completed), nextId: completed.length + 1 };
 }
 function transcriptFromLegacyConversation(messages, compact) {
   return hydrateHistory(messages, compact);
@@ -15891,10 +15895,10 @@ function finishActive(state) {
   if (state.active === void 0) return state;
   const active = withoutModelActivity(state.active);
   return {
-    completed: [...state.completed, {
+    completed: boundedCompleted([...state.completed, {
       ...active,
       completedAt: active.completedAt ?? (/* @__PURE__ */ new Date()).toISOString()
-    }],
+    }]),
     nextId: state.nextId,
     ...state.taskSnapshot === void 0 ? {} : { taskSnapshot: state.taskSnapshot }
   };
@@ -15923,12 +15927,26 @@ function withoutModelActivity(turn, id) {
 function restoreTranscriptState(state) {
   const completed = state.completed.map(cloneTurn);
   if (state.active !== void 0) completed.push(normalizeInterruptedTurn(cloneTurn(state.active)));
+  const nextId = Math.max(state.nextId, ...completed.map((turn) => turn.id + 1), 1);
   return {
-    completed,
-    nextId: Math.max(state.nextId, ...completed.map((turn) => turn.id + 1), 1),
+    completed: boundedCompleted(completed),
+    nextId,
     ...state.taskSnapshot === void 0 ? {} : { taskSnapshot: state.taskSnapshot }
   };
 }
+function boundedCompleted(input) {
+  const recent = input.slice(-MAX_TRANSCRIPT_TURNS);
+  let chars = 0;
+  let start = recent.length;
+  for (let index = recent.length - 1; index >= 0; index -= 1) {
+    const turn = recent[index];
+    const size = JSON.stringify(turn).length;
+    if (start < recent.length && chars + size > MAX_TRANSCRIPT_CHARS) break;
+    chars += size;
+    start = index;
+  }
+  return recent.slice(start);
+}
 function compactionTurn(compact) {
   const text = `Original execution steps before ${compact.compactedAt} are unavailable.`;
   return {
@@ -16752,6 +16770,8 @@ function isCode9(error, code) {
 }
 
 // src/session/tree.ts
+var MAX_SESSION_HISTORY_NODES = 100;
+var MAX_SESSION_HISTORY_CONTEXT_CHARS = 2e6;
 var SessionHistory = class _SessionHistory {
   #path;
   #checkpoints;
@@ -16763,8 +16783,8 @@ var SessionHistory = class _SessionHistory {
     this.#path = join25(resolve21(options.workspace), ".flavor", "session-trees", options.sessionId, "tree.json");
     this.#checkpoints = new WorkspaceCheckpointStore({ workspace: options.workspace });
     this.#restoreContext = options.restoreContext;
-    this.#nodes = document.nodes;
-    this.#leafId = document.leafId;
+    this.#nodes = boundedNodes(document.nodes);
+    this.#leafId = document.leafId !== null && this.#nodes.some((node) => node.id === document.leafId) ? document.leafId : this.#nodes.at(-1)?.id ?? null;
   }
   static async open(options) {
     const path = join25(resolve21(options.workspace), ".flavor", "session-trees", options.sessionId, "tree.json");
@@ -16776,7 +16796,9 @@ var SessionHistory = class _SessionHistory {
     } catch (error) {
       if (!isCode10(error, "ENOENT")) throw error;
     }
-    return new _SessionHistory(options, document);
+    const history = new _SessionHistory(options, document);
+    if (history.#nodes.length !== document.nodes.length) await history.#persist();
+    return history;
   }
   get leafId() {
     return this.#leafId;
@@ -16796,6 +16818,7 @@ var SessionHistory = class _SessionHistory {
     };
     this.#nodes.push(node);
     this.#leafId = node.id;
+    this.#boundNodes();
     await this.#persist();
     return structuredClone(node);
   }
@@ -16837,6 +16860,13 @@ var SessionHistory = class _SessionHistory {
     if (node === void 0) throw new Error(`Unknown session tree node: ${id}`);
     return node;
   }
+  #boundNodes() {
+    const bounded2 = boundedNodes(this.#nodes);
+    this.#nodes.splice(0, this.#nodes.length, ...bounded2);
+    if (this.#leafId !== null && !this.#nodes.some((node) => node.id === this.#leafId)) {
+      this.#leafId = this.#nodes.at(-1)?.id ?? null;
+    }
+  }
   async #persist() {
     const document = {
       version: 1,
@@ -16851,6 +16881,24 @@ var SessionHistory = class _SessionHistory {
     await rename8(temporary, this.#path);
   }
 };
+function boundedNodes(input) {
+  const recent = input.slice(-MAX_SESSION_HISTORY_NODES);
+  let contextChars = 0;
+  let start = recent.length;
+  for (let index = recent.length - 1; index >= 0; index -= 1) {
+    const node = recent[index];
+    const size = JSON.stringify(node.context).length;
+    if (start < recent.length && contextChars + size > MAX_SESSION_HISTORY_CONTEXT_CHARS) break;
+    contextChars += size;
+    start = index;
+  }
+  const retained = recent.slice(start).map((node) => structuredClone(node));
+  const ids = new Set(retained.map((node) => node.id));
+  for (const node of retained) {
+    if (node.parentId !== null && !ids.has(node.parentId)) node.parentId = null;
+  }
+  return retained;
+}
 function validate(value, sessionId) {
   if (value.version !== 1 || value.sessionId !== sessionId || !Array.isArray(value.nodes)) {
     throw new Error("Invalid session tree");
@@ -26325,7 +26373,7 @@ async function createProductionRuntime(options) {
           contextEpoch: harness.main.context.snapshot().epoch?.id ?? "legacy"
         });
         const turnId = harnessJournal.startTurn(turnConfig, { prompt, initialUserMessage: runOptions?.initialUserMessage });
-        return monitorMemoryRestart(durableTurn(persistAndCheckpointAfter(runMain(
+        return monitorMemoryRestart(durableTurn(persistAfter(runMain(
           harness,
           skills,
           prompt,
@@ -26341,11 +26389,7 @@ async function createProductionRuntime(options) {
           runOptions?.initialUserMessage,
           runOptions?.additionalContext,
           ide
-        ), persist, () => sessionHistory.append({
-          prompt,
-          context: harness.main.context.snapshot(),
-          label: `turn: ${prompt.slice(0, 80)}`
-        })), harnessJournal, turnId, turnConfig), () => requestMemoryRestart(workspace, sessionId, emitOutput));
+        ), persist), harnessJournal, turnId, turnConfig), () => requestMemoryRestart(workspace, sessionId, emitOutput));
       },
       runSkill: (skill, prompt, signal) => persistAfter(
         runExplicitSkill(harness, skills, skill, prompt, signal, selectedModels.mainError),
@@ -27055,14 +27099,6 @@ async function* persistAfter(source, persist) {
     await persist();
   }
 }
-async function* persistAndCheckpointAfter(source, persist, checkpoint) {
-  try {
-    for await (const item of source) yield item;
-    await checkpoint();
-  } finally {
-    await persist();
-  }
-}
 async function* persistEach(source, persist) {
   try {
     for await (const item of source) {