@mastra/pg 0.1.6 → 0.1.8-alpha.0

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.1.6-alpha.4 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.1.8-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
 > 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 7672ms
+TSC ⚡️ Build success in 8825ms
 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/pg/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/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10963ms
+DTS ⚡️ Build success in 9901ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 33.20 KB
-ESM ⚡️ Build success in 1374ms
-CJS dist/index.cjs 33.60 KB
-CJS ⚡️ Build success in 1374ms
+ESM dist/index.js 33.26 KB
+ESM ⚡️ Build success in 1052ms
+CJS dist/index.cjs 33.66 KB
+CJS ⚡️ Build success in 1054ms

package/CHANGELOG.md

@@ -1,5 +1,31 @@
 # @mastra/pg
 
+## 0.1.8-alpha.0
+
+### Patch Changes
+
+- Updated dependencies [59df7b6]
+- Updated dependencies [29f3a82]
+- Updated dependencies [59df7b6]
+- Updated dependencies [c139344]
+  - @mastra/core@0.5.0-alpha.0
+
+## 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

@@ -1,6 +1,7 @@
 import type { ArrayOperator } from '@mastra/core/vector/filter';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import type { BasicOperator } from '@mastra/core/vector/filter';
+import type { CreateIndexArgs } from '@mastra/core/vector';
 import type { CreateIndexParams } from '@mastra/core/vector';
 import type { ElementOperator } from '@mastra/core/vector/filter';
 import type { EvalRow } from '@mastra/core/storage';
@@ -13,6 +14,7 @@ import type { NumericOperator } from '@mastra/core/vector/filter';
 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 { RegexOperator } from '@mastra/core/vector/filter';
 import type { StorageColumn } from '@mastra/core/storage';
@@ -135,11 +137,15 @@ declare type OperatorFn = (key: string, paramIndex: number, value?: any) => Filt
 
 export declare type OperatorType = BasicOperator | NumericOperator | ArrayOperator | ElementOperator | LogicalOperator | '$contains' | Exclude<RegexOperator, '$options'>;
 
+declare type PgCreateIndexArgs = [...CreateIndexArgs, IndexConfig?, boolean?];
+
 declare interface PgCreateIndexParams extends CreateIndexParams {
     indexConfig?: IndexConfig;
     buildIndex?: boolean;
 }
 
