1

2026-07-01 11:32:39 +08:00
parent 0aa65c0723
commit 0d345761e8
1 changed files with 21 additions and 51 deletions
@@ -48,70 +48,40 @@ async def get_credit_ratios(
    current_user: User = Depends(get_current_user),
    db: AsyncSession = Depends(get_db),
 ):
-    import json
-
-    async def get_engine_with_ratios(gen_type: str, engines: list):
-        for engine in engines:
+    async def get_ratios_for_engine_type(gen_type: str, engine_ids: list):
+        for engine_id in engine_ids:
            result = await db.execute(
                select(CreditRatio)
                .where(CreditRatio.gen_type == gen_type)
-                .where(CreditRatio.model_config_id == engine.id)
+                .where(CreditRatio.model_config_id == engine_id)
            )
            ratios = result.scalars().all()
            if ratios:
-                ratios_out = [CreditRatioOut.model_validate(r) for r in ratios]
-                engine_info = {
-                    "id": engine.id,
-                    "name": engine.name,
-                    "provider": engine.provider,
-                    "ratios": ratios_out,
-                }
-                if gen_type == "video":
-                    try:
-                        supported_ratios = json.loads(engine.supported_ratios) if engine.supported_ratios else []
-                    except Exception:
-                        supported_ratios = []
-                    try:
-                        supported_resolutions = json.loads(engine.supported_resolutions) if engine.supported_resolutions else []
-                    except Exception:
-                        supported_resolutions = []
-                    try:
-                        supported_durations = json.loads(engine.supported_durations) if engine.supported_durations else []
-                    except Exception:
-                        supported_durations = []
-                    engine_info.update({
-                        "supported_ratios": supported_ratios,
-                        "supported_resolutions": supported_resolutions,
-                        "supported_durations": supported_durations,
-                        "max_duration": engine.max_duration,
-                        "max_image_count": engine.max_image_count,
-                        "max_video_count": engine.max_video_count,
-                    })
-                return engine_info
-        return None
-
+                return [CreditRatioOut.model_validate(r) for r in ratios]
+        return []
+    
    video_engines_result = await db.execute(
-        select(VideoEngine)
+        select(VideoEngine.id)
        .where(VideoEngine.is_active == True)
        .order_by(VideoEngine.priority.desc())
    )
-    video_engines = video_engines_result.scalars().all()
-
+    video_engine_ids = video_engines_result.scalars().all()
+    
    image_engines_result = await db.execute(
-        select(ImageEngine)
+        select(ImageEngine.id)
        .where(ImageEngine.is_active == True)
        .order_by(ImageEngine.priority.desc())
    )
-    image_engines = image_engines_result.scalars().all()
-
+    image_engine_ids = image_engines_result.scalars().all()
+    
    grouped = {}
-
-    video_data = await get_engine_with_ratios("video", video_engines)
-    if video_data:
-        grouped["video"] = video_data
-
-    image_data = await get_engine_with_ratios("image", image_engines)
-    if image_data:
-        grouped["image"] = image_data
-
+    
+    video_ratios = await get_ratios_for_engine_type("video", video_engine_ids)
+    if video_ratios:
+        grouped["video"] = video_ratios
+    
+    image_ratios = await get_ratios_for_engine_type("image", image_engine_ids)
+    if image_ratios:
+        grouped["image"] = image_ratios
+    
    return grouped