added routers on main app router

app/main.py

@@ -23,7 +23,7 @@ from app.redis_client import redis
 from app.deps import get_current_user, rate_limit_org, verify_api_key, check_all_services
 from app.tasks.analytics_worker import redis_listener
 from app.services.vector_service import cleanup_expired_vectors
-from app.routers import health, datasources, reports, flags, scheduler, run, socket, analytics_stream
+from app.routers import health, datasources, reports, flags, scheduler, run, socket, analytics_stream,ai_query,schema
 
 # ─── Logger Configuration ───────────────────────────────────────────────────────
 logging.basicConfig(
@@ -194,11 +194,12 @@ async def add_request_tracking(request: Request, call_next):
     
     return response
 
+# ─── KPI Computation Endpoint ──────────────────────────────────────────────────
 # ─── KPI Computation Endpoint ──────────────────────────────────────────────────
 @app.post("/api/v1/kpi/compute")
 async def compute_kpis(
+    background_tasks: BackgroundTasks,  # ✅ First parameter, no default needed
     source_id: str = Query(..., description="Data source ID"),
-    background_tasks: BackgroundTasks,
     current_user: dict = Depends(get_current_user),
     limited_org: str = Depends(rate_limit_org(max_requests=50))
 ):
@@ -217,13 +218,12 @@ async def compute_kpis(
                 "org_id": org_id,
                 "data": json.loads(cached),
                 "rate_limit": {
-                    "remaining": 50,  # Simplified - get from actual rate limiter
+                    "remaining": 50,
                     "reset_in": 60
                 }
             }
         
         # Trigger background computation
-        # NOTE: Make sure trigger_kpi_computation is defined in deps or imported
         background_tasks.add_task(trigger_kpi_computation, org_id, source_id)
         
         return {
@@ -339,4 +339,6 @@ app.include_router(flags.router, prefix="/api/v1/flags", dependencies=[Depends(v
 app.include_router(scheduler.router, prefix="/api/v1/scheduler", dependencies=[Depends(verify_api_key)])
 app.include_router(run.router, prefix="/api/v1/run", dependencies=[Depends(verify_api_key)])
 app.include_router(socket.router, prefix="/api/v1/socket", dependencies=[Depends(verify_api_key)])
-app.include_router(analytics_stream.router, prefix="/api/v1/analytics", tags=["analytics"])
+app.include_router(analytics_stream.router, prefix="/api/v1/analytics", tags=["analytics"])
+app.include_router(ai_query.router, prefix="/api/v1/ai-query", dependencies=[Depends(verify_api_key)])
+app.include_router(schema.router, prefix="/api/v1/schema", dependencies=[Depends(verify_api_key)])