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

fix(plugin): try catch update run endpoint call

Browse Source
This commit is contained in:
BennyKok 2023-12-22 12:59:05 +08:00
parent 2690db12b5
commit b185d95e93
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
custom_routes.py
View File

@ -202,7 +202,13 @@ def update_run(prompt_id, status: Status):
"status": status.value, "status": status.value,
} }
prompt_metadata[prompt_id]['status'] = status prompt_metadata[prompt_id]['status'] = status
try:
requests.post(status_endpoint, json=body) requests.post(status_endpoint, json=body)
except Exception as e:
error_type = type(e).__name__
stack_trace = traceback.format_exc().strip()
print(f"Error occurred while updating run: {e} {stack_trace}")
async def upload_file(prompt_id, filename, subfolder=None, content_type="image/png", type="output"): async def upload_file(prompt_id, filename, subfolder=None, content_type="image/png", type="output"):