@pyxmate/memory 1.26.3 → 1.26.5

package/dist/cli/pyx-mem.mjs

@@ -866,6 +866,19 @@ function mcpJson(payload) {
 
 // src/mcp/http-client.ts
 var DEFAULT_TIMEOUT_MS = 15e3;
+function parseRetryAfterSeconds(header) {
+  const value = header?.trim() ?? "";
+  return /^\d+$/.test(value) ? Number(value) : void 0;
+}
+var HTTP_DATE = /^(?:[A-Z][a-z]{2}, \d{2} [A-Z][a-z]{2} \d{4} \d{2}:\d{2}:\d{2} GMT|[A-Z][a-z]{5,8}, \d{2}-[A-Z][a-z]{2}-\d{2} \d{2}:\d{2}:\d{2} GMT|[A-Z][a-z]{2} [A-Z][a-z]{2} [ \d]\d \d{2}:\d{2}:\d{2} \d{4})$/;
+function parseRetryAfterMs(header) {
+  const seconds = parseRetryAfterSeconds(header);
+  if (seconds !== void 0) return seconds * 1e3;
+  const value = header?.trim() ?? "";
+  if (!HTTP_DATE.test(value)) return void 0;
+  const at = Date.parse(value.endsWith(" GMT") ? value : `${value} GMT`);
+  return Number.isNaN(at) ? void 0 : Math.max(0, at - Date.now());
+}
 function validateIdempotencyKey(key) {
   if (key.length === 0 || !key.isWellFormed() || new TextEncoder().encode(key).byteLength > 256) {
     throw new Error("idempotencyKey must be 1-256 well-formed UTF-8 bytes");
@@ -2126,6 +2139,7 @@ function createReadCredentials(providerFactory) {
 import { createHash as createHash5, randomUUID as randomUUID5 } from "crypto";
 import { readFile as readFile2, stat as stat2 } from "fs/promises";
 import { basename as basename2, isAbsolute as isAbsolute2, resolve as resolve2 } from "path";
+import { setTimeout as delay2 } from "timers/promises";
 import { Client as McpClient } from "@modelcontextprotocol/sdk/client/index.js";
 import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
@@ -2317,8 +2331,7 @@ async function pause(ms, signal) {
   });
 }
 function retryAfterMs(header) {
-  const value = header?.trim() ?? "";
-  const seconds = /^\d+$/.test(value) ? Number(value) : DEFAULT_RETRY_AFTER_S;
+  const seconds = parseRetryAfterSeconds(header) ?? DEFAULT_RETRY_AFTER_S;
   return Math.min(seconds, MAX_RETRY_AFTER_S) * 1e3;
 }
 function formatCents(microcents, decimals) {
@@ -2736,6 +2749,67 @@ function createGuardedFetch(guard, fetchImpl = fetch) {
   };
   return Object.assign(wrapped, { preconnect: fetch.preconnect });
 }
+var RETRYABLE_MCP_METHODS = /* @__PURE__ */ new Set([
+  "initialize",
+  "notifications/initialized",
+  "tools/list",
+  "prompts/list"
+]);
+var MCP_RETRY_BUDGET_MS = 2e4;
+var MIN_RESEND_SPACING_MS = 1e3;
+function isRetryableMcpMessage(input, init) {
+  const method = init?.method ?? (input instanceof Request ? input.method : "GET");
+  if (method.toUpperCase() !== "POST" || typeof init?.body !== "string") return false;
+  let message;
+  try {
+    message = JSON.parse(init.body);
+  } catch {
+    return false;
+  }
+  if (typeof message !== "object" || message === null) return false;
+  const { method: rpc } = message;
+  return typeof rpc === "string" && RETRYABLE_MCP_METHODS.has(rpc);
+}
+function resendDelayMs(res, deadline) {
+  if (res.status !== 429 && res.status !== 503) return void 0;
+  const retryAfterMs2 = parseRetryAfterMs(res.headers.get("retry-after"));
+  if (retryAfterMs2 === void 0) return void 0;
+  const waitMs = Math.max(retryAfterMs2, MIN_RESEND_SPACING_MS);
+  return Date.now() + waitMs > deadline ? void 0 : waitMs;
+}
+function createMcpRetryAfterFetch(fetchImpl = fetch, retryUntil = () => Date.now() + MCP_RETRY_BUDGET_MS) {
+  const wrapped = async (input, init) => {
+    if (!isRetryableMcpMessage(input, init)) return fetchImpl(input, init);
+    const deadline = retryUntil();
+    const signal = init?.signal ?? (input instanceof Request ? input.signal : void 0);
+    for (; ; ) {
+      const res = await fetchImpl(input, init);
+      const waitMs = resendDelayMs(res, deadline);
+      if (waitMs === void 0) return res;
+      await res.body?.cancel();
+      await delay2(waitMs, void 0, { signal });
+    }
+  };
+  return Object.assign(wrapped, { preconnect: fetch.preconnect });
+}
+async function connectRemoteMcp(client, url, apiKey, fetchImpl, budgetMs = MCP_RETRY_BUDGET_MS) {
+  const connectUntil = Date.now() + budgetMs;
+  let connecting = true;
+  const transport = new StreamableHTTPClientTransport(url, {
+    requestInit: { headers: { Authorization: `Bearer ${apiKey}` } },
+    fetch: createMcpIdempotentFetch(
+      createMcpRetryAfterFetch(
+        fetchImpl,
+        () => connecting ? connectUntil : Date.now() + budgetMs
+      )
+    )
+  });
+  try {
+    await client.connect(transport);
+  } finally {
+    connecting = false;
+  }
+}
 function extractLocalUpload(meta) {
   if (meta === null || typeof meta !== "object") return void 0;
   const descriptor = meta[LOCAL_UPLOAD_META_KEY];
@@ -2877,7 +2951,7 @@ function createProxyServer(client, version, uploadLocalFile, hostedFileIngest) {
   return server;
 }
 async function runMcpProxyServer(opts) {
-  const version = opts.version ?? (true ? "1.26.3" : "0.0.0-dev");
+  const version = opts.version ?? (true ? "1.26.5" : "0.0.0-dev");
   const read = await opts.readCredentials();
   if (!read.ok) {
     throw new Error(
@@ -2888,12 +2962,8 @@ async function runMcpProxyServer(opts) {
   const url = new URL(`${endpoint.replace(/\/+$/, "")}${REMOTE_MCP_PATH}`);
   const guard = createCredentialGuard(opts.readCredentials, read.credentials);
   const client = new McpClient({ name: "pyx-mem-proxy", version }, { capabilities: {} });
-  const clientTransport = new StreamableHTTPClientTransport(url, {
-    requestInit: { headers: { Authorization: `Bearer ${apiKey}` } },
-    fetch: createMcpIdempotentFetch(createGuardedFetch(guard))
-  });
   try {
-    await client.connect(clientTransport);
+    await connectRemoteMcp(client, url, apiKey, createGuardedFetch(guard));
   } catch (err) {
     const msg = err instanceof Error ? err.message : String(err);
     throw new Error(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pyxmate/memory",
-  "version": "1.26.3",
+  "version": "1.26.5",
   "type": "module",
   "description": "SDK for pyx-memory — Memory as a Service for AI agents",
   "license": "MIT",