Extract Redis health‑check interface to its own module, update strategy to import it and drop direct monitor init, clean unused imports, and restore Gemma‑3 providers in models config

Author: Paweł Kędzia

llm_router_api/base/lb/first_available.py

@@ -24,7 +24,6 @@
 
 import time
 import logging
-import random
 
 try:
     import redis
@@ -36,7 +35,6 @@
 from typing import List, Dict, Optional, Any
 
 from llm_router_api.base.constants import REDIS_PORT, REDIS_HOST
-from llm_router_api.base.lb.provider_monitor import RedisProviderMonitor
 from llm_router_api.base.lb.redis_based_interface import RedisBasedStrategyInterface
 
 

llm_router_api/base/lb/redis_based_interface.py

@@ -1,6 +1,8 @@
 import random
 import logging
+
 from abc import ABC
+from typing import List, Dict, Optional, Any, Tuple
 
 try:
     import redis
@@ -9,37 +11,11 @@
 except ImportError:
     REDIS_IS_AVAILABLE = False
 
-from typing import List, Dict, Optional, Any, Tuple
-
 from llm_router_api.base.constants import REDIS_PORT, REDIS_HOST
 from llm_router_api.base.lb.strategy import ChooseProviderStrategyI
-from llm_router_api.base.lb.provider_monitor import RedisProviderMonitor
-
-
-class RedisBasedHealthCheckInterface(ABC):
-    def __init__(
-        self,
-        redis_host: str = REDIS_HOST,
-        redis_port: int = REDIS_PORT,
-        redis_db: int = 0,
-        check_interval: float = 30,
-        clear_buffers: bool = True,
-        logger: Optional[logging.Logger] = None,
-    ):
-        if not REDIS_IS_AVAILABLE:
-            raise RuntimeError("Redis is not available. Please install it first.")
-
-        self.redis_client = redis.Redis(
-            host=redis_host, port=redis_port, db=redis_db, decode_responses=True
-        )
-
-        # Start providers monitor
-        self._monitor = RedisProviderMonitor(
-            redis_client=self.redis_client,
-            check_interval=check_interval,
-            clear_buffers=clear_buffers,
-            logger=logger,
-        )
+from llm_router_api.base.monitor.redis_health_interface import (
+    RedisBasedHealthCheckInterface,
+)
 
 
 class RedisBasedStrategyInterface(
@@ -145,14 +121,6 @@ def __init__(
         if clear_buffers:
             self._clear_buffers()
 
-        # # Start providers monitor
-        # self._monitor = RedisProviderMonitor(
-        #     redis_client=self.redis_client,
-        #     check_interval=30,
-        #     clear_buffers=clear_buffers,
-        #     logger=self.logger,
-        # )
-
     def init_provider(
         self,
         model_name: str,

llm_router_api/base/monitor/__init__.py

@@ -0,0 +1,40 @@
+import logging
+
+from abc import ABC
+from typing import Optional
+
+try:
+    import redis
+
+    REDIS_IS_AVAILABLE = True
+except ImportError:
+    REDIS_IS_AVAILABLE = False
+
+from llm_router_api.base.constants import REDIS_PORT, REDIS_HOST
+from llm_router_api.base.monitor.provider_monitor import RedisProviderMonitor
+
+
+class RedisBasedHealthCheckInterface(ABC):
+    def __init__(
+        self,
+        redis_host: str = REDIS_HOST,
+        redis_port: int = REDIS_PORT,
+        redis_db: int = 0,
+        check_interval: float = 30,
+        clear_buffers: bool = True,
+        logger: Optional[logging.Logger] = None,
+    ):
+        if not REDIS_IS_AVAILABLE:
+            raise RuntimeError("Redis is not available. Please install it first.")
+
+        self.redis_client = redis.Redis(
+            host=redis_host, port=redis_port, db=redis_db, decode_responses=True
+        )
+
+        # Start providers monitor
+        self._monitor = RedisProviderMonitor(
+            redis_client=self.redis_client,
+            check_interval=check_interval,
+            clear_buffers=clear_buffers,
+            logger=logger,
+        )