Update plan model and status handling across backend and frontend

Refactored Plan model fields in frontend to match backend, replacing 'title' and 'description' with 'initial_goal' and updating status fields. Adjusted backend to ensure plan completion only when total steps are greater than zero. Updated API and TaskService to use new fields and status logic for consistency.

Author: Fr4nc3

src/backend/common/models/messages_kernel.py

@@ -232,9 +232,11 @@ def update_step_counts(self):
         self.completed = status_counts[StepStatus.completed]
         self.failed = status_counts[StepStatus.failed]
 
-        # Mark the plan as complete if the sum of completed and failed steps equals the total number of steps
-        if self.completed + self.failed == self.total_steps:
+
+        if self.total_steps > 0 and (self.completed + self.failed) == self.total_steps:
             self.overall_status = PlanStatus.completed
+        # Mark the plan as complete if the sum of completed and failed steps equals the total number of steps
+
 
 
 # Message classes for communication between agents

src/backend/v3/api/router.py

@@ -1030,7 +1030,16 @@ async def get_plans(request: Request):
 
     all_plans = await memory_store.get_all_plans_by_team_id(team_id=current_team.team_id)
 
-    return all_plans
+    steps_for_all_plans = []
+    # Create list of PlanWithSteps and update step counts
+    list_of_plans_with_steps = []
+    for plan in all_plans:
+        plan_with_steps = PlanWithSteps(**plan.model_dump(), steps=[])
+        plan_with_steps.overall_status
+        plan_with_steps.update_step_counts()
+        list_of_plans_with_steps.append(plan_with_steps)
+
+    return list_of_plans_with_steps
 
 
 # Get plans is called in the initial side rendering of the frontend

src/frontend/src/models/plan.tsx

@@ -8,9 +8,9 @@ export interface BaseModel {
     /** Unique identifier */
     id: string;
     /** Timestamp when created */
-    created_at: string;
+
     /** Timestamp when last updated */
-    updated_at: string;
+    timestamp: string;
 }
 
 /**
@@ -24,11 +24,10 @@ export interface Plan extends BaseModel {
     /** User identifier */
     user_id: string;
     /** Plan title */
-    title: string;
-    /** Plan description */
-    description: string;
+    initial_goal: string;
+
     /** Current status of the plan */
-    status: PlanStatus;
+    overall_status: PlanStatus;
     /** Human clarification request text */
     human_clarification_request?: string;
     /** Human clarification response text */

src/frontend/src/services/TaskService.tsx

@@ -27,14 +27,14 @@ export class TaskService {
     plansData.forEach((plan) => {
       const task: Task = {
         id: plan.session_id,
-        name: plan.title,
+        name: plan.initial_goal,
         completed_steps: plan.completed,
         total_steps: plan.total_steps,
-        status: PlanDataService.isPlanComplete(plan) ? "completed" : "inprogress",
+        status: plan.overall_status === PlanStatus.COMPLETED ? "completed" : "inprogress",
         date: new Intl.DateTimeFormat(undefined, {
           dateStyle: "long",
           // timeStyle: "short",
-        }).format(new Date(plan.updated_at)),
+        }).format(new Date(plan.timestamp)),
       };
 
       // Categorize based on plan status and completion