diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 993d81b8fb8..b7842ec27e0 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -11483,11 +11483,11 @@ async def non_admin_all_models( # correctly downstream. user_models = list(user_row.models or []) - # Get all models that are team models, when model team_id == user_row.teams - all_models += _check_if_model_is_team_model( - models=llm_router.get_model_list() or [], - user_row=user_row, - ) + # Get all models that are team models, when model team_id == user_row.teams + all_models += _check_if_model_is_team_model( + models=llm_router.get_model_list() or [], + user_row=user_row, + ) # de-duplicate models. Only return unique model ids unique_models = _deduplicate_litellm_router_models(models=all_models) diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index ae62bf5a67c..e1f0347a7cb 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -6459,7 +6459,7 @@ async def _apply_user_models_filter( model_access_groups: dict[str, list[str]], prisma_client: Optional["PrismaClient"], proxy_logging_obj: Optional["ProxyLogging"], - user_api_key_cache: Optional["DualCache"], + user_api_key_cache: Optional["UserApiKeyCache"], user_models_override: list[str] | None = None, ) -> list[str]: """ @@ -6554,7 +6554,7 @@ async def apply_user_models_filter_to_deployments( llm_router: Optional["Router"], prisma_client: Optional["PrismaClient"], proxy_logging_obj: Optional["ProxyLogging"], - user_api_key_cache: Optional["DualCache"], + user_api_key_cache: Optional["UserApiKeyCache"], user_models_override: list[str] | None = None, ) -> list[dict[str, Any]]: """