chore: tidy up

Author: iziang

aperag/db/repositories/collection.py

@@ -202,13 +202,12 @@ async def _query(session):
 
         return await self._execute_query(_query)
 
-    async def query_collections_by_ids(self, user: str, collection_ids: List[str]):
+    async def query_collections_by_ids(self, collection_ids: List[str]):
         """Query multiple collections by their IDs in a single database call"""
 
         async def _query(session):
             stmt = select(Collection).where(
                 Collection.id.in_(collection_ids),
-                Collection.user == user,
                 Collection.status != CollectionStatus.DELETED,
             )
             result = await session.execute(stmt)

aperag/service/agent_chat_service.py

@@ -175,7 +175,7 @@ async def handle_websocket_agent_chat(self, websocket: WebSocket, user: str, bot
             # Get default collections once for performance
             if bot_config.agent.collections:
                 agent_collection_ids = [collection.id for collection in bot_config.agent.collections]
-                agent_collections = await self.db_ops.query_collections_by_ids(user, agent_collection_ids)
+                agent_collections = await self.db_ops.query_collections_by_ids(agent_collection_ids)
                 for agent_collection in agent_collections:
                     default_collections.append(view_models.Collection(
                         id=agent_collection.id,

aperag/service/collection_service.py

@@ -616,49 +616,6 @@ async def delete_search(self, user: str, collection_id: str, search_id: str) ->
 
         return await self.db_ops.delete_search(user, collection_id, search_id)
 
-    async def validate_collections_batch(
-        self, user: str, collections: list[view_models.Collection]
-    ) -> tuple[bool, str]:
-        """
-        Validate multiple collections in a single database call.
-
-        Args:
-            user: User identifier
-            collections: List of collection objects to validate
-
-        Returns:
-            Tuple of (is_valid, error_message). If valid, error_message is empty.
-        """
-        if not collections:
-            return True, ""
-
-        # Extract collection IDs and validate they exist
-        collection_ids = []
-        for collection in collections:
-            if not collection.id:
-                return False, "Collection object missing 'id' field"
-            collection_ids.append(collection.id)
-
-        # Remove duplicates while preserving order
-        unique_collection_ids = list(dict.fromkeys(collection_ids))
-
-        try:
-            # Single database call to get all collections
-            db_collections = await self.db_ops.query_collections_by_ids(user, unique_collection_ids)
-
-            # Create a set of found collection IDs for fast lookup
-            found_collection_ids = {str(col.id) for col in db_collections}
-
-            # Check if all requested collections were found
-            for collection_id in unique_collection_ids:
-                if collection_id not in found_collection_ids:
-                    return False, f"Collection {collection_id} not found"
-
-            return True, ""
-
-        except Exception as e:
-            return False, f"Failed to validate collections: {str(e)}"
-
     async def test_mineru_token(self, token: str) -> dict:
         """Test the MinerU API token."""
         async with httpx.AsyncClient() as client: