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

(fix) test get_model_cost_map

Browse source
This commit is contained in:
ishaan-jaff 2023-11-04 09:12:22 -07:00
parent 5f26da58a1
commit 38ff9f2b6f
1 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_get_model_cost_map.py
View file

@ -9,11 +9,15 @@ from litellm import get_max_tokens, model_cost, open_ai_chat_completion_models
print(get_max_tokens("gpt-3.5-turbo"))
print(model_cost)
print(open_ai_chat_completion_models)
# print("\n")
# print(model_cost)
# print("\n")
def test_get_gpt3_tokens():
max_tokens = get_max_tokens("gpt-3.5-turbo")
results = max_tokens['max_tokens']
print(results)
# test_get_gpt3_tokens()
# # 🦄🦄🦄🦄🦄🦄🦄🦄
# print(get_max_tokens("palm/chat-bison"))
def test_get_palm_tokens():
# # 🦄🦄🦄🦄🦄🦄🦄🦄
max_tokens = get_max_tokens("palm/chat-bison")
results = max_tokens['max_tokens']
print(results)
# test_get_palm_tokens()