@zilliz/milvus2-sdk-node 2.2.4 → 2.2.5

package/dist/milvus/Data.d.ts

@@ -1,11 +1,7 @@
-import { Client } from './Client';
+import { DataType, DeleteEntitiesReq, FlushReq, GetFlushStateReq, GetQuerySegmentInfoReq, InsertReq, LoadBalanceReq, ImportReq, ListImportTasksReq, FlushResult, GetFlushStateResponse, GetMetricsResponse, GetQuerySegmentInfoResponse, MutationResult, QueryResults, ResStatus, SearchResults, ImportResponse, ListImportTasksResponse, GetMetricsRequest, QueryReq, SearchReq } from '.';
 import { Collection } from './Collection';
-import { DeleteEntitiesReq, FlushReq, GetFlushStateReq, GetQuerySegmentInfoReq, InsertReq, LoadBalanceReq, ImportReq, ListImportTasksReq, FlushResult, GetFlushStateResponse, GetMetricsResponse, GetQuerySegmentInfoResponse, MutationResult, QueryResults, ResStatus, SearchResults, ImportResponse, ListImportTasksResponse, GetMetricsRequest, QueryReq, SearchReq } from './types';
-export declare class Data extends Client {
-    vectorTypes: number[];
-    collectionManager: Collection;
-    private readonly _protoRoot;
-    constructor(client: any, collectionManager: Collection);
+export declare class Data extends Collection {
+    vectorTypes: DataType[];
     /**
      * Insert data into Milvus.
      *
@@ -31,7 +27,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.insert({
+     *  new milvusClient(MILUVS_ADDRESS).insert({
      *    collection_name: COLLECTION_NAME,
      *    fields_data: [{
      *      vector_field: [1,2,2,4],
@@ -63,7 +59,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.deleteEntities({
+     *  new milvusClient(MILUVS_ADDRESS).deleteEntities({
      *    collection_name: COLLECTION_NAME,
      *    expr: 'id in [1,2,3,4]'
      *  });
@@ -98,7 +94,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.search({
+     *  new milvusClient(MILUVS_ADDRESS).search({
      *   collection_name: COLLECTION_NAME,
      *   expr: "",
      *   vectors: [[1, 2, 3, 4]],
@@ -132,7 +128,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.flush({
+     *  new milvusClient(MILUVS_ADDRESS).flush({
      *     collection_names: ['my_collection'],
      *  });
      * ```
@@ -157,7 +153,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.flushSync({
+     *  new milvusClient(MILUVS_ADDRESS).flushSync({
      *     collection_names: ['my_collection'],
      *  });
      * ```
@@ -187,7 +183,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.query({
+     *  new milvusClient(MILUVS_ADDRESS).query({
      *    collection_name: 'my_collection',
      *    expr: "age in [1,2,3,4,5,6,7,8]",
      *    output_fields: ["age"],
@@ -224,7 +220,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *   const res = await milvusClient.dataManager.getFlushState({
+     *   const res = await milvusClient.getFlushState({
      *    segmentIDs: segIds,
      *   });
      * ```
@@ -252,7 +248,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.loadBalance({
+     *   const res = await loadBalance({
      *      src_nodeID: 31,
      *   });
      * ```
@@ -278,7 +274,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.getQuerySegmentInfo({
+     *   const res = await getQuerySegmentInfo({
      *      collectionName: COLLECTION,
      *    });
      * ```
@@ -304,7 +300,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.bulkInsert({
+     *   const res = await bulkInsert({
      *      collection_name: COLLECTION,
      *      files: [`path-to-data-file.json`]
      *    });
@@ -335,7 +331,7 @@ export declare class Data extends Client {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.listImportTasks({
+     *   const res = await listImportTasks({
      *      collection_name: COLLECTION
      *    });
      * ```

package/dist/milvus/Data.js

@@ -70,30 +70,16 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
     }
     return to.concat(ar || Array.prototype.slice.call(from));
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Data = void 0;
-var protobufjs_1 = __importDefault(require("protobufjs"));
-var path_1 = __importDefault(require("path"));
 var utils_1 = require("../utils");
-var Client_1 = require("./Client");
-var ErrorReason_1 = require("./const/ErrorReason");
-var index_1 = require("./utils/index");
-var Blob_1 = require("./utils/Blob");
-var Validate_1 = require("./utils/Validate");
-var Format_1 = require("./utils/Format");
-var Milvus_1 = require("./const/Milvus");
-var types_1 = require("./types");
-var protoPath = path_1.default.resolve(__dirname, '../proto/proto/milvus.proto');
+var _1 = require(".");
+var Collection_1 = require("./Collection");
 var Data = /** @class */ (function (_super) {
     __extends(Data, _super);
-    function Data(client, collectionManager) {
-        var _this = _super.call(this, client) || this;
-        _this.vectorTypes = [Milvus_1.DataType.BinaryVector, Milvus_1.DataType.FloatVector];
-        _this.collectionManager = collectionManager;
-        _this._protoRoot = protobufjs_1.default.loadSync(protoPath);
+    function Data() {
+        var _this = _super !== null && _super.apply(this, arguments) || this;
+        _this.vectorTypes = [_1.DataType.BinaryVector, _1.DataType.FloatVector];
         return _this;
     }
     /**
@@ -121,7 +107,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.insert({
+     *  new milvusClient(MILUVS_ADDRESS).insert({
      *    collection_name: COLLECTION_NAME,
      *    fields_data: [{
      *      vector_field: [1,2,2,4],
@@ -137,19 +123,19 @@ var Data = /** @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.fields_data ||
                             !Array.isArray(data.fields_data) ||
                             !data.fields_data.length) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.INSERT_CHECK_FILEDS_DATA_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.INSERT_CHECK_FILEDS_DATA_IS_REQUIRED);
                         }
                         collection_name = data.collection_name;
-                        return [4 /*yield*/, this.collectionManager.describeCollection({
+                        return [4 /*yield*/, this.describeCollection({
                                 collection_name: collection_name,
                             })];
                     case 1:
                         collectionInfo = _a.sent();
-                        if (collectionInfo.status.error_code !== types_1.ErrorCode.SUCCESS) {
+                        if (collectionInfo.status.error_code !== _1.ErrorCode.SUCCESS) {
                             throw new Error(collectionInfo.status.reason);
                         }
                         fieldsData = collectionInfo.schema.fields
@@ -157,7 +143,7 @@ var Data = /** @class */ (function (_super) {
                             .map(function (v) { return ({
                             name: v.name,
                             type: v.data_type,
-                            dim: Number((0, index_1.findKeyValue)(v.type_params, 'dim')),
+                            dim: Number((0, utils_1.findKeyValue)(v.type_params, 'dim')),
                             value: [],
                         }); });
                         params = __assign(__assign({}, data), { num_rows: data.fields_data.length });
@@ -168,13 +154,13 @@ var Data = /** @class */ (function (_super) {
                             fieldNames.forEach(function (name) {
                                 var target = fieldsData.find(function (item) { return item.name === name; });
                                 if (!target) {
-                                    throw new Error("".concat(ErrorReason_1.ERROR_REASONS.INSERT_CHECK_WRONG_FIELD, " ").concat(i));
+                                    throw new Error("".concat(_1.ERROR_REASONS.INSERT_CHECK_WRONG_FIELD, " ").concat(i));
                                 }
-                                var isVector = _this.vectorTypes.includes(Milvus_1.DataTypeMap[target.type.toLowerCase()]);
+                                var isVector = _this.vectorTypes.includes(_1.DataTypeMap[target.type.toLowerCase()]);
                                 // Check if the dimension is matched when the data type is BinaryVector.
-                                if (Milvus_1.DataTypeMap[target.type.toLowerCase()] === Milvus_1.DataType.BinaryVector &&
+                                if (_1.DataTypeMap[target.type.toLowerCase()] === _1.DataType.BinaryVector &&
                                     v[name].length !== target.dim / 8) {
-                                    throw new Error(ErrorReason_1.ERROR_REASONS.INSERT_CHECK_WRONG_DIM);
+                                    throw new Error(_1.ERROR_REASONS.INSERT_CHECK_WRONG_DIM);
                                 }
                                 // Value in vector field should be array. Therefore you need concat it.
                                 // but array.concat is slow, we need for loop to push the value one by one
@@ -193,55 +179,55 @@ var Data = /** @class */ (function (_super) {
                             var _a, _b, _c, _d;
                             // milvus return string for field type, so we define the DataTypeMap to the value we need.
                             // but if milvus change the string, may casue we cant find value.
-                            var type = Milvus_1.DataTypeMap[v.type.toLowerCase()];
+                            var type = _1.DataTypeMap[v.type.toLowerCase()];
                             var key = _this.vectorTypes.includes(type) ? 'vectors' : 'scalars';
                             var dataKey = 'float_vector';
                             switch (type) {
-                                case Milvus_1.DataType.FloatVector:
+                                case _1.DataType.FloatVector:
                                     dataKey = 'float_vector';
                                     break;
-                                case Milvus_1.DataType.BinaryVector:
+                                case _1.DataType.BinaryVector:
                                     dataKey = 'binary_vector';
                                     break;
-                                case Milvus_1.DataType.Double:
+                                case _1.DataType.Double:
                                     dataKey = 'double_data';
                                     break;
-                                case Milvus_1.DataType.Float:
+                                case _1.DataType.Float:
                                     dataKey = 'float_data';
                                     break;
-                                case Milvus_1.DataType.Int64:
+                                case _1.DataType.Int64:
                                     dataKey = 'long_data';
                                     break;
-                                case Milvus_1.DataType.Int32:
-                                case Milvus_1.DataType.Int16:
-                                case Milvus_1.DataType.Int8:
+                                case _1.DataType.Int32:
+                                case _1.DataType.Int16:
+                                case _1.DataType.Int8:
                                     dataKey = 'int_data';
                                     break;
-                                case Milvus_1.DataType.Bool:
+                                case _1.DataType.Bool:
                                     dataKey = 'bool_data';
                                     break;
-                                case Milvus_1.DataType.VarChar:
+                                case _1.DataType.VarChar:
                                     dataKey = 'string_data';
                                     break;
                                 default:
-                                    throw new Error(ErrorReason_1.ERROR_REASONS.INSERT_CHECK_WRONG_DATA_TYPE);
+                                    throw new Error(_1.ERROR_REASONS.INSERT_CHECK_WRONG_DATA_TYPE);
                             }
                             return _a = {
                                     type: type,
                                     field_name: v.name
                                 },
-                                _a[key] = type === Milvus_1.DataType.FloatVector
+                                _a[key] = type === _1.DataType.FloatVector
                                     ? (_b = {
                                             dim: v.dim
                                         },
                                         _b[dataKey] = {
                                             data: v.value,
                                         },
-                                        _b) : type === Milvus_1.DataType.BinaryVector
+                                        _b) : type === _1.DataType.BinaryVector
                                     ? (_c = {
                                             dim: v.dim
                                         },
-                                        _c[dataKey] = (0, Blob_1.parseBinaryVectorToBytes)(v.value),
+                                        _c[dataKey] = (0, utils_1.parseBinaryVectorToBytes)(v.value),
                                         _c) : (_d = {},
                                     _d[dataKey] = {
                                         data: v.value,
@@ -249,7 +235,7 @@ var Data = /** @class */ (function (_super) {
                                     _d),
                                 _a;
                         });
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Insert', params, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Insert', params, data.timeout)];
                     case 2:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -279,7 +265,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.deleteEntities({
+     *  new milvusClient(MILUVS_ADDRESS).deleteEntities({
      *    collection_name: COLLECTION_NAME,
      *    expr: 'id in [1,2,3,4]'
      *  });
@@ -292,9 +278,9 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.collection_name || !data.expr) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.DELETE_PARAMS_CHECK);
+                            throw new Error(_1.ERROR_REASONS.DELETE_PARAMS_CHECK);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Delete', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Delete', data, data.timeout)];
                     case 1:
                         promise = _a.sent();
                         return [2 /*return*/, promise];
@@ -330,7 +316,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.search({
+     *  new milvusClient(MILUVS_ADDRESS).search({
      *   collection_name: COLLECTION_NAME,
      *   expr: "",
      *   vectors: [[1, 2, 3, 4]],
@@ -352,54 +338,54 @@ var Data = /** @class */ (function (_super) {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
+                        (0, utils_1.checkCollectionName)(data);
                         if (!data.search_params ||
                             !data.search_params.anns_field ||
                             !data.search_params.metric_type ||
                             !data.search_params.topk ||
                             !data.search_params.params) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_PARAMS_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.SEARCH_PARAMS_IS_REQUIRED);
                         }
                         if (!this.vectorTypes.includes(data.vector_type))
-                            throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_MISS_VECTOR_TYPE);
-                        return [4 /*yield*/, this.collectionManager.describeCollection({
+                            throw new Error(_1.ERROR_REASONS.SEARCH_MISS_VECTOR_TYPE);
+                        return [4 /*yield*/, this.describeCollection({
                                 collection_name: data.collection_name,
                             })];
                     case 1:
                         collectionInfo = _c.sent();
                         targetField = collectionInfo.schema.fields.find(function (v) { return v.name === data.search_params.anns_field; });
                         if (!targetField) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_NOT_FIND_VECTOR_FIELD);
+                            throw new Error(_1.ERROR_REASONS.SEARCH_NOT_FIND_VECTOR_FIELD);
                         }
-                        dim = (0, index_1.findKeyValue)(targetField.type_params, 'dim');
-                        vectorType = Milvus_1.DataTypeMap[targetField.data_type.toLowerCase()];
-                        dimension = vectorType === Milvus_1.DataType.BinaryVector ? Number(dim) / 8 : Number(dim);
+                        dim = (0, utils_1.findKeyValue)(targetField.type_params, 'dim');
+                        vectorType = _1.DataTypeMap[targetField.data_type.toLowerCase()];
+                        dimension = vectorType === _1.DataType.BinaryVector ? Number(dim) / 8 : Number(dim);
                         if (!data.vectors[0] || data.vectors[0].length !== dimension) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_DIM_NOT_MATCH);
+                            throw new Error(_1.ERROR_REASONS.SEARCH_DIM_NOT_MATCH);
                         }
                         round_decimal = data.search_params.round_decimal;
                         if (round_decimal !== undefined &&
                             (!Number.isInteger(round_decimal) ||
                                 round_decimal < -1 ||
                                 round_decimal > 6)) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_ROUND_DECIMAL_NOT_VALID);
+                            throw new Error(_1.ERROR_REASONS.SEARCH_ROUND_DECIMAL_NOT_VALID);
                         }
-                        PlaceholderGroup = this._protoRoot.lookupType('milvus.proto.common.PlaceholderGroup');
+                        PlaceholderGroup = this.milvusProto.lookupType('milvus.proto.common.PlaceholderGroup');
                         placeholderGroupParams = PlaceholderGroup.create({
                             placeholders: [
                                 {
                                     tag: '$0',
                                     type: data.vector_type,
                                     values: data.vectors.map(function (v) {
-                                        return data.vector_type === Milvus_1.DataType.BinaryVector
-                                            ? (0, Blob_1.parseBinaryVectorToBytes)(v)
-                                            : (0, Blob_1.parseFloatVectorToBytes)(v);
+                                        return data.vector_type === _1.DataType.BinaryVector
+                                            ? (0, utils_1.parseBinaryVectorToBytes)(v)
+                                            : (0, utils_1.parseFloatVectorToBytes)(v);
                                     }),
                                 },
                             ],
                         });
                         placeholderGroupBytes = PlaceholderGroup.encode(placeholderGroupParams).finish();
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Search', __assign(__assign({}, data), { nq: data.nq || data.vectors.length, dsl: data.expr || '', dsl_type: types_1.DslType.BoolExprV1, placeholder_group: placeholderGroupBytes, search_params: (0, Format_1.parseToKeyValue)(data.search_params) }), data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Search', __assign(__assign({}, data), { nq: data.nq || data.vectors.length, dsl: data.expr || '', dsl_type: _1.DslType.BoolExprV1, placeholder_group: placeholderGroupBytes, search_params: (0, utils_1.parseToKeyValue)(data.search_params) }), data.timeout)];
                     case 2:
                         promise = _c.sent();
                         results = [];
@@ -433,7 +419,7 @@ var Data = /** @class */ (function (_super) {
                                     var i = index === 0 ? scoreIndex : scoreIndex + topk;
                                     var fixedScore = typeof round_decimal === 'undefined' || round_decimal === -1
                                         ? score
-                                        : (0, Format_1.formatNumberPrecision)(score, round_decimal);
+                                        : (0, utils_1.formatNumberPrecision)(score, round_decimal);
                                     var result = {
                                         score: fixedScore,
                                         id: idData_1 ? idData_1[i] : '',
@@ -471,7 +457,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.flush({
+     *  new milvusClient(MILUVS_ADDRESS).flush({
      *     collection_names: ['my_collection'],
      *  });
      * ```
@@ -485,9 +471,9 @@ var Data = /** @class */ (function (_super) {
                         if (!data ||
                             !Array.isArray(data.collection_names) ||
                             !data.collection_names.length) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Flush', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Flush', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -514,7 +500,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.flushSync({
+     *  new milvusClient(MILUVS_ADDRESS).flushSync({
      *     collection_names: ['my_collection'],
      *  });
      * ```
@@ -528,9 +514,9 @@ var Data = /** @class */ (function (_super) {
                         if (!data ||
                             !Array.isArray(data.collection_names) ||
                             !data.collection_names.length) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Flush', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Flush', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         segIDs = Object.keys(res.coll_segIDs)
@@ -544,7 +530,7 @@ var Data = /** @class */ (function (_super) {
                         return [4 /*yield*/, this.getFlushState({ segmentIDs: segIDs })];
                     case 3:
                         flushRes = _a.sent();
-                        return [4 /*yield*/, (0, index_1.sleep)(100)];
+                        return [4 /*yield*/, (0, utils_1.sleep)(100)];
                     case 4:
                         _a.sent();
                         isFlushed = flushRes.flushed;
@@ -580,7 +566,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.query({
+     *  new milvusClient(MILUVS_ADDRESS).query({
      *    collection_name: 'my_collection',
      *    expr: "age in [1,2,3,4,5,6,7,8]",
      *    output_fields: ["age"],
@@ -593,14 +579,14 @@ var Data = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        (0, Validate_1.checkCollectionName)(data);
+                        (0, utils_1.checkCollectionName)(data);
                         if (typeof data.limit === 'number') {
                             limits = { limit: data.limit };
                         }
                         if (typeof data.offset === 'number') {
                             offset = { offset: data.offset };
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Query', __assign(__assign({}, data), { query_params: (0, Format_1.parseToKeyValue)(__assign(__assign({}, limits), offset)) }), data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Query', __assign(__assign({}, data), { query_params: (0, utils_1.parseToKeyValue)(__assign(__assign({}, limits), offset)) }), data.timeout)];
                     case 1:
                         promise = _a.sent();
                         results = [];
@@ -672,9 +658,9 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.request || !data.request.metric_type) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.GET_METRIC_CHECK_PARAMS);
+                            throw new Error(_1.ERROR_REASONS.GET_METRIC_CHECK_PARAMS);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetMetrics', {
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetMetrics', {
                                 request: JSON.stringify(data.request),
                             }, data.timeout)];
                     case 1:
@@ -705,7 +691,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *   const res = await milvusClient.dataManager.getFlushState({
+     *   const res = await milvusClient.getFlushState({
      *    segmentIDs: segIds,
      *   });
      * ```
@@ -717,9 +703,9 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.segmentIDs) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.GET_FLUSH_STATE_CHECK_PARAMS);
+                            throw new Error(_1.ERROR_REASONS.GET_FLUSH_STATE_CHECK_PARAMS);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetFlushState', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetFlushState', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -749,7 +735,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.loadBalance({
+     *   const res = await loadBalance({
      *      src_nodeID: 31,
      *   });
      * ```
@@ -761,9 +747,9 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.src_nodeID) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.LOAD_BALANCE_CHECK_PARAMS);
+                            throw new Error(_1.ERROR_REASONS.LOAD_BALANCE_CHECK_PARAMS);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'LoadBalance', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'LoadBalance', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -791,7 +777,7 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.getQuerySegmentInfo({
+     *   const res = await getQuerySegmentInfo({
      *      collectionName: COLLECTION,
      *    });
      * ```
@@ -803,9 +789,9 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.collectionName) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetQuerySegmentInfo', data, data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetQuerySegmentInfo', data, data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -833,12 +819,13 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.bulkInsert({
+     *   const res = await bulkInsert({
      *      collection_name: COLLECTION,
      *      files: [`path-to-data-file.json`]
      *    });
      * ```
      */
+    /* istanbul ignore next */
     Data.prototype.bulkInsert = function (data) {
         return __awaiter(this, void 0, void 0, function () {
             var res;
@@ -846,12 +833,12 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.collection_name) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
                         }
                         if (!data || !data.files) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.IMPORT_FILE_CHECK);
+                            throw new Error(_1.ERROR_REASONS.IMPORT_FILE_CHECK);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Import', __assign(__assign({}, data), { options: data.options || [] }), data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Import', __assign(__assign({}, data), { options: data.options || [] }), data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -883,11 +870,12 @@ var Data = /** @class */ (function (_super) {
      * #### Example
      *
      * ```
-     *   const res = await dataManager.listImportTasks({
+     *   const res = await listImportTasks({
      *      collection_name: COLLECTION
      *    });
      * ```
      */
+    /* istanbul ignore next */
     Data.prototype.listImportTasks = function (data) {
         return __awaiter(this, void 0, void 0, function () {
             var res;
@@ -895,9 +883,9 @@ var Data = /** @class */ (function (_super) {
                 switch (_a.label) {
                     case 0:
                         if (!data || !data.collection_name) {
-                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                            throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
                         }
-                        return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ListImportTasks', __assign(__assign({}, data), { limit: data.limit || 0 }), data.timeout)];
+                        return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ListImportTasks', __assign(__assign({}, data), { limit: data.limit || 0 }), data.timeout)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -906,6 +894,6 @@ var Data = /** @class */ (function (_super) {
         });
     };
     return Data;
-}(Client_1.Client));
+}(Collection_1.Collection));
 exports.Data = Data;
 //# sourceMappingURL=Data.js.map