@deepagents/context 7.1.0 → 7.1.2

package/dist/index.js

@@ -175,7 +175,7 @@ function fitsFragmentData(value, visited) {
   if (Array.isArray(value)) {
     return value.every((item) => fitsFragmentData(item, visited));
   }
-  if (value instanceof Promise) return true;
+  if (value instanceof Promise) return false;
   if (Symbol.asyncIterator in value || Symbol.iterator in value) return true;
   if (isFragment(value)) return fitsFragmentData(value.data, visited);
   return Object.values(value).every((item) => fitsFragmentData(item, visited));
@@ -2647,6 +2647,7 @@ import {
   isStaticToolUIPart as isStaticToolUIPart2,
   validateUIMessages
 } from "ai";
+import pMemoize2 from "p-memoize";
 
 // packages/context/src/lib/ui-message-guards.ts
 function isRecord2(value) {
@@ -2679,75 +2680,63 @@ function mergeReply(segments) {
     parts: segments.flatMap((segment) => segment.parts)
   };
 }
-var ChainSummaryBuilder = class {
-  #turn = 0;
-  #messageCount = 0;
-  #lastMessageAt;
-  #lastMessage;
-  #lastAssistantMessage;
-  #lastAssistantMessages = [];
-  #replies = [];
-  #replyClosed = true;
-  #firedOnceIds = /* @__PURE__ */ new Set();
-  ingestStored(msg) {
-    this.#messageCount++;
+function summarizeChain(stored, pending) {
+  let turn = 0;
+  let lastMessageAt;
+  let lastMessage;
+  const assistantMessages = [];
+  let reply = [];
+  const replies = [reply];
+  const firedOnceIds = /* @__PURE__ */ new Set();
+  for (const msg of stored) {
     if (msg.name === "assistant") {
       const message3 = requireUIMessage(
         msg.data,
         `Stored assistant message "${msg.id}"`
       );
-      this.#lastAssistantMessage = message3;
-      this.#lastAssistantMessages.push(message3);
-      const open = this.#replies.at(-1);
-      if (open && !this.#replyClosed) open.push(message3);
-      else {
-        this.#replies.push([message3]);
-        this.#replyClosed = false;
-      }
-      return;
-    }
-    if (msg.name !== "user") {
-      return;
+      assistantMessages.push(message3);
+      reply.push(message3);
+      continue;
     }
+    if (msg.name !== "user") continue;
     const message2 = requireUIMessage(
       msg.data,
       `Stored user message "${msg.id}"`
     );
     if (isSyntheticReminderMessage(message2)) {
       for (const id of message2.metadata.synthetic.onceIds ?? []) {
-        this.#firedOnceIds.add(id);
+        firedOnceIds.add(id);
       }
-      return;
+      continue;
     }
-    for (const id of getReminderOnceIds(message2)) this.#firedOnceIds.add(id);
-    this.#turn++;
-    this.#lastMessageAt = msg.createdAt;
-    this.#lastMessage = message2;
-    this.#replyClosed = true;
+    for (const id of getReminderOnceIds(message2)) firedOnceIds.add(id);
+    turn++;
+    lastMessageAt = msg.createdAt;
+    lastMessage = message2;
+    reply = [];
+    replies.push(reply);
   }
-  ingestPending(fragment2) {
-    this.#messageCount++;
-    if (fragment2.name !== "user") return;
+  for (const fragment2 of pending) {
+    if (fragment2.name !== "user") continue;
     const encoded = fragment2.codec?.encode();
-    if (isUIMessage(encoded) && isSyntheticReminderMessage(encoded)) return;
-    this.#turn++;
-  }
-  build() {
-    return {
-      turn: this.#turn,
-      messageCount: this.#messageCount,
-      lastMessageAt: this.#lastMessageAt,
-      lastMessage: this.#lastMessage,
-      lastAssistantMessage: this.#lastAssistantMessage,
-      lastAssistantMessages: this.#lastAssistantMessages,
-      lastAssistantReplies: this.#replies.map(mergeReply),
-      firedOnceIds: this.#firedOnceIds
-    };
+    if (isUIMessage(encoded) && isSyntheticReminderMessage(encoded)) continue;
+    turn++;
   }
-};
+  return {
+    turn,
+    messageCount: stored.length + pending.length,
+    lastMessageAt,
+    lastMessage,
+    lastAssistantMessage: assistantMessages.at(-1),
+    lastAssistantMessages: assistantMessages,
+    lastAssistantReplies: replies.filter((segments) => segments.length > 0).map(mergeReply),
+    firedOnceIds
+  };
+}
 
 // packages/context/src/lib/estimate.ts
 import { encode } from "gpt-tokenizer";
+import pMemoize from "p-memoize";
 import { z as z2 } from "zod";
 var defaultTokenizer = {
   encode(text) {
@@ -2789,14 +2778,17 @@ function inputPricePerMillion(model) {
 }
 var ModelsRegistry = class {
   #cache = /* @__PURE__ */ new Map();
-  #loaded = false;
   #tokenizers = /* @__PURE__ */ new Map();
   #defaultTokenizer = defaultTokenizer;
+  /** Fetches once; concurrent calls share the fetch, and a failed one runs again. */
+  #load = pMemoize(() => this.#fetchModels());
   /**
    * Load models data from models.dev API
    */
   async load() {
-    if (this.#loaded) return;
+    await this.#load();
+  }
+  async #fetchModels() {
     const response = await fetch("https://models.dev/api.json");
     if (!response.ok) {
       throw new Error(`Failed to fetch models: ${response.statusText}`);
@@ -2815,7 +2807,6 @@ var ModelsRegistry = class {
         this.#cache.set(`${providerId}:${modelId}`, info);
       }
     }
-    this.#loaded = true;
   }
   /**
    * Get model info by ID
@@ -3059,20 +3050,6 @@ function pushLimited2(collected, chunk, maxItems, resolverName) {
   collected.push(chunk);
 }
 
-// packages/context/src/lib/resolvers/promise-resolver.ts
-var PromiseResolver = class {
-  name = "PromiseResolver";
-  canResolve(value) {
-    return value instanceof Promise;
-  }
-  async resolve(value, _ctx) {
-    if (!this.canResolve(value)) {
-      throw new TypeError(`${this.name} cannot resolve a ${typeof value}`);
-    }
-    return value;
-  }
-};
-
 // packages/context/src/lib/resolvers/loader-resolver.ts
 var FragmentLoaderResolver = class {
   #resolvers;
@@ -3207,7 +3184,6 @@ function defaultResolvers() {
     new AsyncResolver(),
     new GeneratorResolver(),
     new FunctionResolver(),
-    new PromiseResolver(),
     new IterableResolver()
   ];
 }
@@ -3224,14 +3200,14 @@ var HeadConflictError = class extends Error {
 var SavePipeline = class _SavePipeline {
   #engine;
   #pending;
-  #shouldBranch = true;
-  constructor(engine, pending) {
+  #shouldBranch;
+  constructor(engine, pending, shouldBranch) {
     this.#engine = engine;
     this.#pending = pending;
-  }
-  async applyUpdateBranching(shouldBranch) {
     this.#shouldBranch = shouldBranch;
-    if (!shouldBranch) return this;
+  }
+  async applyUpdateBranching() {
+    if (!this.#shouldBranch) return this;
     for (const fragment2 of this.#pending) {
       if (!fragment2.id) continue;
       const existing = await this.#engine.store.getMessage(fragment2.id);
@@ -4101,6 +4077,25 @@ function mergeTerminalToolParts(message2, liveParts) {
   parts.push(...remaining.values());
   return { ...message2, parts };
 }
+var UNOPENED = {
+  chat: null,
+  branchName: "main",
+  headMessageId: void 0
+};
+var OpenChat = class {
+  chat;
+  branch;
+  constructor(chat2, branch) {
+    this.chat = chat2;
+    this.branch = branch;
+  }
+  get branchName() {
+    return this.branch.name;
+  }
+  get headMessageId() {
+    return this.branch.headMessageId ?? void 0;
+  }
+};
 var ContextEngine = class _ContextEngine {
   /** Non-message fragments (role, hints, etc.) - not persisted in graph */
   #fragments = [];
@@ -4109,23 +4104,13 @@ var ContextEngine = class _ContextEngine {
   #store;
   #chatId;
   #userId;
-  #branchName;
-  #branch = null;
-  #chatData = null;
-  #initialized = false;
-  /** Initial metadata to merge on first initialization */
-  #initialMetadata;
+  /** What this engine knows of its chat: nothing until the first store call opens it. */
+  #session = UNOPENED;
+  /** Opens the chat once; concurrent first calls share it, and a failed open runs again. */
+  #open;
   #loaderResolver;
   /** Retained so forked children inherit the parent's resolver chain */
   #resolvers;
-  get #activeBranch() {
-    if (!this.#branch) {
-      throw new Error(
-        "Branch not initialized. Call #ensureInitialized() first."
-      );
-    }
-    return this.#branch;
-  }
   get #renderableFragments() {
     return this.#fragments.filter((f) => !isConditionalReminder(f));
   }
@@ -4139,38 +4124,30 @@ var ContextEngine = class _ContextEngine {
     this.#store = options.store;
     this.#chatId = options.chatId;
     this.#userId = options.userId;
-    this.#branchName = "main";
-    this.#initialMetadata = options.metadata;
+    this.#open = pMemoize2(() => this.#openChat(options.metadata));
     this.#resolvers = options.resolvers;
     this.#loaderResolver = new FragmentLoaderResolver(
       options.resolvers ?? defaultResolvers()
     );
   }
   /**
-   * Initialize the chat and branch if they don't exist.
+   * Creates the chat if it is new, checks its owner, merges the initial
+   * metadata, and reads the active branch.
    */
-  async #ensureInitialized() {
-    if (this.#initialized) {
-      return;
-    }
-    this.#chatData = await this.#store.upsertChat({
+  async #openChat(metadata) {
+    let chat2 = await this.#store.upsertChat({
       id: this.#chatId,
       userId: this.#userId
     });
-    if (this.#chatData.userId !== this.#userId) {
+    if (chat2.userId !== this.#userId) {
       throw new Error(
-        `chat "${this.#chatId}" belongs to user "${this.#chatData.userId}", not "${this.#userId}"`
+        `chat "${this.#chatId}" belongs to user "${chat2.userId}", not "${this.#userId}"`
       );
     }
-    if (this.#initialMetadata) {
-      const initialMetadata = this.#initialMetadata;
-      this.#chatData = await this.#store.updateChat(
-        this.#chatId,
-        ({ metadata }) => ({
-          metadata: { ...metadata, ...initialMetadata }
-        })
-      );
-      this.#initialMetadata = void 0;
+    if (metadata) {
+      chat2 = await this.#store.updateChat(this.#chatId, (stored) => ({
+        metadata: { ...stored.metadata, ...metadata }
+      }));
     }
     const branch = await this.#store.getActiveBranch(this.#chatId);
     if (!branch) {
@@ -4178,20 +4155,21 @@ var ContextEngine = class _ContextEngine {
         `Active branch not found for chat "${this.#chatId}" after upsertChat`
       );
     }
-    this.#branch = branch;
-    this.#branchName = branch.name;
-    this.#initialized = true;
+    const open = new OpenChat(chat2, branch);
+    this.#session = open;
+    return open;
   }
   /**
    * Create a new branch from a specific message.
    * Shared logic between rewind() and btw().
    */
