youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 09:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

annotation config

Browse Source
This commit is contained in:
JzoNg
2024-10-31 08:43:12 +08:00
parent 339dfe5e02
commit 633768cd2a
7 changed files with 14 additions and 172 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
web/app/components/base/features/new-feature-panel/annotation-reply/config-param-modal.tsx
View File

@ -18,6 +18,7 @@ type Props = {
isShow: boolean
onHide: () => void
onSave: (embeddingModel: {
plugin_id: string
embedding_provider_name: string
embedding_model_name: string
}, score: number) => void
@ -43,11 +44,13 @@ const ConfigParamModal: FC<Props> = ({
const [isLoading, setLoading] = useState(false)
const [embeddingModel, setEmbeddingModel] = useState(oldAnnotationConfig.embedding_model
? {
plugin_id: oldAnnotationConfig.embedding_model.plugin_id,
providerName: oldAnnotationConfig.embedding_model.embedding_provider_name,
modelName: oldAnnotationConfig.embedding_model.embedding_model_name,
}
: (embeddingsDefaultModel
? {
plugin_id: embeddingsDefaultModel.provider.plugin_id,
providerName: embeddingsDefaultModel.provider.provider,
modelName: embeddingsDefaultModel.model,
}
@ -67,6 +70,7 @@ const ConfigParamModal: FC<Props> = ({
}
setLoading(true)
await onSave({
plugin_id: embeddingModel.plugin_id,
embedding_provider_name: embeddingModel.providerName,
embedding_model_name: embeddingModel.modelName,
}, annotationConfig.score_threshold)
@ -107,12 +111,14 @@ const ConfigParamModal: FC<Props> = ({
<div className='pt-1'>
<ModelSelector
defaultModel={embeddingModel && {
plugin_id: '',
provider: embeddingModel.providerName,
model: embeddingModel.modelName,
}}
modelList={embeddingsModelList}
onSelect={(val) => {
setEmbeddingModel({
plugin_id: val.plugin_id,
providerName: val.provider,
modelName: val.model,
})