modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Slim ServerMiddleware to (ctx, call_next) and add OpenTelemetryMiddleware (#2941)

Browse source 
Co-authored-by: Max Isbey <224885523+maxisbey@users.noreply.github.com>
This commit is contained in:
Marcelo Trylesinski 2026-06-22 15:46:30 +02:00 • committed by GitHub
commit ad81ca234a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 374 additions and 86 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/shared/test_jsonrpc_dispatcher.py
View file

@ -2315,11 +2315,11 @@ async def test_server_middleware_observes_cancelled_notification():
await anyio.sleep_forever()
raise NotImplementedError
async def observe(ctx: Any, method: str, params: Mapping[str, Any] | None, call_next: Any) -> Any:
if method == "notifications/cancelled":
observed.append((method, dict(params or {})))
async def observe(ctx: Any, call_next: Any) -> Any:
if ctx.method == "notifications/cancelled":
observed.append((ctx.method, dict(ctx.params or {})))
cancel_observed.set()
return await call_next()
return await call_next(ctx)
server = Server("test-server", on_call_tool=handle_call_tool)
server.middleware.append(observe)