tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-05 20:23:18 +00:00
Code Issues Packages Projects Releases Wiki 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
litellm/tests/test_get_model_cost_map.py
+11 -7
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()