feat: Add use_profile_image and detailed token usage fields to generation models.
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user