elasticsearch 8.18.0__py3-none-any.whl → 9.0.1__py3-none-any.whl

elasticsearch/dsl/field.py

@@ -437,7 +437,9 @@ class Object(Field):
         doc_class: Union[Type["InnerDoc"], "DefaultType"] = DEFAULT,
         *args: Any,
         enabled: Union[bool, "DefaultType"] = DEFAULT,
-        subobjects: Union[bool, "DefaultType"] = DEFAULT,
+        subobjects: Union[
+            Literal["true", "false", "auto"], bool, "DefaultType"
+        ] = DEFAULT,
         copy_to: Union[
             Union[str, "InstrumentedField"],
             Sequence[Union[str, "InstrumentedField"]],
@@ -1109,6 +1111,56 @@ class ConstantKeyword(Field):
         super().__init__(*args, **kwargs)
 
 
+class CountedKeyword(Field):
+    """
+    :arg index:
+    :arg meta: Metadata about the field.
+    :arg properties:
+    :arg ignore_above:
+    :arg dynamic:
+    :arg fields:
+    :arg synthetic_source_keep:
+    """
+
+    name = "counted_keyword"
+    _param_defs = {
+        "properties": {"type": "field", "hash": True},
+        "fields": {"type": "field", "hash": True},
+    }
+
+    def __init__(
+        self,
+        *args: Any,
+        index: Union[bool, "DefaultType"] = DEFAULT,
+        meta: Union[Mapping[str, str], "DefaultType"] = DEFAULT,
+        properties: Union[Mapping[str, Field], "DefaultType"] = DEFAULT,
+        ignore_above: Union[int, "DefaultType"] = DEFAULT,
+        dynamic: Union[
+            Literal["strict", "runtime", "true", "false"], bool, "DefaultType"
+        ] = DEFAULT,
+        fields: Union[Mapping[str, Field], "DefaultType"] = DEFAULT,
+        synthetic_source_keep: Union[
+            Literal["none", "arrays", "all"], "DefaultType"
+        ] = DEFAULT,
+        **kwargs: Any,
+    ):
+        if index is not DEFAULT:
+            kwargs["index"] = index
+        if meta is not DEFAULT:
+            kwargs["meta"] = meta
+        if properties is not DEFAULT:
+            kwargs["properties"] = properties
+        if ignore_above is not DEFAULT:
+            kwargs["ignore_above"] = ignore_above
+        if dynamic is not DEFAULT:
+            kwargs["dynamic"] = dynamic
+        if fields is not DEFAULT:
+            kwargs["fields"] = fields
+        if synthetic_source_keep is not DEFAULT:
+            kwargs["synthetic_source_keep"] = synthetic_source_keep
+        super().__init__(*args, **kwargs)
+
+
 class Date(Field):
     """
     :arg default_timezone: timezone that will be automatically used for tz-naive values
@@ -1118,6 +1170,8 @@ class Date(Field):
     :arg format:
     :arg ignore_malformed:
     :arg index:
+    :arg script:
+    :arg on_script_error:
     :arg null_value:
     :arg precision_step:
     :arg locale:
@@ -1150,6 +1204,8 @@ class Date(Field):
         format: Union[str, "DefaultType"] = DEFAULT,
         ignore_malformed: Union[bool, "DefaultType"] = DEFAULT,
         index: Union[bool, "DefaultType"] = DEFAULT,
+        script: Union["types.Script", Dict[str, Any], "DefaultType"] = DEFAULT,
+        on_script_error: Union[Literal["fail", "continue"], "DefaultType"] = DEFAULT,
         null_value: Any = DEFAULT,
         precision_step: Union[int, "DefaultType"] = DEFAULT,
         locale: Union[str, "DefaultType"] = DEFAULT,
@@ -1182,6 +1238,10 @@ class Date(Field):
             kwargs["ignore_malformed"] = ignore_malformed
         if index is not DEFAULT:
             kwargs["index"] = index
+        if script is not DEFAULT:
+            kwargs["script"] = script
+        if on_script_error is not DEFAULT:
+            kwargs["on_script_error"] = on_script_error
         if null_value is not DEFAULT:
             kwargs["null_value"] = null_value
         if precision_step is not DEFAULT:
@@ -1246,6 +1306,8 @@ class DateNanos(Field):
     :arg format:
     :arg ignore_malformed:
     :arg index:
+    :arg script:
+    :arg on_script_error:
     :arg null_value:
     :arg precision_step:
     :arg doc_values:
@@ -1272,6 +1334,8 @@ class DateNanos(Field):
         format: Union[str, "DefaultType"] = DEFAULT,
         ignore_malformed: Union[bool, "DefaultType"] = DEFAULT,
         index: Union[bool, "DefaultType"] = DEFAULT,
+        script: Union["types.Script", Dict[str, Any], "DefaultType"] = DEFAULT,
+        on_script_error: Union[Literal["fail", "continue"], "DefaultType"] = DEFAULT,
         null_value: Any = DEFAULT,
         precision_step: Union[int, "DefaultType"] = DEFAULT,
         doc_values: Union[bool, "DefaultType"] = DEFAULT,
@@ -1301,6 +1365,10 @@ class DateNanos(Field):
             kwargs["ignore_malformed"] = ignore_malformed
         if index is not DEFAULT:
             kwargs["index"] = index
+        if script is not DEFAULT:
+            kwargs["script"] = script
+        if on_script_error is not DEFAULT:
+            kwargs["on_script_error"] = on_script_error
         if null_value is not DEFAULT:
             kwargs["null_value"] = null_value
         if precision_step is not DEFAULT:
@@ -3085,6 +3153,76 @@ class Nested(Object):
         super().__init__(*args, **kwargs)
 
 
+class Passthrough(Field):
+    """
+    :arg enabled:
+    :arg priority:
+    :arg time_series_dimension:
+    :arg copy_to:
+    :arg store:
+    :arg meta: Metadata about the field.
+    :arg properties:
+    :arg ignore_above:
+    :arg dynamic:
+    :arg fields:
+    :arg synthetic_source_keep:
+    """
+
+    name = "passthrough"
+    _param_defs = {
+        "properties": {"type": "field", "hash": True},
+        "fields": {"type": "field", "hash": True},
+    }
+
+    def __init__(
+        self,
+        *args: Any,
+        enabled: Union[bool, "DefaultType"] = DEFAULT,
+        priority: Union[int, "DefaultType"] = DEFAULT,
+        time_series_dimension: Union[bool, "DefaultType"] = DEFAULT,
+        copy_to: Union[
+            Union[str, "InstrumentedField"],
+            Sequence[Union[str, "InstrumentedField"]],
+            "DefaultType",
+        ] = DEFAULT,
+        store: Union[bool, "DefaultType"] = DEFAULT,
+        meta: Union[Mapping[str, str], "DefaultType"] = DEFAULT,
+        properties: Union[Mapping[str, Field], "DefaultType"] = DEFAULT,
+        ignore_above: Union[int, "DefaultType"] = DEFAULT,
+        dynamic: Union[
+            Literal["strict", "runtime", "true", "false"], bool, "DefaultType"
+        ] = DEFAULT,
+        fields: Union[Mapping[str, Field], "DefaultType"] = DEFAULT,
+        synthetic_source_keep: Union[
+            Literal["none", "arrays", "all"], "DefaultType"
+        ] = DEFAULT,
+        **kwargs: Any,
+    ):
+        if enabled is not DEFAULT:
+            kwargs["enabled"] = enabled
+        if priority is not DEFAULT:
+            kwargs["priority"] = priority
+        if time_series_dimension is not DEFAULT:
+            kwargs["time_series_dimension"] = time_series_dimension
+        if copy_to is not DEFAULT:
+            kwargs["copy_to"] = str(copy_to)
+        if store is not DEFAULT:
+            kwargs["store"] = store
+        if meta is not DEFAULT:
+            kwargs["meta"] = meta
+        if properties is not DEFAULT:
+            kwargs["properties"] = properties
+        if ignore_above is not DEFAULT:
+            kwargs["ignore_above"] = ignore_above
+        if dynamic is not DEFAULT:
+            kwargs["dynamic"] = dynamic
+        if fields is not DEFAULT:
+            kwargs["fields"] = fields
+        if synthetic_source_keep is not DEFAULT:
+            kwargs["synthetic_source_keep"] = synthetic_source_keep
+        super().__init__(*args, **kwargs)
+
+
 class Percolator(Field):
     """
     :arg meta: Metadata about the field.

elasticsearch/dsl/query.py

@@ -1083,6 +1083,8 @@ class Knn(Query):
     :arg filter: Filters for the kNN search query
     :arg similarity: The minimum similarity for a vector to be considered
         a match
+    :arg rescore_vector: Apply oversampling and rescoring to quantized
+        vectors *
     :arg boost: Floating point number used to decrease or increase the
         relevance scores of the query. Boost values are relative to the
         default value of 1.0. A boost value between 0 and 1.0 decreases
@@ -1108,6 +1110,9 @@ class Knn(Query):
         k: Union[int, "DefaultType"] = DEFAULT,
         filter: Union[Query, Sequence[Query], "DefaultType"] = DEFAULT,
         similarity: Union[float, "DefaultType"] = DEFAULT,
+        rescore_vector: Union[
+            "types.RescoreVector", Dict[str, Any], "DefaultType"
+        ] = DEFAULT,
         boost: Union[float, "DefaultType"] = DEFAULT,
         _name: Union[str, "DefaultType"] = DEFAULT,
         **kwargs: Any,
@@ -1120,6 +1125,7 @@ class Knn(Query):
             k=k,
             filter=filter,
             similarity=similarity,
+            rescore_vector=rescore_vector,
             boost=boost,
             _name=_name,
             **kwargs,
@@ -2650,7 +2656,7 @@ class Terms(Query):
         self,
         _field: Union[str, "InstrumentedField", "DefaultType"] = DEFAULT,
         _value: Union[
-            Sequence[Union[int, float, str, bool, None, Any]],
+            Sequence[Union[int, float, str, bool, None]],
             "types.TermsLookup",
             Dict[str, Any],
             "DefaultType",

elasticsearch/dsl/search_base.py

@@ -72,6 +72,9 @@ class QueryProxy(Generic[_S]):
     __bool__ = __nonzero__
 
     def __call__(self, *args: Any, **kwargs: Any) -> _S:
+        """
+        Add a query.
+        """
         s = self._search._clone()
 
         # we cannot use self._proxied since we just cloned self._search and
@@ -354,18 +357,22 @@ class SearchBase(Request[_R]):
     post_filter = ProxyDescriptor[Self]("post_filter")
     _response: Response[_R]
 
-    def __init__(self, **kwargs: Any):
+    def __init__(
+        self,
+        using: AnyUsingType = "default",
+        index: Optional[Union[str, List[str]]] = None,
+        **kwargs: Any,
+    ):
         """
         Search request to elasticsearch.
 
         :arg using: `Elasticsearch` instance to use
         :arg index: limit the search to index
-        :arg doc_type: only query this type.
 
         All the parameters supplied (or omitted) at creation type can be later
         overridden by methods (`using`, `index` and `doc_type` respectively).
         """
-        super().__init__(**kwargs)
+        super().__init__(using=using, index=index, **kwargs)
 
         self.aggs = AggsProxy[_R](self)
         self._sort: List[Union[str, Dict[str, Dict[str, str]]]] = []
@@ -383,9 +390,15 @@ class SearchBase(Request[_R]):
         self._post_filter_proxy = QueryProxy(self, "post_filter")
 
     def filter(self, *args: Any, **kwargs: Any) -> Self:
+        """
+        Add a query in filter context.
+        """
         return self.query(Bool(filter=[Q(*args, **kwargs)]))
 
     def exclude(self, *args: Any, **kwargs: Any) -> Self:
+        """
+        Add a negative query in filter context.
+        """
         return self.query(Bool(filter=[~Q(*args, **kwargs)]))
 
     def __getitem__(self, n: Union[int, slice]) -> Self:

elasticsearch/dsl/types.py

@@ -26,34 +26,6 @@ from .utils import AttrDict
 PipeSeparatedFlags = str
 
 
-class AggregationRange(AttrDict[Any]):
-    """
-    :arg from: Start of the range (inclusive).
-    :arg key: Custom key to return the range with.
-    :arg to: End of the range (exclusive).
-    """
-
-    from_: Union[float, None, DefaultType]
-    key: Union[str, DefaultType]
-    to: Union[float, None, DefaultType]
-
-    def __init__(
-        self,
-        *,
-        from_: Union[float, None, DefaultType] = DEFAULT,
-        key: Union[str, DefaultType] = DEFAULT,
-        to: Union[float, None, DefaultType] = DEFAULT,
-        **kwargs: Any,
-    ):
-        if from_ is not DEFAULT:
-            kwargs["from_"] = from_
-        if key is not DEFAULT:
-            kwargs["key"] = key
-        if to is not DEFAULT:
-            kwargs["to"] = to
-        super().__init__(kwargs)
-
-
 class BucketCorrelationFunction(AttrDict[Any]):
     """
     :arg count_correlation: (required) The configuration to calculate a
@@ -334,34 +306,6 @@ class CustomCategorizeTextAnalyzer(AttrDict[Any]):
         super().__init__(kwargs)
 
 
-class DateRangeExpression(AttrDict[Any]):
-    """
-    :arg from: Start of the range (inclusive).
-    :arg key: Custom key to return the range with.
-    :arg to: End of the range (exclusive).
-    """
-
-    from_: Union[str, float, DefaultType]
-    key: Union[str, DefaultType]
-    to: Union[str, float, DefaultType]
-
-    def __init__(
-        self,
-        *,
-        from_: Union[str, float, DefaultType] = DEFAULT,
-        key: Union[str, DefaultType] = DEFAULT,
-        to: Union[str, float, DefaultType] = DEFAULT,
-        **kwargs: Any,
-    ):
-        if from_ is not DEFAULT:
-            kwargs["from_"] = from_
-        if key is not DEFAULT:
-            kwargs["key"] = key
-        if to is not DEFAULT:
-            kwargs["to"] = to
-        super().__init__(kwargs)
-
-
 class DenseVectorIndexOptions(AttrDict[Any]):
     """
     :arg type: (required) The type of kNN algorithm to use.
@@ -380,15 +324,24 @@ class DenseVectorIndexOptions(AttrDict[Any]):
         `int4_flat` index types.
     :arg ef_construction: The number of candidates to track while
         assembling the list of nearest neighbors for each new node.  Only
-        applicable to `hnsw`, `int8_hnsw`, and `int4_hnsw` index types.
-        Defaults to `100` if omitted.
+        applicable to `hnsw`, `int8_hnsw`, `bbq_hnsw`, and `int4_hnsw`
+        index types. Defaults to `100` if omitted.
     :arg m: The number of neighbors each node will be connected to in the
-        HNSW graph.  Only applicable to `hnsw`, `int8_hnsw`, and
-        `int4_hnsw` index types. Defaults to `16` if omitted.
+        HNSW graph.  Only applicable to `hnsw`, `int8_hnsw`, `bbq_hnsw`,
+        and `int4_hnsw` index types. Defaults to `16` if omitted.
     """
 
     type: Union[
-        Literal["flat", "hnsw", "int4_flat", "int4_hnsw", "int8_flat", "int8_hnsw"],
+        Literal[
+            "bbq_flat",
+            "bbq_hnsw",
+            "flat",
+            "hnsw",
+            "int4_flat",
+            "int4_hnsw",
+            "int8_flat",
+            "int8_hnsw",
+        ],
         DefaultType,
     ]
     confidence_interval: Union[float, DefaultType]
@@ -399,7 +352,16 @@ class DenseVectorIndexOptions(AttrDict[Any]):
         self,
         *,
         type: Union[
-            Literal["flat", "hnsw", "int4_flat", "int4_hnsw", "int8_flat", "int8_hnsw"],
+            Literal[
+                "bbq_flat",
+                "bbq_hnsw",
+                "flat",
+                "hnsw",
+                "int4_flat",
+                "int4_hnsw",
+                "int8_flat",
+                "int8_hnsw",
+            ],
             DefaultType,
         ] = DEFAULT,
         confidence_interval: Union[float, DefaultType] = DEFAULT,
@@ -591,6 +553,7 @@ class FieldSort(AttrDict[Any]):
             "completion",
             "nested",
             "object",
+            "passthrough",
             "version",
             "murmur3",
             "token_count",
@@ -617,6 +580,7 @@ class FieldSort(AttrDict[Any]):
             "shape",
             "histogram",
             "constant_keyword",
+            "counted_keyword",
             "aggregate_metric_double",
             "dense_vector",
             "semantic_text",
@@ -654,6 +618,7 @@ class FieldSort(AttrDict[Any]):
                 "completion",
                 "nested",
                 "object",
+                "passthrough",
                 "version",
                 "murmur3",
                 "token_count",
@@ -680,6 +645,7 @@ class FieldSort(AttrDict[Any]):
                 "shape",
                 "histogram",
                 "constant_keyword",
+                "counted_keyword",
                 "aggregate_metric_double",
                 "dense_vector",
                 "semantic_text",
@@ -2625,7 +2591,7 @@ class PercentageScoreHeuristic(AttrDict[Any]):
 class PinnedDoc(AttrDict[Any]):
     """
     :arg _id: (required) The unique document ID.
-    :arg _index: (required) The index that contains the document.
+    :arg _index: The index that contains the document.
     """
 
     _id: Union[str, DefaultType]
@@ -2850,6 +2816,22 @@ class RegressionInferenceOptions(AttrDict[Any]):
         super().__init__(kwargs)
 
 
+class RescoreVector(AttrDict[Any]):
+    """
+    :arg oversample: (required) Applies the specified oversample factor to
+        k on the approximate kNN search
+    """
+
+    oversample: Union[float, DefaultType]
+
+    def __init__(
+        self, *, oversample: Union[float, DefaultType] = DEFAULT, **kwargs: Any
+    ):
+        if oversample is not DEFAULT:
+            kwargs["oversample"] = oversample
+        super().__init__(kwargs)
+
+
 class ScoreSort(AttrDict[Any]):
     """
     :arg order:
@@ -2880,7 +2862,7 @@ class Script(AttrDict[Any]):
     :arg options:
     """
 
-    source: Union[str, DefaultType]
+    source: Union[str, Dict[str, Any], DefaultType]
     id: Union[str, DefaultType]
     params: Union[Mapping[str, Any], DefaultType]
     lang: Union[Literal["painless", "expression", "mustache", "java"], DefaultType]
@@ -2889,7 +2871,7 @@ class Script(AttrDict[Any]):
     def __init__(
         self,
         *,
-        source: Union[str, DefaultType] = DEFAULT,
+        source: Union[str, Dict[str, Any], DefaultType] = DEFAULT,
         id: Union[str, DefaultType] = DEFAULT,
         params: Union[Mapping[str, Any], DefaultType] = DEFAULT,
         lang: Union[
@@ -3488,14 +3470,14 @@ class SpanTermQuery(AttrDict[Any]):
     :arg _name:
     """
 
-    value: Union[str, DefaultType]
+    value: Union[int, float, str, bool, None, DefaultType]
     boost: Union[float, DefaultType]
     _name: Union[str, DefaultType]
 
     def __init__(
         self,
         *,
-        value: Union[str, DefaultType] = DEFAULT,
+        value: Union[int, float, str, bool, None, DefaultType] = DEFAULT,
         boost: Union[float, DefaultType] = DEFAULT,
         _name: Union[str, DefaultType] = DEFAULT,
         **kwargs: Any,
@@ -3613,7 +3595,7 @@ class TermQuery(AttrDict[Any]):
     :arg _name:
     """
 
-    value: Union[int, float, str, bool, None, Any, DefaultType]
+    value: Union[int, float, str, bool, None, DefaultType]
     case_insensitive: Union[bool, DefaultType]
     boost: Union[float, DefaultType]
     _name: Union[str, DefaultType]
@@ -3621,7 +3603,7 @@ class TermQuery(AttrDict[Any]):
     def __init__(
         self,
         *,
-        value: Union[int, float, str, bool, None, Any, DefaultType] = DEFAULT,
+        value: Union[int, float, str, bool, None, DefaultType] = DEFAULT,
         case_insensitive: Union[bool, DefaultType] = DEFAULT,
         boost: Union[float, DefaultType] = DEFAULT,
         _name: Union[str, DefaultType] = DEFAULT,
@@ -3712,7 +3694,7 @@ class TermsSetQuery(AttrDict[Any]):
     :arg _name:
     """
 
-    terms: Union[Sequence[str], DefaultType]
+    terms: Union[Sequence[Union[int, float, str, bool, None]], DefaultType]
     minimum_should_match: Union[int, str, DefaultType]
     minimum_should_match_field: Union[str, InstrumentedField, DefaultType]
     minimum_should_match_script: Union["Script", Dict[str, Any], DefaultType]
@@ -3722,7 +3704,9 @@ class TermsSetQuery(AttrDict[Any]):
     def __init__(
         self,
         *,
-        terms: Union[Sequence[str], DefaultType] = DEFAULT,
+        terms: Union[
+            Sequence[Union[int, float, str, bool, None]], DefaultType
+        ] = DEFAULT,
         minimum_should_match: Union[int, str, DefaultType] = DEFAULT,
         minimum_should_match_field: Union[
             str, InstrumentedField, DefaultType
@@ -4544,7 +4528,7 @@ class CompositeAggregate(AttrDict[Any]):
     :arg meta:
     """
 
-    after_key: Mapping[str, Union[int, float, str, bool, None, Any]]
+    after_key: Mapping[str, Union[int, float, str, bool, None]]
     buckets: Sequence["CompositeBucket"]
     meta: Mapping[str, Any]
 
@@ -4559,7 +4543,7 @@ class CompositeBucket(AttrDict[Any]):
     :arg doc_count: (required)
     """
 
-    key: Mapping[str, Union[int, float, str, bool, None, Any]]
+    key: Mapping[str, Union[int, float, str, bool, None]]
     doc_count: int
 
 
@@ -5235,9 +5219,7 @@ class Hit(AttrDict[Any]):
     matched_queries: Union[Sequence[str], Mapping[str, float]]
     nested: "NestedIdentity"
     ignored: Sequence[str]
-    ignored_field_values: Mapping[
-        str, Sequence[Union[int, float, str, bool, None, Any]]
-    ]
+    ignored_field_values: Mapping[str, Sequence[Any]]
     shard: str
     node: str
     routing: str
@@ -5246,7 +5228,7 @@ class Hit(AttrDict[Any]):
     seq_no: int
     primary_term: int
     version: int
-    sort: Sequence[Union[int, float, str, bool, None, Any]]
+    sort: Sequence[Union[int, float, str, bool, None]]
 
 
 class HitsMetadata(AttrDict[Any]):
@@ -5271,7 +5253,7 @@ class InferenceAggregate(AttrDict[Any]):
     :arg meta:
     """
 
-    value: Union[int, float, str, bool, None, Any]
+    value: Union[int, float, str, bool, None]
     feature_importance: Sequence["InferenceFeatureImportance"]
     top_classes: Sequence["InferenceTopClassEntry"]
     warning: str
@@ -5307,7 +5289,7 @@ class InferenceTopClassEntry(AttrDict[Any]):
     :arg class_score: (required)
     """
 
-    class_name: Union[int, float, str, bool, None, Any]
+    class_name: Union[int, float, str, bool, None]
     class_probability: float
     class_score: float
 
@@ -5636,7 +5618,7 @@ class MultiTermsBucket(AttrDict[Any]):
     :arg doc_count_error_upper_bound:
     """
 
-    key: Sequence[Union[int, float, str, bool, None, Any]]
+    key: Sequence[Union[int, float, str, bool, None]]
     doc_count: int
     key_as_string: str
     doc_count_error_upper_bound: int
@@ -6187,7 +6169,7 @@ class StringTermsBucket(AttrDict[Any]):
     :arg doc_count_error_upper_bound:
     """
 
-    key: Union[int, float, str, bool, None, Any]
+    key: Union[int, float, str, bool, None]
     doc_count: int
     doc_count_error_upper_bound: int
 
@@ -6291,7 +6273,7 @@ class TimeSeriesBucket(AttrDict[Any]):
     :arg doc_count: (required)
     """
 
-    key: Mapping[str, Union[int, float, str, bool, None, Any]]
+    key: Mapping[str, Union[int, float, str, bool, None]]
     doc_count: int
 
 
@@ -6311,8 +6293,8 @@ class TopMetrics(AttrDict[Any]):
     :arg metrics: (required)
     """
 
-    sort: Sequence[Union[Union[int, float, str, bool, None, Any], None]]
-    metrics: Mapping[str, Union[Union[int, float, str, bool, None, Any], None]]
+    sort: Sequence[Union[Union[int, float, str, bool, None], None]]
+    metrics: Mapping[str, Union[Union[int, float, str, bool, None], None]]
 
 
 class TopMetricsAggregate(AttrDict[Any]):

elasticsearch/dsl/wrappers.py

@@ -18,6 +18,7 @@
 import operator
 from typing import (
     TYPE_CHECKING,
+    Any,
     Callable,
     ClassVar,
     Dict,
@@ -117,3 +118,27 @@ class Range(AttrDict[RangeValT]):
         if "gte" in self._d_:
             return self._d_["gte"], True
         return None, False
+
+
+class AggregationRange(AttrDict[Any]):
+    """
+    :arg from: Start of the range (inclusive).
+    :arg key: Custom key to return the range with.
+    :arg to: End of the range (exclusive).
+    """
+
+    def __init__(
+        self,
+        *,
+        from_: Any = None,
+        key: Optional[str] = None,
+        to: Any = None,
+        **kwargs: Any,
+    ):
+        if from_ is not None:
+            kwargs["from_"] = from_
+        if key is not None:
+            kwargs["key"] = key
+        if to is not None:
+            kwargs["to"] = to
+        super().__init__(kwargs)

elasticsearch/exceptions.py

@@ -61,6 +61,7 @@ class ApiError(_ApiError):
             if self.body and isinstance(self.body, dict) and "error" in self.body:
                 if isinstance(self.body["error"], dict):
                     root_cause = self.body["error"]["root_cause"][0]
+                    caused_by = self.body["error"].get("caused_by", {})
                     cause = ", ".join(
                         filter(
                             None,
@@ -68,6 +69,7 @@ class ApiError(_ApiError):
                                 repr(root_cause["reason"]),
                                 root_cause.get("resource.id"),
                                 root_cause.get("resource.type"),
+                                caused_by.get("reason"),
                             ],
                         )
                     )