Merge pull request #1929 from open-webui/dev

fix

Merge pull request #1929 from open-webui/dev
fix
b1eb08b4 · Timothy Jaeryang Baek · GitHub · 443a256f · 989a4d31 · b1eb08b4
Unverified Commit b1eb08b4 authored May 02, 2024 by Timothy Jaeryang Baek Committed by GitHub May 02, 2024
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

backend/apps/images/main.py backend/apps/images/main.py +2 -2

No files found.
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -402,7 +402,7 @@ def generate_image(
            for image in res["data"]:
                image_filename = save_b64_image(image["b64_json"])
                images.append({"url": f"/cache/image/generations/{image_filename}"})
-                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_id}.json")
+                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_filename}.json")

                with open(file_body_path, "w") as f:
                    json.dump(data, f)
@@ -477,7 +477,7 @@ def generate_image(
            for image in res["images"]:
                image_filename = save_b64_image(image)
                images.append({"url": f"/cache/image/generations/{image_filename}"})
-                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_id}.json")
+                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_filename}.json")

                with open(file_body_path, "w") as f:
                    json.dump({**data, "info": res["info"]}, f)