Add RandomSampler aggregation (#1595)

elastic · Apr 15, 2024 · 380b63b · 380b63b
1 parent 619daa1
commit 380b63b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/elasticsearch_dsl/aggs.py b/elasticsearch_dsl/aggs.py
@@ -278,6 +278,10 @@ class DiversifiedSampler(Bucket):
     name = "diversified_sampler"
 
 
+class RandomSampler(Bucket):
+    name = "random_sampler"
+
+
 class Composite(Bucket):
     name = "composite"
     _param_defs = {

diff --git a/tests/test_aggs.py b/tests/test_aggs.py
@@ -383,3 +383,22 @@ def test_normalize_aggregation():
     assert {
         "normalize": {"buckets_path": "normalized", "method": "percent_of_sum"}
     } == a.to_dict()
+
+
+def test_random_sampler_aggregation():
+    a = aggs.RandomSampler(probability=0.1).metric(
+        "price_percentiles",
+        "percentiles",
+        field="price",
+    )
+
+    assert {
+        "random_sampler": {
+            "probability": 0.1,
+        },
+        "aggs": {
+            "price_percentiles": {
+                "percentiles": {"field": "price"},
+            },
+        },
+    } == a.to_dict()