feat: valid question classify

This commit is contained in:
Joel 2024-03-19 16:42:02 +08:00
parent 653917649d
commit 3dfcd9ca67
3 changed files with 15 additions and 8 deletions

View File

@ -80,7 +80,7 @@ const useOneStepRun = <T>({
moreDataForCheckValid,
}: Params<T>) => {
const { t } = useTranslation()
const { getBeforeNodesInSameBranch } = useWorkflow()
const { getBeforeNodesInSameBranch } = useWorkflow() as any
const isChatMode = useIsChatMode()
const allOutputVars = toNodeOutputVars(getBeforeNodesInSameBranch(id), isChatMode)

View File

@ -3,6 +3,8 @@ import { BlockEnum } from '../../types'
import type { QuestionClassifierNodeType } from './types'
import { ALL_CHAT_AVAILABLE_BLOCKS, ALL_COMPLETION_AVAILABLE_BLOCKS } from '@/app/components/workflow/constants'
const i18nPrefix = 'workflow'
const nodeDefault: NodeDefault<QuestionClassifierNodeType> = {
defaultValue: {
query_variable_selector: [],
@ -26,15 +28,19 @@ const nodeDefault: NodeDefault<QuestionClassifierNodeType> = {
const nodes = isChatMode ? ALL_CHAT_AVAILABLE_BLOCKS : ALL_COMPLETION_AVAILABLE_BLOCKS
return nodes.filter(type => type !== BlockEnum.VariableAssigner)
},
checkValid(payload: QuestionClassifierNodeType) {
let isValid = true
checkValid(payload: QuestionClassifierNodeType, 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.questionClassifiers.inputVars`) })
if (!errorMessages && !payload.model.provider)
errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.nodes.questionClassifiers.model`) })
if (!errorMessages && (!payload.classes || payload.classes.length === 0))
errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.nodes.questionClassifiers.class`) })
return {
isValid,
isValid: !errorMessages,
errorMessage: errorMessages,
}
},

View File

@ -56,6 +56,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
value={inputs.query_variable_selector}
onChange={handleQueryVarChange}
filterVar={filterVar}
width={387}
/>
</Field>
<Field