fix: safety check with when prompt has not yet running

2024-01-12 19:42:50 +08:00 · 2024-01-12 19:42:50 +08:00 · f99b6f34ff
commit f99b6f34ff
parent 8a42506908
1 changed files with 1 additions and 1 deletions
--- a/custom_routes.py
+++ b/custom_routes.py
@ -170,7 +170,7 @@ async def websocket_handler(request):
 async def comfy_deploy_check_status(request):
    prompt_server = server.PromptServer.instance
    prompt_id = request.rel_url.query.get('prompt_id', None)
-    if prompt_id in prompt_metadata:
+    if prompt_id in prompt_metadata and 'status' in prompt_metadata[prompt_id]:
        return web.json_response({
            "status": prompt_metadata[prompt_id]['status'].value
        })