@mastra/chroma 0.0.0-vector-query-sources-20250516172905 → 0.0.0-vector-query-tool-provider-options-20250828222356

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/dist/index.js CHANGED
@@ -1,5 +1,6 @@
1
+ import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
1
2
  import { MastraVector } from '@mastra/core/vector';
2
- import { ChromaClient } from 'chromadb';
3
+ import { CloudClient, ChromaClient } from 'chromadb';
3
4
  import { BaseFilterTranslator } from '@mastra/core/vector/filter';
4
5
 
5
6
  // src/vector/index.ts
@@ -21,7 +22,7 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
21
22
  }
22
23
  translateNode(node, currentPath = "") {
23
24
  if (this.isRegex(node)) {
24
- throw new Error("Regex is not supported in Chroma");
25
+ throw new Error("Regex is supported in Chroma via the `documentFilter` argument");
25
26
  }
26
27
  if (this.isPrimitive(node)) return this.normalizeComparisonValue(node);
27
28
  if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };
@@ -30,6 +31,9 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
30
31
  if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {
31
32
  const [operator, value] = firstEntry;
32
33
  const translated = this.translateOperator(operator, value);
34
+ if (this.isLogicalOperator(operator) && Array.isArray(translated) && translated.length === 1) {
35
+ return translated[0];
36
+ }
33
37
  return this.isLogicalOperator(operator) ? { [operator]: translated } : translated;
34
38
  }
35
39
  const result = {};
@@ -87,31 +91,46 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
87
91
  };
88
92
 
89
93
  // src/vector/index.ts
