feat: question classifer support run

web/app/components/workflow/nodes/question-classifier/panel.tsx

@@ -8,7 +8,8 @@ import AdvancedSetting from './components/advanced-setting'
 import type { QuestionClassifierNodeType } from './types'
 import Field from '@/app/components/workflow/nodes/_base/components/field'
 import ModelParameterModal from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal'
-import type { NodePanelProps } from '@/app/components/workflow/types'
+import { InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
+import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
 
 const i18nPrefix = 'workflow.nodes.questionClassifiers'
 
@@ -27,56 +28,86 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
     handleTopicsChange,
     handleInstructionChange,
     handleMemoryChange,
+    isShowSingleRun,
+    hideSingleRun,
+    runningStatus,
+    handleRun,
+    handleStop,
+    query,
+    setQuery,
   } = useConfig(id, data)
 
   const model = inputs.model
 
   return (
-    <div className='mt-2 px-4 space-y-4'>
-      <Field
-        title={t(`${i18nPrefix}.inputVars`)}
-      >
-        <VarReferencePicker
-          readonly={readOnly}
-          isShowNodeName
-          value={inputs.query_variable_selector}
-          onChange={handleQueryVarChange}
+    <div>
+      <div className='mt-2 px-4 space-y-4'>
+        <Field
+          title={t(`${i18nPrefix}.inputVars`)}
+        >
+          <VarReferencePicker
+            readonly={readOnly}
+            isShowNodeName
+            value={inputs.query_variable_selector}
+            onChange={handleQueryVarChange}
+          />
+        </Field>
+        <Field
+          title={t(`${i18nPrefix}.model`)}
+        >
+          <ModelParameterModal
+            popupClassName='!w-[387px]'
+            isAdvancedMode={true}
+            mode={model?.mode}
+            provider={model?.provider}
+            completionParams={model.completion_params}
+            modelId={model.name}
+            setModel={handleModelChanged}
+            onCompletionParamsChange={handleCompletionParamsChange}
+            hideDebugWithMultipleModel
+            debugWithMultipleModel={false}
+          />
+        </Field>
+        <Field
+          title={t(`${i18nPrefix}.class`)}
+        >
+          <ClassList
+            list={inputs.classes}
+            onChange={handleTopicsChange} />
+        </Field>
+        <Field
+          title={t(`${i18nPrefix}.advancedSetting`)}
+          supportFold
+        >
+          <AdvancedSetting
+            instruction={inputs.instruction}
+            onInstructionChange={handleInstructionChange}
+            memory={inputs.memory}
+            onMemoryChange={handleMemoryChange}
+          />
+        </Field>
+      </div>
+      {isShowSingleRun && (
+        <BeforeRunForm
+          nodeName={inputs.title}
+          onHide={hideSingleRun}
+          forms={[
+            {
+              inputs: [{
+                label: t(`${i18nPrefix}.inputVars`)!,
+                variable: 'query',
+                type: InputVarType.paragraph,
+                required: true,
+              }],
+              values: { query },
+              onChange: keyValue => setQuery((keyValue as any).query),
+            },
+          ]}
+          runningStatus={runningStatus}
+          onRun={handleRun}
+          onStop={handleStop}
         />
-      </Field>
-      <Field
-        title={t(`${i18nPrefix}.model`)}
-      >
-        <ModelParameterModal
-          popupClassName='!w-[387px]'
-          isAdvancedMode={true}
-          mode={model?.mode}
-          provider={model?.provider}
-          completionParams={model.completion_params}
-          modelId={model.name}
-          setModel={handleModelChanged}
-          onCompletionParamsChange={handleCompletionParamsChange}
-          hideDebugWithMultipleModel
-          debugWithMultipleModel={false}
-        />
-      </Field>
-      <Field
-        title={t(`${i18nPrefix}.class`)}
-      >
-        <ClassList
-          list={inputs.classes}
-          onChange={handleTopicsChange} />
-      </Field>
-      <Field
-        title={t(`${i18nPrefix}.advancedSetting`)}
-        supportFold
-      >
-        <AdvancedSetting
-          instruction={inputs.instruction}
-          onInstructionChange={handleInstructionChange}
-          memory={inputs.memory}
-          onMemoryChange={handleMemoryChange}
-        />
-      </Field>
+      )}
     </div>
   )
 }

web/app/components/workflow/nodes/question-classifier/use-config.ts

@@ -3,6 +3,7 @@ import produce from 'immer'
 import type { Memory, ValueSelector } from '../../types'
 import type { QuestionClassifierNodeType } from './types'
 import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
+import useOneStepRun from '@/app/components/workflow/nodes/_base/hooks/use-one-step-run'
 
 const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
   const { inputs, setInputs } = useNodeCrud<QuestionClassifierNodeType>(id, payload)
@@ -52,6 +53,31 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
     setInputs(newInputs)
   }, [inputs, setInputs])
 
+  // single run
+  const {
+    isShowSingleRun,
+    hideSingleRun,
+    runningStatus,
+    handleRun,
+    handleStop,
+    runInputData,
+    setRunInputData,
+  } = useOneStepRun<QuestionClassifierNodeType>({
+    id,
+    data: inputs,
+    defaultRunInputData: {
+      query: 'Negative or positive sentiment?',
+    },
+  })
+
+  const query = runInputData.query
+  const setQuery = useCallback((newQuery: string) => {
+    setRunInputData({
+      ...runInputData,
+      query: newQuery,
+    })
+  }, [runInputData, setRunInputData])
+
   return {
     inputs,
     handleModelChanged,
@@ -60,6 +86,13 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
     handleTopicsChange: handleClassesChange,
     handleInstructionChange,
     handleMemoryChange,
+    isShowSingleRun,
+    hideSingleRun,
+    runningStatus,
+    handleRun,
+    handleStop,
+    query,
+    setQuery,
   }
 }