Update proxy.js

proxy.js

@@ -22,6 +22,7 @@ async function processQueue() {
 
 async function retryWith503Backoff(url, options, startTime) {
   let attempt = 0;
+  let lastResponse = null;
   
   while (true) {
     const elapsed = Date.now() - startTime;
@@ -30,21 +31,43 @@ async function retryWith503Backoff(url, options, startTime) {
     }
     
     const response = await fetch(url, options);
+    lastResponse = response;
     
-    if (response.status !== 503) {
+    // If successful (2xx), return immediately
+    if (response.ok) {
       return response;
     }
     
-    // Exponential backoff: 1s, 2s, 4s, 8s, 16s, 32s, 64s...
-    const delay = Math.min(1000 * Math.pow(2, attempt), 64000);
-    attempt++;
-    
-    // Check if waiting would exceed max time
-    if (elapsed + delay > MAX_WAIT_MS) {
-      throw new Error('Max wait time would be exceeded');
+    // For 503: retry with exponential backoff until max time
+    // For other errors: retry up to 3 times, then return the error
+    if (response.status === 503) {
+      // Exponential backoff for 503: 1s, 2s, 4s, 8s, 16s, 32s, 64s...
+      const delay = Math.min(1000 * Math.pow(2, attempt), 64000);
+      attempt++;
+      
+      // Check if waiting would exceed max time
+      if (elapsed + delay > MAX_WAIT_MS) {
+        return response; // Return 503 if we'd exceed max time
+      }
+      
+      await new Promise(resolve => setTimeout(resolve, delay));
+    } else {
+      // For non-503 errors, retry up to 3 times with shorter delays
+      if (attempt >= 3) {
+        return response; // Return the error after 3 attempts
+      }
+      
+      // Short delay for non-503 errors: 1s, 2s, 3s
+      const delay = (attempt + 1) * 1000;
+      attempt++;
+      
+      // Check if waiting would exceed max time
+      if (elapsed + delay > MAX_WAIT_MS) {
+        return response;
+      }
+      
+      await new Promise(resolve => setTimeout(resolve, delay));
     }
-    
-    await new Promise(resolve => setTimeout(resolve, delay));
   }
 }