-  async #createBranchFrom(messageId, switchTo) {
+  async #createBranchFrom(open, messageId, switchTo) {
     const branches = await this.#store.listBranches(this.#chatId);
+    const baseName = open.branch.name;
     const samePrefix = branches.filter(
-      (it) => it.name === this.#branchName || it.name.startsWith(`${this.#branchName}-v`)
+      (it) => it.name === baseName || it.name.startsWith(`${baseName}-v`)
     );
-    const newBranchName = `${this.#branchName}-v${samePrefix.length + 1}`;
+    const newBranchName = `${baseName}-v${samePrefix.length + 1}`;
     const newBranch = {
       id: crypto.randomUUID(),
       chatId: this.#chatId,
@@ -4203,8 +4181,7 @@ var ContextEngine = class _ContextEngine {
     await this.#store.createBranch(newBranch);
     if (switchTo) {
       await this.#store.setActiveBranch(this.#chatId, newBranch.id);
-      this.#branch = { ...newBranch, isActive: true };
-      this.#branchName = newBranchName;
+      open.branch = { ...newBranch, isActive: true };
       this.#pendingMessages = [];
     }
     const chain = await this.#store.getMessageChain(messageId);
@@ -4236,47 +4213,40 @@ var ContextEngine = class _ContextEngine {
    * Get the current branch name.
    */
   get branch() {
-    return this.#branchName;
+    return this.#session.branchName;
   }
   /**
    * Get the current branch head message ID.
    * Returns undefined if no messages have been saved yet.
    */
   get headMessageId() {
-    return this.#branch?.headMessageId ?? void 0;
+    return this.#session.headMessageId;
   }
   /**
    * Get metadata for the current chat.
    * Returns null if the chat hasn't been initialized yet.
    */
   get chat() {
-    return this.#chatData;
+    return this.#session.chat;
   }
   /**
    * Count user turns in the conversation and return the previous saved user message context.
    * Includes persisted messages and pending messages in the turn count.
    */
   async #getChainContext(pending = this.#pendingMessages) {
-    await this.#ensureInitialized();
-    const builder = new ChainSummaryBuilder();
-    if (this.#branch?.headMessageId) {
-      const chain = await this.#store.getMessageChain(
-        this.#branch.headMessageId
-      );
-      for (const msg of chain) builder.ingestStored(msg);
-    }
-    for (const fragment2 of pending) builder.ingestPending(fragment2);
-    return builder.build();
+    const open = await this.#open();
+    const stored = open.branch.headMessageId ? await this.#store.getMessageChain(open.branch.headMessageId) : [];
+    return summarizeChain(stored, pending);
   }
   async getTurnCount() {
     const { turn } = await this.#getChainContext();
     return turn;
   }
   async firstUserMessage() {
-    await this.#ensureInitialized();
-    if (this.#branch?.headMessageId) {
+    const open = await this.#open();
+    if (open.branch.headMessageId) {
       const chain = await this.#store.getMessageChain(
-        this.#branch.headMessageId
+        open.branch.headMessageId
       );
       for (const msg of chain) {
         if (msg.name === "user") {
@@ -4305,7 +4275,7 @@ var ContextEngine = class _ContextEngine {
    * @throws if the pending tail is missing an id (programming error).
    */
   async headMessage() {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     if (this.#pendingMessages.length > 0) {
       const tail = this.#pendingMessages[this.#pendingMessages.length - 1];
       if (!tail.id) {
@@ -4315,8 +4285,8 @@ var ContextEngine = class _ContextEngine {
       }
       return { id: tail.id, name: tail.name };
     }
-    if (this.#branch?.headMessageId) {
-      const msg = await this.#store.getMessage(this.#branch.headMessageId);
+    if (open.branch.headMessageId) {
+      const msg = await this.#store.getMessage(open.branch.headMessageId);
       if (msg) return { id: msg.id, name: msg.name };
     }
     return void 0;
@@ -4336,12 +4306,12 @@ var ContextEngine = class _ContextEngine {
    * For id-lookup use `headMessage()` instead — that one keeps placeholders.
    */
   async getMessages() {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     const messages = [];
     const indexById = /* @__PURE__ */ new Map();
-    if (this.#branch?.headMessageId) {
+    if (open.branch.headMessageId) {
       const chain = await this.#store.getMessageChain(
-        this.#branch.headMessageId
+        open.branch.headMessageId
       );
       for (const msg of chain) {
         const data = requireUIMessage(msg.data, `Stored message "${msg.id}"`);
@@ -4452,7 +4422,7 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async resolve(options) {
-    await this.#ensureInitialized();
+    await this.#open();
     await this.#loaderResolver.resolve(this.#fragments, {
       sandbox: options.sandbox,
       context: this,
@@ -4477,15 +4447,19 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async save(options) {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     if (this.#pendingMessages.length === 0) {
-      return { headMessageId: this.#branch?.headMessageId ?? void 0 };
+      return { headMessageId: open.branch.headMessageId ?? void 0 };
     }
     const pending = this.#pendingMessages;
     this.#pendingMessages = [];
-    const pipeline = new SavePipeline(this.#asSavePipelineEngine(), pending);
+    const pipeline = new SavePipeline(
+      this.#asSavePipelineEngine(open),
+      pending,
+      options?.branch ?? true
+    );
     try {
-      await pipeline.applyUpdateBranching(options?.branch ?? true);
+      await pipeline.applyUpdateBranching();
       await this.#foldUserReminders(pending);
       return await pipeline.persist();
     } catch (error) {
@@ -4726,13 +4700,14 @@ var ContextEngine = class _ContextEngine {
     return this.#fragments.filter(isConditionalReminder).map((fragment2) => fragment2.metadata.reminder).filter((config) => config.target === target);
   }
   #buildBaseWhenCtx(chain) {
-    const rawUsage = this.#chatData?.metadata?.usage;
-    const usage = isLanguageModelUsage(rawUsage) ? rawUsage : void 0;
-    const elapsed = chain.lastMessageAt !== void 0 ? Date.now() - chain.lastMessageAt : void 0;
-    const chatData = this.#chatData;
-    if (!chatData) {
+    const session = this.#session;
+    if (!(session instanceof OpenChat)) {
       throw new Error("ContextEngine must be initialized before reminders run");
     }
+    const chatData = session.chat;
+    const rawUsage = chatData.metadata?.usage;
+    const usage = isLanguageModelUsage(rawUsage) ? rawUsage : void 0;
+    const elapsed = chain.lastMessageAt !== void 0 ? Date.now() - chain.lastMessageAt : void 0;
     return {
       turn: chain.turn,
       messageCount: chain.messageCount,
@@ -4740,7 +4715,7 @@ var ContextEngine = class _ContextEngine {
       lastMessage: chain.lastMessage,
       chat: chatData,
       usage,
-      branch: this.#branchName,
+      branch: session.branchName,
       elapsed
     };
   }
@@ -4861,36 +4836,36 @@ var ContextEngine = class _ContextEngine {
     }
     return { texts, onceIds: [...onceIds] };
   }
-  #asSavePipelineEngine() {
+  #asSavePipelineEngine(open) {
     return {
       store: this.#store,
       chatId: this.#chatId,
       getActiveBranch: () => ({
-        id: this.#activeBranch.id,
-        headMessageId: this.#activeBranch.headMessageId
+        id: open.branch.id,
+        headMessageId: open.branch.headMessageId
       }),
       commitHead: async (headMessageId, expectedHeadMessageId) => {
         const committed = await this.#store.updateBranchHead(
-          this.#activeBranch.id,
+          open.branch.id,
           headMessageId,
           expectedHeadMessageId
         );
         if (committed) {
-          this.#activeBranch.headMessageId = headMessageId;
+          open.branch.headMessageId = headMessageId;
         }
         return committed;
       },
       refreshBranch: async () => {
         const fresh = await this.#store.getBranch(
           this.#chatId,
-          this.#activeBranch.name
+          open.branch.name
         );
         if (!fresh) {
           throw new Error(
-            `Branch "${this.#activeBranch.name}" not found for chat "${this.#chatId}"`
+            `Branch "${open.branch.name}" not found for chat "${this.#chatId}"`
           );
         }
-        this.#branch = fresh;
+        open.branch = fresh;
         return { id: fresh.id, headMessageId: fresh.headMessageId };
       },
       rewindForUpdate: (parentId) => this.#rewindForUpdate(parentId)
@@ -4909,7 +4884,7 @@ var ContextEngine = class _ContextEngine {
    * @returns Estimate result with token counts, costs, and per-fragment breakdown
    */
   async estimate(modelId, options = {}) {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     await this.#loaderResolver.resolve(this.#fragments, {
       sandbox: options.sandbox,
       context: this,
@@ -4938,9 +4913,9 @@ var ContextEngine = class _ContextEngine {
         cost
       });
     }
-    if (this.#branch?.headMessageId) {
+    if (open.branch.headMessageId) {
       const chain = await this.#store.getMessageChain(
-        this.#branch.headMessageId
+        open.branch.headMessageId
       );
       for (const msg of chain) {
         const content = estimateMessageContent(msg.data);
@@ -5012,7 +4987,7 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async rewind(messageId) {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     const message2 = await this.#store.getMessage(messageId);
     if (!message2) {
       throw new Error(`Message "${messageId}" not found`);
@@ -5020,7 +4995,7 @@ var ContextEngine = class _ContextEngine {
     if (message2.chatId !== this.#chatId) {
       throw new Error(`Message "${messageId}" belongs to a different chat`);
     }
-    return this.#createBranchFrom(messageId, true);
+    return this.#createBranchFrom(open, messageId, true);
   }
   /**
    * Create a checkpoint at the current position.
@@ -5042,15 +5017,15 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async checkpoint(name) {
-    await this.#ensureInitialized();
-    if (!this.#branch?.headMessageId) {
+    const open = await this.#open();
+    if (!open.branch.headMessageId) {
       throw new Error("Cannot create checkpoint: no messages in conversation");
     }
     const checkpoint = {
       id: crypto.randomUUID(),
       chatId: this.#chatId,
       name,
-      messageId: this.#branch.headMessageId,
+      messageId: open.branch.headMessageId,
       createdAt: Date.now()
     };
     await this.#store.createCheckpoint(checkpoint);
@@ -5078,7 +5053,7 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async restore(name) {
-    await this.#ensureInitialized();
+    await this.#open();
     const checkpoint = await this.#store.getCheckpoint(this.#chatId, name);
     if (!checkpoint) {
       throw new Error(
@@ -5103,14 +5078,13 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async switchBranch(name) {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     const branch = await this.#store.getBranch(this.#chatId, name);
     if (!branch) {
       throw new Error(`Branch "${name}" not found in chat "${this.#chatId}"`);
     }
     await this.#store.setActiveBranch(this.#chatId, branch.id);
-    this.#branch = { ...branch, isActive: true };
-    this.#branchName = name;
+    open.branch = { ...branch, isActive: true };
     this.#pendingMessages = [];
   }
   /**
@@ -5145,11 +5119,11 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async btw() {
-    await this.#ensureInitialized();
-    if (!this.#branch?.headMessageId) {
+    const open = await this.#open();
+    if (!open.branch.headMessageId) {
       throw new Error("Cannot create btw branch: no messages in conversation");
     }
-    return this.#createBranchFrom(this.#branch.headMessageId, false);
+    return this.#createBranchFrom(open, open.branch.headMessageId, false);
   }
   /**
    * Update metadata for the current chat.
@@ -5165,14 +5139,11 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async updateChat(updates) {
-    await this.#ensureInitialized();
-    this.#chatData = await this.#store.updateChat(
-      this.#chatId,
-      ({ metadata }) => ({
-        title: updates.title,
-        metadata: updates.metadata === void 0 ? void 0 : { ...metadata, ...updates.metadata }
-      })
-    );
+    const open = await this.#open();
+    open.chat = await this.#store.updateChat(this.#chatId, ({ metadata }) => ({
+      title: updates.title,
+      metadata: updates.metadata === void 0 ? void 0 : { ...metadata, ...updates.metadata }
+    }));
   }
   /**
    * Track token usage for the current chat.
@@ -5188,20 +5159,17 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async trackUsage(usage) {
-    await this.#ensureInitialized();
-    this.#chatData = await this.#store.updateChat(
-      this.#chatId,
-      ({ metadata }) => {
-        const storedUsage = metadata?.usage;
-        const currentUsage = isLanguageModelUsage(storedUsage) ? storedUsage : void 0;
-        return {
-          metadata: {
-            ...metadata,
-            usage: mergeLanguageModelUsage(currentUsage, usage)
-          }
-        };
-      }
-    );
+    const open = await this.#open();
+    open.chat = await this.#store.updateChat(this.#chatId, ({ metadata }) => {
+      const storedUsage = metadata?.usage;
+      const currentUsage = isLanguageModelUsage(storedUsage) ? storedUsage : void 0;
+      return {
+        metadata: {
+          ...metadata,
+          usage: mergeLanguageModelUsage(currentUsage, usage)
+        }
+      };
+    });
   }
   /**
    * Consolidate context fragments (no-op for now).
@@ -5267,7 +5235,7 @@ var ContextEngine = class _ContextEngine {
    * ```
    */
   async inspect(options) {
-    await this.#ensureInitialized();
+    const open = await this.#open();
     const { renderer } = options;
     const estimateResult = await this.estimate(options.modelId, {
       renderer,
@@ -5276,9 +5244,9 @@ var ContextEngine = class _ContextEngine {
     });
     const rendered = renderer.render(this.#renderableFragments);
     const persistedMessages = [];
-    if (this.#branch?.headMessageId) {
+    if (open.branch.headMessageId) {
       const chain = await this.#store.getMessageChain(
-        this.#branch.headMessageId
+        open.branch.headMessageId
       );
       persistedMessages.push(...chain);
     }
@@ -5294,7 +5262,7 @@ var ContextEngine = class _ContextEngine {
       graph,
       meta: {
         chatId: this.#chatId,
-        branch: this.#branchName,
+        branch: open.branch.name,
         timestamp: Date.now()
       }
     };
@@ -10740,7 +10708,7 @@ async function assertMicrosandboxBash(vm) {
 function normalizeMicrosandboxError(error, sdk) {
   const err = toError2(error);
   if (err instanceof MicrosandboxSandboxError) return err;
-  if (err instanceof sdk.LibkrunfwNotFoundError) {
+  if (err instanceof sdk.LibkrunfwNotFoundError || err instanceof sdk.RuntimeNotInstalledError || err instanceof sdk.RuntimeIncompleteError) {
     return new MicrosandboxNotAvailableError(err);
   }
   if (err instanceof sdk.MicrosandboxError) {
@@ -11458,7 +11426,6 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
   #pool;
   #schema;
   #ownsPool;
-  #isInitialized = false;
   constructor(options) {
     super();
     const schema = options.schema ?? "public";
@@ -11488,24 +11455,26 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
   #t(name) {
     return `"${this.#schema}"."${name}"`;
   }
-  async initialize() {
+  /** Opens the store: creates its schema objects before it returns. */
+  static async create(options) {
+    const store = new _PostgresContextStore(options);
+    try {
+      await store.#initialize();
+    } catch (error) {
+      await store.close();
+      throw error;
+    }
+    return store;
+  }
+  async #initialize() {
     const ddl = storeDDL(this.#schema);
     await this.#pool.query(ddl);
-    this.#isInitialized = true;
-  }
-  #ensureInitialized() {
-    if (!this.#isInitialized) {
-      throw new Error(
-        "PostgresContextStore not initialized. Call await store.initialize() after construction."
-      );
-    }
   }
   /**
    * Execute a function within a transaction.
    * Automatically commits on success or rolls back on error.
    */
   async #useTransaction(fn) {
-    this.#ensureInitialized();
     const client = await this.#pool.connect();
     try {
       await client.query("BEGIN");
@@ -11523,7 +11492,6 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
    * Execute a query using the pool (no transaction).
    */
   async #query(sql, params) {
-    this.#ensureInitialized();
     const result = await this.#pool.query(sql, params);
     return result.rows;
   }
@@ -12072,6 +12040,11 @@ IF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_${schema}_branches_ch
 IF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_${schema}_branches_chatId_isActive' AND object_id = OBJECT_ID('[${schema}].[branches]'))
   CREATE INDEX [idx_${schema}_branches_chatId_isActive] ON [${schema}].[branches](chatId, isActive);
 
+-- Deleting a message checks this foreign key. Without the index the check
+-- scans every chat's branches and waits on rows another chat delete holds.
+IF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_${schema}_branches_headMessageId' AND object_id = OBJECT_ID('[${schema}].[branches]'))
+  CREATE INDEX [idx_${schema}_branches_headMessageId] ON [${schema}].[branches](headMessageId);
+
 IF OBJECT_ID('[${schema}].[checkpoints]', 'U') IS NULL
 BEGIN
   CREATE TABLE [${schema}].[checkpoints] (
@@ -12225,7 +12198,6 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
   #pool;
   #schema;
   #ownsPool;
-  #isInitialized = false;
   constructor(options) {
     super();
     const schema = options.schema ?? "dbo";
@@ -12255,7 +12227,18 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
   #t(name) {
     return `[${this.#schema}].[${name}]`;
   }
-  async initialize() {
+  /** Opens the store: connects a pool it owns and creates its schema objects before it returns. */
+  static async create(options) {
+    const store = new _SqlServerContextStore(options);
+    try {
+      await store.#initialize();
+    } catch (error) {
+      await store.close();
+      throw error;
+    }
+    return store;
+  }
+  async #initialize() {
     if (this.#ownsPool) {
       await this.#pool.connect();
     }
@@ -12275,21 +12258,12 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
         await this.#pool.request().batch(batch);
       }
     }
-    this.#isInitialized = true;
-  }
-  #ensureInitialized() {
-    if (!this.#isInitialized) {
-      throw new Error(
-        "SqlServerContextStore not initialized. Call await store.initialize() after construction."
-      );
-    }
   }
   /**
    * Execute a function within a transaction.
    * Automatically commits on success or rolls back on error.
    */
   async #useTransaction(fn) {
-    this.#ensureInitialized();
     const mssql = _SqlServerContextStore.#requireMssql();
     const transaction = new mssql.Transaction(this.#pool);
     try {
@@ -12307,7 +12281,6 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
    * Converts positional params to SQL Server named params (@p0, @p1, ...).
    */
   async #query(sql, params) {
-    this.#ensureInitialized();
     const request = this.#pool.request();
     params?.forEach((value, index) => {
       request.input(`p${index}`, value);
@@ -13159,7 +13132,9 @@ function staticPartText(part, options) {
 }
 
 // packages/context/src/lib/stream/postgres-notify-change-source.ts
+import { EventEmitter, on } from "node:events";
 import { createRequire as createRequire3 } from "node:module";
+import pMemoize3 from "p-memoize";
 import { z as z16 } from "zod";
 
 // packages/context/src/lib/stream/ddl.stream.postgres-notify.ts
@@ -13220,28 +13195,73 @@ EXECUTE FUNCTION "${schema}"."notify_stream_status_update"();
 }
 
 // packages/context/src/lib/stream/postgres-notify-change-source.ts
+var NOT_LISTENING = {
+  release: () => Promise.resolve(),
+  fail: () => {
+  }
+};
+var Listening = class {
+  #client;
+  #handlers;
+  constructor(client, handlers) {
+    this.#client = client;
+    this.#handlers = handlers;
+    client.on("notification", handlers.notification);
+    client.on("error", handlers.error);
+    client.on("end", handlers.end);
+  }
+  /** Stops listening and returns the connection to the pool. */
+  async release(channel) {
+    this.#detach();
+    try {
+      await this.#client.query(`UNLISTEN ${quoteIdentifier(channel)}`);
+    } finally {
+      this.#client.release();
+    }
+  }
+  /** Gives a broken connection back for the pool to destroy. */
+  fail(error) {
+    this.#detach();
+    this.#client.release(error);
+  }
+  #detach() {
+    this.#client.off("notification", this.#handlers.notification);
+    this.#client.off("error", this.#handlers.error);
+    this.#client.off("end", this.#handlers.end);
+  }
+};
+var ChangeDelivery = class {
+  change;
+  constructor(change) {
+    this.change = change;
+  }
+};
 var PostgresNotifyChangeSource = class _PostgresNotifyChangeSource {
   #pool;
   #schema;
   #channel;
   #ownsPool;
-  #listener;
-  #listenPromise;
-  #isInitialized = false;
-  #isClosing = false;
-  #isClosed = false;
-  #subscribers = /* @__PURE__ */ new Map();
-  #onNotification = (message2) => {
-    this.#handleNotification(message2);
-  };
-  #onListenerError = (error) => {
-    this.#failListener(error);
-  };
-  #onListenerEnd = () => {
-    if (!this.#isClosing) {
-      this.#failListener(
-        new Error("PostgreSQL stream notification listener ended")
-      );
+  /** Aborts when close() starts: every subscription ends, and no LISTEN connection opens. */
+  #closed = new AbortController();
+  /** Emits `stream:<id>` with a ChangeDelivery; `error` fails every subscription. */
+  #changes = new EventEmitter().setMaxListeners(0);
+  #subscriptions = 0;
+  #listener = NOT_LISTENING;
+  /** Concurrent subscribers share one LISTEN setup; a failed one runs again. */
+  #listen = pMemoize3(() => this.#openListener(), { cache: false });
+  #handlers = {
+    notification: (message2) => {
+      this.#handleNotification(message2);
+    },
+    error: (error) => {
+      this.#failListener(error);
+    },
+    end: () => {
+      if (!this.#closed.signal.aborted) {
+        this.#failListener(
+          new Error("PostgreSQL stream notification listener ended")
+        );
+      }
     }
   };
   constructor(options) {
@@ -13270,98 +13290,75 @@ var PostgresNotifyChangeSource = class _PostgresNotifyChangeSource {
       );
     }
   }
-  async initialize() {
-    await this.#pool.query(
-      postgresStreamNotifyDDL(this.#schema, this.#channel)
-    );
-    this.#isInitialized = true;
+  /** Opens the source: installs the notify triggers before it returns. */
+  static async create(options) {
+    const source = new _PostgresNotifyChangeSource(options);
+    try {
+      await source.#pool.query(
+        postgresStreamNotifyDDL(source.#schema, source.#channel)
+      );
+    } catch (error) {
+      await source.close();
+      throw error;
+    }
+    return source;
   }
   async *subscribe(streamId, signal) {
-    this.#ensureInitialized();
-    this.#ensureOpen();
-    const queue = new ChangeQueue();
-    this.#addSubscriber(streamId, queue);
-    const cleanup = () => {
-      queue.close();
-      this.#removeSubscriber(streamId, queue);
-    };
-    signal.addEventListener("abort", cleanup, { once: true });
+    if (this.#closed.signal.aborted) {
+      throw new Error("PostgresNotifyChangeSource is closed.");
+    }
+    if (signal.aborted) return;
+    const changes = on(this.#changes, `stream:${streamId}`, {
+      signal: AbortSignal.any([signal, this.#closed.signal])
+    });
+    this.#subscriptions += 1;
     try {
       await this.#ensureListening();
-      if (signal.aborted || this.#isClosed) return;
+      if (signal.aborted || this.#closed.signal.aborted) return;
       yield { kind: "tick" };
-      while (!signal.aborted && !this.#isClosed) {
-        const change = await queue.next(signal);
-        if (!change) return;
-        yield change;
+      for await (const [delivery] of changes) {
+        if (delivery instanceof ChangeDelivery) yield delivery.change;
       }
+    } catch (error) {
+      if (error instanceof Error && error.name === "AbortError") return;
+      throw error;
     } finally {
-      signal.removeEventListener("abort", cleanup);
-      cleanup();
+      this.#subscriptions -= 1;
+      await changes.return?.();
       await this.#releaseListenerIfIdle();
     }
   }
   async close() {
-    if (this.#isClosed) return;
-    this.#isClosed = true;
-    this.#isClosing = true;
-    for (const subscribers of this.#subscribers.values()) {
-      for (const subscriber of subscribers) {
-        subscriber.close();
-      }
-    }
-    this.#subscribers.clear();
-    if (this.#listenPromise) {
-      try {
-        await this.#listenPromise;
-      } catch {
-      }
+    if (this.#closed.signal.aborted) return;
+    this.#closed.abort();
+    try {
+      await this.#listen();
+    } catch {
     }
     await this.#releaseListener();
     if (this.#ownsPool) {
       await this.#pool.end();
     }
   }
-  #ensureInitialized() {
-    if (!this.#isInitialized) {
-      throw new Error(
-        "PostgresNotifyChangeSource not initialized. Call await source.initialize() after construction."
-      );
-    }
+  async #ensureListening() {
+    if (this.#listener instanceof Listening) return;
+    await this.#listen();
   }
-  #ensureOpen() {
-    if (this.#isClosed) {
-      throw new Error("PostgresNotifyChangeSource is closed.");
+  async #openListener() {
+    if (this.#closed.signal.aborted) return;
+    const client = await this.#pool.connect();
+    const listening = new Listening(client, this.#handlers);
+    try {
+      await client.query(`LISTEN ${quoteIdentifier(this.#channel)}`);
+    } catch (error) {
+      listening.fail(true);
+      throw error;
     }
-  }
-  async #ensureListening() {
-    this.#ensureOpen();
-    if (this.#listener) return;
-    if (this.#listenPromise) return this.#listenPromise;
-    this.#listenPromise = (async () => {
-      const listener = await this.#pool.connect();
-      let success = false;
-      try {
-        listener.on("notification", this.#onNotification);
-        listener.on("error", this.#onListenerError);
-        listener.on("end", this.#onListenerEnd);
-        await listener.query(`LISTEN ${quoteIdentifier(this.#channel)}`);
-        if (this.#isClosed) {
-          return;
-        }
-        this.#listener = listener;
-        success = true;
-      } finally {
-        this.#listenPromise = void 0;
-        if (!success) {
-          listener.off("notification", this.#onNotification);
-          listener.off("error", this.#onListenerError);
-          listener.off("end", this.#onListenerEnd);
-          listener.release(true);
-        }
-      }
-    })();
-    return this.#listenPromise;
+    if (this.#closed.signal.aborted) {
+      listening.fail(true);
+      return;
+    }
+    this.#listener = listening;
   }
   #handleNotification(message2) {
     if (message2.channel !== this.#channel || !message2.payload) return;
@@ -13369,125 +13366,28 @@ var PostgresNotifyChangeSource = class _PostgresNotifyChangeSource {
     if (!payload) return;
     if (payload.schema !== this.#schema) return;
     if (payload.kind !== "chunks" && payload.kind !== "status") return;
-    const subscribers = this.#subscribers.get(payload.streamId);
-    if (!subscribers) return;
     const change = payload.kind === "chunks" ? { kind: "chunks" } : { kind: "status" };
-    for (const subscriber of subscribers) {
-      subscriber.push(change);
-    }
+    this.#changes.emit(
+      `stream:${payload.streamId}`,
+      new ChangeDelivery(change)
+    );
   }
   #failListener(error) {
     const listener = this.#listener;
-    this.#listener = void 0;
-    this.#listenPromise = void 0;
-    if (listener) {
-      listener.off("notification", this.#onNotification);
-      listener.off("error", this.#onListenerError);
-      listener.off("end", this.#onListenerEnd);
-      listener.release(error);
-    }
-    for (const subscribers of this.#subscribers.values()) {
-      for (const subscriber of subscribers) {
-        subscriber.fail(error);
-      }
+    this.#listener = NOT_LISTENING;
+    listener.fail(error);
+    if (this.#changes.listenerCount("error") > 0) {
+      this.#changes.emit("error", error);
     }
-    this.#subscribers.clear();
   }
   async #releaseListenerIfIdle() {
-    if (this.#subscribers.size > 0) return;
+    if (this.#subscriptions > 0) return;
     await this.#releaseListener();
   }
   async #releaseListener() {
     const listener = this.#listener;
-    if (!listener) return;
-    this.#listener = void 0;
-    listener.off("notification", this.#onNotification);
-    listener.off("error", this.#onListenerError);
-    listener.off("end", this.#onListenerEnd);
-    try {
-      await listener.query(`UNLISTEN ${quoteIdentifier(this.#channel)}`);
-    } finally {
-      listener.release();
-    }
-  }
-  #addSubscriber(streamId, subscriber) {
-    let subscribers = this.#subscribers.get(streamId);
-    if (!subscribers) {
-      subscribers = /* @__PURE__ */ new Set();
-      this.#subscribers.set(streamId, subscribers);
-    }
-    subscribers.add(subscriber);
-  }
-  #removeSubscriber(streamId, subscriber) {
-    const subscribers = this.#subscribers.get(streamId);
-    if (!subscribers) return;
-    subscribers.delete(subscriber);
-    if (subscribers.size === 0) {
-      this.#subscribers.delete(streamId);
-    }
-  }
-};
-var ChangeQueue = class {
-  #values = [];
-  #waiters = /* @__PURE__ */ new Set();
-  #closed = false;
-  #error;
-  push(value) {
-    if (this.#closed) return;
-    const waiter = this.#waiters.values().next().value;
-    if (waiter) {
-      this.#waiters.delete(waiter);
-      waiter.resolve(value);
-      return;
-    }
-    this.#values.push(value);
-  }
-  fail(error) {
-    if (this.#closed) return;
-    this.#closed = true;
-    this.#error = error;
-    this.#values = [];
-    for (const waiter of this.#waiters) {
-      waiter.reject(error);
-    }
-    this.#waiters.clear();
-  }
-  close() {
-    if (this.#closed) return;
-    this.#closed = true;
-    this.#values = [];
-    for (const waiter of this.#waiters) {
-      waiter.resolve(void 0);
-    }
-    this.#waiters.clear();
-  }
-  async next(signal) {
-    if (this.#error) throw this.#error;
-    if (this.#closed || signal.aborted) return void 0;
-    if (this.#values.length > 0) {
-      return this.#values.shift();
-    }
-    return new Promise((resolve3, reject) => {
-      const waiter = { resolve: resolve3, reject };
-      const cleanup = () => {
-        this.#waiters.delete(waiter);
-        signal.removeEventListener("abort", onAbort);
-      };
-      const onAbort = () => {
-        cleanup();
-        resolve3(void 0);
-      };
-      waiter.resolve = (value) => {
-        cleanup();
-        resolve3(value);
-      };
-      waiter.reject = (error) => {
-        cleanup();
-        reject(error);
-      };
-      this.#waiters.add(waiter);
-      signal.addEventListener("abort", onAbort, { once: true });
-    });
+    this.#listener = NOT_LISTENING;
+    await listener.release(this.#channel);
   }
 };
 var notificationPayload = z16.object({
@@ -13610,8 +13510,6 @@ var PostgresStreamStore = class _PostgresStreamStore extends StreamStore {
   #pool;
   #schema;
   #ownsPool;
-  #isInitialized = false;
-  #isClosed = false;
   constructor(options) {
     super();
     const schema = options.schema ?? "public";
@@ -13639,26 +13537,26 @@ var PostgresStreamStore = class _PostgresStreamStore extends StreamStore {
   #t(name) {
     return `"${this.#schema}"."${name}"`;
   }
-  async initialize() {
+  /** Opens the store: creates its tables before it returns. */
+  static async create(options) {
+    const store = new _PostgresStreamStore(options);
+    try {
+      await store.#initialize();
+    } catch (error) {
+      await store.close();
+      throw error;
+    }
+    return store;
+  }
+  async #initialize() {
     await this.#pool.query(postgresStreamDDL(this.#schema));
-    this.#isInitialized = true;
   }
   async close() {
-    if (this.#isClosed) return;
-    this.#isClosed = true;
-    if (this.#ownsPool) {
+    if (this.#ownsPool && !this.#pool.ending) {
       await this.#pool.end();
     }
   }
-  #ensureInitialized() {
-    if (!this.#isInitialized) {
-      throw new Error(
-        "PostgresStreamStore not initialized. Call await store.initialize() after construction."
-      );
-    }
-  }
   async #query(sql, params) {
-    this.#ensureInitialized();
     const result = await this.#pool.query(sql, params);
     return result.rows;
   }
@@ -13666,7 +13564,6 @@ var PostgresStreamStore = class _PostgresStreamStore extends StreamStore {
     return z18.array(row).parse(await this.#query(sql, params));
   }
   async #useTransaction(fn) {
-    this.#ensureInitialized();
     const client = await this.#pool.connect();
     try {
       await client.query("BEGIN");
@@ -13949,7 +13846,7 @@ import { DatabaseSync as DatabaseSync2 } from "node:sqlite";
 import { z as z19 } from "zod";
 
 // packages/context/src/lib/stream/ddl.stream.sqlite.sql
-var ddl_stream_sqlite_default = "PRAGMA journal_mode = WAL;\nPRAGMA synchronous = NORMAL;\nPRAGMA foreign_keys = ON;\n\nCREATE TABLE IF NOT EXISTS streams (\n  id TEXT PRIMARY KEY,\n  status TEXT NOT NULL DEFAULT 'queued'\n    CHECK(status IN ('queued','running','completed','failed','cancelled')),\n  createdAt INTEGER NOT NULL,\n  startedAt INTEGER,\n  finishedAt INTEGER,\n  cancelRequestedAt INTEGER,\n  error TEXT\n);\n\nCREATE TABLE IF NOT EXISTS stream_chunks (\n  streamId TEXT NOT NULL,\n  seq INTEGER NOT NULL,\n  data TEXT NOT NULL,\n  createdAt INTEGER NOT NULL,\n  PRIMARY KEY (streamId, seq),\n  FOREIGN KEY (streamId) REFERENCES streams(id) ON DELETE CASCADE\n);\n\n-- Supports ordered listing across all streams.\nCREATE INDEX IF NOT EXISTS idx_streams_created_at_id\n  ON streams(createdAt, id);\n\n-- Supports status-filtered ordered listing (e.g. running streams).\nCREATE INDEX IF NOT EXISTS idx_streams_status_created_at_id\n  ON streams(status, createdAt, id);\n";
+var ddl_stream_sqlite_default = "PRAGMA busy_timeout = 5000;\nPRAGMA journal_mode = WAL;\nPRAGMA synchronous = NORMAL;\nPRAGMA foreign_keys = ON;\n\nCREATE TABLE IF NOT EXISTS streams (\n  id TEXT PRIMARY KEY,\n  status TEXT NOT NULL DEFAULT 'queued'\n    CHECK(status IN ('queued','running','completed','failed','cancelled')),\n  createdAt INTEGER NOT NULL,\n  startedAt INTEGER,\n  finishedAt INTEGER,\n  cancelRequestedAt INTEGER,\n  error TEXT\n);\n\nCREATE TABLE IF NOT EXISTS stream_chunks (\n  streamId TEXT NOT NULL,\n  seq INTEGER NOT NULL,\n  data TEXT NOT NULL,\n  createdAt INTEGER NOT NULL,\n  PRIMARY KEY (streamId, seq),\n  FOREIGN KEY (streamId) REFERENCES streams(id) ON DELETE CASCADE\n);\n\n-- Supports ordered listing across all streams.\nCREATE INDEX IF NOT EXISTS idx_streams_created_at_id\n  ON streams(createdAt, id);\n\n-- Supports status-filtered ordered listing (e.g. running streams).\nCREATE INDEX IF NOT EXISTS idx_streams_status_created_at_id\n  ON streams(status, createdAt, id);\n";
 
 // packages/context/src/lib/stream/sqlite.stream-store.ts
 var streamRow = z19.object({
@@ -13972,7 +13869,6 @@ var chunkRow2 = z19.object({
 var SqliteStreamStore = class extends StreamStore {
   #db;
   #statements = /* @__PURE__ */ new Map();
-  #closed = false;
   #stmt(sql) {
     let stmt = this.#statements.get(sql);
     if (!stmt) {
@@ -13982,8 +13878,7 @@ var SqliteStreamStore = class extends StreamStore {
     return stmt;
   }
   close() {
-    if (this.#closed) return;
-    this.#closed = true;
+    if (!this.#db.isOpen) return;
     this.#statements.clear();
     this.#db.close();
   }
@@ -14124,8 +14019,7 @@ var SqliteStreamStore = class extends StreamStore {
   async appendChunks(chunks) {
     if (chunks.length === 0) return;
     const failures = collectStreamFailures(chunks);
-    this.#db.exec("BEGIN TRANSACTION");
-    try {
+    this.#transaction(() => {
       for (const chunk of chunks) {
         this.#stmt(
           `INSERT INTO stream_chunks (streamId, seq, data, createdAt)
@@ -14152,11 +14046,7 @@ var SqliteStreamStore = class extends StreamStore {
           }
         }
       }
-      this.#db.exec("COMMIT");
-    } catch (error) {
-      this.#db.exec("ROLLBACK");
-      throw error;
-    }
+    });
   }
   async getChunks(streamId, fromSeq, limit) {
     let sql = "SELECT * FROM stream_chunks WHERE streamId = ?";
@@ -14212,13 +14102,16 @@ var SqliteStreamStore = class extends StreamStore {
     });
   }
   #transaction(callback) {
+    this.#db.exec("BEGIN IMMEDIATE");
     try {
-      this.#db.exec("BEGIN IMMEDIATE");
       const result = callback();
       this.#db.exec("COMMIT");
       return result;
     } catch (error) {
-      this.#db.exec("ROLLBACK");
+      try {
+        this.#db.exec("ROLLBACK");
+      } catch {
+      }
       throw error;
     }
   }
@@ -14686,7 +14579,6 @@ export {
   PostgresContextStore,
   PostgresNotifyChangeSource,
   PostgresStreamStore,
-  PromiseResolver,
   RuntimeStrategy,
   SqlServerContextStore,
   SqliteContextStore,