@mastra/vectorize 0.1.8-alpha.9 → 0.1.9-alpha.0

package/src/vector/index.ts

@@ -1,161 +0,0 @@
-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';
-
-export class CloudflareVector extends MastraVector {
-  client: Cloudflare;
-  accountId: string;
-
-  constructor({ accountId, apiToken }: { accountId: string; apiToken: string }) {
-    super();
-    this.accountId = accountId;
-
-    this.client = new Cloudflare({
-      apiKey: apiToken,
-    });
-  }
-
-  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
-    const ndjson = vectors
-      .map((vector, index) =>
-        JSON.stringify({
-          id: generatedIds[index]!,
-          values: vector,
-          metadata: metadata?.[index],
-        }),
-      )
-      .join('\n');
-
-    // Note: __binaryRequest is required for proper NDJSON handling
-    await this.client.vectorize.indexes.upsert(
-      indexName,
-      {
-        account_id: this.accountId,
-        body: ndjson,
-      },
-      {
-        __binaryRequest: true,
-      },
-    );
-
-    return generatedIds;
-  }
-
-  transformFilter(filter?: VectorFilter) {
-    const translator = new VectorizeFilterTranslator();
-    return translator.translate(filter);
-  }
-
-  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: {
-        dimensions: dimension,
-        metric: metric === 'dotproduct' ? 'dot-product' : metric,
-      },
-      name: indexName,
-    });
-  }
-
-  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,
-      returnValues: includeVector,
-      returnMetadata: 'all',
-      topK,
-      filter: translatedFilter,
-    });
-
-    return (
-      response?.matches?.map((match: any) => {
-        return {
-          id: match.id,
-          metadata: match.metadata,
-          score: match.score,
-          vector: match.values,
-        };
-      }) || []
-    );
-  }
-
-  async listIndexes(): Promise<string[]> {
-    const res = await this.client.vectorize.indexes.list({
-      account_id: this.accountId,
-    });
-
-    return res?.result?.map(index => index.name!) || [];
-  }
-
-  async describeIndex(indexName: string) {
-    const index = await this.client.vectorize.indexes.get(indexName, {
-      account_id: this.accountId,
-    });
-
-    const described = await this.client.vectorize.indexes.info(indexName, {
-      account_id: this.accountId,
-    });
-
-    return {
-      dimension: described?.dimensions!,
-      // Since vector_count is not available in the response,
-      // we might need a separate API call to get the count if needed
-      count: described?.vectorCount || 0,
-      metric: index?.config?.metric as 'cosine' | 'euclidean' | 'dotproduct',
-    };
-  }
-
-  async deleteIndex(indexName: string): Promise<void> {
-    await this.client.vectorize.indexes.delete(indexName, {
-      account_id: this.accountId,
-    });
-  }
-
-  async createMetadataIndex(indexName: string, propertyName: string, indexType: 'string' | 'number' | 'boolean') {
-    await this.client.vectorize.indexes.metadataIndex.create(indexName, {
-      account_id: this.accountId,
-      propertyName,
-      indexType,
-    });
-  }
-
-  async deleteMetadataIndex(indexName: string, propertyName: string) {
-    await this.client.vectorize.indexes.metadataIndex.delete(indexName, {
-      account_id: this.accountId,
-      propertyName,
-    });
-  }
-
-  async listMetadataIndexes(indexName: string) {
-    const res = await this.client.vectorize.indexes.metadataIndex.list(indexName, {
-      account_id: this.accountId,
-    });
-
-    return res?.metadataIndexes ?? [];
-  }
-}

package/tsconfig.json

@@ -1,5 +0,0 @@
-{
-  "extends": "../../tsconfig.node.json",
-  "include": ["src/**/*"],
-  "exclude": ["node_modules", "**/*.test.ts"]
-}

package/vitest.config.ts

@@ -1,11 +0,0 @@
-import { defineConfig } from 'vitest/config';
-
-export default defineConfig({
-  test: {
-    environment: 'node',
-    include: ['src/**/*.test.ts'],
-    coverage: {
-      reporter: ['text', 'json', 'html'],
-    },
-  },
-});