@zilliz/milvus2-sdk-node 1.0.17 → 1.1.0

package/dist/milvus/Data.d.ts

@@ -1,7 +1,7 @@
 import { Client } from "./Client";
 import { Collection } from "./Collection";
-import { FlushReq, InsertReq } from "./types/Insert";
-import { FlushResult, GetMetricsResponse, MutationResult, QueryResults, SearchResults } from "./types/Response";
+import { CalcDistanceReq, DeleteEntitiesReq, FlushReq, GetFlushStateReq, GetQuerySegmentInfoReq, InsertReq, LoadBalanceReq } from "./types/Data";
+import { CalcDistanceResponse, FlushResult, GetFlushStateResponse, GetMetricsResponse, GetQuerySegmentInfoResponse, MutationResult, QueryResults, ResStatus, SearchResults } from "./types/Response";
 import { GetMetricsRequest, QueryReq, SearchReq } from "./types/Search";
 export declare class Data extends Client {
     vectorTypes: number[];
@@ -40,6 +40,33 @@ export declare class Data extends Client {
      * ```
      */
     insert(data: InsertReq): Promise<MutationResult>;
+    /**
+     * 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.    |
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-------------| :-------------------------------  |
+     *  | status        |  { error_code: number, reason: string }|
+     *  | IDs    |        ID array of the successfully deleted data      |
+     *
+     *
+     * #### Example
+     *
+     * ```
+     *  new milvusClient(MILUVS_ADDRESS).dataManager.deleteEntities({
+     *    collection_name: COLLECTION_NAME,
+     *    expr: 'id in [1,2,3,4]'
+     *  });
+     * ```
+     */
+    deleteEntities(data: DeleteEntitiesReq): Promise<MutationResult>;
     /**
      * Perform vector similarity search.
      *
@@ -53,14 +80,14 @@ export declare class Data extends Client {
      *  | 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.|
   
      * @return
      *  | Property    |           Description              |
      *  | :-------------| :-------------------------------  |
      *  | status        |  { error_code: number, reason: string }|
-     *  | succ_index    |        Insert successful index array      |
-     *  | err_index    |        Insert failed index array      |
-     *  | IDs    |        Insert successful id array      |
+     *  | results    |        {score:number,id:string}[];       |
+     *
      *
      *
      * #### Example
@@ -84,7 +111,7 @@ export declare class Data extends Client {
     search(data: SearchReq): Promise<SearchResults>;
     /**
      * 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              |
      *  | :---------------------- | :----  | :-------------------------------  |
@@ -104,6 +131,29 @@ export declare class Data extends Client {
      * ```
      */
     flush(data: FlushReq): Promise<FlushResult>;
+    /**
+     * It's same function as flush. But flushSync is sync function.
+     * So you can ensure it's flushed after function return the result.
+     *
+     * @param data
+     *  | Property                | Type   |           Description              |
+     *  | :---------------------- | :----  | :-------------------------------  |
+     *  | collection_names        | String[] |        Array of collection names      |
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-------------| :-------------------------------  |
+     *  | status        |  { error_code: number, reason: string }|
+     *
+     * #### Example
+     *
+     * ```
+     *  new milvusClient(MILUVS_ADDRESS).dataManager.flushSync({
+     *     collection_names: ['my_collection'],
+     *  });
+     * ```
+     */
+    flushSync(data: FlushReq): Promise<GetFlushStateResponse>;
     /**
      * Query vector data in Milvus. Current release of Milvus only supports expression as fieldname in [id1,id2,id3]
      *
@@ -143,4 +193,89 @@ export declare class Data extends Client {
      *  | request              | object |        Only allow "system_info" for now    |
      */
     getMetric(data: GetMetricsRequest): Promise<GetMetricsResponse>;
+    /**
+     * @ignore
+     * @param data
+     */
+    calcDistance(data: CalcDistanceReq): Promise<CalcDistanceResponse>;
+    /**
+     * Get flush state by segment ids
+     *
+     * @param data
+     *  | Property                | Type   |           Description              |
+     *  | :---------------------- | :----  | :-------------------------------  |
+     *  | segmentIDs              | Array  |       The segment ids        |
+     *
+     *
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-----------| :-------------------------------  |
+     *  | status      |  { error_code: number,reason:string } |
+     *  | flushed     |  segments flushed or not  |
+     *
+     *
+     * #### Example
+     *
+     * ```
+     *   const res = await milvusClient.dataManager.getFlushState({
+     *    segmentIDs: segIds,
+     *   });
+     * ```
+     */
+    getFlushState(data: GetFlushStateReq): Promise<GetFlushStateResponse>;
+    /**
+     * Do load balancing operation from source query node to destination query node.
+     * 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)       |
+     *
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-----------| :-------------------------------  |
+     *  | status      |  { error_code: number,reason:string } |
+     *  | infos       |  segments infomations  |
+     *
+     *
+     * #### Example
+     *
+     * ```
+     *   const res = await dataManager.loadBalance({
+     *      src_nodeID: 31,
+     *   });
+     * ```
+     */
+    loadBalance(data: LoadBalanceReq): Promise<ResStatus>;
+    /**
+     * 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.       |
+     *
+     *
+     *
+     * @return
+     *  | 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({
+     *      collectionName: COLLECTION,
+     *    });
+     * ```
+     */
+    getQuerySegmentInfo(data: GetQuerySegmentInfoReq): Promise<GetQuerySegmentInfoResponse>;
 }

package/dist/milvus/Data.js

@@ -61,6 +61,11 @@ 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 __spreadArray = (this && this.__spreadArray) || function (to, from) {
+    for (var i = 0, il = from.length, j = to.length; i < il; i++, j++)
+        to[j] = from[i];
+    return to;
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
@@ -72,7 +77,7 @@ var Client_1 = require("./Client");
 var ErrorReason_1 = require("./const/ErrorReason");
 var Common_1 = require("./types/Common");
 var Response_1 = require("./types/Response");
-var utils_2 = require("./utils");
+var index_1 = require("./utils/index");
 var Blob_1 = require("./utils/Blob");
 var path_1 = __importDefault(require("path"));
 var Format_1 = require("./utils/Format");
@@ -124,6 +129,12 @@ var Data = /** @class */ (function (_super) {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        this.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);
+                        }
                         collection_name = data.collection_name;
                         return [4 /*yield*/, this.collectionManager.describeCollection({
                                 collection_name: collection_name,
@@ -138,7 +149,7 @@ var Data = /** @class */ (function (_super) {
                             .map(function (v) { return ({
                             name: v.name,
                             type: v.data_type,
-                            dim: Number(utils_2.findKeyValue(v.type_params, "dim")),
+                            dim: Number(index_1.findKeyValue(v.type_params, "dim")),
                             value: [],
                         }); });
                         params = __assign(__assign({}, data), { num_rows: data.fields_data.length });
@@ -175,8 +186,6 @@ var Data = /** @class */ (function (_super) {
                             // 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()];
-                            if (!type) {
-                            }
                             var key = _this.vectorTypes.includes(type) ? "vectors" : "scalars";
                             var dataKey = "float_vector";
                             switch (type) {
@@ -200,8 +209,11 @@ var Data = /** @class */ (function (_super) {
                                 case Common_1.DataType.Int8:
                                     dataKey = "int_data";
                                     break;
-                                default:
+                                case Common_1.DataType.Bool:
+                                    dataKey = "bool_data";
                                     break;
+                                default:
+                                    throw new Error(ErrorReason_1.ERROR_REASONS.INSERT_CHECK_WRONG_DATA_TYPE);
                             }
                             return _a = {
                                     type: type,
@@ -234,6 +246,49 @@ 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.    |
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-------------| :-------------------------------  |
+     *  | status        |  { error_code: number, reason: string }|
+     *  | IDs    |        ID array of the successfully deleted data      |
+     *
+     *
+     * #### Example
+     *
+     * ```
+     *  new milvusClient(MILUVS_ADDRESS).dataManager.deleteEntities({
+     *    collection_name: COLLECTION_NAME,
+     *    expr: 'id in [1,2,3,4]'
+     *  });
+     * ```
+     */
+    Data.prototype.deleteEntities = function (data) {
+        return __awaiter(this, void 0, void 0, function () {
+            var promise;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        if (!data || !data.collection_name || !data.expr) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.DELETE_PARAMS_CHECK);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "Delete", data)];
+                    case 1:
+                        promise = _a.sent();
+                        return [2 /*return*/, promise];
+                }
+            });
+        });
+    };
     /**
      * Perform vector similarity search.
      *
@@ -247,14 +302,14 @@ var Data = /** @class */ (function (_super) {
      *  | 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.|
   
      * @return
      *  | Property    |           Description              |
      *  | :-------------| :-------------------------------  |
      *  | status        |  { error_code: number, reason: string }|
-     *  | succ_index    |        Insert successful index array      |
-     *  | err_index    |        Insert failed index array      |
-     *  | IDs    |        Insert successful id array      |
+     *  | results    |        {score:number,id:string}[];       |
+     *
      *
      *
      * #### Example
@@ -278,14 +333,20 @@ var Data = /** @class */ (function (_super) {
     Data.prototype.search = function (data) {
         var _a;
         return __awaiter(this, void 0, void 0, function () {
-            var root, collectionInfo, targetField, dim, vectorType, dimension, PlaceholderGroup, placeholderGroupParams, placeholderGroupBytes, promise, results, _b, topks, scores_1, fields_data, ids, fieldsData_1, idData_1;
+            var root, collectionInfo, targetField, dim, vectorType, dimension, PlaceholderGroup, placeholderGroupParams, placeholderGroupBytes, promise, results, round_decimal, _b, topks, scores_1, fields_data, ids, fieldsData_1, idData_1;
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0: return [4 /*yield*/, protobufjs_1.default.load(protoPath)];
                     case 1:
                         root = _c.sent();
-                        if (!root)
-                            throw new Error("Missing milvus proto file");
+                        this.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);
+                        }
                         if (!this.vectorTypes.includes(data.vector_type))
                             throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_MISS_VECTOR_TYPE);
                         return [4 /*yield*/, this.collectionManager.describeCollection({
@@ -297,7 +358,7 @@ var Data = /** @class */ (function (_super) {
                         if (!targetField) {
                             throw new Error(ErrorReason_1.ERROR_REASONS.SEARCH_NOT_FIND_VECTOR_FIELD);
                         }
-                        dim = utils_2.findKeyValue(targetField.type_params, "dim");
+                        dim = 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);
                         if (!data.vectors[0] || data.vectors[0].length !== dimension) {
@@ -322,6 +383,7 @@ var Data = /** @class */ (function (_super) {
                     case 3:
                         promise = _c.sent();
                         results = [];
+                        round_decimal = data.search_params.round_decimal;
                         if (promise.results) {
                             _b = promise.results, topks = _b.topks, scores_1 = _b.scores, fields_data = _b.fields_data, ids = _b.ids;
                             fieldsData_1 = fields_data.map(function (item, i) {
@@ -333,7 +395,7 @@ var Data = /** @class */ (function (_super) {
                                     data: value ? value[value === null || value === void 0 ? void 0 : value.data].data : "",
                                 };
                             });
-                            idData_1 = (_a = ids[ids.id_field]) === null || _a === void 0 ? void 0 : _a.data;
+                            idData_1 = ids ? (_a = ids[ids.id_field]) === null || _a === void 0 ? void 0 : _a.data : undefined;
                             /**
                              *  milvus support mutilple querys to search
                              *  milvus will return all columns data
@@ -344,8 +406,11 @@ var Data = /** @class */ (function (_super) {
                                 var topk = Number(v);
                                 scores_1.splice(0, topk).forEach(function (score, scoreIndex) {
                                     var i = index === 0 ? scoreIndex : scoreIndex + topk;
+                                    var fixedScore = typeof round_decimal === "undefined" || round_decimal === -1
+                                        ? score
+                                        : Format_1.formatNumberPrecision(score, round_decimal);
                                     var result = {
-                                        score: score,
+                                        score: fixedScore,
                                         id: idData_1 ? idData_1[i] : "",
                                     };
                                     fieldsData_1.forEach(function (field) {
@@ -365,7 +430,7 @@ 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              |
      *  | :---------------------- | :----  | :-------------------------------  |
@@ -389,7 +454,13 @@ var Data = /** @class */ (function (_super) {
             var res;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, utils_1.promisify(this.client, "Flush", data)];
+                    case 0:
+                        if (!data ||
+                            !Array.isArray(data.collection_names) ||
+                            !data.collection_names.length) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "Flush", data)];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, res];
@@ -397,6 +468,65 @@ var Data = /** @class */ (function (_super) {
             });
         });
     };
+    /**
+     * It's same function as flush. But flushSync is sync function.
+     * So you can ensure it's flushed after function return the result.
+     *
+     * @param data
+     *  | Property                | Type   |           Description              |
+     *  | :---------------------- | :----  | :-------------------------------  |
+     *  | collection_names        | String[] |        Array of collection names      |
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-------------| :-------------------------------  |
+     *  | status        |  { error_code: number, reason: string }|
+     *
+     * #### Example
+     *
+     * ```
+     *  new milvusClient(MILUVS_ADDRESS).dataManager.flushSync({
+     *     collection_names: ['my_collection'],
+     *  });
+     * ```
+     */
+    Data.prototype.flushSync = function (data) {
+        return __awaiter(this, void 0, void 0, function () {
+            var res, segIDs, isFlushed, flushRes;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        if (!data ||
+                            !Array.isArray(data.collection_names) ||
+                            !data.collection_names.length) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "Flush", data)];
+                    case 1:
+                        res = _a.sent();
+                        segIDs = Object.keys(res.coll_segIDs)
+                            .map(function (v) { return res.coll_segIDs[v].data; })
+                            .reduce(function (pre, cur) { return __spreadArray(__spreadArray([], pre), cur); }, []);
+                        isFlushed = false;
+                        flushRes = null;
+                        _a.label = 2;
+                    case 2:
+                        if (!!isFlushed) return [3 /*break*/, 5];
+                        return [4 /*yield*/, this.getFlushState({ segmentIDs: segIDs })];
+                    case 3:
+                        flushRes = _a.sent();
+                        return [4 /*yield*/, index_1.sleep(100)];
+                    case 4:
+                        _a.sent();
+                        isFlushed = flushRes.flushed;
+                        return [3 /*break*/, 2];
+                    case 5: 
+                    // Before Milvus pre-GA will throw error
+                    return [2 /*return*/, flushRes];
+                }
+            });
+        });
+    };
     /**
      * Query vector data in Milvus. Current release of Milvus only supports expression as fieldname in [id1,id2,id3]
      *
@@ -432,7 +562,9 @@ var Data = /** @class */ (function (_super) {
             var promise, results, fieldsData;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, utils_1.promisify(this.client, "Query", data)];
+                    case 0:
+                        this.checkCollectionName(data);
+                        return [4 /*yield*/, utils_1.promisify(this.client, "Query", data)];
                     case 1:
                         promise = _a.sent();
                         results = [];
@@ -440,7 +572,9 @@ var Data = /** @class */ (function (_super) {
                             var _a;
                             if (item.field === "vectors") {
                                 var key_1 = item.vectors.data;
-                                var vectorValue = item.vectors[key_1].data;
+                                var vectorValue = key_1 === "float_vector"
+                                    ? item.vectors[key_1].data
+                                    : item.vectors[key_1].toJSON().data;
                                 // if binary vector , need use dim / 8 to split vector data
                                 var dim_1 = ((_a = item.vectors) === null || _a === void 0 ? void 0 : _a.data) === "float_vector"
                                     ? Number(item.vectors.dim)
@@ -500,9 +634,13 @@ var Data = /** @class */ (function (_super) {
             var res;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, utils_1.promisify(this.client, "GetMetrics", {
-                            request: JSON.stringify(data.request),
-                        })];
+                    case 0:
+                        if (!data || !data.request || !data.request.metric_type) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.GET_METRIC_CHECK_PARAMS);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "GetMetrics", {
+                                request: JSON.stringify(data.request),
+                            })];
                     case 1:
                         res = _a.sent();
                         return [2 /*return*/, __assign(__assign({}, res), { response: JSON.parse(res.response) })];
@@ -510,6 +648,151 @@ var Data = /** @class */ (function (_super) {
             });
         });
     };
+    /**
+     * @ignore
+     * @param data
+     */
+    Data.prototype.calcDistance = function (data) {
+        return __awaiter(this, void 0, void 0, function () {
+            var res;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, utils_1.promisify(this.client, "CalcDistance", data)];
+                    case 1:
+                        res = _a.sent();
+                        return [2 /*return*/, res];
+                }
+            });
+        });
+    };
+    /**
+     * Get flush state by segment ids
+     *
+     * @param data
+     *  | Property                | Type   |           Description              |
+     *  | :---------------------- | :----  | :-------------------------------  |
+     *  | segmentIDs              | Array  |       The segment ids        |
+     *
+     *
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-----------| :-------------------------------  |
+     *  | status      |  { error_code: number,reason:string } |
+     *  | flushed     |  segments flushed or not  |
+     *
+     *
+     * #### Example
+     *
+     * ```
+     *   const res = await milvusClient.dataManager.getFlushState({
+     *    segmentIDs: segIds,
+     *   });
+     * ```
+     */
+    Data.prototype.getFlushState = 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.segmentIDs) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.GET_FLUSH_STATE_CHECK_PARAMS);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "GetFlushState", data)];
+                    case 1:
+                        res = _a.sent();
+                        return [2 /*return*/, res];
+                }
+            });
+        });
+    };
+    /**
+     * Do load balancing operation from source query node to destination query node.
+     * 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)       |
+     *
+     *
+     * @return
+     *  | Property    |           Description              |
+     *  | :-----------| :-------------------------------  |
+     *  | status      |  { error_code: number,reason:string } |
+     *  | infos       |  segments infomations  |
+     *
+     *
+     * #### Example
+     *
+     * ```
+     *   const res = await dataManager.loadBalance({
+     *      src_nodeID: 31,
+     *   });
+     * ```
+     */
+    Data.prototype.loadBalance = 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.src_nodeID) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.LOAD_BALANCE_CHECK_PARAMS);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "LoadBalance", data)];
+                    case 1:
+                        res = _a.sent();
+                        return [2 /*return*/, res];
+                }
+            });
+        });
+    };
+    /**
+     * 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.       |
+     *
+     *
+     *
+     * @return
+     *  | 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({
+     *      collectionName: COLLECTION,
+     *    });
+     * ```
+     */
+    Data.prototype.getQuerySegmentInfo = 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.collectionName) {
+                            throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
+                        }
+                        return [4 /*yield*/, utils_1.promisify(this.client, "GetQuerySegmentInfo", data)];
+                    case 1:
+                        res = _a.sent();
+                        return [2 /*return*/, res];
+                }
+            });
+        });
+    };
     return Data;
 }(Client_1.Client));
 exports.Data = Data;