chore: add runtime in register k8s command

2026-02-09 15:58:29 +08:00
parent e7890dda3d
commit 198a52e230
4 changed files with 31 additions and 14 deletions
@@ -7,10 +7,11 @@ import { Title } from './constainers';
 import StepCollapse from './step-collapse';

 const K8sRunCommand: React.FC<AddWorkerStepProps> = ({ disabled }) => {
-  const { registrationInfo, stepList } = useAddWorkerContext();
+  const { registrationInfo, stepList, summary } = useAddWorkerContext();
  const intl = useIntl();

  const stepIndex = stepList.indexOf(StepNamesMap.RunCommand) + 1;
+  const currentGPU = summary.get('currentGPU') || '';

  return (
    <StepCollapse
@@ -32,7 +33,10 @@ const K8sRunCommand: React.FC<AddWorkerStepProps> = ({ disabled }) => {
          id: 'clusters.create.addCommand.tips'
        })}
      </Typography.Paragraph>
-      <RegisterClusterInner registrationInfo={registrationInfo} />
+      <RegisterClusterInner
+        registrationInfo={registrationInfo}
+        currentGPU={currentGPU}
+      />
    </StepCollapse>
  );
 };