From 8568f3eab374a7dd13c05d4bf37a624a4b9a3a12 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=E2=82=82=E2=82=82H=E2=82=82=E2=82=85NO=E2=82=86?=
 <96930391+Sisyphbaous-DT-Project@users.noreply.github.com>
Date: Thu, 23 Apr 2026 10:51:20 +0800
Subject: [PATCH 1/4] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81=20OpenAI=20?=
 =?UTF-8?q?=E5=85=BC=E5=AE=B9=E5=B5=8C=E5=85=A5=E6=A8=A1=E5=9E=8B=E5=88=97?=
 =?UTF-8?q?=E8=A1=A8=E5=8F=91=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 astrbot/core/config/default.py                |   1 +
 astrbot/core/provider/provider.py             |   6 +
 .../sources/openai_embedding_source.py        |  32 ++++++
 astrbot/dashboard/routes/config.py            |  79 ++++++++++++++
 .../src/components/shared/AstrBotConfig.vue   |  43 +++++++-
 .../components/shared/ConfigItemRenderer.vue  |  39 ++++++-
 tests/test_dashboard.py                       | 103 ++++++++++++++++++
 tests/test_openai_embedding_source.py         |  64 +++++++++++
 8 files changed, 365 insertions(+), 2 deletions(-)
 create mode 100644 tests/test_openai_embedding_source.py

diff --git a/astrbot/core/config/default.py b/astrbot/core/config/default.py
index 09976f7c41..eac62984d9 100644
--- a/astrbot/core/config/default.py
+++ b/astrbot/core/config/default.py
@@ -2141,6 +2141,7 @@ class ChatProviderTemplate(TypedDict):
                         "description": "嵌入模型",
                         "type": "string",
                         "hint": "嵌入模型名称。",
+                        "_special": "get_embedding_models",
                     },
                     "embedding_api_key": {
                         "description": "API Key",
diff --git a/astrbot/core/provider/provider.py b/astrbot/core/provider/provider.py
index efe9e2e47e..13b4b19608 100644
--- a/astrbot/core/provider/provider.py
+++ b/astrbot/core/provider/provider.py
@@ -334,6 +334,12 @@ def get_dim(self) -> int:
         """获取向量的维度"""
         ...
 
+    async def get_models(self) -> list[str]:
+        """Optional model discovery for embedding providers."""
+        raise NotImplementedError(
+            "This embedding provider does not support model discovery",
+        )
+
     async def test(self) -> None:
         await self.get_embedding("astrbot")
 
diff --git a/astrbot/core/provider/sources/openai_embedding_source.py b/astrbot/core/provider/sources/openai_embedding_source.py
index ae531996ae..3b4d314fd3 100644
--- a/astrbot/core/provider/sources/openai_embedding_source.py
+++ b/astrbot/core/provider/sources/openai_embedding_source.py
@@ -14,6 +14,15 @@
     provider_type=ProviderType.EMBEDDING,
 )
 class OpenAIEmbeddingProvider(EmbeddingProvider):
+    _EMBEDDING_MODEL_HINTS = (
+        "embedding",
+        "bge",
+        "gte",
+        "e5",
+        "m3e",
+        "multilingual-e5",
+    )
+
     def __init__(self, provider_config: dict, provider_settings: dict) -> None:
         super().__init__(provider_config, provider_settings)
         self.provider_config = provider_config
@@ -62,6 +71,29 @@ async def get_embeddings(self, text: list[str]) -> list[list[float]]:
         )
         return [item.embedding for item in embeddings.data]
 
+    async def get_models(self) -> list[str]:
+        try:
+            models = await self.client.models.list()
+            all_model_ids = sorted(
+                {model.id for model in models.data if getattr(model, "id", None)}
+            )
+
+            embedding_model_ids = [
+                model_id
+                for model_id in all_model_ids
+                if self._looks_like_embedding_model(model_id)
+            ]
+
+            # Fall back to all model ids when no embedding-like names are detected.
+            return embedding_model_ids or all_model_ids
+        except Exception as e:
+            raise Exception(f"获取嵌入模型列表失败: {e!s}") from e
+
+    @classmethod
+    def _looks_like_embedding_model(cls, model_id: str) -> bool:
+        normalized = model_id.lower()
+        return any(hint in normalized for hint in cls._EMBEDDING_MODEL_HINTS)
+
     def _embedding_kwargs(self) -> dict:
         """构建嵌入请求的可选参数"""
         kwargs = {}
diff --git a/astrbot/dashboard/routes/config.py b/astrbot/dashboard/routes/config.py
index bcd7e075c7..b9bd8c92c5 100644
--- a/astrbot/dashboard/routes/config.py
+++ b/astrbot/dashboard/routes/config.py
@@ -372,6 +372,10 @@ def __init__(
             "/config/provider/list": ("GET", self.get_provider_config_list),
             "/config/provider/model_list": ("GET", self.get_provider_model_list),
             "/config/provider/get_embedding_dim": ("POST", self.get_embedding_dim),
+            "/config/provider/get_embedding_models": (
+                "POST",
+                self.get_embedding_models,
+            ),
             "/config/provider_sources/models": (
                 "GET",
                 self.get_provider_source_models,
@@ -844,6 +848,7 @@ async def get_embedding_dim(self):
         if not provider_config:
             return Response().error("缺少参数 provider_config").__dict__
 
+        inst = None
         try:
             # 动态导入 EmbeddingProvider
             from astrbot.core.provider.provider import EmbeddingProvider
@@ -907,6 +912,80 @@ async def get_embedding_dim(self):
         except Exception as e:
             logger.error(traceback.format_exc())
             return Response().error(f"获取嵌入维度失败: {e!s}").__dict__
+        finally:
+            terminate_fn = getattr(inst, "terminate", None) if inst else None
+            if inspect.iscoroutinefunction(terminate_fn):
+                try:
+                    await terminate_fn()
+                except Exception:
+                    logger.warning("释放嵌入 provider 资源失败")
+
+    async def get_embedding_models(self):
+        """根据临时 provider_config 获取可用嵌入模型列表"""
+        post_data = await request.json
+        provider_config = post_data.get("provider_config", None)
+        if not provider_config:
+            return Response().error("缺少参数 provider_config").__dict__
+
+        inst = None
+        try:
+            from astrbot.core.provider.provider import EmbeddingProvider
+            from astrbot.core.provider.register import provider_cls_map
+
+            provider_type = provider_config.get("type", None)
+            if not provider_type:
+                return Response().error("provider_config 缺少 type 字段").__dict__
+
+            if provider_type not in provider_cls_map:
+                try:
+                    self.core_lifecycle.provider_manager.dynamic_import_provider(
+                        provider_type,
+                    )
+                except ImportError:
+                    logger.error(traceback.format_exc())
+                    return Response().error("提供商适配器加载失败").__dict__
+
+            if provider_type not in provider_cls_map:
+                return (
+                    Response()
+                    .error(f"未找到适用于 {provider_type} 的提供商适配器")
+                    .__dict__
+                )
+
+            provider_metadata = provider_cls_map[provider_type]
+            cls_type = provider_metadata.cls_type
+            if not cls_type:
+                return Response().error(f"无法找到 {provider_type} 的类").__dict__
+
+            inst = cls_type(provider_config, {})
+            if not isinstance(inst, EmbeddingProvider):
+                return Response().error("提供商不是 EmbeddingProvider 类型").__dict__
+
+            init_fn = getattr(inst, "initialize", None)
+            if inspect.iscoroutinefunction(init_fn):
+                await init_fn()
+
+            try:
+                models = await inst.get_models()
+            except NotImplementedError:
+                return (
+                    Response()
+                    .error("当前提供商暂不支持自动获取模型列表，请手动填写模型 ID")
+                    .__dict__
+                )
+
+            models = sorted(dict.fromkeys(models or []))
+            return Response().ok({"models": models}).__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(f"获取嵌入模型列表失败: {e!s}").__dict__
+        finally:
+            terminate_fn = getattr(inst, "terminate", None) if inst else None
+            if inspect.iscoroutinefunction(terminate_fn):
+                try:
+                    await terminate_fn()
+                except Exception:
+                    logger.warning("释放嵌入 provider 资源失败")
 
     async def get_provider_source_models(self):
         """获取指定 provider_source 支持的模型列表
diff --git a/dashboard/src/components/shared/AstrBotConfig.vue b/dashboard/src/components/shared/AstrBotConfig.vue
index 33273a36c9..7768e3a268 100644
--- a/dashboard/src/components/shared/AstrBotConfig.vue
+++ b/dashboard/src/components/shared/AstrBotConfig.vue
@@ -1,6 +1,6 @@
 <script setup>
 import { VueMonacoEditor } from '@guolao/vue-monaco-editor'
-import { ref, computed } from 'vue'
+import { ref, computed, watch } from 'vue'
 import ConfigItemRenderer from './ConfigItemRenderer.vue'
 import TemplateListEditor from './TemplateListEditor.vue'
 import { useI18n, useModuleI18n } from '@/i18n/composables'
@@ -88,6 +88,8 @@ const currentEditingLanguage = ref('json')
 const currentEditingTheme = ref('vs-light')
 let currentEditingKeyIterable = null
 const loadingEmbeddingDim = ref(false)
+const loadingEmbeddingModels = ref(false)
+const availableEmbeddingModels = ref([])
 
 function openEditorDialog(key, value, theme, language) {
   currentEditingKey.value = key
@@ -125,6 +127,39 @@ async function getEmbeddingDimensions(providerConfig) {
   }
 }
 
+async function getEmbeddingModels(providerConfig) {
+  if (loadingEmbeddingModels.value) return
+
+  loadingEmbeddingModels.value = true
+  try {
+    const response = await axios.post('/api/config/provider/get_embedding_models', {
+      provider_config: providerConfig
+    })
+
+    if (response.data.status !== 'error' && Array.isArray(response.data.data?.models)) {
+      availableEmbeddingModels.value = response.data.data.models
+      useToast().success(`Fetched: ${response.data.data.models.length}`)
+    } else {
+      useToast().error(response.data.message)
+    }
+  } catch (error) {
+    console.error('Error getting embedding models:', error)
+  } finally {
+    loadingEmbeddingModels.value = false
+  }
+}
+
+watch(
+  () => [
+    props.iterable?.type,
+    props.iterable?.embedding_api_key,
+    props.iterable?.embedding_api_base
+  ],
+  () => {
+    availableEmbeddingModels.value = []
+  }
+)
+
 function getValueBySelector(obj, selector) {
   const keys = selector.split('.')
   let current = obj
@@ -266,8 +301,11 @@ function hasVisibleItemsAfter(items, currentIndex) {
                 :plugin-name="pluginName"
                 :config-key="getItemPath(key)"
                 :loading="loadingEmbeddingDim"
+                :loading-models="loadingEmbeddingModels"
+                :available-models="availableEmbeddingModels"
                 :show-fullscreen-btn="!!metadata[metadataKey].items[key]?.editor_mode"
                 @get-embedding-dim="getEmbeddingDimensions(iterable)"
+                @get-embedding-models="getEmbeddingModels(iterable)"
                 @open-fullscreen="openEditorDialog(key, iterable, metadata[metadataKey].items[key]?.editor_theme, metadata[metadataKey].items[key]?.editor_language)"
               />
             </v-col>
@@ -311,6 +349,9 @@ function hasVisibleItemsAfter(items, currentIndex) {
             :item-meta="metadata[metadataKey]"
             :plugin-name="pluginName"
             :config-key="getItemPath(metadataKey)"
+            :loading-models="loadingEmbeddingModels"
+            :available-models="availableEmbeddingModels"
+            @get-embedding-models="getEmbeddingModels(iterable)"
           />
         </v-col>
       </v-row>
diff --git a/dashboard/src/components/shared/ConfigItemRenderer.vue b/dashboard/src/components/shared/ConfigItemRenderer.vue
index 58e41f3ab7..465dee6ca0 100644
--- a/dashboard/src/components/shared/ConfigItemRenderer.vue
+++ b/dashboard/src/components/shared/ConfigItemRenderer.vue
@@ -68,6 +68,30 @@
         </v-btn>
       </div>
     </template>
+    <template v-else-if="itemMeta?._special === 'get_embedding_models'">
+      <div class="d-flex align-center gap-2">
+        <v-combobox
+          :model-value="modelValue"
+          @update:model-value="emitUpdate"
+          :items="availableModels"
+          density="compact"
+          variant="outlined"
+          class="config-field"
+          hide-details
+          clearable
+        ></v-combobox>
+        <v-btn
+          color="primary"
+          variant="tonal"
+          size="small"
+          @click="$emit('get-embedding-models')"
+          :loading="loadingModels"
+          class="ml-2"
+        >
+          {{ t('core.common.autoDetect') }}
+        </v-btn>
+      </div>
+    </template>
 
     <div
       v-else-if="itemMeta?.type === 'list' && itemMeta?.options && itemMeta?.render_type === 'checkbox'"
@@ -264,13 +288,26 @@ const props = defineProps({
     type: Boolean,
     default: false
   },
+  loadingModels: {
+    type: Boolean,
+    default: false
+  },
+  availableModels: {
+    type: Array,
+    default: () => []
+  },
   showFullscreenBtn: {
     type: Boolean,
     default: false
   }
 })
 
-const emit = defineEmits(['update:modelValue', 'get-embedding-dim', 'open-fullscreen'])
+const emit = defineEmits([
+  'update:modelValue',
+  'get-embedding-dim',
+  'get-embedding-models',
+  'open-fullscreen'
+])
 const { t } = useI18n()
 const { getRaw } = useModuleI18n('features/config-metadata')
 
diff --git a/tests/test_dashboard.py b/tests/test_dashboard.py
index 6dc352057c..c4233bd83f 100644
--- a/tests/test_dashboard.py
+++ b/tests/test_dashboard.py
@@ -16,6 +16,7 @@
 from astrbot.core import LogBroker
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
 from astrbot.core.db.sqlite import SQLiteDatabase
+from astrbot.core.provider.provider import EmbeddingProvider
 from astrbot.core.star.star import star_registry
 from astrbot.core.star.star_handler import star_handlers_registry
 from astrbot.core.utils.pip_installer import PipInstallError
@@ -1205,3 +1206,105 @@ def _fake_install_skill_from_zip(
     assert data["data"]["failed"] == [
         {"filename": "bad_skill.zip", "error": "install failed"}
     ]
+
+
+class _DiscoverableEmbeddingProvider(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]:
+        return ["embedding-b", "embedding-a", "embedding-a"]
+
+    async def terminate(self):
+        type(self).terminate_calls += 1
+
+
+class _UnsupportedEmbeddingProvider(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 terminate(self):
+        type(self).terminate_calls += 1
+
+
+@pytest.mark.asyncio
+async def test_get_embedding_models_success_and_terminate(
+    app: Quart,
+    authenticated_header: dict,
+    monkeypatch,
+):
+    from astrbot.core.provider.register import provider_cls_map
+
+    _DiscoverableEmbeddingProvider.terminate_calls = 0
+    monkeypatch.setitem(
+        provider_cls_map,
+        "test_embedding_discovery",
+        SimpleNamespace(cls_type=_DiscoverableEmbeddingProvider),
+    )
+
+    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_discovery",
+            }
+        },
+    )
+
+    assert response.status_code == 200
+    data = await response.get_json()
+    assert data["status"] == "ok"
+    assert data["data"]["models"] == ["embedding-a", "embedding-b"]
+    assert _DiscoverableEmbeddingProvider.terminate_calls == 1
+
+
+@pytest.mark.asyncio
+async def test_get_embedding_models_unsupported_returns_error(
+    app: Quart,
+    authenticated_header: dict,
+    monkeypatch,
+):
+    from astrbot.core.provider.register import provider_cls_map
+
+    _UnsupportedEmbeddingProvider.terminate_calls = 0
+    monkeypatch.setitem(
+        provider_cls_map,
+        "test_embedding_unsupported",
+        SimpleNamespace(cls_type=_UnsupportedEmbeddingProvider),
+    )
+
+    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_unsupported",
+            }
+        },
+    )
+
+    assert response.status_code == 200
+    data = await response.get_json()
+    assert data["status"] == "error"
+    assert _UnsupportedEmbeddingProvider.terminate_calls == 1
diff --git a/tests/test_openai_embedding_source.py b/tests/test_openai_embedding_source.py
new file mode 100644
index 0000000000..b3068a8426
--- /dev/null
+++ b/tests/test_openai_embedding_source.py
@@ -0,0 +1,64 @@
+from types import SimpleNamespace
+
+import pytest
+
+from astrbot.core.provider.sources.openai_embedding_source import (
+    OpenAIEmbeddingProvider,
+)
+
+
+class _FakeModelsAPI:
+    def __init__(self, model_ids: list[str]) -> None:
+        self._model_ids = model_ids
+
+    async def list(self):
+        return SimpleNamespace(
+            data=[SimpleNamespace(id=model_id) for model_id in self._model_ids]
+        )
+
+
+class _FakeClient:
+    def __init__(self, model_ids: list[str]) -> None:
+        self.models = _FakeModelsAPI(model_ids)
+        self.closed = False
+
+    async def close(self):
+        self.closed = True
+
+
+def _make_provider() -> OpenAIEmbeddingProvider:
+    provider_config = {
+        "id": "test-openai-embedding",
+        "type": "openai_embedding",
+        "embedding_api_key": "test-key",
+        "embedding_api_base": "https://api.openai.com/v1",
+        "embedding_model": "text-embedding-3-small",
+    }
+    return OpenAIEmbeddingProvider(
+        provider_config=provider_config,
+        provider_settings={},
+    )
+
+
+@pytest.mark.asyncio
+async def test_openai_embedding_get_models_filters_embedding_like_ids():
+    provider = _make_provider()
+    try:
+        provider.client = _FakeClient(
+            ["gpt-4o-mini", "text-embedding-3-small", "BAAI/bge-m3"]
+        )
+        models = await provider.get_models()
+        assert models == ["BAAI/bge-m3", "text-embedding-3-small"]
+    finally:
+        await provider.terminate()
+
+
+@pytest.mark.asyncio
+async def test_openai_embedding_get_models_falls_back_to_all_when_no_match():
+    provider = _make_provider()
+    try:
+        provider.client = _FakeClient(["chat-model-a", "vision-v1", "chat-model-a"])
+        models = await provider.get_models()
+        assert models == ["chat-model-a", "vision-v1"]
+    finally:
+        await provider.terminate()

From 647c7987b0e9b6cdd01d2ae693b69c5d76b24a3d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=E2=82=82=E2=82=82H=E2=82=82=E2=82=85NO=E2=82=86?=
 <96930391+Sisyphbaous-DT-Project@users.noreply.github.com>
Date: Thu, 23 Apr 2026 11:01:12 +0800
Subject: [PATCH 2/4] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81=20Gemini=20?=
 =?UTF-8?q?=E5=B5=8C=E5=85=A5=E6=A8=A1=E5=9E=8B=E5=88=97=E8=A1=A8=E5=8F=91?=
 =?UTF-8?q?=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../sources/gemini_embedding_source.py        |  50 ++++++++-
 tests/test_gemini_embedding_source.py         | 103 ++++++++++++++++++
 2 files changed, 152 insertions(+), 1 deletion(-)
 create mode 100644 tests/test_gemini_embedding_source.py

diff --git a/astrbot/core/provider/sources/gemini_embedding_source.py b/astrbot/core/provider/sources/gemini_embedding_source.py
index 61ba9cadbe..9cc62d5572 100644
--- a/astrbot/core/provider/sources/gemini_embedding_source.py
+++ b/astrbot/core/provider/sources/gemini_embedding_source.py
@@ -1,4 +1,4 @@
-from typing import cast
+from typing import Any, cast
 
 from google import genai
 from google.genai import types
@@ -78,6 +78,27 @@ async def get_embeddings(self, text: list[str]) -> list[list[float]]:
         except APIError as e:
             raise Exception(f"Gemini Embedding API批量请求失败: {e.message}")
 
+    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", []):
+                model_id = self._extract_model_id(model)
+                if not model_id:
+                    continue
+                all_model_ids.append(model_id)
+                if self._supports_embedding(model, model_id):
+                    embedding_model_ids.append(model_id)
+
+            all_model_ids = sorted(dict.fromkeys(all_model_ids))
+            embedding_model_ids = sorted(dict.fromkeys(embedding_model_ids))
+
+            return embedding_model_ids or all_model_ids
+        except Exception as e:
+            raise Exception(f"获取 Gemini 嵌入模型列表失败: {e!s}") from e
+
     def get_dim(self) -> int:
         """获取向量的维度"""
         return int(self.provider_config.get("embedding_dimensions", 768))
@@ -85,3 +106,30 @@ def get_dim(self) -> int:
     async def terminate(self):
         if self.client:
             await self.client.aclose()
+
+    @staticmethod
+    def _extract_model_id(model: Any) -> str:
+        model_name = getattr(model, "name", "") or getattr(model, "model", "")
+        if not model_name:
+            return ""
+        return str(model_name).removeprefix("models/")
+
+    @classmethod
+    def _supports_embedding(cls, model: Any, model_id: str) -> bool:
+        supported_actions = getattr(model, "supported_actions", None) or getattr(
+            model, "supported_generation_methods", []
+        )
+        if isinstance(supported_actions, list):
+            normalized_actions = {
+                str(action).lower().replace("_", "").replace("-", "")
+                for action in supported_actions
+            }
+            if "embedcontent" in normalized_actions:
+                return True
+
+        return cls._looks_like_embedding_model(model_id)
+
+    @staticmethod
+    def _looks_like_embedding_model(model_id: str) -> bool:
+        normalized_model_id = model_id.lower()
+        return "embedding" in normalized_model_id or "embed" in normalized_model_id
diff --git a/tests/test_gemini_embedding_source.py b/tests/test_gemini_embedding_source.py
new file mode 100644
index 0000000000..8ad4d1be6e
--- /dev/null
+++ b/tests/test_gemini_embedding_source.py
@@ -0,0 +1,103 @@
+from types import SimpleNamespace
+
+import pytest
+
+from astrbot.core.provider.sources.gemini_embedding_source import (
+    GeminiEmbeddingProvider,
+)
+
+
+class _FakeModelsPager:
+    def __init__(self, models) -> None:
+        self.page = models
+
+
+class _FakeModelsAPI:
+    def __init__(self, models) -> None:
+        self._models = models
+
+    async def list(self):
+        return _FakeModelsPager(self._models)
+
+
+class _FakeClient:
+    def __init__(self, models) -> None:
+        self.models = _FakeModelsAPI(models)
+        self.closed = False
+
+    async def aclose(self):
+        self.closed = True
+
+
+def _make_provider() -> GeminiEmbeddingProvider:
+    provider_config = {
+        "id": "test-gemini-embedding",
+        "type": "gemini_embedding",
+        "embedding_api_key": "test-key",
+        "embedding_api_base": "https://generativelanguage.googleapis.com",
+        "embedding_model": "gemini-embedding-exp-03-07",
+    }
+    return GeminiEmbeddingProvider(
+        provider_config=provider_config,
+        provider_settings={},
+    )
+
+
+@pytest.mark.asyncio
+async def test_gemini_embedding_get_models_prefers_embedcontent_capability():
+    provider = _make_provider()
+    try:
+        provider.client = _FakeClient(
+            [
+                SimpleNamespace(
+                    name="models/gemini-2.5-flash",
+                    supported_actions=["generateContent"],
+                ),
+                SimpleNamespace(
+                    name="models/gemini-embedding-001",
+                    supported_actions=["embedContent"],
+                ),
+                SimpleNamespace(
+                    name="models/text-embedding-preview",
+                    supported_generation_methods=["embedContent"],
+                ),
+            ]
+        )
+        models = await provider.get_models()
+        assert models == ["gemini-embedding-001", "text-embedding-preview"]
+    finally:
+        await provider.terminate()
+
+
+@pytest.mark.asyncio
+async def test_gemini_embedding_get_models_falls_back_to_name_matching():
+    provider = _make_provider()
+    try:
+        provider.client = _FakeClient(
+            [
+                SimpleNamespace(name="models/chat-pro"),
+                SimpleNamespace(name="models/embed-lite"),
+                SimpleNamespace(name="models/text-embedding-preview"),
+            ]
+        )
+        models = await provider.get_models()
+        assert models == ["embed-lite", "text-embedding-preview"]
+    finally:
+        await provider.terminate()
+
+
+@pytest.mark.asyncio
+async def test_gemini_embedding_get_models_falls_back_to_all_when_no_match():
+    provider = _make_provider()
+    try:
+        provider.client = _FakeClient(
+            [
+                SimpleNamespace(name="models/gemini-2.5-pro"),
+                SimpleNamespace(name="models/gemini-2.5-flash"),
+                SimpleNamespace(name="models/gemini-2.5-pro"),
+            ]
+        )
+        models = await provider.get_models()
+        assert models == ["gemini-2.5-flash", "gemini-2.5-pro"]
+    finally:
+        await provider.terminate()

From c20931eea126c7b33dfe4c57176cb8df0687c2b3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=E2=82=82=E2=82=82H=E2=82=82=E2=82=85NO=E2=82=86?=
 <96930391+Sisyphbaous-DT-Project@users.noreply.github.com>
Date: Thu, 23 Apr 2026 11:14:57 +0800
Subject: [PATCH 3/4] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20Gemini=20?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=88=86=E9=A1=B5=E9=81=8D=E5=8E=86=E5=B9=B6?=
 =?UTF-8?q?=E8=A1=A5=E5=85=85=E5=BC=82=E5=B8=B8=E5=88=86=E6=94=AF=E6=B5=8B?=
 =?UTF-8?q?=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../sources/gemini_embedding_source.py        |  3 +-
 tests/test_dashboard.py                       | 53 +++++++++++++++++++
 tests/test_gemini_embedding_source.py         |  9 +++-
 3 files changed, 62 insertions(+), 3 deletions(-)

diff --git a/astrbot/core/provider/sources/gemini_embedding_source.py b/astrbot/core/provider/sources/gemini_embedding_source.py
index 9cc62d5572..47596bb327 100644
--- a/astrbot/core/provider/sources/gemini_embedding_source.py
+++ b/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
diff --git a/tests/test_dashboard.py b/tests/test_dashboard.py
index c4233bd83f..a36fa97ce6 100644
--- a/tests/test_dashboard.py
+++ b/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
diff --git a/tests/test_gemini_embedding_source.py b/tests/test_gemini_embedding_source.py
index 8ad4d1be6e..0f2e2d3122 100644
--- a/tests/test_gemini_embedding_source.py
+++ b/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:

From 1ac60e87c1e784eb6844c4536368d4754652db55 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=E2=82=82=E2=82=82H=E2=82=82=E2=82=85NO=E2=82=86?=
 <96930391+Sisyphbaous-DT-Project@users.noreply.github.com>
Date: Thu, 23 Apr 2026 11:28:26 +0800
Subject: [PATCH 4/4] =?UTF-8?q?test:=20=E8=A1=A5=E5=85=85=E4=B8=8D?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E6=A8=A1=E5=9E=8B=E8=87=AA=E5=8A=A8=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E6=97=B6=E7=9A=84=E9=94=99=E8=AF=AF=E6=B6=88=E6=81=AF?=
 =?UTF-8?q?=E6=96=AD=E8=A8=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_dashboard.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_dashboard.py b/tests/test_dashboard.py
index a36fa97ce6..1def897d8f 100644
--- a/tests/test_dashboard.py
+++ b/tests/test_dashboard.py
@@ -1326,6 +1326,7 @@ async def test_get_embedding_models_unsupported_returns_error(
     assert response.status_code == 200
     data = await response.get_json()
     assert data["status"] == "error"
+    assert data["message"] == "当前提供商暂不支持自动获取模型列表，请手动填写模型 ID"
     assert _UnsupportedEmbeddingProvider.terminate_calls == 1