Skip to content

Commit e491c4b

Browse files
desAweisdesAweis
committed
[app-builder] add new feature variable-updater-node
1 parent 859a253 commit e491c4b

8 files changed

Lines changed: 251 additions & 0 deletions

File tree

‎app-builder/plugins/aipp-plugin/src/main/resources/component/basic_node_en.json‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,11 @@
5454
"name": "Variable Aggregation",
5555
"uniqueName": ""
5656
},
57+
{
58+
"type": "variableUpdateNodeState",
59+
"name": "Variable Update",
60+
"uniqueName": ""
61+
},
5762
{
5863
"type": "fileExtractionNodeState",
5964
"name": "File Extraction",

‎app-builder/plugins/aipp-plugin/src/main/resources/component/basic_node_zh.json‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,11 @@
5454
"name": "变量聚合",
5555
"uniqueName": ""
5656
},
57+
{
58+
"type": "variableUpdateNodeState",
59+
"name": "变量更新",
60+
"uniqueName": ""
61+
},
5762
{
5863
"type": "fileExtractionNodeState",
5964
"name": "文件提取",

‎app-builder/plugins/aipp-variable-updater/pom.xml‎

Lines changed: 82 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,82 @@
1+
<?xml version="1.0" encoding="UTF-8"?>
2+
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
3+
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
4+
<modelVersion>4.0.0</modelVersion>
5+
6+
<parent>
7+
<groupId>modelengine.fit.jade</groupId>
8+
<artifactId>app-builder-plugin-parent</artifactId>
9+
<version>1.0.0-SNAPSHOT</version>
10+
</parent>
11+
12+
<groupId>modelengine.fit.jade.plugin</groupId>
13+
<artifactId>aipp-variable-updater</artifactId>
14+
15+
<dependencies>
16+
<dependency>
17+
<groupId>modelengine.fit.jade.waterflow</groupId>
18+
<artifactId>waterflow-runtime-service</artifactId>
19+
</dependency>
20+
<dependency>
21+
<groupId>modelengine.fit.jade</groupId>
22+
<artifactId>aipp-genericable</artifactId>
23+
</dependency>
24+
<dependency>
25+
<groupId>modelengine.fit.jober</groupId>
26+
<artifactId>jober-genericable</artifactId>
27+
</dependency>
28+
29+
<!-- Test -->
30+
<dependency>
31+
<groupId>org.junit.jupiter</groupId>
32+
<artifactId>junit-jupiter</artifactId>
33+
</dependency>
34+
</dependencies>
35+
36+
<build>
37+
<plugins>
38+
<plugin>
39+
<groupId>org.fitframework</groupId>
40+
<artifactId>fit-build-maven-plugin</artifactId>
41+
<version>${fit.version}</version>
42+
<configuration>
43+
<category>user</category>
44+
<level>5</level>
45+
</configuration>
46+
<executions>
47+
<execution>
48+
<id>build-plugin</id>
49+
<goals>
50+
<goal>build-plugin</goal>
51+
</goals>
52+
</execution>
53+
<execution>
54+
<id>package-plugin</id>
55+
<goals>
56+
<goal>package-plugin</goal>
57+
</goals>
58+
</execution>
59+
</executions>
60+
</plugin>
61+
<plugin>
62+
<groupId>org.apache.maven.plugins</groupId>
63+
<artifactId>maven-antrun-plugin</artifactId>
64+
<version>${maven.antrun.version}</version>
65+
<executions>
66+
<execution>
67+
<phase>install</phase>
68+
<configuration>
69+
<target>
70+
<copy file="${project.build.directory}/${project.build.finalName}.jar"
71+
todir="../../../build/plugins"/>
72+
</target>
73+
</configuration>
74+
<goals>
75+
<goal>run</goal>
76+
</goals>
77+
</execution>
78+
</executions>
79+
</plugin>
80+
</plugins>
81+
</build>
82+
</project>

‎app-builder/plugins/aipp-variable-updater/src/main/java/modelengine/fit/jade/aipp/variable/updater/AippVariableUpdater.java‎

Lines changed: 75 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,75 @@
1+
/*---------------------------------------------------------------------------------------------
2+
* Copyright (c) 2025 Huawei Technologies Co., Ltd. All rights reserved.
3+
* This file is a part of the ModelEngine Project.
4+
* Licensed under the MIT License. See License.txt in the project root for license information.
5+
*--------------------------------------------------------------------------------------------*/
6+
7+
package modelengine.fit.jade.aipp.variable.updater;
8+
9+
import modelengine.fit.jober.aipp.constants.AippConst;
10+
import modelengine.fit.jober.common.ErrorCodes;
11+
import modelengine.fit.jober.common.exceptions.JobberException;
12+
import modelengine.fit.waterflow.spi.FlowableService;
13+
import modelengine.fitframework.annotation.Component;
14+
import modelengine.fitframework.annotation.Fitable;
15+
import modelengine.fitframework.util.ObjectUtils;
16+
17+
import java.util.List;
18+
import java.util.Map;
19+
20+
/**
21+
* 变量更新算子服务。
22+
*
23+
* @author 鲁为
24+
* @since 2025-09-26
25+
*/
26+
@Component
27+
public class AippVariableUpdater implements FlowableService {
28+
public static final String UPDATE_VARIABLES = "updateVariables";
29+
public static final String INTERNAL = "_internal";
30+
public static final String OUTPUT_SCOPE = "outputScope";
31+
public static final String KEY = "key";
32+
public static final String VALUE = "value";
33+
34+
@Override
35+
@Fitable("modelengine.fit.jade.aipp.variable.updater")
36+
public List<Map<String, Object>> handleTask(List<Map<String, Object>> flowData) {
37+
Map<String, Object> businessData = this.getBusinessData(flowData);
38+
List<Map<String, Object>> updateVariables = ObjectUtils.cast(businessData.get(UPDATE_VARIABLES));
39+
Map<String, Object> internal = ObjectUtils.cast(businessData.get(INTERNAL));
40+
Map<String, Object> outputScope = ObjectUtils.cast(internal.get(OUTPUT_SCOPE));
41+
for (Map<String, Object> variable : updateVariables) {
42+
List<String> path = ObjectUtils.cast(variable.get(KEY));
43+
Object newValue = variable.get(VALUE);
44+
this.updateNestedMapByPath(outputScope, path, newValue);
45+
}
46+
return flowData;
47+
}
48+
49+
private Map<String, Object> getBusinessData(List<Map<String, Object>> flowData) {
50+
if (flowData.isEmpty() || !flowData.get(0).containsKey(AippConst.BS_DATA_KEY)) {
51+
throw new JobberException(ErrorCodes.INPUT_PARAM_IS_EMPTY, AippConst.BS_DATA_KEY);
52+
}
53+
return ObjectUtils.cast(flowData.get(0).get(AippConst.BS_DATA_KEY));
54+
}
55+
56+
private void updateNestedMapByPath(Map<String, Object> businessData, List<String> path, Object newValue) {
57+
if (businessData == null || path == null || path.isEmpty()) {
58+
return;
59+
}
60+
61+
Map<String, Object> currentMap = businessData;
62+
63+
for (int i = 0; i < path.size() - 1; i++) {
64+
String key = path.get(i);
65+
Object value = currentMap.get(key);
66+
67+
if (value instanceof Map) {
68+
currentMap = ObjectUtils.cast(value);
69+
}
70+
}
71+
72+
String finalKey = path.get(path.size() - 1);
73+
currentMap.put(finalKey, newValue);
74+
}
75+
}

‎app-builder/plugins/aipp-variable-updater/src/main/resources/application.yml‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
fit:
2+
beans:
3+
packages:
4+
- 'modelengine.fit.jade.aipp.variable.updater'

‎app-builder/plugins/aipp-variable-updater/src/test/java/modelengine/fit/jade/aipp/variable/updater/AippVariableUpdaterTest.java‎

Lines changed: 78 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,78 @@
1+
package modelengine.fit.jade.aipp.variable.updater;
2+
3+
import modelengine.fitframework.util.MapBuilder;
4+
import modelengine.fitframework.util.ObjectUtils;
5+
6+
import org.junit.jupiter.api.Assertions;
7+
import org.junit.jupiter.api.DisplayName;
8+
import org.junit.jupiter.api.Test;
9+
10+
import java.util.HashMap;
11+
import java.util.List;
12+
import java.util.Map;
13+
14+
/**
15+
* {@link AippVariableUpdater}的测试方法。
16+
*
17+
* @author 鲁为
18+
* @since 2025-09-28
19+
*/
20+
@DisplayName("变量更新节点测试")
21+
public class AippVariableUpdaterTest {
22+
public static final String UPDATE_VARIABLES = "updateVariables";
23+
public static final String INTERNAL = "_internal";
24+
public static final String OUTPUT_SCOPE = "outputScope";
25+
public static final String KEY = "key";
26+
public static final String VALUE = "value";
27+
28+
private final AippVariableUpdater aippVariableUpdater = new AippVariableUpdater();
29+
30+
@Test
31+
@DisplayName("修改 businessData 中指定路径的值")
32+
void shouldModifySpecificPathValue() {
33+
Map<String, Object> flowDataInner =
34+
this.getFlowData(this.getContextData(List.of("trace1"), "contextId"), this.buildBusinessData());
35+
List<Map<String, Object>> flowData = List.of(flowDataInner);
36+
this.aippVariableUpdater.handleTask(flowData);
37+
Assertions.assertEquals(ObjectUtils.<String>cast(ObjectUtils.<Map<String, Object>>cast(ObjectUtils.<Map<String, Object>>cast(
38+
ObjectUtils.<Map<String, Object>>cast(ObjectUtils.<Map<String, Object>>cast(ObjectUtils.<Map<String, Object>>cast(
39+
flowData.get(0).get("businessData")).get(INTERNAL)).get(OUTPUT_SCOPE)).get(
40+
"jade8xvdq4"))
41+
.get("output")).get("a")), "after");
42+
}
43+
44+
private Map<String, Object> getFlowData(Map<String, Object> contextData, Map<String, Object> businessData) {
45+
return MapBuilder.get(() -> new HashMap<String, Object>())
46+
.put("contextData", contextData)
47+
.put("businessData", businessData)
48+
.build();
49+
}
50+
51+
private Map<String, Object> getContextData(List<String> traceIds, String contextId) {
52+
return MapBuilder.get(() -> new HashMap<String, Object>())
53+
.put("flowTraceIds", traceIds)
54+
.put("contextId", contextId)
55+
.build();
56+
}
57+
58+
private Map<String, Object> buildBusinessData() {
59+
return MapBuilder.<String, Object>get()
60+
.put(UPDATE_VARIABLES,
61+
List.of(MapBuilder.get()
62+
.put(KEY, List.of("jade8xvdq4", "output", "a"))
63+
.put(VALUE, "after")
64+
.build()))
65+
.put(INTERNAL,
66+
MapBuilder.get()
67+
.put(OUTPUT_SCOPE,
68+
MapBuilder.get()
69+
.put("jade8xvdq4",
70+
MapBuilder.get()
71+
.put("output",
72+
MapBuilder.get().put("a", "before").build())
73+
.build())
74+
.build())
75+
.build())
76+
.build();
77+
}
78+
}

‎app-builder/plugins/pom.xml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
<module>aipp-prompt-builder</module>
3131
<module>aipp-rewriter</module>
3232
<module>aipp-variable-aggregation</module>
33+
<module>aipp-variable-updater</module>
3334
<module>aipp-websocket-plugin</module>
3435
<module>app-announcement</module>
3536
<module>app-base</module>

‎frontend/src/pages/addFlow/components/basic-item.tsx‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ const BasicItems = (props: any) => {
5555
'questionClassificationNodeCondition': <ClassificationIcon />,
5656
'httpNodeState': <HttpIcon />,
5757
'variableAggregationNodeState': <VariableAggregation />,
58+
'variableUpdateNodeState': <VariableAggregation />,
5859
'textToImageNodeState': <TextToImageIcon />,
5960
'fileExtractionNodeState': <FileExtractionIcon />,
6061
'noteNode': <ClassificationIcon />,

0 commit comments

Comments
 (0)