@sprucelabs/chroma-data-store 0.0.3 → 0.0.5

package/build/.spruce/errors/chromaDataStore/featureNotSupported.schema.js

@@ -16,4 +16,3 @@ const featureNotSupportedSchema = {
 };
 schema_1.SchemaRegistry.getInstance().trackSchema(featureNotSupportedSchema);
 exports.default = featureNotSupportedSchema;
-//# sourceMappingURL=featureNotSupported.schema.js.map

package/build/.spruce/errors/errors.types.js

@@ -1,3 +1,2 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=errors.types.js.map

package/build/.spruce/errors/options.types.js

@@ -1,3 +1,2 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=options.types.js.map

package/build/.spruce/schemas/fields/fieldClassMap.js

@@ -2,4 +2,3 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const schema_1 = require("@sprucelabs/schema");
 exports.default = schema_1.fieldClassMap;
-//# sourceMappingURL=fieldClassMap.js.map

package/build/.spruce/schemas/fields/fields.types.js

@@ -1,3 +1,2 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=fields.types.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/choice.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.choiceSchema; } });
-//# sourceMappingURL=choice.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/link.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.linkSchema; } });
-//# sourceMappingURL=link.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/location.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.locationSchema; } });
-//# sourceMappingURL=location.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/message.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.messageSchema; } });
-//# sourceMappingURL=message.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/messageSource.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.messageSourceSchema; } });
-//# sourceMappingURL=messageSource.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/messageTarget.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.messageTargetSchema; } });
-//# sourceMappingURL=messageTarget.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/organization.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.organizationSchema; } });
-//# sourceMappingURL=organization.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/person.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.personSchema; } });
-//# sourceMappingURL=person.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/personLocation.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.personLocationSchema; } });
-//# sourceMappingURL=personLocation.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/personOrganization.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.personOrganizationSchema; } });
-//# sourceMappingURL=personOrganization.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/role.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.roleSchema; } });
-//# sourceMappingURL=role.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/sendMessage.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.sendMessageSchema; } });
-//# sourceMappingURL=sendMessage.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/skill.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.skillSchema; } });
-//# sourceMappingURL=skill.schema.js.map

package/build/.spruce/schemas/spruce/v2020_07_22/skillCreator.schema.js

@@ -3,4 +3,3 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = void 0;
 var spruce_core_schemas_1 = require("@sprucelabs/spruce-core-schemas");
 Object.defineProperty(exports, "default", { enumerable: true, get: function () { return spruce_core_schemas_1.skillCreatorSchema; } });
-//# sourceMappingURL=skillCreator.schema.js.map

package/build/ChromaDatabase.js

@@ -383,4 +383,3 @@ class ChromaDatabase {
     }
 }
 exports.default = ChromaDatabase;
-//# sourceMappingURL=ChromaDatabase.js.map

package/build/chroma.types.js

@@ -1,3 +1,2 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=chroma.types.js.map

package/build/errors/SpruceError.js

@@ -23,4 +23,3 @@ class SpruceError extends error_1.default {
     }
 }
 exports.default = SpruceError;
-//# sourceMappingURL=SpruceError.js.map

package/build/errors/featureNotSupported.builder.js

@@ -11,4 +11,3 @@ exports.default = (0, schema_1.buildErrorSchema)({
         },
     },
 });
-//# sourceMappingURL=featureNotSupported.builder.js.map

package/build/esm/.spruce/errors/errors.types.d.ts

@@ -0,0 +1,21 @@
+import { default as SchemaEntity } from '@sprucelabs/schema';
+import * as SpruceSchema from '@sprucelabs/schema';
+export declare namespace SpruceErrors.ChromaDataStore {
+    interface FeatureNotSupported {
+        'operation': string;
+    }
+    interface FeatureNotSupportedSchema extends SpruceSchema.Schema {
+        id: 'featureNotSupported';
+        namespace: 'ChromaDataStore';
+        name: 'Feature not supported';
+        fields: {
+            /** . */
+            'operation': {
+                type: 'text';
+                isRequired: true;
+                options: undefined;
+            };
+        };
+    }
+    type FeatureNotSupportedEntity = SchemaEntity<SpruceErrors.ChromaDataStore.FeatureNotSupportedSchema>;
+}

