inspirations

api/models/GenerationRequest.py

@@ -19,6 +19,7 @@ class GenerationRequest(BaseModel):
     environment_id: Optional[str] = None
     project_id: Optional[str] = None
     idea_id: Optional[str] = None
+    nsfw: bool = False
     count: int = Field(default=1, ge=1, le=10)
 
 
@@ -52,6 +53,7 @@ class GenerationResponse(BaseModel):
     idea_id: Optional[str] = None
     likes_count: int = 0
     is_liked: bool = False
+    nsfw: bool = False
     created_at: datetime = datetime.now(UTC)
     updated_at: datetime = datetime.now(UTC)
 

models/Generation.py

@@ -41,6 +41,7 @@ class Generation(BaseModel):
     project_id: Optional[str] = None
     idea_id: Optional[str] = None
     liked_by: List[str] = Field(default_factory=list)
+    nsfw: bool = False
     created_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
     updated_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
 

repos/generation_repo.py

@@ -290,7 +290,6 @@ class GenerationRepo:
         cursor = self.collection.find(filter_query, {"result_list": 1, "assets_list": 1})
         async for doc in cursor:
             asset_ids.extend(doc.get("result_list", []))
-            asset_ids.extend(doc.get("assets_list", []))
 
         # Мягкое удаление
         res = await self.collection.update_many(