[Misc][V1] Misc code streamlining (#15723)

Signed-off-by: Nick Hill <nhill@redhat.com>
2025-03-28 20:59:47 -07:00
parent 762b424a52
commit 6d531ad7b8
5 changed files with 32 additions and 38 deletions
--- a/vllm/v1/request.py
+++ b/vllm/v1/request.py
@@ -93,9 +93,11 @@ class Request:
        token_ids: Union[int, list[int]],
    ) -> None:
        if isinstance(token_ids, int):
-            token_ids = [token_ids]
-        self._output_token_ids.extend(token_ids)
-        self._all_token_ids.extend(token_ids)
+            self._output_token_ids.append(token_ids)
+            self._all_token_ids.append(token_ids)
+        else:
+            self._output_token_ids.extend(token_ids)
+            self._all_token_ids.extend(token_ids)

    @property
    def num_tokens(self) -> int: