| import { type FC, useEffect, useRef, useState } from 'react' |
| import React from 'react' |
| import type { KnowledgeRetrievalNodeType } from './types' |
| import { Folder } from '@/app/components/base/icons/src/vender/solid/files' |
| import type { NodeProps } from '@/app/components/workflow/types' |
| import { fetchDatasets } from '@/service/datasets' |
| import type { DataSet } from '@/models/datasets' |
|
|
| const Node: FC<NodeProps<KnowledgeRetrievalNodeType>> = ({ |
| data, |
| }) => { |
| const [selectedDatasets, setSelectedDatasets] = useState<DataSet[]>([]) |
| const updateTime = useRef(0) |
| useEffect(() => { |
| (async () => { |
| updateTime.current = updateTime.current + 1 |
| const currUpdateTime = updateTime.current |
|
|
| if (data.dataset_ids?.length > 0) { |
| const { data: dataSetsWithDetail } = await fetchDatasets({ url: '/datasets', params: { page: 1, ids: data.dataset_ids } }) |
| |
| if (currUpdateTime < updateTime.current) |
| return |
| setSelectedDatasets(dataSetsWithDetail) |
| } |
| else { |
| setSelectedDatasets([]) |
| } |
| })() |
| }, [data.dataset_ids]) |
|
|
| if (!selectedDatasets.length) |
| return null |
|
|
| return ( |
| <div className='mb-1 px-3 py-1'> |
| <div className='space-y-0.5'> |
| {selectedDatasets.map(({ id, name }) => ( |
| <div key={id} className='flex items-center h-[26px] bg-workflow-block-parma-bg rounded-md px-1 text-xs font-normal text-gray-700'> |
| <div className='mr-1 shrink-0 p-1 bg-[#F5F8FF] rounded-md border-[0.5px] border-[#E0EAFF]'> |
| <Folder className='w-3 h-3 text-[#444CE7]' /> |
| </div> |
| <div className='grow w-0 text-text-secondary system-xs-regular truncate'> |
| {name} |
| </div> |
| </div> |
| ))} |
| </div> |
| </div> |
| ) |
| } |
|
|
| export default React.memo(Node) |
|
|