refac: images

2025-12-16 11:57:51 +01:00 · 2025-11-04 13:30:59 -05:00
parent 939521b75d
commit 8e5690aab4
8 changed files with 279 additions and 476 deletions
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -45,9 +45,8 @@ from open_webui.routers.retrieval import (
    SearchForm,
 )
 from open_webui.routers.images import (
-    load_b64_image_data,
    image_generations,
-    GenerateImageForm,
+    CreateImageForm,
    upload_image,
 )
 from open_webui.routers.pipelines import (
@@ -770,7 +769,7 @@ async def chat_image_generation_handler(
    try:
        images = await image_generations(
            request=request,
-            form_data=GenerateImageForm(**{"prompt": prompt}),
+            form_data=CreateImageForm(**{"prompt": prompt}),
            user=user,
        )