@zilliz/milvus2-sdk-node 2.2.9 → 2.2.11

package/dist/utils/Validate.js

@@ -1,124 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isStatusCodeMatched = exports.checkSearchParams = exports.checkCollectionAndPartitionName = exports.checkCollectionName = exports.checkCollectionFields = void 0;
-var milvus_1 = require("../milvus");
-var grpc_js_1 = require("@grpc/grpc-js");
-/**
- * when create collection, field must contain 2 Fields.
- * Type is int64 or varchar and primary_key = true
- * Type is one of float_vector and binary_vector
- * Will check fields
- * @param fields
- */
-var checkCollectionFields = function (fields) {
-    // Define arrays of data types that are allowed for vector fields and primary keys, respectively
-    var vectorDataTypes = [milvus_1.DataType.BinaryVector, milvus_1.DataType.FloatVector];
-    var primaryKeyDataTypes = [milvus_1.DataType.Int64, milvus_1.DataType.VarChar];
-    // Check if every field in the `fields` array has a `data_type` property
-    var hasDataTypeKey = fields.every(function (field) {
-        return field.hasOwnProperty('data_type');
-    });
-    // If `hasDataTypeKey` is false, an error is thrown indicating that the `data_type` property is missing
-    if (!hasDataTypeKey) {
-        throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_MISS_DATA_TYPE);
-    }
-    // Check if at least one field in the `fields` array is a primary key with a supported data type
-    var hasPrimaryKey = fields.some(function (field) {
-        var isPrimaryKey = field.is_primary_key;
-        var isSupportedDataType = primaryKeyDataTypes.includes(field.data_type);
-        return isPrimaryKey && isSupportedDataType;
-    });
-    // If `hasPrimaryKey` is false, an error is thrown indicating that a primary key is missing or has an unsupported data type
-    if (!hasPrimaryKey) {
-        throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_PRIMARY_KEY);
-    }
-    // Check if at least one field in the `fields` array is a vector field with a supported data type
-    var hasVectorField = fields.some(function (field) {
-        var isVectorField = vectorDataTypes.includes(field.data_type);
-        return isVectorField;
-    });
-    // If `hasVectorField` is false, an error is thrown indicating that a vector field is missing or has an unsupported data type
-    if (!hasVectorField) {
-        throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_VECTOR_FIELD_EXIST);
-    }
-    fields.forEach(function (field) {
-        var _a, _b;
-        var dataType = field.data_type;
-        var typeParams = field.type_params;
-        var isVectorField = vectorDataTypes.includes(dataType);
-        // Check if field is a vector field
-        if (isVectorField) {
-            var dim = Number((_a = typeParams === null || typeParams === void 0 ? void 0 : typeParams.dim) !== null && _a !== void 0 ? _a : field.dim);
-            // Check if vector field has a dimension
-            if (!dim) {
-                throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_DIM);
-            }
-            // Check if binary vector field has a dimension that is a multiple of 8
-            if (dataType === milvus_1.DataType.BinaryVector && dim % 8 !== 0) {
-                throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_BINARY_DIM);
-            }
-        }
-        // Check if varchar field has a max length
-        if (dataType === milvus_1.DataType.VarChar) {
-            var maxLength = (_b = typeParams === null || typeParams === void 0 ? void 0 : typeParams.max_length) !== null && _b !== void 0 ? _b : field.max_length;
-            if (!maxLength) {
-                throw new Error(milvus_1.ERROR_REASONS.CREATE_COLLECTION_CHECK_MISS_MAXLENGTH);
-            }
-        }
-    });
-    return true;
-};
-exports.checkCollectionFields = checkCollectionFields;
-/**
- * check if the request contains collection_name
- * otherwise throw an error
- * @param data
- */
-var checkCollectionName = function (data) {
-    if (!data || !data.collection_name) {
-        throw new Error(milvus_1.ERROR_REASONS.COLLECTION_NAME_IS_REQUIRED);
-    }
-};
-exports.checkCollectionName = checkCollectionName;
-/**
- * check if the request contains collection_name and partition_name
- * otherwise throw an error
- * @param data
- */
-var checkCollectionAndPartitionName = function (data) {
-    if (!data || !data.collection_name || !data.partition_name) {
-        throw new Error(milvus_1.ERROR_REASONS.COLLECTION_PARTITION_NAME_ARE_REQUIRED);
-    }
-};
-exports.checkCollectionAndPartitionName = checkCollectionAndPartitionName;
-/**
- * Checks the validity of search parameters.
- * @param {Object} data - The search parameters to be checked.
- * @throws {Error} Throws an error if any of the search parameters are invalid.
- */
-var checkSearchParams = function (data) {
-    (0, exports.checkCollectionName)(data);
-    if (!data.vectors && !data.vector) {
-        throw new Error(milvus_1.ERROR_REASONS.VECTORS_OR_VECTOR_IS_MISSING);
-    }
-};
-exports.checkSearchParams = checkSearchParams;
-/**
- * Checks if a gRPC status code matches any of the given codes.
- * DEADLINE_EXCEEDED means that the task wat not completed
- * UNAVAILABLE means that the service is not reachable currently
- * Reference: https://grpc.github.io/grpc/python/grpc.html#grpc-status-code
- *
- * @param {number} code - The gRPC status code to check.
- * @param {number[]} [codesToCheck=[grpcStatus.DEADLINE_EXCEEDED, grpcStatus.UNAVAILABLE]] - An array of gRPC status codes to check against.
- * @returns {boolean} Whether the gRPC status code matches any of the given codes.
- */
-var isStatusCodeMatched = function (code, codesToCheck) {
-    if (codesToCheck === void 0) { codesToCheck = [
-        grpc_js_1.status.DEADLINE_EXCEEDED,
-        grpc_js_1.status.UNAVAILABLE,
-    ]; }
-    return codesToCheck.includes(code);
-};
-exports.isStatusCodeMatched = isStatusCodeMatched;
-//# sourceMappingURL=Validate.js.map

