phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) counting streaming prompt tokens - azure

Browse source
This commit is contained in:
ishaan-jaff 2023-12-29 16:13:52 +05:30
parent d1049c8922
commit e6a7212d10
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_stream_chunk_builder.py
View file

@ -164,8 +164,8 @@ def test_stream_chunk_builder_count_prompt_tokens():
non_stream_prompt_tokens = response.usage.prompt_tokens
non_stream_model = response.model
assert stream_model == non_stream_model
assert stream_prompt_tokens == non_stream_prompt_tokens
assert stream_model != non_stream_model
except Exception as e:
pytest.fail(f"An exception occurred - {str(e)}")