diff --git a/web/src/views/ModelManagement/components/CustomModelModal.tsx b/web/src/views/ModelManagement/components/CustomModelModal.tsx index 47928d87..66c16111 100644 --- a/web/src/views/ModelManagement/components/CustomModelModal.tsx +++ b/web/src/views/ModelManagement/components/CustomModelModal.tsx @@ -44,7 +44,8 @@ const CustomModelModal = forwardRef( }; const handleUpdate = (data: CustomModelForm) => { setLoading(true) - const res = isEdit ? updateCustomModel(model.id, data) : addCustomModel(data) + const { type, provider, ...rest} = data + const res = isEdit ? updateCustomModel(model.id, rest) : addCustomModel(data) res.then(() => { refresh && refresh() diff --git a/web/src/views/ModelManagement/components/GroupModelModal.tsx b/web/src/views/ModelManagement/components/GroupModelModal.tsx index c66288e5..e8734114 100644 --- a/web/src/views/ModelManagement/components/GroupModelModal.tsx +++ b/web/src/views/ModelManagement/components/GroupModelModal.tsx @@ -75,8 +75,9 @@ const GroupModelModal = forwardRef(({ const handleUpdate = (data: CompositeModelForm) => { setLoading(true) + const { type, ...rest } = data const res = isEdit - ? updateCompositeModel(model.id, data) + ? updateCompositeModel(model.id, { ...rest }) : addCompositeModel(data) res.then(() => { diff --git a/web/src/views/ModelManagement/types.ts b/web/src/views/ModelManagement/types.ts index 1967f393..dc143c94 100644 --- a/web/src/views/ModelManagement/types.ts +++ b/web/src/views/ModelManagement/types.ts @@ -17,7 +17,7 @@ export interface DescriptionItem { export interface CompositeModelForm { logo?: any; name: string; - type: string; + type?: string; description: string; api_key_ids: ModelApiKey[] | string[]; } @@ -119,8 +119,8 @@ export interface ModelSquareDetailRef { } export interface CustomModelForm { name: string; - type: string; - provider: string; + type?: string; + provider?: string; logo?: any; description: string; is_official: boolean;