@zilliz/milvus2-sdk-node 2.2.6 → 2.2.8

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 (57) hide show
  1. package/README.md +162 -14
  2. package/dist/milvus/BaseClient.d.ts +9 -8
  3. package/dist/milvus/BaseClient.js +47 -18
  4. package/dist/milvus/BaseClient.js.map +1 -1
  5. package/dist/milvus/Collection.d.ts +49 -49
  6. package/dist/milvus/Collection.js +105 -91
  7. package/dist/milvus/Collection.js.map +1 -1
  8. package/dist/milvus/Data.d.ts +12 -22
  9. package/dist/milvus/Data.js +100 -90
  10. package/dist/milvus/Data.js.map +1 -1
  11. package/dist/milvus/MilvusClient.js +2 -2
  12. package/dist/milvus/MilvusClient.js.map +1 -1
  13. package/dist/milvus/MilvusIndex.d.ts +46 -46
  14. package/dist/milvus/MilvusIndex.js +59 -51
  15. package/dist/milvus/MilvusIndex.js.map +1 -1
  16. package/dist/milvus/Partition.js +7 -7
  17. package/dist/milvus/Partition.js.map +1 -1
  18. package/dist/milvus/Resource.js +6 -6
  19. package/dist/milvus/Resource.js.map +1 -1
  20. package/dist/milvus/User.js +15 -15
  21. package/dist/milvus/User.js.map +1 -1
  22. package/dist/milvus/const/ErrorReason.d.ts +29 -32
  23. package/dist/milvus/const/ErrorReason.js +29 -32
  24. package/dist/milvus/const/ErrorReason.js.map +1 -1
  25. package/dist/milvus/const/Milvus.d.ts +2 -0
  26. package/dist/milvus/const/Milvus.js +7 -1
  27. package/dist/milvus/const/Milvus.js.map +1 -1
  28. package/dist/milvus/index.d.ts +1 -0
  29. package/dist/milvus/index.js +1 -0
  30. package/dist/milvus/index.js.map +1 -1
  31. package/dist/milvus/types/Client.d.ts +9 -0
  32. package/dist/milvus/types/Client.js +3 -0
  33. package/dist/milvus/types/Client.js.map +1 -0
  34. package/dist/milvus/types/Common.d.ts +3 -0
  35. package/dist/milvus/types/Data.d.ts +15 -2
  36. package/dist/milvus/types/Index.d.ts +9 -1
  37. package/dist/milvus/types.d.ts +1 -0
  38. package/dist/milvus/types.js +1 -0
  39. package/dist/milvus/types.js.map +1 -1
  40. package/dist/proto/proto/common.proto +1 -0
  41. package/dist/proto/proto/google/protobuf/descriptor.proto +19 -3
  42. package/dist/proto/proto/milvus.proto +24 -0
  43. package/dist/proto/proto/schema.proto +13 -1
  44. package/dist/sdk.json +1 -1
  45. package/dist/utils/Format.d.ts +8 -0
  46. package/dist/utils/Format.js +31 -1
  47. package/dist/utils/Format.js.map +1 -1
  48. package/dist/utils/Function.d.ts +9 -1
  49. package/dist/utils/Function.js +18 -5
  50. package/dist/utils/Function.js.map +1 -1
  51. package/dist/utils/Validate.d.ts +7 -1
  52. package/dist/utils/Validate.js +64 -18
  53. package/dist/utils/Validate.js.map +1 -1
  54. package/dist/utils/test.d.ts +8 -7
  55. package/dist/utils/test.js +48 -18
  56. package/dist/utils/test.js.map +1 -1
  57. package/package.json +5 -5
@@ -1,8 +1,7 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.checkCollectionAndPartitionName = exports.checkCollectionName = exports.checkCollectionFields = void 0;
4
- var ErrorReason_1 = require("../milvus/const/ErrorReason");
5
- var Milvus_1 = require("../milvus/const/Milvus");
3
+ exports.checkSearchParams = exports.checkCollectionAndPartitionName = exports.checkCollectionName = exports.checkCollectionFields = void 0;
4
+ var milvus_1 = require("../milvus");
6
5
  /**
7
6
  * when create collection, field must contain 2 Fields.
8
7
  * Type is int64 or varchar and primary_key = true
@@ -11,23 +10,58 @@ var Milvus_1 = require("../milvus/const/Milvus");
11
10
  * @param fields
12
11
  */
