From bf945cdf3e5ddf6021c3862da375947ad0ba5608 Mon Sep 17 00:00:00 2001
From: samgst-amazon <samgst@amazon.com>
Date: Thu, 24 Apr 2025 10:46:17 -0700
Subject: [PATCH] add styling to webview

---
 .../services/amazonq/webview/BrowserConnector.kt      | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

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
index 1e5bdb098b6..1afec29c15c 100644
--- 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
@@ -87,13 +87,6 @@ class BrowserConnector(
             .onEach { json ->
                 val node = serializer.toNode(json)
                 when (node.command) {
-                    "ui-is-ready" -> {
-                        uiReady.complete(true)
-                        RunOnceUtil.runOnceForApp("AmazonQ-UI-Ready") {
-                            MeetQSettings.getInstance().reinvent2024OnboardingCount += 1
-                        }
-                    }
-
                     "disclaimer-acknowledged" -> {
                         MeetQSettings.getInstance().disclaimerAcknowledged = true
                     }
@@ -235,6 +228,10 @@ class BrowserConnector(
             }
             CHAT_READY -> {
                 handleChatNotification<ChatReadyNotification, Unit>(node) { server, _ ->
+                    uiReady.complete(true)
+                    RunOnceUtil.runOnceForApp("AmazonQ-UI-Ready") {
+                        MeetQSettings.getInstance().reinvent2024OnboardingCount += 1
+                    }
                     server.chatReady()
                 }
             }