fix(q): Handle monthly conversation limit error for QFeature development (aws#4373)

Author: kumsmrit

.changes/next-release/bugfix-0127ba02-eaca-44c8-bb4a-83ebd3213328.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Amazon Q Feature Development: Update error message for monthly conversation limit reach"
+}

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/FeatureDevExceptions.kt

@@ -13,6 +13,8 @@ class PlanIterationLimitError(override val message: String, override val cause:
 
 class CodeIterationLimitError(override val message: String, override val cause: Throwable?) : RuntimeException()
 
+class MonthlyConversationLimitError(override val message: String, override val cause: Throwable?) : RuntimeException()
+
 internal fun codeGenerationFailedError(): Nothing =
     throw FeatureDevException(message("amazonqFeatureDev.code_generation.failed_generation"))
 

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/controller/FeatureDevController.kt

@@ -33,6 +33,7 @@ import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ContentLengthE
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.DEFAULT_RETRY_LIMIT
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FEATURE_NAME
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.InboundAppMessagesHandler
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.MonthlyConversationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.PlanIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.createUserFacingErrorMessage
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.FeatureDevMessageType
@@ -49,6 +50,7 @@ import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendA
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendAuthenticationInProgressMessage
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendChatInputEnabledMessage
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendError
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendMonthlyLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendSystemPrompt
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.sendUpdatePlaceholder
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.updateFileComponent
@@ -418,6 +420,9 @@ class FeatureDevController(
                         )
                     ),
                 )
+            } else if (err is MonthlyConversationLimitError) {
+                messenger.sendMonthlyLimitError(tabId = tabId)
+                messenger.sendChatInputEnabledMessage(tabId, enabled = false)
             } else if (err is PlanIterationLimitError) {
                 messenger.sendError(tabId = tabId, errMessage = err.message, retries = retriesRemaining(session))
                 messenger.sendSystemPrompt(

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/messages/FeatureDevMessagePublisherExtensions.kt

@@ -169,6 +169,15 @@ suspend fun MessagePublisher.sendError(tabId: String, errMessage: String, retrie
     )
 }
 
+suspend fun MessagePublisher.sendMonthlyLimitError(tabId: String) {
+    this.sendAnswer(
+        tabId = tabId,
+        messageType = FeatureDevMessageType.Answer,
+        message = message("amazonqFeatureDev.exception.monthly_limit_error")
+    )
+    this.sendUpdatePlaceholder(tabId = tabId, newPlaceholder = message("amazonqFeatureDev.placeholder.after_monthly_limit"))
+}
+
 suspend fun MessagePublisher.initialExamples(tabId: String) {
     this.sendAnswer(
         tabId = tabId,

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/util/FeatureDevClientUtil.kt

@@ -19,6 +19,7 @@ import software.aws.toolkits.core.utils.warn
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.CodeIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ContentLengthError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FEATURE_NAME
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.MonthlyConversationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.PlanIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.apiError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.clients.FeatureDevClient
@@ -50,6 +51,10 @@ fun createConversation(proxyClient: FeatureDevClient): String {
         if (e is CodeWhispererRuntimeException) {
             errMssg = e.awsErrorDetails().errorMessage()
             logger.warn(e) { "Start conversation failed for request: ${e.requestId()}" }
+
+            if (e is software.amazon.awssdk.services.codewhispererruntime.model.ServiceQuotaExceededException) {
+                throw MonthlyConversationLimitError(errMssg, e.cause)
+            }
         }
         apiError(errMssg, e.cause)
     }

plugins/toolkit/resources/resources/software/aws/toolkits/resources/MessagesBundle.properties

@@ -61,6 +61,7 @@ amazonqFeatureDev.exception.conversation_not_found=Conversation id must exist be
 amazonqFeatureDev.exception.export_parsing_error=Downloaded code results could not be parsed
 amazonqFeatureDev.exception.insert_code_failed=Failed to insert code changes
 amazonqFeatureDev.exception.message_not_found=Message was not found
+amazonqFeatureDev.exception.monthly_limit_error=Sorry, you have reached the monthly limit for feature development. You can try again next month.
 amazonqFeatureDev.exception.open_diff_failed=Failed to open diff
 amazonqFeatureDev.exception.request_failed=Request failed
 amazonqFeatureDev.exception.retry_request_failed=Retry request failed
@@ -80,6 +81,7 @@ amazonqFeatureDev.follow_up.send_feedback=Send feedback
 amazonqFeatureDev.no_retries.error_text=Sorry, we're unable to provide a response at this time. Please try again later or share feedback with our team to help us troubleshoot.
 amazonqFeatureDev.placeholder.additional_improvements=Provide input on additional improvements
 amazonqFeatureDev.placeholder.after_code_generation=Select an option above to proceed
+amazonqFeatureDev.placeholder.after_monthly_limit=Chat input is disabled
 amazonqFeatureDev.placeholder.closed_session=Your session is now closed
 amazonqFeatureDev.placeholder.generating_approach=Generating plan...
 amazonqFeatureDev.placeholder.generating_code=Generating code...