feat: dataset name

web/app/components/app/chat/answer/index.tsx

@@ -20,6 +20,7 @@ import Tooltip from '@/app/components/base/tooltip'
 import { Markdown } from '@/app/components/base/markdown'
 import AutoHeightTextarea from '@/app/components/base/auto-height-textarea'
 import Button from '@/app/components/base/button'
+import type { DataSet } from '@/models/datasets'
 const Divider: FC<{ name: string }> = ({ name }) => {
   const { t } = useTranslation()
   return <div className='flex items-center my-2'>
@@ -45,9 +46,10 @@ export type IAnswerProps = {
   answerIconClassName?: string
   thoughts?: ThoughtItem[]
   isThinking?: boolean
+  dataSets?: DataSet[]
 }
 // The component needs to maintain its own state to control whether to display input component
-const Answer: FC<IAnswerProps> = ({ item, feedbackDisabled = false, isHideFeedbackEdit = false, onFeedback, onSubmitAnnotation, displayScene = 'web', isResponsing, answerIconClassName, thoughts, isThinking }) => {
+const Answer: FC<IAnswerProps> = ({ item, feedbackDisabled = false, isHideFeedbackEdit = false, onFeedback, onSubmitAnnotation, displayScene = 'web', isResponsing, answerIconClassName, thoughts, isThinking, dataSets }) => {
   const { id, content, more, feedback, adminFeedback, annotation: initAnnotation } = item
   const [showEdit, setShowEdit] = useState(false)
   const [loading, setLoading] = useState(false)
@@ -197,7 +199,11 @@ const Answer: FC<IAnswerProps> = ({ item, feedbackDisabled = false, isHideFeedba
                   </div>
                 )}
                 {(thoughts && thoughts.length > 0) && (
-                  <Thought list={thoughts || []} isThinking={isThinking} />
+                  <Thought
+                    list={thoughts || []}
+                    isThinking={isThinking}
+                    dataSets={dataSets}
+                  />
                 )}
                 {(isResponsing && !content)
                   ? (

web/app/components/app/chat/index.tsx

@@ -19,6 +19,8 @@ import VoiceInput from '@/app/components/base/voice-input'
 import { Microphone01 } from '@/app/components/base/icons/src/vender/line/mediaAndDevices'
 import { Microphone01 as Microphone01Solid } from '@/app/components/base/icons/src/vender/solid/mediaAndDevices'
 import { XCircle } from '@/app/components/base/icons/src/vender/solid/general'
+import type { DataSet } from '@/models/datasets'
+
 export type IChatProps = {
   configElem?: React.ReactNode
   chatList: IChatItem[]
@@ -48,6 +50,7 @@ export type IChatProps = {
   answerIconClassName?: string
   isShowConfigElem?: boolean
   isThoughting?: boolean
+  dataSets?: DataSet[]
 }
 
 const Chat: FC<IChatProps> = ({
@@ -72,6 +75,7 @@ const Chat: FC<IChatProps> = ({
   isShowSpeechToText,
   answerIconClassName,
   isShowConfigElem,
+  dataSets,
 }) => {
   const { t } = useTranslation()
   const { notify } = useContext(ToastContext)
@@ -170,6 +174,7 @@ const Chat: FC<IChatProps> = ({
               answerIconClassName={answerIconClassName}
               thoughts={thoughts}
               isThinking={isThinking}
+              dataSets={dataSets}
             />
           }
           return <Question key={item.id} id={item.id} content={item.content} more={item.more} useCurrentUserAvatar={useCurrentUserAvatar} />

web/app/components/app/chat/thought/index.tsx

@@ -5,20 +5,22 @@ import cn from 'classnames'
 import { useTranslation } from 'react-i18next'
 import type { ThoughtItem } from '../type'
 import s from './style.module.css'
-import { DataSet, Loading as LodingIcon, Search, ThoughtList, WebReader } from '@/app/components/base/icons/src/public/thought'
+import { DataSet as DataSetIcon, Loading as LodingIcon, Search, ThoughtList, WebReader } from '@/app/components/base/icons/src/public/thought'
 import { ChevronDown } from '@/app/components/base/icons/src/vender/line/arrows'
+import type { DataSet } from '@/models/datasets'
 
 // https://www.freecodecamp.org/news/how-to-write-a-regular-expression-for-a-url/
 const urlRegex = /(https:\/\/www\.|http:\/\/www\.|https:\/\/|http:\/\/)?[a-zA-Z]{2,}(\.[a-zA-Z]{2,})(\.[a-zA-Z]{2,})?\/[a-zA-Z0-9]{2,}|((https:\/\/www\.|http:\/\/www\.|https:\/\/|http:\/\/)?[a-zA-Z]{2,}(\.[a-zA-Z]{2,})(\.[a-zA-Z]{2,})?)|(https:\/\/www\.|http:\/\/www\.|https:\/\/|http:\/\/)?[a-zA-Z0-9]{2,}\.[a-zA-Z0-9]{2,}\.[a-zA-Z0-9]{2,}(\.[a-zA-Z0-9]{2,})?/gi
 export type IThoughtProps = {
   list: ThoughtItem[]
   isThinking?: boolean
+  dataSets?: DataSet[]
 }
 
 const getIcon = (toolId: string) => {
   switch (toolId) {
     case 'dataset':
-      return <DataSet />
+      return <DataSetIcon />
     case 'web_reader':
       return <WebReader />
     default:
@@ -29,16 +31,20 @@ const getIcon = (toolId: string) => {
 const Thought: FC<IThoughtProps> = ({
   list,
   isThinking,
+  dataSets,
 }) => {
   const { t } = useTranslation()
   const [isShowDetail, setIsShowDetail] = React.useState(false)
+
   const getThoughtText = (item: ThoughtItem) => {
     try {
       const input = JSON.parse(item.tool_input)
+
       switch (item.tool) {
         case 'dataset':
-        //   debugger
-          return t('explore.universalChat.thought.res.dataset', { input: input.dataset_id })
+          // eslint-disable-next-line no-case-declarations
+          const datasetName = dataSets?.find(item => item.id === input.dataset_id)?.name || 'unknown dataset'
+          return t('explore.universalChat.thought.res.dataset').replace('{datasetName}', `<span class="text-gray-700">${datasetName}</span>`)
         case 'web_reader':
           return t(`explore.universalChat.thought.res.webReader.${!input.cursor ? 'normal' : 'hasPageInfo'}`).replace('{url}', `<a href="${input.url}" class="text-[#155EEF]">${input.url}</a>`)
         default: // google, wikipedia

web/app/components/explore/universal-chat/index.tsx

@@ -662,6 +662,7 @@ const Main: FC<IMainProps> = () => {
                 isShowSuggestion={doShowSuggestion}
                 suggestionList={suggestQuestions}
                 isShowSpeechToText={speechToTextConfig?.enabled}
+                dataSets={dataSets}
               />
             </div>
           </div>

web/i18n/lang/explore.en.ts

@@ -67,7 +67,7 @@ const translation = {
           hasPageInfo: 'Reading next page of {url}',
         },
         search: 'Searching {{query}}',
-        dataset: 'Retrieving dataset {{datasetName}}',
+        dataset: 'Retrieving dataset {datasetName}',
       },
     },
     viewConfigDetailTip: 'In conversation, cannot change above settings',

web/i18n/lang/explore.zh.ts

@@ -67,7 +67,7 @@ const translation = {
           hasPageInfo: '解析链接 {url} 的下一页',
         },
         search: '搜索 {{query}}',
-        dataset: '检索数据集 {{dataset}}',
+        dataset: '检索数据集 {datasetName}',
       },
     },
     viewConfigDetailTip: '在对话中，无法更改上述设置',