Improve ModelTypeEnum type (#3051)

This commit is contained in:
Nanguan Lin 2024-04-04 15:54:59 +08:00 • committed by GitHub
commit 718ac3f83b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 47 additions and 44 deletions

View file

@ -14,6 +14,7 @@ import StepTwo from '@/app/components/datasets/create/step-two'
import AccountSetting from '@/app/components/header/account-setting'
import AppUnavailable from '@/app/components/base/app-unavailable'
import { useDefaultModel } from '@/app/components/header/account-setting/model-provider-page/hooks'
import { ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
type DocumentSettingsProps = {
datasetId: string
@ -26,7 +27,7 @@ const DocumentSettings = ({ datasetId, documentId }: DocumentSettingsProps) => {
const [isShowSetAPIKey, { setTrue: showSetAPIKey, setFalse: hideSetAPIkey }] = useBoolean()
const [hasError, setHasError] = useState(false)
const { indexingTechnique, dataset } = useContext(DatasetDetailContext)
const { data: embeddingsDefaultModel } = useDefaultModel(2)
const { data: embeddingsDefaultModel } = useDefaultModel(ModelTypeEnum.textEmbedding)
const saveHandler = () => router.push(`/datasets/${datasetId}/documents/${documentId}`)