nsfw mark api

aiws.py

@@ -134,7 +134,7 @@ async def start_scheduler(service: GenerationService):
         try:
             logger.info("Running scheduler for stacked generation killing")
             await service.cleanup_stale_generations()
-            await service.cleanup_old_data(days=5)
+            await service.cleanup_old_data(days=14)
         except asyncio.CancelledError:
              break
         except Exception as e:

api/endpoints/generation_router.py

@@ -214,9 +214,7 @@ async def mark_generation_nsfw(
         if not is_member:
             raise HTTPException(status_code=403, detail="Access denied")
 
-    if not await generation_service.dao.generations.mark_nsfw(generation_id, request.is_nsfw):
-        raise HTTPException(status_code=404, detail="Generation not found or already in the requested state")
-    
+    await generation_service.dao.generations.mark_nsfw(generation_id, request.is_nsfw)
     return None
 
 

api/models/__init__.py

@@ -2,6 +2,6 @@ from .AssetDTO import AssetResponse, AssetsResponse
 from .CharacterDTO import CharacterCreateRequest, CharacterUpdateRequest
 from .ExternalGenerationDTO import ExternalGenerationRequest
 from .FinancialUsageDTO import FinancialReport, UsageStats, UsageByEntity
-from .GenerationRequest import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse, PromptRequest, PromptResponse
+from .GenerationRequest import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse, PromptRequest, PromptResponse, NsfwRequest
 from .IdeaRequest import IdeaCreateRequest, IdeaUpdateRequest, IdeaResponse
 from .PostRequest import PostCreateRequest, PostUpdateRequest, AddGenerationsRequest