Skip to content

Commit f2bf243

Browse files
committed
unit test
Signed-off-by: Andrew Xia <[email protected]>
1 parent e366b57 commit f2bf243

File tree

2 files changed

+24
-1
lines changed

2 files changed

+24
-1
lines changed

tests/entrypoints/openai/test_response_api_with_harmony.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,9 @@ async def test_streaming(client: OpenAI, model_name: str, background: bool):
304304
background=background,
305305
)
306306

307+
current_item_id = ""
308+
current_content_index = -1
309+
307310
events = []
308311
current_event_mode = None
309312
resp_id = None
@@ -315,6 +318,26 @@ async def test_streaming(client: OpenAI, model_name: str, background: bool):
315318
current_event_mode = event.type
316319
print(f"\n[{event.type}] ", end="", flush=True)
317320

321+
# verify current_item_id is correct
322+
if event.type == "response.output_item.added":
323+
assert event.item.id != current_item_id
324+
current_item_id = event.item.id
325+
elif event.type in [
326+
"response.output_text.delta",
327+
"response.reasoning_text.delta"
328+
]:
329+
assert event.item_id == current_item_id
330+
331+
# verify content_index_id is correct
332+
if event.type == "response.content_part.added":
333+
assert event.content_index != current_content_index
334+
current_content_index = event.content_index
335+
elif event.type in [
336+
"response.output_text.delta",
337+
"response.reasoning_text.delta"
338+
]:
339+
assert event.content_index == current_content_index
340+
318341
if "text.delta" in event.type:
319342
print(event.delta, end="", flush=True)
320343
elif "reasoning_text.delta" in event.type:

vllm/entrypoints/openai/serving_responses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1504,7 +1504,7 @@ async def _process_harmony_streaming_events(
15041504
raise ValueError(
15051505
f"Unknown function name: {function_name}")
15061506

1507-
current_item_id = f"tool_{random_uuid()}",
1507+
current_item_id = f"tool_{random_uuid()}"
15081508
yield _send_event(
15091509
openai_responses_types.ResponseOutputItemAddedEvent(
15101510
type="response.output_item.added",

0 commit comments

Comments
 (0)