@mastra/pg 1.0.0-beta.4 → 1.0.0-beta.5

package/dist/index.js

@@ -3711,15 +3711,15 @@ var ScoresPG = class extends ScoresStorage {
     } catch (error) {
       throw new MastraError(
         {
-          id: createStorageErrorId("PG", "SAVE_SCORE", "INVALID_PAYLOAD"),
+          id: createStorageErrorId("PG", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: ErrorDomain.STORAGE,
           category: ErrorCategory.USER,
           details: {
-            scorer: score.scorer.id,
-            entityId: score.entityId,
-            entityType: score.entityType,
-            traceId: score.traceId || "",
-            spanId: score.spanId || ""
+            scorer: score.scorer?.id ?? "unknown",
+            entityId: score.entityId ?? "unknown",
+            entityType: score.entityType ?? "unknown",
+            traceId: score.traceId ?? "",
+            spanId: score.spanId ?? ""
           }
         },
         error
@@ -3727,6 +3727,7 @@ var ScoresPG = class extends ScoresStorage {
     }
     try {
       const id = crypto.randomUUID();
+      const now = /* @__PURE__ */ new Date();
       const {
         scorer,
         preprocessStepResult,
@@ -3749,14 +3750,15 @@ var ScoresPG = class extends ScoresStorage {
           scorer: scorer ? JSON.stringify(scorer) : null,
           preprocessStepResult: preprocessStepResult ? JSON.stringify(preprocessStepResult) : null,
           analyzeStepResult: analyzeStepResult ? JSON.stringify(analyzeStepResult) : null,
+          metadata: metadata ? JSON.stringify(metadata) : null,
+          additionalContext: additionalContext ? JSON.stringify(additionalContext) : null,
           requestContext: requestContext ? JSON.stringify(requestContext) : null,
           entity: entity ? JSON.stringify(entity) : null,
-          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+          createdAt: now.toISOString(),
+          updatedAt: now.toISOString()
         }
       });
-      const scoreFromDb = await this.getScoreById({ id });
-      return { score: scoreFromDb };
+      return { score: { ...parsedScore, id, createdAt: now, updatedAt: now } };
     } catch (error) {
       throw new MastraError(
         {
@@ -4138,17 +4140,17 @@ var WorkflowsPG = class extends WorkflowsStorage {
 var PostgresStore = class extends MastraStorage {
   #db;
   #pgp;
-  #config;
   schema;
-  isConnected = false;
+  isInitialized = false;
   stores;
   constructor(config) {
     try {
       validateConfig("PostgresStore", config);
       super({ id: config.id, name: "PostgresStore", disableInit: config.disableInit });
       this.schema = config.schemaName || "public";
+      let pgConfig;
       if (isConnectionStringConfig(config)) {
-        this.#config = {
+        pgConfig = {
           id: config.id,
           connectionString: config.connectionString,
           max: config.max,
@@ -4156,14 +4158,14 @@ var PostgresStore = class extends MastraStorage {
           ssl: config.ssl
         };
       } else if (isCloudSqlConfig(config)) {
-        this.#config = {
+        pgConfig = {
           ...config,
           id: config.id,
           max: config.max,
           idleTimeoutMillis: config.idleTimeoutMillis
         };
       } else if (isHostConfig(config)) {
-        this.#config = {
+        pgConfig = {
           id: config.id,
           host: config.host,
           port: config.port,
@@ -4179,7 +4181,20 @@ var PostgresStore = class extends MastraStorage {
           "PostgresStore: invalid config. Provide either {connectionString}, {host,port,database,user,password}, or a pg ClientConfig (e.g., Cloud SQL connector with `stream`)."
         );
       }
-      this.stores = {};
+      this.#pgp = pgPromise();
+      this.#db = this.#pgp(pgConfig);
+      const operations = new StoreOperationsPG({ client: this.#db, schemaName: this.schema });
+      const scores = new ScoresPG({ client: this.#db, operations, schema: this.schema });
+      const workflows = new WorkflowsPG({ client: this.#db, operations, schema: this.schema });
+      const memory = new MemoryPG({ client: this.#db, schema: this.schema, operations });
+      const observability = new ObservabilityPG({ client: this.#db, operations, schema: this.schema });
+      this.stores = {
+        operations,
+        scores,
+        workflows,
+        memory,
+        observability
+      };
     } catch (e) {
       throw new MastraError(
         {
@@ -4192,33 +4207,19 @@ var PostgresStore = class extends MastraStorage {
     }
   }
   async init() {
-    if (this.isConnected) {
+    if (this.isInitialized) {
       return;
     }
     try {
-      this.isConnected = true;
-      this.#pgp = pgPromise();
-      this.#db = this.#pgp(this.#config);
-      const operations = new StoreOperationsPG({ client: this.#db, schemaName: this.schema });
-      const scores = new ScoresPG({ client: this.#db, operations, schema: this.schema });
-      const workflows = new WorkflowsPG({ client: this.#db, operations, schema: this.schema });
-      const memory = new MemoryPG({ client: this.#db, schema: this.schema, operations });
-      const observability = new ObservabilityPG({ client: this.#db, operations, schema: this.schema });
-      this.stores = {
-        operations,
-        scores,
-        workflows,
-        memory,
-        observability
-      };
+      this.isInitialized = true;
       await super.init();
       try {
-        await operations.createAutomaticIndexes();
+        await this.stores.operations.createAutomaticIndexes();
       } catch (indexError) {
         console.warn("Failed to create indexes:", indexError);
       }
     } catch (error) {
-      this.isConnected = false;
+      this.isInitialized = false;
       throw new MastraError(
         {
           id: createStorageErrorId("PG", "INIT", "FAILED"),
@@ -4230,15 +4231,9 @@ var PostgresStore = class extends MastraStorage {
     }
   }
   get db() {
-    if (!this.#db) {
-      throw new Error(`PostgresStore: Store is not initialized, please call "init()" first.`);
-    }
     return this.#db;
   }
   get pgp() {
-    if (!this.#pgp) {
-      throw new Error(`PostgresStore: Store is not initialized, please call "init()" first.`);
-    }
     return this.#pgp;
   }
   get supports() {