fix(core): openAI custom model not work as expected

close #119
unit-mesh · Mar 27, 2024 · d4eee77 · d4eee77
1 parent 44d6d79
commit d4eee77
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/llms/openai/OpenAIProvider.kt b/src/main/kotlin/cc/unitmesh/devti/llms/openai/OpenAIProvider.kt
@@ -35,11 +35,11 @@ class OpenAIProvider(val project: Project) : LLMProvider {
     private val timeout = Duration.ofSeconds(defaultTimeout)
     private val openAiVersion: String
         get() {
-            val customModel = AutoDevSettingsState.getInstance().customModel
-            if(AutoDevSettingsState.getInstance().openAiModel == SELECT_CUSTOM_MODEL) {
-                AutoDevSettingsState.getInstance().openAiModel = customModel
+            val model = AutoDevSettingsState.getInstance().openAiModel
+            if(model == SELECT_CUSTOM_MODEL) {
+                return AutoDevSettingsState.getInstance().customModel
             }
-            return  AutoDevSettingsState.getInstance().openAiModel
+            return model
         }
     private val openAiKey: String
         get() = AutoDevSettingsState.getInstance().openAiKey