Max
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
"""One tool that rendezvouses with named peers, proving the server dispatches calls concurrently."""
|
|
|
|
from collections import defaultdict
|
|
|
|
import anyio
|
|
|
|
from mcp.server.mcpserver import Context, MCPServer
|
|
from stories._hosting import run_server_from_args
|
|
|
|
|
|
def build_server() -> MCPServer:
|
|
mcp = MCPServer("parallel-calls-example")
|
|
# One Event per tag, shared across every call to this server instance. A handler sets its
|
|
# own tag's event, then waits for every peer's — so no call can return until all named
|
|
# peers are concurrently in-flight. A sequential dispatcher would deadlock here.
|
|
arrivals: dict[str, anyio.Event] = defaultdict(anyio.Event)
|
|
|
|
@mcp.tool()
|
|
async def meet(tag: str, party: list[str], ctx: Context) -> str:
|
|
"""Signal arrival as `tag`, block until every tag in `party` has also arrived, then return."""
|
|
arrivals[tag].set()
|
|
for peer in party:
|
|
await arrivals[peer].wait()
|
|
await ctx.report_progress(1.0, total=1.0, message=tag)
|
|
return tag
|
|
|
|
return mcp
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_server_from_args(build_server)
|