From 7cd9b294c350e67092ebb5ff7c2d57f953463ea4 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Wed, 5 Jun 2024 15:18:54 -0400 Subject: [PATCH] fix another type ignore --- backend/image_generation.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/backend/image_generation.py b/backend/image_generation.py index c99dd40..e3f609f 100644 --- a/backend/image_generation.py +++ b/backend/image_generation.py @@ -24,15 +24,14 @@ async def generate_image( prompt: str, api_key: str, base_url: str | None ) -> Union[str, None]: client = AsyncOpenAI(api_key=api_key, base_url=base_url) - image_params: Dict[str, Union[str, int]] = { - "model": "dall-e-3", - "quality": "standard", - "style": "natural", - "n": 1, - "size": "1024x1024", - "prompt": prompt, - } - res = await client.images.generate(**image_params) # type: ignore + res = await client.images.generate( + model="dall-e-3", + quality="standard", + style="natural", + n=1, + size="1024x1024", + prompt=prompt, + ) await client.close() return res.data[0].url