@mastra/core 0.4.3 → 0.4.4

package/dist/index.d.cts

@@ -18,7 +18,7 @@ export { createTool } from './tools/index.cjs';
 import { MastraTTS as MastraTTS$1, TTSConfig } from './tts/index.cjs';
 export { TagMaskOptions, deepMerge, delay, jsonSchemaPropertiesToTSTypes, jsonSchemaToModel, maskStreamTags } from './utils.cjs';
 import { MastraVector as MastraVector$1 } from './vector/index.cjs';
-export { CreateIndexParams, IndexStats, ParamsToArgs, QueryResult, QueryVectorParams, UpsertVectorParams } from './vector/index.cjs';
+export { CreateIndexArgs, CreateIndexParams, IndexStats, ParamsToArgs, QueryResult, QueryVectorArgs, QueryVectorParams, UpsertVectorArgs, UpsertVectorParams } from './vector/index.cjs';
 import { S as Step, W as Workflow$1 } from './workflow-DlRFMI4Q.cjs';
 export { c as createStep } from './workflow-DlRFMI4Q.cjs';
 export { getActivePathsAndStatus, getResultActivePaths, getStepResult, getSuspendedPaths, isErrorEvent, isFinalState, isTransitionEvent, isVariableReference, mergeChildValue, recursivelyCheckForFinalState, updateStepInHierarchy } from './workflows/index.cjs';

package/dist/index.d.ts

@@ -18,7 +18,7 @@ export { createTool } from './tools/index.js';
 import { MastraTTS as MastraTTS$1, TTSConfig } from './tts/index.js';
 export { TagMaskOptions, deepMerge, delay, jsonSchemaPropertiesToTSTypes, jsonSchemaToModel, maskStreamTags } from './utils.js';
 import { MastraVector as MastraVector$1 } from './vector/index.js';
-export { CreateIndexParams, IndexStats, ParamsToArgs, QueryResult, QueryVectorParams, UpsertVectorParams } from './vector/index.js';
+export { CreateIndexArgs, CreateIndexParams, IndexStats, ParamsToArgs, QueryResult, QueryVectorArgs, QueryVectorParams, UpsertVectorArgs, UpsertVectorParams } from './vector/index.js';
 import { S as Step, W as Workflow$1 } from './workflow-fGgxPZk4.js';
 export { c as createStep } from './workflow-fGgxPZk4.js';
 export { getActivePathsAndStatus, getResultActivePaths, getStepResult, getSuspendedPaths, isErrorEvent, isFinalState, isTransitionEvent, isVariableReference, mergeChildValue, recursivelyCheckForFinalState, updateStepInHierarchy } from './workflows/index.js';

package/dist/vector/index.d.cts

@@ -29,11 +29,13 @@ interface UpsertVectorParams {
     metadata?: Record<string, any>[];
     ids?: string[];
 }
+type UpsertVectorArgs = [string, number[][], Record<string, any>[], string[]?];
 interface CreateIndexParams {
     indexName: string;
     dimension: number;
     metric?: 'cosine' | 'euclidean' | 'dotproduct';
 }
