Update src/rotator_library/client.py

src/rotator_library/client.py

@@ -650,21 +650,6 @@ class RotatingClient:
                 return True
         return False
 
-    def _verify_model_allowed(self, provider: str, model_id: str) -> None:
-        """
-        Verifies if a model is allowed according to whitelist and ignore rules.
-        Raises ValueError if the model is restricted.
-        """
-        is_whitelisted = self._is_model_whitelisted(provider, model_id)
-        is_blacklisted = self._is_model_ignored(provider, model_id)
-
-        # A model is allowed if it's whitelisted OR not blacklisted.
-        # Conversely, it's restricted ONLY if it's NOT whitelisted AND it IS blacklisted.
-        if not is_whitelisted and is_blacklisted:
-            lib_logger.warning(
-                f"Blocking request for restricted model: {model_id} (provider: {provider})"
-            )
-            raise ValueError(f"Model '{model_id}' is restricted and cannot be used.")
 
     def _sanitize_litellm_log(self, log_data: dict) -> dict:
         """
@@ -1255,9 +1240,6 @@ class RotatingClient:
             model = resolved_model
             kwargs["model"] = model  # Ensure kwargs has the resolved model for litellm
 
-        # [STRICT ENFORCEMENT] Verify model is allowed before proceeding
-        self._verify_model_allowed(provider, model)
-
         # [NEW] Filter by model tier requirement and build priority map
         credential_priorities = None
         if provider_plugin and hasattr(provider_plugin, "get_model_tier_requirement"):
@@ -2020,9 +2002,6 @@ class RotatingClient:
             model = resolved_model
             kwargs["model"] = model  # Ensure kwargs has the resolved model for litellm
 
-        # [STRICT ENFORCEMENT] Verify model is allowed before proceeding
-        self._verify_model_allowed(provider, model)
-
         # [NEW] Filter by model tier requirement and build priority map
         credential_priorities = None
         if provider_plugin and hasattr(provider_plugin, "get_model_tier_requirement"):
@@ -3600,4 +3579,4 @@ class RotatingClient:
 
         total_tokens = message_tokens + tool_tokens
 
-        return {"input_tokens": total_tokens}
+        return {"input_tokens": total_tokens}