@zilliz/milvus2-sdk-node 2.2.8 → 2.2.10

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (76) hide show
  1. package/README.md +83 -7
  2. package/dist/milvus/MilvusClient.d.ts +24 -12
  3. package/dist/milvus/MilvusClient.js +30 -124
  4. package/dist/milvus/MilvusClient.js.map +1 -1
  5. package/dist/milvus/const/Milvus.d.ts +3 -0
  6. package/dist/milvus/const/Milvus.js +4 -1
  7. package/dist/milvus/const/Milvus.js.map +1 -1
  8. package/dist/milvus/const/index.d.ts +2 -0
  9. package/dist/milvus/const/index.js +20 -0
  10. package/dist/milvus/const/index.js.map +1 -0
  11. package/dist/milvus/{BaseClient.d.ts → grpc/BaseClient.d.ts} +10 -4
  12. package/dist/milvus/grpc/BaseClient.js +64 -0
  13. package/dist/milvus/grpc/BaseClient.js.map +1 -0
  14. package/dist/milvus/{Collection.d.ts → grpc/Collection.d.ts} +1 -1
  15. package/dist/milvus/{Collection.js → grpc/Collection.js} +35 -35
  16. package/dist/milvus/grpc/Collection.js.map +1 -0
  17. package/dist/milvus/{Data.d.ts → grpc/Data.d.ts} +1 -3
  18. package/dist/milvus/{Data.js → grpc/Data.js} +66 -68
  19. package/dist/milvus/grpc/Data.js.map +1 -0
  20. package/dist/milvus/grpc/GrpcClient.d.ts +36 -0
  21. package/dist/milvus/grpc/GrpcClient.js +219 -0
  22. package/dist/milvus/grpc/GrpcClient.js.map +1 -0
  23. package/dist/milvus/{MilvusIndex.d.ts → grpc/MilvusIndex.d.ts} +1 -1
  24. package/dist/milvus/{MilvusIndex.js → grpc/MilvusIndex.js} +7 -7
  25. package/dist/milvus/grpc/MilvusIndex.js.map +1 -0
  26. package/dist/milvus/{Partition.d.ts → grpc/Partition.d.ts} +1 -1
  27. package/dist/milvus/{Partition.js → grpc/Partition.js} +11 -11
  28. package/dist/milvus/grpc/Partition.js.map +1 -0
  29. package/dist/milvus/{Resource.d.ts → grpc/Resource.d.ts} +1 -1
  30. package/dist/milvus/{Resource.js → grpc/Resource.js} +10 -10
  31. package/dist/milvus/grpc/Resource.js.map +1 -0
  32. package/dist/milvus/{User.d.ts → grpc/User.d.ts} +1 -1
  33. package/dist/milvus/{User.js → grpc/User.js} +25 -24
  34. package/dist/milvus/grpc/User.js.map +1 -0
  35. package/dist/milvus/index.d.ts +4 -11
  36. package/dist/milvus/index.js +6 -12
  37. package/dist/milvus/index.js.map +1 -1
  38. package/dist/milvus/types/Client.d.ts +7 -1
  39. package/dist/milvus/types/Collection.d.ts +1 -1
  40. package/dist/milvus/types/Common.d.ts +1 -1
  41. package/dist/milvus/types/Index.d.ts +10 -59
  42. package/dist/milvus/types/Index.js +25 -0
  43. package/dist/milvus/types/Index.js.map +1 -1
  44. package/dist/milvus/types/MilvusIndex.d.ts +60 -0
  45. package/dist/milvus/types/MilvusIndex.js +3 -0
  46. package/dist/milvus/types/MilvusIndex.js.map +1 -0
  47. package/dist/milvus/types/User.d.ts +1 -1
  48. package/dist/milvus/types.d.ts +1 -1
  49. package/dist/milvus/types.js +2 -1
  50. package/dist/milvus/types.js.map +1 -1
  51. package/dist/proto/proto/google/protobuf/descriptor.proto +1 -1
  52. package/dist/sdk.json +1 -1
  53. package/dist/utils/Format.d.ts +7 -0
  54. package/dist/utils/Format.js +12 -1
  55. package/dist/utils/Format.js.map +1 -1
  56. package/dist/utils/Grpc.d.ts +14 -0
  57. package/dist/utils/Grpc.js +107 -1
  58. package/dist/utils/Grpc.js.map +1 -1
  59. package/dist/utils/Validate.d.ts +11 -0
  60. package/dist/utils/Validate.js +21 -1
  61. package/dist/utils/Validate.js.map +1 -1
  62. package/dist/utils/test.d.ts +14 -0
  63. package/dist/utils/test.js +14 -0
  64. package/dist/utils/test.js.map +1 -1
  65. package/package.json +2 -2
  66. package/dist/milvus/BaseClient.js +0 -87
  67. package/dist/milvus/BaseClient.js.map +0 -1
  68. package/dist/milvus/Collection.js.map +0 -1
  69. package/dist/milvus/Data.js.map +0 -1
  70. package/dist/milvus/MilvusIndex.js.map +0 -1
  71. package/dist/milvus/Partition.js.map +0 -1
  72. package/dist/milvus/Resource.js.map +0 -1
  73. package/dist/milvus/User.js.map +0 -1
  74. package/dist/milvus/Utils.d.ts +0 -1
  75. package/dist/milvus/Utils.js +0 -8
  76. package/dist/milvus/Utils.js.map +0 -1
@@ -74,10 +74,10 @@ var __rest = (this && this.__rest) || function (s, e) {
74
74
  };
75
75
  Object.defineProperty(exports, "__esModule", { value: true });
76
76
  exports.Collection = void 0;
77
- var utils_1 = require("../utils");
77
+ var utils_1 = require("../../utils");
78
78
  var BaseClient_1 = require("./BaseClient");
