@mastra/chroma 0.1.0-alpha.35 → 0.1.0-alpha.37
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/CHANGELOG.md +16 -0
- package/dist/_tsup-dts-rollup.d.ts +43 -0
- package/dist/index.d.ts +1 -25
- package/dist/index.js +3 -1
- package/package.json +5 -5
- package/src/vector/filter.ts +7 -1
- package/src/vector/index.ts +2 -2
- package/tsconfig.json +1 -6
package/CHANGELOG.md
CHANGED
|
@@ -1,5 +1,21 @@
|
|
|
1
1
|
# @mastra/chroma
|
|
2
2
|
|
|
3
|
+
## 0.1.0-alpha.37
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [a9345f9]
|
|
8
|
+
- @mastra/core@0.2.0-alpha.102
|
|
9
|
+
|
|
10
|
+
## 0.1.0-alpha.36
|
|
11
|
+
|
|
12
|
+
### Patch Changes
|
|
13
|
+
|
|
14
|
+
- 4f1d1a1: Enforce types ann cleanup package.json
|
|
15
|
+
- Updated dependencies [66a03ec]
|
|
16
|
+
- Updated dependencies [4f1d1a1]
|
|
17
|
+
- @mastra/core@0.2.0-alpha.101
|
|
18
|
+
|
|
3
19
|
## 0.1.0-alpha.35
|
|
4
20
|
|
|
5
21
|
### Patch Changes
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
import { BaseFilterTranslator } from '@mastra/core/filter';
|
|
2
|
+
import { Filter } from '@mastra/core/filter';
|
|
3
|
+
import { IndexStats } from '@mastra/core/vector';
|
|
4
|
+
import { MastraVector } from '@mastra/core/vector';
|
|
5
|
+
import { OperatorSupport } from '@mastra/core/filter';
|
|
6
|
+
import { QueryResult } from '@mastra/core/vector';
|
|
7
|
+
|
|
8
|
+
/**
|
|
9
|
+
* Translator for Chroma filter queries.
|
|
10
|
+
* Maintains MongoDB-compatible syntax while ensuring proper validation
|
|
11
|
+
* and normalization of values.
|
|
12
|
+
*/
|
|
13
|
+
export declare class ChromaFilterTranslator extends BaseFilterTranslator {
|
|
14
|
+
protected getSupportedOperators(): OperatorSupport;
|
|
15
|
+
translate(filter?: Filter): Filter | undefined;
|
|
16
|
+
private translateNode;
|
|
17
|
+
private translateOperator;
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
declare class ChromaVector extends MastraVector {
|
|
21
|
+
private client;
|
|
22
|
+
private collections;
|
|
23
|
+
constructor({ path, auth, }: {
|
|
24
|
+
path: string;
|
|
25
|
+
auth?: {
|
|
26
|
+
provider: string;
|
|
27
|
+
credentials: string;
|
|
28
|
+
};
|
|
29
|
+
});
|
|
30
|
+
private getCollection;
|
|
31
|
+
private validateVectorDimensions;
|
|
32
|
+
upsert(indexName: string, vectors: number[][], metadata?: Record<string, any>[], ids?: string[]): Promise<string[]>;
|
|
33
|
+
createIndex(indexName: string, dimension: number, metric?: 'cosine' | 'euclidean' | 'dotproduct'): Promise<void>;
|
|
34
|
+
transformFilter(filter?: Filter): Filter | undefined;
|
|
35
|
+
query(indexName: string, queryVector: number[], topK?: number, filter?: Filter, includeVector?: boolean): Promise<QueryResult[]>;
|
|
36
|
+
listIndexes(): Promise<string[]>;
|
|
37
|
+
describeIndex(indexName: string): Promise<IndexStats>;
|
|
38
|
+
deleteIndex(indexName: string): Promise<void>;
|
|
39
|
+
}
|
|
40
|
+
export { ChromaVector }
|
|
41
|
+
export { ChromaVector as ChromaVector_alias_1 }
|
|
42
|
+
|
|
43
|
+
export { }
|
package/dist/index.d.ts
CHANGED
|
@@ -1,25 +1 @@
|
|
|
1
|
-
|
|
2
|
-
import { MastraVector, QueryResult, IndexStats } from '@mastra/core/vector';
|
|
3
|
-
|
|
4
|
-
declare class ChromaVector extends MastraVector {
|
|
5
|
-
private client;
|
|
6
|
-
private collections;
|
|
7
|
-
constructor({ path, auth, }: {
|
|
8
|
-
path: string;
|
|
9
|
-
auth?: {
|
|
10
|
-
provider: string;
|
|
11
|
-
credentials: string;
|
|
12
|
-
};
|
|
13
|
-
});
|
|
14
|
-
private getCollection;
|
|
15
|
-
private validateVectorDimensions;
|
|
16
|
-
upsert(indexName: string, vectors: number[][], metadata?: Record<string, any>[], ids?: string[]): Promise<string[]>;
|
|
17
|
-
createIndex(indexName: string, dimension: number, metric?: 'cosine' | 'euclidean' | 'dotproduct'): Promise<void>;
|
|
18
|
-
transformFilter(filter?: Filter): Filter | undefined;
|
|
19
|
-
query(indexName: string, queryVector: number[], topK?: number, filter?: Filter, includeVector?: boolean): Promise<QueryResult[]>;
|
|
20
|
-
listIndexes(): Promise<string[]>;
|
|
21
|
-
describeIndex(indexName: string): Promise<IndexStats>;
|
|
22
|
-
deleteIndex(indexName: string): Promise<void>;
|
|
23
|
-
}
|
|
24
|
-
|
|
25
|
-
export { ChromaVector };
|
|
1
|
+
export { ChromaVector } from './_tsup-dts-rollup.js';
|
package/dist/index.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
+
import { BaseFilterTranslator } from '@mastra/core/filter';
|
|
1
2
|
import { MastraVector } from '@mastra/core/vector';
|
|
2
3
|
import { ChromaClient } from 'chromadb';
|
|
3
|
-
import { BaseFilterTranslator } from '@mastra/core/filter';
|
|
4
4
|
|
|
5
5
|
// src/vector/index.ts
|
|
6
6
|
var ChromaFilterTranslator = class extends BaseFilterTranslator {
|
|
@@ -88,6 +88,8 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
|
|
|
88
88
|
|
|
89
89
|
// src/vector/index.ts
|
|
90
90
|
var ChromaVector = class extends MastraVector {
|
|
91
|
+
client;
|
|
92
|
+
collections;
|
|
91
93
|
constructor({
|
|
92
94
|
path,
|
|
93
95
|
auth
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/chroma",
|
|
3
|
-
"version": "0.1.0-alpha.
|
|
3
|
+
"version": "0.1.0-alpha.37",
|
|
4
4
|
"description": "Chroma vector store provider for Mastra",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -16,16 +16,16 @@
|
|
|
16
16
|
},
|
|
17
17
|
"dependencies": {
|
|
18
18
|
"chromadb": "^1.9.4",
|
|
19
|
-
"@mastra/core": "^0.2.0-alpha.
|
|
19
|
+
"@mastra/core": "^0.2.0-alpha.102"
|
|
20
20
|
},
|
|
21
21
|
"devDependencies": {
|
|
22
|
-
"@
|
|
23
|
-
"@types/node": "^22.
|
|
22
|
+
"@microsoft/api-extractor": "^7.49.2",
|
|
23
|
+
"@types/node": "^22.13.1",
|
|
24
24
|
"tsup": "^8.0.1",
|
|
25
25
|
"vitest": "^3.0.4"
|
|
26
26
|
},
|
|
27
27
|
"scripts": {
|
|
28
|
-
"build": "tsup src/index.ts --format esm --dts --clean --treeshake",
|
|
28
|
+
"build": "tsup src/index.ts --format esm --experimental-dts --clean --treeshake",
|
|
29
29
|
"build:watch": "pnpm build --watch",
|
|
30
30
|
"test": "vitest run"
|
|
31
31
|
}
|
package/src/vector/filter.ts
CHANGED
|
@@ -1,4 +1,10 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import {
|
|
2
|
+
BaseFilterTranslator,
|
|
3
|
+
type FieldCondition,
|
|
4
|
+
type Filter,
|
|
5
|
+
type OperatorSupport,
|
|
6
|
+
type QueryOperator,
|
|
7
|
+
} from '@mastra/core/filter';
|
|
2
8
|
|
|
3
9
|
/**
|
|
4
10
|
* Translator for Chroma filter queries.
|
package/src/vector/index.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { Filter } from '@mastra/core/filter';
|
|
2
|
-
import { MastraVector, QueryResult, IndexStats } from '@mastra/core/vector';
|
|
1
|
+
import { type Filter } from '@mastra/core/filter';
|
|
2
|
+
import { MastraVector, type QueryResult, type IndexStats } from '@mastra/core/vector';
|
|
3
3
|
import { ChromaClient } from 'chromadb';
|
|
4
4
|
|
|
5
5
|
import { ChromaFilterTranslator } from './filter';
|
package/tsconfig.json
CHANGED
|
@@ -1,10 +1,5 @@
|
|
|
1
1
|
{
|
|
2
|
-
"extends": "../../tsconfig.json",
|
|
3
|
-
"compilerOptions": {
|
|
4
|
-
"moduleResolution": "bundler",
|
|
5
|
-
"outDir": "./dist",
|
|
6
|
-
"rootDir": "./src"
|
|
7
|
-
},
|
|
2
|
+
"extends": "../../tsconfig.node.json",
|
|
8
3
|
"include": ["src/**/*"],
|
|
9
4
|
"exclude": ["node_modules", "**/*.test.ts"]
|
|
10
5
|
}
|