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

* 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

View file

@ -200,9 +200,16 @@ public abstract class BaseChatToolWindowTabPanel implements ChatToolWindowTabPan
}
}));
requestHandler.addErrorListener((error, ex) -> {
responsePanel.enableActions();
responseContainer.displayError(error.getMessage());
stopStreaming(responseContainer);
try {
if ("insufficient_quota".equals(error.getCode())) {
responseContainer.displayQuotaExceeded();
} else {
responseContainer.displayError(error.getMessage());
}
} finally {
responsePanel.enableActions();
stopStreaming(responseContainer);
}
});
requestHandler.addSerpResultsListener(
serpResults -> serpResultsMapping.put(message.getId(), serpResults.stream()