vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-20 17:52:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: settings and credentials refactoring (#360)

Browse source 
* refactor service credential managers

* refactor azure settings

* refactor openai settings

* refactor llama settings

* refactor you settings

* refactor included files settings

* refactor general settings

* refactor advanced settings

* fix advanced settings component init

* refactor project structure

* refactor service settings forms

* remove openai quota exceeded field validator

* fix credential modified conditions

* fix and rearrange minor stuff

* fix you auth logic, add credential cache
This commit is contained in:
Carl-Robert 2024-02-08 01:02:08 +02:00 • committed by GitHub
parent 7c067d9edd
commit 93145098f5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
80 changed files with 1842 additions and 2040 deletions
Download patch file Download diff file Expand all files Collapse all files

142
src/main/java/ee/carlrobert/codegpt/settings/GeneralSettingsConfigurable.java Normal file
View file

@ -0,0 +1,142 @@
package ee.carlrobert.codegpt.settings;
import com.intellij.openapi.Disposable;
import com.intellij.openapi.options.Configurable;
import com.intellij.openapi.util.Disposer;
import ee.carlrobert.codegpt.CodeGPTBundle;
import ee.carlrobert.codegpt.conversations.ConversationsState;
import ee.carlrobert.codegpt.credentials.AzureCredentialsManager;
import ee.carlrobert.codegpt.credentials.LlamaCredentialManager;
import ee.carlrobert.codegpt.credentials.OpenAICredentialManager;
import ee.carlrobert.codegpt.settings.service.azure.AzureSettings;
import ee.carlrobert.codegpt.settings.service.azure.AzureSettingsForm;
import ee.carlrobert.codegpt.settings.service.llama.LlamaSettings;
import ee.carlrobert.codegpt.settings.service.llama.form.LlamaSettingsForm;
import ee.carlrobert.codegpt.settings.service.openai.OpenAISettings;
import ee.carlrobert.codegpt.settings.service.openai.OpenAISettingsForm;
import ee.carlrobert.codegpt.settings.service.you.YouSettings;
import ee.carlrobert.codegpt.settings.service.you.YouSettingsForm;
import ee.carlrobert.codegpt.telemetry.TelemetryAction;
import ee.carlrobert.codegpt.toolwindow.chat.standard.StandardChatToolWindowContentManager;
import ee.carlrobert.codegpt.util.ApplicationUtil;
import javax.swing.JComponent;
import org.jetbrains.annotations.Nls;
import org.jetbrains.annotations.Nullable;
public class GeneralSettingsConfigurable implements Configurable {
private Disposable parentDisposable;
private GeneralSettingsComponent component;
@Nls(capitalization = Nls.Capitalization.Title)
@Override
public String getDisplayName() {
return CodeGPTBundle.get("settings.displayName");
}
@Override
public JComponent getPreferredFocusedComponent() {
return component.getPreferredFocusedComponent();
}
@Nullable
@Override
public JComponent createComponent() {
var settings = GeneralSettings.getInstance();
parentDisposable = Disposer.newDisposable();
component = new GeneralSettingsComponent(parentDisposable, settings);
return component.getPanel();
}
@Override
public boolean isModified() {
var settings = GeneralSettings.getCurrentState();
var serviceSelectionForm = component.getServiceSelectionForm();
return !component.getDisplayName().equals(settings.getDisplayName())
|| component.getSelectedService() != settings.getSelectedService()
|| OpenAISettings.getInstance().isModified(serviceSelectionForm.getOpenAISettingsForm())
|| AzureSettings.getInstance().isModified(serviceSelectionForm.getAzureSettingsForm())
|| YouSettings.getInstance().isModified(serviceSelectionForm.getYouSettingsForm())
|| LlamaSettings.getInstance().isModified(serviceSelectionForm.getLlamaSettingsForm());
}
@Override
public void apply() {
var settings = GeneralSettings.getCurrentState();
settings.setDisplayName(component.getDisplayName());
settings.setSelectedService(component.getSelectedService());
var serviceSelectionForm = component.getServiceSelectionForm();
var openAISettingsForm = serviceSelectionForm.getOpenAISettingsForm();
applyOpenAISettings(openAISettingsForm);
applyAzureSettings(serviceSelectionForm.getAzureSettingsForm());
applyYouSettings(serviceSelectionForm.getYouSettingsForm());
applyLlamaSettings(serviceSelectionForm.getLlamaSettingsForm());
var serviceChanged = component.getSelectedService() != settings.getSelectedService();
var modelChanged = !OpenAISettings.getCurrentState().getModel()
.equals(openAISettingsForm.getModel());
if (serviceChanged || modelChanged) {
resetActiveTab();
if (serviceChanged) {
TelemetryAction.SETTINGS_CHANGED.createActionMessage()
.property("service", component.getSelectedService().getCode().toLowerCase())
.send();
}
}
}
private void applyLlamaSettings(LlamaSettingsForm form) {
LlamaCredentialManager.getInstance()
.setCredential(form.getLlamaServerPreferencesForm().getApiKey());
LlamaSettings.getInstance().loadState(form.getCurrentState());
}
private void applyYouSettings(YouSettingsForm form) {
YouSettings.getInstance().loadState(form.getCurrentState());
}
private void applyAzureSettings(AzureSettingsForm form) {
AzureSettings.getInstance().loadState(form.getCurrentState());
var azureCredentials = AzureCredentialsManager.getInstance();
azureCredentials.setApiKey(form.getApiKey());
azureCredentials.setActiveDirectoryToken(form.getActiveDirectoryToken());
}
@Override
public void reset() {
var settings = GeneralSettings.getCurrentState();
component.setDisplayName(settings.getDisplayName());
component.setSelectedService(settings.getSelectedService());
var serviceSelectionForm = component.getServiceSelectionForm();
serviceSelectionForm.getOpenAISettingsForm().resetForm();
serviceSelectionForm.getAzureSettingsForm().resetForm();
serviceSelectionForm.getLlamaSettingsForm().resetForm();
serviceSelectionForm.getYouSettingsForm().resetForm();
}
@Override
public void disposeUIResources() {
if (parentDisposable != null) {
Disposer.dispose(parentDisposable);
}
component = null;
}
private void applyOpenAISettings(OpenAISettingsForm form) {
OpenAICredentialManager.getInstance().setCredential(form.getApiKey());
OpenAISettings.getInstance().loadState(form.getCurrentState());
}
private void resetActiveTab() {
ConversationsState.getInstance().setCurrentConversation(null);
var project = ApplicationUtil.findCurrentProject();
if (project == null) {
throw new RuntimeException("Could not find current project.");
}
project.getService(StandardChatToolWindowContentManager.class).resetAll();
}
}