test(test_stream_chunk_builder.py): remove completion assert, the test is for prompt tokens

This commit is contained in:
Krrish Dholakia 2024-01-06 14:12:44 +05:30
parent 5c45e69a5e
commit 04c04d62e3

View file

@ -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)}")