@@ -41,6 +41,7 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.JsonRpcNotification
4141import software.aws.toolkits.jetbrains.services.amazonq.lsp.JsonRpcRequest
4242import software.aws.toolkits.jetbrains.services.amazonq.lsp.encryption.JwtEncryptionManager
4343import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.AwsServerCapabilitiesProvider
44+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatAsyncResultManager
4445import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatCommunicationManager
4546import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.FlareUiMessage
4647import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.AUTH_FOLLOW_UP_CLICKED
@@ -98,7 +99,6 @@ import software.aws.toolkits.telemetry.Telemetry
9899import java.util.concurrent.CompletableFuture
99100import java.util.concurrent.CompletionException
100101import java.util.function.Function
101- import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatAsyncResultManager
102102
103103class BrowserConnector (
104104 private val serializer : MessageSerializer = MessageSerializer .getInstance(),
@@ -231,7 +231,6 @@ class BrowserConnector(
231231 val partialResultToken = chatCommunicationManager.addPartialChatMessage(tabId)
232232 chatCommunicationManager.registerPartialResultToken(partialResultToken)
233233
234-
235234 var encryptionManager: JwtEncryptionManager ? = null
236235 val result = AmazonQLspService .executeIfRunning(project) { server ->
237236 encryptionManager = this .encryptionManager
@@ -483,7 +482,7 @@ class BrowserConnector(
483482 chatCommunicationManager.removeInflightRequestForTab(tabId)
484483 } catch (e: CancellationException ) {
485484 LOG .warn { " Cancelled chat generation" }
486- try {
485+ try {
487486 chatAsyncResultManager.createRequestId(partialResultToken)
488487 chatAsyncResultManager.getResult(partialResultToken)
489488 handleCancellation(tabId, partialResultToken, browser)
@@ -500,7 +499,7 @@ class BrowserConnector(
500499 }
501500 }
502501 }
503-
502+
504503 private fun handleCancellation (tabId : String , partialResultToken : String , browser : Browser ) {
505504 // Send a message to hide the stop button without showing an error
506505 val cancelMessage = chatCommunicationManager.getCancellationUiMessage(tabId)
0 commit comments