push

api/models/GenerationRequest.py

@@ -29,6 +29,7 @@ class GenerationResponse(BaseModel):
     prompt: str
     tech_prompt: Optional[str] = None
     assets_list: List[str]
+    result_list: List[str] = []
     result: Optional[str] = None
     execution_time_seconds: Optional[float] = None
     api_execution_time_seconds: Optional[float] = None

api/service/generation_service.py

@@ -254,7 +254,7 @@ class GenerationService:
         # Предполагаем, что у модели Generation есть поле result_asset_ids
         result_ids = [a.id for a in created_assets]
 
-        generation.assets_list = result_ids
+        generation.result_list = result_ids
         generation.status = GenerationStatus.DONE
         generation.progress = 100
         generation.updated_at = datetime.now(UTC)

models/Generation.py

@@ -25,6 +25,7 @@ class Generation(BaseModel):
     prompt: str
     tech_prompt: Optional[str] = None
     assets_list: List[str]
+    result_list: List[str] = Field(default_factory=list)
     result: Optional[str] = None
     progress: int = 0
     execution_time_seconds: Optional[float] = None