feat: Add pagination with total count to generation listings and enable filtering assets by type.

repos/assets_repo.py

@@ -46,7 +46,10 @@ class AssetsRepo:
         res = await self.collection.insert_one(asset.model_dump())
         return str(res.inserted_id)
 
-    async def get_assets(self, limit: int = 10, offset: int = 0, with_data: bool = False) -> List[Asset]:
+    async def get_assets(self, asset_type: Optional[str] = None, limit: int = 10, offset: int = 0, with_data: bool = False) -> List[Asset]:
+        filter = {}
+        if asset_type:
+            filter["type"] = asset_type
         args = {}
         if not with_data:
             args["data"] = 0
@@ -68,7 +71,7 @@ class AssetsRepo:
             # So list DOES NOT return thumbnails by default.
             args["thumbnail"] = 0
             
-        res = await self.collection.find({}, args).sort("created_at", -1).skip(offset).limit(limit).to_list(None)
+        res = await self.collection.find(filter, args).sort("created_at", -1).skip(offset).limit(limit).to_list(None)
         assets = []
         for doc in res:
             doc["id"] = str(doc.pop("_id"))

repos/generation_repo.py

@@ -39,5 +39,13 @@ class GenerationRepo:
             generations.append(Generation(**generation))
         return generations
 
+    async def count_generations(self, character_id: Optional[str] = None, status: Optional[GenerationStatus] = None) -> int:
+        args = {}
+        if character_id is not None:
+            args["linked_character_id"] = character_id
+        if status is not None:
+            args["status"] = status
+        return await self.collection.count_documents(args)
+
     async def update_generation(self, generation: Generation, ):
         res = await self.collection.update_one({"_id": ObjectId(generation.id)}, {"$set": generation.model_dump()})