refac: openai image async gen

2024-11-16 05:24:02 +00:00 · 2024-09-20 03:16:08 +02:00 · 2024-09-20 03:16:08 +02:00 · d12b6cda4e
commit d12b6cda4e
parent bd09b6dbad
1 changed files with 3 additions and 1 deletions
--- a/backend/open_webui/apps/images/main.py
+++ b/backend/open_webui/apps/images/main.py
@ -470,7 +470,9 @@ async def image_generations(
                "response_format": "b64_json",
            }

-            r = requests.post(
+            # Use asyncio.to_thread for the requests.post call
+            r = await asyncio.to_thread(
+                requests.post,
                url=f"{app.state.config.OPENAI_API_BASE_URL}/images/generations",
                json=data,
                headers=headers,