panel.tsx 4.6 KB

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