@sprucelabs/chroma-data-store 0.1.24 → 0.2.1

package/README.md

@@ -0,0 +1,9 @@
+# Running Chroma
+
+1. Clone this rep
+2. Run `yarn start.chroma.docker`
+
+
+# Connecting to Chroma
+
+Use the connection string: `chromadb://localhost:8000`

package/build/ChromaDatabase.js

@@ -189,13 +189,16 @@ class ChromaDatabase {
     async find(collection, query, options, _dbOptions) {
         let { ids, where, skipIds } = this.buildQuery(query);
         let matches;
-        const { limit, includeFields } = options ?? {};
+        let { limit, includeFields } = options ?? {};
         const col = await this.getCollection(collection);
+        if (!limit && where?.$prompt) {
+            limit = 10;
+        }
         if (where?.$prompt) {
             const { $prompt, ...rest } = where;
             const queryResults = await col.query({
                 queryTexts: [$prompt],
-                nResults: 1,
+                nResults: limit,
                 where: Object.keys(rest).length > 0 ? rest : undefined,
             });
             matches = {

package/build/MockChromaDatabase.d.ts

@@ -3,4 +3,6 @@ export default class MockChromaDatabase extends NeDbDatabase {
     private connectionString;
     constructor(connectionString: string);
     assertConnectionStringEquals(connectionString: string): void;
+    connect(): Promise<void>;
+    assertIsConnected(): void;
 }

package/build/MockChromaDatabase.js

@@ -10,5 +10,11 @@ class MockChromaDatabase extends data_stores_1.NeDbDatabase {
     assertConnectionStringEquals(connectionString) {
         test_utils_1.assert.isEqual(this.connectionString, connectionString, 'The connection string you passed to your ChromaDb does not equal what I expected!');
     }
+    async connect() {
+        await super.connect();
+    }
+    assertIsConnected() {
+        test_utils_1.assert.isTrue(this.isConnected(), 'Database is not connected! Try calling db.connect()');
+    }
 }
 exports.default = MockChromaDatabase;

package/build/esm/ChromaDatabase.js

@@ -234,13 +234,16 @@ export default class ChromaDatabase {
         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 : {};
+            let { limit, includeFields } = options !== null && options !== void 0 ? options : {};
             const col = yield this.getCollection(collection);
+            if (!limit && (where === null || where === void 0 ? void 0 : where.$prompt)) {
+                limit = 10;
+            }
             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,
+                    nResults: limit,
                     where: Object.keys(rest).length > 0 ? rest : undefined,
                 });
                 matches = {

package/build/esm/MockChromaDatabase.d.ts

@@ -3,4 +3,6 @@ export default class MockChromaDatabase extends NeDbDatabase {
     private connectionString;
     constructor(connectionString: string);
     assertConnectionStringEquals(connectionString: string): void;
+    connect(): Promise<void>;
+    assertIsConnected(): void;
 }

package/build/esm/MockChromaDatabase.js

@@ -1,3 +1,12 @@
+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());
+    });
+};
 import { NeDbDatabase } from '@sprucelabs/data-stores';
 import { assert } from '@sprucelabs/test-utils';
 export default class MockChromaDatabase extends NeDbDatabase {
@@ -8,4 +17,15 @@ export default class MockChromaDatabase extends NeDbDatabase {
     assertConnectionStringEquals(connectionString) {
         assert.isEqual(this.connectionString, connectionString, 'The connection string you passed to your ChromaDb does not equal what I expected!');
     }
+    connect() {
+        const _super = Object.create(null, {
+            connect: { get: () => super.connect }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            yield _super.connect.call(this);
+        });
+    }
+    assertIsConnected() {
+        assert.isTrue(this.isConnected(), 'Database is not connected! Try calling db.connect()');
+    }
 }

package/package.json

@@ -13,7 +13,7 @@
     "sprucebot",
     "sprucelabs"
   ],
-  "version": "0.1.24",
+  "version": "0.2.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/",