From cc31840d41685ee32aa7fc0193675d7757fb9e8a Mon Sep 17 00:00:00 2001 From: bennykok Date: Sun, 25 Feb 2024 00:17:40 -0800 Subject: [PATCH] fix(plugin): comfy_deploy_check_ws_status --- custom_routes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/custom_routes.py b/custom_routes.py index b320b5b..9c4cefb 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -456,7 +456,7 @@ async def comfy_deploy_check_ws_status(request): client_id = request.rel_url.query.get('client_id', None) if client_id in streaming_prompt_metadata: remaining_queue = 0 # Initialize remaining queue count - for prompt_id in streaming_prompt_metadata[client_id].running_prompt_ids[client_id]: + for prompt_id in streaming_prompt_metadata[client_id].running_prompt_ids: prompt_status = prompt_metadata[prompt_id].status if prompt_status not in [Status.FAILED, Status.SUCCESS]: remaining_queue += 1 # Increment for each prompt still running