Skip to content

Commit 6fcb871

Browse files
committed
P5-PR6: mount hierarchical rate limiter middleware (uses HierarchicalConfigProvider when available)
1 parent 5ce1588 commit 6fcb871

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

api/app/main.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@
6464
from .config_manager.hierarchical_provider import HierarchicalConfigProvider, UserContext
6565
from .services.cache_manager import CacheManager
6666
from fastapi import APIRouter
67+
from .middleware.hierarchical_rate_limiter import HierarchicalRateLimiter
6768

6869

6970
app = FastAPI(
@@ -201,6 +202,13 @@ async def _stop_hf_imap_worker():
201202
except Exception:
202203
pass
203204

205+
# ===== Phase 5: Mount hierarchical rate limiter =====
206+
try:
207+
app.add_middleware(HierarchicalRateLimiter, config_provider=getattr(app.state, "hierarchical_config", None))
208+
except Exception:
209+
# Safe to continue without rate limiting if config provider is unavailable
210+
pass
211+
204212
router_cfg_v4 = APIRouter(prefix="/admin/config/v4", tags=["ConfigurationV4"], dependencies=[Depends(require_admin)])
205213

206214

0 commit comments

Comments
 (0)