refactor(server.py): remove inheritance from Server class and use stdio_server for handling connections to simplify the code structure

bump(dependencies): update mcp dependency version to 1.1.1 for compatibility and improvements
chore(pyproject.toml): add optional dependencies for testing and define pytest plugins in conftest.py for better test organization

Author: tumf

mcp_shell_server/server.py

@@ -1,11 +1,9 @@
-import asyncio
-from mcp import Server
+from mcp.server.stdio import stdio_server
 from .shell_executor import ShellExecutor
 
 
-class ShellServer(Server):
+class ShellServer:
     def __init__(self):
-        super().__init__()
         self.executor = ShellExecutor()
 
     async def handle(self, args: dict) -> dict:
@@ -20,4 +18,4 @@ async def handle(self, args: dict) -> dict:
 
 def main():
     server = ShellServer()
-    server.run()
+    stdio_server(server.handle)

pyproject.toml

@@ -6,15 +6,22 @@ authors = [
     { name = "tumf" }
 ]
 dependencies = [
-    "mcp>=1.1.0",
+    "mcp>=1.1.1",
 ]
 requires-python = ">=3.11"
 readme = "README.md"
 license = { text = "MIT" }
 
+[project.scripts]
+mcp-shell-server = "mcp_shell_server.server:main"
+
+[project.optional-dependencies]
+test = [
+    "pytest>=7.4.0",
+    "pytest-asyncio>=0.23.0",
+    "pytest-env>=1.1.0",
+]
+
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
-
-[project.scripts]
-mcp-shell-server = "mcp_shell_server.server:main"

tests/conftest.py

@@ -0,0 +1,3 @@
+import pytest
+
+pytest_plugins = ('pytest_asyncio',)