Feature/display asst messages (#331)

* Display asst messages in chat

* Use array of array for child_ids in Tasks

Author: ppsreejith

web/src/components/common/Tasks.tsx

@@ -59,7 +59,7 @@ const TreeNode: React.FC<TreeNodeProps> = ({
     if (!Array.isArray(task.child_ids)) {
         return [];
     }
-    return allTasks.filter(t => task.child_ids.includes(t.id));
+    return allTasks.filter(t => task.child_ids.flat().includes(t.id));
   }, [task.child_ids, allTasks]);
 
   const hasChildren = childTasks.length > 0;
@@ -305,10 +305,10 @@ export const Tasks: React.FC = () => {
   const rootTasks = useMemo(() => {
     if (!allTasks || allTasks.length === 0) return [];
     const taskMap = new Map(allTasks.map(t => [t.id, t]));
-    const childIds = new Set<string>();
+    const childIds = new Set<number>();
     allTasks.forEach(task => {
         if (Array.isArray(task.child_ids)) {
-            task.child_ids.forEach(cid => childIds.add(cid));
+            task.child_ids.flat().forEach(cid => childIds.add(cid));
         }
     });
     return allTasks.filter(task =>

web/src/components/common/TasksLite.tsx

@@ -188,6 +188,7 @@ const flattenTasks = (tasks: TasksInfo, expandedRootTasks: Set<string> = new Set
 
     if (shouldShowChildren && Array.isArray(task.child_ids) && task.child_ids.length > 0) {
       const childTasks = task.child_ids
+        .flat()
         .map(childId => taskMap.get(childId))
         .filter((child): child is Task => child !== undefined);
 
@@ -201,10 +202,10 @@ const flattenTasks = (tasks: TasksInfo, expandedRootTasks: Set<string> = new Set
   };
 
   // Find root tasks (tasks without parents or whose parents don't exist)
-  const childIds = new Set<string>();
+  const childIds = new Set<number>();
   tasks.forEach(task => {
     if (Array.isArray(task.child_ids)) {
-      task.child_ids.forEach(cid => childIds.add(cid));
+      task.child_ids.flat().forEach(cid => childIds.add(cid));
     }
   });
 
@@ -240,10 +241,10 @@ export const TasksLite: React.FC = () => {
   const rootTasks = useMemo(() => {
     if (!allTasks || allTasks.length === 0) return [];
     const taskMap = new Map(allTasks.map(t => [t.id, t]));
-    const childIds = new Set<string>();
+    const childIds = new Set<number>();
     allTasks.forEach(task => {
         if (Array.isArray(task.child_ids)) {
-            task.child_ids.forEach(cid => childIds.add(cid));
+            task.child_ids.flat().forEach(cid => childIds.add(cid));
         }
     });
     return allTasks.filter(task =>

web/src/state/chat/reducer.ts

@@ -135,7 +135,7 @@ export interface Task {
   parent_id: number | null;
   run_id: string;
   debug: any;
-  child_ids: number[];
+  child_ids: Array<number[]>;
   result: any; // Key field for completion status
 }
 
@@ -238,6 +238,17 @@ const taskResultToContent = (task: Task): ActionChatMessageContent => {
     ? task.result 
     : JSON.stringify(task.result)
 
+  if (task.agent == "TalkToUser") {
+    return {
+      type: 'DEFAULT',
+      text: content,
+      images: [],
+      renderInfo: {
+        text: content,
+        code: undefined,
+      }
+    }
+  }
   return {
     type: 'BLANK',
     content,