Feat mcp lakebase

pyproject.toml

@@ -36,6 +36,9 @@ classifiers = [
     "Framework :: Hatch",
 ]
 
+[project.optional-dependencies]
+lakebase = ["psycopg[binary]>=3.2.0"]
+
 [build-system]
 requires = ["hatchling", "uv-dynamic-versioning>=0.7.0"]
 build-backend = "hatchling.build"
@@ -54,6 +57,7 @@ dev = [
     "docutils>=0.22.4",
     "fastapi>=0.119.0",
     "mypy>=1.18.2",
+    "psycopg[binary]>=3.2.0",
     "pydantic-settings>=2.11.0",
     "pytest>=8.4.2",
     "pytest-asyncio>=0.24.0",

src/apx/cli/dev/commands.py

@@ -439,6 +439,11 @@ def dev_mcp():
 
     This command should be run from the project root directory.
     """
+    # Load .env file if it exists (for DATABRICKS_CONFIG_PROFILE, etc.)
+    dotenv_path = Path.cwd() / ".env"
+    if dotenv_path.exists():
+        load_dotenv(dotenv_path)
+
     from apx.mcp import run_mcp_server
 
     run_mcp_server()

src/apx/mcp/__init__.py

@@ -11,6 +11,7 @@
 # Import tools to register them with the mcp instance
 # These imports must come after mcp is available
 from apx.mcp import common as _common  # noqa: F401
+from apx.mcp import lakebase as _lakebase  # noqa: F401
 from apx.mcp import sdk as _sdk  # noqa: F401
 
 # Track initialization state