From 3e7077bb736745e952be6453c73099bfb726f042 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 3 Mar 2025 22:24:19 -0800 Subject: [PATCH] refac --- backend/open_webui/main.py | 11 ++++++++++- src/lib/components/chat/ModelSelector/Selector.svelte | 4 ++-- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py index d7c8df7f4..9c31b84f9 100644 --- a/backend/open_webui/main.py +++ b/backend/open_webui/main.py @@ -1142,9 +1142,10 @@ async def get_app_config(request: Request): if data is not None and "id" in data: user = Users.get_user_by_id(data["id"]) + user_count = Users.get_num_users() onboarding = False + if user is None: - user_count = Users.get_num_users() onboarding = user_count == 0 return { @@ -1213,6 +1214,14 @@ async def get_app_config(request: Request): "api_key": GOOGLE_DRIVE_API_KEY.value, }, "onedrive": {"client_id": ONEDRIVE_CLIENT_ID.value}, + **( + { + "record_count": user_count, + "active_entries": app.state.USER_COUNT, + } + if user.role == "admin" + else {} + ), } if user is not None else {} diff --git a/src/lib/components/chat/ModelSelector/Selector.svelte b/src/lib/components/chat/ModelSelector/Selector.svelte index 8a126d85a..ed7a2f419 100644 --- a/src/lib/components/chat/ModelSelector/Selector.svelte +++ b/src/lib/components/chat/ModelSelector/Selector.svelte @@ -334,7 +334,7 @@