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

api/endpoints/assets_router.py

@@ -66,9 +66,9 @@ async def delete_asset(
 
 
 @router.get("")
-async def get_assets(request: Request, dao: DAO = Depends(get_dao), limit: int = 10, offset: int = 0) -> AssetsResponse:
+async def get_assets(request: Request, dao: DAO = Depends(get_dao), type: Optional[str] = None, limit: int = 10, offset: int = 0) -> AssetsResponse:
     logger.info(f"get_assets called. Limit: {limit}, Offset: {offset}")
-    assets = await dao.assets.get_assets(limit, offset)
+    assets = await dao.assets.get_assets(type, limit, offset)
     # assets = await dao.assets.get_assets() # This line seemed redundant/conflicting in original code
     total_count = await dao.assets.get_asset_count()
 

api/endpoints/generation_router.py

@@ -7,7 +7,7 @@ from starlette.requests import Request
 from api import service
 from api.dependency import get_generation_service
 
-from api.models.GenerationRequest import GenerationResponse, GenerationRequest, PromptResponse, PromptRequest
+from api.models.GenerationRequest import GenerationResponse, GenerationRequest,  GenerationsResponse, PromptResponse, PromptRequest
 from api.service.generation_service import GenerationService
 from models.Generation import Generation
 
@@ -43,7 +43,7 @@ async def prompt_from_image(
     return PromptResponse(prompt=generated_prompt)
 
 
-@router.get("", response_model=List[GenerationResponse])
+@router.get("", response_model=GenerationsResponse)
 async def get_generations(character_id: Optional[str] = None, limit: int = 10, offset: int = 0,
                           generation_service: GenerationService = Depends(get_generation_service)):
     logger.info(f"get_generations called. CharacterId: {character_id}, Limit: {limit}, Offset: {offset}")