@zilliz/milvus2-sdk-node 2.2.3 → 2.2.5

package/dist/milvus/Data.js

@@ -40,7 +40,7 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
     function verb(n) { return function (v) { return step([n, v]); }; }
     function step(op) {
         if (f) throw new TypeError("Generator is already executing.");
-        while (_) try {
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
             if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
             if (y = 0, t) op = [op[0] & 2, t.value];
             switch (op[0]) {
@@ -70,56 +70,44 @@ 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 utils_1 = require("../utils");
-var Client_1 = require("./Client");
-var ErrorReason_1 = require("./const/ErrorReason");
-var Common_1 = require("./types/Common");
-var Response_1 = require("./types/Response");
-var index_1 = require("./utils/index");
-var Blob_1 = require("./utils/Blob");
-var path_1 = __importDefault(require("path"));
-var Format_1 = require("./utils/Format");
-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 = [Common_1.DataType.BinaryVector, Common_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;
     }
     /**
      * Insert data into Milvus.
      *
      * @param data
-     *  | Property                | Type                   |           Description              |
-     *  | :---------------------- | :--------------------  | :-------------------------------  |
-     *  | collection_name         | String                 |       Collection name       |
-     *  | partition_name(optional)| String                 |       Partition name       |
-     *  | fields_data             | { [x: string]: any }[] |      If the field type is binary, the vector data length needs to be dimension / 8   |
-     *  | hash_keys(optional)    | Number[]               |  The hash value depends on the primarykey value       |
-     *  | 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       |
-     *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-------------| :-------------------------------  |
-     *  | status        |  { error_code: number, reason: string }|
-     *  | succ_index    |  Index array of the successfully inserted data      |
-     *  | err_index    |   Index array of the unsuccessfully inserted data      |
-     *  | IDs    |        ID array of the successfully inserted data      |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_name | String | Collection name |
+     *  | partition_name(optional)| String | Partition name |
+     *  | fields_data | { [x: string]: any }[] | If the field type is binary, the vector data length needs to be dimension / 8 |
+     *  | hash_keys(optional) | Number[] |  The hash value depends on the primarykey value |
+     *  | 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 } |
+     *  | succ_index |  Index array of the successfully inserted data |
+     *  | err_index | Index array of the unsuccessfully inserted data |
+     *  | IDs | ID array of the successfully inserted data |
      *
      *
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.insert({
+     *  new milvusClient(MILUVS_ADDRESS).insert({
      *    collection_name: COLLECTION_NAME,
      *    fields_data: [{
      *      vector_field: [1,2,2,4],
@@ -135,19 +123,19 @@ var Data = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.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 !== Response_1.ErrorCode.SUCCESS) {
+                        if (collectionInfo.status.error_code !== _1.ErrorCode.SUCCESS) {
                             throw new Error(collectionInfo.status.reason);
                         }
                         fieldsData = collectionInfo.schema.fields
@@ -155,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 });
@@ -166,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(Common_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 (Common_1.DataTypeMap[target.type.toLowerCase()] === Common_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
@@ -191,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 = Common_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 Common_1.DataType.FloatVector:
+                                case _1.DataType.FloatVector:
                                     dataKey = 'float_vector';
                                     break;
-                                case Common_1.DataType.BinaryVector:
+                                case _1.DataType.BinaryVector:
                                     dataKey = 'binary_vector';
                                     break;
-                                case Common_1.DataType.Double:
+                                case _1.DataType.Double:
                                     dataKey = 'double_data';
                                     break;
-                                case Common_1.DataType.Float:
+                                case _1.DataType.Float:
                                     dataKey = 'float_data';
                                     break;
-                                case Common_1.DataType.Int64:
+                                case _1.DataType.Int64:
                                     dataKey = 'long_data';
                                     break;
-                                case Common_1.DataType.Int32:
-                                case Common_1.DataType.Int16:
-                                case Common_1.DataType.Int8:
+                                case _1.DataType.Int32:
+                                case _1.DataType.Int16:
+                                case _1.DataType.Int8:
                                     dataKey = 'int_data';
                                     break;
-                                case Common_1.DataType.Bool:
+                                case _1.DataType.Bool:
                                     dataKey = 'bool_data';
                                     break;
-                                case Common_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 === Common_1.DataType.FloatVector
+                                _a[key] = type === _1.DataType.FloatVector
                                     ? (_b = {
                                             dim: v.dim
                                         },
                                         _b[dataKey] = {
                                             data: v.value,
                                         },
-                                        _b) : type === Common_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,
@@ -247,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];
@@ -259,24 +247,25 @@ var Data = /** @class */ (function (_super) {
      * Delete entities in Milvus
      *
      * @param data
-     *  | Property                | Type                   |           Description              |
-     *  | :---------------------- | :--------------------  | :-------------------------------  |
-     *  | collection_name         | String                 |       Collection name       |
-     *  | partition_name(optional)| String                 |       Partition name       |
-     *  | expr    | String        |  Boolean expression used to filter attribute.    |
-     *  | 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       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_name | String | Collection name |
+     *  | partition_name(optional)| String | Partition name |
+     *  | expr | String | Boolean expression used to filter attribute. |
+     *  | 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 |
+  
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-------------| :-------------------------------  |
-     *  | status        |  { error_code: number, reason: string }|
-     *  | IDs    |        ID array of the successfully deleted data      |
+     * @returns
+     * | Property | Description |
+     *  | :-- | :-- |
+     *  | status |  { error_code: number, reason: string } |
+     *  | IDs | ID array of the successfully deleted data |
      *
      *
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.deleteEntities({
+     *  new milvusClient(MILUVS_ADDRESS).deleteEntities({
      *    collection_name: COLLECTION_NAME,
      *    expr: 'id in [1,2,3,4]'
      *  });
@@ -289,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];
@@ -303,30 +292,31 @@ var Data = /** @class */ (function (_super) {
      * Perform vector similarity search.
      *
      * @param data
-     *  | Property                | Type                   |           Description              |
-     *  | :---------------------- | :--------------------  | :-------------------------------  |
-     *  | collection_name         | String                 |        Collection name       |
-     *  | partition_names(optional)| String[]              |        Array of partition names       |
-     *  | expr(optional)           | String                |      Scalar field filter expression    |
-     *  | search_params            | Object        |    anns_field: vector field name <br/> topk: search result counts <br/> [metric_type](https://milvus.io/docs/v2.0.0/metric.md#floating#Similarity-Metrics) <br/>params: search params   |
-     *  | vectors                  | Number[][]            |  Original vector to search with  |
-     *  | output_fields(optional)  | String[]              |  Support scalar field  |
-     *  | vector_type              | enum                  |  Binary field -> 100, Float field -> 101  |
-     *  | travel_timestamp          | number                  |  We can get timestamp after insert success. Use this timestamp we can time travel in vector search.|
-     *  | 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       |
-     *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-------------| :-------------------------------  |
-     *  | status        |  { error_code: number, reason: string }|
-     *  | results    |        {score:number,id:string}[];       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_name | String | Collection name |
+     *  | partition_names(optional)| String[] | Array of partition names |
+     *  | expr(optional) | String | Scalar field filter expression |
+     *  | search_params | Object | anns_field: vector field name <br/> topk: search result counts <br/> [metric_type](https://milvus.io/docs/v2.0.0/metric.md#floating#Similarity-Metrics) <br/>params: search params |
+     *  | vectors | Number[][] | Original vector to search with |
+     *  | output_fields(optional) | String[] |  Support scalar field |
+     *  | vector_type | enum | Binary field -> 100, Float field -> 101 |
+     *  | travel_timestamp | number |  We can get timestamp after insert success. Use this timestamp we can time travel in vector search.|
+     *  | 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 } |
+     *  | results | {score:number,id:string}[]; |
      *
      *
      *
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.search({
+     *  new milvusClient(MILUVS_ADDRESS).search({
      *   collection_name: COLLECTION_NAME,
      *   expr: "",
      *   vectors: [[1, 2, 3, 4]],
@@ -348,54 +338,54 @@ var Data = /** @class */ (function (_super) {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        this.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 = Common_1.DataTypeMap[targetField.data_type.toLowerCase()];
-                        dimension = vectorType === Common_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 === Common_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: Common_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 = [];
@@ -429,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] : '',
@@ -453,20 +443,21 @@ var Data = /** @class */ (function (_super) {
      * Milvus temporarily buffers the newly inserted vectors in the cache. Call `flush()` to persist them to the object storage.
      * It's async function, so it's will take some times to excute.
      * @param data
-     *  | Property                | Type   |           Description              |
-     *  | :---------------------- | :----  | :-------------------------------  |
-     *  | collection_names        | String[] |        Array of collection names      |
-     *  | 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       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_names | String[] | Array of collection names |
+     *  | 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 |
+  
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-------------| :-------------------------------  |
-     *  | status        |  { error_code: number, reason: string }|
+     * @returns
+     * | Property | Description |
+     *  | :-- | :-- |
+     *  | status | { error_code: number, reason: string } |
      *
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.flush({
+     *  new milvusClient(MILUVS_ADDRESS).flush({
      *     collection_names: ['my_collection'],
      *  });
      * ```
@@ -480,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];
@@ -495,20 +486,21 @@ var Data = /** @class */ (function (_super) {
      * So you can ensure it's flushed after function return the result.
      *
      * @param data
-     *  | Property                | Type   |           Description              |
-     *  | :---------------------- | :----  | :-------------------------------  |
-     *  | collection_names        | String[] |        Array of collection names      |
-     *  | 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       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_names | String[] | Array of collection names |
+     *  | 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 |
+  
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-------------| :-------------------------------  |
-     *  | status        |  { error_code: number, reason: string }|
+     * @returns
+     * | Property | Description |
+     *  | :-- | :-- |
+     *  | status |  { error_code: number, reason: string } |
      *
      * #### Example
      *
      * ```
-     *  new milvusClient(MILUVS_ADDRESS).dataManager.flushSync({
+     *  new milvusClient(MILUVS_ADDRESS).flushSync({
      *     collection_names: ['my_collection'],
      *  });
      * ```
@@ -522,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)
@@ -538,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;
@@ -554,26 +546,27 @@ var Data = /** @class */ (function (_super) {
      * Query vector data in Milvus. Current release of Milvus only supports expression as fieldname in [id1,id2,id3]
      *
      * @param data
-     *  | Property                     | Type   |           Description              |
-     *  | :--------------------------- | :----  | :-------------------------------  |
-     *  | collection_name              | String |        Collection name      |
-     *  | expr                         | String |       Scalar field filter expression     |
-     *  | partitions_names(optional)   | String[] |       Array of partition names      |
-     *  | output_fields                | String[] |       Vector or scalar field to be returned    |
-     *  | 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       |
+     *  | Property | Type  | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_name | String | Collection name |
+     *  | expr | String | Scalar field filter expression |
+     *  | partitions_names(optional) | String[] | Array of partition names |
+     *  | output_fields | String[] | Vector or scalar field to be returned |
+     *  | 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 |
+  
      *  | params | {key: value}[] | An optional key pair json array
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-------------| :-------------------------------  |
-     *  | status        |  { error_code: number,reason:string } |
-     *  | data   |  Data of all fields that you defined in `output_fields`, {field_name: value}[] |
+     * @returns
+     * | Property | Description |
+     *  | :-- | :-- |
+     *  | status | { error_code: number,reason:string } |
+     *  | data | Data of all fields that you defined in `output_fields`, {field_name: value}[] |
      *
      *
      * #### 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"],
@@ -586,14 +579,14 @@ var Data = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.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 = [];
@@ -654,9 +647,9 @@ var Data = /** @class */ (function (_super) {
     /**
      * @ignore
      * @param data
-     *  | Property             | Type   |           Description                      |
-     *  | :------------------- | :----  | :----------------------------------------  |
-     *  | request              | object |        Only allow "system_info" for now    |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | request | object | Only allow "system_info" for now |
      */
     Data.prototype.getMetric = function (data) {
         return __awaiter(this, void 0, void 0, function () {
@@ -665,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:
@@ -681,23 +674,24 @@ var Data = /** @class */ (function (_super) {
      * Get flush state by segment ids
      *
      * @param data
-     *  | Property                | Type   |           Description              |
-     *  | :---------------------- | :----  | :-------------------------------  |
-     *  | segmentIDs              | Array  |       The segment ids        |
-     *  | 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       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | segmentIDs | Array | The segment ids |
+     *  | 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 |
+  
      *
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-----------| :-------------------------------  |
-     *  | status      |  { error_code: number,reason:string } |
-     *  | flushed     |  segments flushed or not  |
+     * @returns
+     * | Property | Description |
+     *  | :--- | :-- |
+     *  | status | { error_code: number,reason:string } |
+     *  | flushed | segments flushed or not |
      *
      *
      * #### Example
      *
      * ```
-     *   const res = await milvusClient.dataManager.getFlushState({
+     *   const res = await milvusClient.getFlushState({
      *    segmentIDs: segIds,
      *   });
      * ```
@@ -709,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];
@@ -724,24 +718,24 @@ var Data = /** @class */ (function (_super) {
      * Only work in cluster milvus.
      *
      * @param data
-     *  | Property            | Type     |           Description              |
-     *  | :-------------------| :----    | :-------------------------------  |
-     *  | src_nodeID          | number   |     The source query node id to balance.        |
-     *  | dst_nodeIDs         | number[] |     The destination query node ids to balance.(optional)        |
-     *  | sealed_segmentIDs   | number[] |     Sealed segment ids to balance.(optional)       |
-     *  | 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       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | src_nodeID | number | The source query node id to balance. |
+     *  | dst_nodeIDs | number[] | The destination query node ids to balance.(optional) |
+     *  | sealed_segmentIDs | number[] | Sealed segment ids to balance.(optional) |
+     *  | 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 |
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-----------| :-------------------------------  |
-     *  | status      |  { error_code: number,reason:string } |
-     *  | infos       |  segments infomations  |
+     * @returns
+     * | Property | Description |
+     *  | :--- | :-- |
+     *  | status | { error_code: number,reason:string } |
+     *  | infos | segments infomations |
      *
      *
      * #### Example
      *
      * ```
-     *   const res = await dataManager.loadBalance({
+     *   const res = await loadBalance({
      *      src_nodeID: 31,
      *   });
      * ```
@@ -753,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];
@@ -767,23 +761,23 @@ var Data = /** @class */ (function (_super) {
      * Notifies Proxy to return segments information from query nodes.
      *
      * @param data
-     *  | Property                | Type   |           Description              |
-     *  | :---------------------- | :----  | :-------------------------------  |
-     *  | collectionName          | String |      The name of the collection to get segments info.       |
-     *  | 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       |
+     *  | Property | Type  | Description |
+     *  | :--- | :-- | :-- |
+     *  | collectionName | String | The name of the collection to get segments info. |
+     *  | 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 |
      *
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-----------| :-------------------------------  |
-     *  | status      |  { error_code: number,reason:string } |
-     *  | infos       |  QuerySegmentInfo is the growing segments's information in query cluster.  |
+     * @returns
+     * | Property | Description |
+     *  | :--- | :-- |
+     *  | status | { error_code: number,reason:string } |
+     *  | infos | QuerySegmentInfo is the growing segments's information in query cluster. |
      *
      *
      * #### Example
      *
      * ```
-     *   const res = await dataManager.getQuerySegmentInfo({
+     *   const res = await getQuerySegmentInfo({
      *      collectionName: COLLECTION,
      *    });
      * ```
@@ -795,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];
@@ -809,28 +803,29 @@ var Data = /** @class */ (function (_super) {
      * Import data from files
      *
      * @param data
-     *  | Property                | Type   |           Description              |
-     *  | :---------------------- | :----  | :-------------------------------  |
-     *  | collection_name          | String |      The name of the collection      |
-     *  | files        | string[] |        File path array       |
+     *  | Property | Type | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_name | String | The name of the collection |
+     *  | files | string[] | File path array |
      *
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-----------| :-------------------------------  |
-     *  | status      |  { error_code: number,reason:string } |
-     *  | tasks       |  taskId array  |
+     * @returns
+     * | Property | Description |
+     *  | :--- | :-- |
+     *  | status | { error_code: number,reason:string } |
+     *  | tasks | taskId array |
      *
      *
      * #### 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;
@@ -838,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];
@@ -855,32 +850,32 @@ var Data = /** @class */ (function (_super) {
      * List import tasks
      *
      * @param data
-     *  | Property                | Type   |           Description              |
-     *  | :---------------------- | :----  | :-------------------------------  |
-     *  | collection_name          | String |      The name of the collection       |
-     *  | limit        | number |       optional, maximum number of tasks returned, list all tasks if the value is 0       |
+     *  | Property | Type  | Description |
+     *  | :--- | :-- | :-- |
+     *  | collection_name | String | The name of the collection |
+     *  | limit | number | optional, maximum number of tasks returned, list all tasks if the value is 0 |
      *
      *
-     * @return
-     *  | Property    |           Description              |
-     *  | :-----------| :-------------------------------  |
-     *  | status      |  { error_code: number,reason:string } |
+     * @returns
+     * | Property | Description |
+     *  | :--- | :-- |
+     *  | status | { error_code: number,reason:string } |
      *  | state | import state |
      *  | row_count | how many rows to import|
      *  | id_list| id lists |
      *  | collection_id | collection to be imported to |
-     *  |
-     *  | tasks       |  taskId array  |
+     *  | tasks | taskId array  |
      *
      *
      * #### 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;
@@ -888,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];
@@ -899,6 +894,6 @@ var Data = /** @class */ (function (_super) {
         });
     };
     return Data;
-}(Client_1.Client));
+}(Collection_1.Collection));
 exports.Data = Data;
 //# sourceMappingURL=Data.js.map