@mastra/turbopuffer 0.0.0-vector-sources-20250516175436 → 0.0.0-vector-extension-schema-20250922130418

package/dist/index.js

@@ -1,3 +1,4 @@
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraVector } from '@mastra/core/vector';
 import { Turbopuffer } from '@turbopuffer/turbopuffer';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -186,28 +187,40 @@ var TurbopufferVector = class extends MastraVector {
   }
   async createIndex({ indexName, dimension, metric }) {
     metric = metric ?? "cosine";
-    if (this.createIndexCache.has(indexName)) {
-      const expected = this.createIndexCache.get(indexName);
-      if (dimension !== expected.dimension || metric !== expected.metric) {
-        throw new Error(
-          `createIndex() called more than once with inconsistent inputs. Index ${indexName} expected dimensions=${expected.dimension} and metric=${expected.metric} but got dimensions=${dimension} and metric=${metric}`
-        );
-      }
-      return;
-    }
-    if (dimension <= 0) {
-      throw new Error("Dimension must be a positive integer");
-    }
     let distanceMetric = "cosine_distance";
-    switch (metric) {
-      case "cosine":
-        distanceMetric = "cosine_distance";
-        break;
-      case "euclidean":
-        distanceMetric = "euclidean_squared";
-        break;
-      case "dotproduct":
-        throw new Error("dotproduct is not supported in Turbopuffer");
+    try {
+      if (this.createIndexCache.has(indexName)) {
+        const expected = this.createIndexCache.get(indexName);
+        if (dimension !== expected.dimension || metric !== expected.metric) {
+          throw new Error(
+            `createIndex() called more than once with inconsistent inputs. Index ${indexName} expected dimensions=${expected.dimension} and metric=${expected.metric} but got dimensions=${dimension} and metric=${metric}`
+          );
+        }
+        return;
+      }
+      if (dimension <= 0) {
+        throw new Error("Dimension must be a positive integer");
+      }
+      switch (metric) {
+        case "cosine":
+          distanceMetric = "cosine_distance";
+          break;
+        case "euclidean":
+          distanceMetric = "euclidean_squared";
+          break;
+        case "dotproduct":
+          throw new Error("dotproduct is not supported in Turbopuffer");
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_CREATE_INDEX_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName, dimension, metric }
+        },
+        error
+      );
     }
     this.createIndexCache.set(indexName, {
       indexName,
@@ -217,15 +230,29 @@ var TurbopufferVector = class extends MastraVector {
     });
   }
   async upsert({ indexName, vectors, metadata, ids }) {
+    let index;
+    let createIndex;
     try {
       if (vectors.length === 0) {
         throw new Error("upsert() called with empty vectors");
       }
-      const index = this.client.namespace(indexName);
-      const createIndex = this.createIndexCache.get(indexName);
+      index = this.client.namespace(indexName);
+      createIndex = this.createIndexCache.get(indexName);
       if (!createIndex) {
         throw new Error(`createIndex() not called for this index`);
       }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_UPSERT_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName }
+        },
+        error
+      );
+    }
+    try {
       const distanceMetric = createIndex.tpufDistanceMetric;
       const vectorIds = ids || vectors.map(() => crypto.randomUUID());
       const records = vectors.map((vector, i) => ({
@@ -253,21 +280,48 @@ var TurbopufferVector = class extends MastraVector {
       }
       return vectorIds;
     } catch (error) {
-      throw new Error(`Failed to upsert vectors into Turbopuffer namespace ${indexName}: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_UPSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
-  async query({ indexName, queryVector, topK, filter, includeVector }) {
-    const schemaConfig = this.opts.schemaConfigForIndex?.(indexName);
-    if (schemaConfig) {
-      if (queryVector.length !== schemaConfig.dimensions) {
-        throw new Error(
-          `Turbopuffer index ${indexName} was configured with dimensions=${schemaConfig.dimensions} but attempting to query with queryVector.length=${queryVector.length}`
-        );
+  async query({
+    indexName,
+    queryVector,
+    topK,
+    filter,
+    includeVector
+  }) {
+    let createIndex;
+    try {
+      const schemaConfig = this.opts.schemaConfigForIndex?.(indexName);
+      if (schemaConfig) {
+        if (queryVector.length !== schemaConfig.dimensions) {
+          throw new Error(
+            `Turbopuffer index ${indexName} was configured with dimensions=${schemaConfig.dimensions} but attempting to query with queryVector.length=${queryVector.length}`
+          );
+        }
       }
-    }
-    const createIndex = this.createIndexCache.get(indexName);
-    if (!createIndex) {
-      throw new Error(`createIndex() not called for this index`);
+      createIndex = this.createIndexCache.get(indexName);
+      if (!createIndex) {
+        throw new Error(`createIndex() not called for this index`);
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_QUERY_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName }
+        },
+        error
+      );
     }
     const distanceMetric = createIndex.tpufDistanceMetric;
     try {
@@ -290,7 +344,15 @@ var TurbopufferVector = class extends MastraVector {
         ...includeVector && item.vector ? { vector: item.vector } : {}
       }));
     } catch (error) {
-      throw new Error(`Failed to query Turbopuffer namespace ${indexName}: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_QUERY_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
   async listIndexes() {
@@ -298,19 +360,23 @@ var TurbopufferVector = class extends MastraVector {
       const namespacesResult = await this.client.namespaces({});
       return namespacesResult.namespaces.map((namespace) => namespace.id);
     } catch (error) {
-      throw new Error(`Failed to list Turbopuffer namespaces: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_LIST_INDEXES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
     }
   }
   /**
    * Retrieves statistics about a vector index.
    *
-   * @param params - The parameters for describing an index
-   * @param params.indexName - The name of the index to describe
+   * @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(...args) {
-    const params = this.normalizeArgs("describeIndex", args);
-    const { indexName } = params;
+  async describeIndex({ indexName }) {
     try {
       const namespace = this.client.namespace(indexName);
       const metadata = await namespace.metadata();
@@ -326,18 +392,32 @@ var TurbopufferVector = class extends MastraVector {
         metric: createIndex.metric
       };
     } catch (error) {
-      throw new Error(`Failed to describe Turbopuffer namespace ${indexName}: ${error}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_DESCRIBE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
-  async deleteIndex(...args) {
-    const params = this.normalizeArgs("deleteIndex", args);
-    const { indexName } = params;
+  async deleteIndex({ indexName }) {
     try {
       const namespace = this.client.namespace(indexName);
       await namespace.deleteAll();
       this.createIndexCache.delete(indexName);
     } catch (error) {
-      throw new Error(`Failed to delete Turbopuffer namespace ${indexName}: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_DELETE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
   /**
@@ -350,25 +430,47 @@ var TurbopufferVector = class extends MastraVector {
    * @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;
+  async updateVector({ indexName, id, update }) {
+    let namespace;
+    let createIndex;
+    let distanceMetric;
+    let record;
     try {
-      const namespace = this.client.namespace(indexName);
-      const createIndex = this.createIndexCache.get(indexName);
+      namespace = this.client.namespace(indexName);
+      createIndex = this.createIndexCache.get(indexName);
       if (!createIndex) {
         throw new Error(`createIndex() not called for this index`);
       }
-      const distanceMetric = createIndex.tpufDistanceMetric;
-      const record = { id };
+      distanceMetric = createIndex.tpufDistanceMetric;
+      record = { id };
       if (update.vector) record.vector = update.vector;
       if (update.metadata) record.attributes = update.metadata;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_UPDATE_VECTOR_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName }
+        },
+        error
+      );
+    }
+    try {
       await namespace.upsert({
         vectors: [record],
         distance_metric: distanceMetric
       });
     } catch (error) {
-      throw new Error(`Failed to update Turbopuffer namespace ${indexName}: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_UPDATE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
   /**
@@ -378,16 +480,24 @@ var TurbopufferVector = class extends MastraVector {
    * @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;
+  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}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_TURBOBUFFER_VECTOR_DELETE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
     }
   }
 };
 
 export { TurbopufferVector };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/vector/filter.ts","../src/vector/index.ts"],"names":["op"],"mappings":";;;;;;AA6BO,IAAM,2BAAA,GAAN,cAA0C,oBAAA,CAAmE;AAAA,EAC/F,qBAAA,GAAyC;AAC1D,IAAA,OAAO;AAAA,MACL,GAAG,oBAAA,CAAqB,iBAAA;AAAA,MACxB,OAAA,EAAS,CAAC,MAAA,EAAQ,KAAK,CAAA;AAAA,MACvB,KAAA,EAAO,CAAC,KAAA,EAAO,MAAA,EAAQ,MAAM,CAAA;AAAA,MAC7B,OAAA,EAAS,CAAC,SAAS,CAAA;AAAA,MACnB,OAAO,EAAC;AAAA;AAAA,MACR,QAAQ;AAAC;AAAA,KACX;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,WAAA,GAA8C;AAAA,IACpD,GAAA,EAAK,IAAA;AAAA,IACL,GAAA,EAAK,OAAA;AAAA,IACL,GAAA,EAAK,IAAA;AAAA,IACL,IAAA,EAAM,KAAA;AAAA,IACN,GAAA,EAAK,IAAA;AAAA,IACL,IAAA,EAAM,KAAA;AAAA,IACN,GAAA,EAAK,IAAA;AAAA,IACL,IAAA,EAAM;AAAA,GACR;AAAA;AAAA;AAAA;AAAA,EAKA,UAAU,MAAA,EAAuD;AAC/D,IAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,MAAM,CAAA,EAAG;AACxB,MAAA,OAAO,MAAA;AAAA,IACT;AAGA,IAAA,IAAA,CAAK,eAAe,MAAM,CAAA;AAG1B,IAAA,MAAM,MAAA,GAAS,IAAA,CAAK,aAAA,CAAc,MAAM,CAAA;AAIxC,IAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,MAAM,KAAK,MAAA,CAAO,MAAA,KAAW,CAAA,IAAM,MAAA,CAAO,CAAC,CAAA,KAAM,KAAA,IAAS,MAAA,CAAO,CAAC,MAAM,IAAA,EAAO;AAChG,MAAA,OAAO,CAAC,KAAA,EAAO,CAAC,MAAyB,CAAC,CAAA;AAAA,IAC5C;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,cAAc,IAAA,EAA0D;AAE9E,IAAA,IAAI,IAAA,KAAS,QAAQ,IAAA,KAAS,MAAA,IAAa,OAAO,IAAA,CAAK,IAAI,CAAA,CAAE,MAAA,KAAW,CAAA,EAAG;AACzE,MAAA,OAAO,CAAC,KAAA,EAAO,EAAE,CAAA;AAAA,IACnB;AAGA,IAAA,IAAI,IAAA,CAAK,WAAA,CAAY,IAAI,CAAA,EAAG;AAC1B,MAAA,MAAM,IAAI,MAAM,mEAAmE,CAAA;AAAA,IACrF;AAGA,IAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,EAAG;AACvB,MAAA,MAAM,IAAI,MAAM,+DAA+D,CAAA;AAAA,IACjF;AAEA,IAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,IAAI,CAAA;AAGnC,IAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,MAAA,OAAO,CAAC,KAAA,EAAO,EAAE,CAAA;AAAA,IACnB;AAEA,IAAA,MAAM,CAAC,GAAA,EAAK,KAAK,CAAA,GAAI,QAAQ,CAAC,CAAA;AAG9B,IAAA,IAAI,GAAA,IAAO,IAAA,CAAK,iBAAA,CAAkB,GAAG,CAAA,EAAG;AACtC,MAAA,OAAO,IAAA,CAAK,gBAAA,CAAiB,GAAA,EAAK,KAAK,CAAA;AAAA,IACzC;AAGA,IAAA,IAAI,OAAA,CAAQ,SAAS,CAAA,EAAG;AACtB,MAAA,MAAM,UAAA,GAAa,OAAA,CAAQ,GAAA,CAAI,CAAC,CAAC,KAAA,EAAO,UAAU,CAAA,KAAM,IAAA,CAAK,uBAAA,CAAwB,KAAA,EAAO,UAAU,CAAC,CAAA;AACvG,MAAA,OAAO,CAAC,OAAO,UAAU,CAAA;AAAA,IAC3B;AAGA,IAAA,OAAO,IAAA,CAAK,uBAAA,CAAwB,GAAA,EAAK,KAAK,CAAA;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA,EAKQ,uBAAA,CAAwB,OAAe,KAAA,EAA6B;AAE1E,IAAA,IAAI,iBAAiB,IAAA,EAAM;AACzB,MAAA,OAAO,CAAC,KAAA,EAAO,IAAA,EAAM,IAAA,CAAK,cAAA,CAAe,KAAK,CAAC,CAAA;AAAA,IACjD;AAGA,IAAA,IAAI,IAAA,CAAK,WAAA,CAAY,KAAK,CAAA,EAAG;AAC3B,MAAA,OAAO,CAAC,KAAA,EAAO,IAAA,EAAM,IAAA,CAAK,cAAA,CAAe,KAAK,CAAC,CAAA;AAAA,IACjD;AAGA,IAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AACxB,MAAA,OAAO,CAAC,KAAA,EAAO,IAAA,EAAM,IAAA,CAAK,oBAAA,CAAqB,KAAK,CAAC,CAAA;AAAA,IACvD;AAGA,IAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAC/C,MAAA,MAAM,SAAA,GAAY,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA;AAGnC,MAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AAExB,QAAA,MAAM,YAAA,GAAe,UAAU,KAAA,CAAM,CAAAA,QAAM,IAAA,CAAK,UAAA,CAAWA,GAAE,CAAC,CAAA;AAC9D,QAAA,IAAI,YAAA,EAAc;AAEhB,UAAA,MAAM,UAAA,GAAa,SAAA,CAAU,GAAA,CAAI,CAAAA,GAAAA,KAAM,IAAA,CAAK,iBAAA,CAAkB,KAAA,EAAOA,GAAAA,EAAI,KAAA,CAAMA,GAAE,CAAC,CAAC,CAAA;AACnF,UAAA,OAAO,CAAC,OAAO,UAAU,CAAA;AAAA,QAC3B,CAAA,MAAO;AAEL,UAAA,MAAM,UAAA,GAAa,SAAA,CAAU,GAAA,CAAI,CAAAA,GAAAA,KAAM;AACrC,YAAA,MAAM,WAAA,GAAc,CAAA,EAAG,KAAK,CAAA,CAAA,EAAIA,GAAE,CAAA,CAAA;AAClC,YAAA,OAAO,IAAA,CAAK,uBAAA,CAAwB,WAAA,EAAa,KAAA,CAAMA,GAAE,CAAC,CAAA;AAAA,UAC5D,CAAC,CAAA;AACD,UAAA,OAAO,CAAC,OAAO,UAAU,CAAA;AAAA,QAC3B;AAAA,MACF;AAGA,MAAA,MAAM,EAAA,GAAK,UAAU,CAAC,CAAA;AACtB,MAAA,IAAI,EAAA,IAAM,IAAA,CAAK,UAAA,CAAW,EAAE,CAAA,EAAG;AAC7B,QAAA,OAAO,KAAK,iBAAA,CAAkB,KAAA,EAAO,EAAA,EAAI,KAAA,CAAM,EAAE,CAAC,CAAA;AAAA,MACpD;AAGA,MAAA,IAAI,EAAA,IAAM,CAAC,IAAA,CAAK,UAAA,CAAW,EAAE,CAAA,EAAG;AAC9B,QAAA,MAAM,WAAA,GAAc,CAAA,EAAG,KAAK,CAAA,CAAA,EAAI,EAAE,CAAA,CAAA;AAClC,QAAA,OAAO,IAAA,CAAK,uBAAA,CAAwB,WAAA,EAAa,KAAA,CAAM,EAAE,CAAC,CAAA;AAAA,MAC5D;AAAA,IACF;AAEA,IAAA,MAAM,IAAI,KAAA,CAAM,CAAA,qCAAA,EAAwC,KAAK,CAAA,CAAE,CAAA;AAAA,EACjE;AAAA;AAAA;AAAA;AAAA,EAKQ,gBAAA,CAAiB,UAAkB,UAAA,EAA4B;AAErE,IAAA,MAAM,SAAA,GAA8B,QAAA,KAAa,MAAA,GAAS,KAAA,GAAQ,IAAA;AAGlE,IAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC9B,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,iBAAA,EAAoB,QAAQ,CAAA,gCAAA,CAAkC,CAAA;AAAA,IAChF;AAGA,IAAA,MAAM,oBAAA,GAAuB,UAAA,CAAW,GAAA,CAAI,CAAA,SAAA,KAAa;AACvD,MAAA,IAAI,OAAO,SAAA,KAAc,QAAA,IAAY,SAAA,KAAc,IAAA,EAAM;AACvD,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,uCAAA,EAA0C,QAAQ,CAAA,CAAE,CAAA;AAAA,MACtE;AACA,MAAA,OAAO,IAAA,CAAK,cAAc,SAAS,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,OAAO,CAAC,WAAW,oBAAoB,CAAA;AAAA,EACzC;AAAA;AAAA;AAAA;AAAA,EAKQ,iBAAA,CAAkB,KAAA,EAAe,QAAA,EAAkB,KAAA,EAA6B;AAEtF,IAAA,IAAI,QAAA,IAAY,IAAA,CAAK,WAAA,CAAY,QAAQ,CAAA,EAAG;AAC1C,MAAA,OAAO,CAAC,OAAO,IAAA,CAAK,WAAA,CAAY,QAAQ,CAAA,EAAG,IAAA,CAAK,cAAA,CAAe,KAAK,CAAC,CAAA;AAAA,IACvE;AAGA,IAAA,QAAQ,QAAA;AAAU,MAChB,KAAK,SAAA;AAGH,QAAA,OAAO,KAAA,GAAQ,CAAC,KAAA,EAAO,OAAA,EAAS,IAAI,CAAA,GAAI,CAAC,KAAA,EAAO,IAAA,EAAM,IAAI,CAAA;AAAA,MAE5D,KAAK,MAAA;AAEH,QAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,KAAA,CAAM,WAAW,CAAA,EAAG;AAC/C,UAAA,MAAM,IAAI,MAAM,0CAA0C,CAAA;AAAA,QAC5D;AAEA,QAAA,MAAM,aAAA,GAAgB,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ,CAAC,KAAA,EAAO,IAAA,EAAM,CAAC,IAAA,CAAK,cAAA,CAAe,IAAI,CAAC,CAAC,CAAoB,CAAA;AAGrG,QAAA,OAAO,CAAC,OAAO,aAAa,CAAA;AAAA,MAE9B;AACE,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,sBAAA,EAAyB,QAAA,IAAY,WAAW,CAAA,CAAE,CAAA;AAAA;AACtE,EACF;AAAA;AAAA;AAAA;AAAA,EAKU,eAAe,KAAA,EAAiB;AAExC,IAAA,IAAI,iBAAiB,IAAA,EAAM;AACzB,MAAA,OAAO,MAAM,WAAA,EAAY;AAAA,IAC3B;AACA,IAAA,OAAO,KAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKU,qBAAqB,MAAA,EAAsB;AACnD,IAAA,OAAO,OAAO,GAAA,CAAI,CAAA,KAAA,KAAS,IAAA,CAAK,cAAA,CAAe,KAAK,CAAC,CAAA;AAAA,EACvD;AACF,CAAA;;;AC3LO,IAAM,iBAAA,GAAN,cAAgC,YAAA,CAAsC;AAAA,EACnE,MAAA;AAAA,EACA,gBAAA;AAAA;AAAA;AAAA;AAAA,EAIA,gBAAA,uBAKA,GAAA,EAAI;AAAA,EACJ,IAAA;AAAA,EAER,YAAY,IAAA,EAAgC;AAC1C,IAAA,KAAA,EAAM;AACN,IAAA,IAAA,CAAK,gBAAA,GAAmB,IAAI,2BAAA,EAA4B;AACxD,IAAA,IAAA,CAAK,IAAA,GAAO,IAAA;AAEZ,IAAA,MAAM,UAAA,GAAa,IAAI,WAAA,CAAY,IAAI,CAAA;AACvC,IAAA,MAAM,SAAA,GAAY,KAAK,cAAA,EAAe;AACtC,IAAA,IAAA,CAAK,MAAA,GACH,SAAA,EAAW,UAAA,CAAW,UAAA,EAAY;AAAA,MAChC,cAAA,EAAgB,oBAAA;AAAA,MAChB,UAAA,EAAY;AAAA,QACV,aAAA,EAAe;AAAA;AACjB,KACD,CAAA,IAAK,UAAA;AAAA,EACV;AAAA,EAEA,MAAM,WAAA,CAAY,EAAE,SAAA,EAAW,SAAA,EAAW,QAAO,EAAqC;AACpF,IAAA,MAAA,GAAS,MAAA,IAAU,QAAA;AACnB,IAAA,IAAI,cAAA,GAAiC,iBAAA;AACrC,IAAA,IAAI;AACF,MAAA,IAAI,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA,EAAG;AAExC,QAAA,MAAM,QAAA,GAAW,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACpD,QAAA,IAAI,SAAA,KAAc,QAAA,CAAS,SAAA,IAAa,MAAA,KAAW,SAAS,MAAA,EAAQ;AAClE,UAAA,MAAM,IAAI,KAAA;AAAA,YACR,CAAA,oEAAA,EAAuE,SAAS,CAAA,qBAAA,EAAwB,QAAA,CAAS,SAAS,CAAA,YAAA,EAAe,QAAA,CAAS,MAAM,CAAA,oBAAA,EAAuB,SAAS,CAAA,YAAA,EAAe,MAAM,CAAA;AAAA,WAC/M;AAAA,QACF;AACA,QAAA;AAAA,MACF;AACA,MAAA,IAAI,aAAa,CAAA,EAAG;AAClB,QAAA,MAAM,IAAI,MAAM,sCAAsC,CAAA;AAAA,MACxD;AACA,MAAA,QAAQ,MAAA;AAAQ,QACd,KAAK,QAAA;AACH,UAAA,cAAA,GAAiB,iBAAA;AACjB,UAAA;AAAA,QACF,KAAK,WAAA;AACH,UAAA,cAAA,GAAiB,mBAAA;AACjB,UAAA;AAAA,QACF,KAAK,YAAA;AACH,UAAA,MAAM,IAAI,MAAM,4CAA4C,CAAA;AAAA;AAChE,IACF,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,sDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,IAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,SAAA,EAAW,MAAA;AAAO,SAC1C;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,gBAAA,CAAiB,IAAI,SAAA,EAAW;AAAA,MACnC,SAAA;AAAA,MACA,SAAA;AAAA,MACA,MAAA;AAAA,MACA,kBAAA,EAAoB;AAAA,KACrB,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,MAAA,CAAO,EAAE,WAAW,OAAA,EAAS,QAAA,EAAU,KAAI,EAA0C;AACzF,IAAA,IAAI,KAAA;AACJ,IAAA,IAAI,WAAA;AACJ,IAAA,IAAI;AACF,MAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,QAAA,MAAM,IAAI,MAAM,oCAAoC,CAAA;AAAA,MACtD;AAEA,MAAA,KAAA,GAAQ,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AACvC,MAAA,WAAA,GAAc,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACjD,MAAA,IAAI,CAAC,WAAA,EAAa;AAChB,QAAA,MAAM,IAAI,MAAM,CAAA,uCAAA,CAAyC,CAAA;AAAA,MAC3D;AAAA,IACF,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,gDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,IAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAEA,IAAA,IAAI;AACF,MAAA,MAAM,iBAAiB,WAAA,CAAY,kBAAA;AACnC,MAAA,MAAM,YAAY,GAAA,IAAO,OAAA,CAAQ,IAAI,MAAM,MAAA,CAAO,YAAY,CAAA;AAC9D,MAAA,MAAM,OAAA,GAAoB,OAAA,CAAQ,GAAA,CAAI,CAAC,QAAQ,CAAA,MAAO;AAAA,QACpD,EAAA,EAAI,UAAU,CAAC,CAAA;AAAA,QACf,MAAA;AAAA,QACA,UAAA,EAAY,QAAA,GAAW,CAAC,CAAA,IAAK;AAAC,OAChC,CAAE,CAAA;AAIF,MAAA,MAAM,SAAA,GAAY,GAAA;AAClB,MAAA,KAAA,IAAS,IAAI,CAAA,EAAG,CAAA,GAAI,OAAA,CAAQ,MAAA,EAAQ,KAAK,SAAA,EAAW;AAClD,QAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,CAAM,CAAA,EAAG,IAAI,SAAS,CAAA;AAC5C,QAAA,MAAM,aAAA,GAKF;AAAA,UACF,OAAA,EAAS,KAAA;AAAA,UACT,eAAA,EAAiB;AAAA,SACnB;AAGA,QAAA,MAAM,YAAA,GAAe,IAAA,CAAK,IAAA,CAAK,oBAAA,GAAuB,SAAS,CAAA;AAC/D,QAAA,IAAI,YAAA,EAAc;AAChB,UAAA,aAAA,CAAc,SAAS,YAAA,CAAa,MAAA;AACpC,UAAA,IAAI,OAAA,CAAQ,CAAC,CAAA,EAAG,MAAA,KAAW,aAAa,UAAA,EAAY;AAClD,YAAA,MAAM,IAAI,KAAA;AAAA,cACR,CAAA,kBAAA,EAAqB,SAAS,CAAA,gCAAA,EAAmC,YAAA,CAAa,UAAU,CAAA,4CAAA,EAA+C,OAAA,CAAQ,CAAC,CAAA,EAAG,MAAM,CAAA;AAAA,aAC3J;AAAA,UACF;AAAA,QACF;AAEA,QAAA,MAAM,KAAA,CAAM,OAAO,aAAa,CAAA;AAAA,MAClC;AAEA,MAAA,OAAO,SAAA;AAAA,IACT,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0CAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,KAAA,CAAM;AAAA,IACV,SAAA;AAAA,IACA,WAAA;AAAA,IACA,IAAA;AAAA,IACA,MAAA;AAAA,IACA;AAAA,GACF,EAAyD;AACvD,IAAA,IAAI,WAAA;AACJ,IAAA,IAAI;AACF,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,IAAA,CAAK,oBAAA,GAAuB,SAAS,CAAA;AAC/D,MAAA,IAAI,YAAA,EAAc;AAChB,QAAA,IAAI,WAAA,CAAY,MAAA,KAAW,YAAA,CAAa,UAAA,EAAY;AAClD,UAAA,MAAM,IAAI,KAAA;AAAA,YACR,qBAAqB,SAAS,CAAA,gCAAA,EAAmC,aAAa,UAAU,CAAA,iDAAA,EAAoD,YAAY,MAAM,CAAA;AAAA,WAChK;AAAA,QACF;AAAA,MACF;AACA,MAAA,WAAA,GAAc,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACjD,MAAA,IAAI,CAAC,WAAA,EAAa;AAChB,QAAA,MAAM,IAAI,MAAM,CAAA,uCAAA,CAAyC,CAAA;AAAA,MAC3D;AAAA,IACF,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,+CAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,IAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAEA,IAAA,MAAM,iBAAiB,WAAA,CAAY,kBAAA;AACnC,IAAA,IAAI;AACF,MAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AAC7C,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,gBAAA,CAAiB,SAAA,CAAU,MAAM,CAAA;AAC/D,MAAA,MAAM,OAAA,GAAwB,MAAM,KAAA,CAAM,KAAA,CAAM;AAAA,QAC9C,eAAA,EAAiB,cAAA;AAAA,QACjB,MAAA,EAAQ,WAAA;AAAA,QACR,KAAA,EAAO,IAAA;AAAA,QACP,OAAA,EAAS,gBAAA;AAAA,QACT,eAAA,EAAiB,aAAA;AAAA,QACjB,kBAAA,EAAoB,IAAA;AAAA,QACpB,WAAA,EAAa,EAAE,KAAA,EAAO,QAAA;AAAS;AAAA,OAChC,CAAA;AACD,MAAA,OAAO,OAAA,CAAQ,IAAI,CAAA,IAAA,MAAS;AAAA,QAC1B,EAAA,EAAI,MAAA,CAAO,IAAA,CAAK,EAAE,CAAA;AAAA,QAClB,OAAO,OAAO,IAAA,CAAK,IAAA,KAAS,QAAA,GAAW,KAAK,IAAA,GAAO,CAAA;AAAA,QACnD,QAAA,EAAU,IAAA,CAAK,UAAA,IAAc,EAAC;AAAA,QAC9B,GAAI,iBAAiB,IAAA,CAAK,MAAA,GAAS,EAAE,MAAA,EAAQ,IAAA,CAAK,MAAA,EAAO,GAAI;AAAC,OAChE,CAAE,CAAA;AAAA,IACJ,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,yCAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,GAAiC;AACrC,IAAA,IAAI;AACF,MAAA,MAAM,mBAAmB,MAAM,IAAA,CAAK,MAAA,CAAO,UAAA,CAAW,EAAE,CAAA;AACxD,MAAA,OAAO,gBAAA,CAAiB,UAAA,CAAW,GAAA,CAAI,CAAA,SAAA,KAAa,UAAU,EAAE,CAAA;AAAA,IAClE,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,gDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc;AAAA,SAC1B;AAAA,QACA;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,SAAA,GAAY,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AACjD,MAAA,MAAM,QAAA,GAAW,MAAM,SAAA,CAAU,QAAA,EAAS;AAC1C,MAAA,MAAM,WAAA,GAAc,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACvD,MAAA,IAAI,CAAC,WAAA,EAAa;AAChB,QAAA,MAAM,IAAI,MAAM,CAAA,uCAAA,CAAyC,CAAA;AAAA,MAC3D;AACA,MAAA,MAAM,YAAY,QAAA,CAAS,UAAA;AAC3B,MAAA,MAAM,QAAQ,QAAA,CAAS,YAAA;AACvB,MAAA,OAAO;AAAA,QACL,SAAA;AAAA,QACA,KAAA;AAAA,QACA,QAAQ,WAAA,CAAY;AAAA,OACtB;AAAA,IACF,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,kDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,CAAY,EAAE,SAAA,EAAU,EAAqC;AACjE,IAAA,IAAI;AACF,MAAA,MAAM,SAAA,GAAY,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AACjD,MAAA,MAAM,UAAU,SAAA,EAAU;AAC1B,MAAA,IAAA,CAAK,gBAAA,CAAiB,OAAO,SAAS,CAAA;AAAA,IACxC,SAAS,KAAA,EAAY;AACnB,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,gDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;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,SAAA;AACJ,IAAA,IAAI,WAAA;AACJ,IAAA,IAAI,cAAA;AACJ,IAAA,IAAI,MAAA;AACJ,IAAA,IAAI;AACF,MAAA,SAAA,GAAY,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AAC3C,MAAA,WAAA,GAAc,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACjD,MAAA,IAAI,CAAC,WAAA,EAAa;AAChB,QAAA,MAAM,IAAI,MAAM,CAAA,uCAAA,CAAyC,CAAA;AAAA,MAC3D;AACA,MAAA,cAAA,GAAiB,WAAA,CAAY,kBAAA;AAC7B,MAAA,MAAA,GAAS,EAAE,EAAA,EAAG;AACd,MAAA,IAAI,MAAA,CAAO,MAAA,EAAQ,MAAA,CAAO,MAAA,GAAS,MAAA,CAAO,MAAA;AAC1C,MAAA,IAAI,MAAA,CAAO,QAAA,EAAU,MAAA,CAAO,UAAA,GAAa,MAAA,CAAO,QAAA;AAAA,IAClD,SAAS,KAAA,EAAO;AACd,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,uDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,IAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AACA,IAAA,IAAI;AACF,MAAA,MAAM,UAAU,MAAA,CAAO;AAAA,QACrB,OAAA,EAAS,CAAC,MAAM,CAAA;AAAA,QAChB,eAAA,EAAiB;AAAA,OAClB,CAAA;AAAA,IACH,SAAS,KAAA,EAAY;AACnB,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,iDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,YAAA,CAAa,EAAE,SAAA,EAAW,IAAG,EAAsC;AACvE,IAAA,IAAI;AACF,MAAA,MAAM,SAAA,GAAY,IAAA,CAAK,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AACjD,MAAA,MAAM,UAAU,MAAA,CAAO,EAAE,KAAK,CAAC,EAAE,GAAG,CAAA;AAAA,IACtC,SAAS,KAAA,EAAY;AACnB,MAAA,MAAM,IAAI,WAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,iDAAA;AAAA,UACJ,QAAQ,WAAA,CAAY,OAAA;AAAA,UACpB,UAAU,aAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACA;AAAA,OACF;AAAA,IACF;AAAA,EACF;AACF","file":"index.js","sourcesContent":["import type {\n  OperatorSupport,\n  VectorFilter,\n  OperatorValueMap,\n  LogicalOperatorValueMap,\n  BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\nimport { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type { FilterCondition, FilterConnective, FilterOperator, Filters } from '@turbopuffer/turbopuffer';\n\ntype TurbopufferOperatorValueMap = Omit<OperatorValueMap, '$regex' | '$options' | '$elemMatch'>;\n\ntype TurbopufferLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor' | '$not'>;\n\ntype TurbopufferBlacklistedRootOperators = BlacklistedRootOperators | '$nor' | '$not';\n\nexport type TurbopufferVectorFilter = VectorFilter<\n  keyof TurbopufferOperatorValueMap,\n  TurbopufferOperatorValueMap,\n  TurbopufferLogicalOperatorValueMap,\n  TurbopufferBlacklistedRootOperators\n>;\n\n/**\n * Translator for converting Mastra filters to Turbopuffer format\n *\n * Mastra filters: { field: { $gt: 10 } }\n * Turbopuffer filters: [\"And\", [[\"field\", \"Gt\", 10]]]\n */\nexport class TurbopufferFilterTranslator extends BaseFilterTranslator<TurbopufferVectorFilter, Filters | undefined> {\n  protected override getSupportedOperators(): OperatorSupport {\n    return {\n      ...BaseFilterTranslator.DEFAULT_OPERATORS,\n      logical: ['$and', '$or'],\n      array: ['$in', '$nin', '$all'],\n      element: ['$exists'],\n      regex: [], // No regex support in Turbopuffer\n      custom: [], // No custom operators\n    };\n  }\n\n  /**\n   * Map Mastra operators to Turbopuffer operators\n   */\n  private operatorMap: Record<string, FilterOperator> = {\n    $eq: 'Eq',\n    $ne: 'NotEq',\n    $gt: 'Gt',\n    $gte: 'Gte',\n    $lt: 'Lt',\n    $lte: 'Lte',\n    $in: 'In',\n    $nin: 'NotIn',\n  };\n\n  /**\n   * Convert the Mastra filter to Turbopuffer format\n   */\n  translate(filter?: TurbopufferVectorFilter): Filters | undefined {\n    if (this.isEmpty(filter)) {\n      return undefined;\n    }\n\n    // Validate the filter structure before translating\n    this.validateFilter(filter);\n\n    // Translate the filter\n    const result = this.translateNode(filter);\n\n    // If we have a single condition (not a logical operator at the top level),\n    // wrap it in an implicit AND to match Turbopuffer's expected format\n    if (!Array.isArray(result) || result.length !== 2 || (result[0] !== 'And' && result[0] !== 'Or')) {\n      return ['And', [result as FilterCondition]];\n    }\n\n    return result as Filters;\n  }\n\n  /**\n   * Recursively translate a filter node\n   */\n  private translateNode(node: TurbopufferVectorFilter): Filters | FilterCondition {\n    // Handle empty or null nodes\n    if (node === null || node === undefined || Object.keys(node).length === 0) {\n      return ['And', []];\n    }\n\n    // Handle primitive values (direct equality comparison)\n    if (this.isPrimitive(node)) {\n      throw new Error('Direct primitive values not valid in this context for Turbopuffer');\n    }\n\n    // Handle direct array value (convert to $in)\n    if (Array.isArray(node)) {\n      throw new Error('Direct array values not valid in this context for Turbopuffer');\n    }\n\n    const entries = Object.entries(node);\n\n    // Process the first operator or field\n    if (entries.length === 0) {\n      return ['And', []];\n    }\n\n    const [key, value] = entries[0] as [string, any];\n\n    // Handle logical operators\n    if (key && this.isLogicalOperator(key)) {\n      return this.translateLogical(key, value);\n    }\n\n    // Multiple fields at top level - implicit AND\n    if (entries.length > 1) {\n      const conditions = entries.map(([field, fieldValue]) => this.translateFieldCondition(field, fieldValue));\n      return ['And', conditions];\n    }\n\n    // Single field with condition(s)\n    return this.translateFieldCondition(key, value);\n  }\n\n  /**\n   * Translate a field condition\n   */\n  private translateFieldCondition(field: string, value: any): FilterCondition {\n    // Handle Date object directly (convert to ISO string)\n    if (value instanceof Date) {\n      return [field, 'Eq', this.normalizeValue(value)];\n    }\n\n    // Handle primitive value (direct equality)\n    if (this.isPrimitive(value)) {\n      return [field, 'Eq', this.normalizeValue(value)];\n    }\n\n    // Handle array value (convert to $in)\n    if (Array.isArray(value)) {\n      return [field, 'In', this.normalizeArrayValues(value)];\n    }\n\n    // Handle object with operators\n    if (typeof value === 'object' && value !== null) {\n      const operators = Object.keys(value);\n\n      // If multiple operators for the same field, create an AND condition\n      if (operators.length > 1) {\n        // Check if all keys are operators\n        const allOperators = operators.every(op => this.isOperator(op));\n        if (allOperators) {\n          // For multiple comparison operators on one field\n          const conditions = operators.map(op => this.translateOperator(field, op, value[op]));\n          return ['And', conditions] as unknown as FilterCondition;\n        } else {\n          // For nested objects with multiple fields\n          const conditions = operators.map(op => {\n            const nestedField = `${field}.${op}`;\n            return this.translateFieldCondition(nestedField, value[op]);\n          });\n          return ['And', conditions] as unknown as FilterCondition;\n        }\n      }\n\n      // Single operator\n      const op = operators[0];\n      if (op && this.isOperator(op)) {\n        return this.translateOperator(field, op, value[op]);\n      }\n\n      // Nested field path (use dot notation)\n      if (op && !this.isOperator(op)) {\n        const nestedField = `${field}.${op}`;\n        return this.translateFieldCondition(nestedField, value[op]);\n      }\n    }\n\n    throw new Error(`Unsupported filter format for field: ${field}`);\n  }\n\n  /**\n   * Translate a logical operator\n   */\n  private translateLogical(operator: string, conditions: any[]): Filters {\n    // Map Mastra logical operators to Turbopuffer\n    const logicalOp: FilterConnective = operator === '$and' ? 'And' : 'Or';\n\n    // Validate conditions\n    if (!Array.isArray(conditions)) {\n      throw new Error(`Logical operator ${operator} requires an array of conditions`);\n    }\n\n    // Translate each condition\n    const translatedConditions = conditions.map(condition => {\n      if (typeof condition !== 'object' || condition === null) {\n        throw new Error(`Invalid condition for logical operator ${operator}`);\n      }\n      return this.translateNode(condition);\n    });\n\n    return [logicalOp, translatedConditions];\n  }\n\n  /**\n   * Translate a specific operator\n   */\n  private translateOperator(field: string, operator: string, value: any): FilterCondition {\n    // Handle comparison operators\n    if (operator && this.operatorMap[operator]) {\n      return [field, this.operatorMap[operator], this.normalizeValue(value)];\n    }\n\n    // Handle special cases\n    switch (operator) {\n      case '$exists':\n        // $exists: true -> use NotEq with null (field exists if it's not null)\n        // $exists: false -> use Eq with null (field doesn't exist if it is null)\n        return value ? [field, 'NotEq', null] : [field, 'Eq', null];\n\n      case '$all':\n        // $all is not directly supported, simulate with AND + IN conditions\n        if (!Array.isArray(value) || value.length === 0) {\n          throw new Error('$all operator requires a non-empty array');\n        }\n\n        const allConditions = value.map(item => [field, 'In', [this.normalizeValue(item)]] as FilterCondition);\n\n        // Return the array of conditions directly without nesting\n        return ['And', allConditions] as unknown as FilterCondition;\n\n      default:\n        throw new Error(`Unsupported operator: ${operator || 'undefined'}`);\n    }\n  }\n\n  /**\n   * Normalize a value for comparison operations\n   */\n  protected normalizeValue(value: any): any {\n    // Handle special value types\n    if (value instanceof Date) {\n      return value.toISOString();\n    }\n    return value;\n  }\n\n  /**\n   * Normalize array values\n   */\n  protected normalizeArrayValues(values: any[]): any[] {\n    return values.map(value => this.normalizeValue(value));\n  }\n}\n","import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';\nimport type {\n  CreateIndexParams,\n  DeleteIndexParams,\n  DeleteVectorParams,\n  DescribeIndexParams,\n  IndexStats,\n  QueryResult,\n  QueryVectorParams,\n  UpdateVectorParams,\n  UpsertVectorParams,\n} from '@mastra/core/vector';\nimport { MastraVector } from '@mastra/core/vector';\nimport { Turbopuffer } from '@turbopuffer/turbopuffer';\nimport type { DistanceMetric, QueryResults, Schema, Vector } from '@turbopuffer/turbopuffer';\nimport { TurbopufferFilterTranslator } from './filter';\nimport type { TurbopufferVectorFilter } from './filter';\n\ntype TurbopufferQueryVectorParams = QueryVectorParams<TurbopufferVectorFilter>;\n\nexport interface TurbopufferVectorOptions {\n  /** The API key to authenticate with. */\n  apiKey: string;\n  /** The base URL. Default is https://api.turbopuffer.com. */\n  baseUrl?: string;\n  /** The timeout to establish a connection, in ms. Default is 10_000. Only applicable in Node and Deno.*/\n  connectTimeout?: number;\n  /** The socket idle timeout, in ms. Default is 60_000. Only applicable in Node and Deno.*/\n  connectionIdleTimeout?: number;\n  /** The number of connections to open initially when creating a new client. Default is 0. */\n  warmConnections?: number;\n  /** Whether to compress requests and accept compressed responses. Default is true. */\n  compression?: boolean;\n  /**\n   * A callback function that takes an index name and returns a config object for that index.\n   * This allows you to define explicit schemas per index.\n   *\n   * Example:\n   * ```typescript\n   * schemaConfigForIndex: (indexName: string) => {\n   *   // Mastra's default embedding model and index for memory messages:\n   *   if (indexName === \"memory_messages_384\") {\n   *     return {\n   *       dimensions: 384,\n   *       schema: {\n   *         thread_id: {\n   *           type: \"string\",\n   *           filterable: true,\n   *         },\n   *       },\n   *     };\n   *   } else {\n   *     throw new Error(`TODO: add schema for index: ${indexName}`);\n   *   }\n   * },\n   * ```\n   */\n  schemaConfigForIndex?: (indexName: string) => {\n    dimensions: number;\n    schema: Schema;\n  };\n}\n\nexport class TurbopufferVector extends MastraVector<TurbopufferVectorFilter> {\n  private client: Turbopuffer;\n  private filterTranslator: TurbopufferFilterTranslator;\n  // There is no explicit create index operation in Turbopuffer, so just register that\n  // someone has called createIndex() and verify that subsequent upsert calls are consistent\n  // with how the index was \"created\"\n  private createIndexCache: Map<\n    string,\n    CreateIndexParams & {\n      tpufDistanceMetric: DistanceMetric;\n    }\n  > = new Map();\n  private opts: TurbopufferVectorOptions;\n\n  constructor(opts: TurbopufferVectorOptions) {\n    super();\n    this.filterTranslator = new TurbopufferFilterTranslator();\n    this.opts = opts;\n\n    const baseClient = new Turbopuffer(opts);\n    const telemetry = this.__getTelemetry();\n    this.client =\n      telemetry?.traceClass(baseClient, {\n        spanNamePrefix: 'turbopuffer-vector',\n        attributes: {\n          'vector.type': 'turbopuffer',\n        },\n      }) ?? baseClient;\n  }\n\n  async createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void> {\n    metric = metric ?? 'cosine'; // default to cosine distance\n    let distanceMetric: DistanceMetric = 'cosine_distance';\n    try {\n      if (this.createIndexCache.has(indexName)) {\n        // verify that the dimensions and distance metric match what we expect\n        const expected = this.createIndexCache.get(indexName)!;\n        if (dimension !== expected.dimension || metric !== expected.metric) {\n          throw new Error(\n            `createIndex() called more than once with inconsistent inputs. Index ${indexName} expected dimensions=${expected.dimension} and metric=${expected.metric} but got dimensions=${dimension} and metric=${metric}`,\n          );\n        }\n        return;\n      }\n      if (dimension <= 0) {\n        throw new Error('Dimension must be a positive integer');\n      }\n      switch (metric) {\n        case 'cosine':\n          distanceMetric = 'cosine_distance';\n          break;\n        case 'euclidean':\n          distanceMetric = 'euclidean_squared';\n          break;\n        case 'dotproduct':\n          throw new Error('dotproduct is not supported in Turbopuffer');\n      }\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_CREATE_INDEX_INVALID_ARGS',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.USER,\n          details: { indexName, dimension, metric },\n        },\n        error,\n      );\n    }\n\n    this.createIndexCache.set(indexName, {\n      indexName,\n      dimension,\n      metric,\n      tpufDistanceMetric: distanceMetric,\n    });\n  }\n\n  async upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]> {\n    let index;\n    let createIndex;\n    try {\n      if (vectors.length === 0) {\n        throw new Error('upsert() called with empty vectors');\n      }\n\n      index = this.client.namespace(indexName);\n      createIndex = this.createIndexCache.get(indexName);\n      if (!createIndex) {\n        throw new Error(`createIndex() not called for this index`);\n      }\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_UPSERT_INVALID_ARGS',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.USER,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n\n    try {\n      const distanceMetric = createIndex.tpufDistanceMetric;\n      const vectorIds = ids || vectors.map(() => crypto.randomUUID());\n      const records: Vector[] = vectors.map((vector, i) => ({\n        id: vectorIds[i]!,\n        vector: vector,\n        attributes: metadata?.[i] || {},\n      }));\n\n      // limit is 256 MB per upsert request, so set a reasonable batch size here that will stay under that for most cases\n      // https://turbopuffer.com/docs/limits\n      const batchSize = 100;\n      for (let i = 0; i < records.length; i += batchSize) {\n        const batch = records.slice(i, i + batchSize);\n        const upsertOptions: {\n          vectors: Vector[];\n          distance_metric: DistanceMetric;\n          schema?: Schema;\n          batchSize?: number;\n        } = {\n          vectors: batch,\n          distance_metric: distanceMetric,\n        };\n\n        // Use the schemaForIndex callback if provided\n        const schemaConfig = this.opts.schemaConfigForIndex?.(indexName);\n        if (schemaConfig) {\n          upsertOptions.schema = schemaConfig.schema;\n          if (vectors[0]?.length !== schemaConfig.dimensions) {\n            throw new Error(\n              `Turbopuffer index ${indexName} was configured with dimensions=${schemaConfig.dimensions} but attempting to upsert vectors[0].length=${vectors[0]?.length}`,\n            );\n          }\n        }\n\n        await index.upsert(upsertOptions);\n      }\n\n      return vectorIds;\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_UPSERT_FAILED',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  async query({\n    indexName,\n    queryVector,\n    topK,\n    filter,\n    includeVector,\n  }: TurbopufferQueryVectorParams): Promise<QueryResult[]> {\n    let createIndex;\n    try {\n      const schemaConfig = this.opts.schemaConfigForIndex?.(indexName);\n      if (schemaConfig) {\n        if (queryVector.length !== schemaConfig.dimensions) {\n          throw new Error(\n            `Turbopuffer index ${indexName} was configured with dimensions=${schemaConfig.dimensions} but attempting to query with queryVector.length=${queryVector.length}`,\n          );\n        }\n      }\n      createIndex = this.createIndexCache.get(indexName);\n      if (!createIndex) {\n        throw new Error(`createIndex() not called for this index`);\n      }\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_QUERY_INVALID_ARGS',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.USER,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n\n    const distanceMetric = createIndex.tpufDistanceMetric;\n    try {\n      const index = this.client.namespace(indexName);\n      const translatedFilter = this.filterTranslator.translate(filter);\n      const results: QueryResults = await index.query({\n        distance_metric: distanceMetric,\n        vector: queryVector,\n        top_k: topK,\n        filters: translatedFilter,\n        include_vectors: includeVector,\n        include_attributes: true,\n        consistency: { level: 'strong' }, // todo: make this configurable somehow?\n      });\n      return results.map(item => ({\n        id: String(item.id),\n        score: typeof item.dist === 'number' ? item.dist : 0,\n        metadata: item.attributes || {},\n        ...(includeVector && item.vector ? { vector: item.vector } : {}),\n      }));\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_QUERY_FAILED',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  async listIndexes(): Promise<string[]> {\n    try {\n      const namespacesResult = await this.client.namespaces({});\n      return namespacesResult.namespaces.map(namespace => namespace.id);\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_LIST_INDEXES_FAILED',\n          domain: ErrorDomain.STORAGE,\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 namespace = this.client.namespace(indexName);\n      const metadata = await namespace.metadata();\n      const createIndex = this.createIndexCache.get(indexName);\n      if (!createIndex) {\n        throw new Error(`createIndex() not called for this index`);\n      }\n      const dimension = metadata.dimensions;\n      const count = metadata.approx_count;\n      return {\n        dimension,\n        count,\n        metric: createIndex.metric,\n      };\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_DESCRIBE_INDEX_FAILED',\n          domain: ErrorDomain.STORAGE,\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      const namespace = this.client.namespace(indexName);\n      await namespace.deleteAll();\n      this.createIndexCache.delete(indexName);\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_DELETE_INDEX_FAILED',\n          domain: ErrorDomain.STORAGE,\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    let namespace;\n    let createIndex;\n    let distanceMetric;\n    let record;\n    try {\n      namespace = this.client.namespace(indexName);\n      createIndex = this.createIndexCache.get(indexName);\n      if (!createIndex) {\n        throw new Error(`createIndex() not called for this index`);\n      }\n      distanceMetric = createIndex.tpufDistanceMetric;\n      record = { id } as Vector;\n      if (update.vector) record.vector = update.vector;\n      if (update.metadata) record.attributes = update.metadata;\n    } catch (error) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_UPDATE_VECTOR_INVALID_ARGS',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.USER,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n    try {\n      await namespace.upsert({\n        vectors: [record],\n        distance_metric: distanceMetric,\n      });\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_UPDATE_VECTOR_FAILED',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes a vector by its ID.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to delete.\n   * @returns A promise that resolves when the deletion is complete.\n   * @throws Will throw an error if the deletion operation fails.\n   */\n  async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n    try {\n      const namespace = this.client.namespace(indexName);\n      await namespace.delete({ ids: [id] });\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: 'STORAGE_TURBOBUFFER_VECTOR_DELETE_VECTOR_FAILED',\n          domain: ErrorDomain.STORAGE,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n}\n"]}

package/dist/vector/filter.d.ts

@@ -0,0 +1,50 @@
+import type { OperatorSupport, VectorFilter, OperatorValueMap, LogicalOperatorValueMap, BlacklistedRootOperators } from '@mastra/core/vector/filter';
+import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import type { Filters } from '@turbopuffer/turbopuffer';
+type TurbopufferOperatorValueMap = Omit<OperatorValueMap, '$regex' | '$options' | '$elemMatch'>;
+type TurbopufferLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor' | '$not'>;
+type TurbopufferBlacklistedRootOperators = BlacklistedRootOperators | '$nor' | '$not';
+export type TurbopufferVectorFilter = VectorFilter<keyof TurbopufferOperatorValueMap, TurbopufferOperatorValueMap, TurbopufferLogicalOperatorValueMap, TurbopufferBlacklistedRootOperators>;
+/**
+ * Translator for converting Mastra filters to Turbopuffer format
+ *
+ * Mastra filters: { field: { $gt: 10 } }
+ * Turbopuffer filters: ["And", [["field", "Gt", 10]]]
+ */
+export declare class TurbopufferFilterTranslator extends BaseFilterTranslator<TurbopufferVectorFilter, Filters | undefined> {
+    protected getSupportedOperators(): OperatorSupport;
+    /**
+     * Map Mastra operators to Turbopuffer operators
+     */
+    private operatorMap;
+    /**
+     * Convert the Mastra filter to Turbopuffer format
+     */
+    translate(filter?: TurbopufferVectorFilter): Filters | undefined;
+    /**
+     * Recursively translate a filter node
+     */
+    private translateNode;
+    /**
+     * Translate a field condition
+     */
+    private translateFieldCondition;
+    /**
+     * Translate a logical operator
+     */
+    private translateLogical;
+    /**
+     * Translate a specific operator
+     */
+    private translateOperator;
+    /**
+     * Normalize a value for comparison operations
+     */
+    protected normalizeValue(value: any): any;
+    /**
+     * Normalize array values
+     */
+    protected normalizeArrayValues(values: any[]): any[];
+}
+export {};
+//# sourceMappingURL=filter.d.ts.map

package/dist/vector/filter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"filter.d.ts","sourceRoot":"","sources":["../../src/vector/filter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EACZ,gBAAgB,EAChB,uBAAuB,EACvB,wBAAwB,EACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAqD,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAE3G,KAAK,2BAA2B,GAAG,IAAI,CAAC,gBAAgB,EAAE,QAAQ,GAAG,UAAU,GAAG,YAAY,CAAC,CAAC;AAEhG,KAAK,kCAAkC,GAAG,IAAI,CAAC,uBAAuB,EAAE,MAAM,GAAG,MAAM,CAAC,CAAC;AAEzF,KAAK,mCAAmC,GAAG,wBAAwB,GAAG,MAAM,GAAG,MAAM,CAAC;AAEtF,MAAM,MAAM,uBAAuB,GAAG,YAAY,CAChD,MAAM,2BAA2B,EACjC,2BAA2B,EAC3B,kCAAkC,EAClC,mCAAmC,CACpC,CAAC;AAEF;;;;;GAKG;AACH,qBAAa,2BAA4B,SAAQ,oBAAoB,CAAC,uBAAuB,EAAE,OAAO,GAAG,SAAS,CAAC;cAC9F,qBAAqB,IAAI,eAAe;IAW3D;;OAEG;IACH,OAAO,CAAC,WAAW,CASjB;IAEF;;OAEG;IACH,SAAS,CAAC,MAAM,CAAC,EAAE,uBAAuB,GAAG,OAAO,GAAG,SAAS;IAoBhE;;OAEG;IACH,OAAO,CAAC,aAAa;IAwCrB;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAsD/B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAoBxB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA6BzB;;OAEG;IACH,SAAS,CAAC,cAAc,CAAC,KAAK,EAAE,GAAG,GAAG,GAAG;IAQzC;;OAEG;IACH,SAAS,CAAC,oBAAoB,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,GAAG,EAAE;CAGrD"}

package/dist/{_tsup-dts-rollup.d.cts → vector/index.d.ts}

@@ -1,105 +1,9 @@
-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 type { CreateIndexParams, DeleteIndexParams, DeleteVectorParams, DescribeIndexParams, IndexStats, QueryResult, QueryVectorParams, UpdateVectorParams, UpsertVectorParams } 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';
-
-/**
- * Translator for converting Mastra filters to Turbopuffer format
- *
- * Mastra filters: { field: { $gt: 10 } }
- * Turbopuffer filters: ["And", [["field", "Gt", 10]]]
- */
-export declare class TurbopufferFilterTranslator extends BaseFilterTranslator {
-    protected getSupportedOperators(): OperatorSupport;
-    /**
-     * Map Mastra operators to Turbopuffer operators
-     */
-    private operatorMap;
-    /**
-     * Convert the Mastra filter to Turbopuffer format
-     */
-    translate(filter?: VectorFilter): Filters | undefined;
-    /**
-     * Recursively translate a filter node
-     */
-    private translateNode;
-    /**
-     * Translate a field condition
-     */
-    private translateFieldCondition;
-    /**
-     * Translate a logical operator
-     */
-    private translateLogical;
-    /**
-     * Translate a specific operator
-     */
-    private translateOperator;
-    /**
-     * Normalize a value for comparison operations
-     */
-    protected normalizeValue(value: any): any;
-    /**
-     * Normalize array values
-     */
-    protected normalizeArrayValues(values: any[]): any[];
-}
-
-declare class TurbopufferVector extends MastraVector {
-    private client;
-    private filterTranslator;
-    private createIndexCache;
-    private opts;
-    constructor(opts: TurbopufferVectorOptions);
-    createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void>;
-    upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
-    query({ indexName, queryVector, topK, filter, includeVector }: QueryVectorParams): Promise<QueryResult[]>;
-    listIndexes(): Promise<string[]>;
-    /**
-     * 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 }
-
-declare interface TurbopufferVectorOptions {
+import type { TurbopufferVectorFilter } from './filter.js';
+type TurbopufferQueryVectorParams = QueryVectorParams<TurbopufferVectorFilter>;
+export interface TurbopufferVectorOptions {
     /** The API key to authenticate with. */
     apiKey: string;
     /** The base URL. Default is https://api.turbopuffer.com. */
@@ -141,7 +45,43 @@ declare interface TurbopufferVectorOptions {
         schema: Schema;
     };
 }
-export { TurbopufferVectorOptions }
-export { TurbopufferVectorOptions as TurbopufferVectorOptions_alias_1 }
-
-export { }
+export declare class TurbopufferVector extends MastraVector<TurbopufferVectorFilter> {
+    private client;
+    private filterTranslator;
+    private createIndexCache;
+    private opts;
+    constructor(opts: TurbopufferVectorOptions);
+    createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void>;
+    upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
+    query({ indexName, queryVector, topK, filter, includeVector, }: TurbopufferQueryVectorParams): Promise<QueryResult[]>;
+    listIndexes(): Promise<string[]>;
+    /**
+     * 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
+     */
+    describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats>;
+    deleteIndex({ indexName }: 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({ indexName, id, update }: 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({ indexName, id }: DeleteVectorParams): Promise<void>;
+}
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/vector/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vector/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,iBAAiB,EACjB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,UAAU,EACV,WAAW,EACX,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEnD,OAAO,KAAK,EAAgC,MAAM,EAAU,MAAM,0BAA0B,CAAC;AAE7F,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAExD,KAAK,4BAA4B,GAAG,iBAAiB,CAAC,uBAAuB,CAAC,CAAC;AAE/E,MAAM,WAAW,wBAAwB;IACvC,wCAAwC;IACxC,MAAM,EAAE,MAAM,CAAC;IACf,4DAA4D;IAC5D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,wGAAwG;IACxG,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,0FAA0F;IAC1F,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,4FAA4F;IAC5F,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,qFAAqF;IACrF,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,oBAAoB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK;QAC5C,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;CACH;AAED,qBAAa,iBAAkB,SAAQ,YAAY,CAAC,uBAAuB,CAAC;IAC1E,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,gBAAgB,CAA8B;IAItD,OAAO,CAAC,gBAAgB,CAKV;IACd,OAAO,CAAC,IAAI,CAA2B;gBAE3B,IAAI,EAAE,wBAAwB;IAgBpC,WAAW,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+C/E,MAAM,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA6EpF,KAAK,CAAC,EACV,SAAS,EACT,WAAW,EACX,IAAI,EACJ,MAAM,EACN,aAAa,GACd,EAAE,4BAA4B,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IA2DlD,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAgBtC;;;;;OAKG;IACG,aAAa,CAAC,EAAE,SAAS,EAAE,EAAE,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC;IA4BtE,WAAW,CAAC,EAAE,SAAS,EAAE,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBlE;;;;;;;;;OASG;IACG,YAAY,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IA4ChF;;;;;;OAMG;IACG,YAAY,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;CAgBzE"}