diff --git a/web/src/interfaces/database/agent.ts b/web/src/interfaces/database/agent.ts index e82d60501..691946bcf 100644 --- a/web/src/interfaces/database/agent.ts +++ b/web/src/interfaces/database/agent.ts @@ -266,7 +266,7 @@ export interface IAgentLogResponse { source: string; user_id: string; dsl: string; - reference: IReference; + reference: IReference[]; name: string; version_title: string; } diff --git a/web/src/pages/agent/explore/components/session-chat.tsx b/web/src/pages/agent/explore/components/session-chat.tsx index e5bd0bb0d..f5a72b0dc 100644 --- a/web/src/pages/agent/explore/components/session-chat.tsx +++ b/web/src/pages/agent/explore/components/session-chat.tsx @@ -8,7 +8,7 @@ import { MessageType } from '@/constants/chat'; import { useUploadAgentFileWithProgress } from '@/hooks/use-agent-request'; import { useFetchUserInfo } from '@/hooks/use-user-setting-request'; import { IAgentLogResponse } from '@/interfaces/database/agent'; -import { IMessage } from '@/interfaces/database/chat'; +import { IMessage, IReferenceObject } from '@/interfaces/database/chat'; import DebugContent from '@/pages/agent/debug-content'; import { useAwaitComponentData } from '@/pages/agent/hooks/use-chat-logic'; import { BeginQuery } from '@/pages/agent/interface'; @@ -149,7 +149,13 @@ export function SessionChat({ session }: SessionChatProps) { nickname={userInfo.nickname} avatar={userInfo.avatar} avatarDialog={canvasInfo?.avatar || ''} - reference={findReferenceByMessageId(message.id)} + reference={ + findReferenceByMessageId(message.id) || + (session?.reference?.[ + Math.floor((i - 1) / 2) + ] as unknown as IReferenceObject) || + {} + } clickDocumentButton={clickDocumentButton} index={i} showLikeButton={false}