Browse Source

refac: filter out arena models from base models

Timothy Jaeryang Baek 5 months ago
parent
commit
c8a3c42ea8
1 changed files with 3 additions and 0 deletions
  1. 3 0
      backend/open_webui/main.py

+ 3 - 0
backend/open_webui/main.py

@@ -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}