refac: gpt-image-1 support

This commit is contained in:
Timothy Jaeryang Baek 2025-04-29 19:34:00 +04:00
parent addf44b21f
commit 6ae2d2c4ea

View File

@ -500,7 +500,11 @@ async def image_generations(
if form_data.size
else request.app.state.config.IMAGE_SIZE
),
"response_format": "b64_json",
**(
{"response_format": "b64_json"}
if "gpt-image-1" in request.app.state.config.IMAGE_GENERATION_MODEL
else {}
),
}
# Use asyncio.to_thread for the requests.post call