From fa13ca7a421fe1464337ca76289ada48be8efc36 Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Mon, 23 Feb 2026 13:47:32 -0800
Subject: [PATCH] Fix: pass api_base to OpenAIEmbedding (#12151)

---
 src/paperless_ai/embedding.py            | 1 +
 src/paperless_ai/tests/test_embedding.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/paperless_ai/embedding.py b/src/paperless_ai/embedding.py
index 993c9ae30..686f73341 100644
--- a/src/paperless_ai/embedding.py
+++ b/src/paperless_ai/embedding.py
@@ -23,6 +23,7 @@ def get_embedding_model() -> BaseEmbedding:
             return OpenAIEmbedding(
                 model=config.llm_embedding_model or "text-embedding-3-small",
                 api_key=config.llm_api_key,
+                api_base=config.llm_endpoint or None,
             )
         case LLMEmbeddingBackend.HUGGINGFACE:
             return HuggingFaceEmbedding(
diff --git a/src/paperless_ai/tests/test_embedding.py b/src/paperless_ai/tests/test_embedding.py
index 9430205fa..1fb69ee06 100644
--- a/src/paperless_ai/tests/test_embedding.py
+++ b/src/paperless_ai/tests/test_embedding.py
@@ -65,12 +65,14 @@ def test_get_embedding_model_openai(mock_ai_config):
     mock_ai_config.return_value.llm_embedding_backend = LLMEmbeddingBackend.OPENAI
     mock_ai_config.return_value.llm_embedding_model = "text-embedding-3-small"
     mock_ai_config.return_value.llm_api_key = "test_api_key"
+    mock_ai_config.return_value.llm_endpoint = "http://test-url"
 
     with patch("paperless_ai.embedding.OpenAIEmbedding") as MockOpenAIEmbedding:
         model = get_embedding_model()
         MockOpenAIEmbedding.assert_called_once_with(
             model="text-embedding-3-small",
             api_key="test_api_key",
+            api_base="http://test-url",
         )
         assert model == MockOpenAIEmbedding.return_value