Commit ae9e7acd authored by Joel's avatar Joel

feat: other node run

parent 149eb38e
...@@ -12,6 +12,7 @@ import Split from '@/app/components/workflow/nodes/_base/components/split' ...@@ -12,6 +12,7 @@ import Split from '@/app/components/workflow/nodes/_base/components/split'
import OutputVars, { VarItem } from '@/app/components/workflow/nodes/_base/components/output-vars' import OutputVars, { VarItem } from '@/app/components/workflow/nodes/_base/components/output-vars'
import { InputVarType, 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' import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
import ResultPanel from '@/app/components/workflow/run/result-panel'
const i18nPrefix = 'workflow.nodes.knowledgeRetrieval' const i18nPrefix = 'workflow.nodes.knowledgeRetrieval'
...@@ -36,6 +37,7 @@ const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({ ...@@ -36,6 +37,7 @@ const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({
handleStop, handleStop,
query, query,
setQuery, setQuery,
runResult,
} = useConfig(id, data) } = useConfig(id, data)
return ( return (
...@@ -139,6 +141,7 @@ const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({ ...@@ -139,6 +141,7 @@ const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({
runningStatus={runningStatus} runningStatus={runningStatus}
onRun={handleRun} onRun={handleRun}
onStop={handleStop} onStop={handleStop}
result={<ResultPanel {...runResult} showSteps={false} />}
/> />
)} )}
</div> </div>
......
...@@ -64,6 +64,7 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => { ...@@ -64,6 +64,7 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => {
handleStop, handleStop,
runInputData, runInputData,
setRunInputData, setRunInputData,
runResult,
} = useOneStepRun<KnowledgeRetrievalNodeType>({ } = useOneStepRun<KnowledgeRetrievalNodeType>({
id, id,
data: inputs, data: inputs,
...@@ -94,6 +95,7 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => { ...@@ -94,6 +95,7 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => {
handleStop, handleStop,
query, query,
setQuery, setQuery,
runResult,
} }
} }
......
...@@ -17,6 +17,7 @@ import { Resolution } from '@/types/app' ...@@ -17,6 +17,7 @@ import { Resolution } from '@/types/app'
import { InputVarType, 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' import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
import type { Props as FormProps } from '@/app/components/workflow/nodes/_base/components/before-run-form/form' import type { Props as FormProps } from '@/app/components/workflow/nodes/_base/components/before-run-form/form'
import ResultPanel from '@/app/components/workflow/run/result-panel'
const i18nPrefix = 'workflow.nodes.llm' const i18nPrefix = 'workflow.nodes.llm'
...@@ -52,6 +53,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({ ...@@ -52,6 +53,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
handleRun, handleRun,
handleStop, handleStop,
varInputs, varInputs,
runResult,
} = useConfig(id, data) } = useConfig(id, data)
const isChatApp = true // TODO: get from app context const isChatApp = true // TODO: get from app context
...@@ -221,6 +223,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({ ...@@ -221,6 +223,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
runningStatus={runningStatus} runningStatus={runningStatus}
onRun={handleRun} onRun={handleRun}
onStop={handleStop} onStop={handleStop}
result={<ResultPanel {...runResult} showSteps={false} />}
/> />
)} )}
</div> </div>
......
...@@ -99,6 +99,7 @@ const useConfig = (id: string, payload: LLMNodeType) => { ...@@ -99,6 +99,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
handleStop, handleStop,
runInputData, runInputData,
setRunInputData, setRunInputData,
runResult,
} = useOneStepRun<LLMNodeType>({ } = useOneStepRun<LLMNodeType>({
id, id,
data: inputs, data: inputs,
...@@ -170,6 +171,7 @@ const useConfig = (id: string, payload: LLMNodeType) => { ...@@ -170,6 +171,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
runningStatus, runningStatus,
handleRun, handleRun,
handleStop, handleStop,
runResult,
} }
} }
......
...@@ -10,6 +10,7 @@ import Field from '@/app/components/workflow/nodes/_base/components/field' ...@@ -10,6 +10,7 @@ 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 ModelParameterModal from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal'
import { InputVarType, 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' import BeforeRunForm from '@/app/components/workflow/nodes/_base/components/before-run-form'
import ResultPanel from '@/app/components/workflow/run/result-panel'
const i18nPrefix = 'workflow.nodes.questionClassifiers' const i18nPrefix = 'workflow.nodes.questionClassifiers'
...@@ -35,6 +36,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({ ...@@ -35,6 +36,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
handleStop, handleStop,
query, query,
setQuery, setQuery,
runResult,
} = useConfig(id, data) } = useConfig(id, data)
const model = inputs.model const model = inputs.model
...@@ -108,6 +110,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({ ...@@ -108,6 +110,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
runningStatus={runningStatus} runningStatus={runningStatus}
onRun={handleRun} onRun={handleRun}
onStop={handleStop} onStop={handleStop}
result={<ResultPanel {...runResult} showSteps={false} />}
/> />
)} )}
</div> </div>
......
...@@ -63,6 +63,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => { ...@@ -63,6 +63,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
handleStop, handleStop,
runInputData, runInputData,
setRunInputData, setRunInputData,
runResult,
} = useOneStepRun<QuestionClassifierNodeType>({ } = useOneStepRun<QuestionClassifierNodeType>({
id, id,
data: inputs, data: inputs,
...@@ -94,6 +95,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => { ...@@ -94,6 +95,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
handleStop, handleStop,
query, query,
setQuery, setQuery,
runResult,
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment