scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 359

test(proxy): type the first-chunk failure test's parameters

Browse source
This commit is contained in:
mateo-berri 2026-10-29 01:20:31 -07:00
parent 86360077c6
commit 5003ef1f31
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/unit/proxy/test_common_request_processing.py
View file

@ -47,6 +47,7 @@ from litellm.proxy.common_request_processing import (
_ClientDisconnectedBeforeFirstChunk,
attach_guardrail_information,
_extract_error_from_sse_chunk,
FirstChunkSseFraming,
_get_cost_breakdown_from_logging_obj,
CostBreakdownHeaderValues,
_has_attribute_error_in_chain,
@ -2373,7 +2374,9 @@ class TestCommonRequestProcessingHelpers:
(RESPONSES_FIRST_CHUNK_FRAMING, ('event: error\ndata: {"type": "error", "code": "server_error"',)),
],
)
async def test_a_stream_that_fails_before_its_first_chunk_ends_in_its_own_api_shape(self, framing, frame_starts):
async def test_a_stream_that_fails_before_its_first_chunk_ends_in_its_own_api_shape(
self, framing: FirstChunkSseFraming, frame_starts: tuple[str, ...]
) -> None:
response: Final = await create_response(
_stream_failing_before_its_first_chunk(), "text/event-stream", {}, framing=framing
)