From f6001f5994ed75eef235f9e5eeda23730664841b Mon Sep 17 00:00:00 2001 From: xds Date: Fri, 27 Feb 2026 13:51:22 +0300 Subject: [PATCH] nsfw mark api --- aiws.py | 2 +- api/endpoints/generation_router.py | 4 +--- api/models/__init__.py | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) 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