feat: adding exception handling middleware for mcp server (#13)

Author: halucinor

pyproject.toml

@@ -5,7 +5,7 @@ description = "A MCP server providing OpenStack services for MCP clients"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "mcp[cli]>=1.12.0",
+    "fastmcp>=2.11.0",
     "openstacksdk>=4.6.0",
     "pydantic>=2.11.7",
 ]

src/openstack_mcp_server/server.py

@@ -1,15 +1,23 @@
-from mcp.server.fastmcp import FastMCP
+from fastmcp.server import FastMCP
+from fastmcp.server.middleware.error_handling import ErrorHandlingMiddleware
+from fastmcp.server.middleware.logging import LoggingMiddleware
 from openstack_mcp_server.tools import register_tool
 
 
 def serve(transport: str, **kwargs):
     """Serve the MCP server with the specified transport."""
-    mcp = FastMCP("openstack_mcp_server")
+    mcp = FastMCP(
+        "openstack_mcp_server",
+    )
 
     register_tool(mcp)
     # resister_resources(mcp)
     # register_prompt(mcp)
 
+    # Add middlewares
+    mcp.add_middleware(ErrorHandlingMiddleware())
+    mcp.add_middleware(LoggingMiddleware())
+
     if transport == "stdio":
         mcp.run(transport="stdio", **kwargs)
     elif transport == "streamable-http":

src/openstack_mcp_server/tools/__init__.py

@@ -1,4 +1,4 @@
-from mcp.server.fastmcp import FastMCP
+from fastmcp import FastMCP
 
 
 def register_tool(mcp: FastMCP):
@@ -9,4 +9,4 @@ def register_tool(mcp: FastMCP):
     from .nova_tools import NovaTools
 
     NovaTools().register_tools(mcp)
-    GlanceTools.register_tools(mcp)
+    GlanceTools().register_tools(mcp)

src/openstack_mcp_server/tools/glance_tools.py

@@ -1,5 +1,5 @@
 from .base import get_openstack_conn
-from mcp.server.fastmcp import FastMCP
+from fastmcp import FastMCP
 
 
 class GlanceTools:

src/openstack_mcp_server/tools/nova_tools.py

@@ -1,5 +1,5 @@
 from .base import get_openstack_conn
-from mcp.server.fastmcp import FastMCP
+from fastmcp import FastMCP
 
 
 class NovaTools: