|
@@ -0,0 +1,385 @@
|
|
|
+import {
|
|
|
+ useCallback,
|
|
|
+ useEffect,
|
|
|
+ useMemo,
|
|
|
+ useRef,
|
|
|
+ useState,
|
|
|
+} from 'react'
|
|
|
+import { useTranslation } from 'react-i18next'
|
|
|
+import useSWR from 'swr'
|
|
|
+import { useLocalStorageState } from 'ahooks'
|
|
|
+import produce from 'immer'
|
|
|
+import type {
|
|
|
+ Callback,
|
|
|
+ ChatConfig,
|
|
|
+ ChatItem,
|
|
|
+ Feedback,
|
|
|
+} from '../types'
|
|
|
+import { CONVERSATION_ID_INFO } from '../constants'
|
|
|
+import {
|
|
|
+ delConversation,
|
|
|
+ fetchAppInfo,
|
|
|
+ fetchAppMeta,
|
|
|
+ fetchAppParams,
|
|
|
+ fetchChatList,
|
|
|
+ fetchConversations,
|
|
|
+ generationConversationName,
|
|
|
+ pinConversation,
|
|
|
+ renameConversation,
|
|
|
+ unpinConversation,
|
|
|
+ updateFeedback,
|
|
|
+} from '@/service/share'
|
|
|
+import type { InstalledApp } from '@/models/explore'
|
|
|
+import type {
|
|
|
+ AppData,
|
|
|
+ ConversationItem,
|
|
|
+} from '@/models/share'
|
|
|
+import { addFileInfos, sortAgentSorts } from '@/app/components/tools/utils'
|
|
|
+import { useToastContext } from '@/app/components/base/toast'
|
|
|
+
|
|
|
+export const useChatWithHistory = (installedAppInfo?: InstalledApp) => {
|
|
|
+ const isInstalledApp = useMemo(() => !!installedAppInfo, [installedAppInfo])
|
|
|
+ const { data: appInfo, isLoading: appInfoLoading } = useSWR(installedAppInfo ? null : 'appInfo', fetchAppInfo)
|
|
|
+
|
|
|
+ const appData = useMemo(() => {
|
|
|
+ if (isInstalledApp) {
|
|
|
+ const { id, app } = installedAppInfo!
|
|
|
+ return {
|
|
|
+ app_id: id,
|
|
|
+ site: { title: app.name, icon: app.icon, icon_background: app.icon_background, prompt_public: false, copyright: '' },
|
|
|
+ plan: 'basic',
|
|
|
+ } as AppData
|
|
|
+ }
|
|
|
+
|
|
|
+ return appInfo
|
|
|
+ }, [isInstalledApp, installedAppInfo, appInfo])
|
|
|
+ const appId = useMemo(() => appData?.app_id, [appData])
|
|
|
+
|
|
|
+ const [conversationIdInfo, setConversationIdInfo] = useLocalStorageState<Record<string, string>>(CONVERSATION_ID_INFO, {
|
|
|
+ defaultValue: {},
|
|
|
+ })
|
|
|
+ const currentConversationId = useMemo(() => conversationIdInfo?.[appId || ''] || '', [appId, conversationIdInfo])
|
|
|
+ const handleConversationIdInfoChange = useCallback((changeConversationId: string) => {
|
|
|
+ if (appId) {
|
|
|
+ setConversationIdInfo({
|
|
|
+ ...conversationIdInfo,
|
|
|
+ [appId || '']: changeConversationId,
|
|
|
+ })
|
|
|
+ }
|
|
|
+ }, [appId, conversationIdInfo, setConversationIdInfo])
|
|
|
+ const [showConfigPanelBeforeChat, setShowConfigPanelBeforeChat] = useState(true)
|
|
|
+
|
|
|
+ const [newConversationId, setNewConversationId] = useState('')
|
|
|
+ const chatShouldReloadKey = useMemo(() => {
|
|
|
+ if (currentConversationId === newConversationId)
|
|
|
+ return ''
|
|
|
+
|
|
|
+ return currentConversationId
|
|
|
+ }, [currentConversationId, newConversationId])
|
|
|
+
|
|
|
+ const { data: appParams } = useSWR(['appParams', isInstalledApp, appId], () => fetchAppParams(isInstalledApp, appId))
|
|
|
+ const { data: appMeta } = useSWR(['appMeta', isInstalledApp, appId], () => fetchAppMeta(isInstalledApp, appId))
|
|
|
+ const { data: appPinnedConversationData, mutate: mutateAppPinnedConversationData } = useSWR(['appConversationData', isInstalledApp, appId, true], () => fetchConversations(isInstalledApp, appId, undefined, true, 100))
|
|
|
+ const { data: appConversationData, isLoading: appConversationDataLoading, mutate: mutateAppConversationData } = useSWR(['appConversationData', isInstalledApp, appId, false], () => fetchConversations(isInstalledApp, appId, undefined, false, 100))
|
|
|
+ const { data: appChatListData, isLoading: appChatListDataLoading } = useSWR(chatShouldReloadKey ? ['appChatList', chatShouldReloadKey, isInstalledApp, appId] : null, () => fetchChatList(chatShouldReloadKey, isInstalledApp, appId))
|
|
|
+
|
|
|
+ const appPrevChatList = useMemo(() => {
|
|
|
+ const data = appChatListData?.data || []
|
|
|
+ const chatList: ChatItem[] = []
|
|
|
+
|
|
|
+ if (currentConversationId && data.length) {
|
|
|
+ data.forEach((item: any) => {
|
|
|
+ chatList.push({
|
|
|
+ id: `question-${item.id}`,
|
|
|
+ content: item.query,
|
|
|
+ isAnswer: false,
|
|
|
+ message_files: item.message_files?.filter((file: any) => file.belongs_to === 'user') || [],
|
|
|
+ })
|
|
|
+ chatList.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: item.message_files?.filter((file: any) => file.belongs_to === 'assistant') || [],
|
|
|
+ })
|
|
|
+ })
|
|
|
+ }
|
|
|
+
|
|
|
+ return chatList
|
|
|
+ }, [appChatListData, currentConversationId])
|
|
|
+
|
|
|
+ const [showNewConversationItemInList, setShowNewConversationItemInList] = useState(false)
|
|
|
+
|
|
|
+ const pinnedConversationList = useMemo(() => {
|
|
|
+ return appPinnedConversationData?.data || []
|
|
|
+ }, [appPinnedConversationData])
|
|
|
+ const { t } = useTranslation()
|
|
|
+ const newConversationInputsRef = useRef<Record<string, any>>({})
|
|
|
+ const [newConversationInputs, setNewConversationInputs] = useState<Record<string, any>>({})
|
|
|
+ const handleNewConversationInputsChange = useCallback((newInputs: Record<string, any>) => {
|
|
|
+ newConversationInputsRef.current = newInputs
|
|
|
+ setNewConversationInputs(newInputs)
|
|
|
+ }, [])
|
|
|
+ const inputsForms = useMemo(() => {
|
|
|
+ return (appParams?.user_input_form || []).filter((item: any) => item.paragraph || item.select || item['text-input']).map((item: any) => {
|
|
|
+ if (item.paragraph) {
|
|
|
+ return {
|
|
|
+ ...item.paragraph,
|
|
|
+ type: 'paragraph',
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (item.select) {
|
|
|
+ return {
|
|
|
+ ...item.select,
|
|
|
+ type: 'select',
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return {
|
|
|
+ ...item['text-input'],
|
|
|
+ type: 'text-input',
|
|
|
+ }
|
|
|
+ })
|
|
|
+ }, [appParams])
|
|
|
+ useEffect(() => {
|
|
|
+ const conversationInputs: Record<string, any> = {}
|
|
|
+
|
|
|
+ inputsForms.forEach((item: any) => {
|
|
|
+ conversationInputs[item.variable] = item.default || ''
|
|
|
+ })
|
|
|
+ handleNewConversationInputsChange(conversationInputs)
|
|
|
+ }, [handleNewConversationInputsChange, inputsForms])
|
|
|
+
|
|
|
+ const { data: newConversation } = useSWR(newConversationId ? [isInstalledApp, appId, newConversationId] : null, () => generationConversationName(isInstalledApp, appId, newConversationId))
|
|
|
+ const [originConversationList, setOriginConversationList] = useState<ConversationItem[]>([])
|
|
|
+ useEffect(() => {
|
|
|
+ if (appConversationData?.data && !appConversationDataLoading)
|
|
|
+ setOriginConversationList(appConversationData?.data)
|
|
|
+ }, [appConversationData, appConversationDataLoading])
|
|
|
+ const conversationList = useMemo(() => {
|
|
|
+ const data = originConversationList.slice()
|
|
|
+
|
|
|
+ if (showNewConversationItemInList && data[0]?.id !== '') {
|
|
|
+ data.unshift({
|
|
|
+ id: '',
|
|
|
+ name: t('share.chat.newChatDefaultName'),
|
|
|
+ inputs: {},
|
|
|
+ introduction: '',
|
|
|
+ })
|
|
|
+ }
|
|
|
+ return data
|
|
|
+ }, [originConversationList, showNewConversationItemInList, t])
|
|
|
+
|
|
|
+ useEffect(() => {
|
|
|
+ if (newConversation) {
|
|
|
+ setOriginConversationList(produce((draft) => {
|
|
|
+ const index = draft.findIndex(item => item.id === newConversation.id)
|
|
|
+
|
|
|
+ if (index > -1)
|
|
|
+ draft[index] = newConversation
|
|
|
+ else
|
|
|
+ draft.unshift(newConversation)
|
|
|
+ }))
|
|
|
+ }
|
|
|
+ }, [newConversation])
|
|
|
+
|
|
|
+ const currentConversationItem = useMemo(() => {
|
|
|
+ let coversationItem = conversationList.find(item => item.id === currentConversationId)
|
|
|
+
|
|
|
+ if (!coversationItem && pinnedConversationList.length)
|
|
|
+ coversationItem = pinnedConversationList.find(item => item.id === currentConversationId)
|
|
|
+
|
|
|
+ return coversationItem
|
|
|
+ }, [conversationList, currentConversationId, pinnedConversationList])
|
|
|
+
|
|
|
+ const { notify } = useToastContext()
|
|
|
+ const checkInputsRequired = useCallback((silent?: boolean) => {
|
|
|
+ if (inputsForms.length) {
|
|
|
+ for (let i = 0; i < inputsForms.length; i += 1) {
|
|
|
+ const item = inputsForms[i]
|
|
|
+
|
|
|
+ if (item.required && !newConversationInputsRef.current[item.variable]) {
|
|
|
+ if (!silent) {
|
|
|
+ notify({
|
|
|
+ type: 'error',
|
|
|
+ message: t('appDebug.errorMessage.valueOfVarRequired', { key: item.variable }),
|
|
|
+ })
|
|
|
+ }
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return true
|
|
|
+ }
|
|
|
+
|
|
|
+ return true
|
|
|
+ }, [inputsForms, notify, t])
|
|
|
+ const handleStartChat = useCallback(() => {
|
|
|
+ if (checkInputsRequired()) {
|
|
|
+ setShowConfigPanelBeforeChat(false)
|
|
|
+ setShowNewConversationItemInList(true)
|
|
|
+ }
|
|
|
+ }, [setShowConfigPanelBeforeChat, setShowNewConversationItemInList, checkInputsRequired])
|
|
|
+ const currentChatInstanceRef = useRef<{ handleStop: () => void }>({ handleStop: () => {} })
|
|
|
+ const handleChangeConversation = useCallback((conversationId: string) => {
|
|
|
+ currentChatInstanceRef.current.handleStop()
|
|
|
+ setNewConversationId('')
|
|
|
+ handleConversationIdInfoChange(conversationId)
|
|
|
+
|
|
|
+ if (conversationId === '' && !checkInputsRequired(true))
|
|
|
+ setShowConfigPanelBeforeChat(true)
|
|
|
+ else
|
|
|
+ setShowConfigPanelBeforeChat(false)
|
|
|
+ }, [handleConversationIdInfoChange, setShowConfigPanelBeforeChat, checkInputsRequired])
|
|
|
+ const handleNewConversation = useCallback(() => {
|
|
|
+ currentChatInstanceRef.current.handleStop()
|
|
|
+ setNewConversationId('')
|
|
|
+
|
|
|
+ if (showNewConversationItemInList) {
|
|
|
+ handleChangeConversation('')
|
|
|
+ }
|
|
|
+ else if (currentConversationId) {
|
|
|
+ handleConversationIdInfoChange('')
|
|
|
+ setShowConfigPanelBeforeChat(true)
|
|
|
+ setShowNewConversationItemInList(true)
|
|
|
+ handleNewConversationInputsChange({})
|
|
|
+ }
|
|
|
+ }, [handleChangeConversation, currentConversationId, handleConversationIdInfoChange, setShowConfigPanelBeforeChat, setShowNewConversationItemInList, showNewConversationItemInList, handleNewConversationInputsChange])
|
|
|
+ const handleUpdateConversationList = useCallback(() => {
|
|
|
+ mutateAppConversationData()
|
|
|
+ mutateAppPinnedConversationData()
|
|
|
+ }, [mutateAppConversationData, mutateAppPinnedConversationData])
|
|
|
+
|
|
|
+ const handlePinConversation = useCallback(async (conversationId: string) => {
|
|
|
+ await pinConversation(isInstalledApp, appId, conversationId)
|
|
|
+ notify({ type: 'success', message: t('common.api.success') })
|
|
|
+ handleUpdateConversationList()
|
|
|
+ }, [isInstalledApp, appId, notify, t, handleUpdateConversationList])
|
|
|
+
|
|
|
+ const handleUnpinConversation = useCallback(async (conversationId: string) => {
|
|
|
+ await unpinConversation(isInstalledApp, appId, conversationId)
|
|
|
+ notify({ type: 'success', message: t('common.api.success') })
|
|
|
+ handleUpdateConversationList()
|
|
|
+ }, [isInstalledApp, appId, notify, t, handleUpdateConversationList])
|
|
|
+
|
|
|
+ const [conversationDeleting, setConversationDeleting] = useState(false)
|
|
|
+ const handleDeleteConversation = useCallback(async (
|
|
|
+ conversationId: string,
|
|
|
+ {
|
|
|
+ onSuccess,
|
|
|
+ }: Callback,
|
|
|
+ ) => {
|
|
|
+ if (conversationDeleting)
|
|
|
+ return
|
|
|
+
|
|
|
+ try {
|
|
|
+ setConversationDeleting(true)
|
|
|
+ await delConversation(isInstalledApp, appId, conversationId)
|
|
|
+ notify({ type: 'success', message: t('common.api.success') })
|
|
|
+ onSuccess()
|
|
|
+ }
|
|
|
+ finally {
|
|
|
+ setConversationDeleting(false)
|
|
|
+ }
|
|
|
+
|
|
|
+ if (conversationId === currentConversationId)
|
|
|
+ handleNewConversation()
|
|
|
+
|
|
|
+ handleUpdateConversationList()
|
|
|
+ }, [isInstalledApp, appId, notify, t, handleUpdateConversationList, handleNewConversation, currentConversationId, conversationDeleting])
|
|
|
+
|
|
|
+ const [conversationRenaming, setConversationRenaming] = useState(false)
|
|
|
+ const handleRenameConversation = useCallback(async (
|
|
|
+ conversationId: string,
|
|
|
+ newName: string,
|
|
|
+ {
|
|
|
+ onSuccess,
|
|
|
+ }: Callback,
|
|
|
+ ) => {
|
|
|
+ if (conversationRenaming)
|
|
|
+ return
|
|
|
+
|
|
|
+ if (!newName.trim()) {
|
|
|
+ notify({
|
|
|
+ type: 'error',
|
|
|
+ message: t('common.chat.conversationNameCanNotEmpty'),
|
|
|
+ })
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ setConversationRenaming(true)
|
|
|
+ try {
|
|
|
+ await renameConversation(isInstalledApp, appId, conversationId, newName)
|
|
|
+
|
|
|
+ notify({
|
|
|
+ type: 'success',
|
|
|
+ message: t('common.actionMsg.modifiedSuccessfully'),
|
|
|
+ })
|
|
|
+ setOriginConversationList(produce((draft) => {
|
|
|
+ const index = originConversationList.findIndex(item => item.id === conversationId)
|
|
|
+ const item = draft[index]
|
|
|
+
|
|
|
+ draft[index] = {
|
|
|
+ ...item,
|
|
|
+ name: newName,
|
|
|
+ }
|
|
|
+ }))
|
|
|
+ onSuccess()
|
|
|
+ }
|
|
|
+ finally {
|
|
|
+ setConversationRenaming(false)
|
|
|
+ }
|
|
|
+ }, [isInstalledApp, appId, notify, t, conversationRenaming, originConversationList])
|
|
|
+
|
|
|
+ const handleNewConversationCompleted = useCallback((newConversationId: string) => {
|
|
|
+ setNewConversationId(newConversationId)
|
|
|
+ handleConversationIdInfoChange(newConversationId)
|
|
|
+ setShowNewConversationItemInList(false)
|
|
|
+ mutateAppConversationData()
|
|
|
+ }, [mutateAppConversationData, handleConversationIdInfoChange])
|
|
|
+
|
|
|
+ const handleFeedback = useCallback(async (messageId: string, feedback: Feedback) => {
|
|
|
+ await updateFeedback({ url: `/messages/${messageId}/feedbacks`, body: { rating: feedback.rating } }, isInstalledApp, appId)
|
|
|
+ notify({ type: 'success', message: t('common.api.success') })
|
|
|
+ }, [isInstalledApp, appId, t, notify])
|
|
|
+
|
|
|
+ return {
|
|
|
+ appInfoLoading,
|
|
|
+ isInstalledApp,
|
|
|
+ appId,
|
|
|
+ currentConversationId,
|
|
|
+ currentConversationItem,
|
|
|
+ handleConversationIdInfoChange,
|
|
|
+ appData,
|
|
|
+ appParams: appParams || {} as ChatConfig,
|
|
|
+ appMeta,
|
|
|
+ appPinnedConversationData,
|
|
|
+ appConversationData,
|
|
|
+ appConversationDataLoading,
|
|
|
+ appChatListData,
|
|
|
+ appChatListDataLoading,
|
|
|
+ appPrevChatList,
|
|
|
+ pinnedConversationList,
|
|
|
+ conversationList,
|
|
|
+ showConfigPanelBeforeChat,
|
|
|
+ setShowConfigPanelBeforeChat,
|
|
|
+ setShowNewConversationItemInList,
|
|
|
+ newConversationInputs,
|
|
|
+ handleNewConversationInputsChange,
|
|
|
+ inputsForms,
|
|
|
+ handleNewConversation,
|
|
|
+ handleStartChat,
|
|
|
+ handleChangeConversation,
|
|
|
+ handlePinConversation,
|
|
|
+ handleUnpinConversation,
|
|
|
+ conversationDeleting,
|
|
|
+ handleDeleteConversation,
|
|
|
+ conversationRenaming,
|
|
|
+ handleRenameConversation,
|
|
|
+ handleNewConversationCompleted,
|
|
|
+ newConversationId,
|
|
|
+ chatShouldReloadKey,
|
|
|
+ handleFeedback,
|
|
|
+ currentChatInstanceRef,
|
|
|
+ }
|
|
|
+}
|