phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) token_counter - prompt tokens == tokens from API

Browse source
This commit is contained in:
ishaan-jaff 2023-12-29 15:15:28 +05:30
parent a300ab9152
commit 4a028d012a
1 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/tests/test_utils.py
View file

@ -316,4 +316,25 @@ def test_token_counter():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_token_counter() # test_token_counter()
def test_token_counter_azure():
# test that prompt_tokens counted == prompt_tokens from Azure API
try:
messages = [{"role": "user", "content": "hi how are you what time is it"}]
tokens_counted = token_counter(model="gpt-3.5-turbo", messages=messages)
print("Tokens Counted=", tokens_counted)
response = litellm.completion(model="azure/chatgpt-v-2", messages=messages)
prompt_tokens = response.usage.prompt_tokens
print("Tokens from Azure API=", prompt_tokens)
assert prompt_tokens == tokens_counted
except:
pytest.fail(f"Error occurred: {e}")
# test_token_counter_azure()