chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: safety check with when prompt has not yet running

Browse Source
This commit is contained in:
BennyKok 2024-01-12 19:42:50 +08:00
parent 8a42506908
commit f99b6f34ff
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_routes.py
View File

@ -170,7 +170,7 @@ async def websocket_handler(request):
async def comfy_deploy_check_status(request): async def comfy_deploy_check_status(request):
prompt_server = server.PromptServer.instance prompt_server = server.PromptServer.instance
prompt_id = request.rel_url.query.get('prompt_id', None) 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({ return web.json_response({
"status": prompt_metadata[prompt_id]['status'].value "status": prompt_metadata[prompt_id]['status'].value
}) })