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
@@ -1,8 +1,9 @@
 import HighlightCode from '@/components/highlight-code';
 import React, { useMemo } from 'react';
-import { generateRegisterCommand } from '../config';
+import { generateK8sRegisterCommand } from '../config';

 type AddModalProps = {
+  currentGPU?: string;
  registrationInfo: {
    token: string;
    image: string;
@@ -10,14 +11,18 @@ type AddModalProps = {
    cluster_id: number | null;
  };
 };
-const AddCluster: React.FC<AddModalProps> = ({ registrationInfo }) => {
+const AddCluster: React.FC<AddModalProps> = ({
+  registrationInfo,
+  currentGPU
+}) => {
  const code = useMemo(() => {
-    return generateRegisterCommand({
+    return generateK8sRegisterCommand({
      server: registrationInfo?.server_url || window.location.origin,
      clusterId: registrationInfo?.cluster_id,
-      registrationToken: registrationInfo?.token
+      registrationToken: registrationInfo?.token,
+      currentGPU
    });
-  }, [registrationInfo]);
+  }, [registrationInfo, currentGPU]);

  return (
    <div>