@reactor-team/js-sdk 2.3.1 → 2.3.2

package/dist/index.mjs

@@ -209,8 +209,9 @@ function closePeerConnection(pc) {
 
 // src/core/CoordinatorClient.ts
 var INITIAL_BACKOFF_MS = 500;
-var MAX_BACKOFF_MS = 3e4;
+var MAX_BACKOFF_MS = 15e3;
 var BACKOFF_MULTIPLIER = 2;
+var DEFAULT_MAX_ATTEMPTS = 6;
 var CoordinatorClient = class {
   constructor(options) {
     this.baseUrl = options.baseUrl;
@@ -403,13 +404,14 @@ var CoordinatorClient = class {
     });
   }
   /**
-   * Polls for the SDP answer with geometric backoff.
+   * Polls for the SDP answer with exponential backoff.
    * Used for async reconnection when the answer is not immediately available.
    * @param sessionId - The session ID to poll for
+   * @param maxAttempts - Optional maximum number of polling attempts before giving up
    * @returns The SDP answer from the server
    */
-  pollSdpAnswer(sessionId) {
-    return __async(this, null, function* () {
+  pollSdpAnswer(_0) {
+    return __async(this, arguments, function* (sessionId, maxAttempts = DEFAULT_MAX_ATTEMPTS) {
       console.debug(
         "[CoordinatorClient] Polling for SDP answer for session:",
         sessionId
@@ -417,9 +419,14 @@ var CoordinatorClient = class {
       let backoffMs = INITIAL_BACKOFF_MS;
       let attempt = 0;
       while (true) {
+        if (attempt >= maxAttempts) {
+          throw new Error(
+            `SDP polling exceeded maximum attempts (${maxAttempts}) for session ${sessionId}`
+          );
+        }
         attempt++;
         console.debug(
-          `[CoordinatorClient] SDP poll attempt ${attempt} for session ${sessionId}`
+          `[CoordinatorClient] SDP poll attempt ${attempt}/${maxAttempts} for session ${sessionId}`
         );
         const response = yield fetch(
           `${this.baseUrl}/sessions/${sessionId}/sdp_params`,
@@ -456,9 +463,10 @@ var CoordinatorClient = class {
    * falls back to polling. If no sdpOffer is provided, goes directly to polling.
    * @param sessionId - The session ID to connect to
    * @param sdpOffer - Optional SDP offer from the local WebRTC peer connection
+   * @param maxAttempts - Optional maximum number of polling attempts before giving up
    * @returns The SDP answer from the server
    */
-  connect(sessionId, sdpOffer) {
+  connect(sessionId, sdpOffer, maxAttempts) {
     return __async(this, null, function* () {
       console.debug("[CoordinatorClient] Connecting to session:", sessionId);
       if (sdpOffer) {
@@ -467,7 +475,7 @@ var CoordinatorClient = class {
           return answer;
         }
       }
-      return this.pollSdpAnswer(sessionId);
+      return this.pollSdpAnswer(sessionId, maxAttempts);
     });
   }
   /**
@@ -1011,8 +1019,9 @@ var Reactor = class {
   }
   /**
    * Public method for reconnecting to an existing session, that may have been interrupted but can be recovered.
+   * @param options Optional connect options (e.g. maxAttempts for SDP polling)
    */
-  reconnect() {
+  reconnect(options) {
     return __async(this, null, function* () {
       if (!this.sessionId || !this.coordinatorClient) {
         console.warn("[Reactor] No active session to reconnect to.");
@@ -1032,7 +1041,8 @@ var Reactor = class {
       try {
         const sdpAnswer = yield this.coordinatorClient.connect(
           this.sessionId,
-          sdpOffer
+          sdpOffer,
+          options == null ? void 0 : options.maxAttempts
         );
         yield this.machineClient.connect(sdpAnswer);
         this.setStatus("ready");
@@ -1056,8 +1066,10 @@ var Reactor = class {
    * Connects to the coordinator and waits for a GPU to be assigned.
    * Once a GPU is assigned, the Reactor will connect to the gpu machine via WebRTC.
    * If no authentication is provided and not in local mode, an error is thrown.
+   * @param jwtToken Optional JWT token for authentication
+   * @param options Optional connect options (e.g. maxAttempts for SDP polling)
    */
-  connect(jwtToken) {
+  connect(jwtToken, options) {
     return __async(this, null, function* () {
       console.debug("[Reactor] Connecting, status:", this.status);
       if (jwtToken == void 0 && !this.local) {
@@ -1074,7 +1086,7 @@ var Reactor = class {
         this.coordinatorClient = this.local ? new LocalCoordinatorClient(this.coordinatorUrl) : new CoordinatorClient({
           baseUrl: this.coordinatorUrl,
           jwtToken,
-          // Safe: validated on line 186-188
+          // Safe: validated above
           model: this.model
         });
         const iceServers = yield this.coordinatorClient.getIceServers();
@@ -1083,7 +1095,11 @@ var Reactor = class {
         const sdpOffer = yield this.machineClient.createOffer();
         const sessionId = yield this.coordinatorClient.createSession(sdpOffer);
         this.setSessionId(sessionId);
-        const sdpAnswer = yield this.coordinatorClient.connect(sessionId);
+        const sdpAnswer = yield this.coordinatorClient.connect(
+          sessionId,
+          void 0,
+          options == null ? void 0 : options.maxAttempts
+        );
         yield this.machineClient.connect(sdpAnswer);
       } catch (error) {
         console.error("[Reactor] Connection failed:", error);
@@ -1093,7 +1109,14 @@ var Reactor = class {
           "coordinator",
           true
         );
-        this.setStatus("disconnected");
+        try {
+          yield this.disconnect(false);
+        } catch (disconnectError) {
+          console.error(
+            "[Reactor] Failed to clean up after connection failure:",
+            disconnectError
+          );
+        }
         throw error;
       }
     });
@@ -1325,13 +1348,13 @@ var createReactorStore = (initProps, publicState = defaultInitState) => {
           throw error;
         }
       }),
-      connect: (jwtToken) => __async(null, null, function* () {
+      connect: (jwtToken, options) => __async(null, null, function* () {
         if (jwtToken === void 0) {
           jwtToken = get().jwtToken;
         }
         console.debug("[ReactorStore] Connect called.");
         try {
-          yield get().internal.reactor.connect(jwtToken);
+          yield get().internal.reactor.connect(jwtToken, options);
           console.debug("[ReactorStore] Connect completed successfully");
         } catch (error) {
           console.error("[ReactorStore] Connect failed:", error);
@@ -1376,10 +1399,10 @@ var createReactorStore = (initProps, publicState = defaultInitState) => {
           throw error;
         }
       }),
-      reconnect: () => __async(null, null, function* () {
+      reconnect: (options) => __async(null, null, function* () {
         console.debug("[ReactorStore] Reconnecting");
         try {
-          yield get().internal.reactor.reconnect();
+          yield get().internal.reactor.reconnect(options);
           console.debug("[ReactorStore] Reconnect completed successfully");
         } catch (error) {
           console.error("[ReactorStore] Failed to reconnect:", error);
@@ -1396,11 +1419,11 @@ import { jsx } from "react/jsx-runtime";
 function ReactorProvider(_a) {
   var _b = _a, {
     children,
-    autoConnect = true,
+    connectOptions,
     jwtToken
   } = _b, props = __objRest(_b, [
     "children",
-    "autoConnect",
+    "connectOptions",
     "jwtToken"
   ]);
   const storeRef = useRef(void 0);
@@ -1415,14 +1438,16 @@ function ReactorProvider(_a) {
     );
     console.debug("[ReactorProvider] Reactor store created successfully");
   }
+  const _a2 = connectOptions != null ? connectOptions : {}, { autoConnect = false } = _a2, pollingOptions = __objRest(_a2, ["autoConnect"]);
   const { coordinatorUrl, modelName, local } = props;
+  const maxAttempts = pollingOptions.maxAttempts;
   useEffect(() => {
     const handleBeforeUnload = () => {
-      var _a2;
+      var _a3;
       console.debug(
         "[ReactorProvider] Page unloading, performing non-recoverable disconnect"
       );
-      (_a2 = storeRef.current) == null ? void 0 : _a2.getState().internal.reactor.disconnect(false);
+      (_a3 = storeRef.current) == null ? void 0 : _a3.getState().internal.reactor.disconnect(false);
     };
     window.addEventListener("beforeunload", handleBeforeUnload);
     return () => {
@@ -1437,7 +1462,7 @@ function ReactorProvider(_a) {
         console.debug(
           "[ReactorProvider] Starting autoconnect in first render..."
         );
-        current2.getState().connect(jwtToken).then(() => {
+        current2.getState().connect(jwtToken, pollingOptions).then(() => {
           console.debug(
             "[ReactorProvider] Autoconnect successful in first render"
           );
@@ -1480,7 +1505,7 @@ function ReactorProvider(_a) {
     );
     if (autoConnect && current.getState().status === "disconnected" && jwtToken) {
       console.debug("[ReactorProvider] Starting autoconnect...");
-      current.getState().connect(jwtToken).then(() => {
+      current.getState().connect(jwtToken, pollingOptions).then(() => {
         console.debug("[ReactorProvider] Autoconnect successful");
       }).catch((error) => {
         console.error("[ReactorProvider] Failed to autoconnect:", error);
@@ -1496,7 +1521,7 @@ function ReactorProvider(_a) {
         console.error("[ReactorProvider] Failed to disconnect:", error);
       });
     };
-  }, [coordinatorUrl, modelName, autoConnect, local, jwtToken]);
+  }, [coordinatorUrl, modelName, autoConnect, local, jwtToken, maxAttempts]);
   return /* @__PURE__ */ jsx(ReactorContext.Provider, { value: storeRef.current, children });
 }
 function useReactorStore(selector) {