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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
| import { useCallback, useMemo } from 'react'
| import produce from 'immer'
| import { useStoreApi } from 'reactflow'
|
| import type { ValueSelector, Var } from '../../types'
| import { InputVarType, VarType } from '../../types'
| import type { DocExtractorNodeType } from './types'
| import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
| import useOneStepRun from '@/app/components/workflow/nodes/_base/hooks/use-one-step-run'
| import {
| useIsChatMode,
| useNodesReadOnly,
| useWorkflow,
| useWorkflowVariables,
| } from '@/app/components/workflow/hooks'
|
| const useConfig = (id: string, payload: DocExtractorNodeType) => {
| const { nodesReadOnly: readOnly } = useNodesReadOnly()
| const { inputs, setInputs } = useNodeCrud<DocExtractorNodeType>(id, payload)
|
| const filterVar = useCallback((varPayload: Var) => {
| return varPayload.type === VarType.file || varPayload.type === VarType.arrayFile
| }, [])
|
| const isChatMode = useIsChatMode()
|
| const store = useStoreApi()
| const { getBeforeNodesInSameBranch } = useWorkflow()
| const {
| getNodes,
| } = store.getState()
| const currentNode = getNodes().find(n => n.id === id)
| const isInIteration = payload.isInIteration
| const iterationNode = isInIteration ? getNodes().find(n => n.id === currentNode!.parentId) : null
| const isInLoop = payload.isInLoop
| const loopNode = isInLoop ? getNodes().find(n => n.id === currentNode!.parentId) : null
| const availableNodes = useMemo(() => {
| return getBeforeNodesInSameBranch(id)
| }, [getBeforeNodesInSameBranch, id])
|
| const { getCurrentVariableType } = useWorkflowVariables()
| const getType = useCallback((variable?: ValueSelector) => {
| const varType = getCurrentVariableType({
| parentNode: isInIteration ? iterationNode : loopNode,
| valueSelector: variable || [],
| availableNodes,
| isChatMode,
| isConstant: false,
| })
| return varType
| }, [getCurrentVariableType, isInIteration, availableNodes, isChatMode, iterationNode, loopNode])
|
| const handleVarChanges = useCallback((variable: ValueSelector | string) => {
| const newInputs = produce(inputs, (draft) => {
| draft.variable_selector = variable as ValueSelector
| draft.is_array_file = getType(draft.variable_selector) === VarType.arrayFile
| })
| setInputs(newInputs)
| }, [getType, inputs, setInputs])
|
| // single run
| const {
| isShowSingleRun,
| hideSingleRun,
| runningStatus,
| isCompleted,
| handleRun,
| handleStop,
| runInputData,
| setRunInputData,
| runResult,
| } = useOneStepRun<DocExtractorNodeType>({
| id,
| data: inputs,
| defaultRunInputData: { files: [] },
| })
| const varInputs = [{
| label: inputs.title,
| variable: 'files',
| type: InputVarType.multiFiles,
| required: true,
| }]
|
| const files = runInputData.files
| const setFiles = useCallback((newFiles: []) => {
| setRunInputData({
| ...runInputData,
| files: newFiles,
| })
| }, [runInputData, setRunInputData])
|
| return {
| readOnly,
| inputs,
| filterVar,
| handleVarChanges,
| // single run
| isShowSingleRun,
| hideSingleRun,
| runningStatus,
| isCompleted,
| handleRun,
| handleStop,
| varInputs,
| files,
| setFiles,
| runResult,
| }
| }
|
| export default useConfig
|
|
