@mastra/chroma 0.0.0-trigger-playground-ui-package-20250506151043 → 0.0.0-tsconfig-compile-20250703214351

package/dist/index.cjs

@@ -1,5 +1,6 @@
 'use strict';
 
+var error = require('@mastra/core/error');
 var vector = require('@mastra/core/vector');
 var chromadb = require('chromadb');
 var filter = require('@mastra/core/vector/filter');
@@ -124,21 +125,32 @@ var ChromaVector = class extends vector.MastraVector {
       }
     }
   }
-  async upsert(...args) {
-    const params = this.normalizeArgs("upsert", args, ["documents"]);
-    const { indexName, vectors, metadata, ids, documents } = params;
-    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,
-      documents
-    });
-    return generatedIds;
+  async upsert({ indexName, vectors, metadata, ids, documents }) {
+    try {
+      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,
+        documents
+      });
+      return generatedIds;
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_UPSERT_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   HnswSpaceMap = {
     cosine: "cosine",
@@ -147,88 +159,215 @@ var ChromaVector = class extends vector.MastraVector {
     l2: "euclidean",
     ip: "dotproduct"
   };
-  async createIndex(...args) {
-    const params = this.normalizeArgs("createIndex", args);
-    const { indexName, dimension, metric = "cosine" } = params;
+  async createIndex({ indexName, dimension, metric = "cosine" }) {
     if (!Number.isInteger(dimension) || dimension <= 0) {
-      throw new Error("Dimension must be a positive integer");
+      throw new error.MastraError({
+        id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_DIMENSION",
+        text: "Dimension must be a positive integer",
+        domain: error.ErrorDomain.MASTRA_VECTOR,
+        category: error.ErrorCategory.USER,
+        details: { dimension }
+      });
     }
     const hnswSpace = this.HnswSpaceMap[metric];
-    if (!["cosine", "l2", "ip"].includes(hnswSpace)) {
-      throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
+    if (!hnswSpace || !["cosine", "l2", "ip"].includes(hnswSpace)) {
+      throw new error.MastraError({
+        id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC",
+        text: `Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`,
+        domain: error.ErrorDomain.MASTRA_VECTOR,
+        category: error.ErrorCategory.USER,
+        details: { metric }
+      });
     }
-    await this.client.createCollection({
-      name: indexName,
-      metadata: {
-        dimension,
-        "hnsw:space": this.HnswSpaceMap[metric]
+    try {
+      await this.client.createCollection({
+        name: indexName,
+        metadata: {
+          dimension,
+          "hnsw:space": hnswSpace
+        }
+      });
+    } catch (error$1) {
+      const message = error$1?.message || error$1?.toString();
+      if (message && message.toLowerCase().includes("already exists")) {
+        await this.validateExistingIndex(indexName, dimension, metric);
+        return;
       }
-    });
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_CREATE_INDEX_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   transformFilter(filter) {
     const translator = new ChromaFilterTranslator();
     return translator.translate(filter);
   }
-  async query(...args) {
-    const params = this.normalizeArgs("query", args, ["documentFilter"]);
-    const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
-    const collection = await this.getCollection(indexName, true);
-    const defaultInclude = ["documents", "metadatas", "distances"];
-    const translatedFilter = this.transformFilter(filter);
-    const results = await collection.query({
-      queryEmbeddings: [queryVector],
-      nResults: topK,
-      where: translatedFilter,
-      whereDocument: documentFilter,
-      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],
-      ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
-    }));
+  async query({
+    indexName,
+    queryVector,
+    topK = 10,
+    filter,
+    includeVector = false,
+    documentFilter
+  }) {
+    try {
+      const collection = await this.getCollection(indexName, true);
+      const defaultInclude = ["documents", "metadatas", "distances"];
+      const translatedFilter = this.transformFilter(filter);
+      const results = await collection.query({
+        queryEmbeddings: [queryVector],
+        nResults: topK,
+        where: translatedFilter,
+        whereDocument: documentFilter,
+        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],
+        ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
+      }));
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_QUERY_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   async listIndexes() {
-    const collections = await this.client.listCollections();
-    return collections.map((collection) => collection);
+    try {
+      const collections = await this.client.listCollections();
+      return collections.map((collection) => collection);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_LIST_INDEXES_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
   }
-  async describeIndex(indexName) {
-    const collection = await this.getCollection(indexName);
-    const count = await collection.count();
-    const metadata = collection.metadata;
-    const hnswSpace = metadata?.["hnsw:space"];
-    return {
-      dimension: metadata?.dimension || 0,
-      count,
-      metric: this.HnswSpaceMap[hnswSpace]
-    };
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param {string} indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex({ indexName }) {
+    try {
+      const collection = await this.getCollection(indexName);
+      const count = await collection.count();
+      const metadata = collection.metadata;
+      const hnswSpace = metadata?.["hnsw:space"];
+      return {
+        dimension: metadata?.dimension || 0,
+        count,
+        metric: this.HnswSpaceMap[hnswSpace]
+      };
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_DESCRIBE_INDEX_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
-  async deleteIndex(indexName) {
-    await this.client.deleteCollection({ name: indexName });
-    this.collections.delete(indexName);
+  async deleteIndex({ indexName }) {
+    try {
+      await this.client.deleteCollection({ name: indexName });
+      this.collections.delete(indexName);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_DELETE_INDEX_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
-  async updateIndexById(indexName, id, update) {
+  /**
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
+  async updateVector({ indexName, id, update }) {
     if (!update.vector && !update.metadata) {
-      throw new Error("No updates provided");
-    }
-    const collection = await this.getCollection(indexName, true);
-    const updateOptions = { ids: [id] };
-    if (update?.vector) {
-      updateOptions.embeddings = [update.vector];
+      throw new error.MastraError({
+        id: "CHROMA_VECTOR_UPDATE_NO_PAYLOAD",
+        text: "No updates provided for vector",
+        domain: error.ErrorDomain.MASTRA_VECTOR,
+        category: error.ErrorCategory.USER,
+        details: { indexName, id }
+      });
     }
-    if (update?.metadata) {
-      updateOptions.metadatas = [update.metadata];
+    try {
+      const collection = await this.getCollection(indexName, true);
+      const updateOptions = { ids: [id] };
+      if (update?.vector) {
+        const stats = await this.describeIndex({ indexName });
+        this.validateVectorDimensions([update.vector], stats.dimension);
+        updateOptions.embeddings = [update.vector];
+      }
+      if (update?.metadata) {
+        updateOptions.metadatas = [update.metadata];
+      }
+      return await collection.update(updateOptions);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_UPDATE_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error$1
+      );
     }
-    return await collection.update(updateOptions);
   }
-  async deleteIndexById(indexName, id) {
+  async deleteVector({ indexName, id }) {
     try {
       const collection = await this.getCollection(indexName, true);
       await collection.delete({ ids: [id] });
-    } catch (error) {
-      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "CHROMA_VECTOR_DELETE_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error$1
+      );
     }
   }
 };

package/dist/index.js

@@ -1,3 +1,4 @@
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraVector } from '@mastra/core/vector';
 import { ChromaClient } from 'chromadb';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -122,21 +123,32 @@ var ChromaVector = class extends MastraVector {
       }
     }
   }
-  async upsert(...args) {
-    const params = this.normalizeArgs("upsert", args, ["documents"]);
-    const { indexName, vectors, metadata, ids, documents } = params;
-    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,
-      documents
-    });
-    return generatedIds;
+  async upsert({ indexName, vectors, metadata, ids, documents }) {
+    try {
+      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,
+        documents
+      });
+      return generatedIds;
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_UPSERT_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
   HnswSpaceMap = {
     cosine: "cosine",
@@ -145,88 +157,215 @@ var ChromaVector = class extends MastraVector {
     l2: "euclidean",
     ip: "dotproduct"
   };
-  async createIndex(...args) {
-    const params = this.normalizeArgs("createIndex", args);
-    const { indexName, dimension, metric = "cosine" } = params;
+  async createIndex({ indexName, dimension, metric = "cosine" }) {
     if (!Number.isInteger(dimension) || dimension <= 0) {
-      throw new Error("Dimension must be a positive integer");
+      throw new MastraError({
+        id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_DIMENSION",
+        text: "Dimension must be a positive integer",
+        domain: ErrorDomain.MASTRA_VECTOR,
+        category: ErrorCategory.USER,
+        details: { dimension }
+      });
     }
     const hnswSpace = this.HnswSpaceMap[metric];
-    if (!["cosine", "l2", "ip"].includes(hnswSpace)) {
-      throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
+    if (!hnswSpace || !["cosine", "l2", "ip"].includes(hnswSpace)) {
+      throw new MastraError({
+        id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC",
+        text: `Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`,
+        domain: ErrorDomain.MASTRA_VECTOR,
+        category: ErrorCategory.USER,
+        details: { metric }
+      });
     }
-    await this.client.createCollection({
-      name: indexName,
-      metadata: {
-        dimension,
-        "hnsw:space": this.HnswSpaceMap[metric]
+    try {
+      await this.client.createCollection({
+        name: indexName,
+        metadata: {
+          dimension,
+          "hnsw:space": hnswSpace
+        }
+      });
+    } catch (error) {
+      const message = error?.message || error?.toString();
+      if (message && message.toLowerCase().includes("already exists")) {
+        await this.validateExistingIndex(indexName, dimension, metric);
+        return;
       }
-    });
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_CREATE_INDEX_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
   transformFilter(filter) {
     const translator = new ChromaFilterTranslator();
     return translator.translate(filter);
   }
-  async query(...args) {
-    const params = this.normalizeArgs("query", args, ["documentFilter"]);
-    const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
-    const collection = await this.getCollection(indexName, true);
-    const defaultInclude = ["documents", "metadatas", "distances"];
-    const translatedFilter = this.transformFilter(filter);
-    const results = await collection.query({
-      queryEmbeddings: [queryVector],
-      nResults: topK,
-      where: translatedFilter,
-      whereDocument: documentFilter,
-      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],
-      ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
-    }));
+  async query({
+    indexName,
+    queryVector,
+    topK = 10,
+    filter,
+    includeVector = false,
+    documentFilter
+  }) {
+    try {
+      const collection = await this.getCollection(indexName, true);
+      const defaultInclude = ["documents", "metadatas", "distances"];
+      const translatedFilter = this.transformFilter(filter);
+      const results = await collection.query({
+        queryEmbeddings: [queryVector],
+        nResults: topK,
+        where: translatedFilter,
+        whereDocument: documentFilter,
+        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],
+        ...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
+      }));
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_QUERY_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
   async listIndexes() {
-    const collections = await this.client.listCollections();
-    return collections.map((collection) => collection);
+    try {
+      const collections = await this.client.listCollections();
+      return collections.map((collection) => collection);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_LIST_INDEXES_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
   }
-  async describeIndex(indexName) {
-    const collection = await this.getCollection(indexName);
-    const count = await collection.count();
-    const metadata = collection.metadata;
-    const hnswSpace = metadata?.["hnsw:space"];
-    return {
-      dimension: metadata?.dimension || 0,
-      count,
-      metric: this.HnswSpaceMap[hnswSpace]
-    };
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param {string} indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex({ indexName }) {
+    try {
+      const collection = await this.getCollection(indexName);
+      const count = await collection.count();
+      const metadata = collection.metadata;
+      const hnswSpace = metadata?.["hnsw:space"];
+      return {
+        dimension: metadata?.dimension || 0,
+        count,
+        metric: this.HnswSpaceMap[hnswSpace]
+      };
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_DESCRIBE_INDEX_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
-  async deleteIndex(indexName) {
-    await this.client.deleteCollection({ name: indexName });
-    this.collections.delete(indexName);
+  async deleteIndex({ indexName }) {
+    try {
+      await this.client.deleteCollection({ name: indexName });
+      this.collections.delete(indexName);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_DELETE_INDEX_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
-  async updateIndexById(indexName, id, update) {
+  /**
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
+  async updateVector({ indexName, id, update }) {
     if (!update.vector && !update.metadata) {
-      throw new Error("No updates provided");
+      throw new MastraError({
+        id: "CHROMA_VECTOR_UPDATE_NO_PAYLOAD",
+        text: "No updates provided for vector",
+        domain: ErrorDomain.MASTRA_VECTOR,
+        category: ErrorCategory.USER,
+        details: { indexName, id }
+      });
     }
-    const collection = await this.getCollection(indexName, true);
-    const updateOptions = { ids: [id] };
-    if (update?.vector) {
-      updateOptions.embeddings = [update.vector];
-    }
-    if (update?.metadata) {
-      updateOptions.metadatas = [update.metadata];
+    try {
+      const collection = await this.getCollection(indexName, true);
+      const updateOptions = { ids: [id] };
+      if (update?.vector) {
+        const stats = await this.describeIndex({ indexName });
+        this.validateVectorDimensions([update.vector], stats.dimension);
+        updateOptions.embeddings = [update.vector];
+      }
+      if (update?.metadata) {
+        updateOptions.metadatas = [update.metadata];
+      }
+      return await collection.update(updateOptions);
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_UPDATE_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
-    return await collection.update(updateOptions);
   }
-  async deleteIndexById(indexName, id) {
+  async deleteVector({ indexName, id }) {
     try {
       const collection = await this.getCollection(indexName, true);
       await collection.delete({ ids: [id] });
     } catch (error) {
-      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: "CHROMA_VECTOR_DELETE_FAILED",
+          domain: ErrorDomain.MASTRA_VECTOR,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
   }
 };