fix: lookup completion request validation

This commit is contained in:
Carl-Robert Linnupuu 2024-05-11 02:18:24 +03:00
parent fcd0808111
commit 0b21652c04
3 changed files with 5 additions and 9 deletions

View file

@ -10,8 +10,6 @@ import com.intellij.openapi.application.ApplicationManager;
import com.intellij.psi.PsiElement;
import com.intellij.psi.util.PsiUtilCore;
import ee.carlrobert.codegpt.Icons;
import ee.carlrobert.codegpt.credentials.CredentialsStore;
import ee.carlrobert.codegpt.credentials.CredentialsStore.CredentialKey;
import ee.carlrobert.codegpt.settings.configuration.ConfigurationSettings;
import java.util.Optional;
import org.jetbrains.annotations.Nullable;
@ -20,15 +18,13 @@ public class MethodNameLookupListener implements LookupManagerListener {
@Override
public void activeLookupChanged(@Nullable Lookup oldLookup, @Nullable Lookup newLookup) {
var application = ApplicationManager.getApplication();
var configuration = ConfigurationSettings.getCurrentState();
if (!configuration.isMethodNameGenerationEnabled()
|| !CredentialsStore.INSTANCE.isCredentialSet(CredentialKey.OPENAI_API_KEY)
if (!ConfigurationSettings.getCurrentState().isMethodNameGenerationEnabled()
|| !CompletionRequestService.getInstance().isRequestAllowed()
|| !(newLookup instanceof LookupImpl lookup)) {
return;
}
var application = ApplicationManager.getApplication();
Optional.ofNullable(lookup.getPsiElement())
.map(PsiElement::getContext)
.ifPresent(context ->

View file

@ -88,7 +88,7 @@ public class ModelComboBoxAction extends ComboBoxAction {
OpenAIChatCompletionModel.GPT_4_0125_128k,
OpenAIChatCompletionModel.GPT_3_5_0125_16k)
.forEach(model -> actionGroup.add(createOpenAIModelAction(model, presentation)));
actionGroup.addSeparator("Custom OpenAI Service");
actionGroup.addSeparator("Custom OpenAI");
actionGroup.add(createModelAction(
CUSTOM_OPENAI,
ApplicationManager.getApplication().getService(CustomServiceSettings.class)