style: auto-format code with black

Author: viragtripathi

Makefile

@@ -25,7 +25,6 @@ install:
 
 format:
 	black cockroachdb_mcp_server
-	black app
 
 lint:
 	ruff cockroachdb_mcp_server

cockroachdb_mcp_server/cli.py

@@ -44,7 +44,9 @@ def serve(
     log_level: str = typer.Option(
         "INFO", "--log-level", help="Logging level (DEBUG, INFO, WARN, etc.)"
     ),
-    demo_mode: bool = typer.Option(False, "--demo-mode", help="Enable demo/debug endpoints")
+    demo_mode: bool = typer.Option(
+        False, "--demo-mode", help="Enable demo/debug endpoints"
+    ),
 ):
     """Start the MCP server."""
 

cockroachdb_mcp_server/main.py

@@ -3,10 +3,7 @@
 from cockroachdb_mcp_server.routes import contexts, health, debug
 from cockroachdb_mcp_server import __version__
 
-app = FastAPI(
-    title="CockroachDB MCP Server",
-    version=__version__
-)
+app = FastAPI(title="CockroachDB MCP Server", version=__version__)
 
 app.include_router(health.router)
 app.include_router(contexts.router, prefix="/contexts")
@@ -22,6 +19,7 @@
 for route in app.routes:
     print(f" - {route.path}")
 
+
 @app.get("/")
 def root():
     return {"message": "MCP Server is running"}

cockroachdb_mcp_server/routes/debug.py

@@ -5,9 +5,11 @@
 
 router = APIRouter()
 
+
 class SQLQuery(BaseModel):
     query: str
 
+
 @router.get("/info")
 def get_database_info():
     """
@@ -23,6 +25,7 @@ def get_database_info():
             "version": version,
         }
 
+
 @router.post("/sql")
 def run_raw_sql(payload: SQLQuery):
     """
@@ -40,6 +43,7 @@ def run_raw_sql(payload: SQLQuery):
     except Exception as e:
         return {"error": f"💥 Query failed: {str(e)}"}
 
+
 @router.get("/tables")
 def list_tables():
     """
@@ -49,13 +53,17 @@ def list_tables():
     engine = get_sqlalchemy_engine()
     try:
         with engine.connect() as conn:
-            result = conn.execute(text("""
+            result = conn.execute(
+                text(
+                    """
                 SELECT table_name
                 FROM information_schema.tables
                 WHERE table_schema = 'public'
                   AND table_type = 'BASE TABLE'
                 ORDER BY table_name;
-            """)).fetchall()
+            """
+                )
+            ).fetchall()
             return {"tables": [row[0] for row in result]}
     except Exception as e:
         return {"error": f"❌ Failed to list tables: {str(e)}"}

cockroachdb_mcp_server/server.py

@@ -1,4 +1,5 @@
 import uvicorn
 
+
 def start_server(host="0.0.0.0", port=8081, reload=False, demo_mode=False):
     uvicorn.run("cockroachdb_mcp_server.main:app", host=host, port=port, reload=reload)