From ae9e7acd77433285dee766dd775ddacb42e5645a Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Wed, 13 Mar 2024 18:49:22 +0800
Subject: [PATCH] feat: other node run

---
 .../components/workflow/nodes/knowledge-retrieval/panel.tsx    | 3 +++
 .../workflow/nodes/knowledge-retrieval/use-config.ts           | 2 ++
 web/app/components/workflow/nodes/llm/panel.tsx                | 3 +++
 web/app/components/workflow/nodes/llm/use-config.ts            | 2 ++
 .../components/workflow/nodes/question-classifier/panel.tsx    | 3 +++
 .../workflow/nodes/question-classifier/use-config.ts           | 2 ++
 6 files changed, 15 insertions(+)

diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/panel.tsx b/web/app/components/workflow/nodes/knowledge-retrieval/panel.tsx
index a7d146bae4..46bd2a0980 100644
--- a/web/app/components/workflow/nodes/knowledge-retrieval/panel.tsx
+++ b/web/app/components/workflow/nodes/knowledge-retrieval/panel.tsx
@@ -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 { InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
 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'
 
@@ -36,6 +37,7 @@ const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({
     handleStop,
     query,
     setQuery,
+    runResult,
   } = useConfig(id, data)
 
   return (
@@ -139,6 +141,7 @@ const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({
             runningStatus={runningStatus}
             onRun={handleRun}
             onStop={handleStop}
+            result={<ResultPanel {...runResult} showSteps={false} />}
           />
         )}
       </div>
diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/use-config.ts b/web/app/components/workflow/nodes/knowledge-retrieval/use-config.ts
index 85abbebdea..7d6e622a4b 100644
--- a/web/app/components/workflow/nodes/knowledge-retrieval/use-config.ts
+++ b/web/app/components/workflow/nodes/knowledge-retrieval/use-config.ts
@@ -64,6 +64,7 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => {
     handleStop,
     runInputData,
     setRunInputData,
+    runResult,
   } = useOneStepRun<KnowledgeRetrievalNodeType>({
     id,
     data: inputs,
@@ -94,6 +95,7 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => {
     handleStop,
     query,
     setQuery,
+    runResult,
   }
 }
 
diff --git a/web/app/components/workflow/nodes/llm/panel.tsx b/web/app/components/workflow/nodes/llm/panel.tsx
index c57f84e420..2f67a29590 100644
--- a/web/app/components/workflow/nodes/llm/panel.tsx
+++ b/web/app/components/workflow/nodes/llm/panel.tsx
@@ -17,6 +17,7 @@ import { Resolution } from '@/types/app'
 import { InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
 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 ResultPanel from '@/app/components/workflow/run/result-panel'
 
 const i18nPrefix = 'workflow.nodes.llm'
 
@@ -52,6 +53,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
     handleRun,
     handleStop,
     varInputs,
+    runResult,
   } = useConfig(id, data)
 
   const isChatApp = true // TODO: get from app context
@@ -221,6 +223,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
           runningStatus={runningStatus}
           onRun={handleRun}
           onStop={handleStop}
+          result={<ResultPanel {...runResult} showSteps={false} />}
         />
       )}
     </div>
diff --git a/web/app/components/workflow/nodes/llm/use-config.ts b/web/app/components/workflow/nodes/llm/use-config.ts
index 9231cbf70b..96ff2cc981 100644
--- a/web/app/components/workflow/nodes/llm/use-config.ts
+++ b/web/app/components/workflow/nodes/llm/use-config.ts
@@ -99,6 +99,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
     handleStop,
     runInputData,
     setRunInputData,
+    runResult,
   } = useOneStepRun<LLMNodeType>({
     id,
     data: inputs,
@@ -170,6 +171,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
     runningStatus,
     handleRun,
     handleStop,
+    runResult,
   }
 }
 
diff --git a/web/app/components/workflow/nodes/question-classifier/panel.tsx b/web/app/components/workflow/nodes/question-classifier/panel.tsx
index 9ca0938117..4ded95a61c 100644
--- a/web/app/components/workflow/nodes/question-classifier/panel.tsx
+++ b/web/app/components/workflow/nodes/question-classifier/panel.tsx
@@ -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 { InputVarType, type NodePanelProps } from '@/app/components/workflow/types'
 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'
 
@@ -35,6 +36,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
     handleStop,
     query,
     setQuery,
+    runResult,
   } = useConfig(id, data)
 
   const model = inputs.model
@@ -108,6 +110,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
           runningStatus={runningStatus}
           onRun={handleRun}
           onStop={handleStop}
+          result={<ResultPanel {...runResult} showSteps={false} />}
         />
       )}
     </div>
diff --git a/web/app/components/workflow/nodes/question-classifier/use-config.ts b/web/app/components/workflow/nodes/question-classifier/use-config.ts
index bb5f22d595..f25a532625 100644
--- a/web/app/components/workflow/nodes/question-classifier/use-config.ts
+++ b/web/app/components/workflow/nodes/question-classifier/use-config.ts
@@ -63,6 +63,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
     handleStop,
     runInputData,
     setRunInputData,
+    runResult,
   } = useOneStepRun<QuestionClassifierNodeType>({
     id,
     data: inputs,
@@ -94,6 +95,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
     handleStop,
     query,
     setQuery,
+    runResult,
   }
 }