Partial chat results

Author: manodnyab

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/toolwindow/AmazonQToolWindow.kt

@@ -61,7 +61,9 @@ class AmazonQToolWindow private constructor(
             AsyncChatUiListener.TOPIC,
             object : AsyncChatUiListener {
                 override fun onChange(message: String) {
-                    chatPanel.browser?.postChat(message)
+                    runInEdt {
+                        chatPanel.browser?.postChat(message)
+                    }
                 }
             }
         )

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/webview/BrowserConnector.kt

@@ -166,18 +166,19 @@ class BrowserConnector(
                         )
                     )
                 )
+
+                val partialResultToken = chatCommunicationManager.addPartialChatMessage(requestFromUi.params.tabId)
                 val chatParams = ChatParams(
                     requestFromUi.params.tabId,
                     chatPrompt,
                     textDocumentIdentifier,
                     cursorState
-                )
-                val partialResultToken = chatCommunicationManager.addPartialChatMessage(requestFromUi.params.tabId)
+                    )
 
                 var encryptionManager: JwtEncryptionManager? = null
                 val result = AmazonQLspService.executeIfRunning(project) { server ->
                     encryptionManager = this.encryptionManager
-                    server.sendChatPrompt(EncryptedChatParams(encryptionManager!!.encrypt(chatParams)))
+                    server.sendChatPrompt(EncryptedChatParams(encryptionManager!!.encrypt(chatParams), partialResultToken))
                 } ?: (CompletableFuture.failedFuture(IllegalStateException("LSP Server not running")))
 
                 result.whenComplete {

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageClientImpl.kt

@@ -102,9 +102,8 @@ class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageC
         val chatCommunicationManager = ChatCommunicationManager.getInstance(project)
         try {
             chatCommunicationManager.handlePartialResultProgressNotification(project, params)
-            notifyInfo("hello")
         } catch (e: Exception) {
-            error("cannot handle partial chat")
+            error("Cannot handle partial chat")
         }
     }
 }

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/model/aws/chat/SendChatPrompt.kt

@@ -9,7 +9,7 @@ data class ChatParams(
     val tabId: String,
     val prompt: ChatPrompt,
     val textDocument: TextDocumentIdentifier,
-    val cursorState: CursorState,
+    val cursorState: CursorState
 )
 
 data class EncryptedChatParams(