Commit 42685382 authored by Joel's avatar Joel

feat: set dataset info

parent c7a1e260
...@@ -37,6 +37,7 @@ import Confirm from '@/app/components/base/confirm' ...@@ -37,6 +37,7 @@ import Confirm from '@/app/components/base/confirm'
import type { DataSet } from '@/models/datasets' import type { DataSet } from '@/models/datasets'
import ConfigSummary from '@/app/components/explore/universal-chat/config-view/summary' import ConfigSummary from '@/app/components/explore/universal-chat/config-view/summary'
import ConfigDetail from '@/app/components/explore/universal-chat/config-view/detail' import ConfigDetail from '@/app/components/explore/universal-chat/config-view/detail'
import { fetchDatasets } from '@/service/datasets'
const APP_ID = 'universal-chat' const APP_ID = 'universal-chat'
const DEFAULT_MODEL_ID = 'claude-2' // gpt-4, claude-2 const DEFAULT_MODEL_ID = 'claude-2' // gpt-4, claude-2
...@@ -165,7 +166,7 @@ const Main: FC<IMainProps> = () => { ...@@ -165,7 +166,7 @@ const Main: FC<IMainProps> = () => {
const conversationName = currConversationInfo?.name || t('share.chat.newChatDefaultName') as string const conversationName = currConversationInfo?.name || t('share.chat.newChatDefaultName') as string
const conversationIntroduction = currConversationInfo?.introduction || '' const conversationIntroduction = currConversationInfo?.introduction || ''
const handleConversationSwitch = () => { const handleConversationSwitch = async () => {
if (!inited) if (!inited)
return return
...@@ -195,8 +196,13 @@ const Main: FC<IMainProps> = () => { ...@@ -195,8 +196,13 @@ const Main: FC<IMainProps> = () => {
pluginConfig[pluginName] = item[pluginName].enabled pluginConfig[pluginName] = item[pluginName].enabled
}) })
setPlugins(pluginConfig) setPlugins(pluginConfig)
console.log('datasetIds', datasetIds) if (datasetIds.length > 0) {
setDateSets([]) const { data } = await fetchDatasets({ url: '/datasets', params: { page: 1, ids: datasetIds } })
setDateSets(data)
}
else {
setDateSets([])
}
} }
else { else {
setModeId(DEFAULT_MODEL_ID) setModeId(DEFAULT_MODEL_ID)
...@@ -238,7 +244,9 @@ const Main: FC<IMainProps> = () => { ...@@ -238,7 +244,9 @@ const Main: FC<IMainProps> = () => {
setControlFocus(Date.now()) setControlFocus(Date.now())
} }
useEffect(handleConversationSwitch, [currConversationId, inited]) useEffect(() => {
handleConversationSwitch()
}, [currConversationId, inited])
const handleConversationIdChange = (id: string) => { const handleConversationIdChange = (id: string) => {
if (id === '-1') { if (id === '-1') {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment