@mastra/pg 0.17.2 → 0.17.3-alpha.1

package/CHANGELOG.md

@@ -1,5 +1,34 @@
 # @mastra/pg
 
+## 0.17.3-alpha.1
+
+### Patch Changes
+
+- feat(pg): add flexible PostgreSQL configuration with shared types ([#8103](https://github.com/mastra-ai/mastra/pull/8103))
+  - Add support for multiple connection methods: connectionString, host/port/database, and Cloud SQL
+  - Introduce shared PostgresConfig type with generic SSL support (ISSLConfig for pg-promise, ConnectionOptions for pg)
+  - Add pgPoolOptions support to PgVector for advanced pool configuration
+  - Create shared validation helpers to reduce code duplication
+  - Maintain backward compatibility with existing configurations
+
+- Updated dependencies [[`1ed9670`](https://github.com/mastra-ai/mastra/commit/1ed9670d3ca50cb60dc2e517738c5eef3968ed27), [`158381d`](https://github.com/mastra-ai/mastra/commit/158381d39335be934b81ef8a1947bccace492c25), [`fb703b9`](https://github.com/mastra-ai/mastra/commit/fb703b9634eeaff1a6eb2b5531ce0f9e8fb04727), [`37a2314`](https://github.com/mastra-ai/mastra/commit/37a23148e0e5a3b40d4f9f098b194671a8a49faf), [`05a9dee`](https://github.com/mastra-ai/mastra/commit/05a9dee3d355694d28847bfffb6289657fcf7dfa), [`e3c1077`](https://github.com/mastra-ai/mastra/commit/e3c107763aedd1643d3def5df450c235da9ff76c), [`1bccdb3`](https://github.com/mastra-ai/mastra/commit/1bccdb33eb90cbeba2dc5ece1c2561fb774b26b6), [`5ef944a`](https://github.com/mastra-ai/mastra/commit/5ef944a3721d93105675cac2b2311432ff8cc393), [`d6b186f`](https://github.com/mastra-ai/mastra/commit/d6b186fb08f1caf1b86f73d3a5ee88fb999ca3be), [`65493b3`](https://github.com/mastra-ai/mastra/commit/65493b31c36f6fdb78f9679f7e1ecf0c250aa5ee), [`a998b8f`](https://github.com/mastra-ai/mastra/commit/a998b8f858091c2ec47683e60766cf12d03001e4), [`8a37bdd`](https://github.com/mastra-ai/mastra/commit/8a37bddb6d8614a32c5b70303d583d80c620ea61)]:
+  - @mastra/core@0.21.0-alpha.1
+
+## 0.17.3-alpha.0
+
+### Patch Changes
+
+- Update peer dependencies to match core package version bump (0.21.0) ([#8619](https://github.com/mastra-ai/mastra/pull/8619))
+
+- Update peer dependencies to match core package version bump (0.21.0) ([#8557](https://github.com/mastra-ai/mastra/pull/8557))
+
+- Update peer dependencies to match core package version bump (0.21.0) ([#8626](https://github.com/mastra-ai/mastra/pull/8626))
+
+- Update peer dependencies to match core package version bump (0.21.0) ([#8686](https://github.com/mastra-ai/mastra/pull/8686))
+
+- Updated dependencies [[`b5a66b7`](https://github.com/mastra-ai/mastra/commit/b5a66b748a14fc8b3f63b04642ddb9621fbcc9e0), [`7b1ef57`](https://github.com/mastra-ai/mastra/commit/7b1ef57fc071c2aa2a2e32905b18cd88719c5a39), [`ee68e82`](https://github.com/mastra-ai/mastra/commit/ee68e8289ea4408d29849e899bc6e78b3bd4e843), [`228228b`](https://github.com/mastra-ai/mastra/commit/228228b0b1de9291cb8887587f5cea1a8757ebad), [`ea33930`](https://github.com/mastra-ai/mastra/commit/ea339301e82d6318257720d811b043014ee44064), [`b5a66b7`](https://github.com/mastra-ai/mastra/commit/b5a66b748a14fc8b3f63b04642ddb9621fbcc9e0), [`135d6f2`](https://github.com/mastra-ai/mastra/commit/135d6f22a326ed1dffff858700669dff09d2c9eb), [`59d036d`](https://github.com/mastra-ai/mastra/commit/59d036d4c2706b430b0e3f1f1e0ee853ce16ca04)]:
+  - @mastra/core@0.21.0-alpha.0
+
 ## 0.17.2
 
 ### Patch Changes

package/dist/index.cjs

@@ -14,11 +14,62 @@ var scores = require('@mastra/core/scores');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
-var pg__default = /*#__PURE__*/_interopDefault(pg);
+function _interopNamespace(e) {
+  if (e && e.__esModule) return e;
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
+        });
+      }
+    });
+  }
+  n.default = e;
+  return Object.freeze(n);
+}
+
+var pg__namespace = /*#__PURE__*/_interopNamespace(pg);
 var xxhash__default = /*#__PURE__*/_interopDefault(xxhash);
 var pgPromise__default = /*#__PURE__*/_interopDefault(pgPromise);
 
 // src/vector/index.ts
+
+// src/shared/config.ts
+var isConnectionStringConfig = (cfg) => {
+  return "connectionString" in cfg;
+};
+var isHostConfig = (cfg) => {
+  return "host" in cfg && "database" in cfg && "user" in cfg && "password" in cfg;
+};
+var isCloudSqlConfig = (cfg) => {
+  return "stream" in cfg || "password" in cfg && typeof cfg.password === "function";
+};
+var validateConfig = (name, config) => {
+  if (isConnectionStringConfig(config)) {
+    if (!config.connectionString || typeof config.connectionString !== "string" || config.connectionString.trim() === "") {
+      throw new Error(
+        `${name}: connectionString must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults.`
+      );
+    }
+  } else if (isCloudSqlConfig(config)) ; else if (isHostConfig(config)) {
+    const required = ["host", "database", "user", "password"];
+    for (const key of required) {
+      if (!config[key] || typeof config[key] !== "string" || config[key].trim() === "") {
+        throw new Error(
+          `${name}: ${key} must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults.`
+        );
+      }
+    }
+  } else {
+    throw new Error(
+      `${name}: invalid config. Provide either {connectionString}, {host,port,database,user,password}, or a pg ClientConfig (e.g., Cloud SQL connector with \`stream\`).`
+    );
+  }
+};
 var PGFilterTranslator = class extends filter.BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -375,29 +426,46 @@ var PgVector = class extends vector.MastraVector {
   vectorExtensionSchema = null;
   schemaSetupComplete = void 0;
   cacheWarmupPromise = null;
-  constructor({
-    connectionString,
-    schemaName,
-    pgPoolOptions
-  }) {
+  constructor(config) {
     try {
-      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."
-        );
-      }
+      validateConfig("PgVector", config);
       super();
-      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,
-        // Fail fast if can't connect
-        ...pgPoolOptions
-      });
+      this.schema = config.schemaName;
+      let poolConfig;
+      if (isConnectionStringConfig(config)) {
+        poolConfig = {
+          connectionString: config.connectionString,
+          ssl: config.ssl,
+          max: config.max ?? 20,
+          idleTimeoutMillis: config.idleTimeoutMillis ?? 3e4,
+          connectionTimeoutMillis: 2e3,
+          ...config.pgPoolOptions
+        };
+      } else if (isCloudSqlConfig(config)) {
+        poolConfig = {
+          ...config,
+          max: config.max ?? 20,
+          idleTimeoutMillis: config.idleTimeoutMillis ?? 3e4,
+          connectionTimeoutMillis: 2e3,
+          ...config.pgPoolOptions
+        };
+      } else if (isHostConfig(config)) {
+        poolConfig = {
+          host: config.host,
+          port: config.port,
+          database: config.database,
+          user: config.user,
+          password: config.password,
+          ssl: config.ssl,
+          max: config.max ?? 20,
+          idleTimeoutMillis: config.idleTimeoutMillis ?? 3e4,
+          connectionTimeoutMillis: 2e3,
+          ...config.pgPoolOptions
+        };
+      } else {
+        throw new Error("PgVector: invalid configuration provided");
+      }
+      const basePool = new pg__namespace.Pool(poolConfig);
       const telemetry = this.__getTelemetry();
       this.pool = telemetry?.traceClass(basePool, {
         spanNamePrefix: "pg-vector",
@@ -431,7 +499,7 @@ var PgVector = class extends vector.MastraVector {
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
-            schemaName: schemaName ?? ""
+            schemaName: "schemaName" in config ? config.schemaName ?? "" : ""
           }
         },
         error$1
@@ -4060,36 +4128,8 @@ var PostgresStore = class extends storage.MastraStorage {
   isConnected = false;
   stores;
   constructor(config) {
-    const isConnectionStringConfig = (cfg) => {
-      return "connectionString" in cfg;
-    };
-    const isHostConfig = (cfg) => {
-      return "host" in cfg && "database" in cfg && "user" in cfg && "password" in cfg;
-    };
-    const isCloudSqlConfig = (cfg) => {
-      return "stream" in cfg || "password" in cfg && typeof cfg.password === "function";
-    };
     try {
-      if (isConnectionStringConfig(config)) {
-        if (!config.connectionString || typeof config.connectionString !== "string" || config.connectionString.trim() === "") {
-          throw new Error(
-            "PostgresStore: connectionString must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults."
-          );
-        }
-      } else if (isCloudSqlConfig(config)) ; else if (isHostConfig(config)) {
-        const required = ["host", "database", "user", "password"];
-        for (const key of required) {
-          if (!config[key] || typeof config[key] !== "string" || config[key].trim() === "") {
-            throw new Error(
-              `PostgresStore: ${key} must be provided and cannot be empty. Passing an empty string may cause fallback to local Postgres defaults.`
-            );
-          }
-        }
-      } else {
-        throw new Error(
-          "PostgresStore: invalid config. Provide either {connectionString}, {host,port,database,user,password}, or a pg ClientConfig (e.g., Cloud SQL connector with `stream`)."
-        );
-      }
+      validateConfig("PostgresStore", config);
       super({ name: "PostgresStore" });
       this.schema = config.schemaName || "public";
       if (isConnectionStringConfig(config)) {