Keep compat with old IndexStats.field_distribution

Author: ellnix

meilisearch/models/index.py

@@ -1,15 +1,43 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import Dict, List, Optional, Union
+from typing import Any, Dict, Iterator, List, Optional, Union
 
 from camel_converter.pydantic_base import CamelBase
+from pydantic import ConfigDict, field_validator
+
+
+class FieldDistribution:
+    __dict: Dict
+
+    def __init__(self, dist: Dict[str, int]) -> None:
+        self.__dict = dist
+        for key in dist:
+            setattr(self, key, dist[key])
+
+    def __getattr__(self, attr: str) -> str:
+        if attr in self.__dict.keys():
+            return attr
+        raise AttributeError(f"{self.__class__.__name__} object has no attribute {attr}")
+
+    def __iter__(self) -> Iterator:
+        return iter(self.__dict__.items())
 
 
 class IndexStats(CamelBase):
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
     number_of_documents: int
     is_indexing: bool
-    field_distribution: Dict[str, int]
+    field_distribution: FieldDistribution
+
+    @field_validator("field_distribution", mode="before")
+    @classmethod
+    def build_field_distribution(cls, v: Any) -> FieldDistribution:
+        if not isinstance(v, dict):
+            raise TypeError('"field_distribution" in IndexStats must be a dict')
+
+        return FieldDistribution(v)
 
 
 class Faceting(CamelBase):

tests/index/test_index_stats_meilisearch.py

@@ -13,5 +13,5 @@ def test_get_stats_default(index_with_documents):
     response = index_with_documents().get_stats()
     assert isinstance(response, IndexStats)
     assert response.number_of_documents == 31
-    assert "genre" in response.field_distribution
-    assert response.field_distribution["genre"] == 11
+    assert hasattr(response.field_distribution, "genre")
+    assert response.field_distribution.genre == 11