diff --git a/aiws.py b/aiws.py index 55e3ad0..5bb94a2 100644 --- a/aiws.py +++ b/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: diff --git a/api/endpoints/generation_router.py b/api/endpoints/generation_router.py index 3817647..641ef70 100644 --- a/api/endpoints/generation_router.py +++ b/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 diff --git a/api/models/__init__.py b/api/models/__init__.py index d591bc8..ac8f3fb 100644 --- a/api/models/__init__.py +++ b/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