weaviate-client
Version:
JS/TS client for Weaviate
184 lines (183 loc) • 10.2 kB
JavaScript
"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.Check = void 0;
const errors_js_1 = require("../../errors.js");
const index_js_1 = require("../serialize/index.js");
class Check {
constructor(connection, name, dbVersionSupport, consistencyLevel, tenant) {
this.getSearcher = () => this.connection.search(this.name, this.consistencyLevel, this.tenant);
this.checkSupportForNamedVectors = (opts) => __awaiter(this, void 0, void 0, function* () {
if (!index_js_1.Serialize.isNamedVectors(opts))
return;
const check = yield this.dbVersionSupport.supportsNamedVectors();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
});
this.checkSupportForBm25AndHybridGroupByQueries = (query, opts) => __awaiter(this, void 0, void 0, function* () {
if (!index_js_1.Serialize.search.isGroupBy(opts))
return;
const check = yield this.dbVersionSupport.supportsBm25AndHybridGroupByQueries();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message(query));
});
this.checkSupportForHybridNearTextAndNearVectorSubSearches = (opts) => __awaiter(this, void 0, void 0, function* () {
if ((opts === null || opts === void 0 ? void 0 : opts.vector) === undefined || Array.isArray(opts.vector))
return;
const check = yield this.dbVersionSupport.supportsHybridNearTextAndNearVectorSubsearchQueries();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
});
this.checkSupportForMultiTargetSearch = (opts) => __awaiter(this, void 0, void 0, function* () {
if (!index_js_1.Serialize.isMultiTarget(opts))
return false;
const check = yield this.dbVersionSupport.supportsMultiTargetVectorSearch();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
return check.supports;
});
this.checkSupportForMultiVectorSearch = (vec) => __awaiter(this, void 0, void 0, function* () {
if (vec === undefined || index_js_1.Serialize.isHybridNearTextSearch(vec))
return false;
if (index_js_1.Serialize.isHybridNearVectorSearch(vec) && !index_js_1.Serialize.isMultiVector(vec.vector))
return false;
if (index_js_1.Serialize.isHybridVectorSearch(vec) && !index_js_1.Serialize.isMultiVector(vec))
return false;
const check = yield this.dbVersionSupport.supportsMultiVectorSearch();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
return check.supports;
});
this.checkSupportForMultiWeightPerTargetSearch = (opts) => __awaiter(this, void 0, void 0, function* () {
if (!index_js_1.Serialize.isMultiWeightPerTarget(opts))
return false;
const check = yield this.dbVersionSupport.supportsMultiWeightsPerTargetSearch();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
return check.supports;
});
this.checkSupportForMultiVectorPerTargetSearch = (vec) => __awaiter(this, void 0, void 0, function* () {
if (vec === undefined || index_js_1.Serialize.isHybridNearTextSearch(vec))
return false;
if (index_js_1.Serialize.isHybridNearVectorSearch(vec) && !index_js_1.Serialize.isMultiVectorPerTarget(vec.vector))
return false;
if (index_js_1.Serialize.isHybridVectorSearch(vec) && !index_js_1.Serialize.isMultiVectorPerTarget(vec))
return false;
const check = yield this.dbVersionSupport.supportsMultiVectorPerTargetSearch();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
return check.supports;
});
this.checkSupportForVectors = (vec) => __awaiter(this, void 0, void 0, function* () {
if (vec === undefined || index_js_1.Serialize.isHybridNearTextSearch(vec))
return false;
const check = yield this.dbVersionSupport.supportsVectorsFieldInGRPC();
return check.supports;
});
this.supportForSingleGroupedGenerative = () => __awaiter(this, void 0, void 0, function* () {
const check = yield this.dbVersionSupport.supportsSingleGrouped();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
return check.supports;
});
this.supportForGenerativeConfigRuntime = (generativeConfig) => __awaiter(this, void 0, void 0, function* () {
if (generativeConfig === undefined)
return true;
const check = yield this.dbVersionSupport.supportsGenerativeConfigRuntime();
if (!check.supports)
throw new errors_js_1.WeaviateUnsupportedFeatureError(check.message);
return check.supports;
});
this.nearSearch = (opts) => {
return Promise.all([
this.getSearcher(),
this.checkSupportForMultiTargetSearch(opts),
this.checkSupportForMultiWeightPerTargetSearch(opts),
this.checkSupportForNamedVectors(opts),
]).then(([search, supportsTargets, supportsWeightsForTargets]) => {
const is126 = supportsTargets;
const is127 = supportsWeightsForTargets;
return { search, supportsTargets: is126 || is127, supportsWeightsForTargets: is127 };
});
};
this.nearVector = (vec, opts) => {
return Promise.all([
this.getSearcher(),
this.checkSupportForMultiTargetSearch(opts),
this.checkSupportForMultiVectorSearch(vec),
this.checkSupportForMultiVectorPerTargetSearch(vec),
this.checkSupportForMultiWeightPerTargetSearch(opts),
this.checkSupportForVectors(vec),
this.checkSupportForNamedVectors(opts),
]).then(([search, supportsMultiTarget, supportsMultiVector, supportsVectorsForTargets, supportsWeightsForTargets, supportsVectors,]) => {
const is126 = supportsMultiTarget || supportsMultiVector;
const is127 = supportsVectorsForTargets || supportsWeightsForTargets;
const is129 = supportsVectors;
return {
search,
supportsTargets: is126 || is127,
supportsVectorsForTargets: is127,
supportsWeightsForTargets: is127,
supportsVectors: is129,
};
});
};
this.hybridSearch = (opts) => {
return Promise.all([
this.getSearcher(),
this.checkSupportForMultiTargetSearch(opts),
this.checkSupportForMultiVectorSearch(opts === null || opts === void 0 ? void 0 : opts.vector),
this.checkSupportForMultiVectorPerTargetSearch(opts === null || opts === void 0 ? void 0 : opts.vector),
this.checkSupportForMultiWeightPerTargetSearch(opts),
this.checkSupportForVectors(opts === null || opts === void 0 ? void 0 : opts.vector),
this.checkSupportForNamedVectors(opts),
this.checkSupportForBm25AndHybridGroupByQueries('Hybrid', opts),
this.checkSupportForHybridNearTextAndNearVectorSubSearches(opts),
]).then(([search, supportsMultiTarget, supportsMultiVector, supportsWeightsForTargets, supportsVectorsForTargets, supportsVectors,]) => {
const is126 = supportsMultiTarget || supportsMultiVector;
const is127 = supportsVectorsForTargets || supportsWeightsForTargets;
const is129 = supportsVectors;
return {
search,
supportsTargets: is126 || is127,
supportsWeightsForTargets: is127,
supportsVectorsForTargets: is127,
supportsVectors: is129,
};
});
};
this.fetchObjects = (opts) => {
return Promise.all([this.getSearcher(), this.checkSupportForNamedVectors(opts)]).then(([search]) => {
return { search };
});
};
this.fetchObjectById = (opts) => {
return Promise.all([this.getSearcher(), this.checkSupportForNamedVectors(opts)]).then(([search]) => {
return { search };
});
};
this.bm25 = (opts) => {
return Promise.all([
this.getSearcher(),
this.checkSupportForNamedVectors(opts),
this.checkSupportForBm25AndHybridGroupByQueries('Bm25', opts),
]).then(([search]) => {
return { search };
});
};
this.connection = connection;
this.name = name;
this.dbVersionSupport = dbVersionSupport;
this.consistencyLevel = consistencyLevel;
this.tenant = tenant;
}
}
exports.Check = Check;