chore: return item on upload

2024-10-04 15:27:31 -07:00 · 2024-10-04 15:27:31 -07:00 · 2f33bcf497
commit 2f33bcf497
parent f812d9d698
1 changed files with 2 additions and 0 deletions
--- a/custom_routes.py
+++ b/custom_routes.py
@ -1365,6 +1365,8 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p
        item["upload_duration"] = end_time - start_time
        if ok.get("is_public") is not None:
            item["is_public"] = ok.get("is_public")
+            
+    return item

 def have_pending_upload(prompt_id):
    if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: