Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
21de6d4b
Unverified
Commit
21de6d4b
authored
Dec 31, 2025
by
Andreas Karatzas
Committed by
GitHub
Dec 31, 2025
Browse files
[CI][Bugfix] Fix token counting in chunked prefill streaming test (#31565)
Signed-off-by:
Andreas Karatzas
<
akaratza@amd.com
>
parent
6c2cfb62
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
tests/entrypoints/openai/test_chunked_prompt.py
tests/entrypoints/openai/test_chunked_prompt.py
+4
-1
No files found.
tests/entrypoints/openai/test_chunked_prompt.py
View file @
21de6d4b
...
@@ -116,7 +116,10 @@ async def test_chat_completion_stream_options_and_logprobs_with_long_prompts(
...
@@ -116,7 +116,10 @@ async def test_chat_completion_stream_options_and_logprobs_with_long_prompts(
assert
chunk
.
choices
[
0
].
logprobs
is
None
assert
chunk
.
choices
[
0
].
logprobs
is
None
empty_chunks_received
+=
1
empty_chunks_received
+=
1
else
:
else
:
tokens_received
+=
1
# Count actual tokens from logprobs since multiple tokens
# can be batched into a single chunk
assert
chunk
.
choices
[
0
].
logprobs
and
chunk
.
choices
[
0
].
logprobs
.
content
tokens_received
+=
len
(
chunk
.
choices
[
0
].
logprobs
.
content
)
if
chunk
.
choices
[
0
].
finish_reason
is
not
None
:
if
chunk
.
choices
[
0
].
finish_reason
is
not
None
:
finished
=
True
finished
=
True
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment