From 30a660ef0c8fa2fecf99840bdd908ee31b26fe57 Mon Sep 17 00:00:00 2001
From: viiccwen <viiccwen@gmail.com>
Date: Fri, 5 Jun 2026 05:39:00 +0000
Subject: [PATCH] Fix wait_for_completion output type docstring

Co-authored-by: OpenAI Codex <codex@openai.com>

Signed-off-by: viiccwen <viiccwen@gmail.com>
---
 vllm/entrypoints/llm.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/vllm/entrypoints/llm.py b/vllm/entrypoints/llm.py
index 7297243f9188..892e5035ab60 100644
--- a/vllm/entrypoints/llm.py
+++ b/vllm/entrypoints/llm.py
@@ -556,7 +556,8 @@ def wait_for_completion(
         and returns their outputs. Use after enqueue() to get results.
 
         Args:
-            output_type: The expected output type, defaults to RequestOutput.
+            output_type: The expected output type(s). If not provided, accepts
+                both RequestOutput and PoolingRequestOutput.
             use_tqdm: If True, shows a tqdm progress bar.
 
         Returns: