fix: append /v1 for openai embedding api base

Author: idiotsj

astrbot/core/provider/sources/openai_embedding_source.py

@@ -14,6 +14,13 @@
     provider_type=ProviderType.EMBEDDING,
 )
 class OpenAIEmbeddingProvider(EmbeddingProvider):
+    @staticmethod
+    def _normalize_embedding_api_base(api_base: str) -> str:
+        normalized_api_base = api_base.rstrip("/")
+        if not normalized_api_base.endswith("/v1"):
+            normalized_api_base = f"{normalized_api_base}/v1"
+        return normalized_api_base
+
     def __init__(self, provider_config: dict, provider_settings: dict) -> None:
         super().__init__(provider_config, provider_settings)
         self.provider_config = provider_config
@@ -27,6 +34,8 @@ def __init__(self, provider_config: dict, provider_settings: dict) -> None:
         api_base = provider_config.get(
             "embedding_api_base", "https://api.openai.com/v1"
         ).strip()
+        if api_base:
+            api_base = self._normalize_embedding_api_base(api_base)
         logger.info(f"[OpenAI Embedding] {provider_id} Using API Base: {api_base}")
         self.client = AsyncOpenAI(
             api_key=provider_config.get("embedding_api_key"),

tests/test_openai_source.py

@@ -4,6 +4,9 @@
 from openai.types.chat.chat_completion import ChatCompletion
 
 from astrbot.core.provider.sources.groq_source import ProviderGroq
+from astrbot.core.provider.sources.openai_embedding_source import (
+    OpenAIEmbeddingProvider,
+)
 from astrbot.core.provider.sources.openai_source import ProviderOpenAIOfficial
 
 
@@ -49,6 +52,20 @@ def _make_groq_provider(overrides: dict | None = None) -> ProviderGroq:
     )
 
 
+def _make_embedding_provider(overrides: dict | None = None) -> OpenAIEmbeddingProvider:
+    provider_config = {
+        "id": "test-openai-embedding",
+        "type": "openai_embedding",
+        "embedding_api_key": "test-key",
+    }
+    if overrides:
+        provider_config.update(overrides)
+    return OpenAIEmbeddingProvider(
+        provider_config=provider_config,
+        provider_settings={},
+    )
+
+
 @pytest.mark.asyncio
 async def test_handle_api_error_content_moderated_removes_images():
     provider = _make_provider(
@@ -234,7 +251,9 @@ async def test_openai_payload_keeps_reasoning_content_in_assistant_history():
         provider._finally_convert_payload(payloads)
 
         assistant_message = payloads["messages"][0]
-        assert assistant_message["content"] == [{"type": "text", "text": "final answer"}]
+        assert assistant_message["content"] == [
+            {"type": "text", "text": "final answer"}
+        ]
         assert assistant_message["reasoning_content"] == "step 1"
     finally:
         await provider.terminate()
@@ -259,7 +278,9 @@ async def test_groq_payload_drops_reasoning_content_from_assistant_history():
         provider._finally_convert_payload(payloads)
 
         assistant_message = payloads["messages"][0]
-        assert assistant_message["content"] == [{"type": "text", "text": "final answer"}]
+        assert assistant_message["content"] == [
+            {"type": "text", "text": "final answer"}
+        ]
         assert "reasoning_content" not in assistant_message
         assert "reasoning" not in assistant_message
     finally:
@@ -533,3 +554,25 @@ async def fake_create(**kwargs):
         assert extra_body["temperature"] == 0.1
     finally:
         await provider.terminate()
+
+
+@pytest.mark.asyncio
+async def test_openai_embedding_provider_appends_v1_to_base_url_when_missing():
+    provider = _make_embedding_provider(
+        {"embedding_api_base": "https://example.com/openai"}
+    )
+    try:
+        assert str(provider.client.base_url) == "https://example.com/openai/v1/"
+    finally:
+        await provider.terminate()
+
+
+@pytest.mark.asyncio
+async def test_openai_embedding_provider_preserves_existing_v1_suffix():
+    provider = _make_embedding_provider(
+        {"embedding_api_base": "https://example.com/openai/v1/"}
+    )
+    try:
+        assert str(provider.client.base_url) == "https://example.com/openai/v1/"
+    finally:
+        await provider.terminate()