fix: Knowledge base workflow variable assignment node

Author: shaohuzhang1

apps/application/flow/step_node/variable_assign_node/i_variable_assign_node.py

@@ -25,5 +25,5 @@ def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
     def _run(self):
         return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
 
-    def execute(self, variable_list, stream, **kwargs) -> NodeResult:
+    def execute(self, variable_list, **kwargs) -> NodeResult:
         pass

apps/application/flow/step_node/variable_assign_node/impl/base_variable_assign_node.py

@@ -58,7 +58,7 @@ def handle(self, variable, evaluation):
             result['output_value'] = reference
         return result
 
-    def execute(self, variable_list, stream, **kwargs) -> NodeResult:
+    def execute(self, variable_list, **kwargs) -> NodeResult:
         #
         result_list = []
         is_chat = False

ui/src/workflow/common/NodeCascader.vue

@@ -93,7 +93,7 @@ const get_up_node_field_list = (contain_self: boolean, use_cache: boolean) => {
   return result.filter((v: any) => v.children && v.children.length > 0)
 }
 const getOptionsValue = () => {
-  if (workflowMode == WorkflowMode.ApplicationLoop) {
+  if ([WorkflowMode.ApplicationLoop, WorkflowMode.KnowledgeLoop].includes(workflowMode)) {
     return props.global
       ? get_up_node_field_list(false, true).filter(
           (v: any) =>
@@ -110,13 +110,6 @@ const getOptionsValue = () => {
       : props.nodeModel
           .get_up_node_field_list(false, true)
           .filter((v: any) => v.children && v.children.length > 0)
-    if (workflowMode == WorkflowMode.Knowledge || workflowMode == WorkflowMode.KnowledgeLoop) {
-      const edgeModel = props.nodeModel.graphModel.getElement('knowledge-base-node')
-      const baseNode = edgeModel
-        .get_node_field_list()
-        .filter((v: any) => v.children && v.children.length > 0)
-      return [...baseNode, ...result]
-    }
     return result
   }
 }

ui/src/workflow/common/app-node.ts

@@ -81,7 +81,7 @@ class AppNode extends HtmlResize.view {
       let globalFields = []
       if (this.props.model.properties.user_input_field_list) {
         globalFields = this.props.model.properties.user_input_field_list.map((item: any) => ({
-          label: item.label.label,
+          label: typeof item.label == 'string' ? item.label : item.label.label,
           value: item.field,
         }))
       }
@@ -129,6 +129,12 @@ class AppNode extends HtmlResize.view {
         this.props.graphModel.getNodeModelById('start-node') ||
         this.props.graphModel.getNodeModelById('loop-start-node')
       )?.get_node_field_list() || []
+    const kbn = this.props.graphModel.getNodeModelById('knowledge-base-node')
+    if (kbn) {
+      const knowledgeBaseFieldList = kbn.get_node_field_list()
+      return [...knowledgeBaseFieldList, ...start_node_field_list, ...result]
+    }
+
     return [...start_node_field_list, ...result]
   }