feat: classify single run

pull/21369/head
Joel 1 year ago
parent db31c1548b
commit eb22bda9da

@ -55,7 +55,7 @@ const useSingleRunFormParams = ({
const inputVarValues = (() => { const inputVarValues = (() => {
const vars: Record<string, any> = {} const vars: Record<string, any> = {}
Object.keys(runInputData) Object.keys(runInputData)
.filter(key => !['#context#', '#files#'].includes(key)) .filter(key => !['#files#'].includes(key))
.forEach((key) => { .forEach((key) => {
vars[key] = runInputData[key] vars[key] = runInputData[key]
}) })
@ -65,7 +65,6 @@ const useSingleRunFormParams = ({
const setInputVarValues = useCallback((newPayload: Record<string, any>) => { const setInputVarValues = useCallback((newPayload: Record<string, any>) => {
const newVars = { const newVars = {
...newPayload, ...newPayload,
'#context#': runInputDataRef.current['#context#'],
'#files#': runInputDataRef.current['#files#'], '#files#': runInputDataRef.current['#files#'],
} }
setRunInputData?.(newVars) setRunInputData?.(newVars)
@ -118,8 +117,29 @@ const useSingleRunFormParams = ({
return forms return forms
})() })()
const getDependentVars = () => {
const promptVars = varInputs.map(item => item.variable.slice(1, -1).split('.'))
const vars = [payload.query_variable_selector, ...promptVars]
if (isVisionModel && payload.vision?.enabled && payload.vision?.configs?.variable_selector) {
const visionVar = payload.vision.configs.variable_selector
vars.push(visionVar)
}
return vars
}
const getDependentVar = (variable: string) => {
if(variable === 'query')
return payload.query_variable_selector
if(variable === '#files#')
return payload.vision.configs?.variable_selector
return false
}
return { return {
forms, forms,
getDependentVars,
getDependentVar,
} }
} }

Loading…
Cancel
Save