Merge pull request #113 from devoxx/issue-112

Issue 112
devoxx · Jun 25, 2024 · 8b445f0 · 8b445f0
2 parents 49d4347 + 53e9330
commit 8b445f0
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 4 deletions.
diff --git a/src/main/java/com/devoxx/genie/ui/DevoxxGenieToolWindowContent.java b/src/main/java/com/devoxx/genie/ui/DevoxxGenieToolWindowContent.java
@@ -72,10 +72,11 @@ public DevoxxGenieToolWindowContent(@NotNull ToolWindow toolWindow) {
      */
     private void setLastSelectedProvider() {
         String lastSelectedProvider = DevoxxGenieStateService.getInstance().getLastSelectedProvider();
-         if (lastSelectedProvider != null && !lastSelectedProvider.isEmpty()) {
+        if (lastSelectedProvider != null && !lastSelectedProvider.isEmpty()) {
             llmProvidersComboBox.setSelectedItem(lastSelectedProvider);
             updateModelNamesComboBox(ModelProvider.valueOf(lastSelectedProvider));
         } else {
+            // If no last selected provider, select the first item in the combobox
             Object selectedItem = llmProvidersComboBox.getSelectedItem();
             if (selectedItem != null) {
                 updateModelNamesComboBox(ModelProvider.valueOf((String) selectedItem));
@@ -90,6 +91,8 @@ private void setupUI() {
         contentPanel.setLayout(new BorderLayout());
         contentPanel.add(createTopPanel(), BorderLayout.NORTH);
         contentPanel.add(createSplitter(), BorderLayout.CENTER);
+
+        setLastSelectedProvider();
     }
 
     /**
@@ -125,8 +128,22 @@ private void setupUI() {
      * Refresh the UI elements because the settings have changed.
      */
     public void settingsChanged() {
+        String currentProvider = (String) llmProvidersComboBox.getSelectedItem();
+        String currentModel = (String) modelNameComboBox.getSelectedItem();
+
         llmProvidersComboBox.removeAllItems();
         addLLMProvidersToComboBox();
+
+        if (currentProvider != null && !currentProvider.isEmpty()) {
+            llmProvidersComboBox.setSelectedItem(currentProvider);
+            updateModelNamesComboBox(ModelProvider.valueOf(currentProvider));
+            if (currentModel != null && !currentModel.isEmpty()) {
+                modelNameComboBox.setSelectedItem(currentModel);
+            }
+        } else {
+            setLastSelectedProvider();
+        }
+
         actionButtonsPanel.configureSearchButtonsVisibility();
     }
 
@@ -242,7 +259,6 @@ private void handleModelProviderSelectionChange(@NotNull ActionEvent e) {
         if (!e.getActionCommand().equals(Constant.COMBO_BOX_CHANGED) || !isInitializationComplete) return;
 
         JComboBox<?> comboBox = (JComboBox<?>) e.getSource();
-
         String selectedLLMProvider = (String) comboBox.getSelectedItem();
         if (selectedLLMProvider == null) return;
 

diff --git a/src/main/java/com/devoxx/genie/ui/settings/DevoxxGenieStateService.java b/src/main/java/com/devoxx/genie/ui/settings/DevoxxGenieStateService.java
@@ -2,7 +2,6 @@
 
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.components.PersistentStateComponent;
-import com.intellij.openapi.components.Service;
 import com.intellij.openapi.components.State;
 import com.intellij.openapi.components.Storage;
 import com.intellij.util.xmlb.XmlSerializerUtil;
@@ -14,7 +13,6 @@
 
 @Getter
 @Setter
-@Service
 @State(
     name = "com.devoxx.genie.ui.SettingsState",
     storages = @Storage("DevoxxGenieSettingsPlugin.xml")

diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
@@ -35,6 +35,7 @@
         <h2>v0.1.19</h2>
         <UL>
             <LI>Feat #98: Allow a streaming response to be stopped</LI>
+            <LI>Feat #112: Keep selected LLM provider after settings page</LI>
         </UL>
         <h2>v0.1.18</h2>
         <UL>
@@ -222,6 +223,7 @@
                     icon="/icons/pluginIcon.svg"
                     factoryClass="com.devoxx.genie.ui.DevoxxGenieToolWindowFactory"/>
 
+        <applicationService serviceImplementation="com.devoxx.genie.ui.settings.DevoxxGenieStateService"/>
         <applicationService serviceImplementation="com.devoxx.genie.service.PromptExecutionService"/>
         <applicationService serviceImplementation="com.devoxx.genie.service.ChatMemoryService"/>
         <applicationService serviceImplementation="com.devoxx.genie.service.MessageCreationService"/>