Merge pull request #137 from ierickson/ie/upgrade-mcp-package

Upgrades the mcp package dependency to 1.8.1

Author: shahar4499

CHANGELOG.md

@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.3.4]
+
+Updates the `mcp` dependency to `1.8.1`.
+
 ## [0.3.3]
 
 Fixes the broken release from 0.3.2.

fastapi_mcp/transport/sse.py

@@ -5,6 +5,7 @@
 from anyio.streams.memory import MemoryObjectSendStream
 from fastapi import Request, Response, BackgroundTasks, HTTPException
 from fastapi.responses import JSONResponse
+from mcp.shared.message import SessionMessage
 from pydantic import ValidationError
 from mcp.server.sse import SseServerTransport
 from mcp.types import JSONRPCMessage, JSONRPCError, ErrorData
@@ -87,7 +88,7 @@ async def handle_fastapi_post_message(self, request: Request) -> Response:
 
         # Create background task to send message
         background_tasks = BackgroundTasks()
-        background_tasks.add_task(self._send_message_safely, writer, message)
+        background_tasks.add_task(self._send_message_safely, writer, SessionMessage(message))
         logger.debug("Accepting message, will send in background")
 
         # Return response with background task
@@ -96,7 +97,7 @@ async def handle_fastapi_post_message(self, request: Request) -> Response:
         return response
 
     async def _send_message_safely(
-        self, writer: MemoryObjectSendStream[JSONRPCMessage], message: Union[JSONRPCMessage, ValidationError]
+        self, writer: MemoryObjectSendStream[SessionMessage], message: Union[SessionMessage, ValidationError]
     ):
         """Send a message to the writer, avoiding ASGI race conditions"""
 
@@ -115,7 +116,7 @@ async def _send_message_safely(
                     id="unknown",  # We don't know the ID from the invalid request
                     error=error_data,
                 )
-                error_message = JSONRPCMessage(root=json_rpc_error)
+                error_message = SessionMessage(JSONRPCMessage(root=json_rpc_error))
                 await writer.send(error_message)
             else:
                 await writer.send(message)

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "fastapi-mcp"
-version = "0.3.3"
+version = "0.3.4"
 description = "Automatic MCP server generator for FastAPI applications - converts FastAPI endpoints to MCP tools for LLM integration"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -29,7 +29,7 @@ dependencies = [
     "fastapi>=0.100.0",
     "typer>=0.9.0",
     "rich>=13.0.0",
-    "mcp>=1.6.0",
+    "mcp>=1.8.1",
     "pydantic>=2.0.0",
     "pydantic-settings>=2.5.2",
     "uvicorn>=0.20.0",

tests/test_sse_mock_transport.py

@@ -3,6 +3,7 @@
 from uuid import UUID
 from unittest.mock import AsyncMock, MagicMock, patch
 from fastapi import HTTPException, Request
+from mcp.shared.message import SessionMessage
 from pydantic import ValidationError
 from anyio.streams.memory import MemoryObjectSendStream
 
@@ -145,9 +146,10 @@ async def test_send_message_safely_with_validation_error(
     # Verify that the writer.send was called with a JSONRPCError
     assert mock_writer.send.called
     sent_message = mock_writer.send.call_args[0][0]
-    assert isinstance(sent_message, JSONRPCMessage)
-    assert isinstance(sent_message.root, JSONRPCError)
-    assert sent_message.root.error.code == -32700  # Parse error code
+    assert isinstance(sent_message, SessionMessage)
+    assert isinstance(sent_message.message, JSONRPCMessage)
+    assert isinstance(sent_message.message.root, JSONRPCError)
+    assert sent_message.message.root.error.code == -32700  # Parse error code
 
 
 @pytest.mark.anyio
@@ -156,7 +158,9 @@ async def test_send_message_safely_with_jsonrpc_message(
 ) -> None:
     """Test sending a JSONRPCMessage safely."""
     # Create a JSONRPCMessage
-    message = JSONRPCMessage.model_validate({"jsonrpc": "2.0", "id": "123", "method": "test_method", "params": {}})
+    message = SessionMessage(
+        JSONRPCMessage.model_validate({"jsonrpc": "2.0", "id": "123", "method": "test_method", "params": {}})
+    )
 
     # Call the function
     await mock_transport._send_message_safely(mock_writer, message)
@@ -176,7 +180,9 @@ async def test_send_message_safely_exception_handling(
     mock_writer.send.side_effect = Exception("Test exception")
 
     # Create a message
-    message = JSONRPCMessage.model_validate({"jsonrpc": "2.0", "id": "123", "method": "test_method", "params": {}})
+    message = SessionMessage(
+        JSONRPCMessage.model_validate({"jsonrpc": "2.0", "id": "123", "method": "test_method", "params": {}})
+    )
 
     # Call the function - it should not raise an exception
     await mock_transport._send_message_safely(mock_writer, message)