diff --git a/web/app/components/workflow/nodes/tool/components/context-generate-modal/components/chat-view.tsx b/web/app/components/workflow/nodes/tool/components/context-generate-modal/components/chat-view.tsx
index fb2b737af3..8e4a957a83 100644
--- a/web/app/components/workflow/nodes/tool/components/context-generate-modal/components/chat-view.tsx
+++ b/web/app/components/workflow/nodes/tool/components/context-generate-modal/components/chat-view.tsx
@@ -1,5 +1,6 @@
import type { ReactNode } from 'react'
import type { ContextGenerateChatMessage } from '../hooks/use-context-generate'
+import type { VersionOption } from '../types'
import type { FormValue } from '@/app/components/header/account-setting/model-provider-page/declarations'
import type { TriggerProps } from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal/trigger'
import type { Model } from '@/types/app'
@@ -12,11 +13,6 @@ import { CodeAssistant } from '@/app/components/base/icons/src/vender/line/gener
import ModelParameterModal from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal'
import { cn } from '@/utils/classnames'
-type VersionOption = {
- index: number
- label: string
-}
-
type ChatViewProps = {
promptMessages: ContextGenerateChatMessage[]
versionOptions: VersionOption[]
@@ -81,9 +77,10 @@ const ChatView = ({
const versionMeta = assistantVersionMap[index]
const isSelected = versionMeta?.index === currentVersionIndex
const assistantContent = message.content || defaultAssistantMessage
+ const messageKey = message.id || `${message.role}-${index}`
return (
{message.role === 'user'
@@ -164,7 +161,7 @@ const ChatView = ({