feat(amazonq): clean up and delete all sessions storage on profileChanged (#2004)

evanliu048 · Will-ShaoHua · web-flow · commit 8993ccaf44e8 · 2025-04-04T15:04:29.000-07:00
* add sessionClear

* change listener to project level

* change listener param

* fx

* ensure dev/test/review work

* linter

* hack

* update endpoint

* delete loh

* revert transform change

* cr

* update session clear

* leave stopModernize in  codeModernizerManager

---------

Co-authored-by: Will Lo &lt;xshaohua@amazon.com&gt;
Co-authored-by: Will Lo &lt;96078566+Will-ShaoHua@users.noreply.github.com&gt;
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeScan/CodeScanChatApp.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeScan/CodeScanChatApp.kt
@@ -21,6 +21,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.auth.isCodeScanAvailable
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.commands.CodeScanActionMessage
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.commands.CodeScanMessageListener
@@ -141,6 +143,15 @@ class CodeScanChatApp(private val scope: CoroutineScope) : AmazonQApp {
                 }
             }
         )
+
+        context.project.messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    chatSessionStorage.deleteAllSessions()
+                }
+            }
+        )
     }
 
     private fun getQTokenProvider(project: Project) = (
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeScan/storage/ChatSessionStorage.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeScan/storage/ChatSessionStorage.kt
@@ -30,4 +30,8 @@ class ChatSessionStorage {
     fun changeAuthenticationNeededNotified(authNeededNotified: Boolean) {
         sessions.keys.forEach { sessions[it]?.authNeededNotified = authNeededNotified }
     }
+
+    fun deleteAllSessions() {
+        sessions.clear()
+    }
 }
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeTestChatApp.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeTestChatApp.kt
@@ -4,13 +4,16 @@
 package software.aws.toolkits.jetbrains.services.amazonqCodeTest
 
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.project.Project
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.launch
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnection
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManagerListener
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.auth.isCodeScanAvailable
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.auth.isCodeTestAvailable
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.controller.CodeTestChatController
@@ -71,6 +74,15 @@ class CodeTestChatApp(private val scope: CoroutineScope) : AmazonQApp {
                 }
             }
         )
+
+        context.project.messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    chatSessionStorage.deleteAllSessions()
+                }
+            }
+        )
     }
 
     private suspend fun handleMessage(message: AmazonQMessage, inboundAppMessagesHandler: InboundAppMessagesHandler) {
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeWhispererUTGChatManager.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeWhispererUTGChatManager.kt
@@ -541,8 +541,8 @@ Please see the unit tests generated below. Click 'View Diff' to review the chang
                     jobGroup = session.testGenerationJobGroupName,
                     jobId = session.testGenerationJob,
                     result = if (e.message == message("testgen.message.cancelled")) MetricResult.Cancelled else MetricResult.Failed,
-                    reason = (e as CodeTestException).code ?: "DefaultError",
-                    reasonDesc = if (e.message == message("testgen.message.cancelled")) "${e.code}: ${e.message}" else e.message,
+                    reason = (e as? CodeTestException)?.code ?: "DefaultError",
+                    reasonDesc = if (e.message == message("testgen.message.cancelled")) "${(e as? CodeTestException)?.code}: ${e.message}" else e.message,
                     perfClientLatency = (Instant.now().toEpochMilli() - session.startTimeOfTestGeneration),
                     isCodeBlockSelected = session.isCodeBlockSelected,
                     artifactsUploadDuration = session.artifactUploadDuration,
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/storage/ChatSessionStorage.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/storage/ChatSessionStorage.kt
@@ -17,4 +17,8 @@ class ChatSessionStorage {
 
     // Find all sessions that are currently waiting to be authenticated
     fun getAuthenticatingSessions(): List<Session> = this.sessions.values.filter { it.isAuthenticating }
+
+    fun deleteAllSessions() {
+        sessions.clear()
+    }
 }
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/DocApp.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/DocApp.kt
@@ -4,13 +4,16 @@
 package software.aws.toolkits.jetbrains.services.amazonqDoc
 
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.project.Project
 import kotlinx.coroutines.launch
 import software.aws.toolkits.jetbrains.core.coroutines.disposableCoroutineScope
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnection
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManagerListener
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.auth.isCodeScanAvailable
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.auth.isCodeTestAvailable
 import software.aws.toolkits.jetbrains.services.amazonqDoc.auth.isDocAvailable
@@ -74,6 +77,15 @@ class DocApp : AmazonQApp {
                 }
             }
         )
