refactor(server.py): update ShellServer to inherit from mcp.Server for better integration

refactor(server.py): rename handle_request method to handle and adjust parameters for consistency
chore(pyproject.toml): update dependencies to use mcp package instead of mcp-core for improved package management

Author: tumf

mcp_shell_server/server.py

@@ -1,27 +1,23 @@
-import os
 import asyncio
-from typing import Dict, List, Optional, Any
-from mcp_core import MCPServer, MCPRequest
+from mcp import Server
 from .shell_executor import ShellExecutor
 
-class ShellServer(MCPServer):
+
+class ShellServer(Server):
     def __init__(self):
         super().__init__()
         self.executor = ShellExecutor()
 
-    async def handle_request(self, request: MCPRequest) -> Dict[str, Any]:
-        command: List[str] = request.args.get("command", [])
-        stdin: Optional[str] = request.args.get("stdin")
-        
+    async def handle(self, args: dict) -> dict:
+        command = args.get("command", [])
+        stdin = args.get("stdin")
+
         if not command:
-            return {
-                "error": "No command provided",
-                "status": 1
-            }
-        
-        result = await self.executor.execute(command, stdin)
-        return result
+            return {"error": "No command provided", "status": 1}
+
+        return await self.executor.execute(command, stdin)
+
 
 def main():
     server = ShellServer()
-    server.run()
+    server.run()

pyproject.toml

@@ -6,9 +6,7 @@ authors = [
     { name = "tumf" }
 ]
 dependencies = [
-    "uvicorn>=0.27.0",
-    "fastapi>=0.109.0",
-    "mcp-core>=0.2.0",
+    "mcp>=1.1.0",
 ]
 requires-python = ">=3.11"
 readme = "README.md"
@@ -18,8 +16,5 @@ license = { text = "MIT" }
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
-[tool.hatch.metadata]
-allow-direct-references = true
-
 [project.scripts]
 mcp-shell-server = "mcp_shell_server.server:main"