diff --git a/models/Asset.py b/models/Asset.py index 0333495..79730a2 100644 --- a/models/Asset.py +++ b/models/Asset.py @@ -2,7 +2,7 @@ from datetime import datetime, UTC from enum import Enum from typing import Optional, Any, List -from pydantic import BaseModel, computed_field +from pydantic import BaseModel, computed_field, Field class AssetType(str, Enum): @@ -19,7 +19,8 @@ class Asset(BaseModel): tg_doc_file_id: Optional[str] = None tg_photo_file_id: Optional[str] = None tags: List[str] = [] - created_at: datetime = datetime.now(UTC) + created_at: datetime = Field(default_factory=lambda: datetime.now(UTC)) + updated_at: datetime = Field(default_factory=lambda: datetime.now(UTC)) # --- CALCULATED FIELD --- @computed_field diff --git a/models/Generation.py b/models/Generation.py index d875b32..e346115 100644 --- a/models/Generation.py +++ b/models/Generation.py @@ -2,7 +2,7 @@ from datetime import datetime, UTC from enum import Enum from typing import List, Optional -from pydantic import BaseModel +from pydantic import BaseModel, Field from models.Asset import Asset from models.enums import AspectRatios, Quality @@ -23,5 +23,5 @@ class Generation(BaseModel): tech_prompt: Optional[str] = None assets_list: List[str] result: Optional[str] = None - created_at: datetime = datetime.now(UTC) - updated_at: datetime = datetime.now(UTC) + created_at: datetime = Field(default_factory=lambda: datetime.now(UTC)) + updated_at: datetime = Field(default_factory=lambda: datetime.now(UTC))