typing of the meta attribute of hits

Author: miguelgrinberg

elasticsearch_dsl/utils.py

@@ -48,6 +48,7 @@
     from .field import Field
     from .index_base import IndexBase
     from .response import Hit  # noqa: F401
+    from .types import Hit as HitBaseType
 
 UsingType: TypeAlias = Union[str, "Elasticsearch"]
 AsyncUsingType: TypeAlias = Union[str, "AsyncElasticsearch"]
@@ -468,7 +469,13 @@ def _clone(self) -> Self:
         return c
 
 
-class HitMeta(AttrDict[Any]):
+if TYPE_CHECKING:
+    HitMetaBase = HitBaseType
+else:
+    HitMetaBase = AttrDict[Any]
+
+
+class HitMeta(HitMetaBase):
     def __init__(
         self,
         document: Dict[str, Any],

examples/async/parent_child.py

@@ -190,7 +190,7 @@ async def get_answers(self) -> List[Any]:
         elasticsearch.
         """
         if "inner_hits" in self.meta and "answer" in self.meta.inner_hits:
-            return cast(List[Any], self.meta.inner_hits.answer.hits)
+            return cast(List[Any], self.meta.inner_hits["answer"].hits)
         return [a async for a in self.search_answers()]
 
     async def save(self, **kwargs: Any) -> None:  # type: ignore[override]

examples/async/sparse_vectors.py

@@ -186,7 +186,7 @@ async def main() -> None:
         )
         print(f"Summary: {hit.summary}")
         if args.show_inner_hits:
-            for passage in hit.meta.inner_hits.passages:
+            for passage in hit.meta.inner_hits["passages"]:
                 print(f"  - [Score: {passage.meta.score}] {passage.content!r}")
         print("")
 

examples/async/vectors.py

@@ -175,7 +175,7 @@ async def main() -> None:
         )
         print(f"Summary: {hit.summary}")
         if args.show_inner_hits:
-            for passage in hit.meta.inner_hits.passages:
+            for passage in hit.meta.inner_hits["passages"]:
                 print(f"  - [Score: {passage.meta.score}] {passage.content!r}")
         print("")
 

examples/parent_child.py

@@ -189,7 +189,7 @@ def get_answers(self) -> List[Any]:
         elasticsearch.
         """
         if "inner_hits" in self.meta and "answer" in self.meta.inner_hits:
-            return cast(List[Any], self.meta.inner_hits.answer.hits)
+            return cast(List[Any], self.meta.inner_hits["answer"].hits)
         return [a for a in self.search_answers()]
 
     def save(self, **kwargs: Any) -> None:  # type: ignore[override]

examples/sparse_vectors.py

@@ -185,7 +185,7 @@ def main() -> None:
         )
         print(f"Summary: {hit.summary}")
         if args.show_inner_hits:
-            for passage in hit.meta.inner_hits.passages:
+            for passage in hit.meta.inner_hits["passages"]:
                 print(f"  - [Score: {passage.meta.score}] {passage.content!r}")
         print("")
 

examples/vectors.py

@@ -174,7 +174,7 @@ def main() -> None:
         )
         print(f"Summary: {hit.summary}")
         if args.show_inner_hits:
-            for passage in hit.meta.inner_hits.passages:
+            for passage in hit.meta.inner_hits["passages"]:
                 print(f"  - [Score: {passage.meta.score}] {passage.content!r}")
         print("")