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

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

Hide file tree

Showing 5 changed files with 6 additions and 0 deletions.
diff --git a/changes/3503.feature.md b/changes/3503.feature.md
@@ -0,0 +1 @@
+Add `enable_model_folders` option to control the visibility of the Models section and MODEL type on the Data & Storage page in the Backend.AI Web UI.
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]
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Add `enable_model_folders` option to control the visibility of the Models section and MODEL type on the Data & Storage page in the Backend.AI Web UI.