From 82e6f43b5cfce5b6334628751d8a1eed378a3ed3 Mon Sep 17 00:00:00 2001
From: Ong Chung Yau <33013947+chungyau97@users.noreply.github.com>
Date: Thu, 8 May 2025 18:43:28 +0800
Subject: [PATCH] fix/import-missing-chat-data (#4401)

fix: add missing imports for messages and feedbacks in AgentFlow and AssistantFlow
---
 packages/server/src/services/export-import/index.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/packages/server/src/services/export-import/index.ts b/packages/server/src/services/export-import/index.ts
index 90317d8fd..791e8a8e6 100644
--- a/packages/server/src/services/export-import/index.ts
+++ b/packages/server/src/services/export-import/index.ts
@@ -178,7 +178,11 @@ async function replaceDuplicateIdsForChatMessage(queryRunner: QueryRunner, origi
         const chatmessageChatflowIds = chatMessages.map((chatMessage) => {
             return { id: chatMessage.chatflowid, qty: 0 }
         })
-        const originalDataChatflowIds = originalData.ChatFlow.map((chatflow) => chatflow.id)
+        const originalDataChatflowIds = [
+            ...originalData.AssistantFlow.map((assistantFlow) => assistantFlow.id),
+            ...originalData.AgentFlow.map((agentflow) => agentflow.id),
+            ...originalData.ChatFlow.map((chatflow) => chatflow.id)
+        ]
         chatmessageChatflowIds.forEach((item) => {
             if (originalDataChatflowIds.includes(item.id)) {
                 item.qty += 1
@@ -229,7 +233,11 @@ async function replaceDuplicateIdsForChatMessageFeedback(
         const feedbackChatflowIds = chatMessageFeedbacks.map((feedback) => {
             return { id: feedback.chatflowid, qty: 0 }
         })
-        const originalDataChatflowIds = originalData.ChatFlow.map((chatflow) => chatflow.id)
+        const originalDataChatflowIds = [
+            ...originalData.AssistantFlow.map((assistantFlow) => assistantFlow.id),
+            ...originalData.AgentFlow.map((agentflow) => agentflow.id),
+            ...originalData.ChatFlow.map((chatflow) => chatflow.id)
+        ]
         feedbackChatflowIds.forEach((item) => {
             if (originalDataChatflowIds.includes(item.id)) {
                 item.qty += 1