Refactor plan creation and make team_id optional

Changed Plan model to use plan_id and made team_id optional. Refactored plan creation logic in router.py to handle plan creation errors, removed mandatory team_id check, and added event tracking for plan creation success and failure.

Author: Fr4nc3

src/backend/common/models/messages_kernel.py

@@ -109,13 +109,14 @@ class Plan(BaseDataModel):
     """Represents a plan containing multiple steps."""
 
     data_type: Literal["plan"] = Field("plan", Literal=True)
-    team_id: str
+    plan_id: str 
     session_id: str
     user_id: str
     initial_goal: str
     overall_status: PlanStatus = PlanStatus.in_progress
     source: str = AgentType.PLANNER.value
     summary: Optional[str] = None
+    team_id: Optional[str] = None
     human_clarification_request: Optional[str] = None
     human_clarification_response: Optional[str] = None
 

src/backend/v3/api/router.py

@@ -216,32 +216,62 @@ async def process_request(background_tasks: BackgroundTasks, input_task: InputTa
         )
         raise HTTPException(status_code=400, detail="no user")
 
-    if not input_task.team_id:
-        track_event_if_configured(
-            "TeamIDNofound", {"status_code": 400, "detail": "no team id"}
-        )
-        raise HTTPException(status_code=400, detail="no team id")
+    # if not input_task.team_id:
+    #     track_event_if_configured(
+    #         "TeamIDNofound", {"status_code": 400, "detail": "no team id"}
+    #     )
+    #     raise HTTPException(status_code=400, detail="no team id")
 
     if not input_task.session_id:
         input_task.session_id = str(uuid.uuid4())
- 
-    plan_id = str(uuid.uuid4())
-
     try:
-        current_user_id.set(user_id)  # Set context
-        current_context = contextvars.copy_context()  # Capture context
-        # background_tasks.add_task(
-        #     lambda: current_context.run(lambda:OrchestrationManager().run_orchestration, user_id, input_task)
-        # )
+        plan_id = str(uuid.uuid4())
+        # Initialize memory store and service
         memory_store = await DatabaseFactory.get_database(user_id=user_id)
         plan = Plan(
             id=plan_id,
+            plan_id=plan_id,
             user_id=user_id,
-            team_id="", #TODO add current_team_id
+            session_id=input_task.session_id,
+            team_id=None,  #TODO add current_team_id
             initial_goal=input_task.description,
             overall_status=PlanStatus.in_progress,
         )
         await memory_store.add_plan(plan)
+
+
+        track_event_if_configured(
+            "PlanCreated",
+            {
+                "status": "success",
+                "plan_id": plan.plan_id,
+                "session_id": input_task.session_id,
+                "user_id": user_id,
+                "team_id": "", #TODO add current_team_id
+                "description": input_task.description,
+            },
+        )
+    except Exception as e:
+        print(f"Error creating plan: {e}")
+        track_event_if_configured(
+            "PlanCreationFailed",
+            {
+                "status": "error",
+                "description": input_task.description,
+                "session_id": input_task.session_id,
+                "user_id": user_id,
+                "error": str(e),
+            },
+        )
+        raise HTTPException(status_code=500, detail="Failed to create plan")
+
+    try:
+        current_user_id.set(user_id)  # Set context
+        current_context = contextvars.copy_context()  # Capture context
+        # background_tasks.add_task(
+        #     lambda: current_context.run(lambda:OrchestrationManager().run_orchestration, user_id, input_task)
+        # )
+
         async def run_with_context():
             return await current_context.run(OrchestrationManager().run_orchestration, user_id, input_task)