@mastra/opensearch 0.10.3 → 0.10.4-alpha.1

package/dist/index.js

@@ -1,3 +1,4 @@
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraVector } from '@mastra/core/vector';
 import { Client } from '@opensearch-project/opensearch';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -9,7 +10,6 @@ var OpenSearchFilterTranslator = class extends BaseFilterTranslator {
       ...BaseFilterTranslator.DEFAULT_OPERATORS,
       logical: ["$and", "$or", "$not"],
       array: ["$in", "$nin", "$all"],
-      element: ["$exists"],
       regex: ["$regex"],
       custom: []
     };
@@ -381,7 +381,13 @@ var OpenSearchVector = class extends MastraVector {
    */
   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: "STORAGE_OPENSEARCH_VECTOR_CREATE_INDEX_INVALID_ARGS",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        text: "Dimension must be a positive integer",
+        details: { indexName, dimension }
+      });
     }
     try {
       await this.client.indices.create({
@@ -412,8 +418,15 @@ var OpenSearchVector = class extends MastraVector {
         await this.validateExistingIndex(indexName, dimension, metric);
         return;
       }
-      console.error(`Failed to create index ${indexName}:`, error);
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_CREATE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, dimension, metric }
+        },
+        error
+      );
     }
   }
   /**
@@ -427,8 +440,14 @@ var OpenSearchVector = class extends MastraVector {
       const indexes = response.body.map((record) => record.index).filter((index) => index !== void 0);
       return indexes;
     } catch (error) {
-      console.error("Failed to list indexes:", error);
-      throw new Error(`Failed to list indexes: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_LIST_INDEXES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   /**
@@ -459,7 +478,17 @@ var OpenSearchVector = class extends MastraVector {
     try {
       await this.client.indices.delete({ index: indexName });
     } catch (error) {
-      console.error(`Failed to delete index ${indexName}:`, error);
+      const mastraError = new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_DELETE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+      this.logger?.error(mastraError.toString());
+      this.logger?.trackException(mastraError);
     }
   }
   /**
@@ -474,31 +503,38 @@ var OpenSearchVector = class extends MastraVector {
   async upsert({ indexName, vectors, metadata = [], ids }) {
     const vectorIds = ids || vectors.map(() => crypto.randomUUID());
     const operations = [];
-    const indexInfo = await this.describeIndex({ indexName });
-    this.validateVectorDimensions(vectors, indexInfo.dimension);
-    for (let i = 0; i < vectors.length; i++) {
-      const operation = {
-        index: {
-          _index: indexName,
-          _id: vectorIds[i]
-        }
-      };
-      const document = {
-        id: vectorIds[i],
-        embedding: vectors[i],
-        metadata: metadata[i] || {}
-      };
-      operations.push(operation);
-      operations.push(document);
-    }
     try {
+      const indexInfo = await this.describeIndex({ indexName });
+      this.validateVectorDimensions(vectors, indexInfo.dimension);
+      for (let i = 0; i < vectors.length; i++) {
+        const operation = {
+          index: {
+            _index: indexName,
+            _id: vectorIds[i]
+          }
+        };
+        const document = {
+          id: vectorIds[i],
+          embedding: vectors[i],
+          metadata: metadata[i] || {}
+        };
+        operations.push(operation);
+        operations.push(document);
+      }
       if (operations.length > 0) {
         await this.client.bulk({ body: operations, refresh: true });
       }
       return vectorIds;
     } catch (error) {
-      console.error("Failed to upsert vectors:", error);
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_UPSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, vectorCount: vectors?.length || 0 }
+        },
+        error
+      );
     }
   }
   /**
@@ -543,8 +579,15 @@ var OpenSearchVector = class extends MastraVector {
       });
       return results;
     } catch (error) {
-      console.error("Failed to query vectors:", error);
-      throw new Error(`Failed to query vectors for index ${indexName}: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_QUERY_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, topK }
+        },
+        error
+      );
     }
   }
   /**
@@ -562,7 +605,7 @@ var OpenSearchVector = class extends MastraVector {
   /**
    * Transforms the filter to the OpenSearch DSL.
    *
-   * @param {VectorFilter} filter - The filter to transform.
+   * @param {OpenSearchVectorFilter} filter - The filter to transform.
    * @returns {Record<string, any>} The transformed filter.
    */
   transformFilter(filter) {
@@ -580,35 +623,52 @@ var OpenSearchVector = class extends MastraVector {
    * @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");
-    }
+    let existingDoc;
     try {
-      const { body: existingDoc } = await this.client.get({
+      if (!update.vector && !update.metadata) {
+        throw new Error("No updates provided");
+      }
+      const { body } = await this.client.get({
         index: indexName,
         id
       }).catch(() => {
         throw new Error(`Document with ID ${id} not found in index ${indexName}`);
       });
-      if (!existingDoc || !existingDoc._source) {
+      if (!body || !body._source) {
         throw new Error(`Document with ID ${id} has no source data in index ${indexName}`);
       }
-      const source = existingDoc._source;
-      const updatedDoc = {
-        id: source.id || id
-      };
+      existingDoc = body;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_UPDATE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName, id }
+        },
+        error
+      );
+    }
+    const source = existingDoc._source;
+    const updatedDoc = {
+      id: source?.id || id
+    };
+    try {
       if (update.vector) {
+        console.log(`1`);
         const indexInfo = await this.describeIndex({ indexName });
+        console.log(`2`);
         this.validateVectorDimensions([update.vector], indexInfo.dimension);
         updatedDoc.embedding = update.vector;
-      } else if (source.embedding) {
+      } else if (source?.embedding) {
         updatedDoc.embedding = source.embedding;
       }
       if (update.metadata) {
         updatedDoc.metadata = update.metadata;
       } else {
-        updatedDoc.metadata = source.metadata || {};
+        updatedDoc.metadata = source?.metadata || {};
       }
+      console.log(`3`);
       await this.client.index({
         index: indexName,
         id,
@@ -616,8 +676,15 @@ var OpenSearchVector = class extends MastraVector {
         refresh: true
       });
     } catch (error) {
-      console.error(`Failed to update document with ID ${id} in index ${indexName}:`, error);
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_UPDATE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
   }
   /**
@@ -635,10 +702,18 @@ var OpenSearchVector = class extends MastraVector {
         refresh: true
       });
     } catch (error) {
-      console.error(`Failed to delete document with ID ${id} from index ${indexName}:`, error);
-      if (error && typeof error === "object" && "statusCode" in error && error.statusCode !== 404) {
-        throw error;
+      if (error && typeof error === "object" && "statusCode" in error && error.statusCode === 404) {
+        return;
       }
+      throw new MastraError(
+        {
+          id: "STORAGE_OPENSEARCH_VECTOR_DELETE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
   }
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/opensearch",
-  "version": "0.10.3",
+  "version": "0.10.4-alpha.1",
   "description": "OpenSearch vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -24,12 +24,12 @@
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
     "@types/node": "^20.19.0",
-    "eslint": "^9.28.0",
+    "eslint": "^9.29.0",
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.3",
-    "@mastra/core": "0.10.6",
-    "@internal/lint": "0.0.13"
+    "@internal/lint": "0.0.13",
+    "@mastra/core": "0.10.7-alpha.2"
   },
   "peerDependencies": {
     "@mastra/core": ">=0.10.4-0 <0.11.0"