diff --git a/react/src/components/ResourceAllocationFormItems.tsx b/react/src/components/ResourceAllocationFormItems.tsx index ccdd2293b..6dfe26785 100644 --- 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< ) : null} {/* TODO: Support cluster mode */} - {!baiClient._config.hideAgents && ( + {!baiClient._config.hideAgents && enableAgentSelect && (