@mastra/pg 0.17.2 → 0.17.3-alpha.1

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export * from './vector/index.js';
 export * from './storage/index.js';
+export type { PostgresConfig, PostgresStoreConfig, PgVectorConfig } from './shared/config.js';
 export { PGVECTOR_PROMPT } from './vector/prompt.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,WAAW,CAAC;AAC1B,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,WAAW,CAAC;AAC1B,YAAY,EAAE,cAAc,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAC3F,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js

@@ -2,7 +2,7 @@ import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import { MastraVector } from '@mastra/core/vector';
 import { Mutex } from 'async-mutex';
-import pg from 'pg';
+import * as pg from 'pg';
 import xxhash from 'xxhash-wasm';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import { MastraStorage, StoreOperations, TABLE_SCHEMAS, TABLE_WORKFLOW_SNAPSHOT, TABLE_AI_SPANS, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_EVALS, TABLE_SCORERS, ScoresStorage, TracesStorage, safelyParseJSON, WorkflowsStorage, LegacyEvalsStorage, MemoryStorage, resolveMessageLimit, TABLE_RESOURCES, ObservabilityStorage } from '@mastra/core/storage';
@@ -11,6 +11,39 @@ import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/scores';
 
 // 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 BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -367,29 +400,46 @@ var PgVector = class extends 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.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.Pool(poolConfig);
       const telemetry = this.__getTelemetry();
       this.pool = telemetry?.traceClass(basePool, {
         spanNamePrefix: "pg-vector",
@@ -423,7 +473,7 @@ var PgVector = class extends MastraVector {
           domain: ErrorDomain.MASTRA_VECTOR,
           category: ErrorCategory.THIRD_PARTY,
           details: {
-            schemaName: schemaName ?? ""
+            schemaName: "schemaName" in config ? config.schemaName ?? "" : ""
           }
         },
         error
@@ -4052,36 +4102,8 @@ var PostgresStore = class extends 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)) {