diff --git a/litellm/router.py b/litellm/router.py index 652cd68b555..67ccec4c6ef 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -1289,6 +1289,11 @@ class Router: ) elif pre_call_check == "enforce_model_rate_limits": _callback = ModelRateLimitingCheck(dual_cache=self.cache) + elif pre_call_check == "encrypted_content_affinity": + from litellm.router_utils.pre_call_checks.encrypted_content_affinity_check import ( + EncryptedContentAffinityCheck, + ) + _callback = EncryptedContentAffinityCheck() if _callback is None: continue