@zilliz/milvus2-sdk-node 2.2.10 → 2.2.11

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 (125) hide show
  1. package/README.md +3 -3
  2. package/dist/milvus/MilvusClient.d.ts +0 -4
  3. package/dist/milvus/MilvusClient.js +20 -48
  4. package/dist/milvus/MilvusClient.js.map +1 -1
  5. package/dist/milvus/OrmClient.d.ts +19 -0
  6. package/dist/milvus/OrmClient.js +77 -0
  7. package/dist/milvus/OrmClient.js.map +1 -0
  8. package/dist/milvus/const/ErrorReason.d.ts +36 -31
  9. package/dist/milvus/const/ErrorReason.js +39 -32
  10. package/dist/milvus/const/ErrorReason.js.map +1 -1
  11. package/dist/milvus/const/Milvus.d.ts +2 -9
  12. package/dist/milvus/const/Milvus.js +16 -33
  13. package/dist/milvus/const/Milvus.js.map +1 -1
  14. package/dist/milvus/const/client.d.ts +5 -0
  15. package/dist/milvus/const/client.js +10 -0
  16. package/dist/milvus/const/client.js.map +1 -0
  17. package/dist/milvus/const/defaults.d.ts +11 -0
  18. package/dist/milvus/const/defaults.js +16 -0
  19. package/dist/milvus/const/defaults.js.map +1 -0
  20. package/dist/milvus/const/highLevel.d.ts +23 -0
  21. package/dist/milvus/const/highLevel.js +26 -0
  22. package/dist/milvus/const/highLevel.js.map +1 -0
  23. package/dist/milvus/const/index.d.ts +4 -1
  24. package/dist/milvus/const/index.js +4 -1
  25. package/dist/milvus/const/index.js.map +1 -1
  26. package/dist/milvus/const/limits.d.ts +2 -0
  27. package/dist/milvus/const/limits.js +7 -0
  28. package/dist/milvus/const/limits.js.map +1 -0
  29. package/dist/milvus/grpc/BaseClient.d.ts +16 -7
  30. package/dist/milvus/grpc/BaseClient.js +37 -18
  31. package/dist/milvus/grpc/BaseClient.js.map +1 -1
  32. package/dist/milvus/grpc/Collection.d.ts +32 -3
  33. package/dist/milvus/grpc/Collection.js +242 -416
  34. package/dist/milvus/grpc/Collection.js.map +1 -1
  35. package/dist/milvus/grpc/Data.d.ts +68 -8
  36. package/dist/milvus/grpc/Data.js +539 -566
  37. package/dist/milvus/grpc/Data.js.map +1 -1
  38. package/dist/milvus/grpc/Database.d.ts +69 -0
  39. package/dist/milvus/grpc/Database.js +98 -0
  40. package/dist/milvus/grpc/Database.js.map +1 -0
  41. package/dist/milvus/grpc/GrpcClient.d.ts +18 -8
  42. package/dist/milvus/grpc/GrpcClient.js +87 -156
  43. package/dist/milvus/grpc/GrpcClient.js.map +1 -1
  44. package/dist/milvus/grpc/MilvusIndex.d.ts +1 -1
  45. package/dist/milvus/grpc/MilvusIndex.js +46 -142
  46. package/dist/milvus/grpc/MilvusIndex.js.map +1 -1
  47. package/dist/milvus/grpc/Partition.d.ts +1 -1
  48. package/dist/milvus/grpc/Partition.js +53 -157
  49. package/dist/milvus/grpc/Partition.js.map +1 -1
  50. package/dist/milvus/grpc/Resource.d.ts +2 -2
  51. package/dist/milvus/grpc/Resource.js +64 -170
  52. package/dist/milvus/grpc/Resource.js.map +1 -1
  53. package/dist/milvus/grpc/User.d.ts +1 -1
  54. package/dist/milvus/grpc/User.js +193 -381
  55. package/dist/milvus/grpc/User.js.map +1 -1
  56. package/dist/milvus/index.d.ts +1 -1
  57. package/dist/milvus/index.js +1 -1
  58. package/dist/milvus/index.js.map +1 -1
  59. package/dist/milvus/orm/Collection.d.ts +45 -0
  60. package/dist/milvus/orm/Collection.js +172 -0
  61. package/dist/milvus/orm/Collection.js.map +1 -0
  62. package/dist/milvus/orm/index.d.ts +2 -0
  63. package/dist/milvus/{types.js → orm/index.js} +3 -11
  64. package/dist/milvus/orm/index.js.map +1 -0
  65. package/dist/milvus/orm/utils.d.ts +27 -0
  66. package/dist/milvus/orm/utils.js +27 -0
  67. package/dist/milvus/orm/utils.js.map +1 -0
  68. package/dist/milvus/types/Client.d.ts +12 -1
  69. package/dist/milvus/types/Collection.d.ts +18 -4
  70. package/dist/milvus/types/Data.d.ts +41 -5
  71. package/dist/milvus/types/Database.d.ts +12 -0
  72. package/dist/milvus/types/Database.js +3 -0
  73. package/dist/milvus/types/Database.js.map +1 -0
  74. package/dist/milvus/types/HighLevel.d.ts +13 -0
  75. package/dist/milvus/types/HighLevel.js +3 -0
  76. package/dist/milvus/types/HighLevel.js.map +1 -0
  77. package/dist/milvus/types/MilvusIndex.d.ts +1 -1
  78. package/dist/milvus/types/Resource.d.ts +1 -1
  79. package/dist/milvus/types/{Index.d.ts → index.d.ts} +2 -0
  80. package/dist/milvus/types/{Index.js → index.js} +3 -1
  81. package/dist/milvus/types/index.js.map +1 -0
  82. package/dist/{utils → milvus/utils}/Blob.js +4 -4
  83. package/dist/milvus/utils/Blob.js.map +1 -0
  84. package/dist/{utils → milvus/utils}/Format.d.ts +66 -2
  85. package/dist/milvus/utils/Format.js +416 -0
  86. package/dist/milvus/utils/Format.js.map +1 -0
  87. package/dist/{utils → milvus/utils}/Function.d.ts +1 -1
  88. package/dist/{utils → milvus/utils}/Function.js +7 -7
  89. package/dist/milvus/utils/Function.js.map +1 -0
  90. package/dist/{utils → milvus/utils}/Grpc.d.ts +9 -8
  91. package/dist/milvus/utils/Grpc.js +190 -0
  92. package/dist/milvus/utils/Grpc.js.map +1 -0
  93. package/dist/{utils → milvus/utils}/Validate.d.ts +7 -1
  94. package/dist/milvus/utils/Validate.js +140 -0
  95. package/dist/milvus/utils/Validate.js.map +1 -0
  96. package/dist/{utils → milvus/utils}/index.d.ts +2 -0
  97. package/dist/{utils → milvus/utils}/index.js +2 -0
  98. package/dist/milvus/utils/index.js.map +1 -0
  99. package/dist/milvus/utils/logger.d.ts +1 -0
  100. package/dist/milvus/utils/logger.js +18 -0
  101. package/dist/milvus/utils/logger.js.map +1 -0
  102. package/dist/proto/proto/common.proto +38 -14
  103. package/dist/proto/proto/google/protobuf/descriptor.proto +8 -4
  104. package/dist/proto/proto/milvus.proto +86 -83
  105. package/dist/proto/proto/schema.proto +5 -0
  106. package/dist/sdk.json +1 -1
  107. package/package.json +13 -10
  108. package/dist/milvus/types/Index.js.map +0 -1
  109. package/dist/milvus/types.d.ts +0 -9
  110. package/dist/milvus/types.js.map +0 -1
  111. package/dist/proto/proto/feder.proto +0 -40
  112. package/dist/proto/proto/msg.proto +0 -107
  113. package/dist/utils/Blob.js.map +0 -1
  114. package/dist/utils/Format.js +0 -252
  115. package/dist/utils/Format.js.map +0 -1
  116. package/dist/utils/Function.js.map +0 -1
  117. package/dist/utils/Grpc.js +0 -170
  118. package/dist/utils/Grpc.js.map +0 -1
  119. package/dist/utils/Validate.js +0 -125
  120. package/dist/utils/Validate.js.map +0 -1
  121. package/dist/utils/index.js.map +0 -1
  122. package/dist/utils/test.d.ts +0 -29
  123. package/dist/utils/test.js +0 -112
  124. package/dist/utils/test.js.map +0 -1
  125. /package/dist/{utils → milvus/utils}/Blob.d.ts +0 -0
