-
Notifications
You must be signed in to change notification settings - Fork 574
fix(mcp): Nest MCP spans under HTTP transactions #5292
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Conversation
sentry_sdk/scope.py
Outdated
| def use_isolation_scope(isolation_scope: "Scope") -> "Generator[Scope, None, None]": | ||
| def use_isolation_scope(isolation_scope: "S") -> "Generator[S, None, None]": | ||
| """ | ||
| .. versionadded:: 2.0.0 | ||
| Context manager that uses the given `isolation_scope` and runs the wrapped code in it. | ||
| The current scope is also forked to not bleed data into the existing current scope. | ||
| After the wrapped code is executed, the original scopes are restored. | ||
| Example Usage: | ||
| .. code-block:: python | ||
| import sentry_sdk | ||
| with sentry_sdk.isolation_scope() as scope: | ||
| scope.set_tag("color", "green") | ||
| sentry_sdk.capture_message("hello") # will include `color` tag. | ||
| sentry_sdk.capture_message("hello, again") # will NOT include `color` tag. | ||
| """ | ||
| if isolation_scope is None: | ||
| yield isolation_scope | ||
| return |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'm not sure about accepting None in the context managers. It's convenient in the MCP integration since it helps avoid code duplication (or the need to break stuff into more functions).
We could also just check if the scopes are None, so I can adjust if needed.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I probably wouldn't allow None in the context managers. If they were just internal utils it'd probably be ok to update them to support this sort of internal usecase, but since they're part of the public API, we should keep them tidy.
You might be able to use nullcontext instead to get around some of the potential mess of having to conditionally wrap code in a with block.
Semver Impact of This PR🟢 Patch (bug fixes) 📋 Changelog PreviewThis is how your changes will appear in the changelog. New Features ✨
Bug Fixes 🐛Integrations
Litellm
Other
Documentation 📚
Internal Changes 🔧Release
Other
🤖 This preview updates automatically when you update the PR. |
sentrivana
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
See comments, but other than that LGTM!
Description
Patch
StreamableHTTPServerTransport.handle_request()to store isolation and current scopes on the Starlette request object'sscopeattribute.Run MCP handlers in the context of the stored scopes, if these are available.
Modify
use_isolation_scope()anduse_scope()context managers to acceptNone. The MCP protocol allows for transport layers that are not HTTP-based, so the scopes may not be available.The Python MCP SDK uses an in-memory queue of requests. The HTTP propagation context is not active in the MCP decorators we patch, since they run after requests are retrieved from the queue.
Issues
Closes #5272
Closes PY-2022
Closes TET-1688
Reminders
tox -e linters.feat:,fix:,ref:,meta:)