panel.tsx 7.2 KB

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