fix: 修复 Gemini 模型分页遍历并补充异常分支测试

Author: Sisyphbaous-DT-Project

astrbot/core/provider/sources/gemini_embedding_source.py

@@ -80,11 +80,10 @@ async def get_embeddings(self, text: list[str]) -> list[list[float]]:
 
     async def get_models(self) -> list[str]:
         try:
-            models = await self.client.models.list()
             all_model_ids: list[str] = []
             embedding_model_ids: list[str] = []
 
-            for model in getattr(models, "page", []):
+            async for model in await self.client.models.list():
                 model_id = self._extract_model_id(model)
                 if not model_id:
                     continue

tests/test_dashboard.py

@@ -1243,6 +1243,25 @@ async def terminate(self):
         type(self).terminate_calls += 1
 
 
+class _ErrorEmbeddingProvider(EmbeddingProvider):
+    terminate_calls = 0
+
+    async def get_embedding(self, text: str) -> list[float]:
+        return [0.1]
+
+    async def get_embeddings(self, text: list[str]) -> list[list[float]]:
+        return [[0.1] for _ in text]
+
+    def get_dim(self) -> int:
+        return 1
+
+    async def get_models(self) -> list[str]:
+        raise RuntimeError("boom")
+
+    async def terminate(self):
+        type(self).terminate_calls += 1
+
+
 @pytest.mark.asyncio
 async def test_get_embedding_models_success_and_terminate(
     app: Quart,
@@ -1308,3 +1327,37 @@ async def test_get_embedding_models_unsupported_returns_error(
     data = await response.get_json()
     assert data["status"] == "error"
     assert _UnsupportedEmbeddingProvider.terminate_calls == 1
+
+
+@pytest.mark.asyncio
+async def test_get_embedding_models_runtime_error_returns_error_and_terminate(
+    app: Quart,
+    authenticated_header: dict,
+    monkeypatch,
+):
+    from astrbot.core.provider.register import provider_cls_map
+
+    _ErrorEmbeddingProvider.terminate_calls = 0
+    monkeypatch.setitem(
+        provider_cls_map,
+        "test_embedding_runtime_error",
+        SimpleNamespace(cls_type=_ErrorEmbeddingProvider),
+    )
+
+    test_client = app.test_client()
+    response = await test_client.post(
+        "/api/config/provider/get_embedding_models",
+        headers=authenticated_header,
+        json={
+            "provider_config": {
+                "id": "test-embedding-provider",
+                "type": "test_embedding_runtime_error",
+            }
+        },
+    )
+
+    assert response.status_code == 200
+    data = await response.get_json()
+    assert data["status"] == "error"
+    assert "获取嵌入模型列表失败" in data["message"]
+    assert _ErrorEmbeddingProvider.terminate_calls == 1

tests/test_gemini_embedding_source.py

@@ -9,7 +9,14 @@
 
 class _FakeModelsPager:
     def __init__(self, models) -> None:
-        self.page = models
+        self._models = models
+
+    def __aiter__(self):
+        async def _gen():
+            for model in self._models:
+                yield model
+
+        return _gen()
 
 
 class _FakeModelsAPI: