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

@@ -14,6 +14,7 @@ class GenerationRequest(BaseModel):
quality: Quality = Quality.ONEK
prompt: str
telegram_id: Optional[int] = None
use_profile_image: bool = True
assets_list: List[str]
@@ -32,6 +33,8 @@ class GenerationResponse(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
progress: int = 0
created_at: datetime = datetime.now(UTC)
updated_at: datetime = datetime.now(UTC)