@mastra/chroma 0.1.6 → 0.1.7

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/chroma@0.1.6-alpha.4 build /home/runner/work/mastra/mastra/stores/chroma
+> @mastra/chroma@0.1.7-alpha.0 build /home/runner/work/mastra/mastra/stores/chroma
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.3.6
 TSC Build start
-TSC ⚡️ Build success in 8888ms
+TSC ⚡️ Build success in 7433ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.7.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/chroma/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.7.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/chroma/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 9024ms
+DTS ⚡️ Build success in 8668ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 7.28 KB
-ESM ⚡️ Build success in 741ms
 CJS dist/index.cjs 7.31 KB
-CJS ⚡️ Build success in 740ms
+CJS ⚡️ Build success in 703ms
+ESM dist/index.js 7.28 KB
+ESM ⚡️ Build success in 703ms

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @mastra/chroma
 
+## 0.1.7
+
+### Patch Changes
+
+- 1da20e7: Update typechecks for positional args
+- Updated dependencies [1da20e7]
+  - @mastra/core@0.4.4
+
+## 0.1.7-alpha.0
+
+### Patch Changes
+
+- 1da20e7: Update typechecks for positional args
+- Updated dependencies [1da20e7]
+  - @mastra/core@0.4.4-alpha.0
+
 ## 0.1.6
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -5,7 +5,9 @@ import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
 import type { ParamsToArgs } from '@mastra/core/vector';
 import type { QueryResult } from '@mastra/core/vector';
+import type { QueryVectorArgs } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
+import type { UpsertVectorArgs } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 
@@ -21,10 +23,14 @@ export declare class ChromaFilterTranslator extends BaseFilterTranslator {
     private translateOperator;
 }
 
+declare type ChromaQueryArgs = [...QueryVectorArgs, VectorFilter?];
+
 declare interface ChromaQueryVectorParams extends QueryVectorParams {
     documentFilter?: VectorFilter;
 }
 
+declare type ChromaUpsertArgs = [...UpsertVectorArgs, string[]?];
+
 declare interface ChromaUpsertVectorParams extends UpsertVectorParams {
     documents?: string[];
 }
@@ -41,11 +47,11 @@ declare class ChromaVector extends MastraVector {
     });
     getCollection(indexName: string, throwIfNotExists?: boolean): Promise<any>;
     private validateVectorDimensions;
-    upsert(...args: ParamsToArgs<ChromaUpsertVectorParams>): Promise<string[]>;
+    upsert(...args: ParamsToArgs<ChromaUpsertVectorParams> | ChromaUpsertArgs): Promise<string[]>;
     private HnswSpaceMap;
     createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
     transformFilter(filter?: VectorFilter): VectorFilter;
-    query(...args: ParamsToArgs<ChromaQueryVectorParams>): Promise<QueryResult[]>;
+    query(...args: ParamsToArgs<ChromaQueryVectorParams> | ChromaQueryArgs): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;

package/dist/_tsup-dts-rollup.d.ts

@@ -5,7 +5,9 @@ import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
 import type { ParamsToArgs } from '@mastra/core/vector';
 import type { QueryResult } from '@mastra/core/vector';
+import type { QueryVectorArgs } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
+import type { UpsertVectorArgs } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 
@@ -21,10 +23,14 @@ export declare class ChromaFilterTranslator extends BaseFilterTranslator {
     private translateOperator;
 }
 
+declare type ChromaQueryArgs = [...QueryVectorArgs, VectorFilter?];
+
 declare interface ChromaQueryVectorParams extends QueryVectorParams {
     documentFilter?: VectorFilter;
 }
 
+declare type ChromaUpsertArgs = [...UpsertVectorArgs, string[]?];
+
 declare interface ChromaUpsertVectorParams extends UpsertVectorParams {
     documents?: string[];
 }
@@ -41,11 +47,11 @@ declare class ChromaVector extends MastraVector {
     });
     getCollection(indexName: string, throwIfNotExists?: boolean): Promise<any>;
     private validateVectorDimensions;
-    upsert(...args: ParamsToArgs<ChromaUpsertVectorParams>): Promise<string[]>;
+    upsert(...args: ParamsToArgs<ChromaUpsertVectorParams> | ChromaUpsertArgs): Promise<string[]>;
     private HnswSpaceMap;
     createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
     transformFilter(filter?: VectorFilter): VectorFilter;
-    query(...args: ParamsToArgs<ChromaQueryVectorParams>): Promise<QueryResult[]>;
+    query(...args: ParamsToArgs<ChromaQueryVectorParams> | ChromaQueryArgs): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/chroma",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "Chroma vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "chromadb": "^1.9.4",
-    "@mastra/core": "^0.4.3"
+    "@mastra/core": "^0.4.4"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.49.2",

package/src/vector/index.ts

@@ -6,6 +6,8 @@ import type {
   UpsertVectorParams,
   QueryVectorParams,
   ParamsToArgs,
+  QueryVectorArgs,
+  UpsertVectorArgs,
 } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 import { ChromaClient } from 'chromadb';
@@ -16,10 +18,14 @@ interface ChromaUpsertVectorParams extends UpsertVectorParams {
   documents?: string[];
 }
 
+type ChromaUpsertArgs = [...UpsertVectorArgs, string[]?];
+
 interface ChromaQueryVectorParams extends QueryVectorParams {
   documentFilter?: VectorFilter;
 }
 
+type ChromaQueryArgs = [...QueryVectorArgs, VectorFilter?];
+
 export class ChromaVector extends MastraVector {
   private client: ChromaClient;
   private collections: Map<string, any>;
@@ -65,8 +71,8 @@ export class ChromaVector extends MastraVector {
     }
   }
 
-  async upsert(...args: ParamsToArgs<ChromaUpsertVectorParams>): Promise<string[]> {
-    const params = this.normalizeArgs<ChromaUpsertVectorParams>('upsert', args, ['documents']);
+  async upsert(...args: ParamsToArgs<ChromaUpsertVectorParams> | ChromaUpsertArgs): Promise<string[]> {
+    const params = this.normalizeArgs<ChromaUpsertVectorParams, ChromaUpsertArgs>('upsert', args, ['documents']);
 
     const { indexName, vectors, metadata, ids, documents } = params;
 
@@ -127,8 +133,8 @@ export class ChromaVector extends MastraVector {
     const translator = new ChromaFilterTranslator();
     return translator.translate(filter);
   }
-  async query(...args: ParamsToArgs<ChromaQueryVectorParams>): Promise<QueryResult[]> {
-    const params = this.normalizeArgs<ChromaQueryVectorParams>('query', args, ['documentFilter']);
+  async query(...args: ParamsToArgs<ChromaQueryVectorParams> | ChromaQueryArgs): Promise<QueryResult[]> {
+    const params = this.normalizeArgs<ChromaQueryVectorParams, ChromaQueryArgs>('query', args, ['documentFilter']);
 
     const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;