Update main.py
Browse files
main.py
CHANGED
|
@@ -30,7 +30,8 @@ class ImageGenerationRequest(BaseModel):
|
|
| 30 |
quality: Optional[Literal["standard", "hd"]] = Field("standard", description="Quality of the image")
|
| 31 |
response_format: Optional[Literal["url", "b64_json"]] = Field("url", description="Response format")
|
| 32 |
size: Optional[Literal["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"]] = Field("1024x1024", description="Size of the generated images")
|
| 33 |
-
style: Optional[Literal["vivid", "natural"]] = Field("vivid", description="Style of the generated images")
|
|
|
|
| 34 |
|
| 35 |
# OpenAI-compatible response models
|
| 36 |
class ImageData(BaseModel):
|
|
|
|
| 30 |
quality: Optional[Literal["standard", "hd"]] = Field("standard", description="Quality of the image")
|
| 31 |
response_format: Optional[Literal["url", "b64_json"]] = Field("url", description="Response format")
|
| 32 |
size: Optional[Literal["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"]] = Field("1024x1024", description="Size of the generated images")
|
| 33 |
+
style: Optional[Literal["vivid", "natural"]] = Field("vivid", description="Style of the generated images")
|
| 34 |
+
user: Optional[str] = Field(None, description="A unique identifier representing your end-user")
|
| 35 |
|
| 36 |
# OpenAI-compatible response models
|
| 37 |
class ImageData(BaseModel):
|