13
12
  var checkCollectionFields = function (fields) {
14
- var vectorTypes = [Milvus_1.DataType.BinaryVector, Milvus_1.DataType.FloatVector];
15
- // primary key only support DataType.Int64 and varchar
16
- var primaryTypes = [Milvus_1.DataType.Int64, Milvus_1.DataType.VarChar];
17
- if (!fields.find(function (v) { return v.data_type && primaryTypes.includes(v.data_type) && v.is_primary_key; })) {
18
- throw new Error(ErrorReason_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PRIMARY_KEY);
13
+ // Define arrays of data types that are allowed for vector fields and primary keys, respectively
14
+ var vectorDataTypes = [milvus_1.DataType.BinaryVector, milvus_1.DataType.FloatVector];
15
+ var primaryKeyDataTypes = [milvus_1.DataType.Int64, milvus_1.DataType.VarChar];
16
+ // Check if every field in the `fields` array has a `data_type` property
17
+ var hasDataTypeKey = fields.every(function (field) {
18
+ return field.hasOwnProperty('data_type');
19
+ });
20
+ // If `hasDataTypeKey` is false, an error is thrown indicating that the `data_type` property is missing
21
+ if (!hasDataTypeKey) {
22
+ throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_MISS_DATA_TYPE);
19
23
  }
20
- if (!fields.find(function (v) { return (v.data_type ? vectorTypes.includes(v.data_type) : false); })) {
21
- throw new Error(ErrorReason_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_VECTOR_FIELD_EXIST);
24
+ // Check if at least one field in the `fields` array is a primary key with a supported data type
25
+ var hasPrimaryKey = fields.some(function (field) {
26
+ var isPrimaryKey = field.is_primary_key;
27
+ var isSupportedDataType = primaryKeyDataTypes.includes(field.data_type);
28
+ return isPrimaryKey && isSupportedDataType;
29
+ });
30
+ // If `hasPrimaryKey` is false, an error is thrown indicating that a primary key is missing or has an unsupported data type
31
+ if (!hasPrimaryKey) {
32
+ throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PRIMARY_KEY);
22
33
  }
23
- fields.forEach(function (v) {
24
- if (v.data_type && vectorTypes.includes(v.data_type)) {
25
- var dim = v.type_params ? v.type_params.dim : undefined;
34
+ // Check if at least one field in the `fields` array is a vector field with a supported data type
35
+ var hasVectorField = fields.some(function (field) {
36
+ var isVectorField = vectorDataTypes.includes(field.data_type);
37
+ return isVectorField;
38
+ });
39
+ // If `hasVectorField` is false, an error is thrown indicating that a vector field is missing or has an unsupported data type
40
+ if (!hasVectorField) {
41
+ throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_VECTOR_FIELD_EXIST);
42
+ }
43
+ fields.forEach(function (field) {
44
+ var _a, _b;
45
+ var dataType = field.data_type;
46
+ var typeParams = field.type_params;
47
+ var isVectorField = vectorDataTypes.includes(dataType);
48
+ // Check if field is a vector field
49
+ if (isVectorField) {
50
+ var dim = Number((_a = typeParams === null || typeParams === void 0 ? void 0 : typeParams.dim) !== null && _a !== void 0 ? _a : field.dim);
51
+ // Check if vector field has a dimension
26
52
  if (!dim) {
27
- throw new Error(ErrorReason_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_DIM);
53
+ throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_DIM);
54
+ }
55
+ // Check if binary vector field has a dimension that is a multiple of 8
56
+ if (dataType === milvus_1.DataType.BinaryVector && dim % 8 !== 0) {
57
+ throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_BINARY_DIM);
28
58
  }
29
- if (v.data_type === Milvus_1.DataType.BinaryVector && Number(dim) % 8 > 0) {
30
- throw new Error(ErrorReason_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_BINARY_DIM);
59
+ }
60
+ // Check if varchar field has a max length
61
+ if (dataType === milvus_1.DataType.VarChar) {
62
+ var maxLength = (_b = typeParams === null || typeParams === void 0 ? void 0 : typeParams.max_length) !== null && _b !== void 0 ? _b : field.max_length;
63
+ if (!maxLength) {
64
+ throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_MAXLENGTH);
31
65
  }
32
66
  }
33
67
  });
@@ -41,7 +75,7 @@ exports.checkCollectionFields = checkCollectionFields;
41
75
  */
42
76
  var checkCollectionName = function (data) {
43
77
  if (!data || !data.collection_name) {
44
- throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
78
+ throw new Error(milvus_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
45
79
  }
46
80
  };
47
81
  exports.checkCollectionName = checkCollectionName;
@@ -52,8 +86,20 @@ exports.checkCollectionName = checkCollectionName;
52
86
  */
53
87
  var checkCollectionAndPartitionName = function (data) {
54
88
  if (!data || !data.collection_name || !data.partition_name) {
55
- throw new Error(ErrorReason_1.ERROR_REASONS.COLLECTION_PARTITION_NAME_ARE_REQUIRED);
89
+ throw new Error(milvus_1.ERROR_REASONS.COLLECTION_PARTITION_NAME_ARE_REQUIRED);
56
90
  }
57
91
  };
58
92
  exports.checkCollectionAndPartitionName = checkCollectionAndPartitionName;
93
+ /**
94
+ * Checks the validity of search parameters.
95
+ * @param {Object} data - The search parameters to be checked.
96
+ * @throws {Error} Throws an error if any of the search parameters are invalid.
97
+ */
98
+ var checkSearchParams = function (data) {
99
+ (0, exports.checkCollectionName)(data);
100
+ if (!data.vectors && !data.vector) {
101
+ throw new Error(milvus_1.ERROR_REASONS.VECTORS_OR_VECTOR_IS_MISSING);
102
+ }
103
+ };
104
+ exports.checkSearchParams = checkSearchParams;
59
105
  //# sourceMappingURL=Validate.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"Validate.js","sourceRoot":"","sources":["../../utils/Validate.ts"],"names":[],"mappings":";;;AAAA,2DAA4D;AAE5D,iDAAkD;AAElD;;;;;;GAMG;AACI,IAAM,qBAAqB,GAAG,UAAC,MAAmB;IACvD,IAAM,WAAW,GAAG,CAAC,iBAAQ,CAAC,YAAY,EAAE,iBAAQ,CAAC,WAAW,CAAC,CAAC;IAClE,sDAAsD;IACtD,IAAM,YAAY,GAAG,CAAC,iBAAQ,CAAC,KAAK,EAAE,iBAAQ,CAAC,OAAO,CAAC,CAAC;IACxD,IACE,CAAC,MAAM,CAAC,IAAI,CACV,UAAA,CAAC,IAAI,OAAA,CAAC,CAAC,SAAS,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,cAAc,EAArE,CAAqE,CAC3E,EACD;QACA,MAAM,IAAI,KAAK,CAAC,2BAAa,CAAC,mCAAmC,CAAC,CAAC;KACpE;IACD,IACE,CAAC,MAAM,CAAC,IAAI,CAAC,UAAA,CAAC,IAAI,OAAA,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAzD,CAAyD,CAAC,EAC5E;QACA,MAAM,IAAI,KAAK,CAAC,2BAAa,CAAC,0CAA0C,CAAC,CAAC;KAC3E;IAED,MAAM,CAAC,OAAO,CAAC,UAAA,CAAC;QACd,IAAI,CAAC,CAAC,SAAS,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE;YACpD,IAAM,GAAG,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;YAC1D,IAAI,CAAC,GAAG,EAAE;gBACR,MAAM,IAAI,KAAK,CAAC,2BAAa,CAAC,gCAAgC,CAAC,CAAC;aACjE;YACD,IAAI,CAAC,CAAC,SAAS,KAAK,iBAAQ,CAAC,YAAY,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBAChE,MAAM,IAAI,KAAK,CAAC,2BAAa,CAAC,kCAAkC,CAAC,CAAC;aACnE;SACF;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AA9BW,QAAA,qBAAqB,yBA8BhC;AAEF;;;;GAIG;AACI,IAAM,mBAAmB,GAAG,UAAC,IAAS;IAC3C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;QAClC,MAAM,IAAI,KAAK,CAAC,2BAAa,CAAC,2BAA2B,CAAC,CAAC;KAC5D;AACH,CAAC,CAAC;AAJW,QAAA,mBAAmB,uBAI9B;AAEF;;;;GAIG;AACI,IAAM,+BAA+B,GAAG,UAAC,IAAS;IACvD,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;QAC1D,MAAM,IAAI,KAAK,CAAC,2BAAa,CAAC,sCAAsC,CAAC,CAAC;KACvE;AACH,CAAC,CAAC;AAJW,QAAA,+BAA+B,mCAI1C"}
1
+ {"version":3,"file":"Validate.js","sourceRoot":"","sources":["../../utils/Validate.ts"],"names":[],"mappings":";;;AAAA,oCAA+D;AAE/D;;;;;;GAMG;AACI,IAAM,qBAAqB,GAAG,UAAC,MAAmB;IACvD,gGAAgG;IAChG,IAAM,eAAe,GAAG,CAAC,iBAAQ,CAAC,YAAY,EAAE,iBAAQ,CAAC,WAAW,CAAC,CAAC;IACtE,IAAM,mBAAmB,GAAG,CAAC,iBAAQ,CAAC,KAAK,EAAE,iBAAQ,CAAC,OAAO,CAAC,CAAC;IAE/D,wEAAwE;IACxE,IAAM,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC,UAAA,KAAK;QACvC,OAAO,KAAK,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,uGAAuG;IACvG,IAAI,CAAC,cAAc,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,gCAAgC,CAAC,CAAC;KACjE;IAED,gGAAgG;IAChG,IAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,UAAA,KAAK;QACrC,IAAM,YAAY,GAAG,KAAK,CAAC,cAAc,CAAC;QAC1C,IAAM,mBAAmB,GAAG,mBAAmB,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAU,CAAC,CAAC;QAC3E,OAAO,YAAY,IAAI,mBAAmB,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,2HAA2H;IAC3H,IAAI,CAAC,aAAa,EAAE;QAClB,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,mCAAmC,CAAC,CAAC;KACpE;IAED,iGAAiG;IACjG,IAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,UAAA,KAAK;QACtC,IAAM,aAAa,GAAG,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAU,CAAC,CAAC;QACjE,OAAO,aAAa,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,6HAA6H;IAC7H,IAAI,CAAC,cAAc,EAAE;QACnB,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,0CAA0C,CAAC,CAAC;KAC3E;IAED,MAAM,CAAC,OAAO,CAAC,UAAA,KAAK;;QAClB,IAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC;QACjC,IAAM,UAAU,GAAG,KAAK,CAAC,WAAW,CAAC;QACrC,IAAM,aAAa,GAAG,eAAe,CAAC,QAAQ,CAAC,QAAS,CAAC,CAAC;QAE1D,mCAAmC;QACnC,IAAI,aAAa,EAAE;YACjB,IAAM,GAAG,GAAG,MAAM,CAAC,MAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,GAAG,mCAAI,KAAK,CAAC,GAAG,CAAC,CAAC;YACjD,wCAAwC;YACxC,IAAI,CAAC,GAAG,EAAE;gBACR,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,gCAAgC,CAAC,CAAC;aACjE;YAED,uEAAuE;YACvE,IAAI,QAAQ,KAAK,iBAAQ,CAAC,YAAY,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE;gBACvD,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,kCAAkC,CAAC,CAAC;aACnE;SACF;QAED,0CAA0C;QAC1C,IAAI,QAAQ,KAAK,iBAAQ,CAAC,OAAO,EAAE;YACjC,IAAM,SAAS,GAAG,MAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,UAAU,mCAAI,KAAK,CAAC,UAAU,CAAC;YAC7D,IAAI,CAAC,SAAS,EAAE;gBACd,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,sCAAsC,CAAC,CAAC;aACvE;SACF;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAnEW,QAAA,qBAAqB,yBAmEhC;AAEF;;;;GAIG;AACI,IAAM,mBAAmB,GAAG,UAAC,IAAS;IAC3C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;QAClC,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,2BAA2B,CAAC,CAAC;KAC5D;AACH,CAAC,CAAC;AAJW,QAAA,mBAAmB,uBAI9B;AAEF;;;;GAIG;AACI,IAAM,+BAA+B,GAAG,UAAC,IAAS;IACvD,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;QAC1D,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,sCAAsC,CAAC,CAAC;KACvE;AACH,CAAC,CAAC;AAJW,QAAA,+BAA+B,mCAI1C;AAEF;;;;GAIG;AACI,IAAM,iBAAiB,GAAG,UAAC,IAAS;IACzC,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;IAE1B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;QACjC,MAAM,IAAI,KAAK,CAAC,sBAAa,CAAC,4BAA4B,CAAC,CAAC;KAC7D;AACH,CAAC,CAAC;AANW,QAAA,iBAAiB,qBAM5B"}
@@ -1,4 +1,4 @@
1
- import { DataType } from '../milvus/const/Milvus';
1
+ import { DataType, FieldType } from '../milvus';
2
2
  export declare const VECTOR_FIELD_NAME = "vector_field";
3
3
  export declare const INDEX_NAME = "index_name";
4
4
  export declare const genCollectionParams: (collectionName: string, dim: string | number, vectorType?: DataType.FloatVector | DataType.BinaryVector, autoID?: boolean, fields?: any[]) => {
@@ -6,9 +6,10 @@ export declare const genCollectionParams: (collectionName: string, dim: string |
6
6
  fields: any[];
7
7
  };
8
8
  export declare const GENERATE_NAME: (pre?: string) => string;
9
- export declare function generateInsertData(fields: {
10
- isVector: boolean;
11
- dim?: number;
12
- name: string;
13
- isBool?: boolean;
14
- }[], count: number): any;
9
+ /**
10
+ * Generates random data for inserting into a collection
11
+ * @param fields An array of objects describing the fields to generate data for
12
+ * @param count The number of data points to generate
13
+ * @returns An array of objects representing the generated data
14
+ */
15
+ export declare function generateInsertData(fields: FieldType[], count?: number): any;
@@ -10,11 +10,11 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
10
10
  };
11
11
  Object.defineProperty(exports, "__esModule", { value: true });
12
12
  exports.generateInsertData = exports.GENERATE_NAME = exports.genCollectionParams = exports.INDEX_NAME = exports.VECTOR_FIELD_NAME = void 0;
13
- var Milvus_1 = require("../milvus/const/Milvus");
13
+ var milvus_1 = require("../milvus");
14
14
  exports.VECTOR_FIELD_NAME = 'vector_field';
15
15
  exports.INDEX_NAME = 'index_name';
16
16
  var genCollectionParams = function (collectionName, dim, vectorType, autoID, fields) {
17
- if (vectorType === void 0) { vectorType = Milvus_1.DataType.FloatVector; }
17
+ if (vectorType === void 0) { vectorType = milvus_1.DataType.FloatVector; }
18
18
  if (autoID === void 0) { autoID = true; }
19
19
  fields = fields || [];
20
20
  return {
@@ -24,16 +24,25 @@ var genCollectionParams = function (collectionName, dim, vectorType, autoID, fie
24
24
  name: exports.VECTOR_FIELD_NAME,
25
25
  description: 'vector field',
26
26
  data_type: vectorType,
27
- type_params: {
28
- dim: dim,
29
- },
27
+ dim: Number(dim),
30
28
  },
31
29
  {
32
30
  name: 'age',
33
- data_type: Milvus_1.DataType.Int64,
34
- autoID: autoID,
31
+ description: '',
32
+ data_type: milvus_1.DataType.Int64,
35
33
  is_primary_key: true,
34
+ autoID: autoID,
35
+ },
36
+ {
37
+ name: 'height',
36
38
  description: '',
39
+ data_type: milvus_1.DataType.Int64,
40
+ },
41
+ {
42
+ name: 'name',
43
+ description: '',
44
+ data_type: milvus_1.DataType.VarChar,
45
+ max_length: 128,
37
46
  }
38
47
  ], fields, true),
39
48
  };
@@ -44,25 +53,46 @@ var GENERATE_NAME = function (pre) {
44
53
  return "".concat(pre, "_").concat(Math.random().toString(36).substr(2, 8));
45
54
  };
46
55
  exports.GENERATE_NAME = GENERATE_NAME;
56
+ /**
57
+ * Generates random data for inserting into a collection
58
+ * @param fields An array of objects describing the fields to generate data for
59
+ * @param count The number of data points to generate
60
+ * @returns An array of objects representing the generated data
61
+ */
47
62
  function generateInsertData(fields, count) {
48
- var results = [];
63
+ if (count === void 0) { count = 10; }
64
+ var results = []; // Initialize an empty array to store the generated data
49
65
  var _loop_1 = function () {
50
- var value = {};
66
+ // Loop until we've generated the desired number of data points
67
+ var value = {}; // Initialize an empty object to store the generated values for this data point
51
68
  fields.forEach(function (v) {
52
- var isVector = v.isVector, dim = v.dim, name = v.name, isBool = v.isBool;
53
- value[name] = isVector
54
- ? __spreadArray([], Array(dim), true).map(function () { return Math.random() * 10; })
55
- : isBool
56
- ? count % 2 === 0
57
- : Math.floor(Math.random() * 100000);
69
+ // bypass autoID
70
+ if (v.autoID) {
71
+ return;
72
+ }
73
+ // Loop through each field we need to generate data for
74
+ var name = v.name; // Destructure the field object to get its properties
75
+ var isVector = v.data_type === milvus_1.DataType.BinaryVector ||
76
+ v.data_type === milvus_1.DataType.FloatVector;
77
+ var dim = v.dim || (v.type_params && v.type_params.dim);
78
+ var isBool = v.data_type === milvus_1.DataType.Bool;
79
+ var isVarChar = v.data_type === milvus_1.DataType.VarChar;
80
+ dim = v.data_type === milvus_1.DataType.BinaryVector ? dim / 8 : dim;
81
+ value[name] = isVector // If the field is a vector field
82
+ ? __spreadArray([], Array(Number(dim)), true).map(function () { return Math.random(); }) // Generate an array of random numbers between 0 and 10 with length equal to the vector dimension
83
+ : isBool // If the field is a boolean field
84
+ ? count % 2 === 0 // Generate a random boolean value based on the current count
85
+ : isVarChar // If the field is a varchar field
86
+ ? Math.random().toString(36).substring(2, 7) // Generate a random string of characters
87
+ : Math.floor(Math.random() * 100000); // Otherwise, generate a random integer between 0 and 100000
58
88
  });
59
- results.push(value);
60
- count--;
89
+ results.push(value); // Add the generated values for this data point to the results array
90
+ count--; // Decrement the count to keep track of how many data points we've generated so far
61
91
  };
62
92
  while (count > 0) {
63
93
  _loop_1();
64
94
  }
65
- return results;
95
+ return results; // Return the array of generated data
66
96
  }
67
97
  exports.generateInsertData = generateInsertData;
68
98
  //# sourceMappingURL=test.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"test.js","sourceRoot":"","sources":["../../utils/test.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,iDAAkD;AAErC,QAAA,iBAAiB,GAAG,cAAc,CAAC;AACnC,QAAA,UAAU,GAAG,YAAY,CAAC;AAChC,IAAM,mBAAmB,GAAG,UACjC,cAAsB,EACtB,GAAoB,EACpB,UAEgD,EAChD,MAAsB,EACtB,MAAc;IAJd,2BAAA,EAAA,aAE4B,iBAAQ,CAAC,WAAW;IAChD,uBAAA,EAAA,aAAsB;IAGtB,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;IACtB,OAAO;QACL,eAAe,EAAE,cAAc;QAC/B,MAAM;YACJ;gBACE,IAAI,EAAE,yBAAiB;gBACvB,WAAW,EAAE,cAAc;gBAC3B,SAAS,EAAE,UAAU;gBAErB,WAAW,EAAE;oBACX,GAAG,KAAA;iBACJ;aACF;YACD;gBACE,IAAI,EAAE,KAAK;gBACX,SAAS,EAAE,iBAAQ,CAAC,KAAK;gBACzB,MAAM,QAAA;gBACN,cAAc,EAAE,IAAI;gBACpB,WAAW,EAAE,EAAE;aAChB;WACE,MAAM,OACV;KACF,CAAC;AACJ,CAAC,CAAC;AAhCW,QAAA,mBAAmB,uBAgC9B;AAEK,IAAM,aAAa,GAAG,UAAC,GAAkB;IAAlB,oBAAA,EAAA,kBAAkB;IAC9C,OAAA,UAAG,GAAG,cAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE;AAAnD,CAAmD,CAAC;AADzC,QAAA,aAAa,iBAC4B;AAEtD,SAAgB,kBAAkB,CAChC,MAA6E,EAC7E,KAAa;IAEb,IAAM,OAAO,GAAQ,EAAE,CAAC;;QAEtB,IAAI,KAAK,GAAQ,EAAE,CAAC;QAEpB,MAAM,CAAC,OAAO,CAAC,UAAA,CAAC;YACN,IAAA,QAAQ,GAAwB,CAAC,SAAzB,EAAE,GAAG,GAAmB,CAAC,IAApB,EAAE,IAAI,GAAa,CAAC,KAAd,EAAE,MAAM,GAAK,CAAC,OAAN,CAAO;YAC1C,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ;gBACpB,CAAC,CAAC,kBAAI,KAAK,CAAC,GAAG,CAAC,QAAE,GAAG,CAAC,cAAM,OAAA,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,EAAlB,CAAkB,CAAC;gBAC/C,CAAC,CAAC,MAAM;oBACR,CAAC,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC;oBACjB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,KAAK,EAAE,CAAC;;IAZV,OAAO,KAAK,GAAG,CAAC;;KAaf;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AApBD,gDAoBC"}
1
+ {"version":3,"file":"test.js","sourceRoot":"","sources":["../../utils/test.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,oCAAgD;AAEnC,QAAA,iBAAiB,GAAG,cAAc,CAAC;AACnC,QAAA,UAAU,GAAG,YAAY,CAAC;AAChC,IAAM,mBAAmB,GAAG,UACjC,cAAsB,EACtB,GAAoB,EACpB,UAEgD,EAChD,MAAsB,EACtB,MAAc;IAJd,2BAAA,EAAA,aAE4B,iBAAQ,CAAC,WAAW;IAChD,uBAAA,EAAA,aAAsB;IAGtB,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;IACtB,OAAO;QACL,eAAe,EAAE,cAAc;QAC/B,MAAM;YACJ;gBACE,IAAI,EAAE,yBAAiB;gBACvB,WAAW,EAAE,cAAc;gBAC3B,SAAS,EAAE,UAAU;gBACrB,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC;aACjB;YACD;gBACE,IAAI,EAAE,KAAK;gBACX,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,iBAAQ,CAAC,KAAK;gBACzB,cAAc,EAAE,IAAI;gBACpB,MAAM,QAAA;aACP;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,iBAAQ,CAAC,KAAK;aAC1B;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,iBAAQ,CAAC,OAAO;gBAC3B,UAAU,EAAE,GAAG;aAChB;WACE,MAAM,OACV;KACF,CAAC;AACJ,CAAC,CAAC;AAxCW,QAAA,mBAAmB,uBAwC9B;AAEK,IAAM,aAAa,GAAG,UAAC,GAAkB;IAAlB,oBAAA,EAAA,kBAAkB;IAC9C,OAAA,UAAG,GAAG,cAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE;AAAnD,CAAmD,CAAC;AADzC,QAAA,aAAa,iBAC4B;AAEtD;;;;;GAKG;AACH,SAAgB,kBAAkB,CAAC,MAAmB,EAAE,KAAkB;IAAlB,sBAAA,EAAA,UAAkB;IACxE,IAAM,OAAO,GAAQ,EAAE,CAAC,CAAC,wDAAwD;;QAE/E,+DAA+D;QAC/D,IAAI,KAAK,GAAQ,EAAE,CAAC,CAAC,+EAA+E;QAEpG,MAAM,CAAC,OAAO,CAAC,UAAA,CAAC;YACd,gBAAgB;YAChB,IAAI,CAAC,CAAC,MAAM,EAAE;gBACZ,OAAO;aACR;YACD,uDAAuD;YAC/C,IAAA,IAAI,GAAK,CAAC,KAAN,CAAO,CAAC,qDAAqD;YACzE,IAAM,QAAQ,GACZ,CAAC,CAAC,SAAS,KAAK,iBAAQ,CAAC,YAAY;gBACrC,CAAC,CAAC,SAAS,KAAK,iBAAQ,CAAC,WAAW,CAAC;YACvC,IAAI,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACxD,IAAM,MAAM,GAAG,CAAC,CAAC,SAAS,KAAK,iBAAQ,CAAC,IAAI,CAAC;YAC7C,IAAM,SAAS,GAAG,CAAC,CAAC,SAAS,KAAK,iBAAQ,CAAC,OAAO,CAAC;YAEnD,GAAG,GAAG,CAAC,CAAC,SAAS,KAAK,iBAAQ,CAAC,YAAY,CAAC,CAAC,CAAE,GAAc,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;YAExE,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,iCAAiC;gBACtD,CAAC,CAAC,kBAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,QAAE,GAAG,CAAC,cAAM,OAAA,IAAI,CAAC,MAAM,EAAE,EAAb,CAAa,CAAC,CAAC,iGAAiG;gBACpJ,CAAC,CAAC,MAAM,CAAC,kCAAkC;oBACzC,CAAC,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC,6DAA6D;oBAC/E,CAAC,CAAC,SAAS,CAAC,kCAAkC;wBAC5C,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,yCAAyC;wBACtF,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,4DAA4D;QAC1G,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,oEAAoE;QACzF,KAAK,EAAE,CAAC,CAAC,mFAAmF;;IA7B9F,OAAO,KAAK,GAAG,CAAC;;KA8Bf;IACD,OAAO,OAAO,CAAC,CAAC,qCAAqC;AACvD,CAAC;AAlCD,gDAkCC"}
package/package.json CHANGED
@@ -1,8 +1,8 @@
1
1
  {
2
2
  "name": "@zilliz/milvus2-sdk-node",
3
3
  "author": "ued@zilliz.com",
4
- "version": "2.2.6",
5
- "milvusVersion": "v2.2.5",
4
+ "version": "2.2.8",
5
+ "milvusVersion": "v2.2.8",
6
6
  "main": "dist/milvus",
7
7
  "files": [
8
8
  "dist"
@@ -19,9 +19,9 @@
19
19
  "doc-json": "npx typedoc milvus --json"
20
20
  },
21
21
  "dependencies": {
22
- "@grpc/grpc-js": "^1.2.12",
23
- "@grpc/proto-loader": "^0.7.3",
24
- "protobufjs": "^7.1.2"
22
+ "@grpc/grpc-js": "^1.8.14",
23
+ "@grpc/proto-loader": "^0.7.6",
24
+ "protobufjs": "^7.2.3"
25
25
  },
26
26
  "devDependencies": {
27
27
  "@babel/plugin-transform-modules-commonjs": "^7.13.8",