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

on quota exceeded suggest user switch to different LLM provider (#221)

Browse source 
* on quota exceeded suggest user switch to different LLM provider

* Improve insufficient quota handling, add more telemetry actions

---------

Co-authored-by: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
This commit is contained in:
keith siilats 2023-10-04 17:18:50 -04:00 • committed by GitHub
parent a45646740e
commit 4c8b8d4e4f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 95 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/main/java/ee/carlrobert/codegpt/settings/SettingsConfigurable.java
View file

@ -9,6 +9,7 @@ import ee.carlrobert.codegpt.credentials.OpenAICredentialsManager;
import ee.carlrobert.codegpt.settings.state.AzureSettingsState;
import ee.carlrobert.codegpt.settings.state.OpenAISettingsState;
import ee.carlrobert.codegpt.settings.state.SettingsState;
import ee.carlrobert.codegpt.telemetry.TelemetryAction;
import ee.carlrobert.codegpt.toolwindow.chat.standard.StandardChatToolWindowContentManager;
import ee.carlrobert.codegpt.util.ApplicationUtils;
import javax.swing.JComponent;
@ -64,7 +65,8 @@ public class SettingsConfigurable implements Configurable, Disposable {
OpenAICredentialsManager.getInstance().setApiKey(serviceSelectionForm.getOpenAIApiKey());
AzureCredentialsManager.getInstance().setApiKey(serviceSelectionForm.getAzureOpenAIApiKey());
AzureCredentialsManager.getInstance().setAzureActiveDirectoryToken(serviceSelectionForm.getAzureActiveDirectoryToken());
AzureCredentialsManager.getInstance()
.setAzureActiveDirectoryToken(serviceSelectionForm.getAzureActiveDirectoryToken());
settings.setDisplayName(settingsComponent.getDisplayName());
settings.setUseOpenAIService(serviceSelectionForm.isOpenAIServiceSelected());
@ -77,6 +79,11 @@ public class SettingsConfigurable implements Configurable, Disposable {
if (serviceChanged || modelChanged) {
resetActiveTab();
if (serviceChanged) {
TelemetryAction.SETTINGS_CHANGED.createActionMessage()
.property("service", getServiceCode(serviceSelectionForm))
.send();
}
}
}
@ -109,7 +116,9 @@ public class SettingsConfigurable implements Configurable, Disposable {
public void dispose() {
}
private boolean isServiceChanged(ServiceSelectionForm serviceSelectionForm, SettingsState settings) {
private boolean isServiceChanged(
ServiceSelectionForm serviceSelectionForm,
SettingsState settings) {
return serviceSelectionForm.isOpenAIServiceSelected() != settings.isUseOpenAIService() ||
serviceSelectionForm.isAzureServiceSelected() != settings.isUseAzureService() ||
serviceSelectionForm.isYouServiceSelected() != settings.isUseYouService();
@ -124,4 +133,17 @@ public class SettingsConfigurable implements Configurable, Disposable {
project.getService(StandardChatToolWindowContentManager.class).resetActiveTab();
}
private String getServiceCode(ServiceSelectionForm serviceSelectionForm) {
if (serviceSelectionForm.isOpenAIServiceSelected()) {
return "openai";
}
if (serviceSelectionForm.isAzureServiceSelected()) {
return "azure";
}
if (serviceSelectionForm.isYouServiceSelected()) {
return "you";
}
return null;
}
}