add sendChatUpdate

Author: samgst-amazon

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

@@ -3,10 +3,13 @@
 
 package software.aws.toolkits.jetbrains.services.amazonq.lsp
 
+import org.eclipse.lsp4j.jsonrpc.services.JsonNotification
 import org.eclipse.lsp4j.jsonrpc.services.JsonRequest
 import org.eclipse.lsp4j.services.LanguageClient
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.OpenTabParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.OpenTabResult
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_SEND_UPDATE
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatUpdateParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.ConnectionMetadata
 import java.util.concurrent.CompletableFuture
 
@@ -20,4 +23,7 @@ interface AmazonQLanguageClient : LanguageClient {
 
     @JsonRequest("aws/chat/openTab")
     fun openTab(params: OpenTabParams): CompletableFuture<OpenTabResult>
+
+    @JsonNotification(CHAT_SEND_UPDATE)
+    fun sendChatUpdate(params: ChatUpdateParams) : CompletableFuture<Unit>
 }

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

@@ -16,9 +16,12 @@ import org.eclipse.lsp4j.ShowMessageRequestParams
 import software.aws.toolkits.jetbrains.core.credentials.AwsBearerTokenConnection
 import software.aws.toolkits.jetbrains.core.credentials.ToolkitConnectionManager
 import software.aws.toolkits.jetbrains.core.credentials.pinning.QConnection
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.AsyncChatUiListener
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatCommunicationManager
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatUpdateParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.OpenTabParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.OpenTabResult
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SEND_CHAT_COMMAND_PROMPT
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.ConnectionMetadata
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.SsoProfileData
 import software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererModelConfigurator
@@ -130,4 +133,25 @@ class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageC
             error("Cannot handle partial chat")
         }
     }
+
+    override fun sendChatUpdate(params: ChatUpdateParams): CompletableFuture<Unit> {
+        // Process the chat update notification from the server
+        // This notification is used to add or update messages in a specific tab
+        val tabId = params.tabId
+        val state = params.state
+        val data = params.data
+
+        val encryptionManager = AmazonQLspService.getInstance(project).encryptionManager
+
+        val uiMessage = ChatCommunicationManager.convertToJsonToSendToChat(
+            command = SEND_CHAT_COMMAND_PROMPT,
+            tabId = tabId,
+            params = encryptionManager.decrypt(params.toString()),
+            isPartialResult = false
+        )
+
+        AsyncChatUiListener.notifyPartialMessageUpdate(uiMessage)
+        
+        return CompletableFuture.completedFuture(Unit)
+    }
 }

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

@@ -0,0 +1,10 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat
+
+data class ChatUpdateParams(
+    val tabId: String,
+    val state: TabState? = null,
+    val data: TabData? = null
+)

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

@@ -18,3 +18,4 @@ const val CHAT_TAB_CHANGE = "aws/chat/tabChange"
 const val CHAT_TAB_REMOVE = "aws/chat/tabRemove"
 const val CHAT_OPEN_TAB = "aws/chat/openTab"
 const val CHAT_BUTTON_CLICK = "aws/chat/buttonClick"
+const val CHAT_SEND_UPDATE = "aws/chat/sendChatUpdate"