@sprucelabs/chroma-data-store 0.0.5 → 0.1.1

package/build/ChromaDatabase.d.ts

@@ -1,14 +1,17 @@
 import { CreateOptions, Database, DatabaseInternalOptions, Index, IndexWithFilter, QueryOptions } from '@sprucelabs/data-stores';
 export default class ChromaDatabase implements Database {
+    static Class?: new (connectionString: string) => Database;
+    private static embeddingFields?;
     private connectionString;
     private client;
     private _isConnected;
     private embeddings;
     private collections;
-    private static embeddingFields?;
     constructor(connectionString: string);
+    static Database(connectionString: string): ChromaDatabase | Database;
     static setEmbeddingsFields(collectionName: string, fields: string[]): void;
     static clearEmbeddingsFields(): void;
+    static getEmbeddingsFields(): Record<string, string[]> | undefined;
     syncUniqueIndexes(_collectionName: string, _indexes: Index[]): Promise<void>;
     syncIndexes(_collectionName: string, _indexes: Index[]): Promise<void>;
     dropIndex(_collectionName: string, _index: Index): Promise<void>;

package/build/ChromaDatabase.js

@@ -24,6 +24,9 @@ class ChromaDatabase {
         }
         this.connectionString = connectionString.replace('chroma://', 'http://');
     }
