panel.tsx 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  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. const i18nPrefix = 'workflow.nodes.questionClassifiers'
  15. const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
  16. id,
  17. data,
  18. }) => {
  19. const { t } = useTranslation()
  20. const {
  21. readOnly,
  22. inputs,
  23. handleModelChanged,
  24. isChatMode,
  25. handleCompletionParamsChange,
  26. handleQueryVarChange,
  27. handleTopicsChange,
  28. handleInstructionChange,
  29. handleMemoryChange,
  30. isShowSingleRun,
  31. hideSingleRun,
  32. runningStatus,
  33. handleRun,
  34. handleStop,
  35. query,
  36. setQuery,
  37. runResult,
  38. filterVar,
  39. } = useConfig(id, data)
  40. const model = inputs.model
  41. return (
  42. <div>
  43. <div className='mt-2 px-4 space-y-4'>
  44. <Field
  45. title={t(`${i18nPrefix}.inputVars`)}
  46. >
  47. <VarReferencePicker
  48. readonly={readOnly}
  49. isShowNodeName
  50. nodeId={id}
  51. value={inputs.query_variable_selector}
  52. onChange={handleQueryVarChange}
  53. filterVar={filterVar}
  54. />
  55. </Field>
  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}.class`)}
  76. >
  77. <ClassList
  78. id={id}
  79. list={inputs.classes}
  80. onChange={handleTopicsChange}
  81. readonly={readOnly}
  82. />
  83. </Field>
  84. <Field
  85. title={t(`${i18nPrefix}.advancedSetting`)}
  86. supportFold
  87. >
  88. <AdvancedSetting
  89. hideMemorySetting={!isChatMode}
  90. instruction={inputs.instruction}
  91. onInstructionChange={handleInstructionChange}
  92. memory={inputs.memory}
  93. onMemoryChange={handleMemoryChange}
  94. readonly={readOnly}
  95. />
  96. </Field>
  97. </div>
  98. {isShowSingleRun && (
  99. <BeforeRunForm
  100. nodeName={inputs.title}
  101. onHide={hideSingleRun}
  102. forms={[
  103. {
  104. inputs: [{
  105. label: t(`${i18nPrefix}.inputVars`)!,
  106. variable: 'query',
  107. type: InputVarType.paragraph,
  108. required: true,
  109. }],
  110. values: { query },
  111. onChange: keyValue => setQuery((keyValue as any).query),
  112. },
  113. ]}
  114. runningStatus={runningStatus}
  115. onRun={handleRun}
  116. onStop={handleStop}
  117. result={<ResultPanel {...runResult} showSteps={false} />}
  118. />
  119. )}
  120. </div>
  121. )
  122. }
  123. export default React.memo(Panel)