@@ -1,125 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.isStatusCodeMatched = exports.checkSearchParams = exports.checkCollectionAndPartitionName = exports.checkCollectionName = exports.checkCollectionFields = void 0;
4
- var milvus_1 = require("../milvus");
5
- var grpc_js_1 = require("@grpc/grpc-js");
6
- /**
7
- * when create collection, field must contain 2 Fields.
8
- * Type is int64 or varchar and primary_key = true
9
- * Type is one of float_vector and binary_vector
10
- * Will check fields
11
- * @param fields
12
- */
13
- var checkCollectionFields = function (fields) {
14
- // Define arrays of data types that are allowed for vector fields and primary keys, respectively
15
- var vectorDataTypes = [milvus_1.DataType.BinaryVector, milvus_1.DataType.FloatVector];
16
- var primaryKeyDataTypes = [milvus_1.DataType.Int64, milvus_1.DataType.VarChar];
17
- // Check if every field in the `fields` array has a `data_type` property
18
- var hasDataTypeKey = fields.every(function (field) {
19
- return field.hasOwnProperty('data_type');
20
- });
21
- // If `hasDataTypeKey` is false, an error is thrown indicating that the `data_type` property is missing
22
- if (!hasDataTypeKey) {
23
- throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_MISS_DATA_TYPE);
24
- }
25
- // Check if at least one field in the `fields` array is a primary key with a supported data type
26
- var hasPrimaryKey = fields.some(function (field) {
27
- var isPrimaryKey = field.is_primary_key;
28
- var isSupportedDataType = primaryKeyDataTypes.includes(field.data_type);
29
- return isPrimaryKey && isSupportedDataType;
30
- });
31
- // If `hasPrimaryKey` is false, an error is thrown indicating that a primary key is missing or has an unsupported data type
32
- if (!hasPrimaryKey) {
33
- throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PRIMARY_KEY);
34
- }
35
- // Check if at least one field in the `fields` array is a vector field with a supported data type
36
- var hasVectorField = fields.some(function (field) {
37
- var isVectorField = vectorDataTypes.includes(field.data_type);
38
- return isVectorField;
39
- });
40
- // If `hasVectorField` is false, an error is thrown indicating that a vector field is missing or has an unsupported data type
41
- if (!hasVectorField) {
42
- throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_VECTOR_FIELD_EXIST);
43
- }
44
- fields.forEach(function (field) {
45
- var _a, _b;
46
- var dataType = field.data_type;
47
- var typeParams = field.type_params;
48
- var isVectorField = vectorDataTypes.includes(dataType);
49
- // Check if field is a vector field
50
- if (isVectorField) {
51
- var dim = Number((_a = typeParams === null || typeParams === void 0 ? void 0 : typeParams.dim) !== null && _a !== void 0 ? _a : field.dim);
52
- // Check if vector field has a dimension
53
- if (!dim) {
54
- throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_DIM);
55
- }
56
- // Check if binary vector field has a dimension that is a multiple of 8
57
- if (dataType === milvus_1.DataType.BinaryVector && dim % 8 !== 0) {
58
- throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_BINARY_DIM);
59
- }
60
- }
61
- // Check if varchar field has a max length
62
- if (dataType === milvus_1.DataType.VarChar) {
63
- var maxLength = (_b = typeParams === null || typeParams === void 0 ? void 0 : typeParams.max_length) !== null && _b !== void 0 ? _b : field.max_length;
64
- if (!maxLength) {
65
- throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_MAXLENGTH);
66
- }
67
- }
68
- });
69
- return true;
70
- };
71
- exports.checkCollectionFields = checkCollectionFields;
72
- /**
73
- * check if the request contains collection_name
74
- * otherwise throw an error
75
- * @param data
76
- */
77
- var checkCollectionName = function (data) {
78
- if (!data || !data.collection_name) {
79
- throw new Error(milvus_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
80
- }
81
- };
82
- exports.checkCollectionName = checkCollectionName;
83
- /**
84
- * check if the request contains collection_name and partition_name
85
- * otherwise throw an error
86
- * @param data
87
- */
88
- var checkCollectionAndPartitionName = function (data) {
89
- if (!data || !data.collection_name || !data.partition_name) {
90
- throw new Error(milvus_1.ERROR_REASONS.COLLECTION_PARTITION_NAME_ARE_REQUIRED);
91
- }
92
- };
93
- exports.checkCollectionAndPartitionName = checkCollectionAndPartitionName;
94
- /**
95
- * Checks the validity of search parameters.
96
- * @param {Object} data - The search parameters to be checked.
97
- * @throws {Error} Throws an error if any of the search parameters are invalid.
98
- */
99
- var checkSearchParams = function (data) {
100
- (0, exports.checkCollectionName)(data);
101
- if (!data.vectors && !data.vector) {
102
- throw new Error(milvus_1.ERROR_REASONS.VECTORS_OR_VECTOR_IS_MISSING);
103
- }
104
- };
105
- exports.checkSearchParams = checkSearchParams;
106
- /**
107
- * Checks if a gRPC status code matches any of the given codes.
108
- * DEADLINE_EXCEEDED means that the task wat not completed
109
- * UNAVAILABLE means that the service is not reachable currently
110
- * Reference: https://grpc.github.io/grpc/python/grpc.html#grpc-status-code
111
- *
112
- * @param {number} code - The gRPC status code to check.
113
- * @param {number[]} [codesToCheck=[grpcStatus.DEADLINE_EXCEEDED, grpcStatus.UNAVAILABLE]] - An array of gRPC status codes to check against.
114
- * @returns {boolean} Whether the gRPC status code matches any of the given codes.
115
- */
116
- var isStatusCodeMatched = function (code, codesToCheck) {
117
- if (codesToCheck === void 0) { codesToCheck = [
118
- grpc_js_1.status.DEADLINE_EXCEEDED,
119
- grpc_js_1.status.UNAVAILABLE,
120
- grpc_js_1.status.INTERNAL,
121
- ]; }
122
- return codesToCheck.includes(code);
123
- };
124
- exports.isStatusCodeMatched = isStatusCodeMatched;
125
- //# sourceMappingURL=Validate.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"Validate.js","sourceRoot":"","sources":["../../utils/Validate.ts"],"names":[],"mappings":";;;AAAA,oCAA+D;AAC/D,yCAAqD;AAErD;;;;;;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;AAEF;;;;;;;;;GASG;AACI,IAAM,mBAAmB,GAAG,UACjC,IAAY,EACZ,YAIC;IAJD,6BAAA,EAAA;QACE,gBAAU,CAAC,iBAAiB;QAC5B,gBAAU,CAAC,WAAW;QACtB,gBAAU,CAAC,QAAQ;KACpB;IAED,OAAO,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC,CAAC;AATW,QAAA,mBAAmB,uBAS9B"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../utils/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,yCAAuB;AACvB,2CAAyB;AACzB,6CAA2B;AAC3B,6CAA2B"}
@@ -1,29 +0,0 @@
1
- import { DataType, FieldType } from '../milvus';
2
- export declare const VECTOR_FIELD_NAME = "vector_field";
3
- export declare const INDEX_NAME = "index_name";
4
- /**
5
- * Generates collection parameters with default fields for a given collection name, dimension, vector type, and optional fields array.
6
- * @param {string} collectionName Name of the collection
7
- * @param {string | number} dim Dimension of the vector field
8
- * @param {DataType.FloatVector | DataType.BinaryVector} vectorType Type of vector field
9
- * @param {boolean} [autoID=true] Whether to automatically generate IDs
10
- * @param {any[]} [fields=[]] Optional array of additional fields
11
- * @returns {{ collection_name: string, fields: any[] }} Object containing the collection name and a fields array
12
- */
13
- export declare const genCollectionParams: (collectionName: string, dim: string | number, vectorType?: DataType.FloatVector | DataType.BinaryVector, autoID?: boolean, fields?: any[]) => {
14
- collection_name: string;
15
- fields: any[];
16
- };
17
- /**
18
- * Generates a random collection name with a prefix and a random string appended to it.
19
- * @param {string} [pre='collection'] - The prefix to use for the collection name.
20
- * @returns {string} The generated collection name.
21
- */
22
- export declare const GENERATE_NAME: (pre?: string) => string;
23
- /**
24
- * Generates random data for inserting into a collection
25
- * @param fields An array of objects describing the fields to generate data for
26
- * @param count The number of data points to generate
27
- * @returns An array of objects representing the generated data
28
- */
29
- export declare function generateInsertData(fields: FieldType[], count?: number): any;
@@ -1,112 +0,0 @@
1
- "use strict";
2
- var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
3
- if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
4
- if (ar || !(i in from)) {
5
- if (!ar) ar = Array.prototype.slice.call(from, 0, i);
6
- ar[i] = from[i];
7
- }
8
- }
9
- return to.concat(ar || Array.prototype.slice.call(from));
10
- };
11
- Object.defineProperty(exports, "__esModule", { value: true });
12
- exports.generateInsertData = exports.GENERATE_NAME = exports.genCollectionParams = exports.INDEX_NAME = exports.VECTOR_FIELD_NAME = void 0;
13
- var milvus_1 = require("../milvus");
14
- exports.VECTOR_FIELD_NAME = 'vector_field';
15
- exports.INDEX_NAME = 'index_name';
16
- /**
17
- * Generates collection parameters with default fields for a given collection name, dimension, vector type, and optional fields array.
18
- * @param {string} collectionName Name of the collection
19
- * @param {string | number} dim Dimension of the vector field
20
- * @param {DataType.FloatVector | DataType.BinaryVector} vectorType Type of vector field
21
- * @param {boolean} [autoID=true] Whether to automatically generate IDs
22
- * @param {any[]} [fields=[]] Optional array of additional fields
23
- * @returns {{ collection_name: string, fields: any[] }} Object containing the collection name and a fields array
24
- */
25
- var genCollectionParams = function (collectionName, dim, vectorType, autoID, fields) {
26
- if (vectorType === void 0) { vectorType = milvus_1.DataType.FloatVector; }
27
- if (autoID === void 0) { autoID = true; }
28
- fields = fields || [];
29
- return {
30
- collection_name: collectionName,
31
- fields: __spreadArray([
32
- {
33
- name: exports.VECTOR_FIELD_NAME,
34
- description: 'vector field',
35
- data_type: vectorType,
36
- dim: Number(dim),
37
- },
38
- {
39
- name: 'age',
40
- description: '',
41
- data_type: milvus_1.DataType.Int64,
42
- is_primary_key: true,
43
- autoID: autoID,
44
- },
45
- {
46
- name: 'height',
47
- description: '',
48
- data_type: milvus_1.DataType.Int64,
49
- },
50
- {
51
- name: 'name',
52
- description: '',
53
- data_type: milvus_1.DataType.VarChar,
54
- max_length: 128,
55
- }
56
- ], fields, true),
57
- };
58
- };
59
- exports.genCollectionParams = genCollectionParams;
60
- /**
61
- * Generates a random collection name with a prefix and a random string appended to it.
62
- * @param {string} [pre='collection'] - The prefix to use for the collection name.
63
- * @returns {string} The generated collection name.
64
- */
65
- var GENERATE_NAME = function (pre) {
66
- if (pre === void 0) { pre = 'collection'; }
67
- return "".concat(pre, "_").concat(Math.random().toString(36).substr(2, 8));
68
- };
69
- exports.GENERATE_NAME = GENERATE_NAME;
70
- /**
71
- * Generates random data for inserting into a collection
72
- * @param fields An array of objects describing the fields to generate data for
73
- * @param count The number of data points to generate
74
- * @returns An array of objects representing the generated data
75
- */
76
- function generateInsertData(fields, count) {
77
- if (count === void 0) { count = 10; }
78
- var results = []; // Initialize an empty array to store the generated data
79
- var _loop_1 = function () {
80
- // Loop until we've generated the desired number of data points
81
- var value = {}; // Initialize an empty object to store the generated values for this data point
82
- fields.forEach(function (v) {
83
- // bypass autoID
84
- if (v.autoID) {
85
- return;
86
- }
87
- // Loop through each field we need to generate data for
88
- var name = v.name; // Destructure the field object to get its properties
89
- var isVector = v.data_type === milvus_1.DataType.BinaryVector ||
90
- v.data_type === milvus_1.DataType.FloatVector;
91
- var dim = v.dim || (v.type_params && v.type_params.dim);
92
- var isBool = v.data_type === milvus_1.DataType.Bool;
93
- var isVarChar = v.data_type === milvus_1.DataType.VarChar;
94
- dim = v.data_type === milvus_1.DataType.BinaryVector ? dim / 8 : dim;
95
- value[name] = isVector // If the field is a vector field
96
- ? __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
97
- : isBool // If the field is a boolean field
98
- ? count % 2 === 0 // Generate a random boolean value based on the current count
99
- : isVarChar // If the field is a varchar field
100
- ? Math.random().toString(36).substring(2, 7) // Generate a random string of characters
101
- : Math.floor(Math.random() * 100000); // Otherwise, generate a random integer between 0 and 100000
102
- });
103
- results.push(value); // Add the generated values for this data point to the results array
104
- count--; // Decrement the count to keep track of how many data points we've generated so far
105
- };
106
- while (count > 0) {
107
- _loop_1();
108
- }
109
- return results; // Return the array of generated data
110
- }
111
- exports.generateInsertData = generateInsertData;
112
- //# sourceMappingURL=test.js.map
@@ -1 +0,0 @@
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;AAEvC;;;;;;;;GAQG;AACI,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;AAEF;;;;GAIG;AACI,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;oBAC3C,CAAC,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC,6DAA6D;oBAC/E,CAAC,CAAC,SAAS,CAAC,kCAAkC;wBAC9C,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;QACtG,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"}
File without changes