@@ -28,19 +28,27 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.AmazonQLspService
2828import software.aws.toolkits.jetbrains.services.amazonq.lsp.encryption.JwtEncryptionManager
2929import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.ChatCommunicationManager
3030import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.getTextDocumentIdentifier
31+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_FEEDBACK
32+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_FOLLOW_UP_CLICK
3133import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_INFO_LINK_CLICK
3234import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_LINK_CLICK
3335import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_QUICK_ACTION
36+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_READY
3437import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_SOURCE_LINK_CLICK
3538import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_ADD
3639import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_CHANGE
3740import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_REMOVE
3841import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatNotification
3942import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatParams
4043import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatPrompt
44+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatReadyNotification
4145import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CursorState
4246import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.EncryptedChatParams
4347import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.EncryptedQuickActionChatParams
48+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.FeedbackNotification
49+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.FeedbackParams
50+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.FollowUpClickNotification
51+ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.FollowUpClickParams
4452import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.InfoLinkClickNotification
4553import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.InfoLinkClickParams
4654import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.LinkClickNotification
@@ -220,6 +228,16 @@ class BrowserConnector(
220228
221229 showResult(result, partialResultToken, tabId, encryptionManager, browser)
222230 }
231+ CHAT_FEEDBACK -> {
232+ handleChatNotification<FeedbackNotification , FeedbackParams >(node) { server, params ->
233+ server.feedback(params)
234+ }
235+ }
236+ CHAT_READY -> {
237+ handleChatNotification<ChatReadyNotification , Unit >(node) { server, _ ->
238+ server.chatReady()
239+ }
240+ }
223241 CHAT_TAB_ADD -> {
224242 handleChatNotification<TabEventRequest , TabEventParams >(node) { server, params ->
225243 server.tabAdd(params)
@@ -253,6 +271,11 @@ class BrowserConnector(
253271 server.sourceLinkClick(params)
254272 }
255273 }
274+ CHAT_FOLLOW_UP_CLICK -> {
275+ handleChatNotification<FollowUpClickNotification , FollowUpClickParams >(node) { server, params ->
276+ server.followUpClick(params)
277+ }
278+ }
256279 }
257280 }
258281
0 commit comments