opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-13 22:57:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: retrieval check valid

Browse Source
This commit is contained in:
Joel 2024-03-19 13:50:03 +08:00
parent a436550dff
commit 6bfd61a887
3 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
web/app/components/workflow/nodes/knowledge-retrieval/default.ts
View File

@ -4,6 +4,7 @@ import type { KnowledgeRetrievalNodeType } from './types'
import { ALL_CHAT_AVAILABLE_BLOCKS, ALL_COMPLETION_AVAILABLE_BLOCKS } from '@/app/components/workflow/constants'
import { RETRIEVE_TYPE } from '@/types/app'
const i18nPrefix = 'workflow'
const nodeDefault: NodeDefault<KnowledgeRetrievalNodeType> = {
defaultValue: {
@ -21,15 +22,22 @@ const nodeDefault: NodeDefault<KnowledgeRetrievalNodeType> = {
const nodes = isChatMode ? ALL_CHAT_AVAILABLE_BLOCKS : ALL_COMPLETION_AVAILABLE_BLOCKS
return nodes
},
checkValid(payload: KnowledgeRetrievalNodeType) {
let isValid = true
checkValid(payload: KnowledgeRetrievalNodeType, t: any) {
let errorMessages = ''
if (payload.type) {
isValid = true
errorMessages = ''
}
if (!errorMessages && (!payload.query_variable_selector || payload.query_variable_selector.length === 0))
errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.nodes.knowledgeRetrieval.queryVariable`) })
if (!errorMessages && (!payload.dataset_ids || payload.dataset_ids.length === 0))
errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.nodes.knowledgeRetrieval.knowledge`) })
if (!errorMessages && payload.retrieval_mode === RETRIEVE_TYPE.multiWay && !payload.multiple_retrieval_config?.reranking_model?.provider)
errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.errorMsg.fields.rerankModel`) })
if (!errorMessages && payload.retrieval_mode === RETRIEVE_TYPE.oneWay && !payload.single_retrieval_config?.model?.provider)
errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t('common.modelProvider.systemReasoningModel.key') })
return {
isValid,
isValid: !errorMessages,
errorMessage: errorMessages,
}
},

1
web/i18n/en-US/workflow.ts
View File

@ -36,6 +36,7 @@ const translation = {
variableValue: 'Variable Value',
code: 'Code',
model: 'Model',
rerankModel: 'Rerank Model',
},
},
singleRun: {

1
web/i18n/zh-Hans/workflow.ts
View File

@ -36,6 +36,7 @@ const translation = {
variableValue: '变量值',
code: '代码',
model: '模型',
rerankModel: 'Rerank 模型',
},
},
singleRun: {