mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-12 23:01:41 +00:00
Fix test
This commit is contained in:
parent
115a1e2e23
commit
f77ee939de
1 changed files with 6 additions and 0 deletions
|
|
@ -2091,6 +2091,12 @@ class TestEnsureOutputItemContentPartAdded:
|
|||
|
||||
events = iterator._pending_response_events
|
||||
assert len(events) == 2
|
||||
assert isinstance(events[0], OutputItemAddedEvent)
|
||||
assert events[0].type == ResponsesAPIStreamEvents.OUTPUT_ITEM_ADDED
|
||||
assert isinstance(events[1], ContentPartAddedEvent)
|
||||
assert events[1].type == ResponsesAPIStreamEvents.CONTENT_PART_ADDED
|
||||
assert events[1].part.type == "output_text"
|
||||
assert iterator.sent_content_part_added_event is True
|
||||
|
||||
def test_emit_response_completed_uses_stream_finish_reason(self):
|
||||
"""
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue