feat: Add use_profile_image and detailed token usage fields to generation models.

This commit is contained in:
xds
2026-02-05 22:19:57 +03:00
parent 002c949f08
commit c0debab0cb
8 changed files with 21 additions and 3 deletions

View File

@@ -19,6 +19,7 @@ class Generation(BaseModel):
failed_reason: Optional[str] = None
linked_character_id: Optional[str] = None
telegram_id: Optional[int] = None
use_profile_image: bool = True
aspect_ratio: AspectRatios
quality: Quality
prompt: str
@@ -29,6 +30,8 @@ class Generation(BaseModel):
execution_time_seconds: Optional[float] = None
api_execution_time_seconds: Optional[float] = None
token_usage: Optional[int] = None
input_token_usage: Optional[int] = None
output_token_usage: Optional[int] = None
created_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
updated_at: datetime = Field(default_factory=lambda: datetime.now(UTC))