Fixing context to show in chat

Author: LokeshDogga13

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

@@ -137,10 +137,16 @@ class ChatCommunicationManager(private val project: Project, private val cs: Cor
         val encryptedPartialChatResult = getObject(params, String::class.java)
         if (encryptedPartialChatResult != null) {
             val partialChatResult = AmazonQLspService.getInstance(project).encryptionManager.decrypt(encryptedPartialChatResult)
+            val partialResultMap: Any = tryOrNull {
+                Gson().fromJson(partialChatResult, Map::class.java)
+            } ?: partialChatResult
+
             notifyUi(
                 FlareUiMessage(
                     command = SEND_CHAT_COMMAND_PROMPT,
-                    params = partialChatResult,
+                    tabId = tabId,
+                    params = partialResultMap,
+                    isPartialResult = true
                 )
             )
         }

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

@@ -7,4 +7,6 @@ data class FlareUiMessage(
     val command: String,
     val params: Any,
     val requestId: String? = null,
+    val tabId: String? = null,
+    val isPartialResult: Boolean? = false
 )