init datamate

2025-10-21 23:00:48 +08:00
commit 1c97afed7d
692 changed files with 135442 additions and 0 deletions
--- a/frontend/src/pages/DataManagement/Create/EditDataset.tsx
+++ b/frontend/src/pages/DataManagement/Create/EditDataset.tsx
@@ -0,0 +1,101 @@
+import BasicInformation from "./components/BasicInformation";
+import {
+  queryDatasetByIdUsingGet,
+  updateDatasetByIdUsingPut,
+} from "../dataset.api";
+import { useEffect, useState } from "react";
+import { Dataset, DatasetType } from "../dataset.model";
+import { App, Button, Drawer, Form, Modal } from "antd";
+
+export default function EditDataset({
+  open,
+  data,
+  onClose,
+  onRefresh,
+}: {
+  open: boolean;
+  data: Dataset | null;
+  onClose: () => void;
+  onRefresh?: () => void;
+}) {
+  const [form] = Form.useForm();
+  const { message } = App.useApp();
+
+  const [newDataset, setNewDataset] = useState({
+    name: "",
+    description: "",
+    datasetType: DatasetType.TEXT,
+    tags: [],
+  });
+  const fetchDataset = async () => {
+    // 如果有id，说明是编辑模式
+    if (data && data.id) {
+      const { data: newData } = await queryDatasetByIdUsingGet(data.id);
+      const updatedDataset = {
+        ...newData,
+        type: newData.type,
+        tags: newData.tags.map((tag) => tag.name) || [],
+      };
+      setNewDataset(updatedDataset);
+      form.setFieldsValue(updatedDataset);
+    }
+  };
+
+  useEffect(() => {
+    fetchDataset();
+  }, [data]);
+
+  const handleValuesChange = (_, allValues) => {
+    setNewDataset({ ...newDataset, ...allValues });
+  };
+
+  const handleSubmit = async () => {
+    const formValues = await form.validateFields();
+
+    const params = {
+      ...formValues,
+      files: undefined,
+    };
+    try {
+      await updateDatasetByIdUsingPut(data?.id, params);
+      onClose();
+      message.success("数据集更新成功");
+      onRefresh?.();
+    } catch (error) {
+      console.error(error);
+      message.error("数据集更新失败，请重试");
+      return;
+    }
+  };
+
+  return (
+    <Modal
+      title={`编辑数据集${data?.name}`}
+      onCancel={onClose}
+      open={open}
+      width={600}
+      maskClosable={false}
+      footer={
+        <>
+          <Button onClick={onClose}>取消</Button>
+          <Button type="primary" onClick={handleSubmit}>
+            确定
+          </Button>
+        </>
+      }
+    >
+      <Form
+        form={form}
+        initialValues={newDataset}
+        onValuesChange={handleValuesChange}
+        layout="vertical"
+      >
+        <BasicInformation
+          data={newDataset}
+          setData={setNewDataset}
+          hidden={["datasetType"]}
+        />
+      </Form>
+    </Modal>
+  );
+}