| | |
| | | return [VarType.number, VarType.string].includes(varPayload.type) |
| | | }, []) |
| | | |
| | | const filterVisionInputVar = useCallback((varPayload: Var) => { |
| | | return [VarType.file, VarType.arrayFile].includes(varPayload.type) |
| | | }, []) |
| | | |
| | | const { |
| | | availableVars, |
| | | availableNodesWithParent, |
| | | } = useAvailableVarList(id, { |
| | | onlyLeafNodeVar: false, |
| | | filterVar: filterInputVar, |
| | | }) |
| | | |
| | | const { |
| | | availableVars: availableVisionVars, |
| | | } = useAvailableVarList(id, { |
| | | onlyLeafNodeVar: false, |
| | | filterVar: filterVisionInputVar, |
| | | }) |
| | | |
| | | const hasSetBlockStatus = { |
| | |
| | | handleRun, |
| | | handleStop, |
| | | runInputData, |
| | | runInputDataRef, |
| | | setRunInputData, |
| | | runResult, |
| | | } = useOneStepRun<QuestionClassifierNodeType>({ |
| | | id, |
| | | data: inputs, |
| | | defaultRunInputData: { |
| | | 'query': '', |
| | | '#files#': [], |
| | | query: '', |
| | | }, |
| | | }) |
| | | |
| | |
| | | setRunInputData(newPayload) |
| | | }, [setRunInputData]) |
| | | |
| | | const visionFiles = runInputData['#files#'] |
| | | const setVisionFiles = useCallback((newFiles: any[]) => { |
| | | setRunInputData({ |
| | | ...runInputDataRef.current, |
| | | '#files#': newFiles, |
| | | }) |
| | | }, [runInputDataRef, setRunInputData]) |
| | | |
| | | const filterVar = useCallback((varPayload: Var) => { |
| | | return varPayload.type === VarType.string |
| | | }, []) |
| | |
| | | hasSetBlockStatus, |
| | | availableVars, |
| | | availableNodesWithParent, |
| | | availableVisionVars, |
| | | handleInstructionChange, |
| | | varInputs, |
| | | inputVarValues, |
| | |
| | | query, |
| | | setQuery, |
| | | runResult, |
| | | visionFiles, |
| | | setVisionFiles, |
| | | } |
| | | } |
| | | |