+
+        context.project.messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    chatSessionStorage.deleteAllSessions()
+                }
+            }
+        )
     }
 
     private suspend fun handleMessage(message: AmazonQMessage, inboundAppMessagesHandler: InboundAppMessagesHandler) {
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/storage/ChatSessionStorage.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/storage/ChatSessionStorage.kt
@@ -23,4 +23,11 @@ class ChatSessionStorage {
 
     // Find all sessions that are currently waiting to be authenticated
     fun getAuthenticatingSessions(): List<DocSession> = this.sessions.values.filter { it.isAuthenticating }
+
+    fun deleteAllSessions() {
+        sessions.values.forEach { session ->
+            session.sessionState.token?.cancel()
+        }
+        sessions.clear()
+    }
 }
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/FeatureDevApp.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/FeatureDevApp.kt
@@ -4,13 +4,16 @@
 package software.aws.toolkits.jetbrains.services.amazonqFeatureDev
 
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.project.Project
 import kotlinx.coroutines.launch
 import software.aws.toolkits.jetbrains.core.coroutines.disposableCoroutineScope
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnection
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManagerListener
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.auth.isCodeScanAvailable
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.auth.isCodeTestAvailable
 import software.aws.toolkits.jetbrains.services.amazonqDoc.auth.isDocAvailable
@@ -75,6 +78,15 @@ class FeatureDevApp : AmazonQApp {
                 }
             }
         )
+
+        context.project.messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    chatSessionStorage.deleteAllSessions()
+                }
+            }
+        )
     }
 
     private suspend fun handleMessage(message: AmazonQMessage, inboundAppMessagesHandler: InboundAppMessagesHandler) {
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/controller/FeatureDevController.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/controller/FeatureDevController.kt
@@ -32,6 +32,8 @@ import software.aws.toolkits.jetbrains.core.coroutines.EDT
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController
 import software.aws.toolkits.jetbrains.services.amazonq.messages.MessagePublisher
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonq.project.RepoSizeError
 import software.aws.toolkits.jetbrains.services.amazonq.toolwindow.AmazonQToolWindowFactory
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.CodeIterationLimitException
@@ -94,6 +96,17 @@ class FeatureDevController(
     private val authController: AuthController = AuthController(),
 ) : InboundAppMessagesHandler {
 
+    init {
+        context.project.messageBus.connect().subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    chatSessionStorage.deleteAllSessions()
+                }
+            }
+        )
+    }
+
     val messenger = context.messagesFromAppToUi
     val toolWindow = ToolWindowManager.getInstance(context.project).getToolWindow(AmazonQToolWindowFactory.WINDOW_ID)
 
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/storage/ChatSessionStorage.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/storage/ChatSessionStorage.kt
@@ -23,4 +23,11 @@ class ChatSessionStorage {
 
     // Find all sessions that are currently waiting to be authenticated
     fun getAuthenticatingSessions(): List<Session> = this.sessions.values.filter { it.isAuthenticating }
+
+    fun deleteAllSessions() {
+        sessions.values.forEach { session ->
+            session.sessionState.token?.cancel()
+        }
+        sessions.clear()
+    }
 }
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/App.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/App.kt
@@ -4,6 +4,7 @@
 package software.aws.toolkits.jetbrains.services.cwc
 
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.project.Project
 import kotlinx.coroutines.flow.merge
 import kotlinx.coroutines.launch
 import software.aws.toolkits.jetbrains.core.coroutines.disposableCoroutineScope
@@ -12,6 +13,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
 import software.aws.toolkits.jetbrains.services.amazonq.onboarding.OnboardingPageInteraction
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonq.util.highlightCommand
 import software.aws.toolkits.jetbrains.services.cwc.commands.ActionRegistrar
 import software.aws.toolkits.jetbrains.services.cwc.commands.CodeScanIssueActionMessage
@@ -75,6 +78,15 @@ class App : AmazonQApp {
                 }
             }
         )
+
+        ApplicationManager.getApplication().messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    inboundAppMessagesHandler.processSessionClear()
+                }
+            }
+        )
     }
 
     private suspend fun handleMessage(message: AmazonQMessage, inboundAppMessagesHandler: InboundAppMessagesHandler) {
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/InboundAppMessagesHandler.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/InboundAppMessagesHandler.kt
@@ -34,4 +34,6 @@ interface InboundAppMessagesHandler {
     suspend fun processCodeScanIssueAction(message: CodeScanIssueActionMessage)
 
     suspend fun processLinkClick(message: IncomingCwcMessage.ClickedLink)
+
+    fun processSessionClear()
 }
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt
@@ -541,6 +541,10 @@ class ChatController private constructor(
         .map { it.tabId }
         .first()
 
+    override fun processSessionClear() {
+        chatSessionStorage.deleteAllSessions()
+    }
+
     companion object {
         private val logger = getLogger<ChatController>()
 
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/inline/InlineChatController.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/inline/InlineChatController.kt
@@ -55,6 +55,8 @@ import software.aws.toolkits.jetbrains.core.gettingstarted.requestCredentialsFor
 import software.aws.toolkits.jetbrains.core.webview.BrowserState
 import software.aws.toolkits.jetbrains.services.amazonq.QWebviewPanel
 import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonq.toolwindow.AMAZON_Q_WINDOW_ID
 import software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererModelConfigurator
 import software.aws.toolkits.jetbrains.services.codewhisperer.model.CaretPosition
@@ -101,6 +103,14 @@ class InlineChatController(
     init {
         Disposer.register(this, listener)
         project.messageBus.connect(this).subscribe(FileEditorManagerListener.FILE_EDITOR_MANAGER, listener)
+        project.messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    sessionStorage.deleteAllSessions()
+                }
+            }
+        )
     }
 
     data class InlineChatMetrics(
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/storage/ChatSessionStorage.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/storage/ChatSessionStorage.kt
@@ -26,4 +26,8 @@ class ChatSessionStorage(
     fun deleteSession(tabId: String) {
         sessions.remove(tabId)?.scope?.cancel()
     }
+
+    fun deleteAllSessions() {
+        sessions.clear()
+    }
 }
diff --git a/plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/CodeModernizerManager.kt b/plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/CodeModernizerManager.kt
@@ -144,6 +144,7 @@ class CodeModernizerManager(private val project: Project) : PersistentStateCompo
                     codeTransformationSession?.let {
                         Disposer.dispose(it)
                     }
+                    managerState = CodeModernizerState()
                     codeTransformationSession = null
                 }
             }
