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 fetchfetchComfyMetadata(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 fetchfetchComfyMetadata(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 fetchfetchComfyMetadata(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 fetchfetchComfyMetadata(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 fetchfetchComfyMetadata(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 fetchfetchComfyMetadata(url);
         if (!result.ok) {
             throw new Error('ComfyUI returned an error.');
+        }