94
+ var spaceMappings = {
95
+ cosine: "cosine",
96
+ euclidean: "l2",
97
+ dotproduct: "ip",
98
+ l2: "euclidean",
99
+ ip: "dotproduct"
100
+ };
90
101
  var ChromaVector = class extends MastraVector {
91
102
  client;
92
103
  collections;
93
- constructor({
94
- path,
95
- auth
96
- }) {
104
+ constructor(chromaClientArgs) {
97
105
  super();
98
- this.client = new ChromaClient({
99
- path,
100
- auth
101
- });
106
+ if (chromaClientArgs?.apiKey) {
107
+ this.client = new CloudClient({
108
+ apiKey: chromaClientArgs.apiKey,
109
+ tenant: chromaClientArgs.tenant,
110
+ database: chromaClientArgs.database
111
+ });
112
+ } else {
113
+ this.client = new ChromaClient(chromaClientArgs);
114
+ }
102
115
  this.collections = /* @__PURE__ */ new Map();
103
116
  }
104
- async getCollection(indexName, throwIfNotExists = true) {
105
- try {
106
- const collection = await this.client.getCollection({ name: indexName, embeddingFunction: void 0 });
107
- this.collections.set(indexName, collection);
108
- } catch {
109
- if (throwIfNotExists) {
110
- throw new Error(`Index ${indexName} does not exist`);
117
+ async getCollection({ indexName, forceUpdate = false }) {
118
+ let collection = this.collections.get(indexName);
119
+ if (forceUpdate || !collection) {
120
+ try {
121
+ collection = await this.client.getCollection({ name: indexName });
122
+ this.collections.set(indexName, collection);
123
+ return collection;
124
+ } catch {
125
+ throw new MastraError({
126
+ id: "CHROMA_COLLECTION_GET_FAILED",
127
+ domain: ErrorDomain.MASTRA_VECTOR,
128
+ category: ErrorCategory.THIRD_PARTY,
129
+ details: { indexName }
130
+ });
111
131
  }
112
- return null;
113
132
  }
114
- return this.collections.get(indexName);
133
+ return collection;
115
134
  }
116
135
  validateVectorDimensions(vectors, dimension) {
117
136
  for (let i = 0; i < vectors.length; i++) {
@@ -122,130 +141,234 @@ var ChromaVector = class extends MastraVector {
122
141
  }
123
142
  }
124
143
  }
125
- async upsert(...args) {
126
- const params = this.normalizeArgs("upsert", args, ["documents"]);
127
- const { indexName, vectors, metadata, ids, documents } = params;
128
- const collection = await this.getCollection(indexName);
129
- const stats = await this.describeIndex({ indexName });
130
- this.validateVectorDimensions(vectors, stats.dimension);
131
- const generatedIds = ids || vectors.map(() => crypto.randomUUID());
132
- const normalizedMetadata = metadata || vectors.map(() => ({}));
133
- await collection.upsert({
134
- ids: generatedIds,
135
- embeddings: vectors,
136
- metadatas: normalizedMetadata,
137
- documents
138
- });
139
- return generatedIds;
144
+ async upsert({ indexName, vectors, metadata, ids, documents }) {
145
+ try {
146
+ const collection = await this.getCollection({ indexName });
147
+ const stats = await this.describeIndex({ indexName });
148
+ this.validateVectorDimensions(vectors, stats.dimension);
149
+ const generatedIds = ids || vectors.map(() => crypto.randomUUID());
150
+ await collection.upsert({
151
+ ids: generatedIds,
152
+ embeddings: vectors,
153
+ metadatas: metadata,
154
+ documents
155
+ });
156
+ return generatedIds;
157
+ } catch (error) {
158
+ if (error instanceof MastraError) throw error;
159
+ throw new MastraError(
160
+ {
161
+ id: "CHROMA_VECTOR_UPSERT_FAILED",
162
+ domain: ErrorDomain.MASTRA_VECTOR,
163
+ category: ErrorCategory.THIRD_PARTY,
164
+ details: { indexName }
165
+ },
166
+ error
167
+ );
168
+ }
140
169
  }
141
- HnswSpaceMap = {
142
- cosine: "cosine",
143
- euclidean: "l2",
144
- dotproduct: "ip",
145
- l2: "euclidean",
146
- ip: "dotproduct"
147
- };
148
- async createIndex(...args) {
149
- const params = this.normalizeArgs("createIndex", args);
150
- const { indexName, dimension, metric = "cosine" } = params;
170
+ async createIndex({ indexName, dimension, metric = "cosine" }) {
151
171
  if (!Number.isInteger(dimension) || dimension <= 0) {
152
- throw new Error("Dimension must be a positive integer");
172
+ throw new MastraError({
173
+ id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_DIMENSION",
174
+ text: "Dimension must be a positive integer",
175
+ domain: ErrorDomain.MASTRA_VECTOR,
176
+ category: ErrorCategory.USER,
177
+ details: { dimension }
178
+ });
153
179
  }
154
- const hnswSpace = this.HnswSpaceMap[metric];
155
- if (!["cosine", "l2", "ip"].includes(hnswSpace)) {
156
- throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
180
+ const hnswSpace = spaceMappings[metric];
181
+ if (!hnswSpace || !["cosine", "l2", "ip"].includes(hnswSpace)) {
182
+ throw new MastraError({
183
+ id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC",
184
+ text: `Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`,
185
+ domain: ErrorDomain.MASTRA_VECTOR,
186
+ category: ErrorCategory.USER,
187
+ details: { metric }
188
+ });
157
189
  }
158
190
  try {
159
- await this.client.createCollection({
191
+ const collection = await this.client.createCollection({
160
192
  name: indexName,
161
- metadata: {
162
- dimension,
163
- "hnsw:space": hnswSpace
164
- }
193
+ metadata: { dimension },
194
+ configuration: { hnsw: { space: hnswSpace } },
195
+ embeddingFunction: null
165
196
  });
197
+ this.collections.set(indexName, collection);
166
198
  } catch (error) {
167
199
  const message = error?.message || error?.toString();
168
200
  if (message && message.toLowerCase().includes("already exists")) {
169
201
  await this.validateExistingIndex(indexName, dimension, metric);
170
202
  return;
171
203
  }
172
- throw error;
204
+ throw new MastraError(
205
+ {
206
+ id: "CHROMA_VECTOR_CREATE_INDEX_FAILED",
207
+ domain: ErrorDomain.MASTRA_VECTOR,
208
+ category: ErrorCategory.THIRD_PARTY,
209
+ details: { indexName }
210
+ },
211
+ error
212
+ );
173
213
  }
174
214
  }
175
215
  transformFilter(filter) {
176
216
  const translator = new ChromaFilterTranslator();
177
- return translator.translate(filter);
217
+ const translatedFilter = translator.translate(filter);
218
+ return translatedFilter ? translatedFilter : void 0;
219
+ }
220
+ async query({
221
+ indexName,
222
+ queryVector,
223
+ topK = 10,
224
+ filter,
225
+ includeVector = false,
226
+ documentFilter
227
+ }) {
228
+ try {
229
+ const collection = await this.getCollection({ indexName });
230
+ const defaultInclude = ["documents", "metadatas", "distances"];
231
+ const translatedFilter = this.transformFilter(filter);
232
+ const results = await collection.query({
233
+ queryEmbeddings: [queryVector],
234
+ nResults: topK,
235
+ where: translatedFilter ?? void 0,
236
+ whereDocument: documentFilter ?? void 0,
237
+ include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
238
+ });
239
+ return (results.ids[0] || []).map((id, index) => ({
240
+ id,
241
+ score: results.distances?.[0]?.[index] || 0,
242
+ metadata: results.metadatas?.[0]?.[index] || {},
243
+ document: results.documents?.[0]?.[index] ?? void 0,
244
+ ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
245
+ }));
246
+ } catch (error) {
247
+ if (error instanceof MastraError) throw error;
248
+ throw new MastraError(
249
+ {
250
+ id: "CHROMA_VECTOR_QUERY_FAILED",
251
+ domain: ErrorDomain.MASTRA_VECTOR,
252
+ category: ErrorCategory.THIRD_PARTY,
253
+ details: { indexName }
254
+ },
255
+ error
256
+ );
257
+ }
178
258
  }
179
- async query(...args) {
180
- const params = this.normalizeArgs("query", args, ["documentFilter"]);
181
- const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
182
- const collection = await this.getCollection(indexName, true);
183
- const defaultInclude = ["documents", "metadatas", "distances"];
184
- const translatedFilter = this.transformFilter(filter);
185
- const results = await collection.query({
186
- queryEmbeddings: [queryVector],
187
- nResults: topK,
188
- where: translatedFilter,
189
- whereDocument: documentFilter,
190
- include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
191
- });
192
- return (results.ids[0] || []).map((id, index) => ({
193
- id,
194
- score: results.distances?.[0]?.[index] || 0,
195
- metadata: results.metadatas?.[0]?.[index] || {},
196
- document: results.documents?.[0]?.[index],
197
- ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
198
- }));
259
+ async get({
260
+ indexName,
261
+ ids,
262
+ filter,
263
+ includeVector = false,
264
+ documentFilter,
265
+ offset,
266
+ limit
267
+ }) {
268
+ try {
269
+ const collection = await this.getCollection({ indexName });
270
+ const defaultInclude = ["documents", "metadatas"];
271
+ const translatedFilter = this.transformFilter(filter);
272
+ const result = await collection.get({
273
+ ids,
274
+ where: translatedFilter ?? void 0,
275
+ whereDocument: documentFilter ?? void 0,
276
+ offset,
277
+ limit,
278
+ include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
279
+ });
280
+ return result.rows();
281
+ } catch (error) {
282
+ if (error instanceof MastraError) throw error;
283
+ throw new MastraError(
284
+ {
285
+ id: "CHROMA_VECTOR_GET_FAILED",
286
+ domain: ErrorDomain.MASTRA_VECTOR,
287
+ category: ErrorCategory.THIRD_PARTY,
288
+ details: { indexName }
289
+ },
290
+ error
291
+ );
292
+ }
199
293
  }
200
294
  async listIndexes() {
201
- const collections = await this.client.listCollections();
202
- return collections.map((collection) => collection);
295
+ try {
296
+ const collections = await this.client.listCollections();
297
+ return collections.map((collection) => collection.name);
298
+ } catch (error) {
299
+ throw new MastraError(
300
+ {
301
+ id: "CHROMA_VECTOR_LIST_INDEXES_FAILED",
302
+ domain: ErrorDomain.MASTRA_VECTOR,
303
+ category: ErrorCategory.THIRD_PARTY
304
+ },
305
+ error
306
+ );
307
+ }
203
308
  }
204
309
  /**
205
310
  * Retrieves statistics about a vector index.
206
311
  *
207
- * @param params - The parameters for describing an index
208
- * @param params.indexName - The name of the index to describe
312
+ * @param {string} indexName - The name of the index to describe
209
313
  * @returns A promise that resolves to the index statistics including dimension, count and metric
210
314
  */
211
- async describeIndex(...args) {
212
- const params = this.normalizeArgs("describeIndex", args);
213
- const { indexName } = params;
214
- const collection = await this.getCollection(indexName);
215
- const count = await collection.count();
216
- const metadata = collection.metadata;
217
- const hnswSpace = metadata?.["hnsw:space"];
218
- return {
219
- dimension: metadata?.dimension || 0,
220
- count,
221
- metric: this.HnswSpaceMap[hnswSpace]
222
- };
315
+ async describeIndex({ indexName }) {
316
+ try {
317
+ const collection = await this.getCollection({ indexName });
318
+ const count = await collection.count();
319
+ const metadata = collection.metadata;
320
+ const space = collection.configuration.hnsw?.space || collection.configuration.spann?.space || void 0;
321
+ return {
322
+ dimension: metadata?.dimension || 0,
323
+ count,
324
+ metric: space ? spaceMappings[space] : void 0
325
+ };
326
+ } catch (error) {
327
+ if (error instanceof MastraError) throw error;
328
+ throw new MastraError(
329
+ {
330
+ id: "CHROMA_VECTOR_DESCRIBE_INDEX_FAILED",
331
+ domain: ErrorDomain.MASTRA_VECTOR,
332
+ category: ErrorCategory.THIRD_PARTY,
333
+ details: { indexName }
334
+ },
335
+ error
336
+ );
337
+ }
223
338
  }
224
- async deleteIndex(...args) {
225
- const params = this.normalizeArgs("deleteIndex", args);
226
- const { indexName } = params;
227
- await this.client.deleteCollection({ name: indexName });
228
- this.collections.delete(indexName);
339
+ async deleteIndex({ indexName }) {
340
+ try {
341
+ await this.client.deleteCollection({ name: indexName });
342
+ this.collections.delete(indexName);
343
+ } catch (error) {
344
+ throw new MastraError(
345
+ {
346
+ id: "CHROMA_VECTOR_DELETE_INDEX_FAILED",
347
+ domain: ErrorDomain.MASTRA_VECTOR,
348
+ category: ErrorCategory.THIRD_PARTY,
349
+ details: { indexName }
350
+ },
351
+ error
352
+ );
353
+ }
229
354
  }
230
- /**
231
- * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
232
- *
233
- * Updates a vector by its ID with the provided vector and/or metadata.
234
- * @param indexName - The name of the index containing the vector.
235
- * @param id - The ID of the vector to update.
236
- * @param update - An object containing the vector and/or metadata to update.
237
- * @param update.vector - An optional array of numbers representing the new vector.
238
- * @param update.metadata - An optional record containing the new metadata.
239
- * @returns A promise that resolves when the update is complete.
240
- * @throws Will throw an error if no updates are provided or if the update operation fails.
241
- */
242
- async updateIndexById(indexName, id, update) {
243
- this.logger.warn(
244
- `Deprecation Warning: updateIndexById() is deprecated.
245
- Please use updateVector() instead.
246
- updateIndexById() will be removed on May 20th, 2025.`
247
- );
248
- await this.updateVector({ indexName, id, update });
355
+ async forkIndex({ indexName, newIndexName }) {
356
+ try {
357
+ const collection = await this.getCollection({ indexName, forceUpdate: true });
358
+ const forkedCollection = await collection.fork({ name: newIndexName });
359
+ this.collections.set(newIndexName, forkedCollection);
360
+ } catch (error) {
361
+ if (error instanceof MastraError) throw error;
362
+ throw new MastraError(
363
+ {
364
+ id: "CHROMA_INDEX_FORK_FAILED",
365
+ domain: ErrorDomain.MASTRA_VECTOR,
366
+ category: ErrorCategory.THIRD_PARTY,
367
+ details: { indexName }
368
+ },
369
+ error
370
+ );
371
+ }
249
372
  }
250
373
  /**
251
374
  * Updates a vector by its ID with the provided vector and/or metadata.
@@ -257,58 +380,56 @@ var ChromaVector = class extends MastraVector {
257
380
  * @returns A promise that resolves when the update is complete.
258
381
  * @throws Will throw an error if no updates are provided or if the update operation fails.
259
382
  */
260
- async updateVector(...args) {
261
- const params = this.normalizeArgs("updateVector", args);
262
- const { indexName, id, update } = params;
383
+ async updateVector({ indexName, id, update }) {
384
+ if (!update.vector && !update.metadata) {
385
+ throw new MastraError({
386
+ id: "CHROMA_VECTOR_UPDATE_NO_PAYLOAD",
387
+ text: "No updates provided for vector",
388
+ domain: ErrorDomain.MASTRA_VECTOR,
389
+ category: ErrorCategory.USER,
390
+ details: { indexName, id }
391
+ });
392
+ }
263
393
  try {
264
- if (!update.vector && !update.metadata) {
265
- throw new Error("No updates provided");
266
- }
267
- const collection = await this.getCollection(indexName, true);
268
- const updateOptions = { ids: [id] };
394
+ const collection = await this.getCollection({ indexName });
395
+ const updateRecordSet = { ids: [id] };
269
396
  if (update?.vector) {
270
397
  const stats = await this.describeIndex({ indexName });
271
398
  this.validateVectorDimensions([update.vector], stats.dimension);
272
- updateOptions.embeddings = [update.vector];
399
+ updateRecordSet.embeddings = [update.vector];
273
400
  }
274
401
  if (update?.metadata) {
275
- updateOptions.metadatas = [update.metadata];
402
+ updateRecordSet.metadatas = [update.metadata];
276
403
  }
277
- return await collection.update(updateOptions);
404
+ return await collection.update(updateRecordSet);
278
405
  } catch (error) {
279
- throw new Error(`Failed to update vector by id: ${id} for index name: ${indexName}: ${error.message}`);
406
+ if (error instanceof MastraError) throw error;
407
+ throw new MastraError(
408
+ {
409
+ id: "CHROMA_VECTOR_UPDATE_FAILED",
410
+ domain: ErrorDomain.MASTRA_VECTOR,
411
+ category: ErrorCategory.THIRD_PARTY,
412
+ details: { indexName, id }
413
+ },
414
+ error
415
+ );
280
416
  }
281
417
  }
282
- /**
283
- * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
284
- *
285
- * Deletes a vector by its ID.
286
- * @param indexName - The name of the index containing the vector.
287
- * @param id - The ID of the vector to delete.
288
- * @returns A promise that resolves when the deletion is complete.
289
- * @throws Will throw an error if the deletion operation fails.
290
- */
291
- async deleteIndexById(indexName, id) {
292
- this.logger.warn(
293
- `Deprecation Warning: deleteIndexById() is deprecated. Please use deleteVector() instead. deleteIndexById() will be removed on May 20th.`
294
- );
295
- await this.deleteVector(indexName, id);
296
- }
297
- /**
298
- * Deletes a vector by its ID.
299
- * @param indexName - The name of the index containing the vector.
300
- * @param id - The ID of the vector to delete.
301
- * @returns A promise that resolves when the deletion is complete.
302
- * @throws Will throw an error if the deletion operation fails.
303
- */
304
- async deleteVector(...args) {
305
- const params = this.normalizeArgs("deleteVector", args);
306
- const { indexName, id } = params;
418
+ async deleteVector({ indexName, id }) {
307
419
  try {
308
- const collection = await this.getCollection(indexName, true);
420
+ const collection = await this.getCollection({ indexName });
309
421
  await collection.delete({ ids: [id] });
310
422
  } catch (error) {
311
- throw new Error(`Failed to delete vector by id: ${id} for index name: ${indexName}: ${error.message}`);
423
+ if (error instanceof MastraError) throw error;
424
+ throw new MastraError(
425
+ {
426
+ id: "CHROMA_VECTOR_DELETE_FAILED",
427
+ domain: ErrorDomain.MASTRA_VECTOR,
428
+ category: ErrorCategory.THIRD_PARTY,
429
+ details: { indexName, id }
430
+ },
431
+ error
432
+ );
312
433
  }
313
434
  }
314
435
  };
@@ -384,3 +505,5 @@ Example Complex Query:
384
505
  }`;
385
506
 
386
507
  export { CHROMA_PROMPT, ChromaVector };
508
+ //# sourceMappingURL=index.js.map
509
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"names":[],"mappings":";;;;;;AA4BO,IAAM,sBAAA,GAAN,cAAqC,oBAAA,CAAyC;AAAA,EAChE,qBAAA,GAAyC;AAC1D,IAAA,OAAO;AAAA,MACL,GAAG,oBAAA,CAAqB,iBAAA;AAAA,MACxB,OAAA,EAAS,CAAC,MAAA,EAAQ,KAAK,CAAA;AAAA,MACvB,KAAA,EAAO,CAAC,KAAA,EAAO,MAAM,CAAA;AAAA,MACrB,SAAS,EAAC;AAAA,MACV,OAAO,EAAC;AAAA,MACR,QAAQ;AAAC,KACX;AAAA,EACF;AAAA,EAEA,UAAU,MAAA,EAAiD;AACzD,IAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,MAAM,CAAA,EAAG,OAAO,MAAA;AACjC,IAAA,IAAA,CAAK,eAAe,MAAM,CAAA;AAE1B,IAAA,OAAO,IAAA,CAAK,cAAc,MAAM,CAAA;AAAA,EAClC;AAAA,EAEQ,aAAA,CAAc,IAAA,EAA0B,WAAA,GAAsB,EAAA,EAAS;AAE7E,IAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA,EAAG;AACtB,MAAA,MAAM,IAAI,MAAM,gEAAgE,CAAA;AAAA,IAClF;AACA,IAAA,IAAI,KAAK,WAAA,CAAY,IAAI,GAAG,OAAO,IAAA,CAAK,yBAAyB,IAAI,CAAA;AACrE,IAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,EAAG,OAAO,EAAE,GAAA,EAAK,IAAA,CAAK,oBAAA,CAAqB,IAAI,CAAA,EAAE;AAEvE,IAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,IAA2B,CAAA;AAC1D,IAAA,MAAM,UAAA,GAAa,QAAQ,CAAC,CAAA;AAE5B,IAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,IAAK,UAAA,IAAc,KAAK,UAAA,CAAW,UAAA,CAAW,CAAC,CAAC,CAAA,EAAG;AACxE,MAAA,MAAM,CAAC,QAAA,EAAU,KAAK,CAAA,GAAI,UAAA;AAC1B,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,iBAAA,CAAkB,QAAA,EAAU,KAAK,CAAA;AACzD,MAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,IAAK,KAAA,CAAM,QAAQ,UAAU,CAAA,IAAK,UAAA,CAAW,MAAA,KAAW,CAAA,EAAG;AAC5F,QAAA,OAAO,WAAW,CAAC,CAAA;AAAA,MACrB;AACA,MAAA,OAAO,IAAA,CAAK,kBAAkB,QAAQ,CAAA,GAAI,EAAE,CAAC,QAAQ,GAAG,UAAA,EAAW,GAAI,UAAA;AAAA,IACzE;AAGA,IAAA,MAAM,SAA8B,EAAC;AACrC,IAAA,MAAM,0BAAiC,EAAC;AAExC,IAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,CAAA,IAAK,OAAA,EAAS;AAClC,MAAA,MAAM,UAAU,WAAA,GAAc,CAAA,EAAG,WAAW,CAAA,CAAA,EAAI,GAAG,CAAA,CAAA,GAAK,GAAA;AAExD,MAAA,IAAI,IAAA,CAAK,UAAA,CAAW,GAAG,CAAA,EAAG;AACxB,QAAA,MAAA,CAAO,GAAG,CAAA,GAAI,IAAA,CAAK,iBAAA,CAAkB,KAAK,KAAK,CAAA;AAC/C,QAAA;AAAA,MACF;AAEA,MAAA,IAAI,OAAO,UAAU,QAAA,IAAY,KAAA,KAAU,QAAQ,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AAExE,QAAA,MAAM,YAAA,GAAe,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA;AACzC,QAAA,IAAI,YAAA,CAAa,KAAA,CAAM,CAAC,CAAC,EAAE,CAAA,KAAM,IAAA,CAAK,UAAA,CAAW,EAAE,CAAC,CAAA,IAAK,YAAA,CAAa,SAAS,CAAA,EAAG;AAChF,UAAA,YAAA,CAAa,OAAA,CAAQ,CAAC,CAAC,EAAA,EAAI,OAAO,CAAA,KAAM;AACtC,YAAA,uBAAA,CAAwB,IAAA,CAAK;AAAA,cAC3B,CAAC,OAAO,GAAG,EAAE,CAAC,EAAE,GAAG,IAAA,CAAK,wBAAA,CAAyB,OAAO,CAAA;AAAE,aAC3D,CAAA;AAAA,UACH,CAAC,CAAA;AACD,UAAA;AAAA,QACF;AAGA,QAAA,IAAI,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA,CAAE,WAAW,CAAA,EAAG;AACnC,UAAA,MAAA,CAAO,OAAO,CAAA,GAAI,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAAA,QAC5C,CAAA,MAAO;AACL,UAAA,MAAM,YAAA,GAAe,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA,CAAE,KAAK,CAAA,CAAA,KAAK,IAAA,CAAK,UAAA,CAAW,CAAC,CAAC,CAAA;AACpE,UAAA,IAAI,YAAA,EAAc;AAEhB,YAAA,MAAM,kBAAuC,EAAC;AAC9C,YAAA,KAAA,MAAW,CAAC,EAAA,EAAI,OAAO,KAAK,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,EAAG;AACjD,cAAA,eAAA,CAAgB,EAAE,CAAA,GAAI,IAAA,CAAK,UAAA,CAAW,EAAE,IAAI,IAAA,CAAK,iBAAA,CAAkB,EAAA,EAAI,OAAO,CAAA,GAAI,OAAA;AAAA,YACpF;AACA,YAAA,MAAA,CAAO,OAAO,CAAA,GAAI,eAAA;AAAA,UACpB,CAAA,MAAO;AAEL,YAAA,MAAA,CAAO,OAAO,MAAA,EAAQ,IAAA,CAAK,aAAA,CAAc,KAAA,EAAO,OAAO,CAAC,CAAA;AAAA,UAC1D;AAAA,QACF;AAAA,MACF,CAAA,MAAO;AACL,QAAA,MAAA,CAAO,OAAO,CAAA,GAAI,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAAA,MAC5C;AAAA,IACF;AAGA,IAAA,IAAI,uBAAA,CAAwB,SAAS,CAAA,EAAG;AACtC,MAAA,OAAO,EAAE,MAAM,uBAAA,EAAwB;AAAA,IACzC;AAGA,IAAA,IAAI,OAAO,IAAA,CAAK,MAAM,EAAE,MAAA,GAAS,CAAA,IAAK,CAAC,WAAA,EAAa;AAClD,MAAA,OAAO;AAAA,QACL,MAAM,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,CAAE,IAAI,CAAC,CAAC,GAAA,EAAK,KAAK,OAAO,EAAE,CAAC,GAAG,GAAG,OAAM,CAAE;AAAA,OACvE;AAAA,IACF;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA,EAEQ,iBAAA,CAAkB,UAAyB,KAAA,EAAiB;AAElE,IAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,EAAG;AACpC,MAAA,OAAO,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,GAAI,MAAM,GAAA,CAAI,CAAA,IAAA,KAAQ,IAAA,CAAK,aAAA,CAAc,IAAI,CAAC,CAAA,GAAI,IAAA,CAAK,cAAc,KAAK,CAAA;AAAA,IACtG;AAGA,IAAA,OAAO,IAAA,CAAK,yBAAyB,KAAK,CAAA;AAAA,EAC5C;AACF,CAAA;;;AC/FA,IAAM,aAAA,GAAgB;AAAA,EACpB,MAAA,EAAQ,QAAA;AAAA,EACR,SAAA,EAAW,IAAA;AAAA,EACX,UAAA,EAAY,IAAA;AAAA,EACZ,EAAA,EAAI,WAAA;AAAA,EACJ,EAAA,EAAI;AACN,CAAA;AAEO,IAAM,YAAA,GAAN,cAA2B,YAAA,CAAiC;AAAA,EACzD,MAAA;AAAA,EACA,WAAA;AAAA,EAER,YAAY,gBAAA,EAAqC;AAC/C,IAAA,KAAA,EAAM;AACN,IAAA,IAAI,kBAAkB,MAAA,EAAQ;AAC5B,MAAA,IAAA,CAAK,MAAA,GAAS,IAAI,WAAA,CAAY;AAAA,QAC5B,QAAQ,gBAAA,CAAiB,MAAA;AAAA,QACzB,QAAQ,gBAAA,CAAiB,MAAA;AAAA,QACzB,UAAU,gBAAA,CAAiB;AAAA,OAC5B,CAAA;AAAA,IACH,CAAA,MAAO;AACL,MAAA,IAAA,CAAK,MAAA,GAAS,IAAI,YAAA,CAAa,gBAAgB,CAAA;AAAA,IACjD;AACA,IAAA,IAAA,CAAK,WAAA,uBAAkB,GAAA,EAAI;AAAA,EAC7B;AAAA,EAEA,MAAM,aAAA,CAAc,EAAE,SAAA,EAAW,WAAA,GAAc,OAAM,EAAiD;AACpG,IAAA,IAAI,UAAA,GAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAS,CAAA;AAC/C,IAAA,IAAI,WAAA,IAAe,CAAC,UAAA,EAAY;AAC9B,MAAA,IAAI;AACF,QAAA,UAAA,GAAa,MAAM,IAAA,CAAK,MAAA,CAAO,cAAc,EAAE,IAAA,EAAM,WAAW,CAAA;AAChE,QAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAA,EAAW,UAAU,CAAA;AAC1C,QAAA,OAAO,UAAA;AAAA,MACT,CAAA,CAAA,MAAQ;AACN,QAAA,MAAM,IAAI,WAAA,CAAY;AAAA,UACpB,EAAA,EAAI,8BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACtB,CAAA;AAAA,MACH;AAAA,IACF;AACA,IAAA,OAAO,UAAA;AAAA,EACT;AAAA,EAEQ,wBAAA,CAAyB,SAAqB,SAAA,EAAyB;AAC7E,IAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,OAAA,CAAQ,QAAQ,CAAA,EAAA,EAAK;AACvC,MAAA,IAAI,OAAA,GAAU,CAAC,CAAA,EAAG,MAAA,KAAW,SAAA,EAAW;AACtC,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,CAAA,gBAAA,EAAmB,CAAC,CAAA,uBAAA,EAA0B,OAAA,GAAU,CAAC,CAAA,EAAG,MAAM,cAAc,SAAS,CAAA,YAAA;AAAA,SAC3F;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,EAAE,SAAA,EAAW,SAAS,QAAA,EAAU,GAAA,EAAK,WAAU,EAAgD;AAC1G,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAEzD,MAAA,MAAM,QAAQ,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACpD,MAAA,IAAA,CAAK,wBAAA,CAAyB,OAAA,EAAS,KAAA,CAAM,SAAS,CAAA;AACtD,MAAA,MAAM,eAAe,GAAA,IAAO,OAAA,CAAQ,IAAI,MAAM,MAAA,CAAO,YAAY,CAAA;AAEjE,MAAA,MAAM,WAAW,MAAA,CAAO;AAAA,QACtB,GAAA,EAAK,YAAA;AAAA,QACL,UAAA,EAAY,OAAA;AAAA,QACZ,SAAA,EAAW,QAAA;AAAA,QACX;AAAA,OACD,CAAA;AAED,MAAA,OAAO,YAAA;AAAA,IACT,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,CAAY,EAAE,WAAW,SAAA,EAAW,MAAA,GAAS,UAAS,EAAqC;AAC/F,IAAA,IAAI,CAAC,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA,IAAK,aAAa,CAAA,EAAG;AAClD,MAAA,MAAM,IAAI,WAAA,CAAY;AAAA,QACpB,EAAA,EAAI,8CAAA;AAAA,QACJ,IAAA,EAAM,sCAAA;AAAA,QACN,QAAQ,WAAA,CAAY,aAAA;AAAA,QACpB,UAAU,aAAA,CAAc,IAAA;AAAA,QACxB,OAAA,EAAS,EAAE,SAAA;AAAU,OACtB,CAAA;AAAA,IACH;AAEA,IAAA,MAAM,SAAA,GAAY,cAAc,MAAM,CAAA;AAEtC,IAAA,IAAI,CAAC,SAAA,IAAa,CAAC,CAAC,QAAA,EAAU,MAAM,IAAI,CAAA,CAAE,QAAA,CAAS,SAAS,CAAA,EAAG;AAC7D,MAAA,MAAM,IAAI,WAAA,CAAY;AAAA,QACpB,EAAA,EAAI,2CAAA;AAAA,QACJ,IAAA,EAAM,oBAAoB,MAAM,CAAA,gDAAA,CAAA;AAAA,QAChC,QAAQ,WAAA,CAAY,aAAA;AAAA,QACpB,UAAU,aAAA,CAAc,IAAA;AAAA,QACxB,OAAA,EAAS,EAAE,MAAA;AAAO,OACnB,CAAA;AAAA,IACH;AAEA,IAAA,IAAI;AACF,MAAA,MAAM,UAAA,GAAa,MAAM,IAAA,CAAK,MAAA,CAAO,gBAAA,CAAiB;AAAA,QACpD,IAAA,EAAM,SAAA;AAAA,QACN,QAAA,EAAU,EAAE,SAAA,EAAU;AAAA,QACtB,eAAe,EAAE,IAAA,EAAM,EAAE,KAAA,EAAO,WAAU,EAAE;AAAA,QAC5C,iBAAA,EAAmB;AAAA,OACpB,CAAA;AACD,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAA,EAAW,UAAU,CAAA;AAAA,IAC5C,SAAS,KAAA,EAAY;AAEnB,MAAA,MAAM,OAAA,GAAU,KAAA,EAAO,OAAA,IAAW,KAAA,EAAO,QAAA,EAAS;AAClD,MAAA,IAAI,WAAW,OAAA,CAAQ,WAAA,EAAY,CAAE,QAAA,CAAS,gBAAgB,CAAA,EAAG;AAE/D,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,SAAA,EAAW,SAAA,EAAW,MAAM,CAAA;AAC7D,QAAA;AAAA,MACF;AACA,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,gBAAgB,MAAA,EAA6B;AAC3C,IAAA,MAAM,UAAA,GAAa,IAAI,sBAAA,EAAuB;AAC9C,IAAA,MAAM,gBAAA,GAAmB,UAAA,CAAW,SAAA,CAAU,MAAM,CAAA;AACpD,IAAA,OAAO,mBAAoB,gBAAA,GAA6B,MAAA;AAAA,EAC1D;AAAA,EAEA,MAAM,KAAA,CAAqC;AAAA,IACzC,SAAA;AAAA,IACA,WAAA;AAAA,IACA,IAAA,GAAO,EAAA;AAAA,IACP,MAAA;AAAA,IACA,aAAA,GAAgB,KAAA;AAAA,IAChB;AAAA,GACF,EAAoD;AAClD,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAEzD,MAAA,MAAM,cAAA,GAA0D,CAAC,WAAA,EAAa,WAAA,EAAa,WAAW,CAAA;AAEtG,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AACpD,MAAA,MAAM,OAAA,GAAU,MAAM,UAAA,CAAW,KAAA,CAAS;AAAA,QACxC,eAAA,EAAiB,CAAC,WAAW,CAAA;AAAA,QAC7B,QAAA,EAAU,IAAA;AAAA,QACV,OAAO,gBAAA,IAAoB,MAAA;AAAA,QAC3B,eAAe,cAAA,IAAkB,MAAA;AAAA,QACjC,SAAS,aAAA,GAAgB,CAAC,GAAG,cAAA,EAAgB,YAAY,CAAA,GAAI;AAAA,OAC9D,CAAA;AAED,MAAA,OAAA,CAAQ,OAAA,CAAQ,IAAI,CAAC,CAAA,IAAK,EAAC,EAAG,GAAA,CAAI,CAAC,EAAA,EAAY,KAAA,MAAmB;AAAA,QAChE,EAAA;AAAA,QACA,OAAO,OAAA,CAAQ,SAAA,GAAY,CAAC,CAAA,GAAI,KAAK,CAAA,IAAK,CAAA;AAAA,QAC1C,UAAU,OAAA,CAAQ,SAAA,GAAY,CAAC,CAAA,GAAI,KAAK,KAAK,EAAC;AAAA,QAC9C,UAAU,OAAA,CAAQ,SAAA,GAAY,CAAC,CAAA,GAAI,KAAK,CAAA,IAAK,MAAA;AAAA,QAC7C,GAAI,aAAA,IAAiB,EAAE,MAAA,EAAQ,OAAA,CAAQ,UAAA,GAAa,CAAC,CAAA,GAAI,KAAK,CAAA,IAAK,EAAC;AAAE,OACxE,CAAE,CAAA;AAAA,IACJ,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,4BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,GAAA,CAAmC;AAAA,IACvC,SAAA;AAAA,IACA,GAAA;AAAA,IACA,MAAA;AAAA,IACA,aAAA,GAAgB,KAAA;AAAA,IAChB,cAAA;AAAA,IACA,MAAA;AAAA,IACA;AAAA,GACF,EAA2B;AACzB,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAEzD,MAAA,MAAM,cAAA,GAA6C,CAAC,WAAA,EAAa,WAAW,CAAA;AAC5E,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AAEpD,MAAA,MAAM,MAAA,GAAS,MAAM,UAAA,CAAW,GAAA,CAAO;AAAA,QACrC,GAAA;AAAA,QACA,OAAO,gBAAA,IAAoB,MAAA;AAAA,QAC3B,eAAe,cAAA,IAAkB,MAAA;AAAA,QACjC,MAAA;AAAA,QACA,KAAA;AAAA,QACA,SAAS,aAAA,GAAgB,CAAC,GAAG,cAAA,EAAgB,YAAY,CAAA,GAAI;AAAA,OAC9D,CAAA;AACD,MAAA,OAAO,OAAO,IAAA,EAAK;AAAA,IACrB,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,GAAiC;AACrC,IAAA,IAAI;AACF,MAAA,MAAM,WAAA,GAAc,MAAM,IAAA,CAAK,MAAA,CAAO,eAAA,EAAgB;AACtD,MAAA,OAAO,WAAA,CAAY,GAAA,CAAI,CAAA,UAAA,KAAc,UAAA,CAAW,IAAI,CAAA;AAAA,IACtD,SAAS,KAAA,EAAY;AACnB,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc;AAAA,SAC1B;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,aAAA,CAAc,EAAE,SAAA,EAAU,EAA6C;AAC3E,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACzD,MAAA,MAAM,KAAA,GAAQ,MAAM,UAAA,CAAW,KAAA,EAAM;AACrC,MAAA,MAAM,WAAW,UAAA,CAAW,QAAA;AAC5B,MAAA,MAAM,KAAA,GAAQ,WAAW,aAAA,CAAc,IAAA,EAAM,SAAS,UAAA,CAAW,aAAA,CAAc,OAAO,KAAA,IAAS,MAAA;AAE/F,MAAA,OAAO;AAAA,QACL,SAAA,EAAW,UAAU,SAAA,IAAa,CAAA;AAAA,QAClC,KAAA;AAAA,QACA,MAAA,EAAQ,KAAA,GAAS,aAAA,CAAc,KAAK,CAAA,GAA8C;AAAA,OACpF;AAAA,IACF,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,qCAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,CAAY,EAAE,SAAA,EAAU,EAAqC;AACjE,IAAA,IAAI;AACF,MAAA,MAAM,KAAK,MAAA,CAAO,gBAAA,CAAiB,EAAE,IAAA,EAAM,WAAW,CAAA;AACtD,MAAA,IAAA,CAAK,WAAA,CAAY,OAAO,SAAS,CAAA;AAAA,IACnC,SAAS,KAAA,EAAY;AACnB,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,SAAA,CAAU,EAAE,SAAA,EAAW,cAAa,EAA+D;AACvG,IAAA,IAAI;AACF,MAAA,MAAM,UAAA,GAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,SAAA,EAAW,WAAA,EAAa,MAAM,CAAA;AAC5E,MAAA,MAAM,mBAAmB,MAAM,UAAA,CAAW,KAAK,EAAE,IAAA,EAAM,cAAc,CAAA;AACrE,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,YAAA,EAAc,gBAAgB,CAAA;AAAA,IACrD,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,YAAA,CAAa,EAAE,SAAA,EAAW,EAAA,EAAI,QAAO,EAAsC;AAC/E,IAAA,IAAI,CAAC,MAAA,CAAO,MAAA,IAAU,CAAC,OAAO,QAAA,EAAU;AACtC,MAAA,MAAM,IAAI,WAAA,CAAY;AAAA,QACpB,EAAA,EAAI,iCAAA;AAAA,QACJ,IAAA,EAAM,gCAAA;AAAA,QACN,QAAQ,WAAA,CAAY,aAAA;AAAA,QACpB,UAAU,aAAA,CAAc,IAAA;AAAA,QACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,OAC1B,CAAA;AAAA,IACH;AAEA,IAAA,IAAI;AACF,MAAA,MAAM,aAAyB,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAErE,MAAA,MAAM,eAAA,GAA6B,EAAE,GAAA,EAAK,CAAC,EAAE,CAAA,EAAE;AAE/C,MAAA,IAAI,QAAQ,MAAA,EAAQ;AAClB,QAAA,MAAM,QAAQ,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACpD,QAAA,IAAA,CAAK,yBAAyB,CAAC,MAAA,CAAO,MAAM,CAAA,EAAG,MAAM,SAAS,CAAA;AAC9D,QAAA,eAAA,CAAgB,UAAA,GAAa,CAAC,MAAA,CAAO,MAAM,CAAA;AAAA,MAC7C;AAEA,MAAA,IAAI,QAAQ,QAAA,EAAU;AACpB,QAAA,eAAA,CAAgB,SAAA,GAAY,CAAC,MAAA,CAAO,QAAQ,CAAA;AAAA,MAC9C;AAEA,MAAA,OAAO,MAAM,UAAA,CAAW,MAAA,CAAO,eAAe,CAAA;AAAA,IAChD,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,YAAA,CAAa,EAAE,SAAA,EAAW,IAAG,EAAsC;AACvE,IAAA,IAAI;AACF,MAAA,MAAM,aAAyB,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACrE,MAAA,MAAM,WAAW,MAAA,CAAO,EAAE,KAAK,CAAC,EAAE,GAAG,CAAA;AAAA,IACvC,SAAS,KAAA,EAAY;AACnB,MAAA,IAAI,KAAA,YAAiB,aAAa,MAAM,KAAA;AACxC,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,aAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AACF;;;AC7ZO,IAAM,aAAA,GAAgB,CAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA","file":"index.js","sourcesContent":["import { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type {\n VectorFilter,\n OperatorSupport,\n QueryOperator,\n OperatorValueMap,\n LogicalOperatorValueMap,\n BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\n\ntype ChromaOperatorValueMap = Omit<OperatorValueMap, '$exists' | '$elemMatch' | '$regex' | '$options'>;\n\ntype ChromaLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor' | '$not'>;\n\ntype ChromaBlacklisted = BlacklistedRootOperators | '$nor' | '$not';\n\nexport type ChromaVectorFilter = VectorFilter<\n keyof ChromaOperatorValueMap,\n ChromaOperatorValueMap,\n ChromaLogicalOperatorValueMap,\n ChromaBlacklisted\n>;\n\n/**\n * Translator for Chroma filter queries.\n * Maintains MongoDB-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class ChromaFilterTranslator extends BaseFilterTranslator<ChromaVectorFilter> {\n protected override getSupportedOperators(): OperatorSupport {\n return {\n ...BaseFilterTranslator.DEFAULT_OPERATORS,\n logical: ['$and', '$or'],\n array: ['$in', '$nin'],\n element: [],\n regex: [],\n custom: [],\n };\n }\n\n translate(filter?: ChromaVectorFilter): ChromaVectorFilter {\n if (this.isEmpty(filter)) return filter;\n this.validateFilter(filter);\n\n return this.translateNode(filter);\n }\n\n private translateNode(node: ChromaVectorFilter, currentPath: string = ''): any {\n // Handle primitive values and arrays\n if (this.isRegex(node)) {\n throw new Error('Regex is supported in Chroma via the `documentFilter` argument');\n }\n if (this.isPrimitive(node)) return this.normalizeComparisonValue(node);\n if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };\n\n const entries = Object.entries(node as Record<string, any>);\n const firstEntry = entries[0];\n // Handle single operator case\n if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {\n const [operator, value] = firstEntry;\n const translated = this.translateOperator(operator, value);\n if (this.isLogicalOperator(operator) && Array.isArray(translated) && translated.length === 1) {\n return translated[0];\n }\n return this.isLogicalOperator(operator) ? { [operator]: translated } : translated;\n }\n\n // Process each entry\n const result: Record<string, any> = {};\n const multiOperatorConditions: any[] = [];\n\n for (const [key, value] of entries) {\n const newPath = currentPath ? `${currentPath}.${key}` : key;\n\n if (this.isOperator(key)) {\n result[key] = this.translateOperator(key, value);\n continue;\n }\n\n if (typeof value === 'object' && value !== null && !Array.isArray(value)) {\n // Check for multiple operators on same field\n const valueEntries = Object.entries(value);\n if (valueEntries.every(([op]) => this.isOperator(op)) && valueEntries.length > 1) {\n valueEntries.forEach(([op, opValue]) => {\n multiOperatorConditions.push({\n [newPath]: { [op]: this.normalizeComparisonValue(opValue) },\n });\n });\n continue;\n }\n\n // Check if the nested object contains operators\n if (Object.keys(value).length === 0) {\n result[newPath] = this.translateNode(value);\n } else {\n const hasOperators = Object.keys(value).some(k => this.isOperator(k));\n if (hasOperators) {\n // For objects with operators, normalize each operator value\n const normalizedValue: Record<string, any> = {};\n for (const [op, opValue] of Object.entries(value)) {\n normalizedValue[op] = this.isOperator(op) ? this.translateOperator(op, opValue) : opValue;\n }\n result[newPath] = normalizedValue;\n } else {\n // For objects without operators, flatten them\n Object.assign(result, this.translateNode(value, newPath));\n }\n }\n } else {\n result[newPath] = this.translateNode(value);\n }\n }\n\n // If we have multiple operators, return them combined with $and\n if (multiOperatorConditions.length > 0) {\n return { $and: multiOperatorConditions };\n }\n\n // Wrap in $and if there are multiple top-level fields\n if (Object.keys(result).length > 1 && !currentPath) {\n return {\n $and: Object.entries(result).map(([key, value]) => ({ [key]: value })),\n };\n }\n\n return result;\n }\n\n private translateOperator(operator: QueryOperator, value: any): any {\n // Handle logical operators\n if (this.isLogicalOperator(operator)) {\n return Array.isArray(value) ? value.map(item => this.translateNode(item)) : this.translateNode(value);\n }\n\n // Handle comparison and element operators\n return this.normalizeComparisonValue(value);\n }\n}\n","import { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport { MastraVector } from '@mastra/core/vector';\nimport type {\n QueryResult,\n IndexStats,\n CreateIndexParams,\n UpsertVectorParams,\n QueryVectorParams,\n DescribeIndexParams,\n DeleteIndexParams,\n DeleteVectorParams,\n UpdateVectorParams,\n} from '@mastra/core/vector';\nimport { ChromaClient, CloudClient } from 'chromadb';\nimport type { ChromaClientArgs, RecordSet, Where, WhereDocument, Collection, Metadata } from 'chromadb';\nimport type { ChromaVectorFilter } from './filter';\nimport { ChromaFilterTranslator } from './filter';\n\ninterface ChromaUpsertVectorParams extends UpsertVectorParams {\n documents?: string[];\n}\n\ninterface ChromaQueryVectorParams extends QueryVectorParams<ChromaVectorFilter> {\n documentFilter?: WhereDocument | null;\n}\n\ninterface ChromaGetRecordsParams {\n indexName: string;\n ids?: string[];\n filter?: ChromaVectorFilter;\n documentFilter?: WhereDocument | null;\n includeVector?: boolean;\n limit?: number;\n offset?: number;\n}\n\ntype MastraMetadata = {\n dimension?: number;\n};\n\ntype ChromaVectorArgs = ChromaClientArgs & { apiKey?: string };\n\nconst spaceMappings = {\n cosine: 'cosine',\n euclidean: 'l2',\n dotproduct: 'ip',\n l2: 'euclidean',\n ip: 'dotproduct',\n};\n\nexport class ChromaVector extends MastraVector<ChromaVectorFilter> {\n private client: ChromaClient;\n private collections: Map<string, Collection>;\n\n constructor(chromaClientArgs?: ChromaVectorArgs) {\n super();\n if (chromaClientArgs?.apiKey) {\n this.client = new CloudClient({\n apiKey: chromaClientArgs.apiKey,\n tenant: chromaClientArgs.tenant,\n database: chromaClientArgs.database,\n });\n } else {\n this.client = new ChromaClient(chromaClientArgs);\n }\n this.collections = new Map();\n }\n\n async getCollection({ indexName, forceUpdate = false }: { indexName: string; forceUpdate?: boolean }) {\n let collection = this.collections.get(indexName);\n if (forceUpdate || !collection) {\n try {\n collection = await this.client.getCollection({ name: indexName });\n this.collections.set(indexName, collection);\n return collection;\n } catch {\n throw new MastraError({\n id: 'CHROMA_COLLECTION_GET_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n });\n }\n }\n return collection;\n }\n\n private validateVectorDimensions(vectors: number[][], dimension: number): void {\n for (let i = 0; i < vectors.length; i++) {\n if (vectors?.[i]?.length !== dimension) {\n throw new Error(\n `Vector at index ${i} has invalid dimension ${vectors?.[i]?.length}. Expected ${dimension} dimensions.`,\n );\n }\n }\n }\n\n async upsert({ indexName, vectors, metadata, ids, documents }: ChromaUpsertVectorParams): Promise<string[]> {\n try {\n const collection = await this.getCollection({ indexName });\n\n const stats = await this.describeIndex({ indexName });\n this.validateVectorDimensions(vectors, stats.dimension);\n const generatedIds = ids || vectors.map(() => crypto.randomUUID());\n\n await collection.upsert({\n ids: generatedIds,\n embeddings: vectors,\n metadatas: metadata,\n documents: documents,\n });\n\n return generatedIds;\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_UPSERT_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n if (!Number.isInteger(dimension) || dimension <= 0) {\n throw new MastraError({\n id: 'CHROMA_VECTOR_CREATE_INDEX_INVALID_DIMENSION',\n text: 'Dimension must be a positive integer',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.USER,\n details: { dimension },\n });\n }\n\n const hnswSpace = spaceMappings[metric] as 'cosine' | 'l2' | 'ip' | undefined;\n\n if (!hnswSpace || !['cosine', 'l2', 'ip'].includes(hnswSpace)) {\n throw new MastraError({\n id: 'CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC',\n text: `Invalid metric: \"${metric}\". Must be one of: cosine, euclidean, dotproduct`,\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.USER,\n details: { metric },\n });\n }\n\n try {\n const collection = await this.client.createCollection({\n name: indexName,\n metadata: { dimension },\n configuration: { hnsw: { space: hnswSpace } },\n embeddingFunction: null,\n });\n this.collections.set(indexName, collection);\n } catch (error: any) {\n // Check for 'already exists' error\n const message = error?.message || error?.toString();\n if (message && message.toLowerCase().includes('already exists')) {\n // Fetch collection info and check dimension\n await this.validateExistingIndex(indexName, dimension, metric);\n return;\n }\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_CREATE_INDEX_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n transformFilter(filter?: ChromaVectorFilter) {\n const translator = new ChromaFilterTranslator();\n const translatedFilter = translator.translate(filter);\n return translatedFilter ? (translatedFilter as Where) : undefined;\n }\n\n async query<T extends Metadata = Metadata>({\n indexName,\n queryVector,\n topK = 10,\n filter,\n includeVector = false,\n documentFilter,\n }: ChromaQueryVectorParams): Promise<QueryResult[]> {\n try {\n const collection = await this.getCollection({ indexName });\n\n const defaultInclude: ['documents', 'metadatas', 'distances'] = ['documents', 'metadatas', 'distances'];\n\n const translatedFilter = this.transformFilter(filter);\n const results = await collection.query<T>({\n queryEmbeddings: [queryVector],\n nResults: topK,\n where: translatedFilter ?? undefined,\n whereDocument: documentFilter ?? undefined,\n include: includeVector ? [...defaultInclude, 'embeddings'] : defaultInclude,\n });\n\n return (results.ids[0] || []).map((id: string, index: number) => ({\n id,\n score: results.distances?.[0]?.[index] || 0,\n metadata: results.metadatas?.[0]?.[index] || {},\n document: results.documents?.[0]?.[index] ?? undefined,\n ...(includeVector && { vector: results.embeddings?.[0]?.[index] || [] }),\n }));\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_QUERY_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async get<T extends Metadata = Metadata>({\n indexName,\n ids,\n filter,\n includeVector = false,\n documentFilter,\n offset,\n limit,\n }: ChromaGetRecordsParams) {\n try {\n const collection = await this.getCollection({ indexName });\n\n const defaultInclude: ['documents', 'metadatas'] = ['documents', 'metadatas'];\n const translatedFilter = this.transformFilter(filter);\n\n const result = await collection.get<T>({\n ids,\n where: translatedFilter ?? undefined,\n whereDocument: documentFilter ?? undefined,\n offset,\n limit,\n include: includeVector ? [...defaultInclude, 'embeddings'] : defaultInclude,\n });\n return result.rows();\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_GET_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async listIndexes(): Promise<string[]> {\n try {\n const collections = await this.client.listCollections();\n return collections.map(collection => collection.name);\n } catch (error: any) {\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_LIST_INDEXES_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n },\n error,\n );\n }\n }\n\n /**\n * Retrieves statistics about a vector index.\n *\n * @param {string} indexName - The name of the index to describe\n * @returns A promise that resolves to the index statistics including dimension, count and metric\n */\n async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {\n try {\n const collection = await this.getCollection({ indexName });\n const count = await collection.count();\n const metadata = collection.metadata as MastraMetadata | undefined;\n const space = collection.configuration.hnsw?.space || collection.configuration.spann?.space || undefined;\n\n return {\n dimension: metadata?.dimension || 0,\n count,\n metric: space ? (spaceMappings[space] as 'cosine' | 'euclidean' | 'dotproduct') : undefined,\n };\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_DESCRIBE_INDEX_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n try {\n await this.client.deleteCollection({ name: indexName });\n this.collections.delete(indexName);\n } catch (error: any) {\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_DELETE_INDEX_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async forkIndex({ indexName, newIndexName }: { indexName: string; newIndexName: string }): Promise<void> {\n try {\n const collection = await this.getCollection({ indexName, forceUpdate: true });\n const forkedCollection = await collection.fork({ name: newIndexName });\n this.collections.set(newIndexName, forkedCollection);\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_INDEX_FORK_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n /**\n * Updates a vector by its ID with the provided vector and/or metadata.\n * @param indexName - The name of the index containing the vector.\n * @param id - The ID of the vector to update.\n * @param update - An object containing the vector and/or metadata to update.\n * @param update.vector - An optional array of numbers representing the new vector.\n * @param update.metadata - An optional record containing the new metadata.\n * @returns A promise that resolves when the update is complete.\n * @throws Will throw an error if no updates are provided or if the update operation fails.\n */\n async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {\n if (!update.vector && !update.metadata) {\n throw new MastraError({\n id: 'CHROMA_VECTOR_UPDATE_NO_PAYLOAD',\n text: 'No updates provided for vector',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.USER,\n details: { indexName, id },\n });\n }\n\n try {\n const collection: Collection = await this.getCollection({ indexName });\n\n const updateRecordSet: RecordSet = { ids: [id] };\n\n if (update?.vector) {\n const stats = await this.describeIndex({ indexName });\n this.validateVectorDimensions([update.vector], stats.dimension);\n updateRecordSet.embeddings = [update.vector];\n }\n\n if (update?.metadata) {\n updateRecordSet.metadatas = [update.metadata];\n }\n\n return await collection.update(updateRecordSet);\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_UPDATE_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, id },\n },\n error,\n );\n }\n }\n\n async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n try {\n const collection: Collection = await this.getCollection({ indexName });\n await collection.delete({ ids: [id] });\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_DELETE_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, id },\n },\n error,\n );\n }\n }\n}\n","/**\n * Vector store specific prompt that details supported operators and examples.\n * This prompt helps users construct valid filters for Chroma Vector.\n */\nexport const CHROMA_PROMPT = `When querying Chroma, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n\nLogical Operators:\n- $and: Logical AND\n Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n\nRestrictions:\n- Regex patterns are not supported\n- Element operators are not supported\n- Only $and and $or logical operators are supported\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Empty arrays in $in/$nin will return no results\n- If multiple top-level fields exist, they're wrapped in $and\n- Only logical operators ($and, $or) can be used at the top level\n- All other operators must be used within a field condition\n Valid: { \"field\": { \"$gt\": 100 } }\n Valid: { \"$and\": [...] }\n Invalid: { \"$gt\": 100 }\n Invalid: { \"$in\": [...] }\n- Logical operators must contain field conditions, not direct operators\n Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- Logical operators ($and, $or):\n - Can only be used at top level or nested within other logical operators\n - Can not be used on a field level, or be nested inside a field\n - Can not be used inside an operator\n - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n\nExample Complex Query:\n{\n \"$and\": [\n { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n { \"$or\": [\n { \"inStock\": true },\n { \"preorder\": true }\n ]}\n ]\n}`;\n"]}
@@ -0,0 +1,19 @@
1
+ import { BaseFilterTranslator } from '@mastra/core/vector/filter';
2
+ import type { VectorFilter, OperatorSupport, OperatorValueMap, LogicalOperatorValueMap, BlacklistedRootOperators } from '@mastra/core/vector/filter';
3
+ type ChromaOperatorValueMap = Omit<OperatorValueMap, '$exists' | '$elemMatch' | '$regex' | '$options'>;
4
+ type ChromaLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor' | '$not'>;
5
+ type ChromaBlacklisted = BlacklistedRootOperators | '$nor' | '$not';
6
+ export type ChromaVectorFilter = VectorFilter<keyof ChromaOperatorValueMap, ChromaOperatorValueMap, ChromaLogicalOperatorValueMap, ChromaBlacklisted>;
7
+ /**
8
+ * Translator for Chroma filter queries.
9
+ * Maintains MongoDB-compatible syntax while ensuring proper validation
10
+ * and normalization of values.
11
+ */
12
+ export declare class ChromaFilterTranslator extends BaseFilterTranslator<ChromaVectorFilter> {
13
+ protected getSupportedOperators(): OperatorSupport;
14
+ translate(filter?: ChromaVectorFilter): ChromaVectorFilter;
15
+ private translateNode;
16
+ private translateOperator;
17
+ }
18
+ export {};
19
+ //# sourceMappingURL=filter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"filter.d.ts","sourceRoot":"","sources":["../../src/vector/filter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EACV,YAAY,EACZ,eAAe,EAEf,gBAAgB,EAChB,uBAAuB,EACvB,wBAAwB,EACzB,MAAM,4BAA4B,CAAC;AAEpC,KAAK,sBAAsB,GAAG,IAAI,CAAC,gBAAgB,EAAE,SAAS,GAAG,YAAY,GAAG,QAAQ,GAAG,UAAU,CAAC,CAAC;AAEvG,KAAK,6BAA6B,GAAG,IAAI,CAAC,uBAAuB,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;AAEpF,KAAK,iBAAiB,GAAG,wBAAwB,GAAG,MAAM,GAAG,MAAM,CAAC;AAEpE,MAAM,MAAM,kBAAkB,GAAG,YAAY,CAC3C,MAAM,sBAAsB,EAC5B,sBAAsB,EACtB,6BAA6B,EAC7B,iBAAiB,CAClB,CAAC;AAEF;;;;GAIG;AACH,qBAAa,sBAAuB,SAAQ,oBAAoB,CAAC,kBAAkB,CAAC;cAC/D,qBAAqB,IAAI,eAAe;IAW3D,SAAS,CAAC,MAAM,CAAC,EAAE,kBAAkB,GAAG,kBAAkB;IAO1D,OAAO,CAAC,aAAa;IAiFrB,OAAO,CAAC,iBAAiB;CAS1B"}