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

fix(plugin): update prompt metadata status properly with realtime prompt

Browse Source
This commit is contained in:
bennykok 2024-02-26 00:09:11 -08:00
parent 1c9c32e9e4
commit 7b653201ae
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

@ -600,6 +600,7 @@ def update_run(prompt_id: str, status: Status):
# if prompt_metadata[prompt_id].start_time is None and status == Status.RUNNING: # if prompt_metadata[prompt_id].start_time is None and status == Status.RUNNING:
# if its realtime prompt we need to skip that. # if its realtime prompt we need to skip that.
if prompt_metadata[prompt_id].is_realtime is True: if prompt_metadata[prompt_id].is_realtime is True:
prompt_metadata[prompt_id].status = status
return return
if (prompt_metadata[prompt_id].status != status): if (prompt_metadata[prompt_id].status != status):
@ -649,7 +650,6 @@ def update_run(prompt_id: str, status: Status):
except Exception as log_error: except Exception as log_error:
print(f"Error reading log file: {log_error}") print(f"Error reading log file: {log_error}")
except Exception as e: except Exception as e:
error_type = type(e).__name__ error_type = type(e).__name__
stack_trace = traceback.format_exc().strip() stack_trace = traceback.format_exc().strip()