feat: Variable aggregation supports outputting dictionary types.

Author: wangliang181230

apps/application/flow/step_node/variable_aggregation_node/i_variable_aggregation_node.py

@@ -11,6 +11,7 @@
 
 class VariableListSerializer(serializers.Serializer):
     v_id = serializers.CharField(required=True, label=_("Variable id"))
+    key = serializers.CharField(required=False, label=_("Key"))
     variable = serializers.ListField(required=True, label=_("Variable"))
 
 

apps/application/flow/step_node/variable_aggregation_node/impl/base_variable_aggregation_node.py

@@ -31,7 +31,6 @@ def save_context(self, details, workflow_manage):
         self.context['exception_message'] = details.get('err_message')
 
     def get_first_non_null(self, variable_list):
-
         for variable in variable_list:
             v = self.workflow_manage.get_reference_field(
                 variable.get('variable')[0],
@@ -40,12 +39,16 @@ def get_first_non_null(self, variable_list):
                 return v
         return None
 
-    def set_variable_to_json(self, variable_list):
-
+    def set_variable_to_array(self, variable_list):
         return [self.workflow_manage.get_reference_field(
             variable.get('variable')[0],
             variable.get('variable')[1:]) for variable in variable_list]
 
+    def set_variable_to_dict(self, variable_list):
+        return {(variable.get('key') or variable.get('variable')[-1]): self.workflow_manage.get_reference_field(
+            variable.get('variable')[0],
+            variable.get('variable')[1:]) for variable in variable_list}
+
     def reset_variable(self, variable):
         value = self.workflow_manage.get_reference_field(
             variable.get('variable')[0],
@@ -65,9 +68,14 @@ def reset_group_list(self, group_list):
 
     def execute(self, strategy, group_list, **kwargs) -> NodeResult:
         strategy_map = {'first_non_null': self.get_first_non_null,
-                        'variable_to_json': self.set_variable_to_json,
+                        'variable_to_array': self.set_variable_to_array,
+                        'variable_to_dict': self.set_variable_to_dict,
                         }
 
+        # 向下兼容
+        if strategy == 'variable_to_json':
+            strategy = 'variable_to_array'
+
         result = {item.get('field'): strategy_map[strategy](item.get('variable_list')) for item in group_list}
 
         return NodeResult(

ui/src/components/execution-detail-card/index.vue

@@ -936,9 +936,11 @@
               </h5>
               <div class="p-8-12 border-t-dashed lighter pre-wrap">
                 {{
-                  data.strategy === 'variable_to_json'
-                    ? t('workflow.nodes.variableAggregationNode.placeholder1')
-                    : t('workflow.nodes.variableAggregationNode.placeholder')
+                  data.strategy === 'first_non_null'
+                    ? t('workflow.nodes.variableAggregationNode.placeholder')
+                    : data.strategy === 'variable_to_dict'
+                      ? t('workflow.nodes.variableAggregationNode.placeholder2')
+                      : t('workflow.nodes.variableAggregationNode.placeholder1')
                 }}
               </div>
             </div>

ui/src/locales/lang/en-US/workflow.ts

@@ -323,7 +323,9 @@ You are a master of problem optimization, adept at accurately inferring user int
       text: 'Aggregate variables of each group according to the aggregation strategy',
       Strategy: 'Aggregation Strategy',
       placeholder: 'Return the first non-null value of each group',
-      placeholder1: 'Return the set of variables for each group',
+      placeholder1: 'Return the array of variables for each group',
+      placeholder2: 'Return the dict of variables for each group',
+      placeholder_key: 'Input key',
       group: {
         noneError: 'Name cannot be empty',
         dupError: 'Name cannot be duplicated',

ui/src/locales/lang/zh-CN/workflow.ts

@@ -304,7 +304,9 @@ export default {
       text: '按聚合策略聚合每组的变量',
       Strategy: '聚合策略',
       placeholder: '返回每组的第一个非空值',
-      placeholder1: '返回每组变量的集合',
+      placeholder1: '返回每组变量的数组（Array）',
+      placeholder2: '返回每组变量的字典（Dict）',
+      placeholder_key: '输入键名',
       group: {
         noneError: '名称不能为空',
         dupError: '名称不能重复',

ui/src/locales/lang/zh-Hant/workflow.ts

@@ -322,7 +322,9 @@ export default {
       text: '按聚合策略聚合每組的變量',
       Strategy: '聚合策略',
       placeholder: '返回每組的第一個非空值',
-      placeholder1: '返回每組變量的集合',
+      placeholder1: '返回每組變量的數組（Array）',
+      placeholder2: '返回每組變量的字典（Dict）',
+      placeholder_key: '輸入鍵名',
       group: {
         noneError: '名稱不能為空',
         dupError: '名稱不能重複',

ui/src/workflow/nodes/variable-aggregation-node/index.ts

@@ -10,8 +10,14 @@ class VariableAggregationNode extends AppNode {
   }
 }
 
+class VariableAggregationNodeModel extends AppNodeModel {
+  get_width() {
+    return 450
+  }
+}
+
 export default {
   type: 'variable-aggregation-node',
-  model: AppNodeModel,
+  model: VariableAggregationNodeModel,
   view: VariableAggregationNode,
 }

ui/src/workflow/nodes/variable-aggregation-node/index.vue

@@ -34,7 +34,11 @@
           />
           <el-option
             :label="t('workflow.nodes.variableAggregationNode.placeholder1')"
-            value="variable_to_json"
+            value="variable_to_array"
+          />
+          <el-option
+            :label="t('workflow.nodes.variableAggregationNode.placeholder2')"
+            value="variable_to_dict"
           />
         </el-select>
       </el-form-item>
@@ -77,10 +81,17 @@
                       trigger: 'change',
                     }"
                   >
+                    <el-input
+                      v-if="form_data.strategy === 'variable_to_dict'"
+                      v-model="item.key"
+                      :placeholder="$t('workflow.variable.placeholder_key')"
+                      style="width: 100px; margin-right: 8px"
+                      maxlength="256"
+                    />
                     <NodeCascader
                       ref="nodeCascaderRef"
                       :nodeModel="nodeModel"
-                      style="width: 200px"
+                      :style="{ width: form_data.strategy === 'variable_to_dict' ? '200px' : '310px'}"
                       :placeholder="$t('workflow.variable.placeholder')"
                       v-model="item.variable"
                     />
@@ -149,7 +160,10 @@ const form = {
 const form_data = computed({
   get: () => {
     if (props.nodeModel.properties.node_data) {
-      return props.nodeModel.properties.node_data
+      // 向下兼容
+      if (props.nodeModel.properties.node_data.strategy === 'variable_to_json') {
+        props.nodeModel.properties.node_data.strategy = 'variable_to_array'
+      }
     } else {
       set(props.nodeModel.properties, 'node_data', form)
     }