adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(知识服务): 添加工作空间模型配置的校验

Browse Source 
在创建知识时检查工作空间是否配置了必要的模型，未配置时抛出异常提示用户
This commit is contained in:
wwq
2026-04-21 17:18:11 +08:00
parent 402c8aef5d
commit cb2a7aa60a
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/app/services/knowledge_service.py
View File

@@ -77,12 +77,18 @@ def create_knowledge(
tenant_id = workspace.tenant_id tenant_id = workspace.tenant_id
if not knowledge.embedding_id: if not knowledge.embedding_id:
if not workspace.embedding:
raise Exception("工作空间未配置 Embedding 模型，请先完善工作空间配置后重试")
knowledge.embedding_id = workspace.embedding knowledge.embedding_id = workspace.embedding
if not knowledge.reranker_id: if not knowledge.reranker_id:
if not workspace.rerank:
raise Exception("工作空间未配置 Rerank 模型，请先完善工作空间配置后重试")
knowledge.reranker_id = workspace.rerank knowledge.reranker_id = workspace.rerank
if not knowledge.llm_id: if not knowledge.llm_id:
if not workspace.llm:
raise Exception("工作空间未配置 LLM 模型，请先完善工作空间配置后重试")
knowledge.llm_id = workspace.llm knowledge.llm_id = workspace.llm
if not knowledge.image2text_id: if not knowledge.image2text_id: