diff --git a/web/src/pages/dataset/dataset-overview/overview-table.tsx b/web/src/pages/dataset/dataset-overview/overview-table.tsx index c2cdb061a..e56d5d179 100644 --- a/web/src/pages/dataset/dataset-overview/overview-table.tsx +++ b/web/src/pages/dataset/dataset-overview/overview-table.tsx @@ -85,8 +85,8 @@ export const getFileLogsTableColumns = ( // row.original.kb_id, // )} > - - {row.original.fileName} + + {row.original.document_name} ), }, diff --git a/web/src/pages/dataset/dataset/generate-button/generate.tsx b/web/src/pages/dataset/dataset/generate-button/generate.tsx index a488966cf..489f81700 100644 --- a/web/src/pages/dataset/dataset/generate-button/generate.tsx +++ b/web/src/pages/dataset/dataset/generate-button/generate.tsx @@ -83,10 +83,10 @@ const MenuItem: React.FC<{ className={cn( 'border cursor-pointer p-2 rounded-md focus:bg-transparent', { - 'hover:border-accent-primary hover:bg-[rgba(59,160,92,0.1)]': + 'hover:border-accent-primary hover:bg-[rgba(59,160,92,0.1)] focus:bg-[rgba(59,160,92,0.1)]': status === generateStatus.start || status === generateStatus.completed, - 'hover:border-border hover:bg-[rgba(59,160,92,0)]': + 'hover:border-border hover:bg-[rgba(59,160,92,0)] focus:bg-[rgba(59,160,92,0)]': status !== generateStatus.start && status !== generateStatus.completed, }, @@ -177,7 +177,11 @@ const MenuItem: React.FC<{ ); }; -const Generate: React.FC = () => { +type GenerateProps = { + disabled?: boolean; +}; +const Generate: React.FC = (props) => { + const { disabled = false } = props; const [open, setOpen] = useState(false); const { graphRunData, raptorRunData } = useTraceGenerate({ open }); const { runGenerate, pauseGenerate } = useDatasetGenerate(); @@ -189,16 +193,21 @@ const Generate: React.FC = () => { return (
- - + +
+ +
{Object.values(GenerateType).map((name) => { diff --git a/web/src/pages/dataset/dataset/index.tsx b/web/src/pages/dataset/dataset/index.tsx index 7994d6be8..be34f7f9d 100644 --- a/web/src/pages/dataset/dataset/index.tsx +++ b/web/src/pages/dataset/dataset/index.tsx @@ -12,6 +12,7 @@ import { } from '@/components/ui/dropdown-menu'; import { useRowSelection } from '@/hooks/logic-hooks/use-row-selection'; import { useFetchDocumentList } from '@/hooks/use-document-request'; +import { useFetchKnowledgeBaseConfiguration } from '@/hooks/use-knowledge-request'; import { Upload } from 'lucide-react'; import { useTranslation } from 'react-i18next'; import { DatasetTable } from './dataset-table'; @@ -30,6 +31,7 @@ export default function Dataset() { onDocumentUploadOk, documentUploadLoading, } = useHandleUploadDocument(); + const { data: dataSetData } = useFetchKnowledgeBaseConfiguration(); const { searchString, @@ -62,7 +64,7 @@ export default function Dataset() { return ( <>
- + 0)} />