mirror of
https://github.com/langgenius/dify.git
synced 2025-10-31 02:42:59 +00:00
31 lines
967 B
TypeScript
31 lines
967 B
TypeScript
|
import type {
|
||
|
|
NodeOutPutVar,
|
||
|
|
ValueSelector,
|
||
|
|
} from '@/app/components/workflow/types'
|
||
|
|
import { InputVarType } from '@/app/components/workflow/types'
|
||
|
|
|
||
|
|
export const findVariableWhenOnLLMVision = (valueSelector: ValueSelector, availableVars: NodeOutPutVar[]) => {
|
||
|
|
const currentVariableNode = availableVars.find((availableVar) => {
|
||
|
|
if (valueSelector[0] === 'sys' && availableVar.isStartNode)
|
||
|
|
return true
|
||
|
|
|
||
|
|
return valueSelector[0] === availableVar.nodeId
|
||
|
|
})
|
||
|
|
const currentVariable = currentVariableNode?.vars.find((variable) => {
|
||
|
|
if (valueSelector[0] === 'sys' && variable.variable === `sys.${valueSelector[1]}`)
|
||
|
|
return true
|
||
|
|
return variable.variable === valueSelector[1]
|
||
|
|
})
|
||
|
|
|
||
|
|
let formType = ''
|
||
|
|
if (currentVariable?.type === 'array[file]')
|
||
|
|
formType = InputVarType.multiFiles
|
||
|
|
if (currentVariable?.type === 'file')
|
||
|
|
formType = InputVarType.singleFile
|
||
|
|
|
||
|
|
return currentVariable && {
|
||
|
|
...currentVariable,
|
||
|
|
formType,
|
||
|
|
}
|
||
|
|
}
|