mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-28 08:49:42 +02:00
fix: harden the base url settings in SaaS mode
This commit is contained in:
parent
88d6ac425b
commit
c7b5ee1ae2
5 changed files with 339 additions and 3 deletions
|
|
@ -11,6 +11,7 @@ from loguru import logger
|
|||
from openai import AsyncOpenAI
|
||||
|
||||
from api.db.db_client import DBClient
|
||||
from api.utils.url_security import validate_user_configured_service_url
|
||||
|
||||
from .base import BaseEmbeddingService
|
||||
|
||||
|
|
@ -54,6 +55,10 @@ class OpenAIEmbeddingService(BaseEmbeddingService):
|
|||
if self._api_key_configured:
|
||||
client_kwargs = {"api_key": api_key}
|
||||
if base_url:
|
||||
validate_user_configured_service_url(
|
||||
base_url,
|
||||
field_name="base_url",
|
||||
)
|
||||
client_kwargs["base_url"] = base_url
|
||||
self.client = AsyncOpenAI(**client_kwargs)
|
||||
logger.info(f"OpenAI embedding service initialized with model: {model_id}")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue