From 68d56be7c0a1485d7cc0127452ab9c865e8cd70e Mon Sep 17 00:00:00 2001
From: Ilango <rajagopalilango@gmail.com>
Date: Thu, 22 Feb 2024 16:13:46 +0530
Subject: [PATCH] Remove related feedback and update stats when clearing chat
 in view messages

---
 packages/server/src/index.ts                              | 4 ++++
 packages/ui/src/ui-component/dialog/ViewMessagesDialog.js | 1 +
 2 files changed, 5 insertions(+)

diff --git a/packages/server/src/index.ts b/packages/server/src/index.ts
index ab998a15c..abfc66ad3 100644
--- a/packages/server/src/index.ts
+++ b/packages/server/src/index.ts
@@ -604,6 +604,10 @@ export class App {
             if (sessionId) deleteOptions.sessionId = sessionId
             if (chatType) deleteOptions.chatType = chatType
 
+            // remove all related feedback records
+            const feedbackDeleteOptions: FindOptionsWhere<ChatMessageFeedback> = { chatId }
+            await this.AppDataSource.getRepository(ChatMessageFeedback).delete(feedbackDeleteOptions)
+
             const results = await this.AppDataSource.getRepository(ChatMessage).delete(deleteOptions)
             return res.json(results)
         })
diff --git a/packages/ui/src/ui-component/dialog/ViewMessagesDialog.js b/packages/ui/src/ui-component/dialog/ViewMessagesDialog.js
index f52723b2f..a9490d273 100644
--- a/packages/ui/src/ui-component/dialog/ViewMessagesDialog.js
+++ b/packages/ui/src/ui-component/dialog/ViewMessagesDialog.js
@@ -214,6 +214,7 @@ const ViewMessagesDialog = ({ show, dialogProps, onCancel }) => {
                     }
                 })
                 getChatmessageApi.request(chatflowid)
+                getStatsApi.request(chatflowid) // update stats
             } catch (error) {
                 const errorData = error.response.data || `${error.response.status}: ${error.response.statusText}`
                 enqueueSnackbar({