Adding session cleanup after auth re-prompt

ashishrp-aws · ashishrp-aws · commit a9a5706a50b5 · 2025-02-11T15:38:59.000-08:00
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt
@@ -959,6 +959,7 @@ class CodeTestChatController(
         authController.handleAuth(context.project, message.authType)
         codeTestChatHelper.sendAuthenticationInProgressMessage(message.tabId) // show user that authentication is in progress
         codeTestChatHelper.sendChatInputEnabledMessage(false) // disable the input field while authentication is in progress
+        sessionCleanUp(codeTestChatHelper.getActiveSession().tabId)
     }
 
     private suspend fun updateBuildAndExecuteProgressCard(

Original file line number	Diff line number	Diff line change
`@@ -959,6 +959,7 @@ class CodeTestChatController(`
`959`	`959`	`authController.handleAuth(context.project, message.authType)`
`960`	`960`	`codeTestChatHelper.sendAuthenticationInProgressMessage(message.tabId) // show user that authentication is in progress`
`961`	`961`	`codeTestChatHelper.sendChatInputEnabledMessage(false) // disable the input field while authentication is in progress`
	`962`	`+ sessionCleanUp(codeTestChatHelper.getActiveSession().tabId)`
`962`	`963`	`}`
`963`	`964`
`964`	`965`	`private suspend fun updateBuildAndExecuteProgressCard(`