flowyml 1.2.0__py3-none-any.whl → 1.4.0__py3-none-any.whl

flowyml/ui/backend/dependencies.py

@@ -0,0 +1,28 @@
+"""Backend dependencies."""
+
+import os
+from flowyml.storage.sql import SQLMetadataStore
+from flowyml.utils.config import get_config
+
+_store = None
+
+
+def get_store() -> SQLMetadataStore:
+    """Get the metadata store instance.
+
+    Uses FLOWYML_DATABASE_URL if set, otherwise defaults to local SQLite.
+    """
+    global _store
+    if _store is None:
+        config = get_config()
+        db_url = os.environ.get("FLOWYML_DATABASE_URL")
+
+        # If no explicit URL, use the config's metadata_db path
+        if not db_url:
+            db_path = config.metadata_db
+            # Ensure it's a string path for SQLMetadataStore
+            _store = SQLMetadataStore(db_path=str(db_path))
+        else:
+            _store = SQLMetadataStore(db_url=db_url)
+
+    return _store

flowyml/ui/backend/main.py

@@ -1,8 +1,12 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.exceptions import RequestValidationError
 import os
+import traceback
+
+from flowyml.monitoring.alerts import alert_manager, AlertLevel
 
 # Include API routers
 from flowyml.ui.backend.routers import (
@@ -17,6 +21,9 @@ from flowyml.ui.backend.routers import (
     leaderboard,
     execution,
     plugins,
+    metrics,
+    client,
+    stats,
 )
 
 app = FastAPI(
@@ -52,6 +59,7 @@ async def get_public_config():
         "remote_server_url": config.remote_server_url,
         "remote_ui_url": config.remote_ui_url,
         "enable_ui": config.enable_ui,
+        "remote_services": config.remote_services,
     }
 
 
@@ -65,86 +73,10 @@ app.include_router(schedules.router, prefix="/api/schedules", tags=["schedules"]
 app.include_router(notifications.router, prefix="/api/notifications", tags=["notifications"])
 app.include_router(leaderboard.router, prefix="/api/leaderboard", tags=["leaderboard"])
 app.include_router(execution.router, prefix="/api/execution", tags=["execution"])
+app.include_router(metrics.router, prefix="/api/metrics", tags=["metrics"])
 app.include_router(plugins.router, prefix="/api", tags=["plugins"])
-
-
-# Stats endpoint for dashboard
-@app.get("/api/stats")
-async def get_stats(project: str = None):
-    """Get overall statistics for the dashboard, optionally filtered by project."""
-    try:
-        from flowyml.storage.metadata import SQLiteMetadataStore
-
-        store = SQLiteMetadataStore()
-
-        # Get base stats
-        stats = store.get_statistics()
-
-        # Get run status counts (not in get_statistics yet)
-        # We can add this to get_statistics later, but for now let's query efficiently
-        import sqlite3
-
-        conn = sqlite3.connect(store.db_path)
-        cursor = conn.cursor()
-
-        if project:
-            cursor.execute(
-                "SELECT COUNT(*) FROM runs WHERE project = ? AND status = 'completed'",
-                [project],
-            )
-            completed_runs = cursor.fetchone()[0]
-
-            cursor.execute(
-                "SELECT COUNT(*) FROM runs WHERE project = ? AND status = 'failed'",
-                [project],
-            )
-            failed_runs = cursor.fetchone()[0]
-
-            cursor.execute(
-                "SELECT AVG(duration) FROM runs WHERE project = ? AND duration IS NOT NULL",
-                [project],
-            )
-            avg_duration = cursor.fetchone()[0] or 0
-
-            cursor.execute(
-                "SELECT COUNT(*) FROM runs WHERE project = ?",
-                [project],
-            )
-            total_runs = cursor.fetchone()[0]
-        else:
-            cursor.execute("SELECT COUNT(*) FROM runs WHERE status = 'completed'")
-            completed_runs = cursor.fetchone()[0]
-
-            cursor.execute("SELECT COUNT(*) FROM runs WHERE status = 'failed'")
-            failed_runs = cursor.fetchone()[0]
-
-            cursor.execute("SELECT AVG(duration) FROM runs WHERE duration IS NOT NULL")
-            avg_duration = cursor.fetchone()[0] or 0
-
-            cursor.execute("SELECT COUNT(*) FROM runs")
-            total_runs = cursor.fetchone()[0]
-
-        conn.close()
-
-        return {
-            "runs": total_runs if project else stats.get("total_runs", 0),
-            "completed_runs": completed_runs,
-            "failed_runs": failed_runs,
-            "pipelines": stats.get("total_pipelines", 0),  # TODO: filter by project
-            "artifacts": stats.get("total_artifacts", 0),  # TODO: filter by project
-            "avg_duration": avg_duration,
-        }
-    except Exception as e:
-        # Return default stats if there's an error
-        return {
-            "runs": 0,
-            "completed_runs": 0,
-            "failed_runs": 0,
-            "pipelines": 0,
-            "artifacts": 0,
-            "avg_duration": 0,
-            "error": str(e),
-        }
+app.include_router(client.router, prefix="/api/client", tags=["client"])
+app.include_router(stats.router, prefix="/api/stats", tags=["stats"])
 
 
 # Static file serving for frontend
@@ -185,3 +117,34 @@ else:
             "message": "flowyml API is running.",
             "detail": "Frontend not built. Run 'npm run build' in flowyml/ui/frontend to enable the UI.",
         }
+
+
+@app.exception_handler(Exception)
+async def global_exception_handler(request, exc):
+    error_msg = str(exc)
+    stack_trace = traceback.format_exc()
+
+    # Log and alert
+    alert_manager.send_alert(
+        title="Backend API Error",
+        message=f"Unhandled exception in {request.method} {request.url.path}: {error_msg}",
+        level=AlertLevel.ERROR,
+        metadata={"traceback": stack_trace, "path": request.url.path},
+    )
+
+    return JSONResponse(
+        status_code=500,
+        content={
+            "error": "Internal Server Error",
+            "message": "Something went wrong on our end. We've been notified.",
+            "detail": error_msg,  # In prod maybe hide this, but for now it's useful
+        },
+    )
+
+
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request, exc):
+    return JSONResponse(
+        status_code=422,
+        content={"error": "Validation Error", "detail": exc.errors()},
+    )