This commit is contained in:
2026-05-25 17:08:18 +08:00
parent df501f6151
commit f1259b71b5
9178 changed files with 1626125 additions and 0 deletions
+48
View File
@@ -0,0 +1,48 @@
import json
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from fastapi import APIRouter, Depends
from app.dependencies import get_db, get_current_user
from app.models.user import User
from app.models.image_engine import ImageEngine
from app.schemas.image_engine import ImageEngineListResponse
router = APIRouter(prefix="/image-engines", tags=["image-engines"])
@router.get("", response_model=ImageEngineListResponse)
async def list_active_engines(
current_user: User = Depends(get_current_user),
db: AsyncSession = Depends(get_db),
):
"""Public endpoint returning active image engine capabilities."""
result = await db.execute(
select(ImageEngine)
.where(ImageEngine.is_active == True)
.order_by(ImageEngine.priority.desc())
)
engines = result.scalars().all()
items = []
for e in engines:
models = []
sizes = {}
try:
models = json.loads(e.supported_models) if e.supported_models else []
except Exception:
pass
try:
sizes = json.loads(e.supported_sizes) if e.supported_sizes else {}
except Exception:
pass
items.append({
"id": e.id,
"name": e.name,
"provider": e.provider,
"supported_models": models,
"supported_sizes": sizes,
"default_size": e.default_size,
})
return {"items": items}