[Feat] /v1/images/generations api supports request cancel (#2621)

Signed-off-by: Semmer <semmer@live.cn>
This commit is contained in:
Ting FU authored and GitHub committed 2026-04-09 07:08:08 +00:00
1 parent 92c788e5fd
commit 0e8e630c5b
1 file changed
+2 -1
+2 -1
View File
@@ -1282,7 +1282,8 @@ async def show_available_models(raw_request: Request) -> JSONResponse:
HTTPStatus.INTERNAL_SERVER_ERROR.value: {"model": ErrorResponse},
},
)
async def generate_images(request: ImageGenerationRequest, raw_request: Request) -> ImageGenerationResponse:
@with_cancellation
async def generate_images(request: ImageGenerationRequest, raw_request: Request):
"""Generate images from text prompts using diffusion models.
OpenAI DALL-E compatible endpoint for text-to-image generation.