From 0e8e630c5b183bcdca74194bb07f2016b7cad3aa Mon Sep 17 00:00:00 2001
From: Ting FU <semmer@live.cn>
Date: Thu, 9 Apr 2026 15:08:08 +0800
Subject: [PATCH] [Feat] /v1/images/generations api supports request cancel
 (#2621)

Signed-off-by: Semmer <semmer@live.cn>
---
 vllm_omni/entrypoints/openai/api_server.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/vllm_omni/entrypoints/openai/api_server.py b/vllm_omni/entrypoints/openai/api_server.py
index d445ad0ec..0706b9898 100644
--- a/vllm_omni/entrypoints/openai/api_server.py
+++ b/vllm_omni/entrypoints/openai/api_server.py
@@ -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.