diff --git a/sentry_sdk/integrations/grpc/aio/server.py b/sentry_sdk/integrations/grpc/aio/server.py index 8ea2d2405b..010337e98c 100644 --- a/sentry_sdk/integrations/grpc/aio/server.py +++ b/sentry_sdk/integrations/grpc/aio/server.py @@ -4,6 +4,7 @@ from sentry_sdk.consts import OP from sentry_sdk.integrations import DidNotEnable from sentry_sdk.integrations.grpc.consts import SPAN_ORIGIN +from sentry_sdk.traces import SegmentSource from sentry_sdk.tracing import TransactionSource from sentry_sdk.tracing_utils import has_span_streaming_enabled from sentry_sdk.utils import event_from_exception @@ -66,7 +67,7 @@ async def wrapped(request: "Any", context: "ServicerContext") -> "Any": name=name, attributes={ "sentry.op": OP.GRPC_SERVER, - "sentry.span.source": TransactionSource.CUSTOM.value, + "sentry.span.source": SegmentSource.CUSTOM.value, "sentry.origin": SPAN_ORIGIN, }, parent_span=None, diff --git a/sentry_sdk/integrations/grpc/server.py b/sentry_sdk/integrations/grpc/server.py index 7918e984c2..1cba1d4b85 100644 --- a/sentry_sdk/integrations/grpc/server.py +++ b/sentry_sdk/integrations/grpc/server.py @@ -4,6 +4,7 @@ from sentry_sdk.consts import OP from sentry_sdk.integrations import DidNotEnable from sentry_sdk.integrations.grpc.consts import SPAN_ORIGIN +from sentry_sdk.traces import SegmentSource from sentry_sdk.tracing import TransactionSource from sentry_sdk.tracing_utils import has_span_streaming_enabled @@ -57,7 +58,7 @@ def behavior(request: "Message", context: "ServicerContext") -> "Message": name=name, attributes={ "sentry.op": OP.GRPC_SERVER, - "sentry.span.source": TransactionSource.CUSTOM.value, + "sentry.span.source": SegmentSource.CUSTOM.value, "sentry.origin": SPAN_ORIGIN, }, parent_span=None,