79
- var _1 = require(".");
80
- var utils_2 = require("../utils");
79
+ var __1 = require("..");
80
+ var utils_2 = require("../../utils");
81
81
  /**
82
82
  * @see [collection operation examples](https://github.com/milvus-io/milvus-sdk-node/blob/main/example/Collection.ts)
83
83
  */
@@ -139,7 +139,7 @@ var Collection = /** @class */ (function (_super) {
139
139
  _b = data || {}, fields = _b.fields, collection_name = _b.collection_name, description = _b.description, _c = _b.consistency_level, consistency_level = _c === void 0 ? 'Bounded' : _c;
140
140
  // Check if fields and collection_name are present, otherwise throw an error.
141
141
  if (!(fields === null || fields === void 0 ? void 0 : fields.length) || !collection_name) {
142
- throw new Error(_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PARAMS);
142
+ throw new Error(__1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PARAMS);
143
143
  }
144
144
  // Check if the fields are valid.
145
145
  (0, utils_1.checkCollectionFields)(fields);
@@ -156,8 +156,8 @@ var Collection = /** @class */ (function (_super) {
156
156
  };
157
157
  collectionParams = CollectionSchema.create(payload);
158
158
  schemaBytes = CollectionSchema.encode(collectionParams).finish();
159
- level = (_a = _1.ConsistencyLevelEnum[consistency_level]) !== null && _a !== void 0 ? _a : _1.ConsistencyLevelEnum.Bounded;
160
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'CreateCollection', __assign(__assign({}, data), { schema: schemaBytes, consistency_level: level }), data.timeout || this.timeout)];
159
+ level = (_a = __1.ConsistencyLevelEnum[consistency_level]) !== null && _a !== void 0 ? _a : __1.ConsistencyLevelEnum.Bounded;
160
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'CreateCollection', __assign(__assign({}, data), { schema: schemaBytes, consistency_level: level }), data.timeout || this.timeout)];
161
161
  case 1:
162
162
  promise = _d.sent();
163
163
  // Return the promise.
@@ -197,7 +197,7 @@ var Collection = /** @class */ (function (_super) {
197
197
  switch (_a.label) {
198
198
  case 0:
199
199
  (0, utils_1.checkCollectionName)(data);
200
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'HasCollection', data, data.timeout || this.timeout)];
200
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'HasCollection', data, data.timeout || this.timeout)];
201
201
  case 1:
202
202
  promise = _a.sent();
203
203
  return [2 /*return*/, promise];
