Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -109,12 +109,16 @@ public CleaningTaskDto createTask(CreateCleaningTaskRequest request) {

ExecutorType executorType = cleanTaskValidator.checkAndGetExecutorType(request.getInstance());

CreateDatasetRequest createDatasetRequest = new CreateDatasetRequest();
createDatasetRequest.setName(request.getDestDatasetName());
createDatasetRequest.setDatasetType(DatasetType.valueOf(request.getDestDatasetType()));
createDatasetRequest.setStatus("ACTIVE");
Dataset destDataset = datasetService.createDataset(createDatasetRequest);

Dataset destDataset;
if (StringUtils.isNotBlank(request.getDestDatasetId())) {
destDataset = datasetService.getDataset(request.getDestDatasetId());
} else {
CreateDatasetRequest createDatasetRequest = new CreateDatasetRequest();
createDatasetRequest.setName(request.getDestDatasetName());
createDatasetRequest.setDatasetType(DatasetType.valueOf(request.getDestDatasetType()));
createDatasetRequest.setStatus("ACTIVE");
destDataset = datasetService.createDataset(createDatasetRequest);
}
Dataset srcDataset = datasetService.getDataset(request.getSrcDatasetId());

CleaningTaskDto task = new CleaningTaskDto();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,10 @@ public class CreateCleaningTaskRequest {
@McpToolParam(description = "清洗任务使用的源数据集名称")
private String srcDatasetName;

@McpToolParam(description = "清洗任务创建的目标数据集名称")
@McpToolParam(description = "清洗任务写入的目标数据集ID", required = false)
private String destDatasetId;

@McpToolParam(description = "清洗任务写入的目标数据集名称,若destDatasetId为空,则创建新数据集。")
private String destDatasetName;

@McpToolParam(description = "清洗任务创建的目标数据集类型,取值范围为TEXT/IMAGE/VIDEO/AUDIO/OTHER")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ import {
DatasetSubType,
DatasetType,
} from "@/pages/DataManagement/dataset.model";
import { Input, Select, Form } from "antd";
import { Input, Select, Form, AutoComplete } from "antd";
import TextArea from "antd/es/input/TextArea";
import { useEffect, useState } from "react";

Expand Down Expand Up @@ -47,6 +47,13 @@ export default function CreateTaskStepOne({
...allValues,
srcDatasetName: dataset?.name || "",
});
} else if (key === "destDatasetName") {
dataset = datasets.find((d) => d.name === value);
setTaskConfig({
...taskConfig,
...allValues,
destDatasetId: dataset?.id || "",
});
} else {
setTaskConfig({ ...taskConfig, ...allValues });
}
Expand Down Expand Up @@ -89,7 +96,26 @@ export default function CreateTaskStepOne({
/>
</Form.Item>
<Form.Item label="目标数据集名称" name="destDatasetName" required>
<Input placeholder="输入目标数据集名称" />
<AutoComplete
options={datasets.map((dataset) => {
return {
label: (
<div className="flex items-center justify-between gap-3 py-2">
<div className="flex items-center font-sm text-gray-900">
<span className="mr-2">{dataset.icon}</span>
<span>{dataset.name}</span>
</div>
<div className="text-xs text-gray-500">{dataset.size}</div>
</div>
),
value: dataset.name,
};
})}
filterOption={(inputValue, option) => {
return option.value.toLowerCase().startsWith(inputValue.toLowerCase());
}}
placeholder="输入或选择目标数据集名称"
/>
</Form.Item>
<Form.Item
label="目标数据集类型"
Expand Down
Loading