From 99cc76690d22fe9c0b78d5b8bf9a5f9be62f8770 Mon Sep 17 00:00:00 2001 From: "Hui." <62495465+Hu1Lcode@users.noreply.github.com> Date: Thu, 19 Mar 2026 18:57:04 +0800 Subject: [PATCH] [Bugfix]Fix bug of online server can not return mutli images (#2007) Signed-off-by: Hui <1779066624@qq.com> Co-authored-by: AlvisGong --- vllm_omni/entrypoints/openai/serving_chat.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/vllm_omni/entrypoints/openai/serving_chat.py b/vllm_omni/entrypoints/openai/serving_chat.py index 4d93dcf0a..a60756f2c 100644 --- a/vllm_omni/entrypoints/openai/serving_chat.py +++ b/vllm_omni/entrypoints/openai/serving_chat.py @@ -2176,11 +2176,17 @@ class OmniOpenAIServingChat(OpenAIServingChat, AudioMixin): # Convert images to base64 content image_contents: list[dict[str, Any]] = [] - for img in images: + flat_images = [] + for item in images: + if isinstance(item, list): + flat_images.extend(item) + else: + flat_images.append(item) + + for img in flat_images: with BytesIO() as buffer: img.save(buffer, format="PNG") - img_bytes = buffer.getvalue() - img_base64 = base64.b64encode(img_bytes).decode("utf-8") + img_base64 = base64.b64encode(buffer.getvalue()).decode("utf-8") image_contents.append( { "type": "image_url",