+type CreateIndexArgs = [string, number, 'cosine' | 'euclidean' | 'dotproduct'];
 interface QueryVectorParams {
     indexName: string;
     queryVector: number[];
@@ -41,18 +43,19 @@ interface QueryVectorParams {
     filter?: VectorFilter;
     includeVector?: boolean;
 }
-type ParamsToArgs<T> = [string | T, ...Array<T[Exclude<keyof T, 'indexName'>]>];
+type QueryVectorArgs = [string, number[], number, VectorFilter?, boolean?];
+type ParamsToArgs<T> = [T] | (T extends QueryVectorParams ? QueryVectorArgs : never) | (T extends UpsertVectorParams ? UpsertVectorArgs : never) | (T extends CreateIndexParams ? CreateIndexArgs : never);
 
 declare abstract class MastraVector extends MastraBase {
     constructor();
     private readonly baseKeys;
-    protected normalizeArgs<T>(method: string, [first, ...rest]: ParamsToArgs<T>, extendedKeys?: string[]): T;
-    abstract query(...args: ParamsToArgs<QueryVectorParams>): Promise<QueryResult[]>;
-    abstract upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
-    abstract createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
+    protected normalizeArgs<T, E extends any[] = never>(method: string, [first, ...rest]: ParamsToArgs<T> | E, extendedKeys?: string[]): T;
+    abstract query<E extends QueryVectorArgs = QueryVectorArgs>(...args: ParamsToArgs<QueryVectorParams> | E): Promise<QueryResult[]>;
+    abstract upsert<E extends UpsertVectorArgs = UpsertVectorArgs>(...args: ParamsToArgs<UpsertVectorParams> | E): Promise<string[]>;
+    abstract createIndex<E extends CreateIndexArgs = CreateIndexArgs>(...args: ParamsToArgs<CreateIndexParams> | E): Promise<void>;
     abstract listIndexes(): Promise<string[]>;
     abstract describeIndex(indexName: string): Promise<IndexStats>;
     abstract deleteIndex(indexName: string): Promise<void>;
 }
 
-export { type CreateIndexParams, type IndexStats, MastraVector, type ParamsToArgs, type QueryResult, type QueryVectorParams, type UpsertVectorParams };
+export { type CreateIndexArgs, type CreateIndexParams, type IndexStats, MastraVector, type ParamsToArgs, type QueryResult, type QueryVectorArgs, type QueryVectorParams, type UpsertVectorArgs, type UpsertVectorParams };

package/dist/vector/index.d.ts

@@ -29,11 +29,13 @@ interface UpsertVectorParams {
     metadata?: Record<string, any>[];
     ids?: string[];
 }
+type UpsertVectorArgs = [string, number[][], Record<string, any>[], string[]?];
 interface CreateIndexParams {
     indexName: string;
     dimension: number;
     metric?: 'cosine' | 'euclidean' | 'dotproduct';
 }
+type CreateIndexArgs = [string, number, 'cosine' | 'euclidean' | 'dotproduct'];
 interface QueryVectorParams {
     indexName: string;
     queryVector: number[];
@@ -41,18 +43,19 @@ interface QueryVectorParams {
     filter?: VectorFilter;
     includeVector?: boolean;
 }
-type ParamsToArgs<T> = [string | T, ...Array<T[Exclude<keyof T, 'indexName'>]>];
+type QueryVectorArgs = [string, number[], number, VectorFilter?, boolean?];
+type ParamsToArgs<T> = [T] | (T extends QueryVectorParams ? QueryVectorArgs : never) | (T extends UpsertVectorParams ? UpsertVectorArgs : never) | (T extends CreateIndexParams ? CreateIndexArgs : never);
 
 declare abstract class MastraVector extends MastraBase {
     constructor();
     private readonly baseKeys;
-    protected normalizeArgs<T>(method: string, [first, ...rest]: ParamsToArgs<T>, extendedKeys?: string[]): T;
-    abstract query(...args: ParamsToArgs<QueryVectorParams>): Promise<QueryResult[]>;
-    abstract upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
-    abstract createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
+    protected normalizeArgs<T, E extends any[] = never>(method: string, [first, ...rest]: ParamsToArgs<T> | E, extendedKeys?: string[]): T;
+    abstract query<E extends QueryVectorArgs = QueryVectorArgs>(...args: ParamsToArgs<QueryVectorParams> | E): Promise<QueryResult[]>;
+    abstract upsert<E extends UpsertVectorArgs = UpsertVectorArgs>(...args: ParamsToArgs<UpsertVectorParams> | E): Promise<string[]>;
+    abstract createIndex<E extends CreateIndexArgs = CreateIndexArgs>(...args: ParamsToArgs<CreateIndexParams> | E): Promise<void>;
     abstract listIndexes(): Promise<string[]>;
     abstract describeIndex(indexName: string): Promise<IndexStats>;
     abstract deleteIndex(indexName: string): Promise<void>;
 }
 
-export { type CreateIndexParams, type IndexStats, MastraVector, type ParamsToArgs, type QueryResult, type QueryVectorParams, type UpsertVectorParams };
+export { type CreateIndexArgs, type CreateIndexParams, type IndexStats, MastraVector, type ParamsToArgs, type QueryResult, type QueryVectorArgs, type QueryVectorParams, type UpsertVectorArgs, type UpsertVectorParams };

package/dist/vector/libsql/index.d.cts

@@ -1,5 +1,5 @@
 import { VectorFilter } from '../filter/index.cjs';
-import { MastraVector, ParamsToArgs, QueryVectorParams, QueryResult, UpsertVectorParams, CreateIndexParams, IndexStats } from '../index.cjs';
+import { MastraVector, ParamsToArgs, QueryVectorParams, QueryVectorArgs, QueryResult, UpsertVectorParams, CreateIndexParams, IndexStats } from '../index.cjs';
 import '../../base-hs9NDAZ2.cjs';
 import '@opentelemetry/api';
 import '../../index-mKY1XrpK.cjs';
@@ -10,6 +10,7 @@ import '@opentelemetry/sdk-trace-base';
 interface LibSQLQueryParams extends QueryVectorParams {
     minScore?: number;
 }
+type LibSQLQueryArgs = [...QueryVectorArgs, number?];
 declare class LibSQLVector extends MastraVector {
     private turso;
     constructor({ connectionUrl, authToken, syncUrl, syncInterval, }: {
@@ -20,7 +21,7 @@ declare class LibSQLVector extends MastraVector {
     });
     protected rewriteDbUrl(url: string): string;
     transformFilter(filter?: VectorFilter): VectorFilter;
-    query(...args: ParamsToArgs<LibSQLQueryParams>): Promise<QueryResult[]>;
+    query(...args: ParamsToArgs<LibSQLQueryParams> | LibSQLQueryArgs): Promise<QueryResult[]>;
     upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
     createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
     deleteIndex(indexName: string): Promise<void>;

package/dist/vector/libsql/index.d.ts

@@ -1,5 +1,5 @@
 import { VectorFilter } from '../filter/index.js';
-import { MastraVector, ParamsToArgs, QueryVectorParams, QueryResult, UpsertVectorParams, CreateIndexParams, IndexStats } from '../index.js';
+import { MastraVector, ParamsToArgs, QueryVectorParams, QueryVectorArgs, QueryResult, UpsertVectorParams, CreateIndexParams, IndexStats } from '../index.js';
 import '../../base-D90KQ4XI.js';
 import '@opentelemetry/api';
 import '../../index-mKY1XrpK.js';
@@ -10,6 +10,7 @@ import '@opentelemetry/sdk-trace-base';
 interface LibSQLQueryParams extends QueryVectorParams {
     minScore?: number;
 }
+type LibSQLQueryArgs = [...QueryVectorArgs, number?];
 declare class LibSQLVector extends MastraVector {
     private turso;
     constructor({ connectionUrl, authToken, syncUrl, syncInterval, }: {
@@ -20,7 +21,7 @@ declare class LibSQLVector extends MastraVector {
     });
     protected rewriteDbUrl(url: string): string;
     transformFilter(filter?: VectorFilter): VectorFilter;
-    query(...args: ParamsToArgs<LibSQLQueryParams>): Promise<QueryResult[]>;
+    query(...args: ParamsToArgs<LibSQLQueryParams> | LibSQLQueryArgs): Promise<QueryResult[]>;
     upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
     createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void>;
     deleteIndex(indexName: string): Promise<void>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/core",
-  "version": "0.4.3",
+  "version": "0.4.4",
   "license": "ISC",
   "type": "module",
   "main": "dist/index.js",