@relevanceai/sdk 1.1.0 → 1.6.0

package/README.md

@@ -56,18 +56,27 @@ import {QueryBuilder,DiscoveryClient,BulkInsertOutput} from "@relevanceai/sdk";
 ```javascript
 const discovery = new DiscoveryClient({ });
 const dataset = discovery.dataset('tshirts-prod');
-// Here we create some demo data. Replace this with your real data
+ // Here we create some demo data. Replace this with your real data
+const fakeVector = [];
+for (let i = 0; i < 768; i++) fakeVector.push(1);
 const tshirtsData = [];
-for (let i = 0; i < 100000; i++) {
-    tshirtsData.push({_id:`tshirt-${i}1`,color:'red',price:i/1000});
-    tshirtsData.push({_id:`tshirt-${i}2`,color:'blue',price:i/1000});
-    tshirtsData.push({_id:`tshirt-${i}3`,color:'orange',price:i/1000});
+for (let i = 0; i < 10000; i++) {
+  tshirtsData.push({_id:`tshirt-${i}1`,color:'red',price:i/1000,'title_text@1-0_vector_':fakeVector});
+  tshirtsData.push({_id:`tshirt-${i}2`,color:'blue',price:i/1000});
+  tshirtsData.push({_id:`tshirt-${i}3`,color:'orange',price:i/1000});
 }
-const res = await dataset.insertDocuments(tshirtsData,{batchSize:50000});
+const res = await dataset.insertDocuments(tshirtsData,{batchSize:10000});
 ```
 ### insertDocuments will output:
 ```javascript
-{"inserted":300000,"failed_documents":[]}
+{"inserted":30000,"failed_documents":[]}
+```
+## Text Search and Vector Search
+```javascript
+const builder = QueryBuilder();
+builder.query('red').text().vector('title_text@1-0_vector_',0.5).minimumRelevance(0.1);
+// .text() searches all fields. alternatively, use .text(field1).text(field2)... to search specific fields
+const searchResults = await dataset.search(builder);
 ```
 ## Filter and retrieve items
 ```javascript
@@ -88,7 +97,7 @@ const filteredItems = await dataset.search(filters);
     }
     ...
   ],
-  resultsSize: 102000,
+  resultsSize: 10200,
   aggregations: {},
   aggregates: {},
   aggregateStats: {}
@@ -105,21 +114,21 @@ const aggregatesResult = await dataset.search(aggregates);
 {
   aggregates:{
     color: {
-      results: { blue: 100000, orange: 100000, red: 100000 },
+      results: { blue: 10000, orange: 10000, red: 10000 },
       aggregates: {}
     },
     price: {
       results: {
-        '0': 30000,
-        '10': 30000,
-        '20': 30000,
-        '30': 30000,
-        '40': 30000,
-        '50': 30000,
-        '60': 30000,
-        '70': 30000,
-        '80': 30000,
-        '90': 30000
+        '0': 0000,
+        '10': 3000,
+        '20': 3000,
+        '30': 3000,
+        '40': 3000,
+        '50': 3000,
+        '60': 3000,
+        '70': 3000,
+        '80': 3000,
+        '90': 3000
       },
       aggregates: {}
     }

package/dist-cjs/generated/VectorApi.js

@@ -430,6 +430,14 @@ class VectorApiClient extends BaseClient_1._GenericClient {
             options
         });
     }
+    async clustercentroidsapiv2servicesclustercentroidslistpost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/list',
+            options
+        });
+    }
     async clustercentroidsgetapiservicesclustercentroidsgetget(input, options) {
         return this.SendRequest({
             input,
@@ -438,7 +446,15 @@ class VectorApiClient extends BaseClient_1._GenericClient {
             options
         });
     }
-    async insertclustercentroidsapiservicesclustercentroidsinsertpost(input, options) {
+    async clustercentroidsgetapiservicesclustercentroidsgetpost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/get',
+            options
+        });
+    }
+    async insertclustercentroids2apiservicesclustercentroidsinsertpost(input, options) {
         return this.SendRequest({
             input,
             method: 'post',
@@ -446,6 +462,38 @@ class VectorApiClient extends BaseClient_1._GenericClient {
             options
         });
     }
+    async updatecentroidsapiv2servicesclustercentroidsupdatepost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/update',
+            options
+        });
+    }
+    async deletecentroidsapiservicesclustercentroidscentroididdeleteget(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'get',
+            path: '/services/cluster/centroids/{centroid_id}/delete',
+            options
+        });
+    }
+    async deletecentroidsapiservicesclustercentroidscentroididdeletepost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/{centroid_id}/delete',
+            options
+        });
+    }
+    async clustercentroidsdeleteapiservicesclustercentroidsdeletepost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/delete',
+            options
+        });
+    }
     async clustercentroidsgetapiservicesclustercentroidsdocumentspost(input, options) {
         return this.SendRequest({
             input,
@@ -454,7 +502,39 @@ class VectorApiClient extends BaseClient_1._GenericClient {
             options
         });
     }
-    async clusteraggregateapiservicesclusteraggregatepost(input, options) {
+    async centroidsmetadatagetapiservicesclustercentroidsmetadataget(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'get',
+            path: '/services/cluster/centroids/metadata',
+            options
+        });
+    }
+    async centroidsmetadatapostapiv2servicesclustercentroidsmetadatapost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/metadata',
+            options
+        });
+    }
+    async centroidslistclosesttocenterv2servicesclustercentroidslistclosesttocenterpost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/list_closest_to_center',
+            options
+        });
+    }
+    async centroidslistfurthestfromcenterv2servicesclustercentroidslistfurthestfromcenterpost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/centroids/list_furthest_from_center',
+            options
+        });
+    }
+    async clusteraggregateapiv2servicesclusteraggregatepost(input, options) {
         return this.SendRequest({
             input,
             method: 'post',
@@ -462,7 +542,7 @@ class VectorApiClient extends BaseClient_1._GenericClient {
             options
         });
     }
-    async clusterfacetsapiservicesclusterfacetsget(input, options) {
+    async advancedclusterfacetsapiservicesclusterfacetsget(input, options) {
         return this.SendRequest({
             input,
             method: 'get',
@@ -470,6 +550,22 @@ class VectorApiClient extends BaseClient_1._GenericClient {
             options
         });
     }
+    async clusterlistservicesclusterlistget(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'get',
+            path: '/services/cluster/list',
+            options
+        });
+    }
+    async clusterlistmultiservicesclusterlistpost(input, options) {
+        return this.SendRequest({
+            input,
+            method: 'post',
+            path: '/services/cluster/list',
+            options
+        });
+    }
     async tagapiservicestaggertagpost(input, options) {
         return this.SendRequest({
             input,

package/dist-cjs/services/discovery/Dataset.js

@@ -27,8 +27,11 @@ class Dataset {
             if (arg instanceof _1._QueryBuilder) {
                 payload = { ...payload, ...arg.build() };
             }
-            else
+            else {
                 options = arg;
+                if (options.rawPayload)
+                    payload = { ...payload, ...options.rawPayload };
+            }
         }
         const reqCallback = async () => await this.client.apiClient.SimpleSearchPost(payload, { dataset_id: this.name });
         if (options.debounce && this.debounceTimer) {
@@ -92,7 +95,7 @@ class Dataset {
         return finalResults;
     }
     async updateDocumentsWhere(filters, partialUpdates) {
-        return (await this.client.apiClient.UpdateWhere({ filters: filters.body.filters, updates: partialUpdates })).body;
+        return (await this.client.apiClient.UpdateWhere({ filters: filters.build().filters, updates: partialUpdates })).body;
     }
     async getDocument(documentId) {
         return (await this.client.apiClient.GetDocument({ document_id: documentId })).body;
@@ -103,11 +106,11 @@ class Dataset {
     async deleteDocuments(documentIds) {
         var _a;
         const filters = (0, _1.QueryBuilder)().match('_id', documentIds);
-        return (await this.client.apiClient.DeleteWhere({ filters: (_a = filters.body.filters) !== null && _a !== void 0 ? _a : [] })).body;
+        return (await this.client.apiClient.DeleteWhere({ filters: (_a = filters.build().filters) !== null && _a !== void 0 ? _a : [] })).body;
     }
     async deleteDocumentsWhere(filters) {
         var _a;
-        return (await this.client.apiClient.DeleteWhere({ filters: (_a = filters.body.filters) !== null && _a !== void 0 ? _a : [] })).body;
+        return (await this.client.apiClient.DeleteWhere({ filters: (_a = filters.build().filters) !== null && _a !== void 0 ? _a : [] })).body;
     }
 }
 exports.Dataset = Dataset;

package/dist-cjs/services/discovery/index.js

@@ -1,41 +1,118 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DiscoveryClient = exports._QueryBuilder = exports.QueryBuilder = void 0;
+exports.DiscoveryClient = exports._QueryBuilder = exports._FilterBuilder = exports.FilterBuilder = exports.QueryBuilder = void 0;
 const __1 = require("../../");
 const Dataset_1 = require("./Dataset");
 function QueryBuilder() {
     return new _QueryBuilder();
 }
 exports.QueryBuilder = QueryBuilder;
-class _QueryBuilder {
+function FilterBuilder() {
+    return new _FilterBuilder();
+}
+exports.FilterBuilder = FilterBuilder;
+class _FilterBuilder {
     constructor() {
         this.body = { filters: [], fieldsToAggregate: [], fieldsToAggregateStats: [] };
     }
+    buildFilters() {
+        return this.body.filters;
+    }
+    rawFilter(filter) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push(filter);
+        return this;
+    }
+    filter(type, key, value, ...options) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({
+            [type]: {
+                key,
+                value,
+                ...options
+            }
+        });
+        return this;
+    }
+    match(field, value) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ match: { key: field, value } });
+        return this;
+    }
+    wildcard(field, value) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ wildcard: { key: field, value } });
+        return this;
+    }
+    selfreference(fielda, fieldb, operation) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ selfreference: { a: fielda, b: fieldb, operation } });
+        return this;
+    }
+    range(field, options) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ range: { key: field, ...options } });
+        return this;
+    }
+    or(filters) {
+        var _a;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ or: filters.map(f => { var _a; return (_a = f.body.filters) !== null && _a !== void 0 ? _a : []; }) });
+        return this;
+    }
+    not(filter) {
+        var _a, _b, _c;
+        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ not: (_c = (_b = filter.body) === null || _b === void 0 ? void 0 : _b.filters) !== null && _c !== void 0 ? _c : [] });
+        return this;
+    }
+}
+exports._FilterBuilder = _FilterBuilder;
+class _QueryBuilder extends _FilterBuilder {
+    constructor() {
+        super();
+        this.shouldPerformTextQuery = false;
+    }
     build() {
+        if (!this.shouldPerformTextQuery)
+            return this.body;
+        if (!this.defaultQueryValue)
+            throw new Error("Please set the search query by calling .query('my search query') before performing a text search.");
+        this.body.query = this.defaultQueryValue;
         return this.body;
     }
-    text(query, ...args) {
-        this.body.query = query;
-        for (const arg of args) {
-            if (Array.isArray(arg))
-                this.body.fieldsToSearch = arg;
-            else
-                this.body.queryConfig = arg;
-        }
+    query(query, fieldsToSearch) {
+        this.defaultQueryValue = query;
+        if (fieldsToSearch)
+            this.body.fieldsToSearch = fieldsToSearch;
         return this;
     }
-    vector(field, model, ...args) {
-        var _a, _b;
+    queryConfig(weight, options) {
+        this.body.queryConfig = { weight, ...(options !== null && options !== void 0 ? options : {}) };
+        return this;
+    }
+    text(field, weight) {
+        this.shouldPerformTextQuery = true;
+        if (!field)
+            return this; // support searching all fields
+        if (!this.body.fieldsToSearch)
+            this.body.fieldsToSearch = [];
+        if (!weight)
+            this.body.fieldsToSearch.push(field);
+        else
+            this.body.fieldsToSearch.push({ key: field, weight });
+        return this;
+    }
+    vector(field, ...args) {
         if (!Array.isArray(this.body.vectorSearchQuery))
             this.body.vectorSearchQuery = [];
-        if (!((_b = (_a = this === null || this === void 0 ? void 0 : this.body) === null || _a === void 0 ? void 0 : _a.vectorSearchQuery) === null || _b === void 0 ? void 0 : _b.length))
-            this.body.vectorSearchQuery = [];
-        let payload = { field, model };
+        let payload = { field };
+        const inferredModelMatch = field.match(/_(.*)_.*vector_/); // title_text@1-0_vector_ -> text@1-0
+        if (inferredModelMatch && inferredModelMatch[1])
+            payload.model = inferredModelMatch[1]; // this can be overridden
         for (const arg of args) {
             if (typeof arg === 'number')
-                payload.weight = arg;
+                payload.weight = arg; // weight
             else
-                payload = { ...payload, ...arg };
+                payload = { ...payload, ...arg }; // options
         }
         this.body.vectorSearchQuery.push(payload);
         return this;
@@ -70,11 +147,6 @@ class _QueryBuilder {
         this.body.pageSize = value;
         return this;
     }
-    rawFilter(filter) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push(filter);
-        return this;
-    }
     includeFields(fields) {
         this.body.includeFields = fields;
     }
@@ -84,47 +156,6 @@ class _QueryBuilder {
     includeVectors(whetherToInclude) {
         this.body.includeVectors = whetherToInclude;
     }
-    filter(type, key, value, ...options) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({
-            [type]: {
-                key,
-                value,
-                ...options
-            }
-        });
-        return this;
-    }
-    match(field, value) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ match: { key: field, value } });
-        return this;
-    }
-    wildcard(field, value) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ wildcard: { key: field, value } });
-        return this;
-    }
-    selfreference(fielda, fieldb, operation) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ selfreference: { a: fielda, b: fieldb, operation } });
-        return this;
-    }
-    range(field, options) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ range: { key: field, ...options } });
-        return this;
-    }
-    or(filters) {
-        var _a;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ or: filters.map(f => { var _a; return (_a = f.body.filters) !== null && _a !== void 0 ? _a : []; }) });
-        return this;
-    }
-    not(filter) {
-        var _a, _b, _c;
-        (_a = this.body.filters) === null || _a === void 0 ? void 0 : _a.push({ not: (_c = (_b = filter.body) === null || _b === void 0 ? void 0 : _b.filters) !== null && _c !== void 0 ? _c : [] });
-        return this;
-    }
     aggregate(field, options) {
         var _a, _b, _c;
         (_a = this.body.fieldsToAggregate) === null || _a === void 0 ? void 0 : _a.push({ key: field, ...options, fieldsToAggregate: (_c = (_b = options === null || options === void 0 ? void 0 : options.aggregates) === null || _b === void 0 ? void 0 : _b.body.fieldsToAggregate) !== null && _c !== void 0 ? _c : [] });

package/dist-es/generated/VectorApi.js

@@ -702,6 +702,18 @@ var VectorApiClient = /** @class */ (function (_super) {
             });
         });
     };
+    VectorApiClient.prototype.clustercentroidsapiv2servicesclustercentroidslistpost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/list',
+                        options: options
+                    })];
+            });
+        });
+    };
     VectorApiClient.prototype.clustercentroidsgetapiservicesclustercentroidsgetget = function (input, options) {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {
@@ -714,7 +726,19 @@ var VectorApiClient = /** @class */ (function (_super) {
             });
         });
     };
-    VectorApiClient.prototype.insertclustercentroidsapiservicesclustercentroidsinsertpost = function (input, options) {
+    VectorApiClient.prototype.clustercentroidsgetapiservicesclustercentroidsgetpost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/get',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.insertclustercentroids2apiservicesclustercentroidsinsertpost = function (input, options) {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 return [2 /*return*/, this.SendRequest({
@@ -726,6 +750,54 @@ var VectorApiClient = /** @class */ (function (_super) {
             });
         });
     };
+    VectorApiClient.prototype.updatecentroidsapiv2servicesclustercentroidsupdatepost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/update',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.deletecentroidsapiservicesclustercentroidscentroididdeleteget = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'get',
+                        path: '/services/cluster/centroids/{centroid_id}/delete',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.deletecentroidsapiservicesclustercentroidscentroididdeletepost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/{centroid_id}/delete',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.clustercentroidsdeleteapiservicesclustercentroidsdeletepost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/delete',
+                        options: options
+                    })];
+            });
+        });
+    };
     VectorApiClient.prototype.clustercentroidsgetapiservicesclustercentroidsdocumentspost = function (input, options) {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {
@@ -738,7 +810,55 @@ var VectorApiClient = /** @class */ (function (_super) {
             });
         });
     };
-    VectorApiClient.prototype.clusteraggregateapiservicesclusteraggregatepost = function (input, options) {
+    VectorApiClient.prototype.centroidsmetadatagetapiservicesclustercentroidsmetadataget = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'get',
+                        path: '/services/cluster/centroids/metadata',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.centroidsmetadatapostapiv2servicesclustercentroidsmetadatapost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/metadata',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.centroidslistclosesttocenterv2servicesclustercentroidslistclosesttocenterpost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/list_closest_to_center',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.centroidslistfurthestfromcenterv2servicesclustercentroidslistfurthestfromcenterpost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/centroids/list_furthest_from_center',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.clusteraggregateapiv2servicesclusteraggregatepost = function (input, options) {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 return [2 /*return*/, this.SendRequest({
@@ -750,7 +870,7 @@ var VectorApiClient = /** @class */ (function (_super) {
             });
         });
     };
-    VectorApiClient.prototype.clusterfacetsapiservicesclusterfacetsget = function (input, options) {
+    VectorApiClient.prototype.advancedclusterfacetsapiservicesclusterfacetsget = function (input, options) {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 return [2 /*return*/, this.SendRequest({
@@ -762,6 +882,30 @@ var VectorApiClient = /** @class */ (function (_super) {
             });
         });
     };
+    VectorApiClient.prototype.clusterlistservicesclusterlistget = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'get',
+                        path: '/services/cluster/list',
+                        options: options
+                    })];
+            });
+        });
+    };
+    VectorApiClient.prototype.clusterlistmultiservicesclusterlistpost = function (input, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.SendRequest({
+                        input: input,
+                        method: 'post',
+                        path: '/services/cluster/list',
+                        options: options
+                    })];
+            });
+        });
+    };
     VectorApiClient.prototype.tagapiservicestaggertagpost = function (input, options) {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {

package/dist-es/services/discovery/Dataset.js

@@ -105,8 +105,11 @@ var Dataset = /** @class */ (function () {
                                 if (arg instanceof _QueryBuilder) {
                                     payload = __assign(__assign({}, payload), arg.build());
                                 }
-                                else
+                                else {
                                     options = arg;
+                                    if (options.rawPayload)
+                                        payload = __assign(__assign({}, payload), options.rawPayload);
+                                }
                             }
                         }
                         catch (e_1_1) { e_1 = { error: e_1_1 }; }
@@ -250,7 +253,7 @@ var Dataset = /** @class */ (function () {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.client.apiClient.UpdateWhere({ filters: filters.body.filters, updates: partialUpdates })];
+                    case 0: return [4 /*yield*/, this.client.apiClient.UpdateWhere({ filters: filters.build().filters, updates: partialUpdates })];
                     case 1: return [2 /*return*/, (_a.sent()).body];
                 }
             });
@@ -284,7 +287,7 @@ var Dataset = /** @class */ (function () {
                 switch (_b.label) {
                     case 0:
                         filters = QueryBuilder().match('_id', documentIds);
-                        return [4 /*yield*/, this.client.apiClient.DeleteWhere({ filters: (_a = filters.body.filters) !== null && _a !== void 0 ? _a : [] })];
+                        return [4 /*yield*/, this.client.apiClient.DeleteWhere({ filters: (_a = filters.build().filters) !== null && _a !== void 0 ? _a : [] })];
                     case 1: return [2 /*return*/, (_b.sent()).body];
                 }
             });
@@ -295,7 +298,7 @@ var Dataset = /** @class */ (function () {
         return __awaiter(this, void 0, void 0, function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
-                    case 0: return [4 /*yield*/, this.client.apiClient.DeleteWhere({ filters: (_a = filters.body.filters) !== null && _a !== void 0 ? _a : [] })];
+                    case 0: return [4 /*yield*/, this.client.apiClient.DeleteWhere({ filters: (_a = filters.build().filters) !== null && _a !== void 0 ? _a : [] })];
                     case 1: return [2 /*return*/, (_b.sent()).body];
                 }
             });