From dd477ef0418c7ce531d829839f4d695856d1d1d3 Mon Sep 17 00:00:00 2001 From: SJpoor <928708321@qq.com> Date: Mon, 8 Dec 2025 20:01:56 +0800 Subject: [PATCH] fix: set default embedding model for TEI profile --- common/settings.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/common/settings.py b/common/settings.py index c8f7a6de4..45dcdb618 100644 --- a/common/settings.py +++ b/common/settings.py @@ -210,7 +210,10 @@ def init_settings(): IMAGE2TEXT_CFG = _resolve_per_model_config(image2text_entry, LLM_FACTORY, API_KEY, LLM_BASE_URL) CHAT_MDL = CHAT_CFG.get("model", "") or "" - EMBEDDING_MDL = os.getenv("TEI_MODEL", "BAAI/bge-small-en-v1.5") if "tei-" in os.getenv("COMPOSE_PROFILES", "") else "" + EMBEDDING_MDL = EMBEDDING_CFG.get("model", "") or "" + compose_profiles = os.getenv("COMPOSE_PROFILES", "") + if "tei-" in compose_profiles: + EMBEDDING_MDL = os.getenv("TEI_MODEL", EMBEDDING_MDL or "BAAI/bge-small-en-v1.5") RERANK_MDL = RERANK_CFG.get("model", "") or "" ASR_MDL = ASR_CFG.get("model", "") or "" IMAGE2TEXT_MDL = IMAGE2TEXT_CFG.get("model", "") or ""