fix: update model list (#1967)

This commit is contained in:
zxhlyh
2024-01-08 18:54:39 +08:00
committed by GitHub
parent 69d42ae95b
commit 853cdd741f
7 changed files with 87 additions and 47 deletions

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)
},
})