feat: implement support for You Pro modes (#399)

* Implement support for You Pro modes: Default, Agent, Custom with various 3rd party models and Research

* Update list of You modes/models depending on user having subscription

* add default value for chatMode

src/main/java/ee/carlrobert/codegpt/completions/CompletionRequestProvider.java

@@ -77,7 +77,7 @@ public class CompletionRequestProvider {
   }
 
   public static String getPromptWithContext(List<ReferencedFile> referencedFiles,
-      String userPrompt) {
+                                            String userPrompt) {
     var includedFilesSettings = IncludedFilesSettings.getCurrentState();
     var repeatableContext = referencedFiles.stream()
         .map(item -> includedFilesSettings.getRepeatableContext()
@@ -158,6 +158,8 @@ public class CompletionRequestProvider {
   public YouCompletionRequest buildYouCompletionRequest(Message message) {
     var requestBuilder = new YouCompletionRequest.Builder(message.getPrompt())
         .setUseGPT4Model(YouSettings.getCurrentState().isUseGPT4Model())
+        .setChatMode(YouSettings.getCurrentState().getChatMode())
+        .setCustomModel(YouSettings.getCurrentState().getCustomModel())
         .setChatHistory(conversation.getMessages().stream()
             .map(prevMessage -> new YouCompletionRequestMessage(
                 prevMessage.getPrompt(),

src/main/java/ee/carlrobert/codegpt/settings/service/you/YouSettingsState.java

@@ -1,5 +1,7 @@
 package ee.carlrobert.codegpt.settings.service.you;
 
+import ee.carlrobert.llm.client.you.completion.YouCompletionCustomModel;
+import ee.carlrobert.llm.client.you.completion.YouCompletionMode;
 import java.util.Objects;
 
 public class YouSettingsState {
@@ -7,6 +9,8 @@ public class YouSettingsState {
   private String email = "";
   private boolean displayWebSearchResults = true;
   private boolean useGPT4Model;
+  private YouCompletionMode chatMode = YouCompletionMode.DEFAULT;
+  private YouCompletionCustomModel customModel;
 
   public String getEmail() {
     return email;
@@ -32,6 +36,22 @@ public class YouSettingsState {
     this.useGPT4Model = useGPT4Model;
   }
 
+  public YouCompletionMode getChatMode() {
+    return chatMode;
+  }
+
+  public void setChatMode(YouCompletionMode chatMode) {
+    this.chatMode = chatMode;
+  }
+
+  public YouCompletionCustomModel getCustomModel() {
+    return customModel;
+  }
+
+  public void setCustomModel(YouCompletionCustomModel customModel) {
+    this.customModel = customModel;
+  }
+
   @Override
   public boolean equals(Object o) {
     if (this == o) {
@@ -43,11 +63,13 @@ public class YouSettingsState {
     YouSettingsState that = (YouSettingsState) o;
     return displayWebSearchResults == that.displayWebSearchResults
         && useGPT4Model == that.useGPT4Model
-        && Objects.equals(email, that.email);
+        && Objects.equals(email, that.email)
+        && chatMode == that.chatMode
+        && customModel == that.customModel;
   }
 
   @Override
   public int hashCode() {
-    return Objects.hash(displayWebSearchResults, useGPT4Model, email);
+    return Objects.hash(displayWebSearchResults, useGPT4Model, email, chatMode, customModel);
   }
 }

src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ui/ChatMessageResponseBody.java

@@ -141,6 +141,9 @@ public class ChatMessageResponseBody extends JPanel {
     if (responseReceived) {
       add(createTextPane(html, false));
     } else {
+      if (currentlyProcessedTextPane == null) {
+        prepareProcessingText(false);
+      }
       currentlyProcessedTextPane.setText(html);
     }
   }

src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ui/textarea/ModelComboBoxAction.java

@@ -2,6 +2,7 @@ package ee.carlrobert.codegpt.toolwindow.chat.ui.textarea;
 
 import static ee.carlrobert.codegpt.settings.service.ServiceType.CUSTOM_OPENAI;
 import static ee.carlrobert.codegpt.settings.service.ServiceType.OPENAI;
+import static ee.carlrobert.codegpt.settings.service.ServiceType.YOU;
 import static java.lang.String.format;
 
 import com.intellij.openapi.actionSystem.AnAction;
@@ -9,9 +10,13 @@ import com.intellij.openapi.actionSystem.AnActionEvent;
 import com.intellij.openapi.actionSystem.DefaultActionGroup;
 import com.intellij.openapi.actionSystem.Presentation;
 import com.intellij.openapi.actionSystem.ex.ComboBoxAction;
+import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.project.DumbAwareAction;
+import com.intellij.util.messages.MessageBusConnection;
 import ee.carlrobert.codegpt.Icons;
 import ee.carlrobert.codegpt.completions.llama.LlamaModel;
+import ee.carlrobert.codegpt.completions.you.YouUserManager;
+import ee.carlrobert.codegpt.completions.you.auth.SignedOutNotifier;
 import ee.carlrobert.codegpt.conversations.ConversationService;
 import ee.carlrobert.codegpt.conversations.ConversationsState;
 import ee.carlrobert.codegpt.settings.GeneralSettings;
@@ -21,7 +26,11 @@ import ee.carlrobert.codegpt.settings.service.custom.CustomServiceSettings;
 import ee.carlrobert.codegpt.settings.service.llama.LlamaSettings;
 import ee.carlrobert.codegpt.settings.service.openai.OpenAISettings;
 import ee.carlrobert.codegpt.settings.service.openai.OpenAISettingsState;
+import ee.carlrobert.codegpt.settings.service.you.YouSettings;
+import ee.carlrobert.codegpt.settings.service.you.YouSettingsState;
 import ee.carlrobert.llm.client.openai.completion.OpenAIChatCompletionModel;
+import ee.carlrobert.llm.client.you.completion.YouCompletionCustomModel;
+import ee.carlrobert.llm.client.you.completion.YouCompletionMode;
 import java.util.List;
 import javax.swing.Icon;
 import javax.swing.JComponent;
@@ -32,12 +41,16 @@ public class ModelComboBoxAction extends ComboBoxAction {
   private final Runnable onAddNewTab;
   private final GeneralSettingsState settings;
   private final OpenAISettingsState openAISettings;
+  private final YouSettingsState youSettings;
 
   public ModelComboBoxAction(Runnable onAddNewTab, ServiceType selectedService) {
     this.onAddNewTab = onAddNewTab;
     settings = GeneralSettings.getCurrentState();
     openAISettings = OpenAISettings.getCurrentState();
+    youSettings = YouSettings.getCurrentState();
     updateTemplatePresentation(selectedService);
+
+    subscribeToYouSignedOutTopic(ApplicationManager.getApplication().getMessageBus().connect());
   }
 
   public JComponent createCustomComponent(@NotNull String place) {
@@ -87,8 +100,22 @@ public class ModelComboBoxAction extends ComboBoxAction {
         getLlamaCppPresentationText(),
         Icons.Llama,
         presentation));
-    actionGroup.addSeparator();
-    actionGroup.add(createModelAction(ServiceType.YOU, "You.com", Icons.YouSmall, presentation));
+
+    if (YouUserManager.getInstance().isSubscribed()) {
+      actionGroup.addSeparator("You.com");
+      List.of(
+              YouCompletionMode.DEFAULT,
+              YouCompletionMode.AGENT,
+              YouCompletionMode.RESEARCH)
+          .forEach(mode -> actionGroup.add(createYouModeAction(mode, presentation)));
+      List.of(
+              YouCompletionCustomModel.values()
+          )
+          .forEach(model -> actionGroup.add(createYouModelAction(model, presentation)));
+    } else {
+      actionGroup.addSeparator();
+      actionGroup.add(createYouModeAction(YouCompletionMode.DEFAULT, presentation));
+    }
     return actionGroup;
   }
 
@@ -97,6 +124,22 @@ public class ModelComboBoxAction extends ComboBoxAction {
     return true;
   }
 
+  private void subscribeToYouSignedOutTopic(
+      MessageBusConnection messageBusConnection
+  ) {
+    messageBusConnection.subscribe(
+        SignedOutNotifier.SIGNED_OUT_TOPIC,
+        (SignedOutNotifier) () -> {
+          var youSettings = YouSettings.getCurrentState();
+          if (!YouUserManager.getInstance().isSubscribed()
+              && youSettings.getChatMode() != YouCompletionMode.DEFAULT) {
+            youSettings.setChatMode(YouCompletionMode.DEFAULT);
+            updateTemplatePresentation(GeneralSettings.getCurrentState().getSelectedService());
+          }
+        }
+    );
+  }
+
   private void updateTemplatePresentation(ServiceType selectedService) {
     var templatePresentation = getTemplatePresentation();
     switch (selectedService) {
@@ -121,7 +164,11 @@ public class ModelComboBoxAction extends ComboBoxAction {
         break;
       case YOU:
         templatePresentation.setIcon(Icons.YouSmall);
-        templatePresentation.setText("You.com");
+        templatePresentation.setText(
+            youSettings.getChatMode() == YouCompletionMode.CUSTOM
+                ? youSettings.getCustomModel().getDescription()
+                : youSettings.getChatMode().getDescription()
+        );
         break;
       case LLAMA_CPP:
         templatePresentation.setText(getLlamaCppPresentationText());
@@ -207,4 +254,53 @@ public class ModelComboBoxAction extends ComboBoxAction {
       }
     };
   }
+
+  private AnAction createYouModeAction(
+      YouCompletionMode mode,
+      Presentation comboBoxPresentation) {
+    createModelAction(YOU, mode.getDescription(), Icons.YouSmall,
+        comboBoxPresentation);
+    return new DumbAwareAction(mode.getDescription(), "", Icons.YouSmall) {
+      @Override
+      public void update(@NotNull AnActionEvent event) {
+        var presentation = event.getPresentation();
+        presentation.setEnabled(!presentation.getText().equals(comboBoxPresentation.getText()));
+      }
+
+      @Override
+      public void actionPerformed(@NotNull AnActionEvent e) {
+        youSettings.setChatMode(mode);
+        handleProviderChange(
+            YOU,
+            mode.getDescription(),
+            Icons.YouSmall,
+            comboBoxPresentation);
+      }
+    };
+  }
+
+  private AnAction createYouModelAction(
+      YouCompletionCustomModel model,
+      Presentation comboBoxPresentation) {
+    createModelAction(YOU, model.getDescription(), Icons.YouSmall,
+        comboBoxPresentation);
+    return new DumbAwareAction(model.getDescription(), "", Icons.YouSmall) {
+      @Override
+      public void update(@NotNull AnActionEvent event) {
+        var presentation = event.getPresentation();
+        presentation.setEnabled(!presentation.getText().equals(comboBoxPresentation.getText()));
+      }
+
+      @Override
+      public void actionPerformed(@NotNull AnActionEvent e) {
+        youSettings.setCustomModel(model);
+        youSettings.setChatMode(YouCompletionMode.CUSTOM);
+        handleProviderChange(
+            YOU,
+            model.getDescription(),
+            Icons.YouSmall,
+            comboBoxPresentation);
+      }
+    };
+  }
 }

src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ui/textarea/UserPromptTextAreaHeader.java

@@ -26,9 +26,6 @@ public class UserPromptTextAreaHeader extends JPanel {
         add(totalTokensPanel, BorderLayout.LINE_START);
         break;
       case YOU:
-        JBCheckBox gpt4CheckBox = new YouProCheckbox();
-        subscribeToYouTopics(gpt4CheckBox);
-        add(gpt4CheckBox, BorderLayout.LINE_START);
         break;
       default:
     }