From 27db32962176f28c5242110bf2510e505921cc4c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 5 Jun 2024 21:42:41 -0700 Subject: [PATCH 1/2] fix - test lowest cost routing --- litellm/tests/test_lowest_cost_routing.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_lowest_cost_routing.py b/litellm/tests/test_lowest_cost_routing.py index 217b4a970..91b8abb68 100644 --- a/litellm/tests/test_lowest_cost_routing.py +++ b/litellm/tests/test_lowest_cost_routing.py @@ -106,14 +106,14 @@ async def test_lowest_cost_routing(): """ model_list = [ { - "model_name": "gpt-3.5-turbo", + "model_name": "gpt-4", "litellm_params": {"model": "gpt-4"}, "model_info": {"id": "openai-gpt-4"}, }, { "model_name": "gpt-3.5-turbo", - "litellm_params": {"model": "groq/llama3-8b-8192"}, - "model_info": {"id": "groq-llama"}, + "litellm_params": {"model": "gpt-3.5-turbo"}, + "model_info": {"id": "gpt-3.5-turbo"}, }, ] @@ -127,7 +127,7 @@ async def test_lowest_cost_routing(): print( response._hidden_params["model_id"] ) # expect groq-llama, since groq/llama has lowest cost - assert "groq-llama" == response._hidden_params["model_id"] + assert "gpt-3.5-turbo" == response._hidden_params["model_id"] async def _deploy(lowest_cost_logger, deployment_id, tokens_used, duration): From 2e95758ada27881ab661e9e23a67a184c58e00dd Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 5 Jun 2024 21:58:59 -0700 Subject: [PATCH 2/2] ci/cd run again --- litellm/tests/test_lowest_cost_routing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_lowest_cost_routing.py b/litellm/tests/test_lowest_cost_routing.py index 91b8abb68..a793ba0a2 100644 --- a/litellm/tests/test_lowest_cost_routing.py +++ b/litellm/tests/test_lowest_cost_routing.py @@ -102,7 +102,7 @@ async def test_get_available_deployments_custom_price(): @pytest.mark.asyncio async def test_lowest_cost_routing(): """ - Test if router returns model with the lowest cost + Test if router, returns model with the lowest cost """ model_list = [ {