change fetch request timeouts to 20 hours from 300s

SaladTechnologies · Jan 21, 2025 · 69e57d1 · 69e57d1
1 parent 1342e22
commit 69e57d1
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "comfyui-api",
-  "version": "1.8.0",
+  "version": "1.8.1",
   "description": "Wraps comfyui to make it easier to use as a stateless web service",
   "main": "dist/src/index.js",
   "scripts": {

diff --git a/src/comfy.ts b/src/comfy.ts
@@ -76,6 +76,7 @@ export async function warmupComfyUI(): Promise<void> {
         "Content-Type": "application/json",
       },
       body: JSON.stringify({ prompt: config.warmupPrompt }),
+      signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
     });
     if (!resp.ok) {
       throw new Error(`Failed to warmup Comfy UI: ${await resp.text()}`);

diff --git a/src/server.ts b/src/server.ts
@@ -334,6 +334,7 @@ server.after(() => {
                     filename,
                     prompt,
                   }),
+                  signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
                 })
                   .catch((e: any) => {
                     app.log.error(
@@ -376,6 +377,7 @@ server.after(() => {
                   prompt,
                   error: e.message,
                 }),
+                signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
               });
 
               if (!resp.ok) {
@@ -495,6 +497,7 @@ server.after(() => {
                   "Content-Type": "application/json",
                 },
                 body: JSON.stringify({ prompt, id, webhook, convert_output }),
+                signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
               }
             );
             const body = await resp.json();

diff --git a/test/test-utils.ts b/test/test-utils.ts
@@ -48,7 +48,7 @@ export async function submitPrompt(
       "Content-Type": "application/json",
     },
     body: JSON.stringify(body),
-    signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 minutes
+    signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
   });
   if (!resp.ok) {
     console.error(await resp.text());