@skydiveai/pi-extensions 0.1.0-beta.2291 → 0.1.0-beta.2292

package/dist/index.mjs

@@ -956,6 +956,27 @@ function createStderrBuffer({ maxBytes }) {
 */
 const DEFAULT_CONNECT_TIMEOUT_MS = 5e3;
 const STDERR_BUFFER_BYTES = 4096;
+const SESSION_NOT_FOUND_RETRY_DELAYS_MS = [
+	250,
+	500,
+	1e3,
+	2e3,
+	4e3,
+	8e3
+];
+function delay(ms) {
+	return new Promise((resolve) => setTimeout(resolve, ms));
+}
+async function fetchWithSessionNotFoundRetry(input, init) {
+	for (let attempt = 0;; attempt += 1) {
+		const response = await fetch(input, init);
+		const sessionId = new Headers(init?.headers).get("mcp-session-id");
+		const retryDelay = SESSION_NOT_FOUND_RETRY_DELAYS_MS[attempt];
+		if (!sessionId || response.status !== 404 || retryDelay === void 0 || !(await response.clone().text()).includes("session_not_found")) return response;
+		await response.body?.cancel();
+		await delay(retryDelay);
+	}
+}
 /**
 * Sandbox egress variables a stdio MCP server needs to make outbound HTTPS
 * calls. The SDK's default child environment is a tiny whitelist (HOME,
@@ -1039,7 +1060,10 @@ function probeAlive(pid) {
 	}
 }
 async function connectHttp(_id, config, client) {
-	const transport = new StreamableHTTPClientTransport(new URL(config.url), { ...config.headers !== null ? { requestInit: { headers: config.headers } } : {} });
+	const transport = new StreamableHTTPClientTransport(new URL(config.url), {
+		...config.headers !== null ? { requestInit: { headers: config.headers } } : {},
+		fetch: fetchWithSessionNotFoundRetry
+	});
 	try {
 		await client.connect(transport);
 		return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skydiveai/pi-extensions",
-  "version": "0.1.0-beta.2291",
+  "version": "0.1.0-beta.2292",
   "homepage": "https://skydive.com",
   "license": "MIT",
   "author": "Create, Inc.",