|
32 | 32 | from mcp.server.lowlevel.server import LifespanResultT |
33 | 33 | from mcp.server.lowlevel.server import Server as MCPServer |
34 | 34 | from mcp.server.lowlevel.server import lifespan as default_lifespan |
35 | | -from mcp.server.message_queue import MessageQueue |
| 35 | +from mcp.server.message_queue import MessageDispatch |
36 | 36 | from mcp.server.session import ServerSession, ServerSessionT |
37 | 37 | from mcp.server.sse import SseServerTransport |
38 | 38 | from mcp.server.stdio import stdio_server |
@@ -77,8 +77,8 @@ class Settings(BaseSettings, Generic[LifespanResultT]): |
77 | 77 | message_path: str = "/messages/" |
78 | 78 |
|
79 | 79 | # SSE message queue settings |
80 | | - message_queue: MessageQueue | None = Field( |
81 | | - None, description="Custom message queue instance" |
| 80 | + message_dispatch: MessageDispatch | None = Field( |
| 81 | + None, description="Custom message dispatch instance" |
82 | 82 | ) |
83 | 83 |
|
84 | 84 | # resource settings |
@@ -486,15 +486,15 @@ async def run_sse_async(self) -> None: |
486 | 486 |
|
487 | 487 | def sse_app(self) -> Starlette: |
488 | 488 | """Return an instance of the SSE server app.""" |
489 | | - message_queue = self.settings.message_queue |
490 | | - if message_queue is None: |
491 | | - from mcp.server.message_queue import InMemoryMessageQueue |
| 489 | + message_dispatch = self.settings.message_dispatch |
| 490 | + if message_dispatch is None: |
| 491 | + from mcp.server.message_queue import InMemoryMessageDispatch |
492 | 492 |
|
493 | | - message_queue = InMemoryMessageQueue() |
494 | | - logger.info("Using default in-memory message queue") |
| 493 | + message_dispatch = InMemoryMessageDispatch() |
| 494 | + logger.info("Using default in-memory message dispatch") |
495 | 495 |
|
496 | 496 | sse = SseServerTransport( |
497 | | - self.settings.message_path, message_queue=message_queue |
| 497 | + self.settings.message_path, message_dispatch=message_dispatch |
498 | 498 | ) |
499 | 499 |
|
500 | 500 | async def handle_sse(request: Request) -> None: |
|
0 commit comments