From df2f18eda98f7346002bac1da1509092d4af9086 Mon Sep 17 00:00:00 2001 From: BennyKok Date: Thu, 21 Dec 2023 00:07:38 +0800 Subject: [PATCH] feat(plugin): upload "files" in node output --- custom_routes.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/custom_routes.py b/custom_routes.py index 27a39af..fa2d30c 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -265,6 +265,11 @@ async def update_run_with_output(prompt_id, data): images = data.get('images', []) for image in images: await upload_file(prompt_id, image.get("filename"), subfolder=image.get("subfolder"), type=image.get("type", "image/png")) + + files = data.get('files', []) + for file in files: + await upload_file(prompt_id, file.get("filename"), subfolder=file.get("subfolder"), type=file.get("type", "image/png")) + except Exception as e: error_type = type(e).__name__ stack_trace = traceback.format_exc().strip()