.

Author: alexander-alderman-webb

sentry_sdk/integrations/mcp.py

@@ -55,7 +55,7 @@
 if TYPE_CHECKING:
     from typing import Any, Callable, ContextManager, Optional, Tuple, Union
 
-    from starlette.types import Receive, Scope, Send  # type: ignore[import-not-found]
+    from starlette.types import Receive, Scope, Send
 
     from sentry_sdk.traces import StreamedSpan
     from sentry_sdk.tracing import Span

sentry_sdk/integrations/starlette.py

@@ -42,8 +42,9 @@
 if TYPE_CHECKING:
     from typing import Any, Awaitable, Callable, Container, Dict, Optional, Tuple, Union
 
-    from sentry_sdk._types import Event, HttpStatusCodeRange
+    from starlette.middleware import _MiddlewareFactory
 
+    from sentry_sdk._types import Event, HttpStatusCodeRange
 try:
     import starlette
     from starlette import __version__ as STARLETTE_VERSION
@@ -156,7 +157,7 @@ def setup_once() -> None:
             patch_templates()
 
 
-def _enable_span_for_middleware(middleware_class: "Any") -> type:
+def _enable_span_for_middleware(middleware_class: "Any") -> _MiddlewareFactory[...,]:
     old_call = middleware_class.__call__
 
     async def _create_span_call(
@@ -701,8 +702,6 @@ class StarletteRequestExtractor:
     (like form data or cookies) and adds it to the Sentry event.
     """
 
-    request: "Request" = None
-
     def __init__(self: "StarletteRequestExtractor", request: "Request") -> None:
         self.request = request