|
@@ -11,10 +11,12 @@ import { useLocalStorageState } from 'ahooks'
|
|
|
import produce from 'immer'
|
|
|
import type {
|
|
|
ChatConfig,
|
|
|
+ ChatItem,
|
|
|
Feedback,
|
|
|
} from '../types'
|
|
|
import { CONVERSATION_ID_INFO } from '../constants'
|
|
|
-import { getPrevChatList, getProcessedInputsFromUrlParams } from '../utils'
|
|
|
+import { buildChatItemTree, getProcessedInputsFromUrlParams } from '../utils'
|
|
|
+import { getProcessedFilesFromResponse } from '../../file-uploader/utils'
|
|
|
import {
|
|
|
fetchAppInfo,
|
|
|
fetchAppMeta,
|
|
@@ -32,6 +34,33 @@ import { useToastContext } from '@/app/components/base/toast'
|
|
|
import { changeLanguage } from '@/i18n/i18next-config'
|
|
|
import { InputVarType } from '@/app/components/workflow/types'
|
|
|
import { TransferMethod } from '@/types/app'
|
|
|
+import { addFileInfos, sortAgentSorts } from '@/app/components/tools/utils'
|
|
|
+
|
|
|
+function getFormattedChatList(messages: any[]) {
|
|
|
+ const newChatList: ChatItem[] = []
|
|
|
+ messages.forEach((item) => {
|
|
|
+ const questionFiles = item.message_files?.filter((file: any) => file.belongs_to === 'user') || []
|
|
|
+ newChatList.push({
|
|
|
+ id: `question-${item.id}`,
|
|
|
+ content: item.query,
|
|
|
+ isAnswer: false,
|
|
|
+ message_files: getProcessedFilesFromResponse(questionFiles.map((item: any) => ({ ...item, related_id: item.id }))),
|
|
|
+ parentMessageId: item.parent_message_id || undefined,
|
|
|
+ })
|
|
|
+ const answerFiles = item.message_files?.filter((file: any) => file.belongs_to === 'assistant') || []
|
|
|
+ newChatList.push({
|
|
|
+ id: item.id,
|
|
|
+ content: item.answer,
|
|
|
+ agent_thoughts: addFileInfos(item.agent_thoughts ? sortAgentSorts(item.agent_thoughts) : item.agent_thoughts, item.message_files),
|
|
|
+ feedback: item.feedback,
|
|
|
+ isAnswer: true,
|
|
|
+ citation: item.retriever_resources,
|
|
|
+ message_files: getProcessedFilesFromResponse(answerFiles.map((item: any) => ({ ...item, related_id: item.id }))),
|
|
|
+ parentMessageId: `question-${item.id}`,
|
|
|
+ })
|
|
|
+ })
|
|
|
+ return newChatList
|
|
|
+}
|
|
|
|
|
|
export const useEmbeddedChatbot = () => {
|
|
|
const isInstalledApp = false
|
|
@@ -77,7 +106,7 @@ export const useEmbeddedChatbot = () => {
|
|
|
|
|
|
const appPrevChatList = useMemo(
|
|
|
() => (currentConversationId && appChatListData?.data.length)
|
|
|
- ? getPrevChatList(appChatListData.data)
|
|
|
+ ? buildChatItemTree(getFormattedChatList(appChatListData.data))
|
|
|
: [],
|
|
|
[appChatListData, currentConversationId],
|
|
|
)
|