inspirations

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
+        return 0.0