scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 182

Fix import issue

Browse source
This commit is contained in:
Sameer Kankute 2026-03-03 18:41:15 +05:30
parent ca597e18c8
commit 2f6279d189
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/router.py
View file

@ -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