From 38ff9f2b6f3167f3b5be73a5fb7eb63c3be36bfd Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 4 Nov 2023 09:12:22 -0700 Subject: [PATCH] (fix) test get_model_cost_map --- litellm/tests/test_get_model_cost_map.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/litellm/tests/test_get_model_cost_map.py b/litellm/tests/test_get_model_cost_map.py index 88213dd8aa..8a86cc51fe 100644 --- a/litellm/tests/test_get_model_cost_map.py +++ b/litellm/tests/test_get_model_cost_map.py @@ -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()