semantic update

backend/src/app.py

@@ -21,7 +21,7 @@ import asyncio
 from .dependencies import get_task_manager, get_scheduler_service
 from .core.utils.network import NetworkError
 from .core.logger import logger, belief_scope
-from .api.routes import plugins, tasks, settings, environments, mappings, migration, connections, git, storage, admin, llm, dashboards, datasets, reports, assistant
+from .api.routes import plugins, tasks, settings, environments, mappings, migration, connections, git, storage, admin, llm, dashboards, datasets, reports, assistant
 from .api import auth
 
 # [DEF:App:Global]
@@ -72,12 +72,12 @@ app.add_middleware(
 )
 
 
-# [DEF:log_requests:Function]
-# @PURPOSE: Middleware to log incoming HTTP requests and their response status.
+# [DEF:network_error_handler:Function]
+# @PURPOSE: Global exception handler for NetworkError.
 # @PRE: request is a FastAPI Request object.
-# @POST: Logs request and response details.
+# @POST: Returns 503 HTTP Exception.
 # @PARAM: request (Request) - The incoming request object.
-# @PARAM: call_next (Callable) - The next middleware or route handler.
+# @PARAM: exc (NetworkError) - The exception instance.
 @app.exception_handler(NetworkError)
 async def network_error_handler(request: Request, exc: NetworkError):
     with belief_scope("network_error_handler"):
@@ -86,26 +86,34 @@ async def network_error_handler(request: Request, exc: NetworkError):
             status_code=503,
             detail="Environment unavailable. Please check if the Superset instance is running."
         )
+# [/DEF:network_error_handler:Function]
 
+# [DEF:log_requests:Function]
+# @PURPOSE: Middleware to log incoming HTTP requests and their response status.
+# @PRE: request is a FastAPI Request object.
+# @POST: Logs request and response details.
+# @PARAM: request (Request) - The incoming request object.
+# @PARAM: call_next (Callable) - The next middleware or route handler.
 @app.middleware("http")
 async def log_requests(request: Request, call_next):
-    # Avoid spamming logs for polling endpoints
-    is_polling = request.url.path.endswith("/api/tasks") and request.method == "GET"
-    
-    if not is_polling:
-        logger.info(f"Incoming request: {request.method} {request.url.path}")
-    
-    try:
-        response = await call_next(request)
+    with belief_scope("log_requests"):
+        # Avoid spamming logs for polling endpoints
+        is_polling = request.url.path.endswith("/api/tasks") and request.method == "GET"
+        
         if not is_polling:
-            logger.info(f"Response status: {response.status_code} for {request.url.path}")
-        return response
-    except NetworkError as e:
-        logger.error(f"Network error caught in middleware: {e}")
-        raise HTTPException(
-            status_code=503,
-            detail="Environment unavailable. Please check if the Superset instance is running."
-        )
+            logger.info(f"Incoming request: {request.method} {request.url.path}")
+        
+        try:
+            response = await call_next(request)
+            if not is_polling:
+                logger.info(f"Response status: {response.status_code} for {request.url.path}")
+            return response
+        except NetworkError as e:
+            logger.error(f"Network error caught in middleware: {e}")
+            raise HTTPException(
+                status_code=503,
+                detail="Environment unavailable. Please check if the Superset instance is running."
+            )
 # [/DEF:log_requests:Function]
 
 # Include API routes
@@ -119,12 +127,12 @@ app.include_router(environments.router, tags=["Environments"])
 app.include_router(mappings.router, prefix="/api/mappings", tags=["Mappings"])
 app.include_router(migration.router)
 app.include_router(git.router, prefix="/api/git", tags=["Git"])
-app.include_router(llm.router, prefix="/api/llm", tags=["LLM"])
-app.include_router(storage.router, prefix="/api/storage", tags=["Storage"])
-app.include_router(dashboards.router)
-app.include_router(datasets.router)
-app.include_router(reports.router)
-app.include_router(assistant.router, prefix="/api/assistant", tags=["Assistant"])
+app.include_router(llm.router, prefix="/api/llm", tags=["LLM"])
+app.include_router(storage.router, prefix="/api/storage", tags=["Storage"])
+app.include_router(dashboards.router)
+app.include_router(datasets.router)
+app.include_router(reports.router)
+app.include_router(assistant.router, prefix="/api/assistant", tags=["Assistant"])
 
 
 # [DEF:api.include_routers:Action]
@@ -249,12 +257,13 @@ if frontend_path.exists():
     # @POST: Returns the requested file or index.html.
     @app.get("/{file_path:path}", include_in_schema=False)
     async def serve_spa(file_path: str):
-        # Only serve SPA for non-API paths
-        # API routes are registered separately and should be matched by FastAPI first
-        if file_path and (file_path.startswith("api/") or file_path.startswith("/api/") or file_path == "api"):
-            # This should not happen if API routers are properly registered
-            # Return 404 instead of serving HTML
-            raise HTTPException(status_code=404, detail=f"API endpoint not found: {file_path}")
+        with belief_scope("serve_spa"):
+            # Only serve SPA for non-API paths
+            # API routes are registered separately and should be matched by FastAPI first
+            if file_path and (file_path.startswith("api/") or file_path.startswith("/api/") or file_path == "api"):
+                # This should not happen if API routers are properly registered
+                # Return 404 instead of serving HTML
+                raise HTTPException(status_code=404, detail=f"API endpoint not found: {file_path}")
         
         full_path = frontend_path / file_path
         if file_path and full_path.is_file():