extend json-node with @Flat command to flatten array

Author: devhelpr

libs/web-flow-executor/src/nodes/json-utils/transform-json.test.ts

@@ -173,4 +173,22 @@ describe('transformJSON', () => {
       role: 'assistant',
     });
   });
+
+  it('should transform a nested array with @flat', () => {
+    const input = {
+      '@flat:output': [1, 2, [3, 4]],
+    };
+    const result = transformJSON(input, undefined, 'root', {});
+    expect((result as any)['output']).toEqual([1, 2, 3, 4]);
+  });
+
+  it('should transform a property from the payload with @flat', () => {
+    const input = {
+      '@flat:output': 'test',
+    };
+    const result = transformJSON(input, undefined, 'root', {
+      test: [1, 2, [3, 4]],
+    });
+    expect((result as any)['output']).toEqual([1, 2, 3, 4]);
+  });
 });

libs/web-flow-executor/src/nodes/json-utils/transform-json.ts

@@ -23,7 +23,19 @@ const evaluateAtProperty = (
   payload: any,
   initialPayload?: any
 ) => {
-  if (key.startsWith('@serialize')) {
+  if (key.startsWith('@flat')) {
+    const flatProperty = key.replace('@flat:', '');
+    if (Array.isArray(value)) {
+      const result = value.flat();
+
+      newJson[flatProperty] = result;
+    } else if (typeof value === 'string') {
+      const flattenValue = initialPayload[value];
+      newJson[flatProperty] = flattenValue.flat();
+    } else {
+      newJson[flatProperty] = value;
+    }
+  } else if (key.startsWith('@serialize')) {
     const setProperty = key.replace('@serialize:', '');
 
     const toSerialize: Record<string, any> = {};