From 9adcfedc04e3a0a96d7af66f37120f1ba2f823ae Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 1 Jan 2024 21:58:48 +0530 Subject: [PATCH] (test) fix test_get_model_cost_map.py --- litellm/tests/test_get_model_cost_map.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/litellm/tests/test_get_model_cost_map.py b/litellm/tests/test_get_model_cost_map.py index e6181bff589..c9f155e5f54 100644 --- a/litellm/tests/test_get_model_cost_map.py +++ b/litellm/tests/test_get_model_cost_map.py @@ -17,7 +17,7 @@ def test_get_gpt3_tokens(): # print(results) -test_get_gpt3_tokens() +# test_get_gpt3_tokens() def test_get_palm_tokens(): @@ -27,7 +27,7 @@ def test_get_palm_tokens(): print(max_tokens) -test_get_palm_tokens() +# test_get_palm_tokens() def test_zephyr_hf_tokens(): @@ -36,7 +36,7 @@ def test_zephyr_hf_tokens(): assert max_tokens == 32768 -test_zephyr_hf_tokens() +# test_zephyr_hf_tokens() def test_cost_ft_gpt_35(): @@ -81,7 +81,7 @@ def test_cost_ft_gpt_35(): ) -test_cost_ft_gpt_35() +# test_cost_ft_gpt_35() def test_cost_azure_gpt_35(): @@ -108,11 +108,11 @@ def test_cost_azure_gpt_35(): ) cost = litellm.completion_cost( - completion_response=resp, model="azure/gpt-3.5-turbo" + completion_response=resp, model="azure/gpt-35-turbo" ) print("\n Calculated Cost for azure/gpt-3.5-turbo", cost) - input_cost = model_cost["azure/gpt-3.5-turbo"]["input_cost_per_token"] - output_cost = model_cost["azure/gpt-3.5-turbo"]["output_cost_per_token"] + input_cost = model_cost["azure/gpt-35-turbo"]["input_cost_per_token"] + output_cost = model_cost["azure/gpt-35-turbo"]["output_cost_per_token"] expected_cost = (input_cost * resp.usage.prompt_tokens) + ( output_cost * resp.usage.completion_tokens )