@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.
- package/README.md +83 -7
- package/dist/milvus/MilvusClient.d.ts +24 -12
- package/dist/milvus/MilvusClient.js +30 -124
- package/dist/milvus/MilvusClient.js.map +1 -1
- package/dist/milvus/const/Milvus.d.ts +3 -0
- package/dist/milvus/const/Milvus.js +4 -1
- package/dist/milvus/const/Milvus.js.map +1 -1
- package/dist/milvus/const/index.d.ts +2 -0
- package/dist/milvus/const/index.js +20 -0
- package/dist/milvus/const/index.js.map +1 -0
- package/dist/milvus/{BaseClient.d.ts → grpc/BaseClient.d.ts} +10 -4
- package/dist/milvus/grpc/BaseClient.js +64 -0
- package/dist/milvus/grpc/BaseClient.js.map +1 -0
- package/dist/milvus/{Collection.d.ts → grpc/Collection.d.ts} +1 -1
- package/dist/milvus/{Collection.js → grpc/Collection.js} +35 -35
- package/dist/milvus/grpc/Collection.js.map +1 -0
- package/dist/milvus/{Data.d.ts → grpc/Data.d.ts} +1 -3
- package/dist/milvus/{Data.js → grpc/Data.js} +66 -68
- package/dist/milvus/grpc/Data.js.map +1 -0
- package/dist/milvus/grpc/GrpcClient.d.ts +36 -0
- package/dist/milvus/grpc/GrpcClient.js +219 -0
- package/dist/milvus/grpc/GrpcClient.js.map +1 -0
- package/dist/milvus/{MilvusIndex.d.ts → grpc/MilvusIndex.d.ts} +1 -1
- package/dist/milvus/{MilvusIndex.js → grpc/MilvusIndex.js} +7 -7
- package/dist/milvus/grpc/MilvusIndex.js.map +1 -0
- package/dist/milvus/{Partition.d.ts → grpc/Partition.d.ts} +1 -1
- package/dist/milvus/{Partition.js → grpc/Partition.js} +11 -11
- package/dist/milvus/grpc/Partition.js.map +1 -0
- package/dist/milvus/{Resource.d.ts → grpc/Resource.d.ts} +1 -1
- package/dist/milvus/{Resource.js → grpc/Resource.js} +10 -10
- package/dist/milvus/grpc/Resource.js.map +1 -0
- package/dist/milvus/{User.d.ts → grpc/User.d.ts} +1 -1
- package/dist/milvus/{User.js → grpc/User.js} +25 -24
- package/dist/milvus/grpc/User.js.map +1 -0
- package/dist/milvus/index.d.ts +4 -11
- package/dist/milvus/index.js +6 -12
- package/dist/milvus/index.js.map +1 -1
- package/dist/milvus/types/Client.d.ts +7 -1
- package/dist/milvus/types/Collection.d.ts +1 -1
- package/dist/milvus/types/Common.d.ts +1 -1
- package/dist/milvus/types/Index.d.ts +10 -59
- package/dist/milvus/types/Index.js +25 -0
- package/dist/milvus/types/Index.js.map +1 -1
- package/dist/milvus/types/MilvusIndex.d.ts +60 -0
- package/dist/milvus/types/MilvusIndex.js +3 -0
- package/dist/milvus/types/MilvusIndex.js.map +1 -0
- package/dist/milvus/types/User.d.ts +1 -1
- package/dist/milvus/types.d.ts +1 -1
- package/dist/milvus/types.js +2 -1
- package/dist/milvus/types.js.map +1 -1
- package/dist/proto/proto/google/protobuf/descriptor.proto +1 -1
- package/dist/sdk.json +1 -1
- package/dist/utils/Format.d.ts +7 -0
- package/dist/utils/Format.js +12 -1
- package/dist/utils/Format.js.map +1 -1
- package/dist/utils/Grpc.d.ts +14 -0
- package/dist/utils/Grpc.js +107 -1
- package/dist/utils/Grpc.js.map +1 -1
- package/dist/utils/Validate.d.ts +11 -0
- package/dist/utils/Validate.js +21 -1
- package/dist/utils/Validate.js.map +1 -1
- package/dist/utils/test.d.ts +14 -0
- package/dist/utils/test.js +14 -0
- package/dist/utils/test.js.map +1 -1
- package/package.json +2 -2
- package/dist/milvus/BaseClient.js +0 -87
- package/dist/milvus/BaseClient.js.map +0 -1
- package/dist/milvus/Collection.js.map +0 -1
- package/dist/milvus/Data.js.map +0 -1
- package/dist/milvus/MilvusIndex.js.map +0 -1
- package/dist/milvus/Partition.js.map +0 -1
- package/dist/milvus/Resource.js.map +0 -1
- package/dist/milvus/User.js.map +0 -1
- package/dist/milvus/Utils.d.ts +0 -1
- package/dist/milvus/Utils.js +0 -8
- 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("
|
|
77
|
+
var utils_1 = require("../../utils");
|
|
78
78
|
var BaseClient_1 = require("./BaseClient");
|
|
79
|
-
var
|
|
80
|
-
var utils_2 = require("
|
|
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(
|
|
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 =
|
|
160
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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.
|
|
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.
|
|
238
|
-
type: data ? data.type :
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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 !==
|
|
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 !==
|
|
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.
|
|
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.
|
|
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.
|
|
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(
|
|
595
|
+
throw new Error(__1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
|
|
596
596
|
}
|
|
597
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
638
|
+
throw new Error(__1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
|
|
639
639
|
}
|
|
640
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
682
|
+
throw new Error(__1.ERROR_REASONS.ALIAS_NAME_IS_REQUIRED);
|
|
683
683
|
}
|
|
684
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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.
|
|
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(
|
|
765
|
+
throw new Error(__1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
|
|
766
766
|
}
|
|
767
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
805
|
+
throw new Error(__1.ERROR_REASONS.COMPACTIONID_IS_REQUIRED);
|
|
806
806
|
}
|
|
807
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
862
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_ID_IS_REQUIRED);
|
|
863
863
|
}
|
|
864
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
905
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
906
906
|
}
|
|
907
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
947
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
948
948
|
}
|
|
949
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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("
|
|
76
|
-
var
|
|
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
|
-
|
|
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(
|
|
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 !==
|
|
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
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
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
|
-
//
|
|
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.
|
|
158
|
+
var target = fieldsData.get(name);
|
|
156
159
|
if (!target) {
|
|
157
|
-
throw new Error("".concat(
|
|
160
|
+
throw new Error("".concat(__1.ERROR_REASONS.INSERT_CHECK_WRONG_FIELD, " ").concat(i));
|
|
158
161
|
}
|
|
159
|
-
var isVector = _this.vectorTypes.includes(
|
|
160
|
-
|
|
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(
|
|
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 =
|
|
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
|
|
186
|
+
case __1.DataType.FloatVector:
|
|
187
187
|
dataKey = 'float_vector';
|
|
188
188
|
break;
|
|
189
|
-
case
|
|
189
|
+
case __1.DataType.BinaryVector:
|
|
190
190
|
dataKey = 'binary_vector';
|
|
191
191
|
break;
|
|
192
|
-
case
|
|
192
|
+
case __1.DataType.Double:
|
|
193
193
|
dataKey = 'double_data';
|
|
194
194
|
break;
|
|
195
|
-
case
|
|
195
|
+
case __1.DataType.Float:
|
|
196
196
|
dataKey = 'float_data';
|
|
197
197
|
break;
|
|
198
|
-
case
|
|
198
|
+
case __1.DataType.Int64:
|
|
199
199
|
dataKey = 'long_data';
|
|
200
200
|
break;
|
|
201
|
-
case
|
|
202
|
-
case
|
|
203
|
-
case
|
|
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
|
|
206
|
+
case __1.DataType.Bool:
|
|
207
207
|
dataKey = 'bool_data';
|
|
208
208
|
break;
|
|
209
|
-
case
|
|
209
|
+
case __1.DataType.VarChar:
|
|
210
210
|
dataKey = 'string_data';
|
|
211
211
|
break;
|
|
212
212
|
default:
|
|
213
|
-
throw new Error("".concat(
|
|
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 ===
|
|
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 ===
|
|
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.
|
|
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(
|
|
281
|
+
throw new Error(__1.ERROR_REASONS.DELETE_PARAMS_CHECK);
|
|
282
282
|
}
|
|
283
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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 =
|
|
346
|
+
type = __1.DataTypeMap[f.data_type.toLowerCase()];
|
|
347
347
|
// filter vector field
|
|
348
|
-
if (type ===
|
|
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 ===
|
|
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
|
-
|
|
367
|
+
__1.DEFAULT_TOPK,
|
|
368
368
|
offset: data.offset || 0,
|
|
369
|
-
metric_type: data.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 ===
|
|
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.
|
|
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:
|
|
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(
|
|
484
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
485
485
|
}
|
|
486
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
527
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
528
528
|
}
|
|
529
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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.
|
|
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(
|
|
671
|
+
throw new Error(__1.ERROR_REASONS.GET_METRIC_CHECK_PARAMS);
|
|
672
672
|
}
|
|
673
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
714
|
+
throw new Error(__1.ERROR_REASONS.GET_FLUSH_STATE_CHECK_PARAMS);
|
|
717
715
|
}
|
|
718
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
758
|
+
throw new Error(__1.ERROR_REASONS.LOAD_BALANCE_CHECK_PARAMS);
|
|
761
759
|
}
|
|
762
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
800
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
803
801
|
}
|
|
804
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
844
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
847
845
|
}
|
|
848
846
|
if (!data || !data.files) {
|
|
849
|
-
throw new Error(
|
|
847
|
+
throw new Error(__1.ERROR_REASONS.IMPORT_FILE_CHECK);
|
|
850
848
|
}
|
|
851
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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(
|
|
894
|
+
throw new Error(__1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
|
|
897
895
|
}
|
|
898
|
-
return [4 /*yield*/, (0, utils_1.promisify)(this.
|
|
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];
|