From c8a3c42ea8a3b341e3d6e0688d0a288730a84f38 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 16 Nov 2024 15:13:33 -0800 Subject: [PATCH] refac: filter out arena models from base models --- backend/open_webui/main.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py index 90c12971d..3aa272699 100644 --- a/backend/open_webui/main.py +++ b/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}