@mastra/turbopuffer 0.0.15-alpha.1 → 0.0.15-alpha.3

package/dist/_tsup-dts-rollup.d.cts

@@ -64,6 +64,28 @@ declare class TurbopufferVector extends MastraVector {
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;
+    /**
+     * 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.
+     */
+    updateVector(indexName: string, id: string, updates: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    /**
+     * Deletes a vector by its ID.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to delete.
+     * @returns A promise that resolves when the deletion is complete.
+     * @throws Will throw an error if the deletion operation fails.
+     */
+    deleteVector(indexName: string, id: string): Promise<void>;
 }
 export { TurbopufferVector }
 export { TurbopufferVector as TurbopufferVector_alias_1 }

package/dist/_tsup-dts-rollup.d.ts

@@ -64,6 +64,28 @@ declare class TurbopufferVector extends MastraVector {
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;
+    /**
+     * 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.
+     */
+    updateVector(indexName: string, id: string, updates: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    /**
+     * Deletes a vector by its ID.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to delete.
+     * @returns A promise that resolves when the deletion is complete.
+     * @throws Will throw an error if the deletion operation fails.
+     */
+    deleteVector(indexName: string, id: string): Promise<void>;
 }
 export { TurbopufferVector }
 export { TurbopufferVector as TurbopufferVector_alias_1 }

package/dist/index.cjs

@@ -331,6 +331,50 @@ var TurbopufferVector = class extends vector.MastraVector {
       throw new Error(`Failed to delete Turbopuffer namespace ${indexName}: ${error.message}`);
     }
   }
+  /**
+   * 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, updates) {
+    try {
+      const namespace = this.client.namespace(indexName);
+      const createIndex = this.createIndexCache.get(indexName);
+      if (!createIndex) {
+        throw new Error(`createIndex() not called for this index`);
+      }
+      const distanceMetric = createIndex.tpufDistanceMetric;
+      const record = { id };
+      if (updates.vector) record.vector = updates.vector;
+      if (updates.metadata) record.attributes = updates.metadata;
+      await namespace.upsert({
+        vectors: [record],
+        distance_metric: distanceMetric
+      });
+    } catch (error) {
+      throw new Error(`Failed to update Turbopuffer namespace ${indexName}: ${error.message}`);
+    }
+  }
+  /**
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
+  async deleteVector(indexName, id) {
+    try {
+      const namespace = this.client.namespace(indexName);
+      await namespace.delete({ ids: [id] });
+    } catch (error) {
+      throw new Error(`Failed to delete Turbopuffer namespace ${indexName}: ${error.message}`);
+    }
+  }
 };
 
 exports.TurbopufferVector = TurbopufferVector;

package/dist/index.js

@@ -329,6 +329,50 @@ var TurbopufferVector = class extends MastraVector {
       throw new Error(`Failed to delete Turbopuffer namespace ${indexName}: ${error.message}`);
     }
   }
+  /**
+   * 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, updates) {
+    try {
+      const namespace = this.client.namespace(indexName);
+      const createIndex = this.createIndexCache.get(indexName);
+      if (!createIndex) {
+        throw new Error(`createIndex() not called for this index`);
+      }
+      const distanceMetric = createIndex.tpufDistanceMetric;
+      const record = { id };
+      if (updates.vector) record.vector = updates.vector;
+      if (updates.metadata) record.attributes = updates.metadata;
+      await namespace.upsert({
+        vectors: [record],
+        distance_metric: distanceMetric
+      });
+    } catch (error) {
+      throw new Error(`Failed to update Turbopuffer namespace ${indexName}: ${error.message}`);
+    }
+  }
+  /**
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
+  async deleteVector(indexName, id) {
+    try {
+      const namespace = this.client.namespace(indexName);
+      await namespace.delete({ ids: [id] });
+    } catch (error) {
+      throw new Error(`Failed to delete Turbopuffer namespace ${indexName}: ${error.message}`);
+    }
+  }
 };
 
 export { TurbopufferVector };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/turbopuffer",
-  "version": "0.0.15-alpha.1",
+  "version": "0.0.15-alpha.3",
   "description": "Turbopuffer vector store provider for Mastra",
   "type": "module",
   "files": [
@@ -24,7 +24,7 @@
   "license": "MIT",
   "dependencies": {
     "@turbopuffer/turbopuffer": "^0.6.4",
-    "@mastra/core": "^0.9.4-alpha.1"
+    "@mastra/core": "^0.9.4-alpha.3"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.5",