scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

(test) test_token_counter_azure

Browse source
This commit is contained in:
ishaan-jaff 2023-12-29 15:37:39 +05:30
parent 037dcbbe10
commit b1077ebc38
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_utils.py
View file

@ -323,7 +323,7 @@ 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)
tokens_counted = token_counter(model="gpt-35-turbo", messages=messages)
print("Tokens Counted=", tokens_counted)
response = litellm.completion(model="azure/chatgpt-v-2", messages=messages)