From 39679d9ee949770a458efa1cae6f0b3eec6e3d6d Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Sun, 21 Apr 2024 01:39:26 +0300
Subject: [PATCH] fix: custom service settings sync

---
 .../settings/service/custom/CustomServiceSettings.kt   | 10 ++++++----
 1 file changed, 6 insertions(+), 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
index aa9556d6..baa46482 100644
--- 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,9 +14,9 @@ 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
@@ -24,7 +24,9 @@ class CustomServiceSettings :
                 url = null
                 body = mutableMapOf()
                 headers = mutableMapOf()
-            }
+            })
+        } else {
+            super.loadState(state)
         }
     }
 }