+declare type PgDefineIndexArgs = [string, 'cosine' | 'euclidean' | 'dotproduct', IndexConfig];
+
 declare interface PgDefineIndexParams {
     indexName: string;
     metric: 'cosine' | 'euclidean' | 'dotproduct';
@@ -175,6 +181,8 @@ declare interface PGIndexStats extends IndexStats {
 export { PGIndexStats }
 export { PGIndexStats as PGIndexStats_alias_1 }
 
+declare type PgQueryVectorArgs = [...QueryVectorArgs, number?, number?, number?];
+
 declare interface PgQueryVectorParams extends QueryVectorParams {
     minScore?: number;
     /**
@@ -195,14 +203,14 @@ declare class PgVector extends MastraVector {
     constructor(connectionString: string);
     transformFilter(filter?: VectorFilter): VectorFilter;
     getIndexInfo(indexName: string): Promise<PGIndexStats>;
-    query(...args: ParamsToArgs<PgQueryVectorParams>): Promise<QueryResult[]>;
+    query(...args: ParamsToArgs<PgQueryVectorParams> | PgQueryVectorArgs): Promise<QueryResult[]>;
     upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
-    createIndex(...args: ParamsToArgs<PgCreateIndexParams>): Promise<void>;
+    createIndex(...args: ParamsToArgs<PgCreateIndexParams> | PgCreateIndexArgs): Promise<void>;
     /**
      * @deprecated This function is deprecated. Use buildIndex instead
      */
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
-    buildIndex(...args: ParamsToArgs<PgDefineIndexParams>): Promise<void>;
+    buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void>;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<PGIndexStats>;
     deleteIndex(indexName: string): Promise<void>;

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

@@ -1,6 +1,7 @@
 import type { ArrayOperator } from '@mastra/core/vector/filter';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import type { BasicOperator } from '@mastra/core/vector/filter';
+import type { CreateIndexArgs } from '@mastra/core/vector';
 import type { CreateIndexParams } from '@mastra/core/vector';
 import type { ElementOperator } from '@mastra/core/vector/filter';
 import type { EvalRow } from '@mastra/core/storage';
@@ -13,6 +14,7 @@ import type { NumericOperator } from '@mastra/core/vector/filter';
 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 { RegexOperator } from '@mastra/core/vector/filter';
 import type { StorageColumn } from '@mastra/core/storage';
@@ -135,11 +137,15 @@ declare type OperatorFn = (key: string, paramIndex: number, value?: any) => Filt
 
 export declare type OperatorType = BasicOperator | NumericOperator | ArrayOperator | ElementOperator | LogicalOperator | '$contains' | Exclude<RegexOperator, '$options'>;
 
+declare type PgCreateIndexArgs = [...CreateIndexArgs, IndexConfig?, boolean?];
+
 declare interface PgCreateIndexParams extends CreateIndexParams {
     indexConfig?: IndexConfig;
     buildIndex?: boolean;
 }
 
+declare type PgDefineIndexArgs = [string, 'cosine' | 'euclidean' | 'dotproduct', IndexConfig];
+
 declare interface PgDefineIndexParams {
     indexName: string;
     metric: 'cosine' | 'euclidean' | 'dotproduct';
@@ -175,6 +181,8 @@ declare interface PGIndexStats extends IndexStats {
 export { PGIndexStats }
 export { PGIndexStats as PGIndexStats_alias_1 }
 
+declare type PgQueryVectorArgs = [...QueryVectorArgs, number?, number?, number?];
+
 declare interface PgQueryVectorParams extends QueryVectorParams {
     minScore?: number;
     /**
@@ -195,14 +203,14 @@ declare class PgVector extends MastraVector {
     constructor(connectionString: string);
     transformFilter(filter?: VectorFilter): VectorFilter;
     getIndexInfo(indexName: string): Promise<PGIndexStats>;
-    query(...args: ParamsToArgs<PgQueryVectorParams>): Promise<QueryResult[]>;
+    query(...args: ParamsToArgs<PgQueryVectorParams> | PgQueryVectorArgs): Promise<QueryResult[]>;
     upsert(...args: ParamsToArgs<UpsertVectorParams>): Promise<string[]>;
-    createIndex(...args: ParamsToArgs<PgCreateIndexParams>): Promise<void>;
+    createIndex(...args: ParamsToArgs<PgCreateIndexParams> | PgCreateIndexArgs): Promise<void>;
     /**
      * @deprecated This function is deprecated. Use buildIndex instead
      */
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
-    buildIndex(...args: ParamsToArgs<PgDefineIndexParams>): Promise<void>;
+    buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void>;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<PGIndexStats>;
     deleteIndex(indexName: string): Promise<void>;

package/dist/index.cjs

@@ -324,7 +324,11 @@ var PgVector = class extends vector.MastraVector {
     return this.indexCache.get(indexName);
   }
   async query(...args) {
-    const params = this.normalizeArgs("query", args, ["minScore", "ef", "probes"]);
+    const params = this.normalizeArgs("query", args, [
+      "minScore",
+      "ef",
+      "probes"
+    ]);
     const { indexName, queryVector, topK = 10, filter, includeVector = false, minScore = 0, ef, probes } = params;
     const client = await this.pool.connect();
     try {
@@ -395,7 +399,10 @@ var PgVector = class extends vector.MastraVector {
     }
   }
   async createIndex(...args) {
-    const params = this.normalizeArgs("createIndex", args, ["indexConfig", "buildIndex"]);
+    const params = this.normalizeArgs("createIndex", args, [
+      "indexConfig",
+      "buildIndex"
+    ]);
     const { indexName, dimension, metric = "cosine", indexConfig = {}, buildIndex = true } = params;
     const client = await this.pool.connect();
     try {
@@ -439,7 +446,10 @@ var PgVector = class extends vector.MastraVector {
     return this.buildIndex({ indexName, metric, indexConfig });
   }
   async buildIndex(...args) {
-    const params = this.normalizeArgs("buildIndex", args, ["metric", "indexConfig"]);
+    const params = this.normalizeArgs("buildIndex", args, [
+      "metric",
+      "indexConfig"
+    ]);
     const { indexName, metric = "cosine", indexConfig } = params;
     const client = await this.pool.connect();
     try {

package/dist/index.js

@@ -317,7 +317,11 @@ var PgVector = class extends MastraVector {
     return this.indexCache.get(indexName);
   }
   async query(...args) {
-    const params = this.normalizeArgs("query", args, ["minScore", "ef", "probes"]);
+    const params = this.normalizeArgs("query", args, [
+      "minScore",
+      "ef",
+      "probes"
+    ]);
     const { indexName, queryVector, topK = 10, filter, includeVector = false, minScore = 0, ef, probes } = params;
     const client = await this.pool.connect();
     try {
@@ -388,7 +392,10 @@ var PgVector = class extends MastraVector {
     }
   }
   async createIndex(...args) {
-    const params = this.normalizeArgs("createIndex", args, ["indexConfig", "buildIndex"]);
+    const params = this.normalizeArgs("createIndex", args, [
+      "indexConfig",
+      "buildIndex"
+    ]);
     const { indexName, dimension, metric = "cosine", indexConfig = {}, buildIndex = true } = params;
     const client = await this.pool.connect();
     try {
@@ -432,7 +439,10 @@ var PgVector = class extends MastraVector {
     return this.buildIndex({ indexName, metric, indexConfig });
   }
   async buildIndex(...args) {
-    const params = this.normalizeArgs("buildIndex", args, ["metric", "indexConfig"]);
+    const params = this.normalizeArgs("buildIndex", args, [
+      "metric",
+      "indexConfig"
+    ]);
     const { indexName, metric = "cosine", indexConfig } = params;
     const client = await this.pool.connect();
     try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.1.6",
+  "version": "0.1.8-alpha.0",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -21,7 +21,7 @@
   "dependencies": {
     "pg": "^8.13.1",
     "pg-promise": "^11.5.4",
-    "@mastra/core": "^0.4.3"
+    "@mastra/core": "^0.5.0-alpha.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.49.2",

package/src/vector/index.ts

@@ -6,6 +6,8 @@ import type {
   CreateIndexParams,
   UpsertVectorParams,
   ParamsToArgs,
+  QueryVectorArgs,
+  CreateIndexArgs,
 } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 import pg from 'pg';
@@ -38,17 +40,23 @@ interface PgQueryVectorParams extends QueryVectorParams {
   probes?: number;
 }
 
+type PgQueryVectorArgs = [...QueryVectorArgs, number?, number?, number?];
+
 interface PgCreateIndexParams extends CreateIndexParams {
   indexConfig?: IndexConfig;
   buildIndex?: boolean;
 }
 
+type PgCreateIndexArgs = [...CreateIndexArgs, IndexConfig?, boolean?];
+
 interface PgDefineIndexParams {
   indexName: string;
   metric: 'cosine' | 'euclidean' | 'dotproduct';
   indexConfig: IndexConfig;
 }
 
+type PgDefineIndexArgs = [string, 'cosine' | 'euclidean' | 'dotproduct', IndexConfig];
+
 export class PgVector extends MastraVector {
   private pool: pg.Pool;
   private indexCache: Map<string, PGIndexStats> = new Map();
@@ -86,8 +94,12 @@ export class PgVector extends MastraVector {
     return this.indexCache.get(indexName)!;
   }
 
-  async query(...args: ParamsToArgs<PgQueryVectorParams>): Promise<QueryResult[]> {
-    const params = this.normalizeArgs<PgQueryVectorParams>('query', args, ['minScore', 'ef', 'probes']);
+  async query(...args: ParamsToArgs<PgQueryVectorParams> | PgQueryVectorArgs): Promise<QueryResult[]> {
+    const params = this.normalizeArgs<PgQueryVectorParams, PgQueryVectorArgs>('query', args, [
+      'minScore',
+      'ef',
+      'probes',
+    ]);
     const { indexName, queryVector, topK = 10, filter, includeVector = false, minScore = 0, ef, probes } = params;
 
     const client = await this.pool.connect();
@@ -174,8 +186,11 @@ export class PgVector extends MastraVector {
     }
   }
 
-  async createIndex(...args: ParamsToArgs<PgCreateIndexParams>): Promise<void> {
-    const params = this.normalizeArgs<PgCreateIndexParams>('createIndex', args, ['indexConfig', 'buildIndex']);
+  async createIndex(...args: ParamsToArgs<PgCreateIndexParams> | PgCreateIndexArgs): Promise<void> {
+    const params = this.normalizeArgs<PgCreateIndexParams, PgCreateIndexArgs>('createIndex', args, [
+      'indexConfig',
+      'buildIndex',
+    ]);
 
     const { indexName, dimension, metric = 'cosine', indexConfig = {}, buildIndex = true } = params;
 
@@ -235,8 +250,11 @@ export class PgVector extends MastraVector {
     return this.buildIndex({ indexName, metric, indexConfig });
   }
 
-  async buildIndex(...args: ParamsToArgs<PgDefineIndexParams>): Promise<void> {
-    const params = this.normalizeArgs<PgDefineIndexParams>('buildIndex', args, ['metric', 'indexConfig']);
+  async buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void> {
+    const params = this.normalizeArgs<PgDefineIndexParams, PgDefineIndexArgs>('buildIndex', args, [
+      'metric',
+      'indexConfig',
+    ]);
 
     const { indexName, metric = 'cosine', indexConfig } = params;