Browse Source

fix: 修改导出到模型

pull/132/head
cp3hnu 1 year ago
parent
commit
751fe25326
1 changed files with 5 additions and 4 deletions
  1. +5
    -4
      react-ui/src/pages/Experiment/components/ExportModelModal/index.tsx

+ 5
- 4
react-ui/src/pages/Experiment/components/ExportModelModal/index.tsx View File

@@ -1,6 +1,6 @@
import editExperimentIcon from '@/assets/img/edit-experiment.png';
import KFModal from '@/components/KFModal';
import { type ResourceData } from '@/pages/Dataset/config';
import { ResourceVersionData, type ResourceData } from '@/pages/Dataset/config';
import {
addModelVersion,
exportModelReq,
@@ -41,7 +41,7 @@ interface ExportModelModalProps extends Omit<ModalProps, 'onOk'> {
function ExportModelModal({ path, onOk, ...rest }: ExportModelModalProps) {
const [form] = Form.useForm();
const [models, setModels] = useState<ResourceData[]>([]);
const [versions, setVersions] = useState<string[]>([]);
const [versions, setVersions] = useState<ResourceVersionData[]>([]);
const [uuid] = useState(Date.now());

const layout = {
@@ -57,8 +57,9 @@ function ExportModelModal({ path, onOk, ...rest }: ExportModelModalProps) {
const getTooltip = () => {
const id = form.getFieldValue('models_id');
const name = models.find((item) => item.id === id)?.name ?? '';
const versionNames = versions.map((item: ResourceVersionData) => item.name).join('、');
const tooltip =
versions.length > 0 ? `${name}有以下版本:\n${versions.join('、')}\n注意不能重复` : undefined;
versions.length > 0 ? `${name}有以下版本:\n${versionNames}\n注意不能重复` : undefined;
return tooltip;
};

@@ -180,7 +181,7 @@ function ExportModelModal({ path, onOk, ...rest }: ExportModelModalProps) {
{ required: true, message: '请输入模型版本' },
{
validator: (_, value) => {
if (value && versions.includes(value)) {
if (value && versions.map((item) => item.name).includes(value)) {
return Promise.reject('模型版本已存在');
} else {
return Promise.resolve();


Loading…
Cancel
Save