Bound offline ComfyUI metadata probes

f4309221b126179b42ec9664e0948610685fba54

permissionBRICK <40219477+permissionBRICK@users.noreply.github.com>

src/endpoints/stable-diffusion.js+17 -6 Viewed

 });
 const comfy = express.Router();
+const COMFY_METADATA_TIMEOUT_MS = 5000;
+/**
+ * Fetches lightweight ComfyUI status or metadata without allowing an offline
+ * backend to occupy a browser-facing request until the OS TCP timeout.
+ * @param {URL} url ComfyUI endpoint URL
+ * @returns {Promise<import('node-fetch').Response>} Fetch response
+ */
+function fetchComfyMetadata(url) {
+    return fetch(url, { signal: AbortSignal.timeout(COMFY_METADATA_TIMEOUT_MS) });
+}
 comfy.post('/ping', async (request, response) => {
     try {
         const url = new URL(urlJoin(request.body.url, '/system_stats'));
-        const result = await fetch(url);
+        const result = await fetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }

     try {
         const url = new URL(urlJoin(request.body.url, '/object_info'));
-        const result = await fetch(url);
+        const result = await fetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }

     try {
         const url = new URL(urlJoin(request.body.url, '/object_info'));
-        const result = await fetch(url);
+        const result = await fetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }

     try {
         const url = new URL(urlJoin(request.body.url, '/object_info'));
-        const result = await fetch(url);
+        const result = await fetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }

     try {
         const url = new URL(urlJoin(request.body.url, '/object_info'));
-        const result = await fetch(url);
+        const result = await fetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }

     try {
         const url = new URL(urlJoin(request.body.url, '/object_info'));
-        const result = await fetch(url);
+        const result = await fetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }