@@ -178,7 +178,11 @@ async function replaceDuplicateIdsForChatMessage(queryRunner: QueryRunner, origi
178
178
const chatmessageChatflowIds = chatMessages . map ( ( chatMessage ) => {
179
179
return { id : chatMessage . chatflowid , qty : 0 }
180
180
} )
181
- const originalDataChatflowIds = originalData . ChatFlow . map ( ( chatflow ) => chatflow . id )
181
+ const originalDataChatflowIds = [
182
+ ...originalData . AssistantFlow . map ( ( assistantFlow ) => assistantFlow . id ) ,
183
+ ...originalData . AgentFlow . map ( ( agentflow ) => agentflow . id ) ,
184
+ ...originalData . ChatFlow . map ( ( chatflow ) => chatflow . id )
185
+ ]
182
186
chatmessageChatflowIds . forEach ( ( item ) => {
183
187
if ( originalDataChatflowIds . includes ( item . id ) ) {
184
188
item . qty += 1
@@ -229,7 +233,11 @@ async function replaceDuplicateIdsForChatMessageFeedback(
229
233
const feedbackChatflowIds = chatMessageFeedbacks . map ( ( feedback ) => {
230
234
return { id : feedback . chatflowid , qty : 0 }
231
235
} )
232
- const originalDataChatflowIds = originalData . ChatFlow . map ( ( chatflow ) => chatflow . id )
236
+ const originalDataChatflowIds = [
237
+ ...originalData . AssistantFlow . map ( ( assistantFlow ) => assistantFlow . id ) ,
238
+ ...originalData . AgentFlow . map ( ( agentflow ) => agentflow . id ) ,
239
+ ...originalData . ChatFlow . map ( ( chatflow ) => chatflow . id )
240
+ ]
233
241
feedbackChatflowIds . forEach ( ( item ) => {
234
242
if ( originalDataChatflowIds . includes ( item . id ) ) {
235
243
item . qty += 1
0 commit comments