diff --git a/api/models/GenerationRequest.py b/api/models/GenerationRequest.py index 02e6436..b9f1469 100644 --- a/api/models/GenerationRequest.py +++ b/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 diff --git a/api/models/__pycache__/GenerationRequest.cpython-313.pyc b/api/models/__pycache__/GenerationRequest.cpython-313.pyc index dce7843..76e1265 100644 Binary files a/api/models/__pycache__/GenerationRequest.cpython-313.pyc and b/api/models/__pycache__/GenerationRequest.cpython-313.pyc differ diff --git a/api/service/__pycache__/generation_service.cpython-313.pyc b/api/service/__pycache__/generation_service.cpython-313.pyc index b67fb05..ffbbcaa 100644 Binary files a/api/service/__pycache__/generation_service.cpython-313.pyc and b/api/service/__pycache__/generation_service.cpython-313.pyc differ diff --git a/api/service/generation_service.py b/api/service/generation_service.py index 93400d5..75ab456 100644 --- a/api/service/generation_service.py +++ b/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) diff --git a/models/Generation.py b/models/Generation.py index 2cb018b..29e9717 100644 --- a/models/Generation.py +++ b/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 diff --git a/models/__pycache__/Generation.cpython-313.pyc b/models/__pycache__/Generation.cpython-313.pyc index cc4f421..0308aa2 100644 Binary files a/models/__pycache__/Generation.cpython-313.pyc and b/models/__pycache__/Generation.cpython-313.pyc differ