package/build/esm/.spruce/errors/errors.types.js

@@ -0,0 +1 @@
+export {};

package/build/esm/.spruce/errors/options.types.d.ts

@@ -0,0 +1,7 @@
+import { SpruceErrors } from "./errors.types";
+import { ErrorOptions as ISpruceErrorOptions } from "@sprucelabs/error";
+export interface FeatureNotSupportedErrorOptions extends SpruceErrors.ChromaDataStore.FeatureNotSupported, ISpruceErrorOptions {
+    code: 'FEATURE_NOT_SUPPORTED';
+}
+type ErrorOptions = FeatureNotSupportedErrorOptions;
+export default ErrorOptions;

package/build/esm/.spruce/errors/options.types.js

@@ -0,0 +1 @@
+export {};

package/build/esm/ChromaDatabase.d.ts

@@ -0,0 +1,46 @@
+import { CreateOptions, Database, DatabaseInternalOptions, Index, IndexWithFilter, QueryOptions } from '@sprucelabs/data-stores';
+export default class ChromaDatabase implements Database {
+    private connectionString;
+    private client;
+    private _isConnected;
+    private embeddings;
+    private collections;
+    private static embeddingFields?;
+    constructor(connectionString: string);
+    static setEmbeddingsFields(collectionName: string, fields: string[]): void;
+    static clearEmbeddingsFields(): void;
+    syncUniqueIndexes(_collectionName: string, _indexes: Index[]): Promise<void>;
+    syncIndexes(_collectionName: string, _indexes: Index[]): Promise<void>;
+    dropIndex(_collectionName: string, _index: Index): Promise<void>;
+    getUniqueIndexes(_collectionName: string): Promise<IndexWithFilter[]>;
+    getIndexes(_collectionName: string): Promise<IndexWithFilter[]>;
+    isConnected(): boolean;
+    generateId(): string;
+    connect(): Promise<void>;
+    close(): Promise<void>;
+    getShouldAutoGenerateId?(): boolean;
+    createOne(collection: string, values: Record<string, any>, _options?: CreateOptions): Promise<Record<string, any>>;
+    private getCollection;
+    private buildPrompt;
+    create(collection: string, values: Record<string, any>[]): Promise<Record<string, any>[]>;
+    private splitValues;
+    private flattenValues;
+    private dropInNullPlaceholders;
+    dropCollection(name: string): Promise<void>;
+    dropDatabase(): Promise<void>;
+    findOne(collection: string, query?: Record<string, any>, options?: QueryOptions, dbOptions?: DatabaseInternalOptions): Promise<Record<string, any> | null>;
+    find(collection: string, query?: Record<string, any>, options?: QueryOptions, _dbOptions?: DatabaseInternalOptions): Promise<Record<string, any>[]>;
+    private mapResultsToRecords;
+    private filterFieldsByInclude;
+    private buildQuery;
+    private expandValues;
+    updateOne(collection: string, query: Record<string, any>, updates: Record<string, any>, _dbOptions?: DatabaseInternalOptions): Promise<Record<string, any>>;
+    update(collection: string, query: Record<string, any>, updates: Record<string, any>): Promise<number>;
+    upsertOne(collection: string, query: Record<string, any>, updates: Record<string, any>): Promise<Record<string, any>>;
+    delete(collection: string, query: Record<string, any>): Promise<number>;
+    deleteOne(collection: string, query: Record<string, any>): Promise<number>;
+    count(collection: string, query?: Record<string, any>): Promise<number>;
+    createUniqueIndex(_collection: string, _index: Index): Promise<void>;
+    createIndex(_collection: string, _index: Index): Promise<void>;
+    query<T>(_query: string, _params?: any[]): Promise<T[]>;
+}

package/build/esm/ChromaDatabase.js

