From d12b6cda4e811d4112a76151169ee938fb52fb92 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Fri, 20 Sep 2024 03:16:08 +0200
Subject: [PATCH] refac: openai image async gen

---
 backend/open_webui/apps/images/main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/backend/open_webui/apps/images/main.py b/backend/open_webui/apps/images/main.py
index 1074e2cb0..768f13954 100644
--- 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,