@mastra/vectorize 0.1.6-alpha.0 → 0.1.6-alpha.3

package/src/vector/index.ts

@@ -1,5 +1,12 @@
-import { type Filter } from '@mastra/core/filter';
-import { MastraVector, type QueryResult } from '@mastra/core/vector';
+import { MastraVector } from '@mastra/core/vector';
+import type {
+  QueryResult,
+  CreateIndexParams,
+  UpsertVectorParams,
+  QueryVectorParams,
+  ParamsToArgs,
+} from '@mastra/core/vector';
+import type { VectorFilter } from '@mastra/core/vector/filter';
 import Cloudflare from 'cloudflare';
 
 import { VectorizeFilterTranslator } from './filter';
@@ -17,12 +24,11 @@ export class CloudflareVector extends MastraVector {
     });
   }
 
-  async upsert(
-    indexName: string,
-    vectors: number[][],
-    metadata?: Record<string, any>[],
-    ids?: string[],
-  ): Promise<string[]> {
+  async upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]> {
+    const params = this.normalizeArgs<UpsertVectorParams>('upsert', args);
+
+    const { indexName, vectors, metadata, ids } = params;
+
     const generatedIds = ids || vectors.map(() => crypto.randomUUID());
 
     // Create NDJSON string - each line is a JSON object
@@ -51,17 +57,16 @@ export class CloudflareVector extends MastraVector {
     return generatedIds;
   }
 
-  transformFilter(filter?: Filter) {
+  transformFilter(filter?: VectorFilter) {
     const translator = new VectorizeFilterTranslator();
-    const translatedFilter = translator.translate(filter);
-    return translatedFilter;
+    return translator.translate(filter);
   }
 
-  async createIndex(
-    indexName: string,
-    dimension: number,
-    metric: 'cosine' | 'euclidean' | 'dotproduct' = 'cosine',
-  ): Promise<void> {
+  async createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void> {
+    const params = this.normalizeArgs<CreateIndexParams>('createIndex', args);
+
+    const { indexName, dimension, metric = 'cosine' } = params;
+
     await this.client.vectorize.indexes.create({
       account_id: this.accountId,
       config: {
@@ -72,14 +77,12 @@ export class CloudflareVector extends MastraVector {
     });
   }
 
-  async query(
-    indexName: string,
-    queryVector: number[],
-    topK: number = 10,
-    filter?: Filter,
-    includeVector: boolean = false,
-  ): Promise<QueryResult[]> {
-    const translatedFilter = this.transformFilter(filter);
+  async query(...args: ParamsToArgs<QueryVectorParams>): Promise<QueryResult[]> {
+    const params = this.normalizeArgs<QueryVectorParams>('query', args);
+
+    const { indexName, queryVector, topK = 10, filter, includeVector = false } = params;
+
+    const translatedFilter = this.transformFilter(filter) ?? {};
     const response = await this.client.vectorize.indexes.query(indexName, {
       account_id: this.accountId,
       vector: queryVector,