mirror of https://github.com/langgenius/dify.git
fix: model list refresh when change credential (#26421)
This commit is contained in:
parent
70bc5ca7f4
commit
595df172a8
|
|
@ -323,15 +323,18 @@ export const useRefreshModel = () => {
|
||||||
const { eventEmitter } = useEventEmitterContextContext()
|
const { eventEmitter } = useEventEmitterContextContext()
|
||||||
const updateModelProviders = useUpdateModelProviders()
|
const updateModelProviders = useUpdateModelProviders()
|
||||||
const updateModelList = useUpdateModelList()
|
const updateModelList = useUpdateModelList()
|
||||||
const handleRefreshModel = useCallback((provider: ModelProvider, configurationMethod: ConfigurationMethodEnum, CustomConfigurationModelFixedFields?: CustomConfigurationModelFixedFields) => {
|
const handleRefreshModel = useCallback((
|
||||||
|
provider: ModelProvider,
|
||||||
|
CustomConfigurationModelFixedFields?: CustomConfigurationModelFixedFields,
|
||||||
|
refreshModelList?: boolean,
|
||||||
|
) => {
|
||||||
updateModelProviders()
|
updateModelProviders()
|
||||||
|
|
||||||
provider.supported_model_types.forEach((type) => {
|
provider.supported_model_types.forEach((type) => {
|
||||||
updateModelList(type)
|
updateModelList(type)
|
||||||
})
|
})
|
||||||
|
|
||||||
if (configurationMethod === ConfigurationMethodEnum.customizableModel
|
if (refreshModelList && provider.custom_configuration.status === CustomConfigurationStatusEnum.active) {
|
||||||
&& provider.custom_configuration.status === CustomConfigurationStatusEnum.active) {
|
|
||||||
eventEmitter?.emit({
|
eventEmitter?.emit({
|
||||||
type: UPDATE_MODEL_PROVIDER_CUSTOM_MODEL_LIST,
|
type: UPDATE_MODEL_PROVIDER_CUSTOM_MODEL_LIST,
|
||||||
payload: provider.provider,
|
payload: provider.provider,
|
||||||
|
|
|
||||||
|
|
@ -90,7 +90,7 @@ export const useAuth = (
|
||||||
type: 'success',
|
type: 'success',
|
||||||
message: t('common.api.actionSuccess'),
|
message: t('common.api.actionSuccess'),
|
||||||
})
|
})
|
||||||
handleRefreshModel(provider, configurationMethod, undefined)
|
handleRefreshModel(provider, undefined, true)
|
||||||
}
|
}
|
||||||
finally {
|
finally {
|
||||||
handleSetDoingAction(false)
|
handleSetDoingAction(false)
|
||||||
|
|
@ -125,7 +125,7 @@ export const useAuth = (
|
||||||
type: 'success',
|
type: 'success',
|
||||||
message: t('common.api.actionSuccess'),
|
message: t('common.api.actionSuccess'),
|
||||||
})
|
})
|
||||||
handleRefreshModel(provider, configurationMethod, undefined)
|
handleRefreshModel(provider, undefined, true)
|
||||||
onRemove?.(pendingOperationCredentialId.current ?? '')
|
onRemove?.(pendingOperationCredentialId.current ?? '')
|
||||||
closeConfirmDelete()
|
closeConfirmDelete()
|
||||||
}
|
}
|
||||||
|
|
@ -147,7 +147,7 @@ export const useAuth = (
|
||||||
|
|
||||||
if (res.result === 'success') {
|
if (res.result === 'success') {
|
||||||
notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
|
notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
|
||||||
handleRefreshModel(provider, configurationMethod, undefined)
|
handleRefreshModel(provider, undefined, !payload.credential_id)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
finally {
|
finally {
|
||||||
|
|
|
||||||
|
|
@ -159,7 +159,7 @@ const ModelLoadBalancingModal = ({
|
||||||
)
|
)
|
||||||
if (res.result === 'success') {
|
if (res.result === 'success') {
|
||||||
notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
|
notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
|
||||||
handleRefreshModel(provider, configurateMethod, currentCustomConfigurationModelFixedFields)
|
handleRefreshModel(provider, currentCustomConfigurationModelFixedFields, false)
|
||||||
onSave?.(provider.provider)
|
onSave?.(provider.provider)
|
||||||
onClose?.()
|
onClose?.()
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue