From e976fe1c58d402b5ed960d54cf7fc60502d9d115 Mon Sep 17 00:00:00 2001
From: xds <iam@voroninv.ru>
Date: Thu, 26 Feb 2026 11:26:18 +0300
Subject: [PATCH] inspirations

---
 api/models/GenerationRequest.py | 2 ++
 models/Generation.py            | 3 ++-
 repos/generation_repo.py        | 1 -
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/api/models/GenerationRequest.py b/api/models/GenerationRequest.py
index d05f2b1..2019720 100644
--- a/api/models/GenerationRequest.py
+++ b/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)
 
diff --git a/models/Generation.py b/models/Generation.py
index b6331a0..e3d2b70 100644
--- a/models/Generation.py
+++ b/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))
 
@@ -50,4 +51,4 @@ class Generation(BaseModel):
             cost_input = self.input_token_usage * 0.000002
             cost_output = self.output_token_usage * 0.00012
             return round(cost_input + cost_output, 3)
-        return 0.0
\ No newline at end of file
+        return 0.0
diff --git a/repos/generation_repo.py b/repos/generation_repo.py
index a4ec38a..753f99c 100644
--- a/repos/generation_repo.py
+++ b/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(