@mastra/chroma 0.0.0-tsconfig-compile-20250703214351 → 0.0.0-unified-sidebar-20251010130811

package/dist/index.cjs

@@ -24,7 +24,7 @@ var ChromaFilterTranslator = class extends filter.BaseFilterTranslator {
   }
   translateNode(node, currentPath = "") {
     if (this.isRegex(node)) {
-      throw new Error("Regex is not supported in Chroma");
+      throw new Error("Regex is supported in Chroma via the `documentFilter` argument");
     }
     if (this.isPrimitive(node)) return this.normalizeComparisonValue(node);
     if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };
@@ -33,6 +33,9 @@ var ChromaFilterTranslator = class extends filter.BaseFilterTranslator {
     if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {
       const [operator, value] = firstEntry;
       const translated = this.translateOperator(operator, value);
+      if (this.isLogicalOperator(operator) && Array.isArray(translated) && translated.length === 1) {
+        return translated[0];
+      }
       return this.isLogicalOperator(operator) ? { [operator]: translated } : translated;
     }
     const result = {};
@@ -90,31 +93,46 @@ var ChromaFilterTranslator = class extends filter.BaseFilterTranslator {
 };
 
 // src/vector/index.ts
+var spaceMappings = {
+  cosine: "cosine",
+  euclidean: "l2",
+  dotproduct: "ip",
+  l2: "euclidean",
+  ip: "dotproduct"
+};
 var ChromaVector = class extends vector.MastraVector {
   client;
   collections;
-  constructor({
-    path,
-    auth
-  }) {
+  constructor(chromaClientArgs) {
     super();
-    this.client = new chromadb.ChromaClient({
-      path,
-      auth
-    });
+    if (chromaClientArgs?.apiKey) {
+      this.client = new chromadb.CloudClient({
+        apiKey: chromaClientArgs.apiKey,
+        tenant: chromaClientArgs.tenant,
+        database: chromaClientArgs.database
+      });
+    } else {
+      this.client = new chromadb.ChromaClient(chromaClientArgs);
+    }
     this.collections = /* @__PURE__ */ new Map();
   }
-  async getCollection(indexName, throwIfNotExists = true) {
-    try {
-      const collection = await this.client.getCollection({ name: indexName, embeddingFunction: void 0 });
-      this.collections.set(indexName, collection);
-    } catch {
-      if (throwIfNotExists) {
-        throw new Error(`Index ${indexName} does not exist`);
+  async getCollection({ indexName, forceUpdate = false }) {
+    let collection = this.collections.get(indexName);
+    if (forceUpdate || !collection) {
+      try {
+        collection = await this.client.getCollection({ name: indexName });
+        this.collections.set(indexName, collection);
+        return collection;
+      } catch {
+        throw new error.MastraError({
+          id: "CHROMA_COLLECTION_GET_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        });
       }
-      return null;
     }
-    return this.collections.get(indexName);
+    return collection;
   }
   validateVectorDimensions(vectors, dimension) {
     for (let i = 0; i < vectors.length; i++) {
@@ -127,15 +145,14 @@ var ChromaVector = class extends vector.MastraVector {
   }
   async upsert({ indexName, vectors, metadata, ids, documents }) {
     try {
-      const collection = await this.getCollection(indexName);
+      const collection = await this.getCollection({ indexName });
       const stats = await this.describeIndex({ indexName });
       this.validateVectorDimensions(vectors, stats.dimension);
       const generatedIds = ids || vectors.map(() => crypto.randomUUID());
-      const normalizedMetadata = metadata || vectors.map(() => ({}));
       await collection.upsert({
         ids: generatedIds,
         embeddings: vectors,
-        metadatas: normalizedMetadata,
+        metadatas: metadata,
         documents
       });
       return generatedIds;
@@ -152,13 +169,6 @@ var ChromaVector = class extends vector.MastraVector {
       );
     }
   }
-  HnswSpaceMap = {
-    cosine: "cosine",
-    euclidean: "l2",
-    dotproduct: "ip",
-    l2: "euclidean",
-    ip: "dotproduct"
-  };
   async createIndex({ indexName, dimension, metric = "cosine" }) {
     if (!Number.isInteger(dimension) || dimension <= 0) {
       throw new error.MastraError({
@@ -169,7 +179,7 @@ var ChromaVector = class extends vector.MastraVector {
         details: { dimension }
       });
     }
-    const hnswSpace = this.HnswSpaceMap[metric];
+    const hnswSpace = spaceMappings[metric];
     if (!hnswSpace || !["cosine", "l2", "ip"].includes(hnswSpace)) {
       throw new error.MastraError({
         id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC",
@@ -180,13 +190,13 @@ var ChromaVector = class extends vector.MastraVector {
       });
     }
     try {
-      await this.client.createCollection({
+      const collection = await this.client.createCollection({
         name: indexName,
-        metadata: {
-          dimension,
-          "hnsw:space": hnswSpace
-        }
+        metadata: { dimension },
+        configuration: { hnsw: { space: hnswSpace } },
+        embeddingFunction: null
       });
+      this.collections.set(indexName, collection);
     } catch (error$1) {
       const message = error$1?.message || error$1?.toString();
       if (message && message.toLowerCase().includes("already exists")) {
@@ -206,7 +216,8 @@ var ChromaVector = class extends vector.MastraVector {
   }
   transformFilter(filter) {
     const translator = new ChromaFilterTranslator();
-    return translator.translate(filter);
+    const translatedFilter = translator.translate(filter);
+    return translatedFilter ? translatedFilter : void 0;
   }
   async query({
     indexName,
@@ -217,21 +228,21 @@ var ChromaVector = class extends vector.MastraVector {
     documentFilter
   }) {
     try {
-      const collection = await this.getCollection(indexName, true);
+      const collection = await this.getCollection({ indexName });
       const defaultInclude = ["documents", "metadatas", "distances"];
       const translatedFilter = this.transformFilter(filter);
       const results = await collection.query({
         queryEmbeddings: [queryVector],
         nResults: topK,
-        where: translatedFilter,
-        whereDocument: documentFilter,
+        where: translatedFilter ?? void 0,
+        whereDocument: documentFilter ?? void 0,
         include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
       });
       return (results.ids[0] || []).map((id, index) => ({
         id,
         score: results.distances?.[0]?.[index] || 0,
         metadata: results.metadatas?.[0]?.[index] || {},
-        document: results.documents?.[0]?.[index],
+        document: results.documents?.[0]?.[index] ?? void 0,
         ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
       }));
     } catch (error$1) {
@@ -247,10 +258,45 @@ var ChromaVector = class extends vector.MastraVector {
       );
     }
   }
+  async get({
+    indexName,
+    ids,
+    filter,
+    includeVector = false,
+    documentFilter,
+    offset,
+    limit
+  }) {
+    try {
+      const collection = await this.getCollection({ indexName });
+      const defaultInclude = ["documents", "metadatas"];
+      const translatedFilter = this.transformFilter(filter);
+      const result = await collection.get({
+        ids,
+        where: translatedFilter ?? void 0,
+        whereDocument: documentFilter ?? void 0,
+        offset,
+        limit,
+        include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
+      });
+      return result.rows();
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_GET_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
+  }
   async listIndexes() {
     try {
       const collections = await this.client.listCollections();
-      return collections.map((collection) => collection);
+      return collections.map((collection) => collection.name);
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -270,14 +316,14 @@ var ChromaVector = class extends vector.MastraVector {
    */
   async describeIndex({ indexName }) {
     try {
-      const collection = await this.getCollection(indexName);
+      const collection = await this.getCollection({ indexName });
       const count = await collection.count();
       const metadata = collection.metadata;
-      const hnswSpace = metadata?.["hnsw:space"];
+      const space = collection.configuration.hnsw?.space || collection.configuration.spann?.space || void 0;
       return {
         dimension: metadata?.dimension || 0,
         count,
-        metric: this.HnswSpaceMap[hnswSpace]
+        metric: space ? spaceMappings[space] : void 0
       };
     } catch (error$1) {
       if (error$1 instanceof error.MastraError) throw error$1;
@@ -308,6 +354,24 @@ var ChromaVector = class extends vector.MastraVector {
       );
     }
   }
+  async forkIndex({ indexName, newIndexName }) {
+    try {
+      const collection = await this.getCollection({ indexName, forceUpdate: true });
+      const forkedCollection = await collection.fork({ name: newIndexName });
+      this.collections.set(newIndexName, forkedCollection);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_INDEX_FORK_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
+  }
   /**
    * Updates a vector by its ID with the provided vector and/or metadata.
    * @param indexName - The name of the index containing the vector.
@@ -329,17 +393,17 @@ var ChromaVector = class extends vector.MastraVector {
       });
     }
     try {
-      const collection = await this.getCollection(indexName, true);
-      const updateOptions = { ids: [id] };
+      const collection = await this.getCollection({ indexName });
+      const updateRecordSet = { ids: [id] };
       if (update?.vector) {
         const stats = await this.describeIndex({ indexName });
         this.validateVectorDimensions([update.vector], stats.dimension);
-        updateOptions.embeddings = [update.vector];
+        updateRecordSet.embeddings = [update.vector];
       }
       if (update?.metadata) {
-        updateOptions.metadatas = [update.metadata];
+        updateRecordSet.metadatas = [update.metadata];
       }
-      return await collection.update(updateOptions);
+      return await collection.update(updateRecordSet);
     } catch (error$1) {
       if (error$1 instanceof error.MastraError) throw error$1;
       throw new error.MastraError(
@@ -355,7 +419,7 @@ var ChromaVector = class extends vector.MastraVector {
   }
   async deleteVector({ indexName, id }) {
     try {
-      const collection = await this.getCollection(indexName, true);
+      const collection = await this.getCollection({ indexName });
       await collection.delete({ ids: [id] });
     } catch (error$1) {
       if (error$1 instanceof error.MastraError) throw error$1;
@@ -444,3 +508,5 @@ Example Complex Query:
 
 exports.CHROMA_PROMPT = CHROMA_PROMPT;
 exports.ChromaVector = ChromaVector;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"names":["BaseFilterTranslator","MastraVector","CloudClient","ChromaClient","MastraError","ErrorDomain","ErrorCategory","error"],"mappings":";;;;;;;;AA4BO,IAAM,sBAAA,GAAN,cAAqCA,2BAAA,CAAyC;AAAA,EAChE,qBAAA,GAAyC;AAC1D,IAAA,OAAO;AAAA,MACL,GAAGA,2BAAA,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,cAA2BC,mBAAA,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,IAAIC,oBAAA,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,IAAIC,qBAAA,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,IAAIC,iBAAA,CAAY;AAAA,UACpB,EAAA,EAAI,8BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,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,SAASC,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,IAAIH,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,8CAAA;AAAA,QACJ,IAAA,EAAM,sCAAA;AAAA,QACN,QAAQC,iBAAA,CAAY,aAAA;AAAA,QACpB,UAAUC,mBAAA,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,IAAIF,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,2CAAA;AAAA,QACJ,IAAA,EAAM,oBAAoB,MAAM,CAAA,gDAAA,CAAA;AAAA,QAChC,QAAQC,iBAAA,CAAY,aAAA;AAAA,QACpB,UAAUC,mBAAA,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,SAASC,OAAA,EAAY;AAEnB,MAAA,MAAM,OAAA,GAAUA,OAAA,EAAO,OAAA,IAAWA,OAAA,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,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,4BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc;AAAA,SAC1B;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,qCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;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,IAAIH,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,iCAAA;AAAA,QACJ,IAAA,EAAM,gCAAA;AAAA,QACN,QAAQC,iBAAA,CAAY,aAAA;AAAA,QACpB,UAAUC,mBAAA,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,SAASC,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACAC;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,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACAC;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.cjs","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"]}

package/dist/index.d.ts

@@ -1,2 +1,3 @@
-export { CHROMA_PROMPT } from './_tsup-dts-rollup.js';
-export { ChromaVector } from './_tsup-dts-rollup.js';
+export * from './vector/index.js';
+export { CHROMA_PROMPT } from './vector/prompt.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,gBAAgB,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC"}