@@ -0,0 +1,445 @@
+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());
+    });
+};
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
+import { DataStoresError, generateId, } from '@sprucelabs/data-stores';
+import { assertOptions, flattenValues, expandValues } from '@sprucelabs/schema';
+import { NULL_PLACEHOLDER } from '@sprucelabs/test-utils';
+import { ChromaClient, OllamaEmbeddingFunction, } from 'chromadb';
+import SpruceError from './errors/SpruceError.js';
+export default class ChromaDatabase {
+    constructor(connectionString) {
+        this._isConnected = false;
+        this.collections = {};
+        assertOptions({ connectionString }, ['connectionString']);
+        this.embeddings = new OllamaEmbeddingFunction({
+            model: 'llama3.2',
+            url: 'http://localhost:11434/api/embeddings',
+        });
+        if (!connectionString.startsWith('chroma://')) {
+            throw new DataStoresError({
+                code: 'INVALID_DB_CONNECTION_STRING',
+            });
+        }
+        this.connectionString = connectionString.replace('chroma://', 'http://');
+    }
+    static setEmbeddingsFields(collectionName, fields) {
+        if (!this.embeddingFields) {
+            this.embeddingFields = {};
+        }
+        this.embeddingFields[collectionName] = fields;
+    }
+    static clearEmbeddingsFields() {
+        delete this.embeddingFields;
+    }
+    syncUniqueIndexes(_collectionName, _indexes) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'syncUniqueIndexes',
+            });
+        });
+    }
+    syncIndexes(_collectionName, _indexes) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'syncIndexes',
+            });
+        });
+    }
+    dropIndex(_collectionName, _index) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'dropIndex',
+            });
+        });
+    }
+    getUniqueIndexes(_collectionName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'getUniqueIndexes',
+            });
+        });
+    }
+    getIndexes(_collectionName) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'getIndexes',
+            });
+        });
+    }
+    isConnected() {
+        return this._isConnected;
+    }
+    generateId() {
+        return generateId();
+    }
+    connect() {
+        return __awaiter(this, void 0, void 0, function* () {
+            this.client = new ChromaClient({ path: this.connectionString });
+            try {
+                yield this.client.heartbeat();
+            }
+            catch (err) {
+                throw new DataStoresError({
+                    code: 'UNABLE_TO_CONNECT_TO_DB',
+                    originalError: err,
+                });
+            }
+            this._isConnected = true;
+        });
+    }
+    close() {
+        return __awaiter(this, void 0, void 0, function* () {
+            this._isConnected = false;
+        });
+    }
+    getShouldAutoGenerateId() {
+        return true;
+    }
+    createOne(collection, values, _options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const results = yield this.create(collection, [values]);
+            return results[0];
+        });
+    }
+    getCollection(collection) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.collections[collection]) {
+                this.collections[collection] =
+                    yield this.client.getOrCreateCollection({
+                        name: collection,
+                        embeddingFunction: this.embeddings,
+                    });
+            }
+            return this.collections[collection];
+        });
+    }
+    buildPrompt(values, collectionName) {
+        var _a;
+        const fields = (_a = ChromaDatabase.embeddingFields) === null || _a === void 0 ? void 0 : _a[collectionName];
+        if ((fields === null || fields === void 0 ? void 0 : fields.length) === 1) {
+            const field = fields[0];
+            return values[field];
+        }
+        let prompt = '';
+        const keys = Object.keys(values);
+        for (const key of keys) {
+            const value = values[key];
+            if (value && typeof value === 'object') {
+                prompt += `${key}:\n\t${this.buildPrompt(value, collectionName)}\n`;
+            }
+            else if (value) {
+                prompt += `${key}: ${value}\n`;
+            }
+        }
+        return prompt.trim();
+    }
+    create(collection, values) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const col = yield this.getCollection(collection);
+            const { documents, ids, metadatas } = this.splitValues(values, collection);
+            yield col.add({
+                documents,
+                ids,
+                metadatas,
+            });
+            return this.find(collection, { id: { $in: ids } });
+        });
+    }
+    splitValues(values, collectionName) {
+        const ids = [];
+        const documents = [];
+        const metadatas = [];
+        for (const v of values) {
+            const { id = this.generateId() } = v, values = __rest(v, ["id"]);
+            ids.push(id);
+            documents.push(this.buildPrompt(values, collectionName));
+            const flattened = this.flattenValues(values);
+            metadatas.push(flattened);
+        }
+        return { documents, ids, metadatas };
+    }
+    flattenValues(values) {
+        const flattened = flattenValues(values, [
+            '$or',
+            '*.$gt',
+            '*.$gte',
+            '*.$lt',
+            '*.$lte',
+            '*.$ne',
+        ]);
+        this.dropInNullPlaceholders(flattened);
+        return flattened;
+    }
+    dropInNullPlaceholders(flattened) {
+        for (const key in flattened) {
+            if (flattened[key] === undefined) {
+                flattened[key] = null;
+            }
+            if (flattened[key] === null) {
+                flattened[key] = NULL_PLACEHOLDER;
+            }
+            if (typeof flattened[key] === 'object') {
+                this.dropInNullPlaceholders(flattened[key]);
+            }
+        }
+    }
+    dropCollection(name) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.client.deleteCollection({ name });
+        });
+    }
+    dropDatabase() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const collections = yield this.client.listCollections();
+            yield Promise.all(collections.map((col) => this.dropCollection(col.name)));
+        });
+    }
+    findOne(collection, query, options, dbOptions) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const matches = yield this.find(collection, query, Object.assign(Object.assign({}, options), { limit: 1 }), dbOptions);
+            return (_a = matches[0]) !== null && _a !== void 0 ? _a : null;
+        });
+    }
+    find(collection, query, options, _dbOptions) {
+        return __awaiter(this, void 0, void 0, function* () {
+            let { ids, where, skipIds } = this.buildQuery(query);
+            let matches;
+            const { limit, includeFields } = options !== null && options !== void 0 ? options : {};
+            const col = yield this.getCollection(collection);
+            if (where === null || where === void 0 ? void 0 : where.$prompt) {
+                const { $prompt } = where, rest = __rest(where, ["$prompt"]);
+                const queryResults = yield col.query({
+                    queryTexts: [$prompt],
+                    nResults: 1,
+                    where: Object.keys(rest).length > 0 ? rest : undefined,
+                });
+                matches = {
+                    ids: queryResults.ids[0],
+                    metadatas: queryResults.metadatas[0],
+                };
+            }
+            else {
+                matches = yield col.get({
+                    ids,
+                    include: ['metadatas'],
+                    where,
+                    limit: limit == 0 ? 1 : limit,
+                });
+            }
+            if (!matches || !matches.ids[0]) {
+                return [];
+            }
+            let records = this.mapResultsToRecords(limit, matches, skipIds);
+            if (includeFields) {
+                records = this.filterFieldsByInclude(records, includeFields);
+            }
+            return records;
+        });
+    }
+    mapResultsToRecords(limit, matches, skipIds) {
+        var _a;
+        let records = [];
+        const total = limit !== null && limit !== void 0 ? limit : matches.ids.length;
+        for (let i = 0; i < total; i++) {
+            const values = (_a = matches.metadatas[i]) !== null && _a !== void 0 ? _a : {};
+            const id = matches.ids[i];
+            if (!skipIds.includes(id)) {
+                records.push(Object.assign({ id }, this.expandValues(values)));
+            }
+        }
+        return records;
+    }
+    filterFieldsByInclude(records, includeFields) {
+        const trimmedRecords = [];
+        // eslint-disable-next-line @typescript-eslint/prefer-for-of
+        for (let c = 0; c < records.length; c++) {
+            const record = {};
+            for (const key of includeFields) {
+                record[key] = records[c][key];
+            }
+            trimmedRecords.push(record);
+        }
+        return trimmedRecords;
+    }
+    buildQuery(query) {
+        const _a = query !== null && query !== void 0 ? query : {}, { id } = _a, rest = __rest(_a, ["id"]);
+        let ids;
+        let where;
+        if (id === null || id === void 0 ? void 0 : id.$in) {
+            ids = id.$in;
+        }
+        else if (id) {
+            ids = [id];
+        }
+        else if (query) {
+            where = rest;
+        }
+        if ((where === null || where === void 0 ? void 0 : where.$or) && where.$or.length === 1) {
+            //@ts-ignore
+            where = where.$or[0];
+        }
+        if (where) {
+            const { $or, $prompt } = where, rest = __rest(where, ["$or", "$prompt"]);
+            if ($or) {
+                where.$or = $or;
+            }
+            else {
+                where = this.flattenValues(rest);
+                if (Object.keys(where).length > 1) {
+                    const $and = [];
+                    for (const key in where) {
+                        //@ts-ignore
+                        $and.push({ [key]: where[key] });
+                    }
+                    where = {
+                        $and,
+                    };
+                }
+            }
+            if ($prompt) {
+                where.$prompt = $prompt;
+            }
+        }
+        const firstId = ids === null || ids === void 0 ? void 0 : ids[0];
+        const skipIds = [];
+        //@ts-ignore
+        if (firstId === null || firstId === void 0 ? void 0 : firstId['$ne']) {
+            //@ts-ignore
+            skipIds.push(firstId.$ne);
+            ids = undefined;
+        }
+        if (Object.keys(where !== null && where !== void 0 ? where : {}).length === 0) {
+            where = undefined;
+        }
+        return { ids, where, skipIds };
+    }
+    expandValues(values) {
+        const nulledValues = {};
+        for (const key in values) {
+            if (values[key] === NULL_PLACEHOLDER) {
+                nulledValues[key] = null;
+            }
+            else {
+                nulledValues[key] = values[key];
+            }
+        }
+        return expandValues(nulledValues);
+    }
+    updateOne(collection, query, updates, _dbOptions) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const col = yield this.getCollection(collection);
+            const match = yield this.findOne(collection, query);
+            if (!match) {
+                throw new DataStoresError({
+                    code: 'RECORD_NOT_FOUND',
+                    query,
+                    storeName: collection,
+                });
+            }
+            const { documents, metadatas } = this.splitValues([updates], collection);
+            yield col.update({
+                ids: [match.id],
+                documents,
+                metadatas,
+            });
+            const { id } = match, values = __rest(match, ["id"]);
+            return Object.assign(Object.assign({ id }, values), updates);
+        });
+    }
+    update(collection, query, updates) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const matches = yield this.find(collection, query);
+            for (const match of matches) {
+                yield this.updateOne(collection, { id: match.id }, updates);
+            }
+            return matches.length;
+        });
+    }
+    upsertOne(collection, query, updates) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const col = yield this.getCollection(collection);
+            const match = yield this.findOne(collection, query);
+            let { documents, ids, metadatas } = this.splitValues([updates], collection);
+            if (match === null || match === void 0 ? void 0 : match.id) {
+                ids = [match.id];
+            }
+            yield col.upsert({
+                documents,
+                ids,
+                metadatas,
+            });
+            const updated = yield this.findOne(collection, { id: ids[0] });
+            return updated;
+        });
+    }
+    delete(collection, query) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const matches = yield this.find(collection, query);
+            const col = yield this.getCollection(collection);
+            yield col.delete({
+                ids: matches.map((m) => m.id),
+            });
+            return matches.length;
+        });
+    }
+    deleteOne(collection, query) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const match = yield this.findOne(collection, query);
+            yield this.delete(collection, { id: match === null || match === void 0 ? void 0 : match.id });
+            return match ? 1 : 0;
+        });
+    }
+    count(collection, query) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const matches = yield this.find(collection, query);
+            return matches.length;
+        });
+    }
+    createUniqueIndex(_collection, _index) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'createUniqueIndex',
+            });
+        });
+    }
+    createIndex(_collection, _index) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'createIndex',
+            });
+        });
+    }
+    query(_query, _params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            throw new SpruceError({
+                code: 'FEATURE_NOT_SUPPORTED',
+                operation: 'query',
+            });
+        });
+    }
+}