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

package/dist/index.cjs

@@ -1,13 +1,13 @@
 'use strict';
 
 var error = require('@mastra/core/error');
+var storage = require('@mastra/core/storage');
 var utils = require('@mastra/core/utils');
 var vector = require('@mastra/core/vector');
 var asyncMutex = require('async-mutex');
 var pg = require('pg');
 var xxhash = require('xxhash-wasm');
 var filter = require('@mastra/core/vector/filter');
-var storage = require('@mastra/core/storage');
 var pgPromise = require('pg-promise');
 var agent = require('@mastra/core/agent');
 var evals = require('@mastra/core/evals');
@@ -576,7 +576,7 @@ var PgVector = class extends vector.MastraVector {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_INITIALIZATION_FAILED",
+          id: storage.createVectorErrorId("PG", "INITIALIZATION", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -670,7 +670,7 @@ var PgVector = class extends vector.MastraVector {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_QUERY_INVALID_INPUT",
+          id: storage.createVectorErrorId("PG", "QUERY", "INVALID_INPUT"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.USER,
           details: {
@@ -726,7 +726,7 @@ var PgVector = class extends vector.MastraVector {
       await client.query("ROLLBACK");
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_QUERY_FAILED",
+          id: storage.createVectorErrorId("PG", "QUERY", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -795,7 +795,7 @@ var PgVector = class extends vector.MastraVector {
           const [, expected, actual] = match;
           const mastraError2 = new error.MastraError(
             {
-              id: "MASTRA_STORAGE_PG_VECTOR_UPSERT_INVALID_INPUT",
+              id: storage.createVectorErrorId("PG", "UPSERT", "INVALID_INPUT"),
               domain: error.ErrorDomain.MASTRA_VECTOR,
               category: error.ErrorCategory.USER,
               text: `Vector dimension mismatch: Index "${indexName}" expects ${expected} dimensions but got ${actual} dimensions. Either use a matching embedding model or delete and recreate the index with the new dimension.`,
@@ -813,7 +813,7 @@ var PgVector = class extends vector.MastraVector {
       }
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_UPSERT_FAILED",
+          id: storage.createVectorErrorId("PG", "UPSERT", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -901,7 +901,7 @@ var PgVector = class extends vector.MastraVector {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_CREATE_INDEX_INVALID_INPUT",
+          id: storage.createVectorErrorId("PG", "CREATE_INDEX", "INVALID_INPUT"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.USER,
           details: {
@@ -951,7 +951,7 @@ var PgVector = class extends vector.MastraVector {
     }).catch((error$1) => {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_CREATE_INDEX_FAILED",
+          id: storage.createVectorErrorId("PG", "CREATE_INDEX", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -971,7 +971,7 @@ var PgVector = class extends vector.MastraVector {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_BUILD_INDEX_FAILED",
+          id: storage.createVectorErrorId("PG", "BUILD_INDEX", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1171,7 +1171,7 @@ var PgVector = class extends vector.MastraVector {
     } catch (e) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_LIST_INDEXES_FAILED",
+          id: storage.createVectorErrorId("PG", "LIST_INDEXES", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -1260,7 +1260,7 @@ var PgVector = class extends vector.MastraVector {
       await client.query("ROLLBACK");
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_DESCRIBE_INDEX_FAILED",
+          id: storage.createVectorErrorId("PG", "DESCRIBE_INDEX", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1285,7 +1285,7 @@ var PgVector = class extends vector.MastraVector {
       await client.query("ROLLBACK");
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_DELETE_INDEX_FAILED",
+          id: storage.createVectorErrorId("PG", "DELETE_INDEX", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1309,7 +1309,7 @@ var PgVector = class extends vector.MastraVector {
       await client.query("ROLLBACK");
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_TRUNCATE_INDEX_FAILED",
+          id: storage.createVectorErrorId("PG", "TRUNCATE_INDEX", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1351,7 +1351,7 @@ var PgVector = class extends vector.MastraVector {
       }
       if (!id && !filter) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_PG_VECTOR_UPDATE_MISSING_PARAMS",
+          id: storage.createVectorErrorId("PG", "UPDATE_VECTOR", "NO_TARGET"),
           text: "Either id or filter must be provided",
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.USER,
@@ -1360,7 +1360,7 @@ var PgVector = class extends vector.MastraVector {
       }
       if (id && filter) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_PG_VECTOR_UPDATE_CONFLICTING_PARAMS",
+          id: storage.createVectorErrorId("PG", "UPDATE_VECTOR", "MUTUALLY_EXCLUSIVE"),
           text: "Cannot provide both id and filter - they are mutually exclusive",
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.USER,
@@ -1394,7 +1394,7 @@ var PgVector = class extends vector.MastraVector {
       } else {
         if (!filter || Object.keys(filter).length === 0) {
           throw new error.MastraError({
-            id: "MASTRA_STORAGE_PG_VECTOR_UPDATE_EMPTY_FILTER",
+            id: storage.createVectorErrorId("PG", "UPDATE_VECTOR", "EMPTY_FILTER"),
             text: "Cannot update with empty filter",
             domain: error.ErrorDomain.MASTRA_VECTOR,
             category: error.ErrorCategory.USER,
@@ -1406,7 +1406,7 @@ var PgVector = class extends vector.MastraVector {
         whereClause = filterQuery.trim().replace(/^WHERE\s+/i, "");
         if (!whereClause) {
           throw new error.MastraError({
-            id: "MASTRA_STORAGE_PG_VECTOR_UPDATE_INVALID_FILTER",
+            id: storage.createVectorErrorId("PG", "UPDATE_VECTOR", "INVALID_FILTER"),
             text: "Filter produced empty WHERE clause",
             domain: error.ErrorDomain.MASTRA_VECTOR,
             category: error.ErrorCategory.USER,
@@ -1437,7 +1437,7 @@ var PgVector = class extends vector.MastraVector {
       }
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_UPDATE_VECTOR_FAILED",
+          id: storage.createVectorErrorId("PG", "UPDATE_VECTOR", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1474,7 +1474,7 @@ var PgVector = class extends vector.MastraVector {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_DELETE_VECTOR_FAILED",
+          id: storage.createVectorErrorId("PG", "DELETE_VECTOR", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1504,7 +1504,7 @@ var PgVector = class extends vector.MastraVector {
       const { tableName } = this.getTableName(indexName);
       if (!filter && !ids) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_PG_VECTOR_DELETE_MISSING_PARAMS",
+          id: storage.createVectorErrorId("PG", "DELETE_VECTORS", "NO_TARGET"),
           text: "Either filter or ids must be provided",
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.USER,
@@ -1513,7 +1513,7 @@ var PgVector = class extends vector.MastraVector {
       }
       if (filter && ids) {
         throw new error.MastraError({
-          id: "MASTRA_STORAGE_PG_VECTOR_DELETE_CONFLICTING_PARAMS",
+          id: storage.createVectorErrorId("PG", "DELETE_VECTORS", "MUTUALLY_EXCLUSIVE"),
           text: "Cannot provide both filter and ids - they are mutually exclusive",
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.USER,
@@ -1525,7 +1525,7 @@ var PgVector = class extends vector.MastraVector {
       if (ids) {
         if (ids.length === 0) {
           throw new error.MastraError({
-            id: "MASTRA_STORAGE_PG_VECTOR_DELETE_EMPTY_IDS",
+            id: storage.createVectorErrorId("PG", "DELETE_VECTORS", "EMPTY_IDS"),
             text: "Cannot delete with empty ids array",
             domain: error.ErrorDomain.MASTRA_VECTOR,
             category: error.ErrorCategory.USER,
@@ -1538,7 +1538,7 @@ var PgVector = class extends vector.MastraVector {
       } else {
         if (!filter || Object.keys(filter).length === 0) {
           throw new error.MastraError({
-            id: "MASTRA_STORAGE_PG_VECTOR_DELETE_EMPTY_FILTER",
+            id: storage.createVectorErrorId("PG", "DELETE_VECTORS", "EMPTY_FILTER"),
             text: "Cannot delete with empty filter. Use deleteIndex to delete all vectors.",
             domain: error.ErrorDomain.MASTRA_VECTOR,
             category: error.ErrorCategory.USER,
@@ -1550,7 +1550,7 @@ var PgVector = class extends vector.MastraVector {
         const whereClause = filterQuery.trim().replace(/^WHERE\s+/i, "");
         if (!whereClause) {
           throw new error.MastraError({
-            id: "MASTRA_STORAGE_PG_VECTOR_DELETE_INVALID_FILTER",
+            id: storage.createVectorErrorId("PG", "DELETE_VECTORS", "INVALID_FILTER"),
             text: "Filter produced empty WHERE clause",
             domain: error.ErrorDomain.MASTRA_VECTOR,
             category: error.ErrorCategory.USER,
@@ -1573,7 +1573,7 @@ var PgVector = class extends vector.MastraVector {
       }
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_VECTOR_DELETE_VECTORS_FAILED",
+          id: storage.createVectorErrorId("PG", "DELETE_VECTORS", "FAILED"),
           domain: error.ErrorDomain.MASTRA_VECTOR,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1713,7 +1713,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_THREAD_BY_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_THREAD_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1728,7 +1728,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
     if (page < 0) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_PG_STORE_INVALID_PAGE",
+        id: storage.createStorageErrorId("PG", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
         text: "Page number must be non-negative",
@@ -1777,7 +1777,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1835,7 +1835,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_SAVE_THREAD_FAILED",
+          id: storage.createStorageErrorId("PG", "SAVE_THREAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1855,7 +1855,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     const existingThread = await this.getThreadById({ threadId: id });
     if (!existingThread) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_PG_STORE_UPDATE_THREAD_FAILED",
+        id: storage.createStorageErrorId("PG", "UPDATE_THREAD", "FAILED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
         text: `Thread ${id} not found`,
@@ -1893,7 +1893,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_UPDATE_THREAD_FAILED",
+          id: storage.createStorageErrorId("PG", "UPDATE_THREAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -1930,7 +1930,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_DELETE_THREAD_FAILED",
+          id: storage.createStorageErrorId("PG", "DELETE_THREAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2037,7 +2037,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_LIST_MESSAGES_BY_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "LIST_MESSAGES_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2059,7 +2059,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new error.MastraError(
         {
-          id: "STORAGE_PG_LIST_MESSAGES_INVALID_THREAD_ID",
+          id: storage.createStorageErrorId("PG", "LIST_MESSAGES", "INVALID_THREAD_ID"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { threadId: Array.isArray(threadId) ? String(threadId) : String(threadId) }
@@ -2069,7 +2069,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     }
     if (page < 0) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_PG_STORE_INVALID_PAGE",
+        id: storage.createStorageErrorId("PG", "LIST_MESSAGES", "INVALID_PAGE"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
         text: "Page number must be non-negative",
@@ -2164,7 +2164,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       const mastraError = new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_LIST_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("PG", "LIST_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2190,7 +2190,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     const threadId = messages[0]?.threadId;
     if (!threadId) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_PG_STORE_SAVE_MESSAGES_FAILED",
+        id: storage.createStorageErrorId("PG", "SAVE_MESSAGES", "FAILED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.THIRD_PARTY,
         text: `Thread ID is required`
@@ -2199,7 +2199,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     const thread = await this.getThreadById({ threadId });
     if (!thread) {
       throw new error.MastraError({
-        id: "MASTRA_STORAGE_PG_STORE_SAVE_MESSAGES_FAILED",
+        id: storage.createStorageErrorId("PG", "SAVE_MESSAGES", "FAILED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.THIRD_PARTY,
         text: `Thread ${threadId} not found`,
@@ -2270,7 +2270,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_SAVE_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("PG", "SAVE_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2399,7 +2399,7 @@ var MemoryPG = class extends storage.MemoryStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_DELETE_MESSAGES_FAILED",
+          id: storage.createStorageErrorId("PG", "DELETE_MESSAGES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: { messageIds: messageIds.join(", ") }
@@ -2520,7 +2520,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_CREATE_SPAN_FAILED",
+          id: storage.createStorageErrorId("PG", "CREATE_SPAN", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -2566,7 +2566,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_GET_TRACE_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_TRACE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -2598,7 +2598,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_UPDATE_SPAN_FAILED",
+          id: storage.createStorageErrorId("PG", "UPDATE_SPAN", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -2634,7 +2634,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
         name = `agent run: '${entityId}'`;
       } else {
         const error$1 = new error.MastraError({
-          id: "PG_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_TRACES_PAGINATED", "INVALID_ENTITY_TYPE"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER,
           details: {
@@ -2702,7 +2702,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_GET_TRACES_PAGINATED_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_TRACES_PAGINATED", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -2731,7 +2731,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_BATCH_CREATE_SPANS_FAILED",
+          id: storage.createStorageErrorId("PG", "BATCH_CREATE_SPANS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -2766,7 +2766,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_BATCH_UPDATE_SPANS_FAILED",
+          id: storage.createStorageErrorId("PG", "BATCH_UPDATE_SPANS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -2785,7 +2785,7 @@ var ObservabilityPG = class extends storage.ObservabilityStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "PG_STORE_BATCH_DELETE_TRACES_FAILED",
+          id: storage.createStorageErrorId("PG", "BATCH_DELETE_TRACES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -2915,7 +2915,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_INSERT_FAILED",
+          id: storage.createStorageErrorId("PG", "INSERT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -2934,7 +2934,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_CLEAR_TABLE_FAILED",
+          id: storage.createStorageErrorId("PG", "CLEAR_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3007,7 +3007,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_CREATE_TABLE_FAILED",
+          id: storage.createStorageErrorId("PG", "CREATE_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3094,7 +3094,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_ALTER_TABLE_FAILED",
+          id: storage.createStorageErrorId("PG", "ALTER_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3128,7 +3128,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_LOAD_FAILED",
+          id: storage.createStorageErrorId("PG", "LOAD", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3150,7 +3150,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
       await this.client.query("ROLLBACK");
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_BATCH_INSERT_FAILED",
+          id: storage.createStorageErrorId("PG", "BATCH_INSERT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3170,7 +3170,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_DROP_TABLE_FAILED",
+          id: storage.createStorageErrorId("PG", "DROP_TABLE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3243,7 +3243,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
       }
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_INDEX_CREATE_FAILED",
+          id: storage.createStorageErrorId("PG", "INDEX_CREATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3275,7 +3275,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_INDEX_DROP_FAILED",
+          id: storage.createStorageErrorId("PG", "INDEX_DROP", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3351,7 +3351,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_INDEX_LIST_FAILED",
+          id: storage.createStorageErrorId("PG", "INDEX_LIST", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: tableName ? {
@@ -3433,7 +3433,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_CREATE_PERFORMANCE_INDEXES_FAILED",
+          id: storage.createStorageErrorId("PG", "CREATE_AUTOMATIC_INDEXES", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3495,7 +3495,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_INDEX_DESCRIBE_FAILED",
+          id: storage.createStorageErrorId("PG", "INDEX_DESCRIBE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3540,7 +3540,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_UPDATE_FAILED",
+          id: storage.createStorageErrorId("PG", "UPDATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3568,7 +3568,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
       await this.client.query("ROLLBACK");
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_BATCH_UPDATE_FAILED",
+          id: storage.createStorageErrorId("PG", "BATCH_UPDATE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3609,7 +3609,7 @@ var StoreOperationsPG = class extends storage.StoreOperations {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_BATCH_DELETE_FAILED",
+          id: storage.createStorageErrorId("PG", "BATCH_DELETE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -3654,7 +3654,7 @@ var ScoresPG = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_SCORE_BY_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_SCORE_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3722,7 +3722,7 @@ var ScoresPG = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_SCORES_BY_SCORER_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_SCORES_BY_SCORER_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3737,15 +3737,15 @@ var ScoresPG = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_SAVE_SCORE_FAILED_INVALID_SCORE_PAYLOAD",
+          id: storage.createStorageErrorId("PG", "SAVE_SCORE", "VALIDATION_FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.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$1
@@ -3753,6 +3753,7 @@ var ScoresPG = class extends storage.ScoresStorage {
     }
     try {
       const id = crypto.randomUUID();
+      const now = /* @__PURE__ */ new Date();
       const {
         scorer,
         preprocessStepResult,
@@ -3775,18 +3776,19 @@ var ScoresPG = class extends storage.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$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_SAVE_SCORE_FAILED",
+          id: storage.createStorageErrorId("PG", "SAVE_SCORE", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3835,7 +3837,7 @@ var ScoresPG = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_SCORES_BY_RUN_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_SCORES_BY_RUN_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3885,7 +3887,7 @@ var ScoresPG = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_SCORES_BY_ENTITY_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_SCORES_BY_ENTITY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -3928,7 +3930,7 @@ var ScoresPG = class extends storage.ScoresStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_SCORES_BY_SPAN_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_SCORES_BY_SPAN", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -4003,7 +4005,7 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_PERSIST_WORKFLOW_SNAPSHOT_FAILED",
+          id: storage.createStorageErrorId("PG", "PERSIST_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -4024,7 +4026,7 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_LOAD_WORKFLOW_SNAPSHOT_FAILED",
+          id: storage.createStorageErrorId("PG", "LOAD_WORKFLOW_SNAPSHOT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -4065,7 +4067,7 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_GET_WORKFLOW_RUN_BY_ID_FAILED",
+          id: storage.createStorageErrorId("PG", "GET_WORKFLOW_RUN_BY_ID", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -4147,7 +4149,7 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_LIST_WORKFLOW_RUNS_FAILED",
+          id: storage.createStorageErrorId("PG", "LIST_WORKFLOW_RUNS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
           details: {
@@ -4164,17 +4166,17 @@ var WorkflowsPG = class extends storage.WorkflowsStorage {
 var PostgresStore = class extends storage.MastraStorage {
   #db;
   #pgp;
-  #config;
   schema;
-  isConnected = false;
+  isInitialized = false;
   stores;
   constructor(config) {
     try {
       validateConfig("PostgresStore", config);
-      super({ id: config.id, name: "PostgresStore" });
+      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,
@@ -4182,14 +4184,14 @@ var PostgresStore = class extends storage.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,
@@ -4205,11 +4207,24 @@ var PostgresStore = class extends storage.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__default.default();
+      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 error.MastraError(
         {
-          id: "MASTRA_STORAGE_PG_STORE_INITIALIZATION_FAILED",
+          id: storage.createStorageErrorId("PG", "INITIALIZATION", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.USER
         },
@@ -4218,36 +4233,22 @@ var PostgresStore = class extends storage.MastraStorage {
     }
   }
   async init() {
-    if (this.isConnected) {
+    if (this.isInitialized) {
       return;
     }
     try {
-      this.isConnected = true;
-      this.#pgp = pgPromise__default.default();
-      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$1) {
-      this.isConnected = false;
+      this.isInitialized = false;
       throw new error.MastraError(
         {
-          id: "MASTRA_STORAGE_POSTGRES_STORE_INIT_FAILED",
+          id: storage.createStorageErrorId("PG", "INIT", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY
         },
@@ -4256,15 +4257,9 @@ var PostgresStore = class extends storage.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() {
@@ -4404,7 +4399,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async createSpan(span) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -4419,7 +4414,7 @@ var PostgresStore = class extends storage.MastraStorage {
   }) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -4430,7 +4425,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async getTrace(traceId) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -4441,7 +4436,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async getTracesPaginated(args) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -4452,7 +4447,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async batchCreateSpans(args) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -4463,7 +4458,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async batchUpdateSpans(args) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"
@@ -4474,7 +4469,7 @@ var PostgresStore = class extends storage.MastraStorage {
   async batchDeleteTraces(args) {
     if (!this.stores.observability) {
       throw new error.MastraError({
-        id: "PG_STORE_OBSERVABILITY_NOT_INITIALIZED",
+        id: storage.createStorageErrorId("PG", "OBSERVABILITY", "NOT_INITIALIZED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.SYSTEM,
         text: "Observability storage is not initialized"