@zilliz/milvus2-sdk-node 2.2.4 → 2.2.6

package/dist/milvus/Collection.js

@@ -61,30 +61,19 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
         if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
     }
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Collection = void 0;
-var protobufjs_1 = __importDefault(require("protobufjs"));
-var path_1 = __importDefault(require("path"));
 var utils_1 = require("../utils");
-var ErrorReason_1 = require("./const/ErrorReason");
-var Validate_1 = require("./utils/Validate");
-var Format_1 = require("./utils/Format");
-var Client_1 = require("./Client");
-var Milvus_1 = require("./const/Milvus");
-var types_1 = require("./types");
-var schemaPath = path_1.default.resolve(__dirname, '../proto/proto/schema.proto');
+var BaseClient_1 = require("./BaseClient");
+var _1 = require(".");
+var utils_2 = require("../utils");
 /**
  * @see [collection operation examples](https://github.com/milvus-io/milvus-sdk-node/blob/main/example/Collection.ts)
  */
 var Collection = /** @class */ (function (_super) {
     __extends(Collection, _super);
-    function Collection(client) {
-        var _this = _super.call(this, client) || this;
-        _this._protoRoot = protobufjs_1.default.loadSync(schemaPath);
-        return _this;
+    function Collection() {
+        return _super !== null && _super.apply(this, arguments) || this;
     }
     /**
      * Create a collection in Milvus.
@@ -107,7 +96,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.createCollection({
+     *  new milvusClient(MILUVS_ADDRESS).createCollection({
      *    collection_name: 'my_collection',
      *    fields: [
      *      {
@@ -131,33 +120,34 @@ var Collection = /** @class */ (function (_super) {
      */
     Collection.prototype.createCollection = function (data) {
         return __awaiter(this, void 0, void 0, function () {
-            var _a, fields, collection_name, description, _b, consistency_level, CollectionSchema, FieldSchema, payload, collectionParams, schemaBtyes, level, promise;
+            var _a, fields, collection_name, description, _b, consistency_level, CollectionSchema, FieldSchema, payload, collectionParams, schemaBytes, level, promise;
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
                         _a = data || {}, fields = _a.fields, collection_name = _a.collection_name, description = _a.description, _b = _a.consistency_level, consistency_level = _b === void 0 ? 'Bounded' : _b;
                         if (!fields || !fields.length || !collection_name) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PARAMS);
+                            throw new Error(_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PARAMS);
                         }
-                        (0, Validate_1.checkCollectionFields)(fields);
-                        CollectionSchema = this._protoRoot.lookupType('milvus.proto.schema.CollectionSchema');
-                        FieldSchema = this._protoRoot.lookupType('milvus.proto.schema.FieldSchema');
+                        (0, utils_1.checkCollectionFields)(fields);
+                        CollectionSchema = this.schemaProto.lookupType('milvus.proto.schema.CollectionSchema');
+                        FieldSchema = this.schemaProto.lookupType('milvus.proto.schema.FieldSchema');
                         payload = {
                             name: collection_name,
                             description: description || '',
                             fields: [],
                         };
                         data.fields.forEach(function (field) {
-                            var value = __assign(__assign({}, field), { typeParams: (0, Format_1.parseToKeyValue)(field.type_params), dataType: field.data_type, isPrimaryKey: field.is_primary_key });
+                            field = (0, utils_2.assignTypeParams)(field);
+                            var value = __assign(__assign({}, field), { typeParams: (0, utils_1.parseToKeyValue)(field.type_params), dataType: field.data_type, isPrimaryKey: field.is_primary_key });
                             var fieldParams = FieldSchema.create(value);
                             payload.fields.push(fieldParams);
                         });
                         collectionParams = CollectionSchema.create(payload);
-                        schemaBtyes = CollectionSchema.encode(collectionParams).finish();
-                        level = Object.keys(Milvus_1.ConsistencyLevelEnum).includes(consistency_level)
-                            ? Milvus_1.ConsistencyLevelEnum[consistency_level]
-                            : Milvus_1.ConsistencyLevelEnum['Bounded'];
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'CreateCollection', __assign(__assign({}, data), { schema: schemaBtyes, consistency_level: level }), data.timeout)];
+                        schemaBytes = CollectionSchema.encode(collectionParams).finish();
+                        level = Object.keys(_1.ConsistencyLevelEnum).includes(consistency_level)
+                            ? _1.ConsistencyLevelEnum[consistency_level]
+                            : _1.ConsistencyLevelEnum['Bounded'];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'CreateCollection', __assign(__assign({}, data), { schema: schemaBytes, consistency_level: level }), data.timeout)];
                     case 1:
                         promise = _c.sent();
                         return [2 /*return*/, promise];
@@ -166,37 +156,37 @@ var Collection = /** @class */ (function (_super) {
         });
     };
     /**
-     * Check if a collection exists.
-     *
-     * @param data
-     *  | Property | Type | Description |
-     *  | :-- | :-- | :-- |
-     *  | collection_name | String | Collection name |
-     *  | timeout? | number | An optional duration of time in millisecond to allow for the RPC. If it is set to undefined, the client keeps waiting until the server responds or error occurs. Default is undefined |
-     *
-     * @returns
-     *  | Property | Description |
-     *  | :-- | :-- |
-     *  | status | { error_code: number, reason: string } |
-     *  | value | `true` or `false` |
-    
-     *
-     * #### Example
-     *
-     * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.hasCollection({
-     *     collection_name: 'my_collection',
-     *  });
-     * ```
-     */
+   * Check if a collection exists.
+   *
+   * @param data
+   *  | Property | Type | Description |
+   *  | :-- | :-- | :-- |
+   *  | collection_name | String | Collection name |
+   *  | timeout? | number | An optional duration of time in millisecond to allow for the RPC. If it is set to undefined, the client keeps waiting until the server responds or error occurs. Default is undefined |
+   *
+   * @returns
+   *  | Property | Description |
+   *  | :-- | :-- |
+   *  | status | { error_code: number, reason: string } |
+   *  | value | `true` or `false` |
+  
+   *
+   * #### Example
+   *
+   * ```
+   *  new milvusClient(MILUVS_ADDRESS).hasCollection({
+   *     collection_name: 'my_collection',
+   *  });
+   * ```
+   */
     Collection.prototype.hasCollection = function (data) {
         return __awaiter(this, void 0, void 0, function () {
             var promise;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'HasCollection', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'HasCollection', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -205,36 +195,36 @@ var Collection = /** @class */ (function (_super) {
         });
     };
     /**
-     * List all collections or get collection loading status.
-     *
-     * @param data
-     *  | Property | Type | Description |
-     *  | :-- | :-- | :-- |
-     *  | type(optional) | enum | All -> 0, Loaded -> 1 |
-     *  | collection_names(optional) | String[] | If `type = Loaded`, Milvus will return `collection_names inMemory_percentages` |
-     *  | timeout? | number | An optional duration of time in millisecond to allow for the RPC. If it is set to undefined, the client keeps waiting until the server responds or error occurs. Default is undefined |
+   * List all collections or get collection loading status.
+   *
+   * @param data
+   *  | Property | Type | Description |
+   *  | :-- | :-- | :-- |
+   *  | type(optional) | enum | All -> 0, Loaded -> 1 |
+   *  | collection_names(optional) | String[] | If `type = Loaded`, Milvus will return `collection_names inMemory_percentages` |
+   *  | timeout? | number | An optional duration of time in millisecond to allow for the RPC. If it is set to undefined, the client keeps waiting until the server responds or error occurs. Default is undefined |
   
-     *
-     * @returns
-     * | Property | Description |
-     *  | :-- | :-- |
-     *  | status | { error_code: number, reason: string } |
-     *  | data |  Contains collection name, ID , timestamp (UTC created time), and loadedPercentage (100 means loaded) |
-     *
-     *
-     * #### Example
-     *
-     * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.showCollections();
-     * ```
-     */
+   *
+   * @returns
+   * | Property | Description |
+   *  | :-- | :-- |
+   *  | status | { error_code: number, reason: string } |
+   *  | data |  Contains collection name, ID , timestamp (UTC created time), and loadedPercentage (100 means loaded) |
+   *
+   *
+   * #### Example
+   *
+   * ```
+   *  new milvusClient(MILUVS_ADDRESS).showCollections();
+   * ```
+   */
     Collection.prototype.showCollections = function (data) {
         return __awaiter(this, void 0, void 0, function () {
             var promise, result;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ShowCollections', {
-                            type: data ? data.type : types_1.ShowCollectionsType.All,
+                    case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ShowCollections', {
+                            type: data ? data.type : _1.ShowCollectionsType.All,
                             collection_names: (data === null || data === void 0 ? void 0 : data.collection_names) || [],
                         }, data === null || data === void 0 ? void 0 : data.timeout)];
                     case 1:
@@ -274,7 +264,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.describeCollection({
+     *  new milvusClient(MILUVS_ADDRESS).describeCollection({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -285,8 +275,8 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DescribeCollection', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'DescribeCollection', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -314,7 +304,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.getCollectionStatistics({
+     *  new milvusClient(MILUVS_ADDRESS).getCollectionStatistics({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -325,11 +315,11 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetCollectionStatistics', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetCollectionStatistics', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
-                        promise.data = (0, Format_1.formatKeyValueData)(promise.stats, ['row_count']);
+                        promise.data = (0, utils_1.formatKeyValueData)(promise.stats, ['row_count']);
                         return [2 /*return*/, promise];
                 }
             });
@@ -356,7 +346,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.loadCollection({
+     *  new milvusClient(MILUVS_ADDRESS).loadCollection({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -367,8 +357,8 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'LoadCollection', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'LoadCollection', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -397,7 +387,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.loadCollectionSync({
+     *  new milvusClient(MILUVS_ADDRESS).loadCollectionSync({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -408,11 +398,11 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'LoadCollection', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'LoadCollection', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
-                        if (promise.error_code !== types_1.ErrorCode.SUCCESS) {
+                        if (promise.error_code !== _1.ErrorCode.SUCCESS) {
                             throw new Error("ErrorCode: ".concat(promise.error_code, ". Reason: ").concat(promise.reason));
                         }
                         loadedPercentage = 0;
@@ -421,11 +411,11 @@ var Collection = /** @class */ (function (_super) {
                         if (!(Number(loadedPercentage) < 100)) return [3 /*break*/, 4];
                         return [4 /*yield*/, this.showCollections({
                                 collection_names: [data.collection_name],
-                                type: types_1.ShowCollectionsType.Loaded,
+                                type: _1.ShowCollectionsType.Loaded,
                             })];
                     case 3:
                         res = _a.sent();
-                        if (res.status.error_code !== types_1.ErrorCode.SUCCESS) {
+                        if (res.status.error_code !== _1.ErrorCode.SUCCESS) {
                             throw new Error("ErrorCode: ".concat(res.status.error_code, ". Reason: ").concat(res.status.reason));
                         }
                         // Because we pass collection_names in showCollections, so it will only this collection in result.
@@ -455,7 +445,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.releaseCollection({
+     *  new milvusClient(MILUVS_ADDRESS).releaseCollection({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -466,8 +456,8 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ReleaseCollection', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ReleaseCollection', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -494,7 +484,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.renameCollection({
+     *  new milvusClient(MILUVS_ADDRESS).renameCollection({
      *    collection_name: 'my_collection',
      *    new_collection_name: 'my_new_collection'
      *  });
@@ -505,7 +495,7 @@ var Collection = /** @class */ (function (_super) {
             var promise;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'RenameCollection', {
+                    case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'RenameCollection', {
                             oldName: data.collection_name,
                             newName: data.new_collection_name,
                         }, data.timeout)];
@@ -534,7 +524,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.dropCollection({
+     *  new milvusClient(MILUVS_ADDRESS).dropCollection({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -545,8 +535,8 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DropCollection', data, data.timeout)];
+                        (0, utils_1.checkCollectionName)(data);
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'DropCollection', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -574,7 +564,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.createAlias({
+     *  new milvusClient(MILUVS_ADDRESS).createAlias({
      *    alias: 'my_collection_alis',
      *    collection_name: 'my_collection',
      *  });
@@ -586,11 +576,11 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
+                        (0, utils_1.checkCollectionName)(data);
                         if (!data.alias) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'CreateAlias', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'CreateAlias', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -618,7 +608,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.dropAlias({
+     *  new milvusClient(MILUVS_ADDRESS).dropAlias({
      *    alias: 'my_collection_alis',
      *    collection_name: 'my_collection',
      *  });
@@ -631,9 +621,9 @@ var Collection = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data.alias) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DropAlias', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'DropAlias', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -661,7 +651,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.alterAlais({
+     *  new milvusClient(MILUVS_ADDRESS).alterAlais({
      *    alias: 'my_collection_alis',
      *    collection_name: 'my_collection',
      *  });
@@ -673,11 +663,11 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
+                        (0, utils_1.checkCollectionName)(data);
                         if (!data.alias) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'AlterAlias', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'AlterAlias', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -703,7 +693,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.compact({
+     *  new milvusClient(MILUVS_ADDRESS).compact({
      *    collection_name: 'my_collection',
      *  });
      * ```
@@ -714,11 +704,11 @@ var Collection = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
+                        (0, utils_1.checkCollectionName)(data);
                         return [4 /*yield*/, this.describeCollection(data)];
                     case 1:
                         collectionInfo = _a.sent();
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ManualCompaction', {
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ManualCompaction', {
                                 collectionID: collectionInfo.collectionID,
                             }, data.timeout)];
                     case 2:
@@ -746,7 +736,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.getCompactionState({
+     *  new milvusClient(MILUVS_ADDRESS).getCompactionState({
      *    compactionID: compactionID,
      *  });
      * ```
@@ -758,9 +748,9 @@ var Collection = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.compactionID) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetCompactionState', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetCompactionState', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -786,7 +776,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.getCompactionStateWithPlans({
+     *  new milvusClient(MILUVS_ADDRESS).getCompactionStateWithPlans({
      *    compactionID: compactionID,
      *  });
      * ```
@@ -798,9 +788,9 @@ var Collection = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.compactionID) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetCompactionStateWithPlans', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetCompactionStateWithPlans', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -826,7 +816,7 @@ var Collection = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).collectionManager.getReplicas({
+     *  new milvusClient(MILUVS_ADDRESS).getReplicas({
      *    collectionID: collectionID,
      *  });
      *
@@ -855,9 +845,94 @@ var Collection = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.collectionID) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_ID_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_ID_IS_REQUIRED);
+                        }
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetReplicas', data, data.timeout)];
+                    case 1:
+                        res = _a.sent();
+                        return [2 /*return*/, res];
+                }
+            });
+        });
+    };
+    /**
+   * Get loading progress of a collection
+   *
+   * @param data
+   *  | Property | Type | Description |
+   *  | :-- | :-- | :-- |
+   *  | collection_name | string | the name of the collection |
+   *  | timeout? | number | An optional duration of time in millisecond to allow for the RPC. If it is set to undefined, the client keeps waiting until the server responds or error occurs. Default is undefined |
+   *
+   * @returns
+   *  | Property | Description |
+   *  | :-- | :-- |
+   *  | status | { error_code: number, reason: string } |
+   *  | total_row_num | the total number of rows in the collection |
+   *  | total_loaded_row_num | the total number of loaded rows in the collection |
+   *
+   * @throws {Error} if `collection_name` property is not present in `data`
+   *
+   * #### Example
+   *
+   * ```
+   *  new milvusClient(MILUVS_ADDRESS).getLoadingProgress({
+   *    collection_name: 'my_collection',
+   *  });
+   * ```
+   */
+    Collection.prototype.getLoadingProgress = function (data) {
+        return __awaiter(this, void 0, void 0, function () {
+            var res;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        if (!data || !data.collection_name) {
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                        }
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetLoadingProgress', data, data.timeout)];
+                    case 1:
+                        res = _a.sent();
+                        return [2 /*return*/, res];
+                }
+            });
+        });
+    };
+    /**
+   * Get the loading state of a collection
+   *
+   * @param data
+   *  | Property | Type | Description |
+   *  | :-- | :-- | :-- |
+   *  | collection_name | string | the name of the collection |
+   *  | timeout? | number | An optional duration of time in milliseconds to allow for the RPC. If it is set to undefined, the client keeps waiting until the server responds or an error occurs. Default is undefined |
+   *
+   * @returns
+   *  | Property | Description |
+   *  | :-- | :-- |
+   *  | status | { error_code: number, reason: string } |
+   *  | state | the loading state of the collection |
+   *
+   * @throws {Error} if `collection_name` property is not present in `data`
+   *
+   * #### Example
+   *
+   * ```
+   *  new milvusClient(MILUVS_ADDRESS).getLoadState({
+   *    collection_name: 'my_collection',
+   *  });
+   * ```
+   */
+    Collection.prototype.getLoadState = function (data) {
+        return __awaiter(this, void 0, void 0, function () {
+            var res;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        if (!data || !data.collection_name) {
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetReplicas', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetLoadState', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -866,6 +941,6 @@ var Collection = /** @class */ (function (_super) {
         });
     };
     return Collection;
-}(Client_1.Client));
+}(BaseClient_1.BaseClient));
 exports.Collection = Collection;
 //# sourceMappingURL=Collection.js.map