package/dist/utils/Validate.js.map

@@ -1 +0,0 @@
-{"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,YAGC;IAHD,6BAAA,EAAA;QACE,gBAAU,CAAC,iBAAiB;QAC5B,gBAAU,CAAC,WAAW;KACvB;IAED,OAAO,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC,CAAC;AARW,QAAA,mBAAmB,uBAQ9B"}

package/dist/utils/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../utils/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,yCAAuB;AACvB,2CAAyB;AACzB,6CAA2B;AAC3B,6CAA2B"}

package/dist/utils/test.d.ts

@@ -1,15 +0,0 @@
-import { DataType, FieldType } from '../milvus';
-export declare const VECTOR_FIELD_NAME = "vector_field";
-export declare const INDEX_NAME = "index_name";
-export declare const genCollectionParams: (collectionName: string, dim: string | number, vectorType?: DataType.FloatVector | DataType.BinaryVector, autoID?: boolean, fields?: any[]) => {
-    collection_name: string;
-    fields: any[];
-};
-export declare const GENERATE_NAME: (pre?: string) => string;
-/**
- * Generates random data for inserting into a collection
- * @param fields An array of objects describing the fields to generate data for
- * @param count The number of data points to generate
- * @returns An array of objects representing the generated data
- */
-export declare function generateInsertData(fields: FieldType[], count?: number): any;

package/dist/utils/test.js

@@ -1,98 +0,0 @@
-"use strict";
-var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
-    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
-        if (ar || !(i in from)) {
-            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
-            ar[i] = from[i];
-        }
-    }
-    return to.concat(ar || Array.prototype.slice.call(from));
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateInsertData = exports.GENERATE_NAME = exports.genCollectionParams = exports.INDEX_NAME = exports.VECTOR_FIELD_NAME = void 0;
-var milvus_1 = require("../milvus");
-exports.VECTOR_FIELD_NAME = 'vector_field';
-exports.INDEX_NAME = 'index_name';
-var genCollectionParams = function (collectionName, dim, vectorType, autoID, fields) {
-    if (vectorType === void 0) { vectorType = milvus_1.DataType.FloatVector; }
-    if (autoID === void 0) { autoID = true; }
-    fields = fields || [];
-    return {
-        collection_name: collectionName,
-        fields: __spreadArray([
-            {
-                name: exports.VECTOR_FIELD_NAME,
-                description: 'vector field',
-                data_type: vectorType,
-                dim: Number(dim),
-            },
-            {
-                name: 'age',
-                description: '',
-                data_type: milvus_1.DataType.Int64,
-                is_primary_key: true,
-                autoID: autoID,
-            },
-            {
-                name: 'height',
-                description: '',
-                data_type: milvus_1.DataType.Int64,
-            },
-            {
-                name: 'name',
-                description: '',
-                data_type: milvus_1.DataType.VarChar,
-                max_length: 128,
-            }
-        ], fields, true),
-    };
-};
-exports.genCollectionParams = genCollectionParams;
-var GENERATE_NAME = function (pre) {
-    if (pre === void 0) { pre = 'collection'; }
-    return "".concat(pre, "_").concat(Math.random().toString(36).substr(2, 8));
-};
-exports.GENERATE_NAME = GENERATE_NAME;
-/**
- * Generates random data for inserting into a collection
- * @param fields An array of objects describing the fields to generate data for
- * @param count The number of data points to generate
- * @returns An array of objects representing the generated data
- */
-function generateInsertData(fields, count) {
-    if (count === void 0) { count = 10; }
-    var results = []; // Initialize an empty array to store the generated data
-    var _loop_1 = function () {
-        // Loop until we've generated the desired number of data points
-        var value = {}; // Initialize an empty object to store the generated values for this data point
-        fields.forEach(function (v) {
-            // bypass autoID
-            if (v.autoID) {
-                return;
-            }
-            // Loop through each field we need to generate data for
-            var name = v.name; // Destructure the field object to get its properties
-            var isVector = v.data_type === milvus_1.DataType.BinaryVector ||
-                v.data_type === milvus_1.DataType.FloatVector;
-            var dim = v.dim || (v.type_params && v.type_params.dim);
-            var isBool = v.data_type === milvus_1.DataType.Bool;
-            var isVarChar = v.data_type === milvus_1.DataType.VarChar;
-            dim = v.data_type === milvus_1.DataType.BinaryVector ? dim / 8 : dim;
-            value[name] = isVector // If the field is a vector field
-                ? __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
-                : isBool // If the field is a boolean field
-                    ? count % 2 === 0 // Generate a random boolean value based on the current count
-                    : isVarChar // If the field is a varchar field
-                        ? Math.random().toString(36).substring(2, 7) // Generate a random string of characters
-                        : Math.floor(Math.random() * 100000); // Otherwise, generate a random integer between 0 and 100000
-        });
-        results.push(value); // Add the generated values for this data point to the results array
-        count--; // Decrement the count to keep track of how many data points we've generated so far
-    };
-    while (count > 0) {
-        _loop_1();
-    }
-    return results; // Return the array of generated data
-}
-exports.generateInsertData = generateInsertData;
-//# sourceMappingURL=test.js.map

package/dist/utils/test.js.map

@@ -1 +0,0 @@
-{"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"}