Refactor team selection to remove session_id and add user default team support

Removed session_id from TeamSelectionRequest and related API logic, simplifying team selection. Added support for setting and deleting a user's default team in backend services and database interfaces. Updated frontend TeamService to match new API contract.

Author: Fr4nc3

src/backend/common/database/cosmosdb.py

@@ -30,6 +30,7 @@
     AgentMessage,
     TeamConfiguration,
     DataType,
+    UserCurrentTeam,
 )
 
 
@@ -431,3 +432,7 @@ async def update_team(self, team: TeamConfiguration) -> None:
             team: The TeamConfiguration to update
         """
         await self.update_item(team)
+    
+    async def set_user_default_team(current_team: UserCurrentTeam) -> bool:
+
+

src/backend/common/database/database_base.py

@@ -9,6 +9,7 @@
     Plan,
     Step,
     TeamConfiguration,
+    UserCurrentTeam,
 )
 
 
@@ -80,6 +81,7 @@ async def add_plan(self, plan: Plan) -> None:
         """Add a plan to the database."""
         pass
 
+    
     @abstractmethod
     async def update_plan(self, plan: Plan) -> None:
         """Update a plan in the database."""
@@ -184,7 +186,13 @@ async def __aexit__(self, exc_type, exc, tb):
         """Async context manager exit."""
         await self.close()
 
-    # Convenience aliases
+    @abstractmethod
     async def get_steps_for_plan(self, plan_id: str) -> List[Step]:
         """Convenience method aliasing get_steps_by_plan for compatibility."""
         pass
+
+    @abstractmethod
+    async def set_user_default_team(self, UserCurrentTeam: UserCurrentTeam) -> bool:
+        """Set the default team for a user."""
+        pass
+

src/backend/common/models/messages_kernel.py

@@ -188,7 +188,6 @@ class StartingTask(KernelBaseModel):
 class TeamSelectionRequest(KernelBaseModel):
     """Request model for team selection."""
     team_id: str
-    session_id: Optional[str] = None
 
 class TeamConfiguration(BaseDataModel):
     """Represents a team configuration stored in the database."""

src/backend/v3/api/router.py

@@ -10,7 +10,7 @@
 from common.database.database_factory import DatabaseFactory
 from common.models.messages_kernel import (GeneratePlanRequest, InputTask,
                                            Plan, PlanStatus,
-                                           TeamSelectionRequest)
+                                           TeamSelectionRequest, UserCurrentTeam)
 from common.utils.event_utils import track_event_if_configured
 from common.utils.utils_kernel import rai_success, rai_validate_team_config
 from fastapi import (APIRouter, BackgroundTasks, Depends, FastAPI, File,
@@ -763,9 +763,15 @@ async def select_team_endpoint(selection: TeamSelectionRequest, request: Request
                 status_code=404, 
                 detail=f"Team configuration '{selection.team_id}' not found or access denied"
             )
-
-        # Generate session ID if not provided
-        session_id = selection.session_id or str(uuid.uuid4())
+        current_team = await team_service.get_user_current_team(user_id=user_id)
+        if current_team and current_team.team_id != selection.team_id:
+            current_team.team_id = selection.team_id
+            await team_service.set_user_current_team(user_id=user_id, current_team=current_team, )
+        else:
+            current_team = UserCurrentTeam(
+                user_id=user_id,
+                team_id=selection.team_id
+            )
 
         # save to in-memory config for current user
         team_config.set_current_team(user_id=user_id, team_configuration=team_configuration)
@@ -777,8 +783,7 @@ async def select_team_endpoint(selection: TeamSelectionRequest, request: Request
                 "status": "success",
                 "team_id": selection.team_id,
                 "team_name": team_configuration.name,
-                "user_id": user_id,
-                "session_id": session_id,
+                "user_id": user_id
             },
         )
 
@@ -787,7 +792,6 @@ async def select_team_endpoint(selection: TeamSelectionRequest, request: Request
             "message": f"Team '{team_configuration.name}' selected successfully",
             "team_id": selection.team_id,
             "team_name": team_configuration.name,
-            "session_id": session_id,
             "agents_count": len(team_configuration.agents),
             "team_description": team_configuration.description,
         }

src/backend/v3/common/services/team_service.py

@@ -218,6 +218,45 @@ async def get_team_configuration(
             self.logger.error("Error retrieving team configuration: %s", str(e))
             return None
 
+    async def delete_user_current_team(self, user_id: str) -> bool:
+        """
+        Delete the current team for a user.
+
+        Args:
+            user_id: User ID to delete the current team for
+
+        Returns:
+            True if successful, False otherwise
+        """
+        try:
+            await self.memory_context.delete_user_current_team(user_id)
+            self.logger.info("Successfully deleted current team for user %s", user_id)
+            return True
+
+        except Exception as e:
+            self.logger.error("Error deleting current team: %s", str(e))
+            return False
+
+    async def set_default_team(self, user_id: str, team_id: str) -> bool:
+        """
+        Set a default team for a user.
+
+        Args:
+            user_id: User ID to set the default team for
+            team_id: Team ID to set as default
+
+        Returns:
+            True if successful, False otherwise
+        """
+        try:
+            await self.memory_context.set_user_default_team(user_id, team_id)
+            self.logger.info("Successfully set default team for user %s: %s", user_id, team_id)
+            return True
+
+        except Exception as e:
+            self.logger.error("Error setting default team: %s", str(e))
+            return False
+
     async def get_all_team_configurations(
         self, user_id: str
     ) -> List[TeamConfiguration]:

src/frontend/src/services/TeamService.tsx

@@ -34,18 +34,18 @@ export class TeamService {
         try {
             console.log('Calling /v3/init_team endpoint...');
             const response = await apiClient.get('/v3/init_team');
-            
+
             console.log('Team initialization response:', response);
-            
+
             return {
                 success: true,
                 data: response
             };
         } catch (error: any) {
             console.error('Team initialization failed:', error);
-            
+
             let errorMessage = 'Failed to initialize team';
-            
+
             if (error.response?.data?.detail) {
                 errorMessage = error.response.data.detail;
             } else if (error.message) {
@@ -170,15 +170,14 @@ export class TeamService {
     /**
      * Select a team for a plan/session
      */
-    static async selectTeam(teamId: string, sessionId?: string): Promise<{
+    static async selectTeam(teamId: string): Promise<{
         success: boolean;
         data?: any;
         error?: string;
     }> {
         try {
             const response = await apiClient.post('/v3/select_team', {
                 team_id: teamId,
-                session_id: sessionId
             });
 
             return {
@@ -187,7 +186,7 @@ export class TeamService {
             };
         } catch (error: any) {
             let errorMessage = 'Failed to select team';
-            
+
             if (error.response?.data?.detail) {
                 errorMessage = error.response.data.detail;
             } else if (error.message) {