@@ -234,8 +234,8 @@ var Collection = /** @class */ (function (_super) {
234
234
  var promise, result;
235
235
  return __generator(this, function (_a) {
236
236
  switch (_a.label) {
237
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ShowCollections', {
238
- type: data ? data.type : _1.ShowCollectionsType.All,
237
+ case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ShowCollections', {
238
+ type: data ? data.type : __1.ShowCollectionsType.All,
239
239
  collection_names: (data === null || data === void 0 ? void 0 : data.collection_names) || [],
240
240
  }, (data === null || data === void 0 ? void 0 : data.timeout) || this.timeout)];
241
241
  case 1:
@@ -287,7 +287,7 @@ var Collection = /** @class */ (function (_super) {
287
287
  switch (_a.label) {
288
288
  case 0:
289
289
  (0, utils_1.checkCollectionName)(data);
290
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'DescribeCollection', data, data.timeout || this.timeout)];
290
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DescribeCollection', data, data.timeout || this.timeout)];
291
291
  case 1:
292
292
  promise = _a.sent();
293
293
  return [2 /*return*/, promise];
@@ -327,7 +327,7 @@ var Collection = /** @class */ (function (_super) {
327
327
  switch (_a.label) {
328
328
  case 0:
329
329
  (0, utils_1.checkCollectionName)(data);
330
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetCollectionStatistics', data, data.timeout || this.timeout)];
330
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetCollectionStatistics', data, data.timeout || this.timeout)];
331
331
  case 1:
332
332
  promise = _a.sent();
333
333
  promise.data = (0, utils_1.formatKeyValueData)(promise.stats, ['row_count']);
@@ -369,7 +369,7 @@ var Collection = /** @class */ (function (_super) {
369
369
  switch (_a.label) {
370
370
  case 0:
371
371
  (0, utils_1.checkCollectionName)(data);
372
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'LoadCollection', data, data.timeout || this.timeout)];
372
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'LoadCollection', data, data.timeout || this.timeout)];
373
373
  case 1:
374
374
  promise = _a.sent();
375
375
  return [2 /*return*/, promise];
@@ -410,10 +410,10 @@ var Collection = /** @class */ (function (_super) {
410
410
  switch (_a.label) {
411
411
  case 0:
412
412
  (0, utils_1.checkCollectionName)(data);
413
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'LoadCollection', data, data.timeout || this.timeout)];
413
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'LoadCollection', data, data.timeout || this.timeout)];
414
414
  case 1:
415
415
  promise = _a.sent();
416
- if (promise.error_code !== _1.ErrorCode.SUCCESS) {
416
+ if (promise.error_code !== __1.ErrorCode.SUCCESS) {
417
417
  throw new Error("ErrorCode: ".concat(promise.error_code, ". Reason: ").concat(promise.reason));
418
418
  }
419
419
  loadedPercentage = 0;
@@ -425,7 +425,7 @@ var Collection = /** @class */ (function (_super) {
425
425
  })];
426
426
  case 3:
427
427
  res = _a.sent();
428
- if (res.status.error_code !== _1.ErrorCode.SUCCESS) {
428
+ if (res.status.error_code !== __1.ErrorCode.SUCCESS) {
429
429
  throw new Error("ErrorCode: ".concat(res.status.error_code, ". Reason: ").concat(res.status.reason));
430
430
  }
431
431
  loadedPercentage = Number(res.progress);
@@ -471,7 +471,7 @@ var Collection = /** @class */ (function (_super) {
471
471
  switch (_a.label) {
472
472
  case 0:
473
473
  (0, utils_1.checkCollectionName)(data);
474
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ReleaseCollection', data, data.timeout || this.timeout)];
474
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ReleaseCollection', data, data.timeout || this.timeout)];
475
475
  case 1:
476
476
  promise = _a.sent();
477
477
  return [2 /*return*/, promise];
@@ -509,7 +509,7 @@ var Collection = /** @class */ (function (_super) {
509
509
  var promise;
510
510
  return __generator(this, function (_a) {
511
511
  switch (_a.label) {
512
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'RenameCollection', {
512
+ case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'RenameCollection', {
513
513
  oldName: data.collection_name,
514
514
  newName: data.new_collection_name,
515
515
  }, data.timeout || this.timeout)];
@@ -550,7 +550,7 @@ var Collection = /** @class */ (function (_super) {
550
550
  switch (_a.label) {
551
551
  case 0:
552
552
  (0, utils_1.checkCollectionName)(data);
553
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'DropCollection', data, data.timeout || this.timeout)];
553
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DropCollection', data, data.timeout || this.timeout)];
554
554
  case 1:
555
555
  promise = _a.sent();
556
556
  return [2 /*return*/, promise];
@@ -592,9 +592,9 @@ var Collection = /** @class */ (function (_super) {
592
592
  case 0:
593
593
  (0, utils_1.checkCollectionName)(data);
594
594
  if (!data.alias) {
595
- throw new Error(_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
595
+ throw new Error(__1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
596
596
  }
597
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'CreateAlias', data, data.timeout || this.timeout)];
597
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'CreateAlias', data, data.timeout || this.timeout)];
598
598
  case 1:
599
599
  promise = _a.sent();
600
600
  return [2 /*return*/, promise];
@@ -635,9 +635,9 @@ var Collection = /** @class */ (function (_super) {
635
635
  switch (_a.label) {
636
636
  case 0:
637
637
  if (!data.alias) {
638
- throw new Error(_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
638
+ throw new Error(__1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
639
639
  }
640
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'DropAlias', data, data.timeout || this.timeout)];
640
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DropAlias', data, data.timeout || this.timeout)];
641
641
  case 1:
642
642
  promise = _a.sent();
643
643
  return [2 /*return*/, promise];
@@ -679,9 +679,9 @@ var Collection = /** @class */ (function (_super) {
679
679
  case 0:
680
680
  (0, utils_1.checkCollectionName)(data);
681
681
  if (!data.alias) {
682
- throw new Error(_1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
682
+ throw new Error(__1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
683
683
  }
684
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'AlterAlias', data, data.timeout || this.timeout)];
684
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'AlterAlias', data, data.timeout || this.timeout)];
685
685
  case 1:
686
686
  promise = _a.sent();
687
687
  return [2 /*return*/, promise];
@@ -722,7 +722,7 @@ var Collection = /** @class */ (function (_super) {
722
722
  return [4 /*yield*/, this.describeCollection(data)];
723
723
  case 1:
724
724
  collectionInfo = _a.sent();
725
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ManualCompaction', {
725
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ManualCompaction', {
726
726
  collectionID: collectionInfo.collectionID,
727
727
  }, data.timeout || this.timeout)];
728
728
  case 2:
@@ -762,9 +762,9 @@ var Collection = /** @class */ (function (_super) {
762
762
  switch (_a.label) {
763
763
  case 0:
764
764
  if (!data || !data.compactionID) {
765
- throw new Error(_1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
765
+ throw new Error(__1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
766
766
  }
767
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetCompactionState', data, data.timeout || this.timeout)];
767
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetCompactionState', data, data.timeout || this.timeout)];
768
768
  case 1:
769
769
  res = _a.sent();
770
770
  return [2 /*return*/, res];
@@ -802,9 +802,9 @@ var Collection = /** @class */ (function (_super) {
802
802
  switch (_a.label) {
803
803
  case 0:
804
804
  if (!data || !data.compactionID) {
805
- throw new Error(_1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
805
+ throw new Error(__1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
806
806
  }
807
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetCompactionStateWithPlans', data, data.timeout || this.timeout)];
807
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetCompactionStateWithPlans', data, data.timeout || this.timeout)];
808
808
  case 1:
809
809
  res = _a.sent();
810
810
  return [2 /*return*/, res];
@@ -859,9 +859,9 @@ var Collection = /** @class */ (function (_super) {
859
859
  switch (_a.label) {
860
860
  case 0:
861
861
  if (!data || !data.collectionID) {
862
- throw new Error(_1.ERROR_REASONS.COLLECTION_ID_IS_REQUIRED);
862
+ throw new Error(__1.ERROR_REASONS.COLLECTION_ID_IS_REQUIRED);
863
863
  }
864
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetReplicas', data, data.timeout || this.timeout)];
864
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetReplicas', data, data.timeout || this.timeout)];
865
865
  case 1:
866
866
  res = _a.sent();
867
867
  return [2 /*return*/, res];
@@ -902,9 +902,9 @@ var Collection = /** @class */ (function (_super) {
902
902
  switch (_a.label) {
903
903
  case 0:
904
904
  if (!data || !data.collection_name) {
905
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
905
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
906
906
  }
907
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetLoadingProgress', data, data.timeout || this.timeout)];
907
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetLoadingProgress', data, data.timeout || this.timeout)];
908
908
  case 1:
909
909
  res = _a.sent();
910
910
  return [2 /*return*/, res];
@@ -944,9 +944,9 @@ var Collection = /** @class */ (function (_super) {
944
944
  switch (_a.label) {
945
945
  case 0:
946
946
  if (!data || !data.collection_name) {
947
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
947
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
948
948
  }
949
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetLoadState', data, data.timeout || this.timeout)];
949
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetLoadState', data, data.timeout || this.timeout)];
950
950
  case 1:
951
951
  res = _a.sent();
952
952
  return [2 /*return*/, res];
@@ -0,0 +1 @@
1
+ {"version":3,"file":"Collection.js","sourceRoot":"","sources":["../../../milvus/grpc/Collection.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,qCAOqB;AACrB,2CAA0C;AAC1C,wBAmCY;AACZ,qCAA+C;AAE/C;;GAEG;AACH;IAAgC,8BAAU;IAA1C;;IAi1BA,CAAC;IAh1BC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA0CG;IACG,qCAAgB,GAAtB,UAAuB,IAAyB;;;;;;;wBAExC,KAKF,IAAI,IAAI,EAAE,EAJZ,MAAM,YAAA,EACN,eAAe,qBAAA,EACf,WAAW,iBAAA,EACX,yBAA6B,EAA7B,iBAAiB,mBAAG,SAAS,KAAA,CAChB;wBAEf,6EAA6E;wBAC7E,IAAI,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,CAAA,IAAI,CAAC,eAAe,EAAE;4BACvC,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,8BAA8B,CAAC,CAAC;yBAC/D;wBAED,iCAAiC;wBACjC,IAAA,6BAAqB,EAAC,MAAM,CAAC,CAAC;wBAGxB,gBAAgB,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAClD,sCAAsC,CACvC,CAAC;wBACI,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAC7C,iCAAiC,CAClC,CAAC;wBAGI,OAAO,GAAG;4BACd,IAAI,EAAE,eAAe;4BACrB,WAAW,EAAE,WAAW,IAAI,EAAE;4BAC9B,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,UAAA,KAAK;gCACtB,gFAAgF;gCAChF,IAAM,KAA2B,IAAA,wBAAgB,EAAC,KAAK,CAAC,EAAhD,WAAW,iBAAA,EAAK,IAAI,cAAtB,eAAwB,CAA0B,CAAC;gCACzD,OAAO,WAAW,CAAC,MAAM,uBACpB,IAAI,KACP,UAAU,EAAE,IAAA,uBAAe,EAAC,WAAW,CAAC,EACxC,QAAQ,EAAE,KAAK,CAAC,SAAS,EACzB,YAAY,EAAE,KAAK,CAAC,cAAc,IAClC,CAAC;4BACL,CAAC,CAAC;yBACH,CAAC;wBAGI,gBAAgB,GAAG,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBAGpD,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,MAAM,EAAE,CAAC;wBAGjE,KAAK,GACT,MAAA,wBAAoB,CAAC,iBAAiB,CAAC,mCAAI,wBAAoB,CAAC,OAAO,CAAC;wBAG1D,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,kBAAkB,wBAEb,IAAI,KACP,MAAM,EAAE,WAAW,EACnB,iBAAiB,EAAE,KAAK,KAE1B,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBATK,OAAO,GAAG,SASf;wBAED,sBAAsB;wBACtB,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;KAuBC;IACK,kCAAa,GAAnB,UAAoB,IAAsB;;;;;;wBACxC,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,eAAe,EACf,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;KAuBC;IACK,oCAAe,GAArB,UACE,IAAyB;;;;;4BAET,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,iBAAiB,EACjB;4BACE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,uBAAmB,CAAC,GAAG;4BAChD,gBAAgB,EAAE,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,gBAAgB,KAAI,EAAE;yBAC/C,EACD,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,KAAI,IAAI,CAAC,OAAO,CAC9B,EAAA;;wBARK,OAAO,GAAG,SAQf;wBACK,MAAM,GAAqB,EAAE,CAAC;wBACpC,OAAO,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAC,IAAY,EAAE,KAAa;4BAC3D,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,MAAA;gCACJ,EAAE,EAAE,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC;gCACjC,SAAS,EAAE,OAAO,CAAC,sBAAsB,CAAC,KAAK,CAAC;gCAChD,gBAAgB,EAAE,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC;6BACtD,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;wBACH,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC;wBAEtB,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACG,uCAAkB,GAAxB,UACE,IAA2B;;;;;;wBAE3B,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,oBAAoB,EACpB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACG,4CAAuB,GAA7B,UACE,IAAgC;;;;;;wBAEhC,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,yBAAyB,EACzB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBAED,OAAO,CAAC,IAAI,GAAG,IAAA,0BAAkB,EAAC,OAAO,CAAC,KAAK,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;wBAEhE,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACG,mCAAc,GAApB,UAAqB,IAAuB;;;;;;wBAC1C,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,gBAAgB,EAChB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACG,uCAAkB,GAAxB,UAAyB,IAAuB;;;;;;wBAC9C,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,gBAAgB,EAChB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBAED,IAAI,OAAO,CAAC,UAAU,KAAK,aAAS,CAAC,OAAO,EAAE;4BAC5C,MAAM,IAAI,KAAK,CACb,qBAAc,OAAO,CAAC,UAAU,uBAAa,OAAO,CAAC,MAAM,CAAE,CAC9D,CAAC;yBACH;wBAEG,gBAAgB,GAAG,CAAC,CAAC;;;6BAClB,CAAA,MAAM,CAAC,gBAAgB,CAAC,GAAG,GAAG,CAAA;wBACzB,qBAAM,IAAI,CAAC,kBAAkB,CAAC;gCACtC,eAAe,EAAE,IAAI,CAAC,eAAe;6BACtC,CAAC,EAAA;;wBAFE,GAAG,GAAG,SAER;wBAEF,IAAI,GAAG,CAAC,MAAM,CAAC,UAAU,KAAK,aAAS,CAAC,OAAO,EAAE;4BAC/C,MAAM,IAAI,KAAK,CACb,qBAAc,GAAG,CAAC,MAAM,CAAC,UAAU,uBAAa,GAAG,CAAC,MAAM,CAAC,MAAM,CAAE,CACpE,CAAC;yBACH;wBACD,gBAAgB,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;wBACxC,cAAc;wBACd,qBAAM,IAAA,aAAK,EAAC,GAAG,CAAC,EAAA;;wBADhB,cAAc;wBACd,SAAgB,CAAC;;4BAGnB,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACG,sCAAiB,GAAvB,UAAwB,IAA8B;;;;;;wBACpD,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,mBAAmB,EACnB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACG,qCAAgB,GAAtB,UAAuB,IAAyB;;;;;4BAC9B,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,kBAAkB,EAClB;4BACE,OAAO,EAAE,IAAI,CAAC,eAAe;4BAC7B,OAAO,EAAE,IAAI,CAAC,mBAAmB;yBAClC,EACD,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBARK,OAAO,GAAG,SAQf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,mCAAc,GAApB,UAAqB,IAAuB;;;;;;wBAC1C,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAEV,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,gBAAgB,EAChB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACG,gCAAW,GAAjB,UAAkB,IAAoB;;;;;;wBACpC,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;4BACf,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,sBAAsB,CAAC,CAAC;yBACvD;wBACe,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,aAAa,EACb,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACG,8BAAS,GAAf,UAAgB,IAAkB;;;;;;wBAChC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;4BACf,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,sBAAsB,CAAC,CAAC;yBACvD;wBACe,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,WAAW,EACX,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACG,+BAAU,GAAhB,UAAiB,IAAmB;;;;;;wBAClC,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;4BACf,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,sBAAsB,CAAC,CAAC;yBACvD;wBACe,qBAAM,IAAA,iBAAS,EAC7B,IAAI,CAAC,MAAM,EACX,YAAY,EACZ,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,OAAO,GAAG,SAKf;wBACD,sBAAO,OAAO,EAAC;;;;KAChB;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,4BAAO,GAAb,UAAc,IAAgB;;;;;;wBAC5B,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;wBACH,qBAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAA;;wBAApD,cAAc,GAAG,SAAmC;wBAC9C,qBAAM,IAAA,iBAAS,EACzB,IAAI,CAAC,MAAM,EACX,kBAAkB,EAClB;gCACE,YAAY,EAAE,cAAc,CAAC,YAAY;6BAC1C,EACD,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBAPK,GAAG,GAAG,SAOX;wBACD,sBAAO,GAAG,EAAC;;;;KACZ;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,uCAAkB,GAAxB,UACE,IAA2B;;;;;;wBAE3B,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;4BAC/B,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,wBAAwB,CAAC,CAAC;yBACzD;wBACW,qBAAM,IAAA,iBAAS,EACzB,IAAI,CAAC,MAAM,EACX,oBAAoB,EACpB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,GAAG,GAAG,SAKX;wBACD,sBAAO,GAAG,EAAC;;;;KACZ;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,gDAA2B,GAAjC,UACE,IAA2B;;;;;;wBAE3B,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;4BAC/B,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,wBAAwB,CAAC,CAAC;yBACzD;wBACW,qBAAM,IAAA,iBAAS,EACzB,IAAI,CAAC,MAAM,EACX,6BAA6B,EAC7B,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,GAAG,GAAG,SAKX;wBACD,sBAAO,GAAG,EAAC;;;;KACZ;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAuCG;IACG,gCAAW,GAAjB,UAAkB,IAAmB;;;;;;wBACnC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;4BAC/B,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,yBAAyB,CAAC,CAAC;yBAC1D;wBACW,qBAAM,IAAA,iBAAS,EACzB,IAAI,CAAC,MAAM,EACX,aAAa,EACb,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,GAAG,GAAG,SAKX;wBACD,sBAAO,GAAG,EAAC;;;;KACZ;IAED;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACG,uCAAkB,GAAxB,UACE,IAA2B;;;;;;wBAE3B,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;4BAClC,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,2BAA2B,CAAC,CAAC;yBAC5D;wBACW,qBAAM,IAAA,iBAAS,EACzB,IAAI,CAAC,MAAM,EACX,oBAAoB,EACpB,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,GAAG,GAAG,SAKX;wBACD,sBAAO,GAAG,EAAC;;;;KACZ;IAED;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACG,iCAAY,GAAlB,UAAmB,IAAqB;;;;;;wBACtC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;4BAClC,MAAM,IAAI,KAAK,CAAC,iBAAa,CAAC,2BAA2B,CAAC,CAAC;yBAC5D;wBACW,qBAAM,IAAA,iBAAS,EACzB,IAAI,CAAC,MAAM,EACX,cAAc,EACd,IAAI,EACJ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAC7B,EAAA;;wBALK,GAAG,GAAG,SAKX;wBACD,sBAAO,GAAG,EAAC;;;;KACZ;IACH,iBAAC;AAAD,CAAC,AAj1BD,CAAgC,uBAAU,GAi1BzC;AAj1BY,gCAAU"}
@@ -1,4 +1,4 @@
1
- import { DataType, DeleteEntitiesReq, FlushReq, GetFlushStateReq, GetQuerySegmentInfoReq, InsertReq, LoadBalanceReq, ImportReq, ListImportTasksReq, FlushResult, GetFlushStateResponse, GetMetricsResponse, GetQuerySegmentInfoResponse, MutationResult, QueryResults, ResStatus, SearchResults, ImportResponse, ListImportTasksResponse, GetMetricsRequest, QueryReq, SearchReq, SearchSimpleReq } from '.';
1
+ import { DataType, DeleteEntitiesReq, FlushReq, GetFlushStateReq, GetQuerySegmentInfoReq, InsertReq, LoadBalanceReq, ImportReq, ListImportTasksReq, FlushResult, GetFlushStateResponse, GetMetricsResponse, GetQuerySegmentInfoResponse, MutationResult, QueryResults, ResStatus, SearchResults, ImportResponse, ListImportTasksResponse, GetMetricsRequest, QueryReq, SearchReq, SearchSimpleReq } from '..';
2
2
  import { Collection } from './Collection';
3
3
  export declare class Data extends Collection {
4
4
  vectorTypes: DataType[];
@@ -197,8 +197,6 @@ export declare class Data extends Collection {
197
197
  * | :--- | :-- | :-- |
198
198
  * | segmentIDs | Array | The segment ids |
199
199
  * | 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 |
200
-
201
- *
202
200
  *
203
201
  * @returns
204
202
  * | Property | Description |
@@ -72,14 +72,15 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
72
72
  };
73
73
  Object.defineProperty(exports, "__esModule", { value: true });
74
74
  exports.Data = void 0;
75
- var utils_1 = require("../utils");
76
- var _1 = require(".");
75
+ var utils_1 = require("../../utils");
76
+ var __1 = require("..");
77
77
  var Collection_1 = require("./Collection");
78
78
  var Data = /** @class */ (function (_super) {
79
79
  __extends(Data, _super);
80
80
  function Data() {
81
81
  var _this = _super !== null && _super.apply(this, arguments) || this;
82
- _this.vectorTypes = [_1.DataType.BinaryVector, _1.DataType.FloatVector];
82
+ // vectorTypes
83
+ _this.vectorTypes = [__1.DataType.BinaryVector, __1.DataType.FloatVector];
83
84
  return _this;
84
85
  }
85
86
  /**
@@ -127,7 +128,7 @@ var Data = /** @class */ (function (_super) {
127
128
  if (!data.fields_data ||
128
129
  !Array.isArray(data.fields_data) ||
129
130
  !data.fields_data.length) {
130
- throw new Error(_1.ERROR_REASONS.INSERT_CHECK_FILEDS_DATA_IS_REQUIRED);
131
+ throw new Error(__1.ERROR_REASONS.INSERT_CHECK_FILEDS_DATA_IS_REQUIRED);
131
132
  }
132
133
  collection_name = data.collection_name;
133
134
  return [4 /*yield*/, this.describeCollection({
@@ -135,35 +136,34 @@ var Data = /** @class */ (function (_super) {
135
136
  })];
136
137
  case 1:
137
138
  collectionInfo = _a.sent();
138
- if (collectionInfo.status.error_code !== _1.ErrorCode.SUCCESS) {
139
+ if (collectionInfo.status.error_code !== __1.ErrorCode.SUCCESS) {
139
140
  throw new Error(collectionInfo.status.reason);
140
141
  }
141
- fieldsData = collectionInfo.schema.fields
142
+ fieldsData = new Map(collectionInfo.schema.fields
142
143
  .filter(function (v) { return !v.is_primary_key || !v.autoID; })
143
- .map(function (v) { return ({
144
- name: v.name,
145
- type: v.data_type,
146
- dim: Number((0, utils_1.findKeyValue)(v.type_params, 'dim')),
147
- value: [],
148
- }); });
144
+ .map(function (v) { return [
145
+ v.name,
146
+ {
147
+ name: v.name,
148
+ type: v.data_type,
149
+ dim: Number((0, utils_1.findKeyValue)(v.type_params, 'dim')),
150
+ value: [],
151
+ },
152
+ ]; }));
149
153
  params = __assign(__assign({}, data), { num_rows: data.fields_data.length });
150
- // You need to parse the original row data to column data for Milvus.
154
+ // Loop through each row and set the corresponding field values in the Map.
151
155
  data.fields_data.forEach(function (v, i) {
152
- // Set the key as the field name to get all names in a row.
153
156
  var fieldNames = Object.keys(v);
154
157
  fieldNames.forEach(function (name) {
155
- var target = fieldsData.find(function (item) { return item.name === name; });
158
+ var target = fieldsData.get(name);
156
159
  if (!target) {
157
- throw new Error("".concat(_1.ERROR_REASONS.INSERT_CHECK_WRONG_FIELD, " ").concat(i));
160
+ throw new Error("".concat(__1.ERROR_REASONS.INSERT_CHECK_WRONG_FIELD, " ").concat(i));
158
161
  }
159
- var isVector = _this.vectorTypes.includes(_1.DataTypeMap[target.type.toLowerCase()]);
160
- // Check if the dimension is matched when the data type is BinaryVector.
161
- if (_1.DataTypeMap[target.type.toLowerCase()] === _1.DataType.BinaryVector &&
162
+ var isVector = _this.vectorTypes.includes(__1.DataTypeMap[target.type.toLowerCase()]);
163
+ if (__1.DataTypeMap[target.type.toLowerCase()] === __1.DataType.BinaryVector &&
162
164
  v[name].length !== target.dim / 8) {
163
- throw new Error(_1.ERROR_REASONS.INSERT_CHECK_WRONG_DIM);
165
+ throw new Error(__1.ERROR_REASONS.INSERT_CHECK_WRONG_DIM);
164
166
  }
165
- // Value in vector field should be array. Therefore you need concat it.
166
- // but array.concat is slow, we need for loop to push the value one by one
167
167
  if (isVector) {
168
168
  for (var _i = 0, _a = v[name]; _i < _a.length; _i++) {
169
169
  var val = _a[_i];
@@ -175,55 +175,55 @@ var Data = /** @class */ (function (_super) {
175
175
  }
176
176
  });
177
177
  });
178
- params.fields_data = fieldsData.map(function (v) {
178
+ params.fields_data = Array.from(fieldsData.values()).map(function (v) {
179
179
  var _a, _b, _c, _d;
180
180
  // milvus return string for field type, so we define the DataTypeMap to the value we need.
181
181
  // but if milvus change the string, may casue we cant find value.
182
- var type = _1.DataTypeMap[v.type.toLowerCase()];
182
+ var type = __1.DataTypeMap[v.type.toLowerCase()];
183
183
  var key = _this.vectorTypes.includes(type) ? 'vectors' : 'scalars';
184
184
  var dataKey = 'float_vector';
185
185
  switch (type) {
186
- case _1.DataType.FloatVector:
186
+ case __1.DataType.FloatVector:
187
187
  dataKey = 'float_vector';
188
188
  break;
189
- case _1.DataType.BinaryVector:
189
+ case __1.DataType.BinaryVector:
190
190
  dataKey = 'binary_vector';
191
191
  break;
192
- case _1.DataType.Double:
192
+ case __1.DataType.Double:
193
193
  dataKey = 'double_data';
194
194
  break;
195
- case _1.DataType.Float:
195
+ case __1.DataType.Float:
196
196
  dataKey = 'float_data';
197
197
  break;
198
- case _1.DataType.Int64:
198
+ case __1.DataType.Int64:
199
199
  dataKey = 'long_data';
200
200
  break;
201
- case _1.DataType.Int32:
202
- case _1.DataType.Int16:
203
- case _1.DataType.Int8:
201
+ case __1.DataType.Int32:
202
+ case __1.DataType.Int16:
203
+ case __1.DataType.Int8:
204
204
  dataKey = 'int_data';
205
205
  break;
206
- case _1.DataType.Bool:
206
+ case __1.DataType.Bool:
207
207
  dataKey = 'bool_data';
208
208
  break;
209
- case _1.DataType.VarChar:
209
+ case __1.DataType.VarChar:
210
210
  dataKey = 'string_data';
211
211
  break;
212
212
  default:
213
- throw new Error("".concat(_1.ERROR_REASONS.INSERT_CHECK_WRONG_DATA_TYPE, " \"").concat(v.type, ".\""));
213
+ throw new Error("".concat(__1.ERROR_REASONS.INSERT_CHECK_WRONG_DATA_TYPE, " \"").concat(v.type, ".\""));
214
214
  }
215
215
  return _a = {
216
216
  type: type,
217
217
  field_name: v.name
218
218
  },
219
- _a[key] = type === _1.DataType.FloatVector
219
+ _a[key] = type === __1.DataType.FloatVector
220
220
  ? (_b = {
221
221
  dim: v.dim
222
222
  },
223
223
  _b[dataKey] = {
224
224
  data: v.value,
225
225
  },
226
- _b) : type === _1.DataType.BinaryVector
226
+ _b) : type === __1.DataType.BinaryVector
227
227
  ? (_c = {
228
228
  dim: v.dim
229
229
  },
@@ -235,7 +235,7 @@ var Data = /** @class */ (function (_super) {
235
235
  _d),
236
236
  _a;
237
237
  });
238
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Insert', params, data.timeout || this.timeout)];
238
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Insert', params, data.timeout || this.timeout)];
239
239
  case 2:
240
240
  promise = _a.sent();
241
241
  return [2 /*return*/, promise];
@@ -278,9 +278,9 @@ var Data = /** @class */ (function (_super) {
278
278
  switch (_a.label) {
279
279
  case 0:
280
280
  if (!data || !data.collection_name || !data.expr) {
281
- throw new Error(_1.ERROR_REASONS.DELETE_PARAMS_CHECK);
281
+ throw new Error(__1.ERROR_REASONS.DELETE_PARAMS_CHECK);
282
282
  }
283
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Delete', data, data.timeout || this.timeout)];
283
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Delete', data, data.timeout || this.timeout)];
284
284
  case 1:
285
285
  promise = _a.sent();
286
286
  return [2 /*return*/, promise];
@@ -343,13 +343,13 @@ var Data = /** @class */ (function (_super) {
343
343
  anns_field = void 0;
344
344
  for (i = 0; i < collectionInfo.schema.fields.length; i++) {
345
345
  f = collectionInfo.schema.fields[i];
346
- type = _1.DataTypeMap[f.data_type.toLowerCase()];
346
+ type = __1.DataTypeMap[f.data_type.toLowerCase()];
347
347
  // filter vector field
348
- if (type === _1.DataType.FloatVector || type === _1.DataType.BinaryVector) {
348
+ if (type === __1.DataType.FloatVector || type === __1.DataType.BinaryVector) {
349
349
  // get dimension
350
350
  dim = Number((0, utils_1.findKeyValue)(f.type_params, 'dim'));
351
351
  // correcting dimension if necessary for binary vectors
352
- dim = type === _1.DataType.BinaryVector ? dim / 8 : dim;
352
+ dim = type === __1.DataType.BinaryVector ? dim / 8 : dim;
353
353
  // anns field
354
354
  anns_field = f.name;
355
355
  // vector type
@@ -364,9 +364,9 @@ var Data = /** @class */ (function (_super) {
364
364
  anns_field: anns_field,
365
365
  topk: data.limit ||
366
366
  data.topk ||
367
- _1.DEFAULT_TOPK,
367
+ __1.DEFAULT_TOPK,
368
368
  offset: data.offset || 0,
369
- metric_type: data.metric_type || _1.DEFAULT_METRIC_TYPE,
369
+ metric_type: data.metric_type || __1.DEFAULT_METRIC_TYPE,
370
370
  params: JSON.stringify(data.params || {}),
371
371
  };
372
372
  searchVectors = data.vectors || [
@@ -380,20 +380,20 @@ var Data = /** @class */ (function (_super) {
380
380
  tag: '$0',
381
381
  type: vectorType_1,
382
382
  values: searchVectors.map(function (v) {
383
- return vectorType_1 === _1.DataType.BinaryVector
383
+ return vectorType_1 === __1.DataType.BinaryVector
384
384
  ? (0, utils_1.parseBinaryVectorToBytes)(v)
385
385
  : (0, utils_1.parseFloatVectorToBytes)(v);
386
386
  }),
387
387
  },
388
388
  ],
389
389
  })).finish();
390
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Search', {
390
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Search', {
391
391
  collection_name: data.collection_name,
392
392
  partition_names: data.partition_names,
393
393
  output_fields: data.output_fields || defaultOutputFields,
394
394
  nq: data.nq || searchVectors.length,
395
395
  dsl: data.expr || data.filter || '',
396
- dsl_type: _1.DslType.BoolExprV1,
396
+ dsl_type: __1.DslType.BoolExprV1,
397
397
  placeholder_group: placeholderGroupBytes,
398
398
  search_params: (0, utils_1.parseToKeyValue)(search_params),
399
399
  }, data.timeout || this.timeout)];
@@ -481,9 +481,9 @@ var Data = /** @class */ (function (_super) {
481
481
  if (!data ||
482
482
  !Array.isArray(data.collection_names) ||
483
483
  !data.collection_names.length) {
484
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
484
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
485
485
  }
486
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Flush', data, data.timeout || this.timeout)];
486
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Flush', data, data.timeout || this.timeout)];
487
487
  case 1:
488
488
  res = _a.sent();
489
489
  return [2 /*return*/, res];
@@ -524,9 +524,9 @@ var Data = /** @class */ (function (_super) {
524
524
  if (!data ||
525
525
  !Array.isArray(data.collection_names) ||
526
526
  !data.collection_names.length) {
527
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
527
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
528
528
  }
529
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Flush', data, data.timeout || this.timeout)];
529
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Flush', data, data.timeout || this.timeout)];
530
530
  case 1:
531
531
  res = _a.sent();
532
532
  segIDs = Object.keys(res.coll_segIDs)
@@ -596,7 +596,7 @@ var Data = /** @class */ (function (_super) {
596
596
  if (typeof data.offset === 'number') {
597
597
  offset = { offset: data.offset };
598
598
  }
599
- 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 || this.timeout)];
599
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Query', __assign(__assign({}, data), { query_params: (0, utils_1.parseToKeyValue)(__assign(__assign({}, limits), offset)) }), data.timeout || this.timeout)];
600
600
  case 1:
601
601
  promise = _a.sent();
602
602
  results = [];
@@ -668,9 +668,9 @@ var Data = /** @class */ (function (_super) {
668
668
  switch (_a.label) {
669
669
  case 0:
670
670
  if (!data || !data.request || !data.request.metric_type) {
671
- throw new Error(_1.ERROR_REASONS.GET_METRIC_CHECK_PARAMS);
671
+ throw new Error(__1.ERROR_REASONS.GET_METRIC_CHECK_PARAMS);
672
672
  }
673
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetMetrics', {
673
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetMetrics', {
674
674
  request: JSON.stringify(data.request),
675
675
  }, data.timeout || this.timeout)];
676
676
  case 1:
@@ -688,8 +688,6 @@ var Data = /** @class */ (function (_super) {
688
688
  * | :--- | :-- | :-- |
689
689
  * | segmentIDs | Array | The segment ids |
690
690
  * | 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 |
691
-
692
- *
693
691
  *
694
692
  * @returns
695
693
  * | Property | Description |
@@ -713,9 +711,9 @@ var Data = /** @class */ (function (_super) {
713
711
  switch (_a.label) {
714
712
  case 0:
715
713
  if (!data || !data.segmentIDs) {
716
- throw new Error(_1.ERROR_REASONS.GET_FLUSH_STATE_CHECK_PARAMS);
714
+ throw new Error(__1.ERROR_REASONS.GET_FLUSH_STATE_CHECK_PARAMS);
717
715
  }
718
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetFlushState', data, data.timeout || this.timeout)];
716
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetFlushState', data, data.timeout || this.timeout)];
719
717
  case 1:
720
718
  res = _a.sent();
721
719
  return [2 /*return*/, res];
@@ -757,9 +755,9 @@ var Data = /** @class */ (function (_super) {
757
755
  switch (_a.label) {
758
756
  case 0:
759
757
  if (!data || !data.src_nodeID) {
760
- throw new Error(_1.ERROR_REASONS.LOAD_BALANCE_CHECK_PARAMS);
758
+ throw new Error(__1.ERROR_REASONS.LOAD_BALANCE_CHECK_PARAMS);
761
759
  }
762
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'LoadBalance', data, data.timeout || this.timeout)];
760
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'LoadBalance', data, data.timeout || this.timeout)];
763
761
  case 1:
764
762
  res = _a.sent();
765
763
  return [2 /*return*/, res];
@@ -799,9 +797,9 @@ var Data = /** @class */ (function (_super) {
799
797
  switch (_a.label) {
800
798
  case 0:
801
799
  if (!data || !data.collectionName) {
802
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
800
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
803
801
  }
804
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'GetQuerySegmentInfo', data, data.timeout || this.timeout)];
802
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'GetQuerySegmentInfo', data, data.timeout || this.timeout)];
805
803
  case 1:
806
804
  res = _a.sent();
807
805
  return [2 /*return*/, res];
@@ -843,12 +841,12 @@ var Data = /** @class */ (function (_super) {
843
841
  switch (_a.label) {
844
842
  case 0:
845
843
  if (!data || !data.collection_name) {
846
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
844
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
847
845
  }
848
846
  if (!data || !data.files) {
849
- throw new Error(_1.ERROR_REASONS.IMPORT_FILE_CHECK);
847
+ throw new Error(__1.ERROR_REASONS.IMPORT_FILE_CHECK);
850
848
  }
851
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'Import', __assign(__assign({}, data), { options: data.options || [] }), data.timeout || this.timeout)];
849
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'Import', __assign(__assign({}, data), { options: data.options || [] }), data.timeout || this.timeout)];
852
850
  case 1:
853
851
  res = _a.sent();
854
852
  return [2 /*return*/, res];
@@ -893,9 +891,9 @@ var Data = /** @class */ (function (_super) {
893
891
  switch (_a.label) {
894
892
  case 0:
895
893
  if (!data || !data.collection_name) {
896
- throw new Error(_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
894
+ throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
897
895
  }
898
- return [4 /*yield*/, (0, utils_1.promisify)(this.grpcClient, 'ListImportTasks', __assign(__assign({}, data), { limit: data.limit || 0 }), data.timeout || this.timeout)];
896
+ return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ListImportTasks', __assign(__assign({}, data), { limit: data.limit || 0 }), data.timeout || this.timeout)];
899
897
  case 1:
900
898
  res = _a.sent();
901
899
  return [2 /*return*/, res];