diff --git a/plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/CodeTransformChatApp.kt b/plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/CodeTransformChatApp.kt
@@ -4,6 +4,7 @@
 package software.aws.toolkits.jetbrains.services.codemodernizer
 
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.project.Project
 import kotlinx.coroutines.flow.merge
 import kotlinx.coroutines.launch
 import software.aws.toolkits.jetbrains.core.coroutines.disposableCoroutineScope
@@ -15,6 +16,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
 import software.aws.toolkits.jetbrains.services.amazonqCodeScan.auth.isCodeScanAvailable
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.auth.isCodeTestAvailable
 import software.aws.toolkits.jetbrains.services.amazonqDoc.auth.isDocAvailable
@@ -161,6 +164,15 @@ class CodeTransformChatApp : AmazonQApp {
                 }
             }
         )
+
+        context.project.messageBus.connect(this).subscribe(
+            QRegionProfileSelectedListener.TOPIC,
+            object : QRegionProfileSelectedListener {
+                override fun onProfileSelected(project: Project, profile: QRegionProfile?) {
+                    chatSessionStorage.deleteAllSessions()
+                }
+            }
+        )
     }
 
     private suspend fun handleMessage(message: AmazonQMessage, inboundAppMessagesHandler: InboundAppMessagesHandler) {
diff --git a/plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/session/ChatSessionStorage.kt b/plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/session/ChatSessionStorage.kt
@@ -28,4 +28,8 @@ class ChatSessionStorage {
     fun changeAuthenticationNeededNotified(authNeededNotified: Boolean) {
         sessions.keys.forEach { sessions[it]?.authNeededNotified = authNeededNotified }
     }
+
+    fun deleteAllSessions() {
+        sessions.clear()
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQApp`
`21`	`21`	`import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext`
`22`	`22`	`import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController`
`23`	`23`	`import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage`
	`24`	`+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile`
	`25`	`+import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener`
`24`	`26`	`import software.aws.toolkits.jetbrains.services.amazonqCodeScan.auth.isCodeScanAvailable`
`25`	`27`	`import software.aws.toolkits.jetbrains.services.amazonqCodeScan.commands.CodeScanActionMessage`
`26`	`28`	`import software.aws.toolkits.jetbrains.services.amazonqCodeScan.commands.CodeScanMessageListener`
`@@ -141,6 +143,15 @@ class CodeScanChatApp(private val scope: CoroutineScope) : AmazonQApp {`
`141`	`143`	`}`
`142`	`144`	`}`
`143`	`145`	`)`
	`146`	`+`
	`147`	`+ context.project.messageBus.connect(this).subscribe(`
	`148`	`+ QRegionProfileSelectedListener.TOPIC,`
	`149`	`+ object : QRegionProfileSelectedListener {`
	`150`	`+ override fun onProfileSelected(project: Project, profile: QRegionProfile?) {`
	`151`	`+ chatSessionStorage.deleteAllSessions()`
	`152`	`+ }`
	`153`	`+ }`
	`154`	`+ )`
`144`	`155`	`}`
`145`	`156`
`146`	`157`	`private fun getQTokenProvider(project: Project) = (`
Original file line number	Diff line number	Diff line change
`@@ -30,4 +30,8 @@ class ChatSessionStorage {`
`30`	`30`	`fun changeAuthenticationNeededNotified(authNeededNotified: Boolean) {`
`31`	`31`	`sessions.keys.forEach { sessions[it]?.authNeededNotified = authNeededNotified }`
`32`	`32`	`}`
	`33`	`+`
	`34`	`+ fun deleteAllSessions() {`
	`35`	`+ sessions.clear()`
	`36`	`+ }`
`33`	`37`	`}`
Original file line number	Diff line number	Diff line change
`@@ -17,4 +17,8 @@ class ChatSessionStorage {`
`17`	`17`
`18`	`18`	`// Find all sessions that are currently waiting to be authenticated`
`19`	`19`	`fun getAuthenticatingSessions(): List<Session> = this.sessions.values.filter { it.isAuthenticating }`
	`20`	`+`
	`21`	`+ fun deleteAllSessions() {`
	`22`	`+ sessions.clear()`
	`23`	`+ }`
`20`	`24`	`}`