diff --git a/src/sentry/search/eap/spans/aggregates.py b/src/sentry/search/eap/spans/aggregates.py index 55bfa21f4fae4b..b901a2a390c023 100644 --- a/src/sentry/search/eap/spans/aggregates.py +++ b/src/sentry/search/eap/spans/aggregates.py @@ -711,7 +711,16 @@ def resolve_bounded_sample(args: ResolvedArguments) -> tuple[AttributeKey, Trace processor=count_processor, arguments=[ AttributeArgumentDefinition( - attribute_types={"string"}, + attribute_types={ + "string", + "duration", + "number", + "integer", + "percentage", + "currency", + *constants.SIZE_TYPE, + *constants.DURATION_TYPE, + }, ) ], ), @@ -741,7 +750,16 @@ def resolve_bounded_sample(args: ResolvedArguments) -> tuple[AttributeKey, Trace processor=count_processor, arguments=[ AttributeArgumentDefinition( - attribute_types={"string"}, + attribute_types={ + "string", + "duration", + "number", + "integer", + "percentage", + "currency", + *constants.SIZE_TYPE, + *constants.DURATION_TYPE, + }, ) ], ), diff --git a/tests/snuba/api/endpoints/test_organization_events_span_indexed.py b/tests/snuba/api/endpoints/test_organization_events_span_indexed.py index d81ff7a5920408..2e4fe412d8412f 100644 --- a/tests/snuba/api/endpoints/test_organization_events_span_indexed.py +++ b/tests/snuba/api/endpoints/test_organization_events_span_indexed.py @@ -2040,6 +2040,7 @@ def test_aggregate_numeric_attr(self) -> None: "count_unique(bar)", "count_unique(tags[bar])", "count_unique(tags[bar,string])", + "count_unique(tags[foo,number])", "count()", "count(span.duration)", "count(tags[foo, number])", @@ -2068,6 +2069,7 @@ def test_aggregate_numeric_attr(self) -> None: "count_unique(bar)": 2, "count_unique(tags[bar])": 2, "count_unique(tags[bar,string])": 2, + "count_unique(tags[foo,number])": 1, "count()": 2, "count(span.duration)": 2, "count(tags[foo, number])": 1,