update get sku types API

src/api/v2/clients/clusterClient.ts

@@ -0,0 +1,26 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { Util } from '@pai/commom/util';
+
+import { ISkuType } from '../models/virtualCluster';
+
+import { OpenPAIBaseClient } from './baseClient';
+
+/**
+ * OpenPAI Cluster client.
+ */
+export class ClusterClient extends OpenPAIBaseClient {
+
+    /**
+     * Get sku types.
+     * @param vcName The name of virtual cluster.
+     */
+    public async getSkuTypes(vcName?: string): Promise<{ [id: string]: ISkuType; }> {
+        const url: string = Util.fixUrl(
+            `${this.cluster.rest_server_uri}/api/v2/cluster/sku-types`,
+            this.cluster.https
+        );
+        return await this.httpClient.get(url, undefined, undefined, { vc: vcName });
+    }
+}

src/api/v2/clients/virtualClusterClient.ts

@@ -4,8 +4,6 @@
 import { IPAICluster, IVirtualCluster } from '@api/v2';
 import { Util } from '@pai/commom/util';
 
-import { ISkuType } from '../models/virtualCluster';
-
 import { OpenPAIBaseClient } from './baseClient';
 
 /**
@@ -38,16 +36,4 @@ export class VirtualClusterClient extends OpenPAIBaseClient {
         );
         return await this.httpClient.get(url);
     }
-
-    /**
-     * Get sku types in the virtual cluster.
-     * @param vcName The name of virtual cluster.
-     */
-    public async getVirtualClusterSkuTypes(vcName: string): Promise<{ [id: string]: ISkuType; }> {
-        const url: string = Util.fixUrl(
-            `${this.cluster.rest_server_uri}/api/v2/virtual-clusters/${vcName}/sku-types`,
-            this.cluster.https
-        );
-        return await this.httpClient.get(url);
-    }
 }

src/api/v2/index.ts

@@ -15,6 +15,7 @@ import {
     UserClient,
     VirtualClusterClient
 } from './clients';
+import { ClusterClient } from './clients/clusterClient';
 import { GroupClient } from './clients/groupClient';
 import { StorageNodeV2 as StorageNode } from './clients/storageClient';
 import { IAuthnInfo, ILoginInfo } from './models/authn';
@@ -32,6 +33,7 @@ import { INodeResource, IVirtualCluster } from './models/virtualCluster';
  */
 export {
     AuthnClient,
+    ClusterClient,
     JobClient,
     OpenPAIClient,
     OpenPAIBaseClient,