feat: fix src import

Author: datnguye

pyproject.toml

@@ -40,6 +40,8 @@ build-backend = "pdm.backend"
 
 [tool.pdm]
 version = {source = "scm"}
+[tool.pdm.build]
+package-dir = "src"
 
 [tool.ruff]
 line-length = 120

src/__init__.py

@@ -0,0 +1 @@
+"""MetricFlow MCP server package."""

src/main/main_sse.py

@@ -4,8 +4,8 @@
 
 import uvicorn
 
-from src.server.sse_server import app
-from src.utils.logger import configure_logging
+from server.sse_server import app
+from utils.logger import configure_logging
 
 # Configure logging
 logger = configure_logging()

src/main/main_stdio.py

@@ -1,12 +1,17 @@
 """Main entry point for MetricFlow MCP server (STDIO mode)."""
 
-from src.server.stdio_server import main as stdio_main
-from src.utils.logger import configure_logging
+from server.stdio_server import main as stdio_main
+from utils.logger import configure_logging
 
 # Configure logging
 logger = configure_logging()
 
 
+def main() -> None:
+    """Run the MetricFlow MCP server in STDIO mode."""
+    stdio_main()
+
+
 if __name__ == "__main__":
     # Run in STDIO mode
-    stdio_main()
+    main()

src/server/auth.py

@@ -9,7 +9,7 @@
 from fastapi import Depends, HTTPException, Request, status
 from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
 
-from src.utils.logger import logger
+from utils.logger import logger
 
 # Security scheme for Bearer token authentication
 security = HTTPBearer(auto_error=False)

src/server/sse_server.py

@@ -9,10 +9,10 @@
 from mcp.server.fastmcp import FastMCP
 from mcp.server.sse import SseServerTransport
 
-from src.config.config import load_mf_config
-from src.server.auth import Authenticated, validate_auth_config
-from src.tools.cli_tools import register_mf_cli_tools
-from src.utils.logger import logger
+from config.config import load_mf_config
+from server.auth import Authenticated, validate_auth_config
+from tools.cli_tools import register_mf_cli_tools
+from utils.logger import logger
 
 # Global variables
 transport = SseServerTransport("/messages/")

src/server/stdio_server.py

@@ -6,9 +6,9 @@
 from mcp.server.fastmcp import FastMCP
 from mcp.server.stdio import stdio_server
 
-from src.config.config import load_mf_config
-from src.tools.cli_tools import register_mf_cli_tools
-from src.utils.logger import logger
+from config.config import load_mf_config
+from tools.cli_tools import register_mf_cli_tools
+from utils.logger import logger
 
 
 async def _register_tools(mcp_server: FastMCP, config: Any) -> None:

src/tools/cli_tools.py

@@ -2,15 +2,15 @@
 
 from mcp.server.fastmcp import FastMCP
 
-from src.config.config import MfCliConfig
-from src.tools.metricflow.health_checks import health_checks
-from src.tools.metricflow.list_dimension_values import list_dimension_values
-from src.tools.metricflow.list_dimensions import list_dimensions
-from src.tools.metricflow.list_entities import list_entities
-from src.tools.metricflow.list_metrics import list_metrics
-from src.tools.metricflow.query import query
-from src.tools.metricflow.validate_configs import validate_configs
-from src.utils.prompts import load_prompt
+from config.config import MfCliConfig
+from tools.metricflow.health_checks import health_checks
+from tools.metricflow.list_dimension_values import list_dimension_values
+from tools.metricflow.list_dimensions import list_dimensions
+from tools.metricflow.list_entities import list_entities
+from tools.metricflow.list_metrics import list_metrics
+from tools.metricflow.query import query
+from tools.metricflow.validate_configs import validate_configs
+from utils.prompts import load_prompt
 
 
 def register_mf_cli_tools(mf_mcp: FastMCP, config: MfCliConfig) -> None:

src/tools/metricflow/base.py

@@ -3,8 +3,8 @@
 import os
 import subprocess  # nosec B404
 
-from src.config.config import MfCliConfig
-from src.utils.logger import logger
+from config.config import MfCliConfig
+from utils.logger import logger
 
 
 def run_mf_command(command: list[str], config: MfCliConfig) -> tuple[str, str]:

src/tools/metricflow/health_checks.py

@@ -2,8 +2,8 @@
 
 import json
 
-from src.config.config import MfCliConfig
-from src.tools.metricflow.base import run_mf_command
+from config.config import MfCliConfig
+from tools.metricflow.base import run_mf_command
 
 
 def health_checks(config: MfCliConfig) -> tuple[str, str]: