From 672b8f14f2a6b4e7e08134a01bf7e8eb06d15db8 Mon Sep 17 00:00:00 2001
From: StyleZhang <jasonapring2015@outlook.com>
Date: Mon, 18 Mar 2024 16:14:05 +0800
Subject: [PATCH] chat

---
 web/app/components/workflow/panel/chat-record/index.tsx | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/web/app/components/workflow/panel/chat-record/index.tsx b/web/app/components/workflow/panel/chat-record/index.tsx
index decdebdfd9..d1285b0a54 100644
--- a/web/app/components/workflow/panel/chat-record/index.tsx
+++ b/web/app/components/workflow/panel/chat-record/index.tsx
@@ -8,6 +8,7 @@ import {
 import { useStore } from '../../store'
 import UserInput from './user-input'
 import Chat from '@/app/components/base/chat/chat'
+import type { ChatItem } from '@/app/components/base/chat/types'
 import { fetchConvesationMessages } from '@/service/debug'
 import { useStore as useAppStore } from '@/app/components/app/store'
 import Loading from '@/app/components/base/loading'
@@ -19,7 +20,7 @@ const ChatRecord = () => {
   const currentConversationID = useStore(s => s.currentConversationID)
 
   const chatMessageList = useMemo(() => {
-    const res: any = []
+    const res: ChatItem[] = []
     if (chatList.length) {
       chatList.forEach((item: any) => {
         res.push({
@@ -35,10 +36,11 @@ const ChatRecord = () => {
           isAnswer: true,
           citation: item.retriever_resources,
           message_files: item.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
+          workflow_run_id: item.workflow_run_id,
         })
       })
     }
-    return res as any
+    return res
   }, [chatList])
 
   const handleFetchConvesationMessages = useCallback(async () => {