feat(BA-543): Add enable_model_folders option to webserver to contr…

…ol Models visibility in WebUI
lablup · Jan 20, 2025 · 4e419f6 · 4e419f6
1 parent 166f2bd
commit 4e419f6
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/configs/webserver/sample.conf b/configs/webserver/sample.conf
@@ -124,6 +124,8 @@ menu_blocklist = "pipeline"
 #menu_inactivelist = "statistics"
 # Enable/disable the LLM Playground tab in the service page
 # enable_LLM_playground = false
+# Enable/disable the Models tab and MODEL usage mode in the data & storage page
+# enable_model_folders = true
 
 [api]
 domain = "default"

diff --git a/src/ai/backend/web/config.py b/src/ai/backend/web/config.py
@@ -66,6 +66,7 @@
         t.Key("edu_appname_prefix", default=""): t.String(allow_blank=True),
         t.Key("enable_model_store", default=True): t.ToBool(),
         t.Key("enable_extend_login_session", default=False): t.ToBool(),
+        t.Key("enable_model_folders", default=True): t.ToBool(),
     }).allow_extra("*"),
     t.Key("resources"): t.Dict({
         t.Key("open_port_to_public", default=False): t.ToBool,

diff --git a/src/ai/backend/web/templates/config.toml.j2 b/src/ai/backend/web/templates/config.toml.j2
@@ -5,6 +5,7 @@
 {% toml_field "defaultImportEnvironment" config["ui"]["default_import_environment"] %}
 {% toml_field "siteDescription" config["ui"]["brand"] %}
 {% toml_field "enableLLMPlayground" config["ui"]["enable_LLM_playground"] %}
+{% toml_field "enableModelFolders" config["ui"]["enable_model_folders"] %}
 connectionMode = "SESSION"
 {% toml_field "signupSupport" config["service"]["enable_signup"] %}
 {% toml_field "allowChangeSigninMode" config["service"]["allow_change_signin_mode"] %}

diff --git a/src/ai/backend/web/templates/config_ini.toml.j2 b/src/ai/backend/web/templates/config_ini.toml.j2
@@ -4,6 +4,7 @@
 {% toml_field "defaultSessionEnvironment" config["ui"]["default_environment"] %}
 {% toml_field "siteDescription" config["ui"]["brand"] %}
 {% toml_field "enableLLMPlayground" config["ui"]["enable_LLM_playground"] %}
+{% toml_field "enableModelFolders" config["ui"]["enable_model_folders"] %}
 connectionMode = "SESSION"
 
 [environments]