diff --git a/vllm_omni/entrypoints/openai/serving_chat.py b/vllm_omni/entrypoints/openai/serving_chat.py index 3296c0b4d..66a86e7c9 100644 --- a/vllm_omni/entrypoints/openai/serving_chat.py +++ b/vllm_omni/entrypoints/openai/serving_chat.py @@ -2116,7 +2116,7 @@ class OmniOpenAIServingChat(OpenAIServingChat, AudioMixin): index=output.index, delta=DeltaMessage(role=role, content=audio_base64), logprobs=None, - finish_reason="stop", + finish_reason=output.finish_reason, stop_reason=output.stop_reason, token_ids=(as_list(output.token_ids) if request.return_token_ids else None), ) @@ -2126,7 +2126,7 @@ class OmniOpenAIServingChat(OpenAIServingChat, AudioMixin): message=ChatMessage(role=role, audio=audio_obj), logprobs=None, finish_reason="stop", - stop_reason=None, + stop_reason=output.stop_reason, ) choices.append(choice_data) return choices