feat: model filter frontend integration

2025-12-16 11:57:51 +01: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/ollama/main.py
+++ b/backend/apps/ollama/main.py
@@ -135,11 +135,14 @@ async def get_ollama_tags(
 ):
    if url_idx == None:
        models = await get_all_models()
+
        if app.state.MODEL_FILTER_ENABLED:
            if user.role == "user":
-                models["models"] = filter(
-                    lambda model: model["name"] in app.state.MODEL_LIST,
-                    models["models"],
+                models["models"] = list(
+                    filter(
+                        lambda model: model["name"] in app.state.MODEL_LIST,
+                        models["models"],
+                    )
                )
                return models
        return models