- Fix 71 invalid-syntax files (class-body newline-broken assignments) - Add from/None chain to 307 B904 raise-without-from sites - Add B008 ignore to ruff.toml (already in pyproject.toml) - Noqa F401 on __init__.py re-exports (137 sites) - Noqa E402 on deferred imports (63 sites) - Bulk-add stdlib/FastAPI/project imports for F821 (127 sites) - Replace ×→x, –→-, …→... in docstrings (4093 chars) - Manual refactor of 5 SIM103/SIM116 patterns Tests: 791 passed (66 deselected due to pre-existing Redis issues in test_rag.py) Co-authored-by: opencode <opencode@rugmunch.io>
58 lines
2 KiB
Python
58 lines
2 KiB
Python
"""#2 SSE Streaming + #3 Provider Dashboard endpoints."""
|
|
|
|
import os
|
|
|
|
import httpx
|
|
from fastapi import APIRouter
|
|
|
|
router = APIRouter(prefix="/api/v1/databus", tags=["databus-extras"])
|
|
|
|
BACKEND = os.getenv("BACKEND_URL", "http://localhost:8000")
|
|
|
|
|
|
@router.get("/providers/dashboard")
|
|
async def provider_dashboard():
|
|
"""Real-time provider health dashboard data - feed into Grafana."""
|
|
try:
|
|
async with httpx.AsyncClient(timeout=10) as c:
|
|
r = await c.get(
|
|
f"{BACKEND}/api/v1/databus/providers/health",
|
|
headers={"X-RMI-Key": os.getenv("RMI_INTERNAL_KEY", "rmi-internal-2026")},
|
|
)
|
|
if r.status_code == 200:
|
|
data = r.json()
|
|
providers = data.get("providers", data)
|
|
|
|
# Format for Grafana
|
|
panels = []
|
|
for name, health in providers.items() if isinstance(providers, dict) else []:
|
|
panels.append(
|
|
{
|
|
"provider": name,
|
|
"status": "healthy" if health.get("healthy", True) else "degraded",
|
|
"latency_ms": health.get("avg_latency_ms", 0),
|
|
"error_rate": health.get("error_rate", 0),
|
|
"circuit": health.get("circuit_state", "closed"),
|
|
}
|
|
)
|
|
|
|
return {
|
|
"providers": panels,
|
|
"summary": {
|
|
"total": len(panels),
|
|
"healthy": sum(1 for p in panels if p["status"] == "healthy"),
|
|
"degraded": sum(1 for p in panels if p["status"] != "healthy"),
|
|
},
|
|
}
|
|
except Exception:
|
|
pass
|
|
|
|
return {"providers": [], "note": "Provider health API unavailable - check backend"}
|
|
|
|
|
|
@router.get("/queue/stats")
|
|
async def task_queue_stats():
|
|
"""Background task queue statistics."""
|
|
from app.core.task_queue import get_queue_stats
|
|
|
|
return await get_queue_stats()
|