dify/web/app/components/workflow/nodes/knowledge-base/components/embedding-model.tsx

import type { DefaultModel } from '@/app/components/header/account-setting/model-provider-page/declarations'
import { memo, useCallback, useMemo } from 'react'
import { useTranslation } from 'react-i18next'
import { ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
import { useModelList } from '@/app/components/header/account-setting/model-provider-page/hooks'
import ModelSelector from '@/app/components/header/account-setting/model-provider-page/model-selector'
import { Field } from '@/app/components/workflow/nodes/_base/components/layout'

type EmbeddingModelProps = {
  embeddingModel?: string
  embeddingModelProvider?: string
  onEmbeddingModelChange?: (model: {
    embeddingModel: string
    embeddingModelProvider: string
  }) => void
  warningDot?: boolean
  readonly?: boolean
}
const EmbeddingModel = ({
  embeddingModel,
  embeddingModelProvider,
  onEmbeddingModelChange,
  warningDot = false,
  readonly = false,
}: EmbeddingModelProps) => {
  const { t } = useTranslation()
  const { data: embeddingModelList } = useModelList(ModelTypeEnum.textEmbedding)
  const embeddingModelConfig = useMemo(() => {
    if (!embeddingModel || !embeddingModelProvider) return undefined

    return {
      providerName: embeddingModelProvider,
      modelName: embeddingModel,
    }
  }, [embeddingModel, embeddingModelProvider])

  const handleEmbeddingModelChange = useCallback(
    (model: DefaultModel) => {
      onEmbeddingModelChange?.({
        embeddingModelProvider: model.provider,
        embeddingModel: model.model,
      })
    },
    [onEmbeddingModelChange],
  )

  return (
    <Field
      fieldTitleProps={{
        title: t(($) => $['form.embeddingModel'], { ns: 'datasetSettings' }),
        warningDot,
      }}
    >
      <ModelSelector
        defaultModel={
          embeddingModelConfig && {
            provider: embeddingModelConfig.providerName,
            model: embeddingModelConfig.modelName,
          }
        }
        modelList={embeddingModelList}
        onSelect={handleEmbeddingModelChange}
        readonly={readonly}
        showDeprecatedWarnIcon
      />
    </Field>
  )
}
export default memo(EmbeddingModel)