feat: enhance dataset management features with improved tag handling, download functionality, and UI updates

This commit is contained in:
chenghh-9609
2025-10-23 10:33:42 +08:00
parent add6998601
commit a6d4b51601
10 changed files with 88 additions and 55 deletions

View File

@@ -4,6 +4,7 @@ import {
EditOutlined,
DeleteOutlined,
PlusOutlined,
UploadOutlined,
} from "@ant-design/icons";
import TagManager from "@/components/TagManagement";
import { Link, useNavigate } from "react-router";
@@ -25,6 +26,7 @@ import {
} from "../dataset.api";
import { formatBytes } from "@/utils/unit";
import EditDataset from "../Create/EditDataset";
import ImportConfiguration from "../Detail/components/ImportConfiguration";
export default function DatasetManagementPage() {
const navigate = useNavigate();
@@ -32,7 +34,7 @@ export default function DatasetManagementPage() {
const [viewMode, setViewMode] = useState<"card" | "list">("card");
const [editDatasetOpen, setEditDatasetOpen] = useState(false);
const [currentDataset, setCurrentDataset] = useState<Dataset | null>(null);
const [showUploadDialog, setShowUploadDialog] = useState(false);
const [statisticsData, setStatisticsData] = useState<any>({
count: {},
size: {},
@@ -131,6 +133,11 @@ export default function DatasetManagementPage() {
message.success("数据删除成功");
};
const handleImportData = (dataset: Dataset) => {
setCurrentDataset(dataset);
setShowUploadDialog(true);
};
useEffect(() => {
fetchStatistics();
}, []);
@@ -141,11 +148,18 @@ export default function DatasetManagementPage() {
label: "编辑",
icon: <EditOutlined />,
onClick: (item: Dataset) => {
console.log(item);
setCurrentDataset(item);
setEditDatasetOpen(true);
},
},
{
key: "import",
label: "导入",
icon: <UploadOutlined />,
onClick: (item: Dataset) => {
handleImportData(item);
},
},
{
key: "download",
label: "下载",
@@ -343,7 +357,19 @@ export default function DatasetManagementPage() {
<EditDataset
open={editDatasetOpen}
data={currentDataset}
onClose={() => setEditDatasetOpen(false)}
onClose={() => {
setCurrentDataset(null);
setEditDatasetOpen(false);
}}
onRefresh={fetchData}
/>
<ImportConfiguration
data={currentDataset}
open={showUploadDialog}
onClose={() => {
setCurrentDataset(null);
setShowUploadDialog(false);
}}
onRefresh={fetchData}
/>
</div>