opensource/mateclaw
2
0
Fork 0
mirror of https://gitee.com/mateos/mateclaw.git synced 2026-09-15 03:55:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(embedding): skip unconfigured provider in embedding model resolution

Browse Source
This commit is contained in:
matevip 2026-04-22 10:12:57 +08:00
parent bc002fd302
commit f8c7e5271b
1 changed files with 24 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
mateclaw-server/src/main/java/vip/mate/wiki/service/WikiEmbeddingService.java
View File

@ -47,6 +47,7 @@ public class WikiEmbeddingService {
private final ModelConfigService modelConfigService; private final ModelConfigService modelConfigService;
private final WikiKnowledgeBaseService kbService; private final WikiKnowledgeBaseService kbService;
private final SystemSettingMapper systemSettingMapper; private final SystemSettingMapper systemSettingMapper;
private final vip.mate.llm.service.ModelProviderService modelProviderService;
/** 系统默认 embedding 模型的 mate_system_setting key */ /** 系统默认 embedding 模型的 mate_system_setting key */
public static final String SYSTEM_SETTING_DEFAULT_EMBEDDING_ID = "embedding.default.model.id"; public static final String SYSTEM_SETTING_DEFAULT_EMBEDDING_ID = "embedding.default.model.id";
@ -87,15 +88,24 @@ public class WikiEmbeddingService {
if (defaultId != null) { if (defaultId != null) {
ModelConfigEntity model = safeGetModel(defaultId); ModelConfigEntity model = safeGetModel(defaultId);
if (isUsable(model)) { if (isUsable(model)) {
return new Resolved(factory.build(model), model.getModelName()); try {
return new Resolved(factory.build(model), model.getModelName());
} catch (Exception e) {
log.warn("[WikiEmbedding] System default embedding model {} build failed: {}", defaultId, e.getMessage());
}
} else {
log.warn("[WikiEmbedding] System default embedding model {} is unusable, falling back", defaultId);
} }
log.warn("[WikiEmbedding] System default embedding model {} is unusable, falling back", defaultId);
} }
// 优先级 3：任意 enabled // 优先级 3：任意 enabled
ModelConfigEntity anyEnabled = modelConfigService.findFirstEnabledEmbedding(); ModelConfigEntity anyEnabled = modelConfigService.findFirstEnabledEmbedding();
if (isUsable(anyEnabled)) { if (isUsable(anyEnabled)) {
return new Resolved(factory.build(anyEnabled), anyEnabled.getModelName()); try {
return new Resolved(factory.build(anyEnabled), anyEnabled.getModelName());
} catch (Exception e) {
log.warn("[WikiEmbedding] Fallback embedding model {} build failed: {}", anyEnabled.getId(), e.getMessage());
}
} }
log.warn("[WikiEmbedding] No usable embedding model configured. " log.warn("[WikiEmbedding] No usable embedding model configured. "
@ -355,9 +365,17 @@ public class WikiEmbeddingService {
} }
private boolean isUsable(ModelConfigEntity model) { private boolean isUsable(ModelConfigEntity model) {
return model != null if (model == null || !Boolean.TRUE.equals(model.getEnabled())
&& Boolean.TRUE.equals(model.getEnabled()) || !"embedding".equals(model.getModelType())) {
&& "embedding".equals(model.getModelType()); return false;
}
// Skip models whose provider lacks a valid API key (mirrors chat model path fix 341ad1f)
try {
return modelProviderService.isProviderConfigured(model.getProvider());
} catch (Exception e) {
log.debug("[WikiEmbedding] Provider check failed for model {}: {}", model.getId(), e.getMessage());
return false;
}
} }
private Long readSystemDefaultEmbeddingId() { private Long readSystemDefaultEmbeddingId() {