Add method to query data by type and session ID

Introduces get_data_by_type_and_session_id to DatabaseBase and implements it in CosmosDBClient. This method allows querying documents by data_type, session_id, and user_id, improving data retrieval granularity.

Author: Fr4nc3

src/backend/common/database/cosmosdb.py

@@ -361,6 +361,27 @@ async def delete_team(self, team_id: str) -> bool:
             logging.exception(f"Failed to delete team from Cosmos DB: {e}")
             return False
 
+    async def get_data_by_type_and_session_id(
+        self, data_type: str, session_id: str
+    ) -> List[BaseDataModel]:
+        """Query the Cosmos DB for documents with the matching data_type, session_id and user_id."""
+        await self._ensure_initialized()
+        if self.container is None:
+            return []
+
+        model_class = self.MODEL_CLASS_MAPPING.get(data_type, BaseDataModel)
+        try:
+            query = "SELECT * FROM c WHERE c.session_id=@session_id AND c.user_id=@user_id AND c.data_type=@data_type ORDER BY c._ts ASC"
+            parameters = [
+                {"name": "@session_id", "value": session_id},
+                {"name": "@data_type", "value": data_type},
+                {"name": "@user_id", "value": self.user_id},
+            ]
+            return await self.query_items(query, parameters, model_class)
+        except Exception as e:
+            logging.exception(f"Failed to query data by type from Cosmos DB: {e}")
+            return []
+
     # Data Management Operations
     async def get_data_by_type(self, data_type: str) -> List[BaseDataModel]:
         """Retrieve all data of a specific type."""

src/backend/common/database/database_base.py

@@ -105,6 +105,12 @@ async def get_all_plans(self) -> List[Plan]:
         """Retrieve all plans for the user."""
         pass
 
+    @abstractmethod
+    async def get_data_by_type_and_session_id(
+        self, data_type: str, session_id: str
+    ) -> List[BaseDataModel]:
+        pass
+
     # Step Operations
     @abstractmethod
     async def add_step(self, step: Step) -> None: