diff --git a/litellm/proxy/hooks/__init__.py b/litellm/proxy/hooks/__init__.py index 467565d748..83d7e17343 100644 --- a/litellm/proxy/hooks/__init__.py +++ b/litellm/proxy/hooks/__init__.py @@ -17,13 +17,13 @@ except ImportError: # List of all available hooks that can be enabled PROXY_HOOKS = { "max_budget_limiter": _PROXY_MaxBudgetLimiter, - "parallel_request_limiter": _PROXY_MaxParallelRequestsHandler_v3, + "parallel_request_limiter": _PROXY_MaxParallelRequestsHandler, "cache_control_check": _PROXY_CacheControlCheck, } ## FEATURE FLAG HOOKS ## -if os.getenv("LEGACY_MULTI_INSTANCE_RATE_LIMITING", "false").lower() == "true": - PROXY_HOOKS["parallel_request_limiter"] = _PROXY_MaxParallelRequestsHandler +if os.getenv("EXPERIMENTAL_MULTI_INSTANCE_RATE_LIMITING", "false").lower() == "true": + PROXY_HOOKS["parallel_request_limiter"] = _PROXY_MaxParallelRequestsHandler_v3 ### update PROXY_HOOKS with ENTERPRISE_PROXY_HOOKS ###