From 04c04d62e3fe59613755feec4513a09bfb5cb332 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 6 Jan 2024 14:12:44 +0530 Subject: [PATCH] test(test_stream_chunk_builder.py): remove completion assert, the test is for prompt tokens --- litellm/tests/test_stream_chunk_builder.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/litellm/tests/test_stream_chunk_builder.py b/litellm/tests/test_stream_chunk_builder.py index 37ea1c3d3b..b8bc7d45cf 100644 --- a/litellm/tests/test_stream_chunk_builder.py +++ b/litellm/tests/test_stream_chunk_builder.py @@ -165,8 +165,6 @@ def test_stream_chunk_builder_count_prompt_tokens(): assert stream_model == non_stream_model assert stream_prompt_tokens == non_stream_prompt_tokens - assert stream_completion_tokens == non_stream_completion_tokens - except Exception as e: pytest.fail(f"An exception occurred - {str(e)}")