From 841e941ecd232e73a476e38f0d570d40f1fb9283 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 14 Dec 2023 13:08:35 +0530 Subject: [PATCH] (test) router - model_group_alias --- litellm/tests/test_router_get_deployments.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_router_get_deployments.py b/litellm/tests/test_router_get_deployments.py index f9ca124c267..1e642babae4 100644 --- a/litellm/tests/test_router_get_deployments.py +++ b/litellm/tests/test_router_get_deployments.py @@ -291,7 +291,6 @@ def test_weighted_selection_router_no_rpm_set(): def test_model_group_aliases(): try: litellm.set_verbose = False - litellm.model_group_alias_map = {"gpt-4": "gpt-3.5-turbo"} model_list = [ { "model_name": "gpt-3.5-turbo", @@ -321,7 +320,10 @@ def test_model_group_aliases(): ] router = Router( model_list=model_list, + model_group_alias={"gpt-4": "gpt-3.5-turbo"} # gpt-4 requests sent to gpt-3.5-turbo ) + + # test that gpt-4 requests are sent to gpt-3.5-turbo for _ in range(20): selected_model = router.get_available_deployment("gpt-4") print("\n selected model", selected_model) @@ -330,8 +332,7 @@ def test_model_group_aliases(): pytest.fail(f"Selected model {selected_model_name} is not gpt-3.5-turbo") router.reset() - litellm.model_group_alias_map = {} except Exception as e: traceback.print_exc() pytest.fail(f"Error occurred: {e}") -# test_model_group_aliases() \ No newline at end of file +test_model_group_aliases() \ No newline at end of file