fix: retry api only when request failed to comfyui api

gazai-io · Dec 23, 2024 · 188b994 · 188b994
1 parent a836879
commit 188b994
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rp_handler.py b/src/rp_handler.py
@@ -173,7 +173,7 @@ def queue_workflow(workflow):
         try:
             req = urllib.request.Request(f"http://{COMFY_HOST}/prompt", data=data)
             return json.loads(urllib.request.urlopen(req).read())
-        except Exception as e:
+        except requests.RequestException as e:
             print(f"Error queuing workflow: {str(e)}. Retrying...")
             time.sleep(COMFY_API_AVAILABLE_INTERVAL_MS / 1000)
             retries += 1