Switch image generation to DALL-E 3

Use the dall-e-3 model in generate_images and default to 1024x1024.
dall-e-3 only supports n=1, so request a single image per call.
This commit is contained in:
Father Bot
2026-04-08 11:08:52 +03:00
parent 04174251b5
commit 09ebbbf996
+5 -2
View File
@@ -341,8 +341,11 @@ async def transcribe_audio(audio_file) -> str:
return r.text or ""
async def generate_images(prompt, n_images=4, size="512x512"):
r = await openai_client.images.generate(prompt=prompt, n=n_images, size=size)
async def generate_images(prompt, n_images=1, size="1024x1024"):
# dall-e-3 only supports generating one image per request
r = await openai_client.images.generate(
model="dall-e-3", prompt=prompt, n=1, size=size
)
image_urls = [item.url for item in r.data]
return image_urls