feat: model filter frontend integration

2025-06-26 18:26:48 +00:00 · 2024-03-09 21:29:04 -08:00
parent b550e23bf6
commit 81daf4ceb9
4 changed files with 83 additions and 6 deletions
--- a/backend/apps/openai/main.py
+++ b/backend/apps/openai/main.py
@@ -196,9 +196,11 @@ async def get_models(url_idx: Optional[int] = None, user=Depends(get_current_use
        models = await get_all_models()
        if app.state.MODEL_FILTER_ENABLED:
            if user.role == "user":
-                models["data"] = filter(
-                    lambda model: model["id"] in app.state.MODEL_LIST,
-                    models["data"],
+                models["data"] = list(
+                    filter(
+                        lambda model: model["id"] in app.state.MODEL_LIST,
+                        models["data"],
+                    )
                )
                return models
        return models