remove tabID from ChatCommunicationManager

samgst-amazon · samgst-amazon · commit c93517130b46 · 2025-04-14T11:37:52.000-07:00
diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/webview/BrowserConnector.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/webview/BrowserConnector.kt
@@ -210,11 +210,13 @@ class BrowserConnector(
             }
             SEND_END_CHAT -> {
                 val requestFromUi = serializer.deserializeChatMessages(node, EndChatParams::class.java)
-
                 val endChatParams = EndChatParams(requestFromUi.tabId)
                 val result = AmazonQLspService.executeIfRunning(project) { server ->
                     server.endChat(endChatParams)
                 } ?: CompletableFuture.failedFuture(IllegalStateException("LSP Server not running"))
+                result.whenComplete { value, error ->
+                    chatCommunicationManager.removePartialChatMessage(requestFromUi.tabId)
+                }
             }
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -210,11 +210,13 @@ class BrowserConnector(`
`210`	`210`	`}`
`211`	`211`	`SEND_END_CHAT -> {`
`212`	`212`	`val requestFromUi = serializer.deserializeChatMessages(node, EndChatParams::class.java)`
`213`		`-`
`214`	`213`	`val endChatParams = EndChatParams(requestFromUi.tabId)`
`215`	`214`	`val result = AmazonQLspService.executeIfRunning(project) { server ->`
`216`	`215`	`server.endChat(endChatParams)`
`217`	`216`	`} ?: CompletableFuture.failedFuture(IllegalStateException("LSP Server not running"))`
	`217`	`+ result.whenComplete { value, error ->`
	`218`	`+ chatCommunicationManager.removePartialChatMessage(requestFromUi.tabId)`
	`219`	`+ }`
`218`	`220`	`}`
`219`	`221`	`}`
`220`	`222`	`}`