refactor: add worker

2025-11-15 22:37:44 +08:00
parent 864cf76c07
commit 0f3f31ef45
39 changed files with 1746 additions and 606 deletions
@@ -0,0 +1,56 @@
+export const StepNamesMap = {
+  SelectCluster: 'SelectCluster',
+  SelectGPU: 'SelectGPU',
+  CheckEnv: 'CheckEnv',
+  SpecifyArgs: 'SpecifyArgs',
+  RunCommand: 'RunCommand'
+} as const;
+
+export type StepName = keyof typeof StepNamesMap;
+
+export const DockerStepsFromCluster = [
+  StepNamesMap.SelectGPU,
+  StepNamesMap.CheckEnv,
+  StepNamesMap.SpecifyArgs,
+  StepNamesMap.RunCommand
+];
+
+export const DockerStepsFromWorker = [
+  StepNamesMap.SelectCluster,
+  StepNamesMap.SelectGPU,
+  StepNamesMap.CheckEnv,
+  StepNamesMap.SpecifyArgs,
+  StepNamesMap.RunCommand
+];
+
+export const K8sStepsFromCluter = [
+  StepNamesMap.SelectGPU,
+  StepNamesMap.CheckEnv,
+  StepNamesMap.RunCommand
+];
+
+export interface SummaryDataKeys {
+  currentGPU: string;
+  cluster_id: number;
+  clusterName: string;
+  workerCommand: {
+    label: string;
+    link: string;
+    notes: string[];
+  };
+  modelDirConfig: {
+    enabled: boolean;
+    path: string;
+  };
+  workerIPConfig: {
+    enabled: boolean;
+    ip: string;
+    required: boolean;
+  };
+}
+
+export type SummaryDataMap = {
+  [key in keyof SummaryDataKeys]: SummaryDataKeys[key];
+};
+
+export type SummaryDataKey = keyof SummaryDataKeys;