feat(FR-404): show or hide model folder related info

lablup · Jan 20, 2025 · 8894192 · 8894192
1 parent 5e36c40
commit 8894192
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 2 deletions.
diff --git a/config.toml.sample b/config.toml.sample
@@ -34,6 +34,7 @@ enableModelStore = false                # Enable model store feature. (From Back
 enableLLMPlayground = false             # Enable LLM Playground feature. (From Backend.AI 24.03)
 enableExtendLoginSession = false        # If true, enables login session extension UI. (From Backend.AI 24.09)
 enableImportFromHuggingFace = false     # Enable import from Hugging Face feature. (From Backend.AI 24.09)
+enableModelFolders = true               # Enable model folders feature. (From Backend.AI 23.03)
 
 [wsproxy]
 proxyURL = "[Proxy URL]"

diff --git a/react/src/components/FolderCreateModal.tsx b/react/src/components/FolderCreateModal.tsx
@@ -227,7 +227,9 @@ const FolderCreateModal: React.FC<FolderCreateModalProps> = ({
         <Form.Item label={t('data.UsageMode')} name={'usage_mode'}>
           <Radio.Group>
             <Radio value={'general'}>General</Radio>
-            <Radio value={'model'}>Model</Radio>
+            {baiClient._config.enableModelFolders ? (
+              <Radio value={'model'}>Model</Radio>
+            ) : null}
           </Radio.Group>
         </Form.Item>
         <Divider />

diff --git a/react/src/hooks/index.tsx b/react/src/hooks/index.tsx
@@ -432,6 +432,7 @@ type BackendAIConfig = {
   enableLLMPlayground: boolean;
   enableImportFromHuggingFace: boolean;
   enableContainerCommit: boolean;
+  enableModelFolders: boolean;
   appDownloadUrl: string;
   systemSSHImage: string;
   fasttrackEndpoint: string;

diff --git a/react/src/pages/VFolderListPage.tsx b/react/src/pages/VFolderListPage.tsx
@@ -99,7 +99,7 @@ const VFolderListPage: React.FC<VFolderListPageProps> = (props) => {
       key: 'automount',
       tab: t('data.AutomountFolders'),
     },
-    {
+    baiClient._config.enableModelFolders && {
       key: 'model',
       tab: t('data.Models'),
     },

diff --git a/src/components/backend-ai-login.ts b/src/components/backend-ai-login.ts
@@ -144,6 +144,7 @@ export default class BackendAILogin extends BackendAIPage {
   @property({ type: Boolean }) enableLLMPlayground = false;
   @property({ type: Boolean }) enableImportFromHuggingFace = false;
   @property({ type: Boolean }) enableExtendLoginSession = false;
+  @property({ type: Boolean }) enableModelFolders = true;
   @property({ type: Boolean }) showNonInstalledImages = false;
   @property({ type: String }) eduAppNamePrefix;
   @property({ type: String }) pluginPages;
@@ -880,6 +881,13 @@ export default class BackendAILogin extends BackendAIPage {
         value: generalConfig?.enableExtendLoginSession,
       } as ConfigValueObject,
     ) as boolean;
+
+    // Enable model folders support
+    this.enableModelFolders = this._getConfigValueByExists(generalConfig, {
+      valueType: 'boolean',
+      defaultValue: true,
+      value: generalConfig?.enableModelFolders,
+    } as ConfigValueObject) as boolean;
   }
 
   /**
@@ -1916,6 +1924,8 @@ export default class BackendAILogin extends BackendAIPage {
           this.enableImportFromHuggingFace;
         globalThis.backendaiclient._config.enableExtendLoginSession =
           this.enableExtendLoginSession;
+        globalThis.backendaiclient._config.enableModelFolders =
+          this.enableModelFolders;
         globalThis.backendaiclient._config.pluginPages = this.pluginPages;
         globalThis.backendaiclient._config.blockList = this.blockList;
         globalThis.backendaiclient._config.inactiveList = this.inactiveList;