vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-19 16:28:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Next Edit model migration (ProxyAI)

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2025-10-08 12:13:56 +01:00
parent 70bb94c0ce
commit 6490b1879e
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/main/kotlin/ee/carlrobert/codegpt/settings/models/ModelSettings.kt
View file

@ -64,7 +64,7 @@ class ModelSettings : SimplePersistentStateComponent<ModelSettingsState>(ModelSe
migrateMissingProviderInformation()
migrateEditCodeModel()
migrateProxyAIApplyModel()
migrateProxyAICodeModels()
migrateProxyAIAutocompleteAndNextEditModels()
notifyIfChanged(oldState, this.state)
}
@ -158,11 +158,13 @@ class ModelSettings : SimplePersistentStateComponent<ModelSettingsState>(ModelSe
}
}
private fun migrateProxyAICodeModels() {
val modelSelection =
service<ModelSelectionService>().getModelSelectionForFeature(FeatureType.CODE_COMPLETION)
if (modelSelection.provider == PROXYAI && modelSelection.model != ModelRegistry.MERCURY_CODER) {
setModelWithProvider(FeatureType.CODE_COMPLETION, ModelRegistry.MERCURY_CODER, PROXYAI)
private fun migrateProxyAIAutocompleteAndNextEditModels() {
val modelService = service<ModelSelectionService>()
listOf(FeatureType.CODE_COMPLETION, FeatureType.NEXT_EDIT).forEach {
val modelSelection = modelService.getModelSelectionForFeature(it)
if (modelSelection.provider == PROXYAI && modelSelection.model != ModelRegistry.MERCURY_CODER) {
setModelWithProvider(it, ModelRegistry.MERCURY_CODER, PROXYAI)
}
}
}