P0lar1s/Ombre_Brain
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

单独配置embedding模型

Browse Source
This commit is contained in:
mousongzhe
2026-04-21 13:21:23 +08:00
parent e9d61b5d9d
commit 2d2de45d5a
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
embedding_engine.py
View File

@@ -34,8 +34,12 @@ class EmbeddingEngine:
dehy_cfg = config.get("dehydration", {})
embed_cfg = config.get("embedding", {})
self.api_key = dehy_cfg.get("api_key", "")
self.base_url = dehy_cfg.get("base_url", "https://generativelanguage.googleapis.com/v1beta/openai/")
self.api_key = (embed_cfg.get("api_key") or dehy_cfg.get("api_key") or "").strip()
self.base_url = (
(embed_cfg.get("base_url") or "").strip()
or (dehy_cfg.get("base_url") or "").strip()
or "https://generativelanguage.googleapis.com/v1beta/openai/"
)
self.model = embed_cfg.get("model", "gemini-embedding-001")
self.enabled = bool(self.api_key) and embed_cfg.get("enabled", True)