@mastra/turbopuffer 0.0.0-switch-to-core-20250424015131 → 0.0.0-vector-query-sources-20250516172905

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

@@ -1,12 +1,17 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import type { CreateIndexParams } from '@mastra/core/vector';
+import type { DeleteIndexParams } from '@mastra/core/vector';
+import type { DeleteVectorParams } from '@mastra/core/vector';
+import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { Filters } from '@turbopuffer/turbopuffer';
 import type { IndexStats } from '@mastra/core/vector';
 import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
+import type { ParamsToArgs } from '@mastra/core/vector';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
 import type { Schema } from '@turbopuffer/turbopuffer';
+import type { UpdateVectorParams } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 
@@ -62,8 +67,34 @@ declare class TurbopufferVector extends MastraVector {
     upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
     query({ indexName, queryVector, topK, filter, includeVector }: QueryVectorParams): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
-    describeIndex(indexName: string): Promise<IndexStats>;
-    deleteIndex(indexName: string): Promise<void>;
+    /**
+     * Retrieves statistics about a vector index.
+     *
+     * @param params - The parameters for describing an index
+     * @param params.indexName - The name of the index to describe
+     * @returns A promise that resolves to the index statistics including dimension, count and metric
+     */
+    describeIndex(...args: ParamsToArgs<DescribeIndexParams>): Promise<IndexStats>;
+    deleteIndex(...args: ParamsToArgs<DeleteIndexParams>): 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(...args: ParamsToArgs<UpdateVectorParams>): 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(...args: ParamsToArgs<DeleteVectorParams>): Promise<void>;
 }
 export { TurbopufferVector }
 export { TurbopufferVector as TurbopufferVector_alias_1 }

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

@@ -1,12 +1,17 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import type { CreateIndexParams } from '@mastra/core/vector';
+import type { DeleteIndexParams } from '@mastra/core/vector';
+import type { DeleteVectorParams } from '@mastra/core/vector';
+import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { Filters } from '@turbopuffer/turbopuffer';
 import type { IndexStats } from '@mastra/core/vector';
 import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
+import type { ParamsToArgs } from '@mastra/core/vector';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
 import type { Schema } from '@turbopuffer/turbopuffer';
+import type { UpdateVectorParams } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 
@@ -62,8 +67,34 @@ declare class TurbopufferVector extends MastraVector {
     upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
     query({ indexName, queryVector, topK, filter, includeVector }: QueryVectorParams): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
-    describeIndex(indexName: string): Promise<IndexStats>;
-    deleteIndex(indexName: string): Promise<void>;
+    /**
+     * Retrieves statistics about a vector index.
+     *
+     * @param params - The parameters for describing an index
+     * @param params.indexName - The name of the index to describe
+     * @returns A promise that resolves to the index statistics including dimension, count and metric
+     */
+    describeIndex(...args: ParamsToArgs<DescribeIndexParams>): Promise<IndexStats>;
+    deleteIndex(...args: ParamsToArgs<DeleteIndexParams>): 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(...args: ParamsToArgs<UpdateVectorParams>): 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(...args: ParamsToArgs<DeleteVectorParams>): Promise<void>;
 }
 export { TurbopufferVector }
 export { TurbopufferVector as TurbopufferVector_alias_1 }

package/dist/index.cjs

@@ -303,7 +303,16 @@ var TurbopufferVector = class extends vector.MastraVector {
       throw new Error(`Failed to list Turbopuffer namespaces: ${error}`);
     }
   }
-  async describeIndex(indexName) {
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param params - The parameters for describing an index
+   * @param params.indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex(...args) {
+    const params = this.normalizeArgs("describeIndex", args);
+    const { indexName } = params;
     try {
       const namespace = this.client.namespace(indexName);
       const metadata = await namespace.metadata();
@@ -322,7 +331,9 @@ var TurbopufferVector = class extends vector.MastraVector {
       throw new Error(`Failed to describe Turbopuffer namespace ${indexName}: ${error}`);
     }
   }
-  async deleteIndex(indexName) {
+  async deleteIndex(...args) {
+    const params = this.normalizeArgs("deleteIndex", args);
+    const { indexName } = params;
     try {
       const namespace = this.client.namespace(indexName);
       await namespace.deleteAll();
@@ -331,6 +342,54 @@ 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(...args) {
+    const params = this.normalizeArgs("updateVector", args);
+    const { indexName, id, update } = params;
+    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 (update.vector) record.vector = update.vector;
+      if (update.metadata) record.attributes = update.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(...args) {
+    const params = this.normalizeArgs("deleteVector", args);
+    const { indexName, id } = params;
+    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

@@ -301,7 +301,16 @@ var TurbopufferVector = class extends MastraVector {
       throw new Error(`Failed to list Turbopuffer namespaces: ${error}`);
     }
   }
-  async describeIndex(indexName) {
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param params - The parameters for describing an index
+   * @param params.indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex(...args) {
+    const params = this.normalizeArgs("describeIndex", args);
+    const { indexName } = params;
     try {
       const namespace = this.client.namespace(indexName);
       const metadata = await namespace.metadata();
@@ -320,7 +329,9 @@ var TurbopufferVector = class extends MastraVector {
       throw new Error(`Failed to describe Turbopuffer namespace ${indexName}: ${error}`);
     }
   }
-  async deleteIndex(indexName) {
+  async deleteIndex(...args) {
+    const params = this.normalizeArgs("deleteIndex", args);
+    const { indexName } = params;
     try {
       const namespace = this.client.namespace(indexName);
       await namespace.deleteAll();
@@ -329,6 +340,54 @@ 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(...args) {
+    const params = this.normalizeArgs("updateVector", args);
+    const { indexName, id, update } = params;
+    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 (update.vector) record.vector = update.vector;
+      if (update.metadata) record.attributes = update.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(...args) {
+    const params = this.normalizeArgs("deleteVector", args);
+    const { indexName, id } = params;
+    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.0-switch-to-core-20250424015131",
+  "version": "0.0.0-vector-query-sources-20250516172905",
   "description": "Turbopuffer vector store provider for Mastra",
   "type": "module",
   "files": [
@@ -24,17 +24,17 @@
   "license": "MIT",
   "dependencies": {
     "@turbopuffer/turbopuffer": "^0.6.4",
-    "@mastra/core": "0.0.0-switch-to-core-20250424015131"
+    "@mastra/core": "0.0.0-vector-query-sources-20250516172905"
   },
   "devDependencies": {
-    "@microsoft/api-extractor": "^7.52.1",
+    "@microsoft/api-extractor": "^7.52.5",
     "@types/node": "^20.17.27",
     "dotenv": "^16.4.7",
     "eslint": "^9.23.0",
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
-    "vitest": "^3.0.9",
-    "@internal/lint": "0.0.2"
+    "vitest": "^3.1.2",
+    "@internal/lint": "0.0.0-vector-query-sources-20250516172905"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",