From 25d8835db1233989c2d43e5e313c6636a42f1108 Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Thu, 14 Nov 2024 10:03:30 +0000
Subject: [PATCH] feat: disable code completions when rate limit exceeded

---
 .../codecompletions/CodeCompletionEventListener.kt    | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/CodeCompletionEventListener.kt b/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/CodeCompletionEventListener.kt
index df156af7..9bd070e8 100644
--- a/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/CodeCompletionEventListener.kt
+++ b/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/CodeCompletionEventListener.kt
@@ -2,11 +2,15 @@ package ee.carlrobert.codegpt.codecompletions
 
 import com.intellij.notification.NotificationType
 import com.intellij.openapi.application.runReadAction
+import com.intellij.openapi.components.service
 import com.intellij.openapi.diagnostic.thisLogger
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.util.TextRange
 import ee.carlrobert.codegpt.CodeGPTKeys.IS_FETCHING_COMPLETION
 import ee.carlrobert.codegpt.CodeGPTKeys.REMAINING_EDITOR_COMPLETION
+import ee.carlrobert.codegpt.settings.GeneralSettings
+import ee.carlrobert.codegpt.settings.service.ServiceType
+import ee.carlrobert.codegpt.settings.service.codegpt.CodeGPTServiceSettings
 import ee.carlrobert.codegpt.ui.OverlayUtil.showNotification
 import ee.carlrobert.codegpt.util.StringUtil
 import ee.carlrobert.llm.client.openai.completion.ErrorDetails
@@ -70,6 +74,13 @@ abstract class CodeCompletionEventListener(
     }
 
     override fun onError(error: ErrorDetails, ex: Throwable) {
+        val isCodeGPTService = GeneralSettings.getSelectedService() == ServiceType.CODEGPT
+        if (isCodeGPTService && "RATE_LIMIT_EXCEEDED" == error.code) {
+            service<CodeGPTServiceSettings>().state
+                .codeCompletionSettings
+                .codeCompletionsEnabled = false
+        }
+
         if (ex.message == null || (ex.message != null && ex.message != "Canceled")) {
             showNotification(error.message, NotificationType.ERROR)
             logger.error(error.message, ex)