@mastra/pg 0.3.1-alpha.2 → 0.3.1-alpha.3

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.3.1-alpha.2 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.3.1-alpha.3 build /home/runner/work/mastra/mastra/stores/pg
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 9212ms
+TSC ⚡️ Build success in 11190ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 11600ms
+DTS ⚡️ Build success in 11666ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 48.63 KB
-CJS ⚡️ Build success in 1328ms
-ESM dist/index.js 48.21 KB
-ESM ⚡️ Build success in 1328ms
+ESM dist/index.js 48.40 KB
+ESM ⚡️ Build success in 1450ms
+CJS dist/index.cjs 48.82 KB
+CJS ⚡️ Build success in 1450ms

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/pg
 
+## 0.3.1-alpha.3
+
+### Patch Changes
+
+- 6a76949: add PgVector pgPoolOptions constructor parameter
+- Updated dependencies [6262bd5]
+  - @mastra/core@0.9.1-alpha.3
+
 ## 0.3.1-alpha.2
 
 ### Patch Changes

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

@@ -14,6 +14,7 @@ import type { MessageType } from '@mastra/core/memory';
 import type { NumericOperator } from '@mastra/core/vector/filter';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
 import type { ParamsToArgs } from '@mastra/core/vector';
+import pg from 'pg';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorArgs } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
@@ -212,6 +213,7 @@ declare class PgVector extends MastraVector {
     constructor(config: {
         connectionString: string;
         schemaName?: string;
+        pgPoolOptions?: Omit<pg.PoolConfig, 'connectionString'>;
     });
     private getMutexByName;
     private getTableName;

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

@@ -14,6 +14,7 @@ import type { MessageType } from '@mastra/core/memory';
 import type { NumericOperator } from '@mastra/core/vector/filter';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
 import type { ParamsToArgs } from '@mastra/core/vector';
+import pg from 'pg';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorArgs } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
@@ -212,6 +213,7 @@ declare class PgVector extends MastraVector {
     constructor(config: {
         connectionString: string;
         schemaName?: string;
+        pgPoolOptions?: Omit<pg.PoolConfig, 'connectionString'>;
     });
     private getMutexByName;
     private getTableName;

package/dist/index.cjs

@@ -342,22 +342,34 @@ var PgVector = class extends vector.MastraVector {
   vectorExtensionInstalled = void 0;
   schemaSetupComplete = void 0;
   constructor(config) {
-    const connectionString = typeof config === "string" ? config : config.connectionString;
-    if (!connectionString || typeof connectionString !== "string" || connectionString.trim() === "") {
+    let connectionString;
+    let pgPoolOptions;
+    let schemaName;
+    if (typeof config === "string") {
+      connectionString = config;
+      schemaName = void 0;
+      pgPoolOptions = void 0;
+    } else {
+      connectionString = config.connectionString;
+      schemaName = config.schemaName;
+      pgPoolOptions = config.pgPoolOptions;
+    }
+    if (!connectionString || connectionString.trim() === "") {
       throw new Error(
         "PgVector: connectionString must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults."
       );
     }
     super();
-    this.schema = typeof config === "string" ? void 0 : config.schemaName;
+    this.schema = schemaName;
     const basePool = new pg__default.default.Pool({
       connectionString,
       max: 20,
       // Maximum number of clients in the pool
       idleTimeoutMillis: 3e4,
       // Close idle connections after 30 seconds
-      connectionTimeoutMillis: 2e3
+      connectionTimeoutMillis: 2e3,
       // Fail fast if can't connect
+      ...pgPoolOptions
     });
     const telemetry = this.__getTelemetry();
     this.pool = telemetry?.traceClass(basePool, {

package/dist/index.js

@@ -334,22 +334,34 @@ var PgVector = class extends MastraVector {
   vectorExtensionInstalled = void 0;
   schemaSetupComplete = void 0;
   constructor(config) {
-    const connectionString = typeof config === "string" ? config : config.connectionString;
-    if (!connectionString || typeof connectionString !== "string" || connectionString.trim() === "") {
+    let connectionString;
+    let pgPoolOptions;
+    let schemaName;
+    if (typeof config === "string") {
+      connectionString = config;
+      schemaName = void 0;
+      pgPoolOptions = void 0;
+    } else {
+      connectionString = config.connectionString;
+      schemaName = config.schemaName;
+      pgPoolOptions = config.pgPoolOptions;
+    }
+    if (!connectionString || connectionString.trim() === "") {
       throw new Error(
         "PgVector: connectionString must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults."
       );
     }
     super();
-    this.schema = typeof config === "string" ? void 0 : config.schemaName;
+    this.schema = schemaName;
     const basePool = new pg.Pool({
       connectionString,
       max: 20,
       // Maximum number of clients in the pool
       idleTimeoutMillis: 3e4,
       // Close idle connections after 30 seconds
-      connectionTimeoutMillis: 2e3
+      connectionTimeoutMillis: 2e3,
       // Fail fast if can't connect
+      ...pgPoolOptions
     });
     const telemetry = this.__getTelemetry();
     this.pool = telemetry?.traceClass(basePool, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.3.1-alpha.2",
+  "version": "0.3.1-alpha.3",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -24,7 +24,7 @@
     "pg": "^8.13.3",
     "pg-promise": "^11.11.0",
     "xxhash-wasm": "^1.1.0",
-    "@mastra/core": "^0.9.1-alpha.2"
+    "@mastra/core": "^0.9.1-alpha.3"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",

package/src/vector/index.test.ts

@@ -2436,4 +2436,45 @@ describe('PgVector', () => {
       });
     });
   });
+
+  describe('PoolConfig Custom Options', () => {
+    it('should apply custom values to properties with default values', async () => {
+      const db = new PgVector({
+        connectionString,
+        pgPoolOptions: {
+          max: 5,
+          idleTimeoutMillis: 10000,
+          connectionTimeoutMillis: 1000,
+        },
+      });
+
+      expect(db['pool'].options.max).toBe(5);
+      expect(db['pool'].options.idleTimeoutMillis).toBe(10000);
+      expect(db['pool'].options.connectionTimeoutMillis).toBe(1000);
+    });
+
+    it('should pass properties with no default values', async () => {
+      const db = new PgVector({
+        connectionString,
+        pgPoolOptions: {
+          ssl: false,
+        },
+      });
+
+      expect(db['pool'].options.ssl).toBe(false);
+    });
+    it('should keep default values when custom values are added', async () => {
+      const db = new PgVector({
+        connectionString,
+        pgPoolOptions: {
+          ssl: false,
+        },
+      });
+
+      expect(db['pool'].options.max).toBe(20);
+      expect(db['pool'].options.idleTimeoutMillis).toBe(30000);
+      expect(db['pool'].options.connectionTimeoutMillis).toBe(2000);
+      expect(db['pool'].options.ssl).toBe(false);
+    });
+  });
 });

package/src/vector/index.ts

@@ -71,23 +71,49 @@ export class PgVector extends MastraVector {
   private schemaSetupComplete: boolean | undefined = undefined;
 
   constructor(connectionString: string);
-  constructor(config: { connectionString: string; schemaName?: string });
-  constructor(config: string | { connectionString: string; schemaName?: string }) {
-    const connectionString = typeof config === 'string' ? config : config.connectionString;
-    if (!connectionString || typeof connectionString !== 'string' || connectionString.trim() === '') {
+  constructor(config: {
+    connectionString: string;
+    schemaName?: string;
+    pgPoolOptions?: Omit<pg.PoolConfig, 'connectionString'>;
+  });
+  constructor(
+    config:
+      | string
+      | {
+          connectionString: string;
+          schemaName?: string;
+          pgPoolOptions?: Omit<pg.PoolConfig, 'connectionString'>;
+        },
+  ) {
+    let connectionString: string;
+    let pgPoolOptions: Omit<pg.PoolConfig, 'connectionString'> | undefined;
+    let schemaName: string | undefined;
+
+    if (typeof config === 'string') {
+      connectionString = config;
+      schemaName = undefined;
+      pgPoolOptions = undefined;
+    } else {
+      connectionString = config.connectionString;
+      schemaName = config.schemaName;
+      pgPoolOptions = config.pgPoolOptions;
+    }
+
+    if (!connectionString || connectionString.trim() === '') {
       throw new Error(
         'PgVector: connectionString must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults.',
       );
     }
     super();
 
-    this.schema = typeof config === 'string' ? undefined : config.schemaName;
+    this.schema = schemaName;
 
     const basePool = new pg.Pool({
       connectionString,
       max: 20, // Maximum number of clients in the pool
       idleTimeoutMillis: 30000, // Close idle connections after 30 seconds
       connectionTimeoutMillis: 2000, // Fail fast if can't connect
+      ...pgPoolOptions,
     });
 
     const telemetry = this.__getTelemetry();