some minor cleanup, and coalesce image/images into one thing, and use itertools.cycle

2024-01-27 19:07:29 -05:00
parent 86af896150
commit 4416f36ae9
8 changed files with 62 additions and 50 deletions
--- a/src/dailyai/services/fal_ai_services.py
+++ b/src/dailyai/services/fal_ai_services.py
@@ -11,7 +11,7 @@ from dailyai.services.ai_services import LLMService, TTSService, ImageGenService


 class FalImageGenService(ImageGenService):
-    def __init__(self, image_size, aiohttp_session:aiohttp.ClientSession):
+    def __init__(self, image_size, aiohttp_session: aiohttp.ClientSession):
        super().__init__(image_size)
        self._aiohttp_session = aiohttp_session