diff --git a/tests/base_sdk_tests/check_sdk_http.py b/tests/base_sdk_tests/check_sdk_http.py index f7a1e30ac99..d5b2b33a219 100644 --- a/tests/base_sdk_tests/check_sdk_http.py +++ b/tests/base_sdk_tests/check_sdk_http.py @@ -69,7 +69,11 @@ def enqueue(body: object, status: int = 200) -> None: def enqueue_stream() -> None: - chunk: Final = {**CHAT, "object": "chat.completion.chunk", "choices": [{"index": 0, "delta": {"content": "pong"}}]} + chunk: Final = { + **CHAT, + "object": "chat.completion.chunk", + "choices": [{"index": 0, "delta": {"content": "pong"}, "finish_reason": "stop"}], + } RESPONSES.put((200, f"data: {json.dumps(chunk)}\n\ndata: [DONE]\n\n".encode()))