jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: update model list (#1967)

Browse source
This commit is contained in:
zxhlyh 2024-01-08 18:54:39 +08:00 • committed by GitHub
commit 853cdd741f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 87 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

9
web/app/components/header/account-setting/model-provider-page/model-selector/popup-item.tsx
View file

@ -8,7 +8,7 @@ import type {
import {
useLanguage,
useUpdateModelList,
useUpdateModelProvidersAndModelList,
useUpdateModelProviders,
} from '../hooks'
import ModelIcon from '../model-icon'
import ModelName from '../model-name'
@ -16,7 +16,6 @@ import {
ConfigurateMethodEnum,
MODEL_STATUS_TEXT,
ModelStatusEnum,
ModelTypeEnum,
} from '../declarations'
import { Check } from '@/app/components/base/icons/src/vender/line/general'
import { useModalContext } from '@/context/modal-context'
@ -38,7 +37,7 @@ const PopupItem: FC<PopupItemProps> = ({
const { setShowModelModal } = useModalContext()
const { modelProviders } = useProviderContext()
const updateModelList = useUpdateModelList()
const updateModelProvidersAndModelList = useUpdateModelProvidersAndModelList()
const updateModelProviders = useUpdateModelProviders()
const currentProvider = modelProviders.find(provider => provider.provider === model.provider)!
const handleSelect = (provider: string, modelItem: ModelItem) => {
if (modelItem.status !== ModelStatusEnum.active)
@ -53,11 +52,11 @@ const PopupItem: FC<PopupItemProps> = ({
currentConfigurateMethod: ConfigurateMethodEnum.predefinedModel,
},
onSaveCallback: () => {
updateModelProvidersAndModelList()
updateModelProviders()
const modelType = model.models[0].model_type
if (modelType !== ModelTypeEnum.textGeneration)
if (modelType)
updateModelList(modelType)
},
})