| @@ -323,15 +323,18 @@ export const useRefreshModel = () => { | |||
| const { eventEmitter } = useEventEmitterContextContext() | |||
| const updateModelProviders = useUpdateModelProviders() | |||
| const updateModelList = useUpdateModelList() | |||
| const handleRefreshModel = useCallback((provider: ModelProvider, configurationMethod: ConfigurationMethodEnum, CustomConfigurationModelFixedFields?: CustomConfigurationModelFixedFields) => { | |||
| const handleRefreshModel = useCallback(( | |||
| provider: ModelProvider, | |||
| CustomConfigurationModelFixedFields?: CustomConfigurationModelFixedFields, | |||
| refreshModelList?: boolean, | |||
| ) => { | |||
| updateModelProviders() | |||
| provider.supported_model_types.forEach((type) => { | |||
| updateModelList(type) | |||
| }) | |||
| if (configurationMethod === ConfigurationMethodEnum.customizableModel | |||
| && provider.custom_configuration.status === CustomConfigurationStatusEnum.active) { | |||
| if (refreshModelList && provider.custom_configuration.status === CustomConfigurationStatusEnum.active) { | |||
| eventEmitter?.emit({ | |||
| type: UPDATE_MODEL_PROVIDER_CUSTOM_MODEL_LIST, | |||
| payload: provider.provider, | |||
| @@ -90,7 +90,7 @@ export const useAuth = ( | |||
| type: 'success', | |||
| message: t('common.api.actionSuccess'), | |||
| }) | |||
| handleRefreshModel(provider, configurationMethod, undefined) | |||
| handleRefreshModel(provider, undefined, true) | |||
| } | |||
| finally { | |||
| handleSetDoingAction(false) | |||
| @@ -125,7 +125,7 @@ export const useAuth = ( | |||
| type: 'success', | |||
| message: t('common.api.actionSuccess'), | |||
| }) | |||
| handleRefreshModel(provider, configurationMethod, undefined) | |||
| handleRefreshModel(provider, undefined, true) | |||
| onRemove?.(pendingOperationCredentialId.current ?? '') | |||
| closeConfirmDelete() | |||
| } | |||
| @@ -147,7 +147,7 @@ export const useAuth = ( | |||
| if (res.result === 'success') { | |||
| notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') }) | |||
| handleRefreshModel(provider, configurationMethod, undefined) | |||
| handleRefreshModel(provider, undefined, !payload.credential_id) | |||
| } | |||
| } | |||
| finally { | |||
| @@ -159,7 +159,7 @@ const ModelLoadBalancingModal = ({ | |||
| ) | |||
| if (res.result === 'success') { | |||
| notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') }) | |||
| handleRefreshModel(provider, configurateMethod, currentCustomConfigurationModelFixedFields) | |||
| handleRefreshModel(provider, currentCustomConfigurationModelFixedFields, false) | |||
| onSave?.(provider.provider) | |||
| onClose?.() | |||
| } | |||