2867 - created getAllTasksRecursively because the initial implementation didn't account for tasks in cluster elements, rag and tools

Author: marko-kriskovic

client/src/pages/platform/workflow-editor/utils/getAllTasksRecursively.ts

@@ -0,0 +1,36 @@
+import {TASK_DISPATCHER_NAMES} from '@/shared/constants';
+import {WorkflowTask} from '@/shared/middleware/platform/configuration';
+
+import {TASK_DISPATCHER_CONFIG} from './taskDispatcherConfig';
+
+/**
+ * Recursively gets all tasks from a workflow, including tasks nested inside task dispatchers
+ */
+export default function getAllTasksRecursively(tasks: Array<WorkflowTask>): Array<WorkflowTask> {
+    const allTasks: Array<WorkflowTask> = [];
+
+    const extractTasks = (taskList: Array<WorkflowTask>) => {
+        taskList.forEach((task) => {
+            allTasks.push(task);
+
+            // Check if this is a task dispatcher
+            const componentName = task.type?.split('/')[0];
+
+            if (componentName && TASK_DISPATCHER_NAMES.includes(componentName)) {
+                const config = TASK_DISPATCHER_CONFIG[componentName as keyof typeof TASK_DISPATCHER_CONFIG];
+
+                if (config) {
+                    const subtasks = config.getSubtasks({getAllSubtasks: true, task});
+
+                    if (subtasks && subtasks.length > 0) {
+                        extractTasks(subtasks);
+                    }
+                }
+            }
+        });
+    };
+
+    extractTasks(tasks);
+
+    return allTasks;
+}

client/src/pages/platform/workflow-editor/utils/getFormattedName.ts

@@ -3,6 +3,7 @@ import {ClusterElementsType, NodeDataType} from '@/shared/types';
 
 import {isPlainObject} from '../../cluster-element-editor/utils/clusterElementsUtils';
 import useWorkflowDataStore from '../stores/useWorkflowDataStore';
+import getAllTasksRecursively from './getAllTasksRecursively';
 
 export default function getFormattedName(itemName: string): string {
     const {nodes, workflow} = useWorkflowDataStore.getState();
@@ -13,7 +14,9 @@ export default function getFormattedName(itemName: string): string {
 
     const workflowDefinition = JSON.parse(workflow.definition!);
 
-    const clusterElementNames = workflowDefinition.tasks.map((task: WorkflowTask) => {
+    const allTasks = getAllTasksRecursively(workflowDefinition.tasks);
+
+    const clusterElementNames = allTasks.map((task: WorkflowTask) => {
         const elementNames: string[] = [];
 
         const {clusterElements} = task;