refactor: update environment variable handling for configuration in main.py and config.py

main.py

@@ -5,6 +5,7 @@ from fastmcp import FastMCP, Context
 from loguru import logger
 from dataclasses import dataclass
 import uuid
+import os
 
 from utils.config import API_CONFIG
 from utils.model import MultiModuleRequest, SingleModuleRequest, SearchSpaceRoutingRequest
@@ -13,6 +14,9 @@ from fastmcp.server.dependencies import get_http_request
 from starlette.requests import Request
 
 
+from dotenv import load_dotenv
+load_dotenv()
+
 @dataclass
 class DSContext:
     """Typed context for DS MCP server"""
@@ -386,9 +390,9 @@ Return a 2-3x more detailed/enhanced bug report with precise technical terms for
 
 
 if __name__ == "__main__":
-    transport_mode = API_CONFIG["transport_mode"]
-    host = API_CONFIG["host"]
-    port = API_CONFIG["port"]
-    log_level = API_CONFIG["log_level"]
+    TRANSPORT_MODE = os.getenv("TRANSPORT_MODE", "streamable-http")
+    HOST = os.getenv("HOST", "0.0.0.0")
+    PORT = int(os.getenv("PORT", "8000"))
+    LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO")
 
-    mcp.run(transport=transport_mode, host=host, port=port, log_level=log_level)
+    mcp.run(transport=TRANSPORT_MODE, host=HOST, port=PORT, log_level=LOG_LEVEL)

utils/config.py

@@ -5,14 +5,10 @@ load_dotenv()
 
 # API Configuration
 API_CONFIG = {
-    "api_base_url": os.getenv("DS_API_BASE_URL", "https://dnext-ds-api.hf.space"),
-    "api_multimodule_endpoint": os.getenv("DS_API_MULTIMODULE_ENDPOINT", "/mcp_multi_module_bug_localization"),
-    "api_singlemodule_endpoint": os.getenv("DS_API_SINGLEMODULE_ENDPOINT", "/mcp_single_module_bug_localization"),
-    "api_searchspace_endpoint": os.getenv("DS_API_SEARCHSPACE_ENDPOINT", "/mcp_search_space_routing"),
+    "api_base_url": str(os.getenv("DS_API_BASE_URL", "https://dnext-ds-api.hf.space")),
+    "api_multimodule_endpoint": str(os.getenv("DS_API_MULTIMODULE_ENDPOINT", "/mcp_multi_module_bug_localization")),
+    "api_singlemodule_endpoint": str(os.getenv("DS_API_SINGLEMODULE_ENDPOINT", "/mcp_single_module_bug_localization")),
+    "api_searchspace_endpoint": str(os.getenv("DS_API_SEARCHSPACE_ENDPOINT", "/mcp_search_space_routing")),
     "hf_access_token": os.getenv("HF_ACCESS_TOKEN", None),
-    "timeout": os.getenv("TIMEOUT", 120),
-    "transport_mode": os.getenv("TRANSPORT_MODE", "streamable-http"),
-    "host": os.getenv("HOST", "0.0.0.0"),
-    "port": os.getenv("PORT", 8000),
-    "log_level": os.getenv("LOG_LEVEL", "INFO"),
+    "timeout": int(os.getenv("TIMEOUT", 120)),
 }