nsfw mark api

This commit is contained in:
xds
2026-02-27 13:51:22 +03:00
parent e4a39e90c3
commit f6001f5994
3 changed files with 3 additions and 5 deletions

View File

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

View File

@@ -214,9 +214,7 @@ async def mark_generation_nsfw(
if not is_member: if not is_member:
raise HTTPException(status_code=403, detail="Access denied") raise HTTPException(status_code=403, detail="Access denied")
if not await generation_service.dao.generations.mark_nsfw(generation_id, request.is_nsfw): 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")
return None return None

View File

@@ -2,6 +2,6 @@ from .AssetDTO import AssetResponse, AssetsResponse
from .CharacterDTO import CharacterCreateRequest, CharacterUpdateRequest from .CharacterDTO import CharacterCreateRequest, CharacterUpdateRequest
from .ExternalGenerationDTO import ExternalGenerationRequest from .ExternalGenerationDTO import ExternalGenerationRequest
from .FinancialUsageDTO import FinancialReport, UsageStats, UsageByEntity 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 .IdeaRequest import IdeaCreateRequest, IdeaUpdateRequest, IdeaResponse
from .PostRequest import PostCreateRequest, PostUpdateRequest, AddGenerationsRequest from .PostRequest import PostCreateRequest, PostUpdateRequest, AddGenerationsRequest