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
@ -1366,6 +1366,8 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p
        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:
        logger.info(f"Have pending upload {len(prompt_metadata[prompt_id].uploading_nodes)}")