mirror of
https://github.com/langgenius/dify.git
synced 2025-07-15 05:04:52 +00:00

The frontend of feat: Persist Variables for Enhanced Debugging Workflow (#20699). Co-authored-by: jZonG <jzongcode@gmail.com>
93 lines
2.7 KiB
TypeScript
93 lines
2.7 KiB
TypeScript
import type { MutableRefObject } from 'react'
|
|
import type { InputVar, ValueSelector, Variable } from '@/app/components/workflow/types'
|
|
import { useCallback } from 'react'
|
|
import type { VariableAssignerNodeType } from './types'
|
|
|
|
type Params = {
|
|
id: string,
|
|
payload: VariableAssignerNodeType,
|
|
runInputData: Record<string, any>
|
|
runInputDataRef: MutableRefObject<Record<string, any>>
|
|
getInputVars: (textList: string[]) => InputVar[]
|
|
setRunInputData: (data: Record<string, any>) => void
|
|
toVarInputs: (variables: Variable[]) => InputVar[]
|
|
varSelectorsToVarInputs: (variables: ValueSelector[]) => InputVar[]
|
|
}
|
|
const useSingleRunFormParams = ({
|
|
payload,
|
|
runInputData,
|
|
setRunInputData,
|
|
varSelectorsToVarInputs,
|
|
}: Params) => {
|
|
const setInputVarValues = useCallback((newPayload: Record<string, any>) => {
|
|
setRunInputData(newPayload)
|
|
}, [setRunInputData])
|
|
const inputVarValues = (() => {
|
|
const vars: Record<string, any> = {}
|
|
Object.keys(runInputData)
|
|
.forEach((key) => {
|
|
vars[key] = runInputData[key]
|
|
})
|
|
return vars
|
|
})()
|
|
|
|
const forms = (() => {
|
|
const allInputs: ValueSelector[] = []
|
|
const isGroupEnabled = !!payload.advanced_settings?.group_enabled
|
|
if (!isGroupEnabled && payload.variables && payload.variables.length) {
|
|
payload.variables.forEach((varSelector) => {
|
|
allInputs.push(varSelector)
|
|
})
|
|
}
|
|
if (isGroupEnabled && payload.advanced_settings && payload.advanced_settings.groups && payload.advanced_settings.groups.length) {
|
|
payload.advanced_settings.groups.forEach((group) => {
|
|
group.variables?.forEach((varSelector) => {
|
|
allInputs.push(varSelector)
|
|
})
|
|
})
|
|
}
|
|
|
|
const varInputs = varSelectorsToVarInputs(allInputs)
|
|
// remove duplicate inputs
|
|
const existVarsKey: Record<string, boolean> = {}
|
|
const uniqueVarInputs: InputVar[] = []
|
|
varInputs.forEach((input) => {
|
|
if(!input)
|
|
return
|
|
if (!existVarsKey[input.variable]) {
|
|
existVarsKey[input.variable] = true
|
|
uniqueVarInputs.push({
|
|
...input,
|
|
required: false, // just one of the inputs is required
|
|
})
|
|
}
|
|
})
|
|
return [
|
|
{
|
|
inputs: uniqueVarInputs,
|
|
values: inputVarValues,
|
|
onChange: setInputVarValues,
|
|
},
|
|
]
|
|
})()
|
|
|
|
const getDependentVars = () => {
|
|
if(payload.advanced_settings?.group_enabled) {
|
|
const vars: ValueSelector[][] = []
|
|
payload.advanced_settings.groups.forEach((group) => {
|
|
if(group.variables)
|
|
vars.push([...group.variables])
|
|
})
|
|
return vars
|
|
}
|
|
return [payload.variables]
|
|
}
|
|
|
|
return {
|
|
forms,
|
|
getDependentVars,
|
|
}
|
|
}
|
|
|
|
export default useSingleRunFormParams
|