+ ideas

api/models/GenerationRequest.py

@@ -17,6 +17,7 @@ class GenerationRequest(BaseModel):
     use_profile_image: bool = True
     assets_list: List[str]
     project_id: Optional[str] = None
+    idea_id: Optional[str] = None
     count: int = Field(default=1, ge=1, le=10)
 
 
@@ -47,6 +48,7 @@ class GenerationResponse(BaseModel):
     cost: Optional[float] = None
     created_by: Optional[str] = None
     generation_group_id: Optional[str] = None
+    idea_id: Optional[str] = None
     created_at: datetime = datetime.now(UTC)
     updated_at: datetime = datetime.now(UTC)