youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 02:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: question classifer support run

Browse Source
This commit is contained in:
Joel
2024-03-07 18:43:19 +08:00
parent 425e162a91
commit 783f7a9b13
2 changed files with 109 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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,
}
}