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

feat: check before run

Browse Source
This commit is contained in:
Joel
2024-03-15 17:00:54 +08:00
parent 68f947c7e0
commit ff5ab43f9c
4 changed files with 51 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
web/app/components/workflow/nodes/_base/hooks/use-one-step-run.ts
View File

@ -7,7 +7,7 @@ import {
} from '@/app/components/workflow/hooks'
import { toNodeOutputVars } from '@/app/components/workflow/nodes/_base/components/variable/utils'
import type { CheckValidRes, CommonNodeType, InputVar, ValueSelector, Var, Variable } from '@/app/components/workflow/types'
import type { CommonNodeType, InputVar, ValueSelector, Var, Variable } from '@/app/components/workflow/types'
import { BlockEnum, InputVarType, NodeRunningStatus, VarType } from '@/app/components/workflow/types'
import { useStore as useAppStore } from '@/app/components/app/store'
import { singleNodeRun } from '@/service/workflow'
@ -25,7 +25,7 @@ type Params<T> = {
id: string
data: CommonNodeType<T>
defaultRunInputData: Record<string, any>
beforeRunCheckValid?: () => CheckValidRes
// beforeRunCheckValid?: () => CheckValidRes // has checked before run button clicked
}
const varTypeToInputVarType = (type: VarType, {
@ -53,7 +53,7 @@ const useOneStepRun = <T>({
id,
data,
defaultRunInputData,
beforeRunCheckValid = () => ({ isValid: true }),
// beforeRunCheckValid = () => ({ isValid: true }),
}: Params<T>) => {
const { t } = useTranslation()
const { getBeforeNodesInSameBranch } = useWorkflow()
@ -91,6 +91,11 @@ const useOneStepRun = <T>({
const [canShowSingleRun, setCanShowSingleRun] = useState(false)
const isShowSingleRun = data._isSingleRun && canShowSingleRun
useEffect(() => {
if (!checkValid) {
setCanShowSingleRun(true)
return
}
if (data._isSingleRun) {
const { isValid, errorMessage } = checkValid(data, t)
setCanShowSingleRun(isValid)
@ -122,14 +127,14 @@ const useOneStepRun = <T>({
const runningStatus = data._singleRunningStatus || NodeRunningStatus.NotStart
const isCompleted = runningStatus === NodeRunningStatus.Succeeded || runningStatus === NodeRunningStatus.Failed
const handleRun = async () => {
const { isValid, errorMessage } = beforeRunCheckValid()
if (!isValid) {
Toast.notify({
type: 'error',
message: errorMessage!,
})
return false
}
// const { isValid, errorMessage } = beforeRunCheckValid()
// if (!isValid) {
// Toast.notify({
// type: 'error',
// message: errorMessage!,
// })
// return false
// }
handleNodeDataUpdate({
id,
data: {