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

feat(model): auto-activate Ollama model as default on startup discovery

Browse Source
This commit is contained in:
matevip 2026-04-06 12:34:12 +08:00
parent d8f6ac7ea3
commit 08d1a89c0e
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mateclaw-server/src/main/java/vip/mate/llm/config/OllamaAutoDiscoveryRunner.java
View File

@ -14,6 +14,9 @@ import vip.mate.llm.model.TestResult;
import vip.mate.llm.service.ModelConfigService; import vip.mate.llm.service.ModelConfigService;
import vip.mate.llm.service.ModelDiscoveryService; import vip.mate.llm.service.ModelDiscoveryService;
import vip.mate.llm.service.ModelProviderService;
import java.util.List;
import java.util.Set; import java.util.Set;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -36,6 +39,7 @@ public class OllamaAutoDiscoveryRunner implements ApplicationRunner {
private final ModelDiscoveryService modelDiscoveryService; private final ModelDiscoveryService modelDiscoveryService;
private final ModelConfigService modelConfigService; private final ModelConfigService modelConfigService;
private final ModelProviderService modelProviderService;
@Async @Async
@Override @Override
@ -82,8 +86,32 @@ public class OllamaAutoDiscoveryRunner implements ApplicationRunner {
log.info("Ollama: auto-enabled model '{}'", modelTag); log.info("Ollama: auto-enabled model '{}'", modelTag);
} }
} }
// Auto-activate first Ollama model if no valid default exists
tryAutoActivateOllamaModel();
} catch (Exception e) { } catch (Exception e) {
log.debug("Ollama auto-discovery skipped: {}", e.getMessage()); log.debug("Ollama auto-discovery skipped: {}", e.getMessage());
} }
} }
private void tryAutoActivateOllamaModel() {
List<ModelConfigEntity> ollamaModels = modelConfigService.listModelsByProvider(OLLAMA_PROVIDER_ID);
List<ModelConfigEntity> enabledModels = ollamaModels.stream()
.filter(m -> Boolean.TRUE.equals(m.getEnabled()))
.toList();
if (enabledModels.isEmpty()) {
return;
}
boolean hasValidDefault;
try {
ModelConfigEntity current = modelConfigService.getDefaultModel();
hasValidDefault = modelProviderService.isProviderAvailable(current.getProvider());
} catch (Exception e) {
hasValidDefault = false;
}
if (!hasValidDefault) {
ModelConfigEntity first = enabledModels.get(0);
modelConfigService.setDefaultModel(OLLAMA_PROVIDER_ID, first.getModelName());
log.info("Ollama: auto-activated default model '{}'", first.getModelName());
}
}
} }