panel.tsx 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  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 ConfigVision from '../_base/components/config-vision'
  6. import useConfig from './use-config'
  7. import ClassList from './components/class-list'
  8. import AdvancedSetting from './components/advanced-setting'
  9. import type { QuestionClassifierNodeType } from './types'
  10. import Field from '@/app/components/workflow/nodes/_base/components/field'
  11. import ModelParameterModal from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal'
  12. import { InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
  13. import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
  14. import ResultPanel from '@/app/components/workflow/run/result-panel'
  15. import Split from '@/app/components/workflow/nodes/_base/components/split'
  16. import OutputVars, { VarItem } from '@/app/components/workflow/nodes/_base/components/output-vars'
  17. import { FieldCollapse } from '@/app/components/workflow/nodes/_base/components/collapse'
  18. const i18nPrefix = 'workflow.nodes.questionClassifiers'
  19. const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
  20. id,
  21. data,
  22. }) => {
  23. const { t } = useTranslation()
  24. const {
  25. readOnly,
  26. inputs,
  27. handleModelChanged,
  28. isChatMode,
  29. isChatModel,
  30. handleCompletionParamsChange,
  31. handleQueryVarChange,
  32. handleTopicsChange,
  33. hasSetBlockStatus,
  34. availableVars,
  35. availableNodesWithParent,
  36. handleInstructionChange,
  37. inputVarValues,
  38. varInputs,
  39. setInputVarValues,
  40. handleMemoryChange,
  41. isVisionModel,
  42. handleVisionResolutionChange,
  43. handleVisionResolutionEnabledChange,
  44. isShowSingleRun,
  45. hideSingleRun,
  46. runningStatus,
  47. handleRun,
  48. handleStop,
  49. runResult,
  50. filterVar,
  51. } = useConfig(id, data)
  52. const model = inputs.model
  53. return (
  54. <div className='pt-2'>
  55. <div className='px-4 space-y-4'>
  56. <Field
  57. title={t(`${i18nPrefix}.model`)}
  58. >
  59. <ModelParameterModal
  60. popupClassName='!w-[387px]'
  61. isInWorkflow
  62. isAdvancedMode={true}
  63. mode={model?.mode}
  64. provider={model?.provider}
  65. completionParams={model.completion_params}
  66. modelId={model.name}
  67. setModel={handleModelChanged}
  68. onCompletionParamsChange={handleCompletionParamsChange}
  69. hideDebugWithMultipleModel
  70. debugWithMultipleModel={false}
  71. readonly={readOnly}
  72. />
  73. </Field>
  74. <Field
  75. title={t(`${i18nPrefix}.inputVars`)}
  76. >
  77. <VarReferencePicker
  78. readonly={readOnly}
  79. isShowNodeName
  80. nodeId={id}
  81. value={inputs.query_variable_selector}
  82. onChange={handleQueryVarChange}
  83. filterVar={filterVar}
  84. />
  85. </Field>
  86. <Split />
  87. <ConfigVision
  88. nodeId={id}
  89. readOnly={readOnly}
  90. isVisionModel={isVisionModel}
  91. enabled={inputs.vision?.enabled}
  92. onEnabledChange={handleVisionResolutionEnabledChange}
  93. config={inputs.vision?.configs}
  94. onConfigChange={handleVisionResolutionChange}
  95. />
  96. <Field
  97. title={t(`${i18nPrefix}.class`)}
  98. >
  99. <ClassList
  100. id={id}
  101. list={inputs.classes}
  102. onChange={handleTopicsChange}
  103. readonly={readOnly}
  104. />
  105. </Field>
  106. <Split />
  107. </div>
  108. <FieldCollapse
  109. title={t(`${i18nPrefix}.advancedSetting`)}
  110. >
  111. <AdvancedSetting
  112. hideMemorySetting={!isChatMode}
  113. instruction={inputs.instruction}
  114. onInstructionChange={handleInstructionChange}
  115. memory={inputs.memory}
  116. onMemoryChange={handleMemoryChange}
  117. readonly={readOnly}
  118. isChatApp={isChatMode}
  119. isChatModel={isChatModel}
  120. hasSetBlockStatus={hasSetBlockStatus}
  121. nodesOutputVars={availableVars}
  122. availableNodes={availableNodesWithParent}
  123. />
  124. </FieldCollapse>
  125. <Split />
  126. <div>
  127. <OutputVars>
  128. <>
  129. <VarItem
  130. name='class_name'
  131. type='string'
  132. description={t(`${i18nPrefix}.outputVars.className`)}
  133. />
  134. </>
  135. </OutputVars>
  136. </div>
  137. {isShowSingleRun && (
  138. <BeforeRunForm
  139. nodeName={inputs.title}
  140. onHide={hideSingleRun}
  141. forms={[
  142. {
  143. inputs: [{
  144. label: t(`${i18nPrefix}.inputVars`)!,
  145. variable: 'query',
  146. type: InputVarType.paragraph,
  147. required: true,
  148. }, ...varInputs],
  149. values: inputVarValues,
  150. onChange: setInputVarValues,
  151. },
  152. ]}
  153. runningStatus={runningStatus}
  154. onRun={handleRun}
  155. onStop={handleStop}
  156. result={<ResultPanel {...runResult} showSteps={false} />}
  157. />
  158. )}
  159. </div>
  160. )
  161. }
  162. export default React.memo(Panel)