Fix: Correct MCPServerConfig import to resolve Docker container crash (#20)

* fix: correct MCPServerConfig import to resolve Docker container crash

Fixes #19

The application was attempting to import MCPServerConfig from src.models,
but it's actually defined in src.mcp_client as a dataclass. Updated the
import statement to import from the correct module.

Changes:
- Remove MCPServerConfig from src.models import
- Add MCPServerConfig to src.mcp_client import

* fix: correct constants import in tool_manager.py

Also fixes import issue that would cause container to crash after the
MCPServerConfig fix. The import was missing the src. prefix.

* fix: correct rate_limit_endpoint decorator usage

Fixes incorrect parameter usage in rate_limit_endpoint decorators.
The decorator expects an endpoint category string (e.g., "general",
"chat") but was being called with numeric limit= parameters.

Changes:
- Replace @rate_limit_endpoint(limit=X) with @rate_limit_endpoint("general")
- Affects 9 endpoints: /v1/tools/* and /v1/mcp/* endpoints
- Also includes the tool_manager.py constants import fix

All import and decorator issues are now resolved. Docker container
starts successfully and all endpoints are properly configured.

Author: RichardAtCT

src/main.py

@@ -32,15 +32,14 @@
     MCPServerInfoResponse,
     MCPServersListResponse,
     MCPConnectionRequest,
-    MCPServerConfig,
 )
 from src.claude_cli import ClaudeCodeCLI
 from src.message_adapter import MessageAdapter
 from src.auth import verify_api_key, security, validate_claude_code_auth, get_claude_code_auth_info
 from src.parameter_validator import ParameterValidator, CompatibilityReporter
 from src.session_manager import session_manager
 from src.tool_manager import tool_manager
-from src.mcp_client import mcp_client
+from src.mcp_client import mcp_client, MCPServerConfig
 from src.rate_limiter import (
     limiter,
     rate_limit_exceeded_handler,
@@ -897,7 +896,7 @@ async def delete_session(
 
 
 @app.get("/v1/tools", response_model=ToolListResponse)
-@rate_limit_endpoint(limit=100)
+@rate_limit_endpoint("general")
 async def list_tools(
     request: Request, credentials: Optional[HTTPAuthorizationCredentials] = Depends(security)
 ):
@@ -922,7 +921,7 @@ async def list_tools(
 
 
 @app.get("/v1/tools/config", response_model=ToolConfigurationResponse)
-@rate_limit_endpoint(limit=100)
+@rate_limit_endpoint("general")
 async def get_tool_config(
     request: Request,
     session_id: Optional[str] = None,
@@ -944,7 +943,7 @@ async def get_tool_config(
 
 
 @app.post("/v1/tools/config", response_model=ToolConfigurationResponse)
-@rate_limit_endpoint(limit=30)
+@rate_limit_endpoint("general")
 async def update_tool_config(
     config_request: ToolConfigurationRequest,
     request: Request,
@@ -991,7 +990,7 @@ async def update_tool_config(
 
 
 @app.get("/v1/tools/stats")
-@rate_limit_endpoint(limit=100)
+@rate_limit_endpoint("general")
 async def get_tool_stats(
     request: Request, credentials: Optional[HTTPAuthorizationCredentials] = Depends(security)
 ):
@@ -1004,7 +1003,7 @@ async def get_tool_stats(
 
 
 @app.get("/v1/mcp/servers", response_model=MCPServersListResponse)
-@rate_limit_endpoint(limit=100)
+@rate_limit_endpoint("general")
 async def list_mcp_servers(
     request: Request, credentials: Optional[HTTPAuthorizationCredentials] = Depends(security)
 ):
@@ -1040,7 +1039,7 @@ async def list_mcp_servers(
 
 
 @app.post("/v1/mcp/servers")
-@rate_limit_endpoint(limit=30)
+@rate_limit_endpoint("general")
 async def register_mcp_server(
     body: MCPServerConfigRequest,
     request: Request,
@@ -1069,7 +1068,7 @@ async def register_mcp_server(
 
 
 @app.post("/v1/mcp/connect")
-@rate_limit_endpoint(limit=30)
+@rate_limit_endpoint("general")
 async def connect_mcp_server(
     body: MCPConnectionRequest,
     request: Request,
@@ -1100,7 +1099,7 @@ async def connect_mcp_server(
 
 
 @app.post("/v1/mcp/disconnect")
-@rate_limit_endpoint(limit=30)
+@rate_limit_endpoint("general")
 async def disconnect_mcp_server(
     body: MCPConnectionRequest,
     request: Request,
@@ -1125,7 +1124,7 @@ async def disconnect_mcp_server(
 
 
 @app.get("/v1/mcp/stats")
-@rate_limit_endpoint(limit=100)
+@rate_limit_endpoint("general")
 async def get_mcp_stats(
     request: Request, credentials: Optional[HTTPAuthorizationCredentials] = Depends(security)
 ):

src/tool_manager.py

@@ -10,7 +10,7 @@
 from threading import Lock
 from datetime import datetime
 
-from constants import CLAUDE_TOOLS, DEFAULT_ALLOWED_TOOLS, DEFAULT_DISALLOWED_TOOLS
+from src.constants import CLAUDE_TOOLS, DEFAULT_ALLOWED_TOOLS, DEFAULT_DISALLOWED_TOOLS
 
 logger = logging.getLogger(__name__)