@sprucelabs/chroma-data-store 0.2.0 → 0.2.2

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 = {
@@ -354,6 +357,9 @@ class ChromaDatabase {
     }
     async delete(collection, query) {
         const matches = await this.find(collection, query);
+        if (matches.length === 0) {
+            return 0;
+        }
         const col = await this.getCollection(collection);
         await col.delete({
             ids: matches.map((m) => m.id),

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 = {
@@ -405,6 +408,9 @@ export default class ChromaDatabase {
     delete(collection, query) {
         return __awaiter(this, void 0, void 0, function* () {
             const matches = yield this.find(collection, query);
+            if (matches.length === 0) {
+                return 0;
+            }
             const col = yield this.getCollection(collection);
             yield col.delete({
                 ids: matches.map((m) => m.id),

package/package.json

@@ -13,7 +13,7 @@
     "sprucebot",
     "sprucelabs"
   ],
-  "version": "0.2.0",
+  "version": "0.2.2",
   "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/",