@zilliz/milvus2-sdk-node 2.2.5 → 2.2.7
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 +167 -13
- package/dist/milvus/BaseClient.d.ts +9 -8
- package/dist/milvus/BaseClient.js +47 -18
- package/dist/milvus/BaseClient.js.map +1 -1
- package/dist/milvus/Collection.d.ts +54 -1
- package/dist/milvus/Collection.js +141 -40
- package/dist/milvus/Collection.js.map +1 -1
- package/dist/milvus/Data.d.ts +10 -22
- package/dist/milvus/Data.js +96 -90
- package/dist/milvus/Data.js.map +1 -1
- package/dist/milvus/MilvusClient.js +2 -2
- package/dist/milvus/MilvusClient.js.map +1 -1
- package/dist/milvus/MilvusIndex.d.ts +46 -46
- package/dist/milvus/MilvusIndex.js +59 -51
- package/dist/milvus/MilvusIndex.js.map +1 -1
- package/dist/milvus/Partition.js +7 -7
- package/dist/milvus/Partition.js.map +1 -1
- package/dist/milvus/Resource.js +6 -6
- package/dist/milvus/Resource.js.map +1 -1
- package/dist/milvus/User.js +15 -15
- package/dist/milvus/User.js.map +1 -1
- package/dist/milvus/const/ErrorReason.d.ts +29 -32
- package/dist/milvus/const/ErrorReason.js +29 -32
- package/dist/milvus/const/ErrorReason.js.map +1 -1
- package/dist/milvus/const/Milvus.d.ts +25 -19
- package/dist/milvus/const/Milvus.js +31 -31
- package/dist/milvus/const/Milvus.js.map +1 -1
- package/dist/milvus/index.d.ts +1 -0
- package/dist/milvus/index.js +1 -0
- package/dist/milvus/index.js.map +1 -1
- package/dist/milvus/types/Client.d.ts +9 -0
- package/dist/milvus/types/Client.js +3 -0
- package/dist/milvus/types/Client.js.map +1 -0
- package/dist/milvus/types/Collection.d.ts +33 -16
- package/dist/milvus/types/Collection.js.map +1 -1
- package/dist/milvus/types/Common.d.ts +3 -0
- package/dist/milvus/types/Data.d.ts +13 -2
- package/dist/milvus/types/Index.d.ts +9 -1
- package/dist/milvus/types.d.ts +1 -0
- package/dist/milvus/types.js +1 -0
- package/dist/milvus/types.js.map +1 -1
- package/dist/proto/proto/common.proto +1 -0
- package/dist/proto/proto/google/protobuf/descriptor.proto +29 -5
- package/dist/proto/proto/milvus.proto +24 -0
- package/dist/proto/proto/schema.proto +17 -1
- package/dist/sdk.json +1 -1
- package/dist/utils/Format.d.ts +28 -1
- package/dist/utils/Format.js +73 -7
- package/dist/utils/Format.js.map +1 -1
- package/dist/utils/Function.d.ts +9 -1
- package/dist/utils/Function.js +18 -5
- package/dist/utils/Function.js.map +1 -1
- package/dist/utils/Validate.d.ts +7 -1
- package/dist/utils/Validate.js +64 -18
- package/dist/utils/Validate.js.map +1 -1
- package/dist/utils/test.d.ts +9 -8
- package/dist/utils/test.js +48 -18
- package/dist/utils/test.js.map +1 -1
- package/package.json +4 -4
|
@@ -57,6 +57,7 @@ message FieldSchema {
|
|
|
57
57
|
FieldState state = 9; // To keep compatible with older version, the default
|
|
58
58
|
// state is `Created`.
|
|
59
59
|
DataType element_type = 10; // For array type, the element type is stored here
|
|
60
|
+
ValueField default_value = 11; // default_value only support scalars except array and json for now
|
|
60
61
|
}
|
|
61
62
|
|
|
62
63
|
/**
|
|
@@ -84,10 +85,25 @@ message BytesArray { repeated bytes data = 1; }
|
|
|
84
85
|
|
|
85
86
|
message StringArray { repeated string data = 1; }
|
|
86
87
|
|
|
87
|
-
message ArrayArray {
|
|
88
|
+
message ArrayArray {
|
|
89
|
+
repeated ScalarField data = 1;
|
|
90
|
+
DataType element_type = 2;
|
|
91
|
+
}
|
|
88
92
|
|
|
89
93
|
message JSONArray { repeated bytes data = 1; }
|
|
90
94
|
|
|
95
|
+
message ValueField {
|
|
96
|
+
oneof data {
|
|
97
|
+
bool bool_data = 1;
|
|
98
|
+
int32 int_data = 2;
|
|
99
|
+
int64 long_data = 3;
|
|
100
|
+
float float_data = 4;
|
|
101
|
+
double double_data = 5;
|
|
102
|
+
string string_data = 6;
|
|
103
|
+
bytes bytes_data = 7;
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
|
|
91
107
|
message ScalarField {
|
|
92
108
|
oneof data {
|
|
93
109
|
BoolArray bool_data = 1;
|
package/dist/sdk.json
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":"2.2.
|
|
1
|
+
{"version":"2.2.7","milvusVersion":"v2.2.5"}
|
package/dist/utils/Format.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { KeyValuePair } from '../milvus
|
|
1
|
+
import { KeyValuePair, FieldType } from '../milvus';
|
|
2
2
|
/**
|
|
3
3
|
* parse [{key:"row_count",value:4}] to {row_count:4}
|
|
4
4
|
* @param data key value pair array
|
|
@@ -93,5 +93,32 @@ export declare const unixtimeToHybridts: (unixtime: bigint | string) => string;
|
|
|
93
93
|
* ```
|
|
94
94
|
*/
|
|
95
95
|
export declare const datetimeToHybrids: (datetime: Date) => string;
|
|
96
|
+
/**
|
|
97
|
+
* Converts a string to base64 encoding.
|
|
98
|
+
* @param str The string to convert.
|
|
99
|
+
* @returns The base64 encoded string.
|
|
100
|
+
*/
|
|
96
101
|
export declare const stringToBase64: (str: string) => string;
|
|
102
|
+
/**
|
|
103
|
+
* Formats the given address by removing the http or https prefix and appending the default Milvus port if necessary.
|
|
104
|
+
* @param address The address to format.
|
|
105
|
+
* @returns The formatted address.
|
|
106
|
+
*/
|
|
97
107
|
export declare const formatAddress: (address: string) => string;
|
|
108
|
+
/**
|
|
109
|
+
* Assigns properties with keys `dim` or `max_length` to the `type_params` object of a `FieldType` object.
|
|
110
|
+
* If the property exists in the `field` object, it is converted to a string and then deleted from the `field` object.
|
|
111
|
+
* If the property already exists in the `type_params` object, it is also converted to a string.
|
|
112
|
+
*
|
|
113
|
+
* @param field The `FieldType` object to modify.
|
|
114
|
+
* @returns The modified `FieldType` object.
|
|
115
|
+
*/
|
|
116
|
+
export declare const assignTypeParams: (field: FieldType, typeParamKeys?: string[]) => any;
|
|
117
|
+
/**
|
|
118
|
+
* Parses a time token and returns the corresponding number of milliseconds.
|
|
119
|
+
*
|
|
120
|
+
* @param {string} token - The time token to parse.
|
|
121
|
+
* @returns {number} The number of milliseconds corresponding to the time token.
|
|
122
|
+
* @throws {Error} If the time token is invalid.
|
|
123
|
+
*/
|
|
124
|
+
export declare const parseTimeToken: (token: string) => number;
|
package/dist/utils/Format.js
CHANGED
|
@@ -9,10 +9,9 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
|
|
|
9
9
|
return to.concat(ar || Array.prototype.slice.call(from));
|
|
10
10
|
};
|
|
11
11
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
12
|
-
exports.formatAddress = exports.stringToBase64 = exports.datetimeToHybrids = exports.unixtimeToHybridts = exports.hybridtsToUnixtime = exports.checkTimeParam = exports.formatNumberPrecision = exports.parseToKeyValue = exports.formatKeyValueData = void 0;
|
|
12
|
+
exports.parseTimeToken = exports.assignTypeParams = exports.formatAddress = exports.stringToBase64 = exports.datetimeToHybrids = exports.unixtimeToHybridts = exports.hybridtsToUnixtime = exports.checkTimeParam = exports.formatNumberPrecision = exports.parseToKeyValue = exports.formatKeyValueData = void 0;
|
|
13
13
|
var _1 = require("./");
|
|
14
|
-
var
|
|
15
|
-
var ErrorReason_1 = require("../milvus/const/ErrorReason");
|
|
14
|
+
var milvus_1 = require("../milvus");
|
|
16
15
|
/**
|
|
17
16
|
* parse [{key:"row_count",value:4}] to {row_count:4}
|
|
18
17
|
* @param data key value pair array
|
|
@@ -95,7 +94,7 @@ exports.checkTimeParam = checkTimeParam;
|
|
|
95
94
|
*/
|
|
96
95
|
var hybridtsToUnixtime = function (hybridts) {
|
|
97
96
|
if (!(0, exports.checkTimeParam)(hybridts)) {
|
|
98
|
-
throw new Error("hybridts ".concat(
|
|
97
|
+
throw new Error("hybridts ".concat(milvus_1.ERROR_REASONS.TIMESTAMP_PARAM_CHECK));
|
|
99
98
|
}
|
|
100
99
|
var timestamp = typeof hybridts === 'bigint' ? hybridts : BigInt(hybridts);
|
|
101
100
|
var physical = timestamp >> LOGICAL_BITS;
|
|
@@ -126,7 +125,7 @@ exports.hybridtsToUnixtime = hybridtsToUnixtime;
|
|
|
126
125
|
*/
|
|
127
126
|
var unixtimeToHybridts = function (unixtime) {
|
|
128
127
|
if (!(0, exports.checkTimeParam)(unixtime)) {
|
|
129
|
-
throw new Error("hybridts ".concat(
|
|
128
|
+
throw new Error("hybridts ".concat(milvus_1.ERROR_REASONS.TIMESTAMP_PARAM_CHECK));
|
|
130
129
|
}
|
|
131
130
|
var timestamp = typeof unixtime === 'bigint' ? unixtime : BigInt(unixtime);
|
|
132
131
|
var physical = (timestamp * BigInt(1000)) << LOGICAL_BITS;
|
|
@@ -157,19 +156,86 @@ exports.unixtimeToHybridts = unixtimeToHybridts;
|
|
|
157
156
|
*/
|
|
158
157
|
var datetimeToHybrids = function (datetime) {
|
|
159
158
|
if (!(datetime instanceof Date)) {
|
|
160
|
-
throw new Error("hybridts ".concat(
|
|
159
|
+
throw new Error("hybridts ".concat(milvus_1.ERROR_REASONS.DATE_TYPE_CHECK));
|
|
161
160
|
}
|
|
162
161
|
return (0, exports.unixtimeToHybridts)((datetime.getTime() / 1000).toString());
|
|
163
162
|
};
|
|
164
163
|
exports.datetimeToHybrids = datetimeToHybrids;
|
|
164
|
+
/**
|
|
165
|
+
* Converts a string to base64 encoding.
|
|
166
|
+
* @param str The string to convert.
|
|
167
|
+
* @returns The base64 encoded string.
|
|
168
|
+
*/
|
|
165
169
|
var stringToBase64 = function (str) {
|
|
166
170
|
return Buffer.from(str, 'utf-8').toString('base64');
|
|
167
171
|
};
|
|
168
172
|
exports.stringToBase64 = stringToBase64;
|
|
173
|
+
/**
|
|
174
|
+
* Formats the given address by removing the http or https prefix and appending the default Milvus port if necessary.
|
|
175
|
+
* @param address The address to format.
|
|
176
|
+
* @returns The formatted address.
|
|
177
|
+
*/
|
|
169
178
|
var formatAddress = function (address) {
|
|
170
179
|
// remove http or https prefix from address
|
|
171
180
|
var ip = address.replace(/(http|https)*:\/\//, '');
|
|
172
|
-
return ip.includes(':') ? ip : "".concat(ip, ":").concat(
|
|
181
|
+
return ip.includes(':') ? ip : "".concat(ip, ":").concat(milvus_1.DEFAULT_MILVUS_PORT);
|
|
173
182
|
};
|
|
174
183
|
exports.formatAddress = formatAddress;
|
|
184
|
+
/**
|
|
185
|
+
* Assigns properties with keys `dim` or `max_length` to the `type_params` object of a `FieldType` object.
|
|
186
|
+
* If the property exists in the `field` object, it is converted to a string and then deleted from the `field` object.
|
|
187
|
+
* If the property already exists in the `type_params` object, it is also converted to a string.
|
|
188
|
+
*
|
|
189
|
+
* @param field The `FieldType` object to modify.
|
|
190
|
+
* @returns The modified `FieldType` object.
|
|
191
|
+
*/
|
|
192
|
+
var assignTypeParams = function (field, typeParamKeys) {
|
|
193
|
+
if (typeParamKeys === void 0) { typeParamKeys = ['dim', 'max_length']; }
|
|
194
|
+
var newField = JSON.parse(JSON.stringify(field));
|
|
195
|
+
typeParamKeys.forEach(function (key) {
|
|
196
|
+
if (newField.hasOwnProperty(key)) {
|
|
197
|
+
// if the property exists in the field object, assign it to the type_params object
|
|
198
|
+
newField.type_params = newField.type_params || {};
|
|
199
|
+
newField.type_params[key] = String(newField[key]);
|
|
200
|
+
// delete the property from the field object
|
|
201
|
+
delete newField[key];
|
|
202
|
+
}
|
|
203
|
+
if (newField.type_params && newField.type_params[key]) {
|
|
204
|
+
// if the property already exists in the type_params object, convert it to a string
|
|
205
|
+
newField.type_params[key] = String(newField.type_params[key]);
|
|
206
|
+
}
|
|
207
|
+
});
|
|
208
|
+
return newField;
|
|
209
|
+
};
|
|
210
|
+
exports.assignTypeParams = assignTypeParams;
|
|
211
|
+
/**
|
|
212
|
+
* Parses a time token and returns the corresponding number of milliseconds.
|
|
213
|
+
*
|
|
214
|
+
* @param {string} token - The time token to parse.
|
|
215
|
+
* @returns {number} The number of milliseconds corresponding to the time token.
|
|
216
|
+
* @throws {Error} If the time token is invalid.
|
|
217
|
+
*/
|
|
218
|
+
var parseTimeToken = function (token) {
|
|
219
|
+
var num = parseInt(token.slice(0, -1));
|
|
220
|
+
var unit = token.slice(-1);
|
|
221
|
+
switch (unit) {
|
|
222
|
+
case 's':
|
|
223
|
+
return num * 1000;
|
|
224
|
+
case 'm':
|
|
225
|
+
return num * 60 * 1000;
|
|
226
|
+
case 'h':
|
|
227
|
+
return num * 60 * 60 * 1000;
|
|
228
|
+
case 'd':
|
|
229
|
+
return num * 24 * 60 * 60 * 1000;
|
|
230
|
+
case 'w':
|
|
231
|
+
return num * 7 * 24 * 60 * 60 * 1000;
|
|
232
|
+
case 'M':
|
|
233
|
+
return num * 30 * 24 * 60 * 60 * 1000;
|
|
234
|
+
case 'Y':
|
|
235
|
+
return num * 365 * 24 * 60 * 60 * 1000;
|
|
236
|
+
default:
|
|
237
|
+
throw new Error("Invalid time token: ".concat(token));
|
|
238
|
+
}
|
|
239
|
+
};
|
|
240
|
+
exports.parseTimeToken = parseTimeToken;
|
|
175
241
|
//# sourceMappingURL=Format.js.map
|
package/dist/utils/Format.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"Format.js","sourceRoot":"","sources":["../../utils/Format.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,uBAAkC;AAClC,
|
|
1
|
+
{"version":3,"file":"Format.js","sourceRoot":"","sources":["../../utils/Format.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,uBAAkC;AAClC,oCAKmB;AAEnB;;;;;GAKG;AACI,IAAM,kBAAkB,GAAG,UAAC,IAAoB,EAAE,IAAc;IACrE,IAAM,MAAM,GAAyB,EAAE,CAAC;IACxC,IAAI,CAAC,OAAO,CAAC,UAAA,CAAC;QACZ,IAAM,KAAK,GAAG,IAAA,eAAY,EAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC;AARW,QAAA,kBAAkB,sBAQ7B;AAEF;;;;GAIG;AACI,IAAM,eAAe,GAAG,UAAC,IAE/B;IACC,OAAO,IAAI;QACT,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,CACtB,UAAC,GAAU,EAAE,GAAW,IAAK,uCAAI,GAAG,UAAE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,WAAvC,CAAwC,EACrE,EAAE,CACH;QACH,CAAC,CAAC,EAAE,CAAC;AACT,CAAC,CAAC;AATW,QAAA,eAAe,mBAS1B;AAEF;;;;;GAKG;AACI,IAAM,qBAAqB,GAAG,UAAC,MAAc,EAAE,SAAiB;IACrE,OAAO,MAAM,CACX,MAAM;SACH,QAAQ,EAAE;SACV,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,UAAC,CAAC,EAAE,CAAC;QACR,IAAI,CAAC,KAAK,CAAC,EAAE;YACX,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;SAC9B;QACD,OAAO,CAAC,CAAC;IACX,CAAC,CAAC;SACD,IAAI,CAAC,GAAG,CAAC,CACb,CAAC;AACJ,CAAC,CAAC;AAbW,QAAA,qBAAqB,yBAahC;AAEF,IAAM,YAAY,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;AAChC,qDAAqD;AAE9C,IAAM,cAAc,GAAG,UAAC,EAAO;IACpC,QAAQ,OAAO,EAAE,EAAE;QACjB,KAAK,QAAQ;YACX,OAAO,IAAI,CAAC;QACd,KAAK,QAAQ;YACX,OAAO,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1C;YACE,OAAO,KAAK,CAAC;KAChB;AACH,CAAC,CAAC;AATW,QAAA,cAAc,kBASzB;AAEF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACI,IAAM,kBAAkB,GAAG,UAAC,QAAyB;IAC1D,IAAI,CAAC,IAAA,sBAAc,EAAC,QAAQ,CAAC,EAAE;QAC7B,MAAM,IAAI,KAAK,CAAC,mBAAY,sBAAa,CAAC,qBAAqB,CAAE,CAAC,CAAC;KACpE;IACD,IAAM,SAAS,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC7E,IAAM,QAAQ,GAAG,SAAS,IAAI,YAAY,CAAC;IAC3C,OAAO,CAAC,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;AAC9C,CAAC,CAAC;AAPW,QAAA,kBAAkB,sBAO7B;AAEF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACI,IAAM,kBAAkB,GAAG,UAAC,QAAyB;IAC1D,IAAI,CAAC,IAAA,sBAAc,EAAC,QAAQ,CAAC,EAAE;QAC7B,MAAM,IAAI,KAAK,CAAC,mBAAY,sBAAa,CAAC,qBAAqB,CAAE,CAAC,CAAC;KACpE;IACD,IAAM,SAAS,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAE7E,IAAM,QAAQ,GAAG,CAAC,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,YAAY,CAAC;IAC5D,OAAO,QAAQ,CAAC,QAAQ,EAAE,CAAC;AAC7B,CAAC,CAAC;AARW,QAAA,kBAAkB,sBAQ7B;AAEF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACI,IAAM,iBAAiB,GAAG,UAAC,QAAc;IAC9C,IAAI,CAAC,CAAC,QAAQ,YAAY,IAAI,CAAC,EAAE;QAC/B,MAAM,IAAI,KAAK,CAAC,mBAAY,sBAAa,CAAC,eAAe,CAAE,CAAC,CAAC;KAC9D;IACD,OAAO,IAAA,0BAAkB,EAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;AACpE,CAAC,CAAC;AALW,QAAA,iBAAiB,qBAK5B;AAEF;;;;GAIG;AACI,IAAM,cAAc,GAAG,UAAC,GAAW;IACxC,OAAA,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;AAA5C,CAA4C,CAAC;AADlC,QAAA,cAAc,kBACoB;AAE/C;;;;GAIG;AACI,IAAM,aAAa,GAAG,UAAC,OAAe;IAC3C,2CAA2C;IAC3C,IAAM,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;IACrD,OAAO,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAG,EAAE,cAAI,4BAAmB,CAAE,CAAC;AAChE,CAAC,CAAC;AAJW,QAAA,aAAa,iBAIxB;AAEF;;;;;;;GAOG;AACI,IAAM,gBAAgB,GAAG,UAC9B,KAAgB,EAChB,aAA+C;IAA/C,8BAAA,EAAA,iBAA2B,KAAK,EAAE,YAAY,CAAC;IAE/C,IAAI,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;IACjD,aAAa,CAAC,OAAO,CAAC,UAAA,GAAG;QACvB,IAAI,QAAQ,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE;YAChC,kFAAkF;YAClF,QAAQ,CAAC,WAAW,GAAG,QAAQ,CAAC,WAAW,IAAI,EAAE,CAAC;YAClD,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAClD,4CAA4C;YAC5C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;SACtB;QAED,IAAI,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE;YACrD,mFAAmF;YACnF,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;SAC/D;IACH,CAAC,CAAC,CAAC;IACH,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AApBW,QAAA,gBAAgB,oBAoB3B;AAEF;;;;;;GAMG;AACI,IAAM,cAAc,GAAG,UAAC,KAAa;IAC1C,IAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,IAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7B,QAAQ,IAAI,EAAE;QACZ,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,IAAI,CAAC;QACpB,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC;QACzB,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAC9B,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACnC,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACvC,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACxC,KAAK,GAAG;YACN,OAAO,GAAG,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACzC;YACE,MAAM,IAAI,KAAK,CAAC,8BAAuB,KAAK,CAAE,CAAC,CAAC;KACnD;AACH,CAAC,CAAC;AArBW,QAAA,cAAc,kBAqBzB"}
|
package/dist/utils/Function.d.ts
CHANGED
|
@@ -1,4 +1,12 @@
|
|
|
1
1
|
import { KeyValuePair } from '../milvus/types/Common';
|
|
2
|
-
|
|
2
|
+
/**
|
|
3
|
+
* Promisify a function call with optional timeout
|
|
4
|
+
* @param obj - The object containing the target function
|
|
5
|
+
* @param target - The name of the target function to call
|
|
6
|
+
* @param params - The parameters to pass to the target function
|
|
7
|
+
* @param timeout - Optional timeout in milliseconds
|
|
8
|
+
* @returns A Promise that resolves with the result of the target function call
|
|
9
|
+
*/
|
|
10
|
+
export declare function promisify(obj: any, target: string, params: any, timeout: number): Promise<any>;
|
|
3
11
|
export declare const findKeyValue: (obj: KeyValuePair[], key: string) => string | number | undefined;
|
|
4
12
|
export declare const sleep: (time: number) => Promise<unknown>;
|
package/dist/utils/Function.js
CHANGED
|
@@ -1,26 +1,39 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.sleep = exports.findKeyValue = exports.promisify = void 0;
|
|
4
|
-
|
|
4
|
+
/**
|
|
5
|
+
* Promisify a function call with optional timeout
|
|
6
|
+
* @param obj - The object containing the target function
|
|
7
|
+
* @param target - The name of the target function to call
|
|
8
|
+
* @param params - The parameters to pass to the target function
|
|
9
|
+
* @param timeout - Optional timeout in milliseconds
|
|
10
|
+
* @returns A Promise that resolves with the result of the target function call
|
|
11
|
+
*/
|
|
5
12
|
function promisify(obj, target, params, timeout) {
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
13
|
+
// Calculate the deadline for the function call
|
|
14
|
+
var deadline = new Date(Date.now() + timeout);
|
|
15
|
+
// Create a new Promise that wraps the target function call
|
|
9
16
|
var res = new Promise(function (resolve, reject) {
|
|
10
17
|
try {
|
|
18
|
+
// Call the target function with the provided parameters and deadline
|
|
11
19
|
obj[target](params, { deadline: deadline }, function (err, result) {
|
|
12
20
|
if (err) {
|
|
21
|
+
// If there was an error, reject the Promise with the error
|
|
13
22
|
reject(err);
|
|
14
23
|
}
|
|
24
|
+
// Otherwise, resolve the Promise with the result
|
|
15
25
|
resolve(result);
|
|
16
26
|
});
|
|
17
27
|
}
|
|
18
28
|
catch (e) {
|
|
29
|
+
// If there was an exception, throw a new Error
|
|
19
30
|
throw new Error(e);
|
|
20
31
|
}
|
|
21
32
|
}).catch(function (err) {
|
|
22
|
-
|
|
33
|
+
// Return a rejected Promise with the error
|
|
34
|
+
return Promise.reject(err);
|
|
23
35
|
});
|
|
36
|
+
// Return the Promise
|
|
24
37
|
return res;
|
|
25
38
|
}
|
|
26
39
|
exports.promisify = promisify;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"Function.js","sourceRoot":"","sources":["../../utils/Function.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"Function.js","sourceRoot":"","sources":["../../utils/Function.ts"],"names":[],"mappings":";;;AAGA;;;;;;;GAOG;AACH,SAAgB,SAAS,CACvB,GAAQ,EACR,MAAc,EACd,MAAW,EACX,OAAe;IAEf,+CAA+C;IAC/C,IAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,CAAC;IAEhD,2DAA2D;IAC3D,IAAM,GAAG,GAAG,IAAI,OAAO,CAAC,UAAC,OAAO,EAAE,MAAM;QACtC,IAAI;YACF,qEAAqE;YACrE,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,QAAQ,UAAA,EAAE,EAAE,UAAC,GAAQ,EAAE,MAAW;gBACtD,IAAI,GAAG,EAAE;oBACP,2DAA2D;oBAC3D,MAAM,CAAC,GAAG,CAAC,CAAC;iBACb;gBACD,iDAAiD;gBACjD,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;SACJ;QAAC,OAAO,CAAM,EAAE;YACf,+CAA+C;YAC/C,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;SACpB;IACH,CAAC,CAAC,CAAC,KAAK,CAAC,UAAA,GAAG;QACV,2CAA2C;QAC3C,OAAO,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,qBAAqB;IACrB,OAAO,GAAG,CAAC;AACb,CAAC;AAhCD,8BAgCC;AAEM,IAAM,YAAY,GAAG,UAAC,GAAmB,EAAE,GAAW,YAC3D,OAAA,MAAA,GAAG,CAAC,IAAI,CAAC,UAAA,CAAC,IAAI,OAAA,CAAC,CAAC,GAAG,KAAK,GAAG,EAAb,CAAa,CAAC,0CAAE,KAAK,CAAA,EAAA,CAAC;AADzB,QAAA,YAAY,gBACa;AAE/B,IAAM,KAAK,GAAG,UAAC,IAAY;IAChC,OAAO,IAAI,OAAO,CAAC,UAAA,OAAO,IAAI,OAAA,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,EAAzB,CAAyB,CAAC,CAAC;AAC3D,CAAC,CAAC;AAFW,QAAA,KAAK,SAEhB"}
|
package/dist/utils/Validate.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { FieldType } from '../milvus
|
|
1
|
+
import { FieldType } from '../milvus';
|
|
2
2
|
/**
|
|
3
3
|
* when create collection, field must contain 2 Fields.
|
|
4
4
|
* Type is int64 or varchar and primary_key = true
|
|
@@ -19,3 +19,9 @@ export declare const checkCollectionName: (data: any) => void;
|
|
|
19
19
|
* @param data
|
|
20
20
|
*/
|
|
21
21
|
export declare const checkCollectionAndPartitionName: (data: any) => void;
|
|
22
|
+
/**
|
|
23
|
+
* Checks the validity of search parameters.
|
|
24
|
+
* @param {Object} data - The search parameters to be checked.
|
|
25
|
+
* @throws {Error} Throws an error if any of the search parameters are invalid.
|
|
26
|
+
*/
|
|
27
|
+
export declare const checkSearchParams: (data: any) => void;
|
package/dist/utils/Validate.js
CHANGED
|
@@ -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
|
|
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
|
-
|
|
15
|
-
|
|
16
|
-
var
|
|
17
|
-
if
|
|
18
|
-
|
|
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
|
|
21
|
-
|
|
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
|
|
24
|
-
|
|
25
|
-
|
|
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(
|
|
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
|
-
|
|
30
|
-
|
|
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(
|
|
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(
|
|
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,
|
|
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"}
|
package/dist/utils/test.d.ts
CHANGED
|
@@ -1,14 +1,15 @@
|
|
|
1
|
-
import { DataType } from '../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
|
-
export declare const genCollectionParams: (collectionName: string, dim: string, vectorType?: DataType.FloatVector | DataType.BinaryVector, autoID?: boolean, fields?: any[]) => {
|
|
4
|
+
export declare const genCollectionParams: (collectionName: string, dim: string | number, vectorType?: DataType.FloatVector | DataType.BinaryVector, autoID?: boolean, fields?: any[]) => {
|
|
5
5
|
collection_name: string;
|
|
6
6
|
fields: any[];
|
|
7
7
|
};
|
|
8
8
|
export declare const GENERATE_NAME: (pre?: string) => string;
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
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;
|
package/dist/utils/test.js
CHANGED
|
@@ -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
|
|
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 =
|
|
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
|
-
|
|
28
|
-
dim: dim,
|
|
29
|
-
},
|
|
27
|
+
dim: Number(dim),
|
|
30
28
|
},
|
|
31
29
|
{
|
|
32
30
|
name: 'age',
|
|
33
|
-
|
|
34
|
-
|
|
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
|
-
|
|
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
|
-
|
|
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
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
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
|
package/dist/utils/test.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"test.js","sourceRoot":"","sources":["../../utils/test.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,
|
|
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,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@zilliz/milvus2-sdk-node",
|
|
3
3
|
"author": "ued@zilliz.com",
|
|
4
|
-
"version": "2.2.
|
|
4
|
+
"version": "2.2.7",
|
|
5
5
|
"milvusVersion": "v2.2.5",
|
|
6
6
|
"main": "dist/milvus",
|
|
7
7
|
"files": [
|
|
@@ -19,9 +19,9 @@
|
|
|
19
19
|
"doc-json": "npx typedoc milvus --json"
|
|
20
20
|
},
|
|
21
21
|
"dependencies": {
|
|
22
|
-
"@grpc/grpc-js": "^1.
|
|
23
|
-
"@grpc/proto-loader": "^0.7.
|
|
24
|
-
"protobufjs": "^7.
|
|
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",
|