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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
| import { useCallback } from 'react'
| import {
| useNodes,
| } from 'reactflow'
| import { uniqBy } from 'lodash-es'
| import {
| useIsChatMode,
| useWorkflow,
| useWorkflowVariables,
| } from '../../hooks'
| import type {
| Node,
| Var,
| } from '../../types'
| import { AssignerNodeInputType, WriteMode } from './types'
|
| export const useGetAvailableVars = () => {
| const nodes: Node[] = useNodes()
| const { getBeforeNodesInSameBranchIncludeParent } = useWorkflow()
| const { getNodeAvailableVars } = useWorkflowVariables()
| const isChatMode = useIsChatMode()
| const getAvailableVars = useCallback((nodeId: string, handleId: string, filterVar: (v: Var) => boolean, hideEnv = false) => {
| const availableNodes: Node[] = []
| const currentNode = nodes.find(node => node.id === nodeId)!
|
| if (!currentNode)
| return []
|
| const beforeNodes = getBeforeNodesInSameBranchIncludeParent(nodeId)
| availableNodes.push(...beforeNodes)
| const parentNode = nodes.find(node => node.id === currentNode.parentId)
|
| if (hideEnv) {
| return getNodeAvailableVars({
| parentNode,
| beforeNodes: uniqBy(availableNodes, 'id').filter(node => node.id !== nodeId),
| isChatMode,
| hideEnv,
| hideChatVar: hideEnv,
| filterVar,
| })
| .map(node => ({
| ...node,
| vars: node.isStartNode ? node.vars.filter(v => !v.variable.startsWith('sys.')) : node.vars,
| }))
| .filter(item => item.vars.length > 0)
| }
|
| return getNodeAvailableVars({
| parentNode,
| beforeNodes: uniqBy(availableNodes, 'id').filter(node => node.id !== nodeId),
| isChatMode,
| filterVar,
| })
| }, [nodes, getBeforeNodesInSameBranchIncludeParent, getNodeAvailableVars, isChatMode])
|
| return getAvailableVars
| }
|
| export const useHandleAddOperationItem = () => {
| return useCallback((list: any[]) => {
| const newItem = {
| variable_selector: [],
| write_mode: WriteMode.overwrite,
| input_type: AssignerNodeInputType.variable,
| value: '',
| }
| return [...list, newItem]
| }, [])
| }
|
|
