@ponythewhite/base-context 1.0.7 → 1.0.8

package/dist/bundle/{chunk-VSIYFWYI.js → chunk-KT2BNEPE.js}

@@ -14,7 +14,7 @@ import {
   calculateCost,
   classifyStreamFailure,
   withRequestBody
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 
 // ../ai/src/providers/openai-responses-shared.ts
 function encodeTextSignatureV1(id, phase) {
@@ -56,11 +56,10 @@ function matchesResponsesTextSignature(signature, item) {
   const parsed = parseTextSignature(signature);
   return parsed !== void 0 && parsed.id === item.id && parsed.phase === phase;
 }
-function captureResponsesProjection(model, context, transformed, input, messageIndices, pendingPublicMessageGroups) {
+function captureResponsesProjection(model, context, unchangedSource, positionDependentMessageIndices, input, messageIndices, pendingPublicMessageGroups) {
   if (!(model.provider === "openai" && model.api === "openai-responses" || model.provider === "openai-codex" && model.api === "openai-codex-responses"))
     return;
-  if (transformed.length !== context.messages.length || transformed.some((message, index) => JSON.stringify(message) !== JSON.stringify(context.messages[index])))
-    return;
+  if (!unchangedSource) return;
   const items = context.messages.map(() => []);
   for (const [index, sourceIndex] of messageIndices.entries()) {
     if (sourceIndex !== null) items[sourceIndex].push(input[index]);
@@ -112,7 +111,7 @@ function captureResponsesProjection(model, context, transformed, input, messageI
         const item = rendered[blockIndex];
         if (block.type === "text") {
           if (!matchesResponsesTextSignature(block.textSignature, item)) return;
-          if (block.textSignature === void 0 && !generatedMessageIndices.includes(index))
+          if (positionDependentMessageIndices.has(index) && !generatedMessageIndices.includes(index))
             generatedMessageIndices.push(index);
         } else if (block.type === "thinking") {
           if (!block.thinkingSignature || item.type !== "reasoning") return;
@@ -240,6 +239,13 @@ function convertResponsesMessages(model, context, allowedToolCallProviders, opti
     normalizeToolCallId,
     pendingPublicMessageGroups
   );
+  const unchangedSource = (options?.onProjection !== void 0 || options?.responsesMessageIds !== void 0) && transformedMessages.length === context.messages.length && transformedMessages.every(
+    (message, index) => JSON.stringify(message) === JSON.stringify(context.messages[index])
+  );
+  const sourceIds = options?.responsesMessageIds;
+  const assignedIds = sourceIds?.filter((id) => id !== void 0);
+  const responsesMessageIds = unchangedSource && sourceIds?.length === context.messages.length && assignedIds?.every((id) => Number.isSafeInteger(id) && id >= 0) && new Set(assignedIds).size === assignedIds.length ? sourceIds : void 0;
+  const positionDependentMessageIndices = /* @__PURE__ */ new Set();
   const includeSystemPrompt = options?.includeSystemPrompt ?? true;
   if (includeSystemPrompt && context.systemPrompt) {
     const role = model.reasoning ? "developer" : "system";
@@ -282,7 +288,7 @@ function convertResponsesMessages(model, context, allowedToolCallProviders, opti
       const output = [];
       const assistantMsg = msg;
       const isDifferentModel = assistantMsg.model !== model.id && assistantMsg.provider === model.provider && assistantMsg.api === model.api;
-      for (const block of msg.content) {
+      for (const [blockIndex, block] of msg.content.entries()) {
         if (block.type === "thinking") {
           if (block.thinkingSignature) {
             const reasoningItem = JSON.parse(block.thinkingSignature);
@@ -293,7 +299,13 @@ function convertResponsesMessages(model, context, allowedToolCallProviders, opti
           const parsedSignature = parseTextSignature(textBlock.textSignature);
           let msgId = parsedSignature?.id;
           if (!msgId) {
-            msgId = `msg_${msgIndex}`;
+            const sourceId = responsesMessageIds?.[sourceIndex];
+            if (sourceId !== void 0) {
+              msgId = `msg_src_${sourceId}_${blockIndex}`;
+            } else {
+              msgId = `msg_${msgIndex}_${blockIndex}`;
+              positionDependentMessageIndices.add(sourceIndex);
+            }
           } else if (msgId.length > 64) {
             msgId = `msg_${shortHash(msgId)}`;
           }
@@ -365,7 +377,8 @@ function convertResponsesMessages(model, context, allowedToolCallProviders, opti
     const projection = captureResponsesProjection(
       model,
       context,
-      transformedMessages,
+      unchangedSource,
+      positionDependentMessageIndices,
       messages,
       messageIndices,
       pendingPublicMessageGroups

package/dist/bundle/{chunk-KTNETTHI.js → chunk-Q3MBSBEB.js}

@@ -3,6 +3,9 @@ import {
   sanitizeSurrogates,
   transformMessages
 } from "./chunk-4GXPNXOJ.js";
+import {
+  require_base64_js
+} from "./chunk-D3KOF57W.js";
 import {
   __commonJS,
   __require,
@@ -3847,107 +3850,6 @@ var require_src3 = __commonJS({
   }
 });
 
-// ../../node_modules/base64-js/index.js
-var require_base64_js = __commonJS({
-  "../../node_modules/base64-js/index.js"(exports) {
-    "use strict";
-    exports.byteLength = byteLength;
-    exports.toByteArray = toByteArray;
-    exports.fromByteArray = fromByteArray;
-    var lookup = [];
-    var revLookup = [];
-    var Arr = typeof Uint8Array !== "undefined" ? Uint8Array : Array;
-    var code = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
-    for (i = 0, len = code.length; i < len; ++i) {
-      lookup[i] = code[i];
-      revLookup[code.charCodeAt(i)] = i;
-    }
-    var i;
-    var len;
-    revLookup["-".charCodeAt(0)] = 62;
-    revLookup["_".charCodeAt(0)] = 63;
-    function getLens(b64) {
-      var len2 = b64.length;
-      if (len2 % 4 > 0) {
-        throw new Error("Invalid string. Length must be a multiple of 4");
-      }
-      var validLen = b64.indexOf("=");
-      if (validLen === -1) validLen = len2;
-      var placeHoldersLen = validLen === len2 ? 0 : 4 - validLen % 4;
-      return [validLen, placeHoldersLen];
-    }
-    function byteLength(b64) {
-      var lens = getLens(b64);
-      var validLen = lens[0];
-      var placeHoldersLen = lens[1];
-      return (validLen + placeHoldersLen) * 3 / 4 - placeHoldersLen;
-    }
-    function _byteLength(b64, validLen, placeHoldersLen) {
-      return (validLen + placeHoldersLen) * 3 / 4 - placeHoldersLen;
-    }
-    function toByteArray(b64) {
-      var tmp;
-      var lens = getLens(b64);
-      var validLen = lens[0];
-      var placeHoldersLen = lens[1];
-      var arr = new Arr(_byteLength(b64, validLen, placeHoldersLen));
-      var curByte = 0;
-      var len2 = placeHoldersLen > 0 ? validLen - 4 : validLen;
-      var i2;
-      for (i2 = 0; i2 < len2; i2 += 4) {
-        tmp = revLookup[b64.charCodeAt(i2)] << 18 | revLookup[b64.charCodeAt(i2 + 1)] << 12 | revLookup[b64.charCodeAt(i2 + 2)] << 6 | revLookup[b64.charCodeAt(i2 + 3)];
-        arr[curByte++] = tmp >> 16 & 255;
-        arr[curByte++] = tmp >> 8 & 255;
-        arr[curByte++] = tmp & 255;
-      }
-      if (placeHoldersLen === 2) {
-        tmp = revLookup[b64.charCodeAt(i2)] << 2 | revLookup[b64.charCodeAt(i2 + 1)] >> 4;
-        arr[curByte++] = tmp & 255;
-      }
-      if (placeHoldersLen === 1) {
-        tmp = revLookup[b64.charCodeAt(i2)] << 10 | revLookup[b64.charCodeAt(i2 + 1)] << 4 | revLookup[b64.charCodeAt(i2 + 2)] >> 2;
-        arr[curByte++] = tmp >> 8 & 255;
-        arr[curByte++] = tmp & 255;
-      }
-      return arr;
-    }
-    function tripletToBase64(num) {
-      return lookup[num >> 18 & 63] + lookup[num >> 12 & 63] + lookup[num >> 6 & 63] + lookup[num & 63];
-    }
-    function encodeChunk(uint8, start, end) {
-      var tmp;
-      var output = [];
-      for (var i2 = start; i2 < end; i2 += 3) {
-        tmp = (uint8[i2] << 16 & 16711680) + (uint8[i2 + 1] << 8 & 65280) + (uint8[i2 + 2] & 255);
-        output.push(tripletToBase64(tmp));
-      }
-      return output.join("");
-    }
-    function fromByteArray(uint8) {
-      var tmp;
-      var len2 = uint8.length;
-      var extraBytes = len2 % 3;
-      var parts = [];
-      var maxChunkLength = 16383;
-      for (var i2 = 0, len22 = len2 - extraBytes; i2 < len22; i2 += maxChunkLength) {
-        parts.push(encodeChunk(uint8, i2, i2 + maxChunkLength > len22 ? len22 : i2 + maxChunkLength));
-      }
-      if (extraBytes === 1) {
-        tmp = uint8[len2 - 1];
-        parts.push(
-          lookup[tmp >> 2] + lookup[tmp << 4 & 63] + "=="
-        );
-      } else if (extraBytes === 2) {
-        tmp = (uint8[len2 - 2] << 8) + uint8[len2 - 1];
-        parts.push(
-          lookup[tmp >> 10] + lookup[tmp >> 4 & 63] + lookup[tmp << 2 & 63] + "="
-        );
-      }
-      return parts.join("");
-    }
-  }
-});
-
 // ../../node_modules/google-auth-library/build/src/crypto/shared.js
 var require_shared = __commonJS({
   "../../node_modules/google-auth-library/build/src/crypto/shared.js"(exports) {

package/dist/bundle/{chunk-LFPWHI42.js → chunk-QKIFBA7N.js}

@@ -2,11 +2,11 @@ import { createRequire as __piBundleCreateRequire } from 'node:module'; const re
 import {
   appendRotatingLog,
   getAgentLogPath
-} from "./chunk-UPVDERMD.js";
+} from "./chunk-2IPDSTDR.js";
 import {
   setLogSink,
   stringifyLogEntry
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 
 // dist/core/logging.js
 var AGENT_LOG_MAX_BYTES = 20 * 1024 * 1024;

package/dist/bundle/{chunk-PFROC4EV.js → chunk-RVJVXZCS.js}

@@ -348,7 +348,7 @@ import {
   watchWithErrorHandler,
   workerRosterEntryFromSummary,
   wrapTextWithAnsi
-} from "./chunk-VCW3YDFF.js";
+} from "./chunk-EXEFSSF4.js";
 import {
   array,
   boolean,
@@ -367,7 +367,7 @@ import {
 import {
   installFileLogSink,
   setLogContext
-} from "./chunk-LFPWHI42.js";
+} from "./chunk-QKIFBA7N.js";
 import {
   APP_NAME,
   APP_TITLE,
@@ -413,7 +413,7 @@ import {
   shouldUseWindowsShell,
   signalProcessGroupOrProcess,
   typebox_exports
-} from "./chunk-UPVDERMD.js";
+} from "./chunk-2IPDSTDR.js";
 import {
   PRODUCT,
   PRODUCT_ENV
@@ -435,7 +435,7 @@ import {
   getProviders,
   modelsAreEqual,
   supportsFastMode
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 import {
   __commonJS,
   __export,
@@ -6370,6 +6370,9 @@ var SettingsManager = class _SettingsManager {
     const model = this.settings.autoRefine?.model;
     return model === void 0 ? void 0 : structuredClone(model);
   }
+  getLearningEnabled() {
+    return this.settings.learning?.enabled === true;
+  }
   getAutoRefineSettings() {
     const turnInterval = this.settings.autoRefine?.turnInterval;
     const cooldownMs = this.settings.autoRefine?.cooldownMs;
@@ -52785,7 +52788,7 @@ ${message}`, ["Yes", "No"], opts);
         if (this.sideQuestionBashDiscarded !== void 0) {
           this.sideQuestionBashDiscarded = void 0;
           this.activeBashComponent = void 0;
-          this.ui.requestRender();
+          this.ui.flushRender();
           break;
         }
         const component = this.activeBashComponent;
@@ -52803,7 +52806,7 @@ ${message}`, ["Yes", "No"], opts);
           this.sideQuestionBashComponent = void 0;
           this.finishSideQuestionBash(event, component.getOutput());
         }
-        this.ui.requestRender();
+        this.ui.flushRender();
         break;
       }
       case "message_start":
@@ -52876,7 +52879,7 @@ ${message}`, ["Yes", "No"], opts);
           this.streamingMessage = void 0;
           this.footer.invalidate();
         }
-        this.ui.requestRender();
+        this.ui.flushRender();
         break;
       case "tool_execution_start": {
         this.startedToolCalls.add(event.toolCallId);
@@ -52908,7 +52911,7 @@ ${message}`, ["Yes", "No"], opts);
           component.updateResult({ ...event.result, isError: event.isError });
           this.pendingTools.delete(event.toolCallId);
           this.startedToolCalls.delete(event.toolCallId);
-          this.ui.requestRender();
+          this.ui.flushRender();
         }
         break;
       }
@@ -52945,8 +52948,8 @@ ${message}`, ["Yes", "No"], opts);
         this.renderRecap();
         this.applyOptimisticContextUsage();
         void this.refreshConnectionContextUsage();
+        this.ui.flushRender();
         await this.checkShutdownRequested();
-        this.ui.requestRender();
         break;
       case "compaction_start": {
         this.startCompactionLoader(event.reason, event.customInstructions);

package/dist/bundle/{cli-main-MMHTK7EB.js → cli-main-BUK7QTHO.js}

@@ -5,10 +5,10 @@ import {
   isOwnedSessionWorkerProcess,
   maybeRunOwnedSessionWorkerFrontend,
   maybeStartDaemonEarly
-} from "./chunk-VCW3YDFF.js";
+} from "./chunk-EXEFSSF4.js";
 import {
   APP_NAME
-} from "./chunk-UPVDERMD.js";
+} from "./chunk-2IPDSTDR.js";
 import "./chunk-5QCJ5DQU.js";
 import "./chunk-CRBVRZBV.js";
 import "./chunk-GRJFXLBM.js";
@@ -16,7 +16,7 @@ import "./chunk-2T7M7VJ4.js";
 import "./chunk-UIRUOAIQ.js";
 import "./chunk-HXOBFFJX.js";
 import "./chunk-DGNX7HQ3.js";
-import "./chunk-W46VZFI6.js";
+import "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 
 // dist/cli-main.js
@@ -39,7 +39,7 @@ async function runCli() {
     }
     const [{ EnvHttpProxyAgent, setGlobalDispatcher }, { main }] = await Promise.all([
       import("undici"),
-      import("./main-XAIVH4Y7.js")
+      import("./main-2FWHT2WZ.js")
     ]);
     setGlobalDispatcher(new EnvHttpProxyAgent({ bodyTimeout: 0, headersTimeout: 0 }));
     try {

package/dist/bundle/cli.js

@@ -52,6 +52,6 @@ var supported = assertNodeVersion({
   exit: (code) => process.exit(code)
 });
 if (supported) {
-  const { runCli } = await import("./cli-main-MMHTK7EB.js");
+  const { runCli } = await import("./cli-main-BUK7QTHO.js");
   await runCli();
 }

package/dist/bundle/{google-T32FMD4G.js → google-HTMCUJ7X.js}

@@ -10,7 +10,7 @@ import {
   mapToolChoice,
   observeGoogleUsage,
   retainThoughtSignature
-} from "./chunk-KTNETTHI.js";
+} from "./chunk-Q3MBSBEB.js";
 import {
   buildBaseOptions,
   sanitizeSurrogates
@@ -26,7 +26,7 @@ import {
   formatStreamFailureMessage,
   recordStreamFailure,
   streamFailureFromStopReason
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 
 // ../ai/src/providers/google.ts

package/dist/bundle/{google-vertex-56KP6I3B.js → google-vertex-ZQO65AXF.js}

@@ -12,7 +12,7 @@ import {
   mapToolChoice,
   observeGoogleUsage,
   retainThoughtSignature
-} from "./chunk-KTNETTHI.js";
+} from "./chunk-Q3MBSBEB.js";
 import {
   buildBaseOptions,
   sanitizeSurrogates
@@ -25,7 +25,7 @@ import {
   formatStreamFailureMessage,
   recordStreamFailure,
   streamFailureFromStopReason
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 
 // ../ai/src/providers/google-vertex.ts

package/dist/bundle/{main-XAIVH4Y7.js → main-2FWHT2WZ.js}

@@ -17,11 +17,11 @@ import {
   shouldUseDaemonClientRuntime,
   shouldUseDaemonInteractive,
   shouldUseEphemeralSessionManagerForDaemonInteractive
-} from "./chunk-PFROC4EV.js";
-import "./chunk-VCW3YDFF.js";
+} from "./chunk-RVJVXZCS.js";
+import "./chunk-EXEFSSF4.js";
 import "./chunk-CFNASUYO.js";
-import "./chunk-LFPWHI42.js";
-import "./chunk-UPVDERMD.js";
+import "./chunk-QKIFBA7N.js";
+import "./chunk-2IPDSTDR.js";
 import "./chunk-5QCJ5DQU.js";
 import "./chunk-CRBVRZBV.js";
 import "./chunk-GRJFXLBM.js";
@@ -29,7 +29,7 @@ import "./chunk-2T7M7VJ4.js";
 import "./chunk-UIRUOAIQ.js";
 import "./chunk-HXOBFFJX.js";
 import "./chunk-DGNX7HQ3.js";
-import "./chunk-W46VZFI6.js";
+import "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 export {
   createDefaultRuntimeFactory,

package/dist/bundle/{mistral-WGBMYIVU.js → mistral-SB7LU56G.js}

@@ -45,7 +45,7 @@ import {
   clampThinkingLevel,
   recordStreamFailure,
   streamFailureFromStopReason
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 
 // ../../node_modules/@mistralai/mistralai/esm/lib/url.js

package/dist/bundle/{openai-codex-responses-PIV2RQ3G.js → openai-codex-responses-5CEW4VN3.js}

@@ -8,7 +8,7 @@ import {
   convertResponsesTools,
   observeResponsesEvent,
   processResponsesStream
-} from "./chunk-VSIYFWYI.js";
+} from "./chunk-KT2BNEPE.js";
 import "./chunk-NQB5DCFC.js";
 import {
   headersToRecord
@@ -33,9 +33,32 @@ import {
   formatThrownValue,
   isLocalRequestPreparationError,
   recordStreamFailure
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 
+// ../ai/src/utils/abortable-delay.ts
+function abortableDelay(ms, signal) {
+  return new Promise((resolve, reject) => {
+    if (signal?.aborted) {
+      reject(new Error("Request was aborted"));
+      return;
+    }
+    const cleanup = () => {
+      clearTimeout(timeout);
+      signal?.removeEventListener("abort", onAbort);
+    };
+    const onAbort = () => {
+      cleanup();
+      reject(new Error("Request was aborted"));
+    };
+    const timeout = setTimeout(() => {
+      cleanup();
+      resolve();
+    }, ms);
+    signal?.addEventListener("abort", onAbort, { once: true });
+  });
+}
+
 // ../ai/src/providers/openai-codex-responses.ts
 var _os = null;
 var dynamicImport = (specifier) => import(specifier);
@@ -78,19 +101,6 @@ function codexTransportError(error) {
     { cause: error }
   );
 }
-function sleep(ms, signal) {
-  return new Promise((resolve, reject) => {
-    if (signal?.aborted) {
-      reject(new Error("Request was aborted"));
-      return;
-    }
-    const timeout = setTimeout(resolve, ms);
-    signal?.addEventListener("abort", () => {
-      clearTimeout(timeout);
-      reject(new Error("Request was aborted"));
-    });
-  });
-}
 var streamOpenAICodexResponses = (model, context, options) => {
   const stream = new AssistantMessageEventStream();
   (async () => {
@@ -279,8 +289,10 @@ var streamOpenAICodexResponses = (model, context, options) => {
         }
         if (!(attempt < MAX_RETRIES && isRetryableError(lastError))) throw lastError;
         await attempts.settle("failed");
-        const delayMs = BASE_DELAY_MS * 2 ** attempt;
-        await sleep(delayMs, options?.signal);
+        const retryAfter = response?.headers.get("retry-after");
+        const retryAfterMs = retryAfter ? /^\d+(\.\d+)?$/.test(retryAfter) ? Number(retryAfter) * 1e3 : Date.parse(retryAfter) - Date.now() : NaN;
+        const delayMs = Number.isFinite(retryAfterMs) ? Math.max(0, retryAfterMs) : BASE_DELAY_MS * 2 ** attempt;
+        await abortableDelay(delayMs, options?.signal);
       }
       if (!response?.ok) {
         throw lastError ?? new Error("Failed after retries");
@@ -325,7 +337,8 @@ function buildRequestBody(model, context, options, onProjection) {
   const messages = convertResponsesMessages(model, context, CODEX_TOOL_CALL_PROVIDERS, {
     includeSystemPrompt: false,
     onProjection,
-    pendingPublicMessageGroups: options?.attempts?.pendingPublicMessageGroups
+    pendingPublicMessageGroups: options?.attempts?.pendingPublicMessageGroups,
+    responsesMessageIds: options?.attempts?.responsesMessageIds
   });
   const body = {
     model: model.id,
@@ -335,7 +348,7 @@ function buildRequestBody(model, context, options, onProjection) {
     input: messages,
     text: { verbosity: options?.textVerbosity || "low" },
     include: ["reasoning.encrypted_content"],
-    prompt_cache_key: options?.sessionId,
+    prompt_cache_key: options?.cacheRetention === "none" ? void 0 : options?.sessionId,
     tool_choice: "auto",
     parallel_tool_calls: true
   };
@@ -505,6 +518,13 @@ async function* parseSSE(response) {
 }
 var OPENAI_BETA_RESPONSES_WEBSOCKETS = "responses_websockets=2026-02-06";
 var SESSION_WEBSOCKET_CACHE_TTL_MS = 5 * 60 * 1e3;
+var WEBSOCKET_AUTH_HEADERS = [
+  "authorization",
+  "chatgpt-account-id",
+  "openai-organization",
+  "openai-project",
+  "x-api-key"
+];
 var websocketSessionCache = /* @__PURE__ */ new Map();
 var websocketDebugStats = /* @__PURE__ */ new Map();
 var websocketSseFallbackSessions = /* @__PURE__ */ new Set();
@@ -613,7 +633,9 @@ function scheduleSessionWebSocketExpiry(sessionId, entry) {
   entry.idleTimer = setTimeout(() => {
     if (entry.busy) return;
     closeWebSocketSilently(entry.socket, 1e3, "idle_timeout");
-    websocketSessionCache.delete(sessionId);
+    if (websocketSessionCache.get(sessionId) === entry) {
+      websocketSessionCache.delete(sessionId);
+    }
   }, SESSION_WEBSOCKET_CACHE_TTL_MS);
 }
 async function connectWebSocket(url, headers, signal, timeoutMs = 3e4) {
@@ -705,16 +727,19 @@ async function acquireWebSocket(url, headers, sessionId, signal, timeoutMs) {
       clearTimeout(cached.idleTimer);
       cached.idleTimer = void 0;
     }
-    if (!cached.busy && isWebSocketReusable(cached.socket)) {
+    const matchesConnection = cached.url === url && WEBSOCKET_AUTH_HEADERS.every((name) => cached.headers.get(name) === headers.get(name));
+    if (!cached.busy && matchesConnection && isWebSocketReusable(cached.socket)) {
       cached.busy = true;
       return {
         socket: cached.socket,
         entry: cached,
         reused: true,
         release: ({ keep } = {}) => {
-          if (!keep || !isWebSocketReusable(cached.socket)) {
+          if (!keep || !isWebSocketReusable(cached.socket) || websocketSessionCache.get(sessionId) !== cached) {
             closeWebSocketSilently(cached.socket);
-            websocketSessionCache.delete(sessionId);
+            if (websocketSessionCache.get(sessionId) === cached) {
+              websocketSessionCache.delete(sessionId);
+            }
             return;
           }
           cached.busy = false;
@@ -732,20 +757,21 @@ async function acquireWebSocket(url, headers, sessionId, signal, timeoutMs) {
         }
       };
     }
-    if (!isWebSocketReusable(cached.socket)) {
-      closeWebSocketSilently(cached.socket);
+    cached.continuation = void 0;
+    closeWebSocketSilently(cached.socket);
+    if (websocketSessionCache.get(sessionId) === cached) {
       websocketSessionCache.delete(sessionId);
     }
   }
   const socket = await connectWebSocket(url, headers, signal, timeoutMs);
-  const entry = { socket, url, busy: true };
+  const entry = { socket, url, headers: new Headers(headers), busy: true };
   websocketSessionCache.set(sessionId, entry);
   return {
     socket,
     entry,
     reused: false,
     release: ({ keep } = {}) => {
-      if (!keep || !isWebSocketReusable(entry.socket)) {
+      if (!keep || !isWebSocketReusable(entry.socket) || websocketSessionCache.get(sessionId) !== entry) {
         closeWebSocketSilently(entry.socket);
         if (entry.idleTimer) clearTimeout(entry.idleTimer);
         if (websocketSessionCache.get(sessionId) === entry) {

package/dist/bundle/{openai-completions-64C3JPR4.js → openai-completions-BAZJQSS2.js}

@@ -36,7 +36,7 @@ import {
   formatStreamFailureMessage,
   recordStreamFailure,
   withRequestBody
-} from "./chunk-W46VZFI6.js";
+} from "./chunk-D3KOF57W.js";
 import "./chunk-BAWSWWEU.js";
 
 // ../ai/src/providers/openai-completions.ts