Add team configuration support and refactor data structure

Introduces team-specific methods to CosmosMemoryContext for managing TeamConfiguration objects, including add, update, retrieve, and delete operations. Refactors JsonService to use the new team methods and ensures unique IDs and timestamps are generated for team configs. Adds a default team JSON, a comprehensive test for team methods, and updates documentation. Also reorganizes data directory structure for consistency.

Author: Fr4nc3

docs/DeploymentGuide.md

@@ -206,43 +206,9 @@ Once you've opened the project in [Codespaces](#github-codespaces), [Dev Contain
 
 5. Once the deployment has completed successfully, open the [Azure Portal](https://portal.azure.com/), go to the deployed resource group, find the App Service, and get the app URL from `Default domain`.
 
-6. If you are done trying out the application, you can delete the resources by running `azd down`.
+6. When Deployment is complete, follow steps in [Set Up Authentication in Azure App Service](../docs/azure_app_service_auth_setup.md) to add app authentication to your web app running on Azure App Service
 
-### Publishing Local Build Container to Azure Container Registry
-
-If you need to rebuild the source code and push the updated container to the deployed Azure Container Registry, follow these steps:
-
-1. Set the environment variable `USE_LOCAL_BUILD` to `True`:
-
-   - **Linux/macOS**:
-
-     ```bash
-     export USE_LOCAL_BUILD=True
-     ```
-
-   - **Windows (PowerShell)**:
-     ```powershell
-     $env:USE_LOCAL_BUILD = $true
-     ```
-
-2. Run the `az login` command
-
-   ```bash
-   az login
-   ```
-
-3. Run the `azd up` command again to rebuild and push the updated container:
-   ```bash
-   azd up
-   ```
-
-This will rebuild the source code, package it into a container, and push it to the Azure Container Registry associated with your deployment.
-
-This guide provides step-by-step instructions for deploying your application using Azure Container Registry (ACR) and Azure Container Apps.
-
-There are several ways to deploy the solution. You can deploy to run in Azure in one click, or manually, or you can deploy locally.
-
-When Deployment is complete, follow steps in [Set Up Authentication in Azure App Service](../docs/azure_app_service_auth_setup.md) to add app authentication to your web app running on Azure App Service
+7. If you are done trying out the application, you can delete the resources by running `azd down`.
 
 # Local setup
 

src/backend/app_kernel.py

@@ -81,7 +81,9 @@
 # Add this near the top of your app.py, after initializing the app
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=[frontend_url],
+    allow_origins=[
+        frontend_url
+    ],  # Allow all origins for development; restrict in production
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -202,8 +204,8 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
                 "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"
-            ]
+                "Ensure your request complies with content policies",
+            ],
         }
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
@@ -287,9 +289,7 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
                 r"Rate limit is exceeded\. Try again in (\d+) seconds?\.", error_msg
             )
             if match:
-                error_msg = (
-                    f"Rate limit is exceeded. Try again in {match.group(1)} seconds."
-                )
+                error_msg = "Application temporarily unavailable due to quota limits. Please try again later."
 
         track_event_if_configured(
             "InputTaskError",
@@ -308,7 +308,7 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
 async def create_plan_endpoint(input_task: InputTask, request: Request):
     """
     Create a new plan without full processing.
-    
+
     ---
     tags:
       - Plans
@@ -365,19 +365,19 @@ async def create_plan_endpoint(input_task: InputTask, request: Request):
             },
         )
         raise HTTPException(
-            status_code=400, 
+            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", 
+                    "Use more professional and constructive language",
                     "Focus on legitimate business or educational objectives",
-                    "Ensure your request complies with content policies"
+                    "Ensure your request complies with content policies",
                 ],
-                "user_action": "Please revise your request and try again"
-            }
+                "user_action": "Please revise your request and try again",
+            },
         )
 
     # Get authenticated user
@@ -406,7 +406,7 @@ async def create_plan_endpoint(input_task: InputTask, request: Request):
             user_id=user_id,
             initial_goal=input_task.description,
             overall_status=PlanStatus.in_progress,
-            source=AgentType.PLANNER.value
+            source=AgentType.PLANNER.value,
         )
 
         # Save the plan to the database
@@ -442,10 +442,12 @@ async def create_plan_endpoint(input_task: InputTask, request: Request):
 
 
 @app.post("/api/generate_plan")
-async def generate_plan_endpoint(generate_plan_request: GeneratePlanRequest, request: Request):
+async def generate_plan_endpoint(
+    generate_plan_request: GeneratePlanRequest, request: Request
+):
     """
     Generate plan steps for an existing plan using the planner agent.
-    
+
     ---
     tags:
       - Plans
@@ -511,11 +513,17 @@ async def generate_plan_endpoint(generate_plan_request: GeneratePlanRequest, req
         kernel, memory_store = await initialize_runtime_and_context("", user_id)
 
         # Get the existing plan
-        plan = await memory_store.get_plan_by_plan_id(plan_id=generate_plan_request.plan_id)
+        plan = await memory_store.get_plan_by_plan_id(
+            plan_id=generate_plan_request.plan_id
+        )
         if not plan:
             track_event_if_configured(
                 "GeneratePlanNotFound",
-                {"status_code": 404, "detail": "Plan not found", "plan_id": generate_plan_request.plan_id}
+                {
+                    "status_code": 404,
+                    "detail": "Plan not found",
+                    "plan_id": generate_plan_request.plan_id,
+                },
             )
             raise HTTPException(status_code=404, detail="Plan not found")
 
@@ -538,16 +546,19 @@ async def generate_plan_endpoint(generate_plan_request: GeneratePlanRequest, req
 
         # Create an InputTask from the plan's initial goal
         input_task = InputTask(
-            session_id=plan.session_id,
-            description=plan.initial_goal
+            session_id=plan.session_id, description=plan.initial_goal
         )
 
         # Use the group chat manager to generate the plan steps
         await group_chat_manager.handle_input_task(input_task)
 
         # Get the updated plan with steps
-        updated_plan = await memory_store.get_plan_by_plan_id(plan_id=generate_plan_request.plan_id)
-        steps = await memory_store.get_steps_by_plan(plan_id=generate_plan_request.plan_id)
+        updated_plan = await memory_store.get_plan_by_plan_id(
+            plan_id=generate_plan_request.plan_id
+        )
+        steps = await memory_store.get_steps_by_plan(
+            plan_id=generate_plan_request.plan_id
+        )
 
         # Log successful plan generation
         track_event_if_configured(
@@ -760,7 +771,7 @@ async def human_clarification_endpoint(
             },
         )
         raise HTTPException(
-            status_code=400, 
+            status_code=400,
             detail={
                 "error_type": "RAI_VALIDATION_FAILED",
                 "message": "Clarification Safety Check Failed",
@@ -769,10 +780,10 @@ async def human_clarification_endpoint(
                     "Use clear and professional language",
                     "Avoid potentially harmful or inappropriate content",
                     "Focus on providing constructive feedback or clarification",
-                    "Ensure your message complies with content policies"
+                    "Ensure your message complies with content policies",
                 ],
-                "user_action": "Please revise your clarification and try again"
-            }
+                "user_action": "Please revise your clarification and try again",
+            },
         )
 
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)

src/backend/context/cosmos_memory_kernel.py

@@ -17,7 +17,14 @@
 
 # Import the AppConfig instance
 from app_config import config
-from models.messages_kernel import BaseDataModel, Plan, Session, Step, AgentMessage
+from models.messages_kernel import (
+    BaseDataModel,
+    Plan,
+    Session,
+    Step,
+    AgentMessage,
+    TeamConfiguration,
+)
 
 
 # Add custom JSON encoder class for datetime objects
@@ -38,6 +45,7 @@ class CosmosMemoryContext(MemoryStoreBase):
         "plan": Plan,
         "step": Step,
         "agent_message": AgentMessage,
+        "team_config": TeamConfiguration,
         # Messages are handled separately
     }
 
@@ -341,6 +349,131 @@ async def get_agent_messages_by_session(
         messages = await self.query_items(query, parameters, AgentMessage)
         return messages
 
+    async def add_team(self, team: TeamConfiguration) -> None:
+        """Add a team configuration to Cosmos DB.
+
+        Args:
+            team: The TeamConfiguration to add
+        """
+        await self.add_item(team)
+
+    async def update_team(self, team: TeamConfiguration) -> None:
+        """Update an existing team configuration in Cosmos DB.
+
+        Args:
+            team: The TeamConfiguration to update
+        """
+        await self.update_item(team)
+
+    async def get_team(self, team_id: str) -> Optional[TeamConfiguration]:
+        """Retrieve a specific team configuration by team_id.
+
+        Args:
+            team_id: The team_id of the team configuration to retrieve
+
+        Returns:
+            TeamConfiguration object or None if not found
+        """
+        query = "SELECT * FROM c WHERE c.team_id=@team_id AND c.data_type=@data_type"
+        parameters = [
+            {"name": "@team_id", "value": team_id},
+            {"name": "@data_type", "value": "team_config"},
+        ]
+        teams = await self.query_items(query, parameters, TeamConfiguration)
+        return teams[0] if teams else None
+
+    async def get_team_by_id(self, id: str) -> Optional[TeamConfiguration]:
+        """Retrieve a specific team configuration by its document id.
+
+        Args:
+            id: The document id of the team configuration to retrieve
+
+        Returns:
+            TeamConfiguration object or None if not found
+        """
+        query = "SELECT * FROM c WHERE c.id=@id AND c.data_type=@data_type"
+        parameters = [
+            {"name": "@id", "value": id},
+            {"name": "@data_type", "value": "team_config"},
+        ]
+        teams = await self.query_items(query, parameters, TeamConfiguration)
+        return teams[0] if teams else None
+
+    async def get_all_teams_by_user(self, user_id: str) -> List[TeamConfiguration]:
+        """Retrieve all team configurations for a specific user.
+
+        Args:
+            user_id: The user_id to get team configurations for
+
+        Returns:
+            List of TeamConfiguration objects
+        """
+        query = "SELECT * FROM c WHERE c.user_id=@user_id AND c.data_type=@data_type ORDER BY c.created DESC"
+        parameters = [
+            {"name": "@user_id", "value": user_id},
+            {"name": "@data_type", "value": "team_config"},
+        ]
+        teams = await self.query_items(query, parameters, TeamConfiguration)
+        return teams
+
+    async def delete_team(self, team_id: str) -> bool:
+        """Delete a team configuration by team_id.
+
+        Args:
+            team_id: The team_id of the team configuration to delete
+
+        Returns:
+            True if team was found and deleted, False otherwise
+        """
+        await self.ensure_initialized()
+
+        try:
+            # First find the team to get its document id and partition key
+            team = await self.get_team(team_id)
+            if team:
+                # Use the session_id as partition key, or fall back to user_id if no session_id
+                partition_key = (
+                    team.session_id
+                    if hasattr(team, "session_id") and team.session_id
+                    else team.user_id
+                )
+                await self._container.delete_item(
+                    item=team.id, partition_key=partition_key
+                )
+                return True
+            return False
+        except Exception as e:
+            logging.exception(f"Failed to delete team from Cosmos DB: {e}")
+            return False
+
+    async def delete_team_by_id(self, id: str) -> bool:
+        """Delete a team configuration by its document id.
+
+        Args:
+            id: The document id of the team configuration to delete
+
+        Returns:
+            True if team was found and deleted, False otherwise
+        """
+        await self.ensure_initialized()
+
+        try:
+            # First find the team to get its partition key
+            team = await self.get_team_by_id(id)
+            if team:
+                # Use the session_id as partition key, or fall back to user_id if no session_id
+                partition_key = (
+                    team.session_id
+                    if hasattr(team, "session_id") and team.session_id
+                    else team.user_id
+                )
+                await self._container.delete_item(item=id, partition_key=partition_key)
+                return True
+            return False
+        except Exception as e:
+            logging.exception(f"Failed to delete team from Cosmos DB: {e}")
+            return False
+
     async def add_message(self, message: ChatMessageContent) -> None:
         """Add a message to the memory and save to Cosmos DB."""
         await self.ensure_initialized()