panel.tsx 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  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. handleCompletionParamsChange,
  28. handleQueryVarChange,
  29. handleTopicsChange,
  30. handleInstructionChange,
  31. handleMemoryChange,
  32. isShowSingleRun,
  33. hideSingleRun,
  34. runningStatus,
  35. handleRun,
  36. handleStop,
  37. query,
  38. setQuery,
  39. runResult,
  40. filterVar,
  41. } = useConfig(id, data)
  42. const model = inputs.model
  43. return (
  44. <div className='mt-2'>
  45. <div className='px-4 pb-4 space-y-4'>
  46. <Field
  47. title={t(`${i18nPrefix}.inputVars`)}
  48. >
  49. <VarReferencePicker
  50. readonly={readOnly}
  51. isShowNodeName
  52. nodeId={id}
  53. value={inputs.query_variable_selector}
  54. onChange={handleQueryVarChange}
  55. filterVar={filterVar}
  56. />
  57. </Field>
  58. <Field
  59. title={t(`${i18nPrefix}.model`)}
  60. >
  61. <ModelParameterModal
  62. popupClassName='!w-[387px]'
  63. isInWorkflow
  64. isAdvancedMode={true}
  65. mode={model?.mode}
  66. provider={model?.provider}
  67. completionParams={model.completion_params}
  68. modelId={model.name}
  69. setModel={handleModelChanged}
  70. onCompletionParamsChange={handleCompletionParamsChange}
  71. hideDebugWithMultipleModel
  72. debugWithMultipleModel={false}
  73. readonly={readOnly}
  74. />
  75. </Field>
  76. <Field
  77. title={t(`${i18nPrefix}.class`)}
  78. >
  79. <ClassList
  80. id={id}
  81. list={inputs.classes}
  82. onChange={handleTopicsChange}
  83. readonly={readOnly}
  84. />
  85. </Field>
  86. <Field
  87. title={t(`${i18nPrefix}.advancedSetting`)}
  88. supportFold
  89. >
  90. <AdvancedSetting
  91. hideMemorySetting={!isChatMode}
  92. instruction={inputs.instruction}
  93. onInstructionChange={handleInstructionChange}
  94. memory={inputs.memory}
  95. onMemoryChange={handleMemoryChange}
  96. readonly={readOnly}
  97. />
  98. </Field>
  99. </div>
  100. <Split />
  101. <div className='px-4 pt-4 pb-2'>
  102. <OutputVars>
  103. <>
  104. <VarItem
  105. name='class_name'
  106. type='string'
  107. description={t(`${i18nPrefix}.outputVars.className`)}
  108. />
  109. </>
  110. </OutputVars>
  111. </div>
  112. {isShowSingleRun && (
  113. <BeforeRunForm
  114. nodeName={inputs.title}
  115. onHide={hideSingleRun}
  116. forms={[
  117. {
  118. inputs: [{
  119. label: t(`${i18nPrefix}.inputVars`)!,
  120. variable: 'query',
  121. type: InputVarType.paragraph,
  122. required: true,
  123. }],
  124. values: { query },
  125. onChange: keyValue => setQuery((keyValue as any).query),
  126. },
  127. ]}
  128. runningStatus={runningStatus}
  129. onRun={handleRun}
  130. onStop={handleStop}
  131. result={<ResultPanel {...runResult} showSteps={false} />}
  132. />
  133. )}
  134. </div>
  135. )
  136. }
  137. export default React.memo(Panel)