diff --git a/frontend/packages/core/src/pages/guide/LocalAiDeploy.tsx b/frontend/packages/core/src/pages/guide/LocalAiDeploy.tsx index c9e42636..64d3fc3c 100644 --- a/frontend/packages/core/src/pages/guide/LocalAiDeploy.tsx +++ b/frontend/packages/core/src/pages/guide/LocalAiDeploy.tsx @@ -57,9 +57,12 @@ const LocalAiDeploy = forwardRef((props: any, ref: any * @returns */ const deployPopularModel = async (id: string) => { - await deployLocalModel({ + const response = await deployLocalModel({ modelID: id }) + if (response.code !== STATUS_CODE.SUCCESS) { + return + } onClose?.() } @@ -84,10 +87,13 @@ const LocalAiDeploy = forwardRef((props: any, ref: any form .validateFields() .then(async (value) => { - await deployLocalModel({ + const response = await deployLocalModel({ modelID: value.model, team: value.team }) + if (response.code !== STATUS_CODE.SUCCESS) { + return + } resolve(true) }) .catch((errorInfo) => reject(errorInfo)) diff --git a/frontend/packages/core/src/pages/guide/deployModelUtil.ts b/frontend/packages/core/src/pages/guide/deployModelUtil.ts index c08b566b..a0477234 100644 --- a/frontend/packages/core/src/pages/guide/deployModelUtil.ts +++ b/frontend/packages/core/src/pages/guide/deployModelUtil.ts @@ -23,6 +23,7 @@ const useDeployLocalModel = () => { const { code, msg } = response if (code === STATUS_CODE.SUCCESS) { message.success(msg || $t(RESPONSE_TIPS.success)) + return response } else { message.error(msg || $t(RESPONSE_TIPS.error)) }