models + refactor

scheduler/daily_scheduler.py

@@ -7,7 +7,7 @@ from aiogram import Bot
 from aiogram.types import BufferedInputFile, InlineKeyboardButton, InlineKeyboardMarkup
 
 from adapters.google_adapter import GoogleAdapter
-from adapters.ai_proxy_adapter import AIProxyAdapter
+from adapters.ai_proxy_adapter import AIProxyAdapter, AIProxyException
 from adapters.s3_adapter import S3Adapter
 from api.service.generation_service import GenerationService
 from models.Asset import Asset
@@ -272,11 +272,11 @@ class DailyScheduler:
         try:
             # Delegate all heavy lifting to the existing service
             await self.generation_service.create_generation(generation)
-        except Exception:
-            # create_generation doesn't mark FAILED itself — the caller (_queued_generation_runner) does.
-            # So we need to handle failure here.
-            await self.generation_service._handle_generation_failure(generation, Exception("Image generation failed"))
-            raise
+        except AIProxyException as e:
+            # error code is already saved by create_generation
+            raise ValueError(f"AI Proxy Error: {e.error_code or str(e)}")
+        except Exception as e:
+            raise ValueError(f"Image generation failed: {str(e)}")
 
         # After create_generation, generation.result_list is populated
         if not generation.result_list: