1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
| import useNodeInfo from './use-node-info'
| import {
| useIsChatMode,
| useWorkflow,
| useWorkflowVariables,
| } from '@/app/components/workflow/hooks'
| import type { Node, ValueSelector, Var } from '@/app/components/workflow/types'
| type Params = {
| onlyLeafNodeVar?: boolean
| hideEnv?: boolean
| hideChatVar?: boolean
| filterVar: (payload: Var, selector: ValueSelector) => boolean
| passedInAvailableNodes?: Node[]
| }
|
| // TODO: loop type?
| const useAvailableVarList = (nodeId: string, {
| onlyLeafNodeVar,
| filterVar,
| hideEnv,
| hideChatVar,
| passedInAvailableNodes,
| }: Params = {
| onlyLeafNodeVar: false,
| filterVar: () => true,
| }) => {
| const { getTreeLeafNodes, getBeforeNodesInSameBranchIncludeParent } = useWorkflow()
| const { getNodeAvailableVars } = useWorkflowVariables()
| const isChatMode = useIsChatMode()
|
| const availableNodes = passedInAvailableNodes || (onlyLeafNodeVar ? getTreeLeafNodes(nodeId) : getBeforeNodesInSameBranchIncludeParent(nodeId))
|
| const {
| parentNode: iterationNode,
| } = useNodeInfo(nodeId)
|
| const availableVars = getNodeAvailableVars({
| parentNode: iterationNode,
| beforeNodes: availableNodes,
| isChatMode,
| filterVar,
| hideEnv,
| hideChatVar,
| })
|
| return {
| availableVars,
| availableNodes,
| availableNodesWithParent: availableNodes,
| }
| }
|
| export default useAvailableVarList
|
|
