panel.tsx 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238
  1. import type { FC } from 'react'
  2. import React from 'react'
  3. import { useTranslation } from 'react-i18next'
  4. import MemoryConfig from '../_base/components/memory-config'
  5. import VarReferencePicker from '../_base/components/variable/var-reference-picker'
  6. import Editor from '../_base/components/prompt/editor'
  7. import ResultPanel from '../../run/result-panel'
  8. import ConfigVision from '../_base/components/config-vision'
  9. import useConfig from './use-config'
  10. import type { ParameterExtractorNodeType } from './types'
  11. import ExtractParameter from './components/extract-parameter/list'
  12. import ImportFromTool from './components/extract-parameter/import-from-tool'
  13. import AddExtractParameter from './components/extract-parameter/update'
  14. import ReasoningModePicker from './components/reasoning-mode-picker'
  15. import Field from '@/app/components/workflow/nodes/_base/components/field'
  16. import Split from '@/app/components/workflow/nodes/_base/components/split'
  17. import ModelParameterModal from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal'
  18. import OutputVars, { VarItem } from '@/app/components/workflow/nodes/_base/components/output-vars'
  19. import { InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
  20. import Tooltip from '@/app/components/base/tooltip'
  21. import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
  22. import { VarType } from '@/app/components/workflow/types'
  23. import { FieldCollapse } from '@/app/components/workflow/nodes/_base/components/collapse'
  24. const i18nPrefix = 'workflow.nodes.parameterExtractor'
  25. const i18nCommonPrefix = 'workflow.common'
  26. const Panel: FC<NodePanelProps<ParameterExtractorNodeType>> = ({
  27. id,
  28. data,
  29. }) => {
  30. const { t } = useTranslation()
  31. const {
  32. readOnly,
  33. inputs,
  34. handleInputVarChange,
  35. filterVar,
  36. isChatModel,
  37. isChatMode,
  38. isCompletionModel,
  39. handleModelChanged,
  40. handleImportFromTool,
  41. handleCompletionParamsChange,
  42. addExtractParameter,
  43. handleExactParamsChange,
  44. handleInstructionChange,
  45. hasSetBlockStatus,
  46. handleMemoryChange,
  47. isSupportFunctionCall,
  48. handleReasoningModeChange,
  49. availableVars,
  50. availableNodesWithParent,
  51. inputVarValues,
  52. varInputs,
  53. isVisionModel,
  54. handleVisionResolutionChange,
  55. handleVisionResolutionEnabledChange,
  56. isShowSingleRun,
  57. hideSingleRun,
  58. runningStatus,
  59. handleRun,
  60. handleStop,
  61. runResult,
  62. setInputVarValues,
  63. } = useConfig(id, data)
  64. const model = inputs.model
  65. return (
  66. <div className='pt-2'>
  67. <div className='px-4 space-y-4'>
  68. <Field
  69. title={t(`${i18nCommonPrefix}.model`)}
  70. >
  71. <ModelParameterModal
  72. popupClassName='!w-[387px]'
  73. isInWorkflow
  74. isAdvancedMode={true}
  75. mode={model?.mode}
  76. provider={model?.provider}
  77. completionParams={model?.completion_params}
  78. modelId={model?.name}
  79. setModel={handleModelChanged}
  80. onCompletionParamsChange={handleCompletionParamsChange}
  81. hideDebugWithMultipleModel
  82. debugWithMultipleModel={false}
  83. readonly={readOnly}
  84. />
  85. </Field>
  86. <Field
  87. title={t(`${i18nPrefix}.inputVar`)}
  88. >
  89. <>
  90. <VarReferencePicker
  91. readonly={readOnly}
  92. nodeId={id}
  93. isShowNodeName
  94. value={inputs.query || []}
  95. onChange={handleInputVarChange}
  96. filterVar={filterVar}
  97. />
  98. </>
  99. </Field>
  100. <Split />
  101. <ConfigVision
  102. nodeId={id}
  103. readOnly={readOnly}
  104. isVisionModel={isVisionModel}
  105. enabled={inputs.vision?.enabled}
  106. onEnabledChange={handleVisionResolutionEnabledChange}
  107. config={inputs.vision?.configs}
  108. onConfigChange={handleVisionResolutionChange}
  109. />
  110. <Field
  111. title={t(`${i18nPrefix}.extractParameters`)}
  112. operations={
  113. !readOnly
  114. ? (
  115. <div className='flex items-center space-x-1'>
  116. {!readOnly && (
  117. <ImportFromTool onImport={handleImportFromTool} />
  118. )}
  119. {!readOnly && (<div className='w-px h-3 bg-gray-200'></div>)}
  120. <AddExtractParameter type='add' onSave={addExtractParameter} />
  121. </div>
  122. )
  123. : undefined
  124. }
  125. >
  126. <ExtractParameter
  127. readonly={readOnly}
  128. list={inputs.parameters || []}
  129. onChange={handleExactParamsChange}
  130. />
  131. </Field>
  132. <Editor
  133. title={
  134. <div className='flex items-center space-x-1'>
  135. <span className='uppercase'>{t(`${i18nPrefix}.instruction`)}</span>
  136. <Tooltip
  137. popupContent={
  138. <div className='w-[120px]'>
  139. {t(`${i18nPrefix}.instructionTip`)}
  140. </div>
  141. }
  142. triggerClassName='w-3.5 h-3.5 ml-0.5'
  143. />
  144. </div>
  145. }
  146. value={inputs.instruction}
  147. onChange={handleInstructionChange}
  148. readOnly={readOnly}
  149. isChatModel={isChatModel}
  150. isChatApp={isChatMode}
  151. isShowContext={false}
  152. hasSetBlockStatus={hasSetBlockStatus}
  153. nodesOutputVars={availableVars}
  154. availableNodes={availableNodesWithParent}
  155. />
  156. </div>
  157. <FieldCollapse title={t(`${i18nPrefix}.advancedSetting`)}>
  158. <>
  159. {/* Memory */}
  160. {isChatMode && (
  161. <div className='mt-4'>
  162. <MemoryConfig
  163. readonly={readOnly}
  164. config={{ data: inputs.memory }}
  165. onChange={handleMemoryChange}
  166. canSetRoleName={isCompletionModel}
  167. />
  168. </div>
  169. )}
  170. {isSupportFunctionCall && (
  171. <div className='mt-2'>
  172. <ReasoningModePicker
  173. type={inputs.reasoning_mode}
  174. onChange={handleReasoningModeChange}
  175. />
  176. </div>
  177. )}
  178. </>
  179. </FieldCollapse>
  180. {inputs.parameters?.length > 0 && (<>
  181. <Split />
  182. <div>
  183. <OutputVars>
  184. <>
  185. {inputs.parameters.map((param, index) => (
  186. <VarItem
  187. key={index}
  188. name={param.name}
  189. type={param.type}
  190. description={param.description}
  191. />
  192. ))}
  193. <VarItem
  194. name='__is_success'
  195. type={VarType.number}
  196. description={t(`${i18nPrefix}.isSuccess`)}
  197. />
  198. <VarItem
  199. name='__reason'
  200. type={VarType.string}
  201. description={t(`${i18nPrefix}.errorReason`)}
  202. />
  203. </>
  204. </OutputVars>
  205. </div>
  206. </>)}
  207. {isShowSingleRun && (
  208. <BeforeRunForm
  209. nodeName={inputs.title}
  210. onHide={hideSingleRun}
  211. forms={[
  212. {
  213. inputs: [{
  214. label: t(`${i18nPrefix}.inputVar`)!,
  215. variable: 'query',
  216. type: InputVarType.paragraph,
  217. required: true,
  218. }, ...varInputs],
  219. values: inputVarValues,
  220. onChange: setInputVarValues,
  221. },
  222. ]}
  223. runningStatus={runningStatus}
  224. onRun={handleRun}
  225. onStop={handleStop}
  226. result={<ResultPanel {...runResult} showSteps={false} />}
  227. />
  228. )}
  229. </div>
  230. )
  231. }
  232. export default React.memo(Panel)