Federated Search Addition

Author: MuddyHope

meilisearch/client.py

@@ -219,14 +219,17 @@ def index(self, uid: str) -> Index:
             return Index(self.config, uid=uid)
         raise ValueError("The index UID should not be None")
 
-    def multi_search(self, queries: Sequence[Mapping[str, Any]]) -> Dict[str, List[Dict[str, Any]]]:
+    def multi_search(self, queries: Sequence[Mapping[str, Any]], federation: Optional[Dict[str, Any]] = None) -> Dict[str, List[Dict[str, Any]]]:
         """Multi-index search.
 
         Parameters
         ----------
         queries:
             List of dictionaries containing the specified indexes and their search queries
             https://www.meilisearch.com/docs/reference/api/search#search-in-an-index
+        federation: (optional):
+            List of dictionaries containing offset and limit
+            https://www.meilisearch.com/docs/reference/api/multi_search
 
         Returns
         -------
@@ -240,7 +243,7 @@ def multi_search(self, queries: Sequence[Mapping[str, Any]]) -> Dict[str, List[D
         """
         return self.http.post(
             f"{self.config.paths.multi_search}",
-            body={"queries": queries},
+            body={"queries": queries, "federation": federation},
         )
 
     def get_all_stats(self) -> Dict[str, Any]:

tests/client/test_client_multi_search_meilisearch.py

@@ -32,3 +32,28 @@ def test_multi_search_on_no_index(client):
     """Tests multi-search on a non existing index."""
     with pytest.raises(MeilisearchApiError):
         client.multi_search([{"indexUid": "indexDoesNotExist", "q": ""}])
+
+
+def test_multi_search_with_no_value_in_federation(client, empty_index):
+    """Tests multi-search with federation, but no value"""
+    empty_index("indexA")
+    response = client.multi_search([{"indexUid": "indexA", "q": ""}], {})
+
+    assert response["hits"][0]["_federation"]['indexUid'] == "indexA" if len(response["hits"]) >= 1 else isinstance(response, dict)
+
+
+def test_multi_search_with_offset_and_limit_in_federation(client, empty_index):
+    """Tests multi-search with federation, with offset and limit value"""
+    empty_index("indexA")
+    response = client.multi_search([{"indexUid": "indexA", "q": ""}], {"offset": 2, "limit": 2})
+    
+    assert len(response["hits"]) == 2 if response['hits'] else True
+
+
+def test_multi_search_with_federation_options(client, empty_index):
+    """Tests multi-search with federation, with federation options"""
+    empty_index("indexA")
+    response = client.multi_search([{"indexUid": "indexA", "q": "", "federationOptions": {"weight": 0.99}}], {"offset": 2, "limit": 2})
+
+    assert isinstance(response["hits"], list)
+    assert response["hits"][0]["_federation"]["weightedRankingScore"] < 0.99 if len(response["hits"]) >= 1 else True