Add team-based plan retrieval and refactor endpoints

Introduces get_all_plans_by_team_id to database interfaces and updates plan retrieval to be team-specific in the backend. Refactors several API endpoint function names for consistency. Removes step approval/rejection buttons from TaskDetails component and streamlines team initialization logic in HomePage, improving fallback handling and user feedback.

Author: Fr4nc3

src/backend/app_kernel.py

@@ -697,7 +697,11 @@ async def get_plans(
 
         return [plan_with_steps, formatted_messages]
 
-    all_plans = await memory_store.get_all_plans()
+    current_team = await memory_store.get_current_team(user_id=user_id)
+    if not current_team:
+      return []
+    
+    all_plans = await memory_store.get_all_plans_by_team_id(team_id=current_team.id)
     # Fetch steps for all plans concurrently
     steps_for_all_plans = await asyncio.gather(
         *[memory_store.get_steps_by_plan(plan_id=plan.id) for plan in all_plans]

src/backend/common/database/cosmosdb.py

@@ -259,6 +259,16 @@ async def get_all_plans(self) -> List[Plan]:
         ]
         return await self.query_items(query, parameters, Plan)
 
+    async def get_all_plans_by_team_id(self, team_id: str) -> List[Plan]:
+        """Retrieve all plans for a specific team."""
+        query = "SELECT * FROM c WHERE c.team_id=@team_id AND c.data_type=@data_type and c.user_id=@user_id"
+        parameters = [
+            {"name": "@user_id", "value": self.user_id},
+            {"name": "@team_id", "value": team_id},
+            {"name": "@data_type", "value": "plan"},
+        ]
+        return await self.query_items(query, parameters, Plan)
+
     # Step Operations
     async def add_step(self, step: Step) -> None:
         """Add a step to CosmosDB."""

src/backend/common/database/database_base.py

@@ -106,7 +106,10 @@ async def get_plan(self, plan_id: str) -> Optional[Plan]:
     async def get_all_plans(self) -> List[Plan]:
         """Retrieve all plans for the user."""
         pass
-
+    @abstractmethod
+    async def get_all_plans_by_team_id(self, team_id: str) -> List[Plan]:
+        """Retrieve all plans for a specific team."""
+        pass
     @abstractmethod
     async def get_data_by_type_and_session_id(
         self, data_type: str, session_id: str

src/backend/v3/api/router.py

@@ -189,30 +189,30 @@ async def process_request(background_tasks: BackgroundTasks, input_task: InputTa
     """
 
 
-    # if not await rai_success(input_task.description, False):
-    #     track_event_if_configured(
-    #         "RAI failed",
-    #         {
-    #             "status": "Plan not created - RAI check failed",
-    #             "description": input_task.description,
-    #             "session_id": input_task.session_id,
-    #         },
-    #     )
-    #     raise HTTPException(
-    #         status_code=400,
-    #         detail={
-    #             "error_type": "RAI_VALIDATION_FAILED",
-    #             "message": "Content Safety Check Failed",
-    #             "description": "Your request contains content that doesn't meet our safety guidelines. Please modify your request to ensure it's appropriate and try again.",
-    #             "suggestions": [
-    #                 "Remove any potentially harmful, inappropriate, or unsafe content",
-    #                 "Use more professional and constructive language",
-    #                 "Focus on legitimate business or educational objectives",
-    #                 "Ensure your request complies with content policies",
-    #             ],
-    #             "user_action": "Please revise your request and try again",
-    #         },
-    #     )
+    if not await rai_success(input_task.description, False):
+        track_event_if_configured(
+            "RAI failed",
+            {
+                "status": "Plan not created - RAI check failed",
+                "description": input_task.description,
+                "session_id": input_task.session_id,
+            },
+        )
+        raise HTTPException(
+            status_code=400,
+            detail={
+                "error_type": "RAI_VALIDATION_FAILED",
+                "message": "Content Safety Check Failed",
+                "description": "Your request contains content that doesn't meet our safety guidelines. Please modify your request to ensure it's appropriate and try again.",
+                "suggestions": [
+                    "Remove any potentially harmful, inappropriate, or unsafe content",
+                    "Use more professional and constructive language",
+                    "Focus on legitimate business or educational objectives",
+                    "Ensure your request complies with content policies",
+                ],
+                "user_action": "Please revise your request and try again",
+            },
+        )
 
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
@@ -360,7 +360,7 @@ async def user_clarification(human_feedback: messages.UserClarificationResponse,
 
 
 @app_v3.post("/upload_team_config")
-async def upload_team_config_endpoint(request: Request, file: UploadFile = File(...),     team_id: Optional[str] = Query(None),):
+async def upload_team_config(request: Request, file: UploadFile = File(...),     team_id: Optional[str] = Query(None),):
     """
     Upload and save a team configuration JSON file.
 
@@ -534,7 +534,7 @@ async def upload_team_config_endpoint(request: Request, file: UploadFile = File(
 
 
 @app_v3.get("/team_configs")
-async def get_team_configs_endpoint(request: Request):
+async def get_team_configs(request: Request):
     """
     Retrieve all team configurations for the current user.
 
@@ -607,7 +607,7 @@ async def get_team_configs_endpoint(request: Request):
 
 
 @app_v3.get("/team_configs/{team_id}")
-async def get_team_config_by_id_endpoint(team_id: str, request: Request):
+async def get_team_config_by_id(team_id: str, request: Request):
     """
     Retrieve a specific team configuration by ID.
 
@@ -689,7 +689,7 @@ async def get_team_config_by_id_endpoint(team_id: str, request: Request):
 
 
 @app_v3.delete("/team_configs/{team_id}")
-async def delete_team_config_endpoint(team_id: str, request: Request):
+async def delete_team_config(team_id: str, request: Request):
     """
     Delete a team configuration by ID.
 
@@ -767,7 +767,7 @@ async def delete_team_config_endpoint(team_id: str, request: Request):
 
 
 @app_v3.get("/model_deployments")
-async def get_model_deployments_endpoint(request: Request):
+async def get_model_deployments(request: Request):
     """
     Get information about available model deployments for debugging/validation.
 
@@ -799,7 +799,7 @@ async def get_model_deployments_endpoint(request: Request):
 
 
 @app_v3.post("/select_team")
-async def select_team_endpoint(selection: TeamSelectionRequest, request: Request):
+async def select_team(selection: TeamSelectionRequest, request: Request):
     """
     Select the current team for the user session.
     """
@@ -883,7 +883,7 @@ async def select_team_endpoint(selection: TeamSelectionRequest, request: Request
 
 
 @app_v3.get("/search_indexes")
-async def get_search_indexes_endpoint(request: Request):
+async def get_search_indexes(request: Request):
     """
     Get information about available search indexes for debugging/validation.
 

src/frontend/src/components/content/TaskDetails.tsx

@@ -149,94 +149,6 @@ const TaskDetails: React.FC<TaskDetailsProps> = ({
                       <Caption1>{functionOrDetails}</Caption1>
                     )}
                   </Body1>
-                    <div className="task-details-action-buttons">
-                    {step.human_approval_status !== "accepted" &&
-                      step.human_approval_status !== "rejected" && (
-                      <>
-                        <Tooltip relationship="label" content={canInteract ? "Approve" : "You must first provide feedback to the planner"}>
-                        <Button
-                          icon={<Checkmark20Regular />}
-                          appearance="subtle"
-                          onClick={
-                          canInteract
-                            ? async (e) => {
-                              // Disable buttons for this step
-                              setSteps((prev) =>
-                              prev.map((s) =>
-                                s.id === step.id
-                                ? { ...s, _isActionLoading: true }
-                                : s
-                              )
-                              );
-                              try {
-                              await preOnApproved(step);
-                              } finally {
-                              // Remove loading state after API call
-                              setSteps((prev) =>
-                                prev.map((s) =>
-                                s.id === step.id
-                                  ? { ...s, _isActionLoading: false }
-                                  : s
-                                )
-                              );
-                              }
-                            }
-                            : undefined
-                          }
-                          disabled={
-                          !canInteract ||
-                          !!step._isActionLoading
-                          }
-                          className={
-                          canInteract
-                            ? "task-details-action-button"
-                            : "task-details-action-button-disabled"
-                          }
-                        />
-                        </Tooltip>
-
-                        <Tooltip relationship="label" content={canInteract ? "Reject" : "You must first provide feedback to the planner"}>
-                        <Button
-                          icon={<Dismiss20Regular />}
-                          appearance="subtle"
-                          onClick={
-                          canInteract
-                            ? async (e) => {
-                              setSteps((prev) =>
-                              prev.map((s) =>
-                                s.id === step.id
-                                ? { ...s, _isActionLoading: true }
-                                : s
-                              )
-                              );
-                              try {
-                              await preOnRejected(step);
-                              } finally {
-                              setSteps((prev) =>
-                                prev.map((s) =>
-                                s.id === step.id
-                                  ? { ...s, _isActionLoading: false }
-                                  : s
-                                )
-                              );
-                              }
-                            }
-                            : undefined
-                          }
-                          disabled={
-                          !canInteract ||
-                          !!step._isActionLoading
-                          }
-                          className={
-                          canInteract
-                            ? "task-details-action-button"
-                            : "task-details-action-button-disabled"
-                          }
-                        />
-                        </Tooltip>
-                      </>
-                      )}
-                    </div>
                 </div>
               </div>
             );