Create OmniOpenAIServeImage Class To manage edit_images and generate_image

Signed-off-by: bash000000 <m2588953@outlook.com>

Author: zhcn000000

vllm_omni/entrypoints/openai/protocol/images.py

@@ -144,8 +144,8 @@ class ImageEditResponse(BaseModel):
 
 
 class ImageEditRequest(BaseModel):
-    image: list[UploadFile] = Field(..., description="Image file to edit")
-    url: str | None = Field(default=None, description="URL of the image to edit")
+    image: list[UploadFile] | None = Field(default=None, description="Image file to edit")
+    url: list[str] | None = Field(default=None, description="URL of the image to edit")
     prompt: str = Field(..., description="Text description of the desired image edit")
     model: str | None = Field(
         default=None,

vllm_omni/entrypoints/openai/serving_image.py

@@ -60,7 +60,7 @@ def __init__(
             log_error_stack=log_error_stack,
         )
         self._engine_client = engine_client
-        self._model_name = models.model_name
+        self._model_name = models.model_name()
         self._stage_configs = stage_configs
 
         self._stage_types: list[str] = []
@@ -218,7 +218,7 @@ def _encode_image_base64_with_compression(
         """
         buffer = BytesIO()
         save_kwargs = {}
-        if format in {"jpg", "jpeg", "webp"}:
+        if format in ("jpg", "jpeg", "webp"):
             save_kwargs["quality"] = output_compression
         elif format == "png":
             save_kwargs["compress_level"] = max(0, min(9, 9 - output_compression // 11))  # Map 0-100 to 9-0
@@ -281,7 +281,7 @@ async def _load_input_images(
     def _choose_output_format(output_format: str | None, background: str | None) -> str:
         # Normalize and choose extension
         fmt = (output_format or "").lower()
-        if fmt in {"jpg", "png", "webp", "jpeg"}:
+        if fmt in ("jpg", "png", "webp", "jpeg"):
             return fmt
         # If transparency requested, prefer png
         if (background or "auto").lower() == "transparent":