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

chore: return item on upload

Browse Source
This commit is contained in:
bennykok 2024-10-04 15:27:31 -07:00
parent f812d9d698
commit 2f33bcf497
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_routes.py
View File

@ -1365,6 +1365,8 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p
item["upload_duration"] = end_time - start_time item["upload_duration"] = end_time - start_time
if ok.get("is_public") is not None: if ok.get("is_public") is not None:
item["is_public"] = ok.get("is_public") item["is_public"] = ok.get("is_public")
return item
def have_pending_upload(prompt_id): def have_pending_upload(prompt_id):
if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: