From e3c059aff4e0d4e6557a637435c344ccc976cd08 Mon Sep 17 00:00:00 2001
From: Tai Lai <tlaiamz@amazon.com>
Date: Tue, 22 Apr 2025 13:21:17 -0700
Subject: [PATCH] fix(amazonq): forward chat update notification

---
 packages/amazonq/src/lsp/chat/messages.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/packages/amazonq/src/lsp/chat/messages.ts b/packages/amazonq/src/lsp/chat/messages.ts
index c82d4c24570..3dddd12ab8c 100644
--- a/packages/amazonq/src/lsp/chat/messages.ts
+++ b/packages/amazonq/src/lsp/chat/messages.ts
@@ -49,6 +49,8 @@ import {
     buttonClickRequestType,
     ButtonClickResult,
     CancellationTokenSource,
+    chatUpdateNotificationType,
+    ChatUpdateParams,
 } from '@aws/language-server-runtimes/protocol'
 import { v4 as uuidv4 } from 'uuid'
 import * as vscode from 'vscode'
@@ -463,6 +465,13 @@ export function registerMessageListeners(
             true
         )
     })
+
+    languageClient.onNotification(chatUpdateNotificationType.method, (params: ChatUpdateParams) => {
+        void provider.webview?.postMessage({
+            command: chatUpdateNotificationType.method,
+            params: params,
+        })
+    })
 }
 
 function isServerEvent(command: string) {