Add proximityPrecision setting

Author: serendipty01

.code-samples.meilisearch.yaml

@@ -714,3 +714,9 @@ update_search_cutoff_1: |-
   client.index('movies').update_search_cutoff_ms(150)
 reset_search_cutoff_1: |-
   client.index('movies').reset_search_cutoff_ms()
+get_proximity_precision_settings_1: |-
+  client.index('books').get_proximity_precision()
+update_proximity_precision_settings_1: |-
+  client.index('books').update_proximity_precision(ProximityPrecision.BY_ATTRIBUTE)
+reset_proximity_precision_settings_1: |-
+  client.index('books').reset_proximity_precision()

meilisearch/_httprequests.py

@@ -12,6 +12,7 @@
     MeilisearchCommunicationError,
     MeilisearchTimeoutError,
 )
+from meilisearch.models.index import ProximityPrecision
 from meilisearch.version import qualified_version
 
 
@@ -28,7 +29,14 @@ def send_request(
         http_method: Callable,
         path: str,
         body: Optional[
-            Union[Mapping[str, Any], Sequence[Mapping[str, Any]], List[str], str, int]
+            Union[
+                Mapping[str, Any],
+                Sequence[Mapping[str, Any]],
+                List[str],
+                str,
+                int,
+                ProximityPrecision,
+            ]
         ] = None,
         content_type: Optional[str] = None,
     ) -> Any:
@@ -90,7 +98,14 @@ def put(
         self,
         path: str,
         body: Optional[
-            Union[Mapping[str, Any], Sequence[Mapping[str, Any]], List[str], str, int]
+            Union[
+                Mapping[str, Any],
+                Sequence[Mapping[str, Any]],
+                List[str],
+                str,
+                int,
+                ProximityPrecision,
+            ]
         ] = None,
         content_type: Optional[str] = "application/json",
     ) -> Any:

meilisearch/config.py

@@ -40,6 +40,7 @@ class Paths:
         swap = "swap-indexes"
         embedders = "embedders"
         search_cutoff_ms = "search-cutoff-ms"
+        proximity_precision = "proximity-precision"
 
     def __init__(
         self,

meilisearch/index.py

@@ -17,6 +17,7 @@
     IndexStats,
     OpenAiEmbedder,
     Pagination,
+    ProximityPrecision,
     TypoTolerance,
     UserProvidedEmbedder,
 )
@@ -1917,6 +1918,68 @@ def reset_search_cutoff_ms(self) -> TaskInfo:
 
         return TaskInfo(**task)
 
+    # PROXIMITY PRECISION SETTINGS
+
+    def get_proximity_precision(self) -> ProximityPrecision | None:
+        """Get the proximity_precision of the index.
+
+        Returns
+        -------
+        settings:
+            proximity_precision of the index.
+
+        Raises
+        ------
+        MeilisearchApiError
+            An error containing details about why Meilisearch can't process your request. Meilisearch error codes are described here: https://www.meilisearch.com/docs/reference/errors/error_codes#meilisearch-errors
+        """
+        return self.http.get(self.__settings_url_for(self.config.paths.proximity_precision))
+
+    def update_proximity_precision(self, body: ProximityPrecision) -> TaskInfo:
+        """Update the proximity_precision of the index.
+
+        Parameters
+        ----------
+        body:
+            proximity_precision
+
+        Returns
+        -------
+        task_info:
+            TaskInfo instance containing information about a task to track the progress of an asynchronous process.
+            https://www.meilisearch.com/docs/reference/api/tasks#get-one-task
+
+        Raises
+        ------
+        MeilisearchApiError
+            An error containing details about why Meilisearch can't process your request. Meilisearch error codes are described here: https://www.meilisearch.com/docs/reference/errors/error_codes#meilisearch-errors
+        """
+        task = self.http.put(
+            self.__settings_url_for(self.config.paths.proximity_precision), body.value
+        )
+
+        return TaskInfo(**task)
+
+    def reset_proximity_precision(self) -> TaskInfo:
+        """Reset the proximity_precision of the index
+
+        Returns
+        -------
+        task_info:
+            TaskInfo instance containing information about a task to track the progress of an asynchronous process.
+            https://www.meilisearch.com/docs/reference/api/tasks#get-one-task
+
+        Raises
+        ------
+        MeilisearchApiError
+            An error containing details about why Meilisearch can't process your request. Meilisearch error codes are described here: https://www.meilisearch.com/docs/reference/errors/error_codes#meilisearch-errors
+        """
+        task = self.http.delete(
+            self.__settings_url_for(self.config.paths.proximity_precision),
+        )
+
+        return TaskInfo(**task)
+
     @staticmethod
     def _batch(
         documents: Sequence[Mapping[str, Any]], batch_size: int

meilisearch/models/index.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional, Union
 
 from camel_converter import to_snake
@@ -48,6 +49,11 @@ class TypoTolerance(CamelBase):
     min_word_size_for_typos: Optional[MinWordSizeForTypos] = None
 
 
+class ProximityPrecision(Enum):
+    BY_WORD = "byWord"
+    BY_ATTRIBUTE = "byAttribute"
+
+
 class OpenAiEmbedder(CamelBase):
     source: str = "openAi"
     model: Optional[str] = None  # Defaults to text-embedding-ada-002

tests/settings/test_settings_proximity_precision_meilisearch.py

@@ -0,0 +1,37 @@
+from meilisearch.models.index import ProximityPrecision
+
+NEW_PROXIMITY_PRECISION = ProximityPrecision.BY_ATTRIBUTE
+
+
+def test_get_proximity_precision(empty_index):
+    """Tests getting default proximity precision."""
+    response = empty_index().get_proximity_precision()
+    assert response == ProximityPrecision.BY_WORD.value
+
+
+def test_update_proximity_precision(empty_index):
+    """Tests updating proximity precision."""
+    index = empty_index()
+    response = index.update_proximity_precision(NEW_PROXIMITY_PRECISION)
+    update = index.wait_for_task(response.task_uid)
+    assert update.status == "succeeded"
+    response = index.get_proximity_precision()
+    assert NEW_PROXIMITY_PRECISION.value == response
+
+
+def test_reset_proximity_precision(empty_index):
+    """Tests resetting the proximity precision to its default value."""
+    index = empty_index()
+    # Update the settings first
+    response = index.update_proximity_precision(NEW_PROXIMITY_PRECISION)
+    update = index.wait_for_task(response.task_uid)
+    assert update.status == "succeeded"
+    # Check the settings have been correctly updated
+    response = index.get_proximity_precision()
+    assert NEW_PROXIMITY_PRECISION.value == response
+    # Check the reset of the settings
+    response = index.reset_proximity_precision()
+    update = index.wait_for_task(response.task_uid)
+    assert update.status == "succeeded"
+    response = index.get_proximity_precision()
+    assert response == ProximityPrecision.BY_WORD.value