diff --git a/python/tests/integration/completions/test_text_completion.py b/python/tests/integration/completions/test_text_completion.py index 1cae0965f984..3e9b34ef76aa 100644 --- a/python/tests/integration/completions/test_text_completion.py +++ b/python/tests/integration/completions/test_text_completion.py @@ -296,7 +296,7 @@ async def get_text_completion_response( ) parts: list[StreamingTextContent] = [part async for part in response if part is not None] if parts: - return sum(parts[:1], parts[0]) + return sum(parts[1:], parts[0]) raise AssertionError("No response") return await service.get_text_content( prompt=prompt,