missing one file

Author: aponcedeleonch

src/codegate/dashboard/dashboard.py

@@ -6,8 +6,9 @@
 import structlog
 from fastapi import APIRouter, Depends, FastAPI
 from fastapi.responses import StreamingResponse
-from codegate import __version__
 
+from codegate import __version__
+from codegate.api.v1 import v1
 from codegate.dashboard.post_processing import (
     parse_get_alert_conversation,
     parse_messages_in_conversations,
@@ -20,23 +21,23 @@
 dashboard_router = APIRouter(tags=["Dashboard"])
 db_reader = None
 
+
 def get_db_reader():
     global db_reader
     if db_reader is None:
         db_reader = DbReader()
     return db_reader
 
+
 def fetch_latest_version() -> str:
     url = "https://api.github.com/repos/stacklok/codegate/releases/latest"
-    headers = {
-        "Accept": "application/vnd.github+json",
-        "X-GitHub-Api-Version": "2022-11-28"
-    }
+    headers = {"Accept": "application/vnd.github+json", "X-GitHub-Api-Version": "2022-11-28"}
     response = requests.get(url, headers=headers, timeout=5)
     response.raise_for_status()
     data = response.json()
     return data.get("tag_name", "unknown")
 
+
 @dashboard_router.get("/dashboard/messages")
 def get_messages(db_reader: DbReader = Depends(get_db_reader)) -> List[Conversation]:
     """
@@ -72,17 +73,18 @@ async def stream_sse():
     """
     return StreamingResponse(generate_sse_events(), media_type="text/event-stream")
 
+
 @dashboard_router.get("/dashboard/version")
 def version_check():
     try:
         latest_version = fetch_latest_version()
 
         # normalize the versions as github will return them with a 'v' prefix
-        current_version = __version__.lstrip('v')
-        latest_version_stripped = latest_version.lstrip('v')
+        current_version = __version__.lstrip("v")
+        latest_version_stripped = latest_version.lstrip("v")
 
         is_latest: bool = latest_version_stripped == current_version
-        
+
         return {
             "current_version": current_version,
             "latest_version": latest_version_stripped,
@@ -95,15 +97,15 @@ def version_check():
             "current_version": __version__,
             "latest_version": "unknown",
             "is_latest": None,
-            "error": "An error occurred while fetching the latest version"
+            "error": "An error occurred while fetching the latest version",
         }
     except Exception as e:
         logger.error(f"Unexpected error: {str(e)}")
         return {
             "current_version": __version__,
             "latest_version": "unknown",
             "is_latest": None,
-            "error": "An unexpected error occurred"
+            "error": "An unexpected error occurred",
         }
 
 
@@ -113,6 +115,7 @@ def generate_openapi():
 
     # Include your defined router
     app.include_router(dashboard_router)
+    app.include_router(v1, prefix="/api/v1", tags=["CodeGate API"])
 
     # Generate OpenAPI JSON
     openapi_schema = app.openapi()