@mastra/pg 0.15.2 → 0.15.3-alpha.0

package/dist/index.js

@@ -2800,16 +2800,17 @@ var WorkflowsPG = class extends WorkflowsStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
     try {
       const now = (/* @__PURE__ */ new Date()).toISOString();
       await this.client.none(
-        `INSERT INTO ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} (workflow_name, run_id, snapshot, "createdAt", "updatedAt")
-                 VALUES ($1, $2, $3, $4, $5)
+        `INSERT INTO ${getTableName({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: this.schema })} (workflow_name, run_id, "resourceId", snapshot, "createdAt", "updatedAt")
+                 VALUES ($1, $2, $3, $4, $5, $6)
                  ON CONFLICT (workflow_name, run_id) DO UPDATE
-                 SET snapshot = $3, "updatedAt" = $5`,
-        [workflowName, runId, JSON.stringify(snapshot), now, now]
+                 SET "resourceId" = $3, snapshot = $4, "updatedAt" = $6`,
+        [workflowName, runId, resourceId, JSON.stringify(snapshot), now, now]
       );
     } catch (error) {
       throw new MastraError(
@@ -2971,37 +2972,68 @@ 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 ("connectionString" in config) {
+      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 {
+      } else if (isCloudSqlConfig(config)) ; else if (isHostConfig(config)) {
         const required = ["host", "database", "user", "password"];
         for (const key of required) {
-          if (!(key in config) || typeof config[key] !== "string" || config[key].trim() === "") {
+          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`)."
+        );
       }
       super({ name: "PostgresStore" });
       this.schema = config.schemaName || "public";
-      this.#config = {
-        max: config.max,
-        idleTimeoutMillis: config.idleTimeoutMillis,
-        ...`connectionString` in config ? { connectionString: config.connectionString } : {
+      if (isConnectionStringConfig(config)) {
+        this.#config = {
+          connectionString: config.connectionString,
+          max: config.max,
+          idleTimeoutMillis: config.idleTimeoutMillis
+        };
+      } else if (isCloudSqlConfig(config)) {
+        this.#config = {
+          ...config,
+          max: config.max,
+          idleTimeoutMillis: config.idleTimeoutMillis
+        };
+      } else if (isHostConfig(config)) {
+        this.#config = {
           host: config.host,
           port: config.port,
           database: config.database,
           user: config.user,
           password: config.password,
-          ssl: config.ssl
-        }
-      };
+          ssl: config.ssl,
+          max: config.max,
+          idleTimeoutMillis: config.idleTimeoutMillis
+        };
+      } else {
+        this.#config = {
+          ...config,
+          max: config.max,
+          idleTimeoutMillis: config.idleTimeoutMillis
+        };
+      }
       this.stores = {};
     } catch (e) {
       throw new MastraError(
@@ -3203,9 +3235,10 @@ var PostgresStore = class extends MastraStorage {
   async persistWorkflowSnapshot({
     workflowName,
     runId,
+    resourceId,
     snapshot
   }) {
-    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, snapshot });
+    return this.stores.workflows.persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot });
   }
   async loadWorkflowSnapshot({
     workflowName,