fix: custom service settings sync

carlrobertoh · Apr 20, 2024 · 39679d9 · 39679d9
1 parent 6e6a499
commit 39679d9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/kotlin/ee/carlrobert/codegpt/settings/service/custom/CustomServiceSettings.kt b/src/main/kotlin/ee/carlrobert/codegpt/settings/service/custom/CustomServiceSettings.kt
@@ -14,17 +14,19 @@ class CustomServiceSettings :
     SimplePersistentStateComponent<CustomServiceState>(CustomServiceState()) {
 
     override fun loadState(state: CustomServiceState) {
-        this.state.run {
-            // Migrate old settings
-            if (state.url != null || state.body.isNotEmpty() || state.headers.isNotEmpty()) {
+        if (state.url != null || state.body.isNotEmpty() || state.headers.isNotEmpty()) {
+            super.loadState(this.state.apply {
+                // Migrate old settings
                 template = state.template
                 chatCompletionSettings.url = state.url
                 chatCompletionSettings.body = state.body
                 chatCompletionSettings.headers = state.headers
                 url = null
                 body = mutableMapOf()
                 headers = mutableMapOf()
-            }
+            })
+        } else {
+            super.loadState(state)
         }
     }
 }