refac: filter out arena models from base models

This commit is contained in:
Timothy Jaeryang Baek 2024-11-16 15:13:33 -08:00
parent 345c9d12a9
commit c8a3c42ea8

View File

@ -1120,6 +1120,9 @@ async def get_models(user=Depends(get_verified_user)):
@app.get("/api/models/base")
async def get_base_models(user=Depends(get_admin_user)):
models = await get_all_base_models()
# Filter out arena models
models = [model for model in models if not model.get("arena", False)]
return {"data": models}