ref(grpc): Use SegmentSource constants for streamed spans (#6542)

Author: alexander-alderman-webb

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,

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,