From 711a1428f8a5893ddaa06e5ba12a29daa3b77b36 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 29 Nov 2024 21:03:31 -0800 Subject: [PATCH] fix: fix tests --- litellm/proxy/auth/auth_checks.py | 2 -- tests/llm_translation/test_max_completion_tokens.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/proxy/auth/auth_checks.py b/litellm/proxy/auth/auth_checks.py index 315d9dd36..21a25c8c1 100644 --- a/litellm/proxy/auth/auth_checks.py +++ b/litellm/proxy/auth/auth_checks.py @@ -904,7 +904,6 @@ async def can_key_call_model( if llm_router: access_groups = llm_router.get_model_access_groups(model_name=model) - models_in_current_access_groups = [] if ( len(access_groups) > 0 and llm_router is not None ): # check if token contains any model access groups @@ -917,7 +916,6 @@ async def can_key_call_model( # Filter out models that are access_groups filtered_models = [m for m in valid_token.models if m not in access_groups] - filtered_models += models_in_current_access_groups verbose_proxy_logger.debug(f"model: {model}; allowed_models: {filtered_models}") all_model_access: bool = False diff --git a/tests/llm_translation/test_max_completion_tokens.py b/tests/llm_translation/test_max_completion_tokens.py index de335a3c5..7b10bcd3d 100644 --- a/tests/llm_translation/test_max_completion_tokens.py +++ b/tests/llm_translation/test_max_completion_tokens.py @@ -81,7 +81,7 @@ async def test_bedrock_max_completion_tokens(model: str, respx_mock: MockRouter) @pytest.mark.parametrize( "model", - ["anthropic/claude-3-sonnet-20240229", "anthropic/claude-3-opus-20240229,"], + ["anthropic/claude-3-sonnet-20240229", "anthropic/claude-3-opus-20240229"], ) @pytest.mark.respx @pytest.mark.asyncio()