@sprucelabs/chroma-data-store 0.2.0 → 0.2.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/build/ChromaDatabase.js
CHANGED
|
@@ -189,13 +189,16 @@ class ChromaDatabase {
|
|
|
189
189
|
async find(collection, query, options, _dbOptions) {
|
|
190
190
|
let { ids, where, skipIds } = this.buildQuery(query);
|
|
191
191
|
let matches;
|
|
192
|
-
|
|
192
|
+
let { limit, includeFields } = options ?? {};
|
|
193
193
|
const col = await this.getCollection(collection);
|
|
194
|
+
if (!limit && where?.$prompt) {
|
|
195
|
+
limit = 10;
|
|
196
|
+
}
|
|
194
197
|
if (where?.$prompt) {
|
|
195
198
|
const { $prompt, ...rest } = where;
|
|
196
199
|
const queryResults = await col.query({
|
|
197
200
|
queryTexts: [$prompt],
|
|
198
|
-
nResults:
|
|
201
|
+
nResults: limit,
|
|
199
202
|
where: Object.keys(rest).length > 0 ? rest : undefined,
|
|
200
203
|
});
|
|
201
204
|
matches = {
|
|
@@ -234,13 +234,16 @@ export default class ChromaDatabase {
|
|
|
234
234
|
return __awaiter(this, void 0, void 0, function* () {
|
|
235
235
|
let { ids, where, skipIds } = this.buildQuery(query);
|
|
236
236
|
let matches;
|
|
237
|
-
|
|
237
|
+
let { limit, includeFields } = options !== null && options !== void 0 ? options : {};
|
|
238
238
|
const col = yield this.getCollection(collection);
|
|
239
|
+
if (!limit && (where === null || where === void 0 ? void 0 : where.$prompt)) {
|
|
240
|
+
limit = 10;
|
|
241
|
+
}
|
|
239
242
|
if (where === null || where === void 0 ? void 0 : where.$prompt) {
|
|
240
243
|
const { $prompt } = where, rest = __rest(where, ["$prompt"]);
|
|
241
244
|
const queryResults = yield col.query({
|
|
242
245
|
queryTexts: [$prompt],
|
|
243
|
-
nResults:
|
|
246
|
+
nResults: limit,
|
|
244
247
|
where: Object.keys(rest).length > 0 ? rest : undefined,
|
|
245
248
|
});
|
|
246
249
|
matches = {
|
package/package.json
CHANGED
|
@@ -13,7 +13,7 @@
|
|
|
13
13
|
"sprucebot",
|
|
14
14
|
"sprucelabs"
|
|
15
15
|
],
|
|
16
|
-
"version": "0.2.
|
|
16
|
+
"version": "0.2.1",
|
|
17
17
|
"scripts": {
|
|
18
18
|
"build.ci": "yarn run build.tsc && yarn run build.resolve-paths && yarn run lint",
|
|
19
19
|
"build.copy-files": "mkdir -p build && rsync -avzq --exclude='*.ts' ./src/ ./build/",
|