feat: Added sharding support for python sdk

Author: jawad-khan

meilisearch/models/task.py

@@ -21,6 +21,7 @@ class Task(CamelBase):
     enqueued_at: datetime
     started_at: Optional[datetime] = None
     finished_at: Optional[datetime] = None
+    network: Optional[Dict[str, Any]] = None
 
     if is_pydantic_2():
 

tests/client/test_client_multi_search_meilisearch.py

@@ -84,14 +84,17 @@ def test_multi_search_with_network(client, index_with_documents):
     resp = client.add_or_update_networks(
         {
             "self": REMOTE_MS_1,
+            "sharding": True,
             "remotes": {
                 REMOTE_MS_1: {
                     "url": "http://ms-1235.example.meilisearch.io",
                     "searchApiKey": "xxxxxxxx",
+                    "writeApiKey": "xxxxxxxx",
                 },
                 REMOTE_MS_2: {
                     "url": "http://ms-1255.example.meilisearch.io",
                     "searchApiKey": "xxxxxxxx",
+                    "writeApiKey": "xxxxxxxx",
                 },
             },
         }

tests/client/test_client_network.py

@@ -16,15 +16,25 @@ def test_add_or_update_networks(client):
     """Tests upsert network remote instance."""
     body = {
         "self": REMOTE_MS_1,
+        "sharding": True,
         "remotes": {
-            REMOTE_MS_1: {"url": "http://localhost:7700", "searchApiKey": "xxxxxxxxxxxxxx"},
-            REMOTE_MS_2: {"url": "http://localhost:7720", "searchApiKey": "xxxxxxxxxxxxxxx"},
+            REMOTE_MS_1: {
+                "url": "http://localhost:7700",
+                "searchApiKey": "xxxxxxxxxxxxxx",
+                "writeApiKey": "xxxxxxxxx",
+            },
+            REMOTE_MS_2: {
+                "url": "http://localhost:7720",
+                "searchApiKey": "xxxxxxxxxxxxxxx",
+                "writeApiKey": "xxxxxxxx",
+            },
         },
     }
     response = client.add_or_update_networks(body=body)
 
     assert isinstance(response, dict)
     assert response["self"] == REMOTE_MS_1
+    assert response["sharding"] is True
     assert len(response["remotes"]) >= 2
     assert REMOTE_MS_2 in response["remotes"]
     assert REMOTE_MS_1 in response["remotes"]

tests/client/test_client_sharding.py

@@ -0,0 +1,35 @@
+import pytest
+
+from tests.common import BASE_URL, REMOTE_MS_1
+
+
+@pytest.mark.usefixtures("enable_network_options")
+def test_update_and_get_network_settings(client):
+    """Test updating and getting network settings."""
+    instance_name = REMOTE_MS_1
+    options = {
+        "self": instance_name,
+        "remotes": {
+            instance_name: {
+                "url": BASE_URL,
+                "searchApiKey": "search-key-1",
+                "writeApiKey": "write-key-1",
+            }
+        },
+        "sharding": True,
+    }
+
+    client.add_or_update_networks(options)
+    response = client.get_all_networks()
+
+    assert response["self"] == options["self"]
+    assert response["remotes"][instance_name]["url"] == options["remotes"][instance_name]["url"]
+    assert (
+        response["remotes"][instance_name]["searchApiKey"]
+        == options["remotes"][instance_name]["searchApiKey"]
+    )
+    assert (
+        response["remotes"][instance_name]["writeApiKey"]
+        == options["remotes"][instance_name]["writeApiKey"]
+    )
+    assert response["sharding"] == options["sharding"]