add: option to show/hide agent select option by launcher

lablup · Nov 21, 2024 · 916a5df · 916a5df
1 parent ae54e8a
commit 916a5df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/react/src/components/ResourceAllocationFormItems.tsx b/react/src/components/ResourceAllocationFormItems.tsx
@@ -84,6 +84,7 @@ type MergedResourceAllocationFormValue = ResourceAllocationFormValue &
   ImageEnvironmentFormInput;
 
 interface ResourceAllocationFormItemsProps {
+  enableAgentSelect?: boolean;
   enableNumOfSessions?: boolean;
   enableResourcePresets?: boolean;
   showRemainingWarning?: boolean;
@@ -93,6 +94,7 @@ interface ResourceAllocationFormItemsProps {
 const ResourceAllocationFormItems: React.FC<
   ResourceAllocationFormItemsProps
 > = ({
+  enableAgentSelect = false,
   enableNumOfSessions,
   enableResourcePresets,
   forceImageMinValues = false,
@@ -1206,7 +1208,7 @@ const ResourceAllocationFormItems: React.FC<
         </Card>
       ) : null}
       {/* TODO: Support cluster mode */}
-      {!baiClient._config.hideAgents && (
+      {!baiClient._config.hideAgents && enableAgentSelect && (
         <Form.Item
           label={t('session.launcher.SelectAgent')}
           required

diff --git a/react/src/pages/SessionLauncherPage.tsx b/react/src/pages/SessionLauncherPage.tsx
@@ -1233,6 +1233,7 @@ const SessionLauncherPage = () => {
                   }}
                 >
                   <ResourceAllocationFormItems
+                    enableAgentSelect
                     enableNumOfSessions
                     enableResourcePresets
                     showRemainingWarning