+    static Database(connectionString) {
+        return new (this.Class ?? this)(connectionString);
+    }
     static setEmbeddingsFields(collectionName, fields) {
         if (!this.embeddingFields) {
             this.embeddingFields = {};
@@ -33,6 +36,9 @@ class ChromaDatabase {
     static clearEmbeddingsFields() {
         delete this.embeddingFields;
     }
+    static getEmbeddingsFields() {
+        return this.embeddingFields;
+    }
     async syncUniqueIndexes(_collectionName, _indexes) {
         throw new SpruceError_1.default({
             code: 'FEATURE_NOT_SUPPORTED',
@@ -174,7 +180,7 @@ class ChromaDatabase {
     }
     async dropDatabase() {
         const collections = await this.client.listCollections();
-        await Promise.all(collections.map((col) => this.dropCollection(col.name)));
+        await Promise.all(collections.map((col) => this.dropCollection(col)));
     }
     async findOne(collection, query, options, dbOptions) {
         const matches = await this.find(collection, query, { ...options, limit: 1 }, dbOptions);

package/build/MockChromaDatabase.d.ts

@@ -0,0 +1,6 @@
+import { NeDbDatabase } from '@sprucelabs/data-stores';
+export default class MockChromaDatabase extends NeDbDatabase {
+    private connectionString;
+    constructor(connectionString: string);
+    assertConnectionStringEquals(connectionString: string): void;
+}

package/build/MockChromaDatabase.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const data_stores_1 = require("@sprucelabs/data-stores");
+const test_utils_1 = require("@sprucelabs/test-utils");
+class MockChromaDatabase extends data_stores_1.NeDbDatabase {
+    constructor(connectionString) {
+        super();
+        this.connectionString = connectionString;
+    }
+    assertConnectionStringEquals(connectionString) {
+        test_utils_1.assert.isEqual(this.connectionString, connectionString, 'The connection string you passed to your ChromaDb does not equal what I expected!');
+    }
+}
+exports.default = MockChromaDatabase;

package/build/chromaDbAssert.utility.d.ts

@@ -0,0 +1,4 @@
+declare const chromaDbAssert: {
+    embeddingFieldsForCollectionEqual(collection: string, fields: string[]): void;
+};
+export default chromaDbAssert;

package/build/chromaDbAssert.utility.js

@@ -0,0 +1,20 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const schema_1 = require("@sprucelabs/schema");
+const test_utils_1 = require("@sprucelabs/test-utils");
+const ChromaDatabase_1 = __importDefault(require("./ChromaDatabase"));
+const chromaDbAssert = {
+    embeddingFieldsForCollectionEqual(collection, fields) {
+        (0, schema_1.assertOptions)({
+            collection,
+            fields,
+        }, ['collection', 'fields']);
+        const actual = ChromaDatabase_1.default.getEmbeddingsFields()?.[collection];
+        test_utils_1.assert.isTruthy(actual, `I could not find any embedding fields for the collection called ${collection}`);
+        test_utils_1.assert.isEqualDeep(actual, fields, `The embedding fields for the collection did not match.`);
+    },
+};
+exports.default = chromaDbAssert;

package/build/esm/ChromaDatabase.d.ts

@@ -1,14 +1,17 @@
 import { CreateOptions, Database, DatabaseInternalOptions, Index, IndexWithFilter, QueryOptions } from '@sprucelabs/data-stores';
 export default class ChromaDatabase implements Database {
+    static Class?: new (connectionString: string) => Database;
+    private static embeddingFields?;
     private connectionString;
     private client;
     private _isConnected;
     private embeddings;
     private collections;
-    private static embeddingFields?;
     constructor(connectionString: string);
+    static Database(connectionString: string): ChromaDatabase | Database;
     static setEmbeddingsFields(collectionName: string, fields: string[]): void;
     static clearEmbeddingsFields(): void;
+    static getEmbeddingsFields(): Record<string, string[]> | undefined;
     syncUniqueIndexes(_collectionName: string, _indexes: Index[]): Promise<void>;
     syncIndexes(_collectionName: string, _indexes: Index[]): Promise<void>;
     dropIndex(_collectionName: string, _index: Index): Promise<void>;

package/build/esm/ChromaDatabase.js

@@ -39,6 +39,10 @@ export default class ChromaDatabase {
         }
         this.connectionString = connectionString.replace('chroma://', 'http://');
     }
+    static Database(connectionString) {
+        var _a;
+        return new ((_a = this.Class) !== null && _a !== void 0 ? _a : this)(connectionString);
+    }
     static setEmbeddingsFields(collectionName, fields) {
         if (!this.embeddingFields) {
             this.embeddingFields = {};
@@ -48,6 +52,9 @@ export default class ChromaDatabase {
     static clearEmbeddingsFields() {
         delete this.embeddingFields;
     }
+    static getEmbeddingsFields() {
+        return this.embeddingFields;
+    }
     syncUniqueIndexes(_collectionName, _indexes) {
         return __awaiter(this, void 0, void 0, function* () {
             throw new SpruceError({
@@ -213,7 +220,7 @@ export default class ChromaDatabase {
     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)));
+            yield Promise.all(collections.map((col) => this.dropCollection(col)));
         });
     }
     findOne(collection, query, options, dbOptions) {

package/build/esm/MockChromaDatabase.d.ts

@@ -0,0 +1,6 @@
+import { NeDbDatabase } from '@sprucelabs/data-stores';
+export default class MockChromaDatabase extends NeDbDatabase {
+    private connectionString;
+    constructor(connectionString: string);
+    assertConnectionStringEquals(connectionString: string): void;
+}

package/build/esm/MockChromaDatabase.js

@@ -0,0 +1,11 @@
+import { NeDbDatabase } from '@sprucelabs/data-stores';
+import { assert } from '@sprucelabs/test-utils';
+export default class MockChromaDatabase extends NeDbDatabase {
+    constructor(connectionString) {
+        super();
+        this.connectionString = connectionString;
+    }
+    assertConnectionStringEquals(connectionString) {
+        assert.isEqual(this.connectionString, connectionString, 'The connection string you passed to your ChromaDb does not equal what I expected!');
+    }
+}

package/build/esm/chromaDbAssert.utility.d.ts

@@ -0,0 +1,4 @@
+declare const chromaDbAssert: {
+    embeddingFieldsForCollectionEqual(collection: string, fields: string[]): void;
+};
+export default chromaDbAssert;

package/build/esm/chromaDbAssert.utility.js

@@ -0,0 +1,16 @@
+import { assertOptions } from '@sprucelabs/schema';
+import { assert } from '@sprucelabs/test-utils';
+import ChromaDatabase from './ChromaDatabase.js';
+const chromaDbAssert = {
+    embeddingFieldsForCollectionEqual(collection, fields) {
+        var _a;
+        assertOptions({
+            collection,
+            fields,
+        }, ['collection', 'fields']);
+        const actual = (_a = ChromaDatabase.getEmbeddingsFields()) === null || _a === void 0 ? void 0 : _a[collection];
+        assert.isTruthy(actual, `I could not find any embedding fields for the collection called ${collection}`);
+        assert.isEqualDeep(actual, fields, `The embedding fields for the collection did not match.`);
+    },
+};
+export default chromaDbAssert;

package/build/esm/index.d.ts

@@ -1,2 +1,4 @@
 export { default as ChromaDatabase } from './ChromaDatabase';
+export { default as chromaDbAssert } from './chromaDbAssert.utility';
+export { default as MockChromaDatabase } from './MockChromaDatabase';
 export * from './ChromaDatabase';

package/build/esm/index.js

@@ -1,5 +1,7 @@
 export { default as ChromaDatabase } from './ChromaDatabase.js';
 import { DatabaseFactory } from '@sprucelabs/data-stores';
 import ChromaDatabase from './ChromaDatabase.js';
+export { default as chromaDbAssert } from './chromaDbAssert.utility.js';
+export { default as MockChromaDatabase } from './MockChromaDatabase.js';
 export * from './ChromaDatabase.js';
 DatabaseFactory.addAdapter('chroma://', ChromaDatabase);

package/build/index.d.ts

@@ -1,2 +1,4 @@
 export { default as ChromaDatabase } from './ChromaDatabase';
+export { default as chromaDbAssert } from './chromaDbAssert.utility';
+export { default as MockChromaDatabase } from './MockChromaDatabase';
 export * from './ChromaDatabase';

package/build/index.js

@@ -17,10 +17,14 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChromaDatabase = void 0;
+exports.MockChromaDatabase = exports.chromaDbAssert = exports.ChromaDatabase = void 0;
 var ChromaDatabase_1 = require("./ChromaDatabase");
 Object.defineProperty(exports, "ChromaDatabase", { enumerable: true, get: function () { return __importDefault(ChromaDatabase_1).default; } });
 const data_stores_1 = require("@sprucelabs/data-stores");
 const ChromaDatabase_2 = __importDefault(require("./ChromaDatabase"));
+var chromaDbAssert_utility_1 = require("./chromaDbAssert.utility");
+Object.defineProperty(exports, "chromaDbAssert", { enumerable: true, get: function () { return __importDefault(chromaDbAssert_utility_1).default; } });
+var MockChromaDatabase_1 = require("./MockChromaDatabase");
+Object.defineProperty(exports, "MockChromaDatabase", { enumerable: true, get: function () { return __importDefault(MockChromaDatabase_1).default; } });
 __exportStar(require("./ChromaDatabase"), exports);
 data_stores_1.DatabaseFactory.addAdapter('chroma://', ChromaDatabase_2.default);

package/package.json

@@ -13,7 +13,7 @@
     "sprucebot",
     "sprucelabs"
   ],
-  "version": "0.0.5",
+  "version": "0.1.1",
   "scripts": {
     "build.ci": "yarn run build.tsc && yarn run build.resolve-paths && yarn run lint",
     "build.copy-files": "mkdir -p build && rsync -avzq --exclude='*.ts' ./src/ ./build/",
@@ -66,7 +66,7 @@
   "jest": {
     "testRunner": "jest-circus/runner",
     "maxWorkers": 4,
-    "testTimeout": 1000000,
+    "testTimeout": 10000000,
     "testEnvironment": "node",
     "testPathIgnorePatterns": [
       "<rootDir>/tmp/",
@@ -82,12 +82,12 @@
     }
   },
   "dependencies": {
-    "@sprucelabs/data-stores": "^28.5.1",
+    "@sprucelabs/data-stores": "^28.5.2",
     "@sprucelabs/error": "^6.0.556",
     "@sprucelabs/schema": "^31.0.3",
     "@sprucelabs/spruce-core-schemas": "^40.1.586",
     "@sprucelabs/spruce-skill-utils": "^31.1.9",
-    "chromadb": "^1.9.4",
+    "chromadb": "^1.10.0",
     "chromadb-default-embed": "^2.13.2"
   }
 }