From 2f6279d1894ce630afcbf80c1e96231ec06d1e82 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Tue, 3 Mar 2026 18:41:15 +0530 Subject: [PATCH] Fix import issue --- litellm/router.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index 67ccec4c6ef..8eb2c417511 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -1255,6 +1255,10 @@ class Router: # Encrypted content affinity # --------------------------------------------------------------------- if "encrypted_content_affinity" in optional_pre_call_checks: + from litellm.router_utils.pre_call_checks.encrypted_content_affinity_check import ( + EncryptedContentAffinityCheck, + ) + if self.optional_callbacks is None: self.optional_callbacks = [] @@ -1289,11 +1293,6 @@ 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