@mastra/vectorize 0.0.0-afterToolExecute-20250414225911

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/LICENSE.md ADDED
@@ -0,0 +1,46 @@
1
+ # Elastic License 2.0 (ELv2)
2
+
3
+ Copyright (c) 2025 Mastra AI, Inc.
4
+
5
+ **Acceptance**
6
+ By using the software, you agree to all of the terms and conditions below.
7
+
8
+ **Copyright License**
9
+ The licensor grants you a non-exclusive, royalty-free, worldwide, non-sublicensable, non-transferable license to use, copy, distribute, make available, and prepare derivative works of the software, in each case subject to the limitations and conditions below
10
+
11
+ **Limitations**
12
+ You may not provide the software to third parties as a hosted or managed service, where the service provides users with access to any substantial set of the features or functionality of the software.
13
+
14
+ You may not move, change, disable, or circumvent the license key functionality in the software, and you may not remove or obscure any functionality in the software that is protected by the license key.
15
+
16
+ You may not alter, remove, or obscure any licensing, copyright, or other notices of the licensor in the software. Any use of the licensor’s trademarks is subject to applicable law.
17
+
18
+ **Patents**
19
+ The licensor grants you a license, under any patent claims the licensor can license, or becomes able to license, to make, have made, use, sell, offer for sale, import and have imported the software, in each case subject to the limitations and conditions in this license. This license does not cover any patent claims that you cause to be infringed by modifications or additions to the software. If you or your company make any written claim that the software infringes or contributes to infringement of any patent, your patent license for the software granted under these terms ends immediately. If your company makes such a claim, your patent license ends immediately for work on behalf of your company.
20
+
21
+ **Notices**
22
+ You must ensure that anyone who gets a copy of any part of the software from you also gets a copy of these terms.
23
+
24
+ If you modify the software, you must include in any modified copies of the software prominent notices stating that you have modified the software.
25
+
26
+ **No Other Rights**
27
+ These terms do not imply any licenses other than those expressly granted in these terms.
28
+
29
+ **Termination**
30
+ If you use the software in violation of these terms, such use is not licensed, and your licenses will automatically terminate. If the licensor provides you with a notice of your violation, and you cease all violation of this license no later than 30 days after you receive that notice, your licenses will be reinstated retroactively. However, if you violate these terms after such reinstatement, any additional violation of these terms will cause your licenses to terminate automatically and permanently.
31
+
32
+ **No Liability**
33
+ As far as the law allows, the software comes as is, without any warranty or condition, and the licensor will not be liable to you for any damages arising out of these terms or the use or nature of the software, under any kind of legal claim.
34
+
35
+ **Definitions**
36
+ The _licensor_ is the entity offering these terms, and the _software_ is the software the licensor makes available under these terms, including any portion of it.
37
+
38
+ _you_ refers to the individual or entity agreeing to these terms.
39
+
40
+ _your company_ is any legal entity, sole proprietorship, or other kind of organization that you work for, plus all organizations that have control over, are under the control of, or are under common control with that organization. _control_ means ownership of substantially all the assets of an entity, or the power to direct its management and policies by vote, contract, or otherwise. Control can be direct or indirect.
41
+
42
+ _your licenses_ are all the licenses granted to you for the software under these terms.
43
+
44
+ _use_ means anything you do with the software requiring one of your licenses.
45
+
46
+ _trademark_ means trademarks, service marks, and similar rights.
package/README.md ADDED
@@ -0,0 +1,74 @@
1
+ # @mastra/vectorize
2
+
3
+ Vector store implementation for Vectorize, a managed vector database service optimized for AI applications.
4
+
5
+ ## Installation
6
+
7
+ ```bash
8
+ npm install @mastra/vectorize
9
+ ```
10
+
11
+ ## Usage
12
+
13
+ ```typescript
14
+ import { VectorizeStore } from '@mastra/vectorize';
15
+
16
+ const vectorStore = new VectorizeStore({
17
+ apiKey: process.env.VECTORIZE_API_KEY,
18
+ projectId: process.env.VECTORIZE_PROJECT_ID
19
+ });
20
+
21
+ // Create a new index
22
+ await vectorStore.createIndex({
23
+ indexName: 'my-index',
24
+ dimension: 1536,
25
+ metric: 'cosine'
26
+ });
27
+
28
+ // Add vectors
29
+ const vectors = [[0.1, 0.2, ...], [0.3, 0.4, ...]];
30
+ const metadata = [{ text: 'doc1' }, { text: 'doc2' }];
31
+ const ids = await vectorStore.upsert({
32
+ indexName: 'my-index',
33
+ vectors,
34
+ metadata
35
+ });
36
+
37
+ // Query vectors
38
+ const results = await vectorStore.query({
39
+ indexName: 'my-index',
40
+ queryVector: [0.1, 0.2, ...],
41
+ topK: 10,
42
+ filter: { text: { $eq: 'doc1' } },
43
+ includeVector: false
44
+ });
45
+ ```
46
+
47
+ ## Configuration
48
+
49
+ The Vectorize vector store requires the following configuration:
50
+
51
+ - `VECTORIZE_API_KEY`: Your Vectorize API key
52
+ - `VECTORIZE_INDEX_NAME`: Name of the index to use
53
+ - `VECTORIZE_PROJECT_ID`: Your Vectorize project ID
54
+
55
+ ## Features
56
+
57
+ - Purpose-built for AI and ML workloads
58
+ - High-performance vector similarity search
59
+ - Automatic index optimization
60
+ - Scalable architecture
61
+ - Real-time updates and queries
62
+
63
+ ## Methods
64
+
65
+ - `createIndex({ indexName, dimension, metric? })`: Create a new index
66
+ - `upsert({ indexName, vectors, metadata?, ids? })`: Add or update vectors
67
+ - `query({ indexName, queryVector, topK?, filter?, includeVector? })`: Search for similar vectors
68
+ - `listIndexes()`: List all indexes
69
+ - `describeIndex(indexName)`: Get index statistics
70
+ - `deleteIndex(indexName)`: Delete an index
71
+
72
+ ## Related Links
73
+
74
+ - [Vectorize Documentation](https://www.vectorize.com/docs)
@@ -0,0 +1,49 @@
1
+ import { BaseFilterTranslator } from '@mastra/core/vector/filter';
2
+ import Cloudflare from 'cloudflare';
3
+ import type { CreateIndexParams } from '@mastra/core/vector';
4
+ import { MastraVector } from '@mastra/core/vector';
5
+ import type { OperatorSupport } from '@mastra/core/vector/filter';
6
+ import type { ParamsToArgs } from '@mastra/core/vector';
7
+ import type { QueryResult } from '@mastra/core/vector';
8
+ import type { QueryVectorParams } from '@mastra/core/vector';
9
+ import type { UpsertVectorParams } from '@mastra/core/vector';
10
+ import type { VectorFilter } from '@mastra/core/vector/filter';
11
+
12
+ declare class CloudflareVector extends MastraVector {
13
+ client: Cloudflare;
14
+ accountId: string;
15
+ constructor({ accountId, apiToken }: {
16
+ accountId: string;
17
+ apiToken: string;
18
+ });
19
+ upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
20
+ transformFilter(filter?: VectorFilter): VectorFilter;
21
+ private verifyIndexExists;
22
+ createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
23
+ query(...args: ParamsToArgs<QueryVectorParams>): Promise<QueryResult[]>;
24
+ listIndexes(): Promise<string[]>;
25
+ describeIndex(indexName: string): Promise<{
26
+ dimension: number;
27
+ count: number;
28
+ metric: "cosine" | "euclidean" | "dotproduct";
29
+ }>;
30
+ deleteIndex(indexName: string): Promise<void>;
31
+ createMetadataIndex(indexName: string, propertyName: string, indexType: 'string' | 'number' | 'boolean'): Promise<void>;
32
+ deleteMetadataIndex(indexName: string, propertyName: string): Promise<void>;
33
+ listMetadataIndexes(indexName: string): Promise<Cloudflare.Vectorize.Indexes.MetadataIndex.MetadataIndexListResponse.MetadataIndex[]>;
34
+ updateIndexById(indexName: string, id: string, update: {
35
+ vector?: number[];
36
+ metadata?: Record<string, any>;
37
+ }): Promise<void>;
38
+ deleteIndexById(indexName: string, id: string): Promise<void>;
39
+ }
40
+ export { CloudflareVector }
41
+ export { CloudflareVector as CloudflareVector_alias_1 }
42
+
43
+ export declare class VectorizeFilterTranslator extends BaseFilterTranslator {
44
+ protected getSupportedOperators(): OperatorSupport;
45
+ translate(filter?: VectorFilter): VectorFilter;
46
+ private translateNode;
47
+ }
48
+
49
+ export { }
@@ -0,0 +1,49 @@
1
+ import { BaseFilterTranslator } from '@mastra/core/vector/filter';
2
+ import Cloudflare from 'cloudflare';
3
+ import type { CreateIndexParams } from '@mastra/core/vector';
4
+ import { MastraVector } from '@mastra/core/vector';
5
+ import type { OperatorSupport } from '@mastra/core/vector/filter';
6
+ import type { ParamsToArgs } from '@mastra/core/vector';
7
+ import type { QueryResult } from '@mastra/core/vector';
8
+ import type { QueryVectorParams } from '@mastra/core/vector';
9
+ import type { UpsertVectorParams } from '@mastra/core/vector';
10
+ import type { VectorFilter } from '@mastra/core/vector/filter';
11
+
12
+ declare class CloudflareVector extends MastraVector {
13
+ client: Cloudflare;
14
+ accountId: string;
15
+ constructor({ accountId, apiToken }: {
16
+ accountId: string;
17
+ apiToken: string;
18
+ });
19
+ upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
20
+ transformFilter(filter?: VectorFilter): VectorFilter;
21
+ private verifyIndexExists;
22
+ createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
23
+ query(...args: ParamsToArgs<QueryVectorParams>): Promise<QueryResult[]>;
24
+ listIndexes(): Promise<string[]>;
25
+ describeIndex(indexName: string): Promise<{
26
+ dimension: number;
27
+ count: number;
28
+ metric: "cosine" | "euclidean" | "dotproduct";
29
+ }>;
30
+ deleteIndex(indexName: string): Promise<void>;
31
+ createMetadataIndex(indexName: string, propertyName: string, indexType: 'string' | 'number' | 'boolean'): Promise<void>;
32
+ deleteMetadataIndex(indexName: string, propertyName: string): Promise<void>;
33
+ listMetadataIndexes(indexName: string): Promise<Cloudflare.Vectorize.Indexes.MetadataIndex.MetadataIndexListResponse.MetadataIndex[]>;
34
+ updateIndexById(indexName: string, id: string, update: {
35
+ vector?: number[];
36
+ metadata?: Record<string, any>;
37
+ }): Promise<void>;
38
+ deleteIndexById(indexName: string, id: string): Promise<void>;
39
+ }
40
+ export { CloudflareVector }
41
+ export { CloudflareVector as CloudflareVector_alias_1 }
42
+
43
+ export declare class VectorizeFilterTranslator extends BaseFilterTranslator {
44
+ protected getSupportedOperators(): OperatorSupport;
45
+ translate(filter?: VectorFilter): VectorFilter;
46
+ private translateNode;
47
+ }
48
+
49
+ export { }
package/dist/index.cjs ADDED
@@ -0,0 +1,233 @@
1
+ 'use strict';
2
+
3
+ var vector = require('@mastra/core/vector');
4
+ var Cloudflare = require('cloudflare');
5
+ var filter = require('@mastra/core/vector/filter');
6
+
7
+ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
8
+
9
+ var Cloudflare__default = /*#__PURE__*/_interopDefault(Cloudflare);
10
+
11
+ // src/vector/index.ts
12
+ var VectorizeFilterTranslator = class extends filter.BaseFilterTranslator {
13
+ getSupportedOperators() {
14
+ return {
15
+ ...filter.BaseFilterTranslator.DEFAULT_OPERATORS,
16
+ logical: [],
17
+ array: ["$in", "$nin"],
18
+ element: [],
19
+ regex: [],
20
+ custom: []
21
+ };
22
+ }
23
+ translate(filter) {
24
+ if (this.isEmpty(filter)) return filter;
25
+ this.validateFilter(filter);
26
+ return this.translateNode(filter);
27
+ }
28
+ translateNode(node, currentPath = "") {
29
+ if (this.isRegex(node)) {
30
+ throw new Error("Regex is not supported in Vectorize");
31
+ }
32
+ if (this.isPrimitive(node)) return { $eq: this.normalizeComparisonValue(node) };
33
+ if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };
34
+ const entries = Object.entries(node);
35
+ const firstEntry = entries[0];
36
+ if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {
37
+ const [operator, value] = firstEntry;
38
+ return { [operator]: this.normalizeComparisonValue(value) };
39
+ }
40
+ const result = {};
41
+ for (const [key, value] of entries) {
42
+ const newPath = currentPath ? `${currentPath}.${key}` : key;
43
+ if (this.isOperator(key)) {
44
+ result[key] = this.normalizeComparisonValue(value);
45
+ continue;
46
+ }
47
+ if (typeof value === "object" && value !== null && !Array.isArray(value)) {
48
+ if (Object.keys(value).length === 0) {
49
+ result[newPath] = this.translateNode(value);
50
+ continue;
51
+ }
52
+ const hasOperators = Object.keys(value).some((k) => this.isOperator(k));
53
+ if (hasOperators) {
54
+ result[newPath] = this.translateNode(value);
55
+ } else {
56
+ Object.assign(result, this.translateNode(value, newPath));
57
+ }
58
+ } else {
59
+ result[newPath] = this.translateNode(value);
60
+ }
61
+ }
62
+ return result;
63
+ }
64
+ };
65
+
66
+ // src/vector/index.ts
67
+ var CloudflareVector = class extends vector.MastraVector {
68
+ client;
69
+ accountId;
70
+ constructor({ accountId, apiToken }) {
71
+ super();
72
+ this.accountId = accountId;
73
+ this.client = new Cloudflare__default.default({
74
+ apiToken
75
+ });
76
+ }
77
+ async upsert(...args) {
78
+ const params = this.normalizeArgs("upsert", args);
79
+ const { indexName, vectors, metadata, ids } = params;
80
+ const generatedIds = ids || vectors.map(() => crypto.randomUUID());
81
+ const ndjson = vectors.map(
82
+ (vector, index) => JSON.stringify({
83
+ id: generatedIds[index],
84
+ values: vector,
85
+ metadata: metadata?.[index]
86
+ })
87
+ ).join("\n");
88
+ await this.client.vectorize.indexes.upsert(
89
+ indexName,
90
+ {
91
+ account_id: this.accountId,
92
+ body: ndjson
93
+ },
94
+ {
95
+ __binaryRequest: true
96
+ }
97
+ );
98
+ return generatedIds;
99
+ }
100
+ transformFilter(filter) {
101
+ const translator = new VectorizeFilterTranslator();
102
+ return translator.translate(filter);
103
+ }
104
+ async verifyIndexExists(indexName, dimension) {
105
+ try {
106
+ const info = await this.client.vectorize.indexes.info(indexName, {
107
+ account_id: this.accountId
108
+ });
109
+ if (!info) {
110
+ return false;
111
+ }
112
+ if (info.dimensions !== dimension) {
113
+ throw new Error(
114
+ `Index "${indexName}" already exists with ${info.dimensions} dimensions, but ${dimension} dimensions were requested`
115
+ );
116
+ }
117
+ return true;
118
+ } catch (error) {
119
+ const message = error?.errors?.[0]?.message || error?.message;
120
+ if (error.status === 404 || error.status === 410 || message?.toLowerCase().includes("not found") || message?.toLowerCase().includes("deleted")) {
121
+ return false;
122
+ }
123
+ throw error;
124
+ }
125
+ }
126
+ async createIndex(...args) {
127
+ const params = this.normalizeArgs("createIndex", args);
128
+ const { indexName, dimension, metric = "cosine" } = params;
129
+ const exists = await this.verifyIndexExists(indexName, dimension);
130
+ if (exists) {
131
+ this.logger.info(
132
+ `Index "${indexName}" already exists with ${dimension} dimensions and metric ${metric}, skipping creation.`
133
+ );
134
+ return;
135
+ }
136
+ await this.client.vectorize.indexes.create({
137
+ account_id: this.accountId,
138
+ config: {
139
+ dimensions: dimension,
140
+ metric: metric === "dotproduct" ? "dot-product" : metric
141
+ },
142
+ name: indexName
143
+ });
144
+ }
145
+ async query(...args) {
146
+ const params = this.normalizeArgs("query", args);
147
+ const { indexName, queryVector, topK = 10, filter, includeVector = false } = params;
148
+ const translatedFilter = this.transformFilter(filter) ?? {};
149
+ const response = await this.client.vectorize.indexes.query(indexName, {
150
+ account_id: this.accountId,
151
+ vector: queryVector,
152
+ returnValues: includeVector,
153
+ returnMetadata: "all",
154
+ topK,
155
+ filter: translatedFilter
156
+ });
157
+ return response?.matches?.map((match) => {
158
+ return {
159
+ id: match.id,
160
+ metadata: match.metadata,
161
+ score: match.score,
162
+ vector: match.values
163
+ };
164
+ }) || [];
165
+ }
166
+ async listIndexes() {
167
+ const res = await this.client.vectorize.indexes.list({
168
+ account_id: this.accountId
169
+ });
170
+ return res?.result?.map((index) => index.name) || [];
171
+ }
172
+ async describeIndex(indexName) {
173
+ const index = await this.client.vectorize.indexes.get(indexName, {
174
+ account_id: this.accountId
175
+ });
176
+ const described = await this.client.vectorize.indexes.info(indexName, {
177
+ account_id: this.accountId
178
+ });
179
+ return {
180
+ dimension: described?.dimensions,
181
+ // Since vector_count is not available in the response,
182
+ // we might need a separate API call to get the count if needed
183
+ count: described?.vectorCount || 0,
184
+ metric: index?.config?.metric
185
+ };
186
+ }
187
+ async deleteIndex(indexName) {
188
+ await this.client.vectorize.indexes.delete(indexName, {
189
+ account_id: this.accountId
190
+ });
191
+ }
192
+ async createMetadataIndex(indexName, propertyName, indexType) {
193
+ await this.client.vectorize.indexes.metadataIndex.create(indexName, {
194
+ account_id: this.accountId,
195
+ propertyName,
196
+ indexType
197
+ });
198
+ }
199
+ async deleteMetadataIndex(indexName, propertyName) {
200
+ await this.client.vectorize.indexes.metadataIndex.delete(indexName, {
201
+ account_id: this.accountId,
202
+ propertyName
203
+ });
204
+ }
205
+ async listMetadataIndexes(indexName) {
206
+ const res = await this.client.vectorize.indexes.metadataIndex.list(indexName, {
207
+ account_id: this.accountId
208
+ });
209
+ return res?.metadataIndexes ?? [];
210
+ }
211
+ async updateIndexById(indexName, id, update) {
212
+ if (!update.vector && !update.metadata) {
213
+ throw new Error("No update data provided");
214
+ }
215
+ const updatePayload = {
216
+ };
217
+ if (update.vector) {
218
+ updatePayload.vectors = [update.vector];
219
+ }
220
+ if (update.metadata) {
221
+ updatePayload.metadata = [update.metadata];
222
+ }
223
+ await this.upsert({ indexName, vectors: updatePayload.vectors, metadata: updatePayload.metadata });
224
+ }
225
+ async deleteIndexById(indexName, id) {
226
+ await this.client.vectorize.indexes.deleteByIds(indexName, {
227
+ ids: [id],
228
+ account_id: this.accountId
229
+ });
230
+ }
231
+ };
232
+
233
+ exports.CloudflareVector = CloudflareVector;
@@ -0,0 +1 @@
1
+ export { CloudflareVector } from './_tsup-dts-rollup.cjs';
@@ -0,0 +1 @@
1
+ export { CloudflareVector } from './_tsup-dts-rollup.js';
package/dist/index.js ADDED
@@ -0,0 +1,227 @@
1
+ import { MastraVector } from '@mastra/core/vector';
2
+ import Cloudflare from 'cloudflare';
3
+ import { BaseFilterTranslator } from '@mastra/core/vector/filter';
4
+
5
+ // src/vector/index.ts
6
+ var VectorizeFilterTranslator = class extends BaseFilterTranslator {
7
+ getSupportedOperators() {
8
+ return {
9
+ ...BaseFilterTranslator.DEFAULT_OPERATORS,
10
+ logical: [],
11
+ array: ["$in", "$nin"],
12
+ element: [],
13
+ regex: [],
14
+ custom: []
15
+ };
16
+ }
17
+ translate(filter) {
18
+ if (this.isEmpty(filter)) return filter;
19
+ this.validateFilter(filter);
20
+ return this.translateNode(filter);
21
+ }
22
+ translateNode(node, currentPath = "") {
23
+ if (this.isRegex(node)) {
24
+ throw new Error("Regex is not supported in Vectorize");
25
+ }
26
+ if (this.isPrimitive(node)) return { $eq: this.normalizeComparisonValue(node) };
27
+ if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };
28
+ const entries = Object.entries(node);
29
+ const firstEntry = entries[0];
30
+ if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {
31
+ const [operator, value] = firstEntry;
32
+ return { [operator]: this.normalizeComparisonValue(value) };
33
+ }
34
+ const result = {};
35
+ for (const [key, value] of entries) {
36
+ const newPath = currentPath ? `${currentPath}.${key}` : key;
37
+ if (this.isOperator(key)) {
38
+ result[key] = this.normalizeComparisonValue(value);
39
+ continue;
40
+ }
41
+ if (typeof value === "object" && value !== null && !Array.isArray(value)) {
42
+ if (Object.keys(value).length === 0) {
43
+ result[newPath] = this.translateNode(value);
44
+ continue;
45
+ }
46
+ const hasOperators = Object.keys(value).some((k) => this.isOperator(k));
47
+ if (hasOperators) {
48
+ result[newPath] = this.translateNode(value);
49
+ } else {
50
+ Object.assign(result, this.translateNode(value, newPath));
51
+ }
52
+ } else {
53
+ result[newPath] = this.translateNode(value);
54
+ }
55
+ }
56
+ return result;
57
+ }
58
+ };
59
+
60
+ // src/vector/index.ts
61
+ var CloudflareVector = class extends MastraVector {
62
+ client;
63
+ accountId;
64
+ constructor({ accountId, apiToken }) {
65
+ super();
66
+ this.accountId = accountId;
67
+ this.client = new Cloudflare({
68
+ apiToken
69
+ });
70
+ }
71
+ async upsert(...args) {
72
+ const params = this.normalizeArgs("upsert", args);
73
+ const { indexName, vectors, metadata, ids } = params;
74
+ const generatedIds = ids || vectors.map(() => crypto.randomUUID());
75
+ const ndjson = vectors.map(
76
+ (vector, index) => JSON.stringify({
77
+ id: generatedIds[index],
78
+ values: vector,
79
+ metadata: metadata?.[index]
80
+ })
81
+ ).join("\n");
82
+ await this.client.vectorize.indexes.upsert(
83
+ indexName,
84
+ {
85
+ account_id: this.accountId,
86
+ body: ndjson
87
+ },
88
+ {
89
+ __binaryRequest: true
90
+ }
91
+ );
92
+ return generatedIds;
93
+ }
94
+ transformFilter(filter) {
95
+ const translator = new VectorizeFilterTranslator();
96
+ return translator.translate(filter);
97
+ }
98
+ async verifyIndexExists(indexName, dimension) {
99
+ try {
100
+ const info = await this.client.vectorize.indexes.info(indexName, {
101
+ account_id: this.accountId
102
+ });
103
+ if (!info) {
104
+ return false;
105
+ }
106
+ if (info.dimensions !== dimension) {
107
+ throw new Error(
108
+ `Index "${indexName}" already exists with ${info.dimensions} dimensions, but ${dimension} dimensions were requested`
109
+ );
110
+ }
111
+ return true;
112
+ } catch (error) {
113
+ const message = error?.errors?.[0]?.message || error?.message;
114
+ if (error.status === 404 || error.status === 410 || message?.toLowerCase().includes("not found") || message?.toLowerCase().includes("deleted")) {
115
+ return false;
116
+ }
117
+ throw error;
118
+ }
119
+ }
120
+ async createIndex(...args) {
121
+ const params = this.normalizeArgs("createIndex", args);
122
+ const { indexName, dimension, metric = "cosine" } = params;
123
+ const exists = await this.verifyIndexExists(indexName, dimension);
124
+ if (exists) {
125
+ this.logger.info(
126
+ `Index "${indexName}" already exists with ${dimension} dimensions and metric ${metric}, skipping creation.`
127
+ );
128
+ return;
129
+ }
130
+ await this.client.vectorize.indexes.create({
131
+ account_id: this.accountId,
132
+ config: {
133
+ dimensions: dimension,
134
+ metric: metric === "dotproduct" ? "dot-product" : metric
135
+ },
136
+ name: indexName
137
+ });
138
+ }
139
+ async query(...args) {
140
+ const params = this.normalizeArgs("query", args);
141
+ const { indexName, queryVector, topK = 10, filter, includeVector = false } = params;
142
+ const translatedFilter = this.transformFilter(filter) ?? {};
143
+ const response = await this.client.vectorize.indexes.query(indexName, {
144
+ account_id: this.accountId,
145
+ vector: queryVector,
146
+ returnValues: includeVector,
147
+ returnMetadata: "all",
148
+ topK,
149
+ filter: translatedFilter
150
+ });
151
+ return response?.matches?.map((match) => {
152
+ return {
153
+ id: match.id,
154
+ metadata: match.metadata,
155
+ score: match.score,
156
+ vector: match.values
157
+ };
158
+ }) || [];
159
+ }
160
+ async listIndexes() {
161
+ const res = await this.client.vectorize.indexes.list({
162
+ account_id: this.accountId
163
+ });
164
+ return res?.result?.map((index) => index.name) || [];
165
+ }
166
+ async describeIndex(indexName) {
167
+ const index = await this.client.vectorize.indexes.get(indexName, {
168
+ account_id: this.accountId
169
+ });
170
+ const described = await this.client.vectorize.indexes.info(indexName, {
171
+ account_id: this.accountId
172
+ });
173
+ return {
174
+ dimension: described?.dimensions,
175
+ // Since vector_count is not available in the response,
176
+ // we might need a separate API call to get the count if needed
177
+ count: described?.vectorCount || 0,
178
+ metric: index?.config?.metric
179
+ };
180
+ }
181
+ async deleteIndex(indexName) {
182
+ await this.client.vectorize.indexes.delete(indexName, {
183
+ account_id: this.accountId
184
+ });
185
+ }
186
+ async createMetadataIndex(indexName, propertyName, indexType) {
187
+ await this.client.vectorize.indexes.metadataIndex.create(indexName, {
188
+ account_id: this.accountId,
189
+ propertyName,
190
+ indexType
191
+ });
192
+ }
193
+ async deleteMetadataIndex(indexName, propertyName) {
194
+ await this.client.vectorize.indexes.metadataIndex.delete(indexName, {
195
+ account_id: this.accountId,
196
+ propertyName
197
+ });
198
+ }
199
+ async listMetadataIndexes(indexName) {
200
+ const res = await this.client.vectorize.indexes.metadataIndex.list(indexName, {
201
+ account_id: this.accountId
202
+ });
203
+ return res?.metadataIndexes ?? [];
204
+ }
205
+ async updateIndexById(indexName, id, update) {
206
+ if (!update.vector && !update.metadata) {
207
+ throw new Error("No update data provided");
208
+ }
209
+ const updatePayload = {
210
+ };
211
+ if (update.vector) {
212
+ updatePayload.vectors = [update.vector];
213
+ }
214
+ if (update.metadata) {
215
+ updatePayload.metadata = [update.metadata];
216
+ }
217
+ await this.upsert({ indexName, vectors: updatePayload.vectors, metadata: updatePayload.metadata });
218
+ }
219
+ async deleteIndexById(indexName, id) {
220
+ await this.client.vectorize.indexes.deleteByIds(indexName, {
221
+ ids: [id],
222
+ account_id: this.accountId
223
+ });
224
+ }
225
+ };
226
+
227
+ export { CloudflareVector };
package/package.json ADDED
@@ -0,0 +1,45 @@
1
+ {
2
+ "name": "@mastra/vectorize",
3
+ "version": "0.0.0-afterToolExecute-20250414225911",
4
+ "description": "Cloudflare Vectorize store provider for Mastra",
5
+ "type": "module",
6
+ "files": [
7
+ "dist"
8
+ ],
9
+ "main": "dist/index.js",
10
+ "types": "dist/index.d.ts",
11
+ "exports": {
12
+ ".": {
13
+ "import": {
14
+ "types": "./dist/index.d.ts",
15
+ "default": "./dist/index.js"
16
+ },
17
+ "require": {
18
+ "types": "./dist/index.d.cts",
19
+ "default": "./dist/index.cjs"
20
+ }
21
+ },
22
+ "./package.json": "./package.json"
23
+ },
24
+ "license": "Elastic-2.0",
25
+ "dependencies": {
26
+ "cloudflare": "^4.1.0",
27
+ "@mastra/core": "0.0.0-afterToolExecute-20250414225911"
28
+ },
29
+ "devDependencies": {
30
+ "@microsoft/api-extractor": "^7.52.1",
31
+ "@types/node": "^20.17.27",
32
+ "dotenv": "^16.4.7",
33
+ "eslint": "^9.23.0",
34
+ "tsup": "^8.4.0",
35
+ "typescript": "^5.8.2",
36
+ "vitest": "^3.0.9",
37
+ "@internal/lint": "0.0.0-afterToolExecute-20250414225911"
38
+ },
39
+ "scripts": {
40
+ "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",
41
+ "build:watch": "pnpm build --watch",
42
+ "test": "vitest run",
43
+ "lint": "eslint ."
44
+ }
45
+ }