You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. import type { FC } from 'react'
  2. import React from 'react'
  3. import { useTranslation } from 'react-i18next'
  4. import VarReferencePicker from '../_base/components/variable/var-reference-picker'
  5. import Split from '../_base/components/split'
  6. import ResultPanel from '../../run/result-panel'
  7. import { MAX_ITERATION_PARALLEL_NUM, MIN_ITERATION_PARALLEL_NUM } from '../../constants'
  8. import type { IterationNodeType } from './types'
  9. import useConfig from './use-config'
  10. import { ErrorHandleMode, InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
  11. import Field from '@/app/components/workflow/nodes/_base/components/field'
  12. import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
  13. import Switch from '@/app/components/base/switch'
  14. import Select from '@/app/components/base/select'
  15. import Slider from '@/app/components/base/slider'
  16. import Input from '@/app/components/base/input'
  17. import formatTracing from '@/app/components/workflow/run/utils/format-log'
  18. import { useLogs } from '@/app/components/workflow/run/hooks'
  19. const i18nPrefix = 'workflow.nodes.iteration'
  20. const Panel: FC<NodePanelProps<IterationNodeType>> = ({
  21. id,
  22. data,
  23. }) => {
  24. const { t } = useTranslation()
  25. const responseMethod = [
  26. {
  27. value: ErrorHandleMode.Terminated,
  28. name: t(`${i18nPrefix}.ErrorMethod.operationTerminated`),
  29. },
  30. {
  31. value: ErrorHandleMode.ContinueOnError,
  32. name: t(`${i18nPrefix}.ErrorMethod.continueOnError`),
  33. },
  34. {
  35. value: ErrorHandleMode.RemoveAbnormalOutput,
  36. name: t(`${i18nPrefix}.ErrorMethod.removeAbnormalOutput`),
  37. },
  38. ]
  39. const {
  40. readOnly,
  41. inputs,
  42. filterInputVar,
  43. handleInputChange,
  44. childrenNodeVars,
  45. iterationChildrenNodes,
  46. handleOutputVarChange,
  47. isShowSingleRun,
  48. hideSingleRun,
  49. runningStatus,
  50. handleRun,
  51. handleStop,
  52. runResult,
  53. inputVarValues,
  54. setInputVarValues,
  55. usedOutVars,
  56. iterator,
  57. setIterator,
  58. iteratorInputKey,
  59. iterationRunResult,
  60. changeParallel,
  61. changeErrorResponseMode,
  62. changeParallelNums,
  63. } = useConfig(id, data)
  64. const nodeInfo = formatTracing(iterationRunResult, t)[0]
  65. const logsParams = useLogs()
  66. return (
  67. <div className='pb-2 pt-2'>
  68. <div className='space-y-4 px-4 pb-4'>
  69. <Field
  70. title={t(`${i18nPrefix}.input`)}
  71. required
  72. operations={(
  73. <div className='system-2xs-medium-uppercase flex h-[18px] items-center rounded-[5px] border border-divider-deep px-1 capitalize text-text-tertiary'>Array</div>
  74. )}
  75. >
  76. <VarReferencePicker
  77. readonly={readOnly}
  78. nodeId={id}
  79. isShowNodeName
  80. value={inputs.iterator_selector || []}
  81. onChange={handleInputChange}
  82. filterVar={filterInputVar}
  83. />
  84. </Field>
  85. </div>
  86. <Split />
  87. <div className='mt-2 space-y-4 px-4 pb-4'>
  88. <Field
  89. title={t(`${i18nPrefix}.output`)}
  90. required
  91. operations={(
  92. <div className='system-2xs-medium-uppercase flex h-[18px] items-center rounded-[5px] border border-divider-deep px-1 capitalize text-text-tertiary'>Array</div>
  93. )}
  94. >
  95. <VarReferencePicker
  96. readonly={readOnly}
  97. nodeId={id}
  98. isShowNodeName
  99. value={inputs.output_selector || []}
  100. onChange={handleOutputVarChange}
  101. availableNodes={iterationChildrenNodes}
  102. availableVars={childrenNodeVars}
  103. />
  104. </Field>
  105. </div>
  106. <div className='px-4 pb-2'>
  107. <Field title={t(`${i18nPrefix}.parallelMode`)} tooltip={<div className='w-[230px]'>{t(`${i18nPrefix}.parallelPanelDesc`)}</div>} inline>
  108. <Switch defaultValue={inputs.is_parallel} onChange={changeParallel} />
  109. </Field>
  110. </div>
  111. {
  112. inputs.is_parallel && (<div className='px-4 pb-2'>
  113. <Field title={t(`${i18nPrefix}.MaxParallelismTitle`)} isSubTitle tooltip={<div className='w-[230px]'>{t(`${i18nPrefix}.MaxParallelismDesc`)}</div>}>
  114. <div className='row flex'>
  115. <Input type='number' wrapperClassName='w-18 mr-4 ' max={MAX_ITERATION_PARALLEL_NUM} min={MIN_ITERATION_PARALLEL_NUM} value={inputs.parallel_nums} onChange={(e) => { changeParallelNums(Number(e.target.value)) }} />
  116. <Slider
  117. value={inputs.parallel_nums}
  118. onChange={changeParallelNums}
  119. max={MAX_ITERATION_PARALLEL_NUM}
  120. min={MIN_ITERATION_PARALLEL_NUM}
  121. className=' mt-4 flex-1 shrink-0'
  122. />
  123. </div>
  124. </Field>
  125. </div>)
  126. }
  127. <Split />
  128. <div className='px-4 py-2'>
  129. <Field title={t(`${i18nPrefix}.errorResponseMethod`)} >
  130. <Select items={responseMethod} defaultValue={inputs.error_handle_mode} onSelect={changeErrorResponseMode} allowSearch={false} />
  131. </Field>
  132. </div>
  133. {isShowSingleRun && (
  134. <BeforeRunForm
  135. nodeName={inputs.title}
  136. onHide={hideSingleRun}
  137. forms={[
  138. {
  139. inputs: [...usedOutVars],
  140. values: inputVarValues,
  141. onChange: setInputVarValues,
  142. },
  143. {
  144. label: t(`${i18nPrefix}.input`)!,
  145. inputs: [{
  146. label: '',
  147. variable: iteratorInputKey,
  148. type: InputVarType.iterator,
  149. required: false,
  150. }],
  151. values: { [iteratorInputKey]: iterator },
  152. onChange: keyValue => setIterator(keyValue[iteratorInputKey]),
  153. },
  154. ]}
  155. runningStatus={runningStatus}
  156. onRun={handleRun}
  157. onStop={handleStop}
  158. {...logsParams}
  159. result={
  160. <ResultPanel {...runResult} showSteps={false} nodeInfo={nodeInfo} {...logsParams} />
  161. }
  162. />
  163. )}
  164. </div>
  165. )
  166. }
  167. export default React.memo(Panel)