@mastra/chroma 0.1.6 → 0.1.8-alpha.0
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/.turbo/turbo-build.log +5 -5
- package/CHANGELOG.md +26 -0
- package/dist/_tsup-dts-rollup.d.cts +8 -2
- package/dist/_tsup-dts-rollup.d.ts +8 -2
- package/package.json +2 -2
- package/src/vector/index.ts +10 -4
package/.turbo/turbo-build.log
CHANGED
|
@@ -1,23 +1,23 @@
|
|
|
1
1
|
|
|
2
|
-
> @mastra/chroma@0.1.
|
|
2
|
+
> @mastra/chroma@0.1.8-alpha.0 build /home/runner/work/mastra/mastra/stores/chroma
|
|
3
3
|
> tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake
|
|
4
4
|
|
|
5
5
|
[34mCLI[39m Building entry: src/index.ts
|
|
6
6
|
[34mCLI[39m Using tsconfig: tsconfig.json
|
|
7
7
|
[34mCLI[39m tsup v8.3.6
|
|
8
8
|
[34mTSC[39m Build start
|
|
9
|
-
[32mTSC[39m ⚡️ Build success in
|
|
9
|
+
[32mTSC[39m ⚡️ Build success in 6377ms
|
|
10
10
|
[34mDTS[39m Build start
|
|
11
11
|
[34mCLI[39m Target: es2022
|
|
12
12
|
Analysis will use the bundled TypeScript version 5.7.3
|
|
13
13
|
[36mWriting package typings: /home/runner/work/mastra/mastra/stores/chroma/dist/_tsup-dts-rollup.d.ts[39m
|
|
14
14
|
Analysis will use the bundled TypeScript version 5.7.3
|
|
15
15
|
[36mWriting package typings: /home/runner/work/mastra/mastra/stores/chroma/dist/_tsup-dts-rollup.d.cts[39m
|
|
16
|
-
[32mDTS[39m ⚡️ Build success in
|
|
16
|
+
[32mDTS[39m ⚡️ Build success in 8892ms
|
|
17
17
|
[34mCLI[39m Cleaning output folder
|
|
18
18
|
[34mESM[39m Build start
|
|
19
19
|
[34mCJS[39m Build start
|
|
20
20
|
[32mESM[39m [1mdist/index.js [22m[32m7.28 KB[39m
|
|
21
|
-
[32mESM[39m ⚡️ Build success in
|
|
21
|
+
[32mESM[39m ⚡️ Build success in 599ms
|
|
22
22
|
[32mCJS[39m [1mdist/index.cjs [22m[32m7.31 KB[39m
|
|
23
|
-
[32mCJS[39m ⚡️ Build success in
|
|
23
|
+
[32mCJS[39m ⚡️ Build success in 600ms
|
package/CHANGELOG.md
CHANGED
|
@@ -1,5 +1,31 @@
|
|
|
1
1
|
# @mastra/chroma
|
|
2
2
|
|
|
3
|
+
## 0.1.8-alpha.0
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [59df7b6]
|
|
8
|
+
- Updated dependencies [29f3a82]
|
|
9
|
+
- Updated dependencies [59df7b6]
|
|
10
|
+
- Updated dependencies [c139344]
|
|
11
|
+
- @mastra/core@0.5.0-alpha.0
|
|
12
|
+
|
|
13
|
+
## 0.1.7
|
|
14
|
+
|
|
15
|
+
### Patch Changes
|
|
16
|
+
|
|
17
|
+
- 1da20e7: Update typechecks for positional args
|
|
18
|
+
- Updated dependencies [1da20e7]
|
|
19
|
+
- @mastra/core@0.4.4
|
|
20
|
+
|
|
21
|
+
## 0.1.7-alpha.0
|
|
22
|
+
|
|
23
|
+
### Patch Changes
|
|
24
|
+
|
|
25
|
+
- 1da20e7: Update typechecks for positional args
|
|
26
|
+
- Updated dependencies [1da20e7]
|
|
27
|
+
- @mastra/core@0.4.4-alpha.0
|
|
28
|
+
|
|
3
29
|
## 0.1.6
|
|
4
30
|
|
|
5
31
|
### Patch Changes
|
|
@@ -5,7 +5,9 @@ import { MastraVector } from '@mastra/core/vector';
|
|
|
5
5
|
import type { OperatorSupport } from '@mastra/core/vector/filter';
|
|
6
6
|
import type { ParamsToArgs } from '@mastra/core/vector';
|
|
7
7
|
import type { QueryResult } from '@mastra/core/vector';
|
|
8
|
+
import type { QueryVectorArgs } from '@mastra/core/vector';
|
|
8
9
|
import type { QueryVectorParams } from '@mastra/core/vector';
|
|
10
|
+
import type { UpsertVectorArgs } from '@mastra/core/vector';
|
|
9
11
|
import type { UpsertVectorParams } from '@mastra/core/vector';
|
|
10
12
|
import type { VectorFilter } from '@mastra/core/vector/filter';
|
|
11
13
|
|
|
@@ -21,10 +23,14 @@ export declare class ChromaFilterTranslator extends BaseFilterTranslator {
|
|
|
21
23
|
private translateOperator;
|
|
22
24
|
}
|
|
23
25
|
|
|
26
|
+
declare type ChromaQueryArgs = [...QueryVectorArgs, VectorFilter?];
|
|
27
|
+
|
|
24
28
|
declare interface ChromaQueryVectorParams extends QueryVectorParams {
|
|
25
29
|
documentFilter?: VectorFilter;
|
|
26
30
|
}
|
|
27
31
|
|
|
32
|
+
declare type ChromaUpsertArgs = [...UpsertVectorArgs, string[]?];
|
|
33
|
+
|
|
28
34
|
declare interface ChromaUpsertVectorParams extends UpsertVectorParams {
|
|
29
35
|
documents?: string[];
|
|
30
36
|
}
|
|
@@ -41,11 +47,11 @@ declare class ChromaVector extends MastraVector {
|
|
|
41
47
|
});
|
|
42
48
|
getCollection(indexName: string, throwIfNotExists?: boolean): Promise<any>;
|
|
43
49
|
private validateVectorDimensions;
|
|
44
|
-
upsert(...args: ParamsToArgs<ChromaUpsertVectorParams>): Promise<string[]>;
|
|
50
|
+
upsert(...args: ParamsToArgs<ChromaUpsertVectorParams> | ChromaUpsertArgs): Promise<string[]>;
|
|
45
51
|
private HnswSpaceMap;
|
|
46
52
|
createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
|
|
47
53
|
transformFilter(filter?: VectorFilter): VectorFilter;
|
|
48
|
-
query(...args: ParamsToArgs<ChromaQueryVectorParams>): Promise<QueryResult[]>;
|
|
54
|
+
query(...args: ParamsToArgs<ChromaQueryVectorParams> | ChromaQueryArgs): Promise<QueryResult[]>;
|
|
49
55
|
listIndexes(): Promise<string[]>;
|
|
50
56
|
describeIndex(indexName: string): Promise<IndexStats>;
|
|
51
57
|
deleteIndex(indexName: string): Promise<void>;
|
|
@@ -5,7 +5,9 @@ import { MastraVector } from '@mastra/core/vector';
|
|
|
5
5
|
import type { OperatorSupport } from '@mastra/core/vector/filter';
|
|
6
6
|
import type { ParamsToArgs } from '@mastra/core/vector';
|
|
7
7
|
import type { QueryResult } from '@mastra/core/vector';
|
|
8
|
+
import type { QueryVectorArgs } from '@mastra/core/vector';
|
|
8
9
|
import type { QueryVectorParams } from '@mastra/core/vector';
|
|
10
|
+
import type { UpsertVectorArgs } from '@mastra/core/vector';
|
|
9
11
|
import type { UpsertVectorParams } from '@mastra/core/vector';
|
|
10
12
|
import type { VectorFilter } from '@mastra/core/vector/filter';
|
|
11
13
|
|
|
@@ -21,10 +23,14 @@ export declare class ChromaFilterTranslator extends BaseFilterTranslator {
|
|
|
21
23
|
private translateOperator;
|
|
22
24
|
}
|
|
23
25
|
|
|
26
|
+
declare type ChromaQueryArgs = [...QueryVectorArgs, VectorFilter?];
|
|
27
|
+
|
|
24
28
|
declare interface ChromaQueryVectorParams extends QueryVectorParams {
|
|
25
29
|
documentFilter?: VectorFilter;
|
|
26
30
|
}
|
|
27
31
|
|
|
32
|
+
declare type ChromaUpsertArgs = [...UpsertVectorArgs, string[]?];
|
|
33
|
+
|
|
28
34
|
declare interface ChromaUpsertVectorParams extends UpsertVectorParams {
|
|
29
35
|
documents?: string[];
|
|
30
36
|
}
|
|
@@ -41,11 +47,11 @@ declare class ChromaVector extends MastraVector {
|
|
|
41
47
|
});
|
|
42
48
|
getCollection(indexName: string, throwIfNotExists?: boolean): Promise<any>;
|
|
43
49
|
private validateVectorDimensions;
|
|
44
|
-
upsert(...args: ParamsToArgs<ChromaUpsertVectorParams>): Promise<string[]>;
|
|
50
|
+
upsert(...args: ParamsToArgs<ChromaUpsertVectorParams> | ChromaUpsertArgs): Promise<string[]>;
|
|
45
51
|
private HnswSpaceMap;
|
|
46
52
|
createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
|
|
47
53
|
transformFilter(filter?: VectorFilter): VectorFilter;
|
|
48
|
-
query(...args: ParamsToArgs<ChromaQueryVectorParams>): Promise<QueryResult[]>;
|
|
54
|
+
query(...args: ParamsToArgs<ChromaQueryVectorParams> | ChromaQueryArgs): Promise<QueryResult[]>;
|
|
49
55
|
listIndexes(): Promise<string[]>;
|
|
50
56
|
describeIndex(indexName: string): Promise<IndexStats>;
|
|
51
57
|
deleteIndex(indexName: string): Promise<void>;
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/chroma",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.8-alpha.0",
|
|
4
4
|
"description": "Chroma vector store provider for Mastra",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -20,7 +20,7 @@
|
|
|
20
20
|
},
|
|
21
21
|
"dependencies": {
|
|
22
22
|
"chromadb": "^1.9.4",
|
|
23
|
-
"@mastra/core": "^0.
|
|
23
|
+
"@mastra/core": "^0.5.0-alpha.0"
|
|
24
24
|
},
|
|
25
25
|
"devDependencies": {
|
|
26
26
|
"@microsoft/api-extractor": "^7.49.2",
|
package/src/vector/index.ts
CHANGED
|
@@ -6,6 +6,8 @@ import type {
|
|
|
6
6
|
UpsertVectorParams,
|
|
7
7
|
QueryVectorParams,
|
|
8
8
|
ParamsToArgs,
|
|
9
|
+
QueryVectorArgs,
|
|
10
|
+
UpsertVectorArgs,
|
|
9
11
|
} from '@mastra/core/vector';
|
|
10
12
|
import type { VectorFilter } from '@mastra/core/vector/filter';
|
|
11
13
|
import { ChromaClient } from 'chromadb';
|
|
@@ -16,10 +18,14 @@ interface ChromaUpsertVectorParams extends UpsertVectorParams {
|
|
|
16
18
|
documents?: string[];
|
|
17
19
|
}
|
|
18
20
|
|
|
21
|
+
type ChromaUpsertArgs = [...UpsertVectorArgs, string[]?];
|
|
22
|
+
|
|
19
23
|
interface ChromaQueryVectorParams extends QueryVectorParams {
|
|
20
24
|
documentFilter?: VectorFilter;
|
|
21
25
|
}
|
|
22
26
|
|
|
27
|
+
type ChromaQueryArgs = [...QueryVectorArgs, VectorFilter?];
|
|
28
|
+
|
|
23
29
|
export class ChromaVector extends MastraVector {
|
|
24
30
|
private client: ChromaClient;
|
|
25
31
|
private collections: Map<string, any>;
|
|
@@ -65,8 +71,8 @@ export class ChromaVector extends MastraVector {
|
|
|
65
71
|
}
|
|
66
72
|
}
|
|
67
73
|
|
|
68
|
-
async upsert(...args: ParamsToArgs<ChromaUpsertVectorParams>): Promise<string[]> {
|
|
69
|
-
const params = this.normalizeArgs<ChromaUpsertVectorParams>('upsert', args, ['documents']);
|
|
74
|
+
async upsert(...args: ParamsToArgs<ChromaUpsertVectorParams> | ChromaUpsertArgs): Promise<string[]> {
|
|
75
|
+
const params = this.normalizeArgs<ChromaUpsertVectorParams, ChromaUpsertArgs>('upsert', args, ['documents']);
|
|
70
76
|
|
|
71
77
|
const { indexName, vectors, metadata, ids, documents } = params;
|
|
72
78
|
|
|
@@ -127,8 +133,8 @@ export class ChromaVector extends MastraVector {
|
|
|
127
133
|
const translator = new ChromaFilterTranslator();
|
|
128
134
|
return translator.translate(filter);
|
|
129
135
|
}
|
|
130
|
-
async query(...args: ParamsToArgs<ChromaQueryVectorParams>): Promise<QueryResult[]> {
|
|
131
|
-
const params = this.normalizeArgs<ChromaQueryVectorParams>('query', args, ['documentFilter']);
|
|
136
|
+
async query(...args: ParamsToArgs<ChromaQueryVectorParams> | ChromaQueryArgs): Promise<QueryResult[]> {
|
|
137
|
+
const params = this.normalizeArgs<ChromaQueryVectorParams, ChromaQueryArgs>('query', args, ['documentFilter']);
|
|
132
138
|
|
|
133
139
|
const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
|
|
134
140
|
|