List all available voices by pagination

Author: kunalk16

src/esperanto/providers/tts/mistral.py

@@ -66,20 +66,34 @@ def _get_models(self) -> List[Model]:
     def available_voices(self) -> Dict[str, Voice]:
         if self._voices_cache is not None:
             return self._voices_cache
-        items = []
-        page = 1
+
+        items: List[Dict[str, Any]] = []
+        offset = 0
+        limit = 100
         while True:
             response = self.client.get(
                 f"{self.base_url}/audio/voices",
                 headers=self._get_headers(),
-                params={"page": page},
+                params={"limit": limit, "offset": offset},
             )
             self._handle_error(response)
             body = response.json()
-            items.extend(body.get("items", []))
-            if page >= body.get("total_pages", 1):
+            page_items = body.get("items", [])
+            items.extend(page_items)
+
+            total = body.get("total")
+            if total is not None:
+                if len(items) >= total:
+                    break
+            elif len(page_items) < limit:
+                break
+
+            if not page_items:
+                # To avoid an infinite loop.
                 break
-            page += 1
+
+            offset += len(page_items)
+
         self._voices_cache = {
             item["id"]: Voice(
                 id=item["id"],

tests/providers/tts/test_mistral.py

@@ -163,7 +163,7 @@ def test_available_voices(tts_model):
     tts_model.client.get.assert_called_once_with(
         "https://api.mistral.ai/v1/audio/voices",
         headers=tts_model._get_headers(),
-        params={"page": 1},
+        params={"limit": 100, "offset": 0},
     )
 
     assert "gb_jane_neutral" in voices
@@ -180,6 +180,83 @@ def test_available_voices_cached(tts_model):
     assert tts_model.client.get.call_count == 1
 
 
+def test_available_voices_paginated(tts_model):
+    """Multi-page response: all pages' voices should be returned."""
+    page_one_items = [
+        {
+            "id": f"voice_{i}",
+            "name": f"Voice {i}",
+            "gender": "NEUTRAL",
+            "languages": ["en"],
+        }
+        for i in range(100)
+    ]
+    page_two_items = [
+        {
+            "id": "voice_100",
+            "name": "Voice 100",
+            "gender": "NEUTRAL",
+            "languages": ["en"],
+        },
+        {
+            "id": "voice_101",
+            "name": "Voice 101",
+            "gender": "NEUTRAL",
+            "languages": ["en"],
+        },
+    ]
+
+    def make_response(status_code, json_data):
+        response = Mock()
+        response.status_code = status_code
+        response.json.return_value = json_data
+        response.text = ""
+        return response
+
+    def paginated_get(url, **kwargs):
+        params = kwargs.get("params", {})
+        offset = params.get("offset", 0)
+        if offset == 0:
+            return make_response(
+                200,
+                {
+                    "items": page_one_items,
+                    "page": 1,
+                    "page_size": 100,
+                    "total": 102,
+                    "total_pages": 2,
+                },
+            )
+        if offset == 100:
+            return make_response(
+                200,
+                {
+                    "items": page_two_items,
+                    "page": 2,
+                    "page_size": 100,
+                    "total": 102,
+                    "total_pages": 2,
+                },
+            )
+        return make_response(200, {"items": [], "total": 102})
+
+    tts_model.client.get.side_effect = paginated_get
+
+    voices = tts_model.available_voices
+
+    assert len(voices) == 102
+    assert "voice_0" in voices
+    assert "voice_99" in voices
+    assert "voice_100" in voices
+    assert "voice_101" in voices
+    assert tts_model.client.get.call_count == 2
+
+    first_call = tts_model.client.get.call_args_list[0]
+    second_call = tts_model.client.get.call_args_list[1]
+    assert first_call.kwargs["params"] == {"limit": 100, "offset": 0}
+    assert second_call.kwargs["params"] == {"limit": 100, "offset": 100}
+
+
 def test_error_handling_4xx(tts_model):
     tts_model.client.post.side_effect = None
     error_response = Mock()