@mastra/pg 1.0.0-beta.0 → 1.0.0-beta.10

package/dist/shared/config.d.ts

@@ -1,6 +1,8 @@
-import type { ConnectionOptions } from 'tls';
+import type { ConnectionOptions } from 'node:tls';
+import type { CreateIndexOptions } from '@mastra/core/storage';
 import type { ClientConfig } from 'pg';
 import type * as pg from 'pg';
+import type pgPromise from 'pg-promise';
 import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
 /**
  * Generic PostgreSQL configuration type.
@@ -11,6 +13,55 @@ export type PostgresConfig<SSLType = ISSLConfig | ConnectionOptions> = {
     schemaName?: string;
     max?: number;
     idleTimeoutMillis?: number;
+    /**
+     * When true, automatic initialization (table creation/migrations) is disabled.
+     * This is useful for CI/CD pipelines where you want to:
+     * 1. Run migrations explicitly during deployment (not at runtime)
+     * 2. Use different credentials for schema changes vs runtime operations
+     *
+     * When disableInit is true:
+     * - The storage will not automatically create/alter tables on first use
+     * - You must call `storage.init()` explicitly in your CI/CD scripts
+     *
+     * @example
+     * // In CI/CD script:
+     * const storage = new PostgresStore({ ...config, disableInit: false });
+     * await storage.init(); // Explicitly run migrations
+     *
+     * // In runtime application:
+     * const storage = new PostgresStore({ ...config, disableInit: true });
+     * // No auto-init, tables must already exist
+     */
+    disableInit?: boolean;
+    /**
+     * When true, default indexes will not be created during initialization.
+     * This is useful when:
+     * 1. You want to manage indexes separately or use custom indexes only
+     * 2. Default indexes don't match your query patterns
+     * 3. You want to reduce initialization time in development
+     *
+     * @default false
+     */
+    skipDefaultIndexes?: boolean;
+    /**
+     * Custom indexes to create during initialization.
+     * These indexes are created in addition to default indexes (unless skipDefaultIndexes is true).
+     *
+     * Each index must specify which table it belongs to. The store will route each index
+     * to the appropriate domain based on the table name.
+     *
+     * @example
+     * ```typescript
+     * const store = new PostgresStore({
+     *   connectionString: '...',
+     *   indexes: [
+     *     { name: 'my_threads_type_idx', table: 'mastra_threads', columns: ['metadata->>\'type\''] },
+     *     { name: 'my_messages_status_idx', table: 'mastra_messages', columns: ['metadata->>\'status\''] },
+     *   ],
+     * });
+     * ```
+     */
+    indexes?: CreateIndexOptions[];
 } & ({
     host: string;
     port: number;
@@ -24,8 +75,41 @@ export type PostgresConfig<SSLType = ISSLConfig | ConnectionOptions> = {
 } | ClientConfig);
 /**
  * PostgreSQL configuration for PostgresStore (uses pg-promise with ISSLConfig)
+ *
+ * Accepts either:
+ * - A pre-configured pg-promise client: `{ id, client, schemaName? }`
+ * - Connection string: `{ id, connectionString, ... }`
+ * - Host/port config: `{ id, host, port, database, user, password, ... }`
+ * - Cloud SQL connector config: `{ id, stream, ... }`
  */
-export type PostgresStoreConfig = PostgresConfig<ISSLConfig>;
+export type PostgresStoreConfig = PostgresConfig<ISSLConfig> | {
+    id: string;
+    /**
+     * Pre-configured pg-promise database client.
+     * Use this when you need to configure the client before initialization,
+     * e.g., to add pool listeners or set connection-level settings.
+     *
+     * @example
+     * ```typescript
+     * import pgPromise from 'pg-promise';
+     *
+     * const pgp = pgPromise();
+     * const client = pgp({ connectionString: '...' });
+     *
+     * // Custom setup before using
+     * client.$pool.on('connect', async (poolClient) => {
+     *   await poolClient.query('SET ROLE my_role;');
+     * });
+     *
+     * const store = new PostgresStore({ id: 'my-store', client });
+     * ```
+     */
+    client: pgPromise.IDatabase<{}>;
+    schemaName?: string;
+    disableInit?: boolean;
+    skipDefaultIndexes?: boolean;
+    indexes?: CreateIndexOptions[];
+};
 /**
  * PostgreSQL configuration for PgVector (uses pg with ConnectionOptions)
  */
@@ -45,5 +129,16 @@ export declare const isHostConfig: <SSLType>(cfg: PostgresConfig<SSLType>) => cf
     ssl?: boolean | SSLType;
 };
 export declare const isCloudSqlConfig: <SSLType>(cfg: PostgresConfig<SSLType>) => cfg is PostgresConfig<SSLType> & ClientConfig;
-export declare const validateConfig: (name: string, config: PostgresConfig<ISSLConfig | ConnectionOptions>) => void;
+/**
+ * Type guard for pre-configured client config (PostgresStore only)
+ */
+export declare const isClientConfig: (cfg: PostgresStoreConfig) => cfg is {
+    id: string;
+    client: pgPromise.IDatabase<{}>;
+    schemaName?: string;
+    disableInit?: boolean;
+    skipDefaultIndexes?: boolean;
+    indexes?: CreateIndexOptions[];
+};
+export declare const validateConfig: (name: string, config: PostgresStoreConfig) => void;
 //# sourceMappingURL=config.d.ts.map

package/dist/shared/config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/shared/config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,KAAK,CAAC;AAC7C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,KAAK,KAAK,EAAE,MAAM,IAAI,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iCAAiC,CAAC;AAElE;;;GAGG;AACH,MAAM,MAAM,cAAc,CAAC,OAAO,GAAG,UAAU,GAAG,iBAAiB,IAAI;IACrE,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,GAAG,CACA;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;CACzB,GACD;IACE,gBAAgB,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;CACzB,GAED,YAAY,CACf,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC;AAE7D;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,cAAc,CAAC,iBAAiB,CAAC,GAAG;IAC/D,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;CACzD,CAAC;AAEF,eAAO,MAAM,wBAAwB,GAAI,OAAO,EAC9C,KAAK,cAAc,CAAC,OAAO,CAAC,KAC3B,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG;IAAE,gBAAgB,EAAE,MAAM,CAAC;IAAC,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAA;CAEtF,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,EAClC,KAAK,cAAc,CAAC,OAAO,CAAC,KAC3B,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;CAGzB,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,EACtC,KAAK,cAAc,CAAC,OAAO,CAAC,KAC3B,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG,YAEnC,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,MAAM,MAAM,EAAE,QAAQ,cAAc,CAAC,UAAU,GAAG,iBAAiB,CAAC,SA+BlG,CAAC"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/shared/config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,KAAK,KAAK,EAAE,MAAM,IAAI,CAAC;AAC9B,OAAO,KAAK,SAAS,MAAM,YAAY,CAAC;AACxC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iCAAiC,CAAC;AAElE;;;GAGG;AACH,MAAM,MAAM,cAAc,CAAC,OAAO,GAAG,UAAU,GAAG,iBAAiB,IAAI;IACrE,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B;;;;;;;;;;;;;;;;;;OAkBG;IACH,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB;;;;;;;;OAQG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B;;;;;;;;;;;;;;;;;OAiBG;IACH,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,GAAG,CACA;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;CACzB,GACD;IACE,gBAAgB,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;CACzB,GAED,YAAY,CACf,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,MAAM,mBAAmB,GAC3B,cAAc,CAAC,UAAU,CAAC,GAC1B;IACE,EAAE,EAAE,MAAM,CAAC;IACX;;;;;;;;;;;;;;;;;;;OAmBG;IACH,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,CAAC;AAEN;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,cAAc,CAAC,iBAAiB,CAAC,GAAG;IAC/D,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;CACzD,CAAC;AAEF,eAAO,MAAM,wBAAwB,GAAI,OAAO,EAC9C,KAAK,cAAc,CAAC,OAAO,CAAC,KAC3B,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG;IAAE,gBAAgB,EAAE,MAAM,CAAC;IAAC,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAA;CAEtF,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,OAAO,EAClC,KAAK,cAAc,CAAC,OAAO,CAAC,KAC3B,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;CAGzB,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,OAAO,EACtC,KAAK,cAAc,CAAC,OAAO,CAAC,KAC3B,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,GAAG,YAEnC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,cAAc,GACzB,KAAK,mBAAmB,KACvB,GAAG,IAAI;IACR,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAGhC,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,MAAM,MAAM,EAAE,QAAQ,mBAAmB,SAuCvE,CAAC"}

package/dist/storage/{domains/operations → db}/index.d.ts

@@ -1,16 +1,71 @@
-import { StoreOperations } from '@mastra/core/storage';
+import { MastraBase } from '@mastra/core/base';
 import type { StorageColumn, TABLE_NAMES, CreateIndexOptions, IndexInfo, StorageIndexStats } from '@mastra/core/storage';
 import type { IDatabase } from 'pg-promise';
-export type { CreateIndexOptions, IndexInfo, StorageIndexStats };
-export declare class StoreOperationsPG extends StoreOperations {
+import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
+/**
+ * Configuration for standalone domain usage.
+ * Accepts either:
+ * 1. An existing pg-promise database instance
+ * 2. Config to create a new client internally
+ */
+export type PgDomainConfig = PgDomainClientConfig | PgDomainRestConfig;
+/**
+ * Pass an existing pg-promise database instance
+ */
+export interface PgDomainClientConfig {
+    /** The pg-promise database instance */
     client: IDatabase<{}>;
+    /** Optional schema name (defaults to 'public') */
     schemaName?: string;
-    private setupSchemaPromise;
-    private schemaSetupComplete;
-    constructor({ client, schemaName }: {
-        client: IDatabase<{}>;
-        schemaName?: string;
-    });
+    /** When true, default indexes will not be created during initialization */
+    skipDefaultIndexes?: boolean;
+    /** Custom indexes to create for this domain's tables */
+    indexes?: CreateIndexOptions[];
+}
+/**
+ * Pass config to create a new pg-promise client internally
+ */
+export type PgDomainRestConfig = {
+    /** Optional schema name (defaults to 'public') */
+    schemaName?: string;
+    /** When true, default indexes will not be created during initialization */
+    skipDefaultIndexes?: boolean;
+    /** Custom indexes to create for this domain's tables */
+    indexes?: CreateIndexOptions[];
+} & ({
+    host: string;
+    port: number;
+    database: string;
+    user: string;
+    password: string;
+    ssl?: boolean | ISSLConfig;
+} | {
+    connectionString: string;
+    ssl?: boolean | ISSLConfig;
+});
+/**
+ * Resolves PgDomainConfig to a pg-promise database instance and schema.
+ * Handles creating a new client if config is provided.
+ */
+export declare function resolvePgConfig(config: PgDomainConfig): {
+    client: IDatabase<{}>;
+    schemaName?: string;
+    skipDefaultIndexes?: boolean;
+    indexes?: CreateIndexOptions[];
+};
+/**
+ * Internal config for PgDB - accepts already-resolved client
+ */
+export interface PgDBInternalConfig {
+    client: IDatabase<{}>;
+    schemaName?: string;
+    skipDefaultIndexes?: boolean;
+}
+export declare class PgDB extends MastraBase {
+    client: IDatabase<{}>;
+    schemaName?: string;
+    skipDefaultIndexes?: boolean;
+    constructor(config: PgDBInternalConfig);
     hasColumn(table: string, column: string): Promise<boolean>;
     /**
      * Prepares values for insertion, handling JSONB columns by stringifying them
@@ -21,11 +76,12 @@ export declare class StoreOperationsPG extends StoreOperations {
      */
     private addTimestampZColumns;
     /**
-     * Prepares a value for database operations, handling Date objects and JSON serialization
-     * This is schema-aware and only stringifies objects for JSONB columns
+     * Prepares a value for database operations
      */
     private prepareValue;
     private setupSchema;
+    protected getSqlType(type: StorageColumn['type']): string;
+    protected getDefaultValue(type: StorageColumn['type']): string;
     insert({ tableName, record }: {
         tableName: TABLE_NAMES;
         record: Record<string, any>;
@@ -33,15 +89,16 @@ export declare class StoreOperationsPG extends StoreOperations {
     clearTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
-    protected getDefaultValue(type: StorageColumn['type']): string;
     createTable({ tableName, schema, }: {
         tableName: TABLE_NAMES;
         schema: Record<string, StorageColumn>;
     }): Promise<void>;
+    private setupTimestampTriggers;
     /**
-     * Set up timestamp triggers for a table to automatically manage createdAt/updatedAt
+     * Migrates the spans table schema from OLD_SPAN_SCHEMA to current SPAN_SCHEMA.
+     * This adds new columns that don't exist in old schema.
      */
-    private setupTimestampTriggers;
+    private migrateSpansTable;
     /**
      * Alters table schema to add columns if they don't exist
      * @param tableName Name of the table
@@ -64,43 +121,15 @@ export declare class StoreOperationsPG extends StoreOperations {
     dropTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
-    /**
-     * Create a new index on a table
-     */
     createIndex(options: CreateIndexOptions): Promise<void>;
-    /**
-     * Drop an existing index
-     */
     dropIndex(indexName: string): Promise<void>;
-    /**
-     * List indexes for a specific table or all tables
-     */
     listIndexes(tableName?: string): Promise<IndexInfo[]>;
-    /**
-     * Returns definitions for automatic performance indexes
-     * These composite indexes cover both filtering and sorting in single index
-     */
-    protected getAutomaticIndexDefinitions(): CreateIndexOptions[];
-    /**
-     * Creates automatic indexes for optimal query performance
-     * Uses getAutomaticIndexDefinitions() to determine which indexes to create
-     */
-    createAutomaticIndexes(): Promise<void>;
-    /**
-     * Get detailed statistics for a specific index
-     */
     describeIndex(indexName: string): Promise<StorageIndexStats>;
-    /**
-     * Update a single record in the database
-     */
     update({ tableName, keys, data, }: {
         tableName: TABLE_NAMES;
         keys: Record<string, any>;
         data: Record<string, any>;
     }): Promise<void>;
-    /**
-     * Update multiple records in a single batch transaction
-     */
     batchUpdate({ tableName, updates, }: {
         tableName: TABLE_NAMES;
         updates: Array<{
@@ -108,12 +137,15 @@ export declare class StoreOperationsPG extends StoreOperations {
             data: Record<string, any>;
         }>;
     }): Promise<void>;
-    /**
-     * Delete multiple records by keys
-     */
     batchDelete({ tableName, keys }: {
         tableName: TABLE_NAMES;
         keys: Record<string, any>[];
     }): Promise<void>;
+    /**
+     * Delete all data from a table (alias for clearTable for consistency with other stores)
+     */
+    deleteData({ tableName }: {
+        tableName: TABLE_NAMES;
+    }): Promise<void>;
 }
 //# sourceMappingURL=index.d.ts.map

package/dist/storage/db/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/storage/db/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAU/C,OAAO,KAAK,EACV,aAAa,EACb,WAAW,EACX,kBAAkB,EAClB,SAAS,EACT,iBAAiB,EAClB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iCAAiC,CAAC;AAElE;;;;;GAKG;AACH,MAAM,MAAM,cAAc,GAAG,oBAAoB,GAAG,kBAAkB,CAAC;AAEvE;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,uCAAuC;IACvC,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACtB,kDAAkD;IAClD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,wDAAwD;IACxD,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B,kDAAkD;IAClD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,wDAAwD;IACxD,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,GAAG,CACA;IACE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,OAAO,GAAG,UAAU,CAAC;CAC5B,GACD;IACE,gBAAgB,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,GAAG,UAAU,CAAC;CAC5B,CACJ,CAAC;AAEF;;;GAGG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,cAAc,GAAG;IACvD,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,CAoBA;AAaD;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAOD,qBAAa,IAAK,SAAQ,UAAU;IAC3B,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;gBAExB,MAAM,EAAE,kBAAkB;IAWhC,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAWhE;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAY9B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAY5B;;OAEG;IACH,OAAO,CAAC,YAAY;YAuBN,WAAW;IA6DzB,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAWzD,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,MAAM;IAWxD,MAAM,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA4BrG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCpE,WAAW,CAAC,EAChB,SAAS,EACT,MAAM,GACP,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;KACvC,GAAG,OAAO,CAAC,IAAI,CAAC;YA+FH,sBAAsB;IAwCpC;;;OAGG;YACW,iBAAiB;IAqD/B;;;;;OAKG;IACG,UAAU,CAAC,EACf,SAAS,EACT,MAAM,EACN,WAAW,GACZ,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtC,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,GAAG,OAAO,CAAC,IAAI,CAAC;IAyCX,IAAI,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAuCzG,WAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB9G,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBnE,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuFvD,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgC3C,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAiFrD,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAoE5D,MAAM,CAAC,EACX,SAAS,EACT,IAAI,EACJ,IAAI,GACL,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC3B,GAAG,OAAO,CAAC,IAAI,CAAC;IA6CX,WAAW,CAAC,EAChB,SAAS,EACT,OAAO,GACR,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,CAAC;YACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SAC3B,CAAC,CAAC;KACJ,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBX,WAAW,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA2C9G;;OAEG;IACG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAG3E"}

package/dist/storage/domains/agents/index.d.ts

@@ -0,0 +1,39 @@
+import { AgentsStorage } from '@mastra/core/storage';
+import type { StorageAgentType, StorageCreateAgentInput, StorageUpdateAgentInput, StorageListAgentsInput, StorageListAgentsOutput, CreateIndexOptions } from '@mastra/core/storage';
+import type { PgDomainConfig } from '../../db/index.js';
+export declare class AgentsPG extends AgentsStorage {
+    #private;
+    /** Tables managed by this domain */
+    static readonly MANAGED_TABLES: readonly ["mastra_agents"];
+    constructor(config: PgDomainConfig);
+    /**
+     * Returns default index definitions for the agents domain tables.
+     * Currently no default indexes are defined for agents.
+     */
+    getDefaultIndexDefinitions(): CreateIndexOptions[];
+    /**
+     * Creates default indexes for optimal query performance.
+     * Currently no default indexes are defined for agents.
+     */
+    createDefaultIndexes(): Promise<void>;
+    init(): Promise<void>;
+    /**
+     * Creates custom user-defined indexes for this domain's tables.
+     */
+    createCustomIndexes(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
+    private parseJson;
+    private parseRow;
+    getAgentById({ id }: {
+        id: string;
+    }): Promise<StorageAgentType | null>;
+    createAgent({ agent }: {
+        agent: StorageCreateAgentInput;
+    }): Promise<StorageAgentType>;
+    updateAgent({ id, ...updates }: StorageUpdateAgentInput): Promise<StorageAgentType>;
+    deleteAgent({ id }: {
+        id: string;
+    }): Promise<void>;
+    listAgents(args?: StorageListAgentsInput): Promise<StorageListAgentsOutput>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/storage/domains/agents/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/agents/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,EAMd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,gBAAgB,EAChB,uBAAuB,EACvB,uBAAuB,EACvB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAG/C,qBAAa,QAAS,SAAQ,aAAa;;IAMzC,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,6BAA2B;gBAE7C,MAAM,EAAE,cAAc;IAUlC;;;OAGG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAIlD;;;OAGG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOrC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAM3B;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI1C,OAAO,CAAC,SAAS;IA2BjB,OAAO,CAAC,QAAQ;IAqBV,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAwBtE,WAAW,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,uBAAuB,CAAA;KAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAoDrF,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,EAAE,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAsInF,WAAW,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBlD,UAAU,CAAC,IAAI,CAAC,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;CA+DlF"}

package/dist/storage/domains/memory/index.d.ts

@@ -1,18 +1,27 @@
 import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
 import { MemoryStorage } from '@mastra/core/storage';
-import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsByResourceIdInput, StorageListThreadsByResourceIdOutput } from '@mastra/core/storage';
-import type { IDatabase } from 'pg-promise';
-import type { StoreOperationsPG } from '../operations/index.js';
+import type { StorageResourceType, StorageListMessagesInput, StorageListMessagesOutput, StorageListThreadsByResourceIdInput, StorageListThreadsByResourceIdOutput, CreateIndexOptions } from '@mastra/core/storage';
+import type { PgDomainConfig } from '../../db/index.js';
 export declare class MemoryPG extends MemoryStorage {
-    private client;
-    private schema;
-    private operations;
-    constructor({ client, schema, operations, }: {
-        client: IDatabase<{}>;
-        schema: string;
-        operations: StoreOperationsPG;
-    });
+    #private;
+    /** Tables managed by this domain */
+    static readonly MANAGED_TABLES: readonly ["mastra_threads", "mastra_messages", "mastra_resources"];
+    constructor(config: PgDomainConfig);
+    init(): Promise<void>;
+    /**
+     * Returns default index definitions for the memory domain tables.
+     */
+    getDefaultIndexDefinitions(): CreateIndexOptions[];
+    /**
+     * Creates default indexes for optimal query performance.
+     */
+    createDefaultIndexes(): Promise<void>;
+    /**
+     * Creates custom user-defined indexes for this domain's tables.
+     */
+    createCustomIndexes(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
     /**
      * Normalizes message row from database by applying createdAtZ fallback
      */
@@ -32,6 +41,16 @@ export declare class MemoryPG extends MemoryStorage {
     deleteThread({ threadId }: {
         threadId: string;
     }): Promise<void>;
+    /**
+     * Fetches messages around target messages using cursor-based pagination.
+     *
+     * This replaces the previous ROW_NUMBER() approach which caused severe performance
+     * issues on large tables (see GitHub issue #11150). The old approach required
+     * scanning and sorting ALL messages in a thread to assign row numbers.
+     *
+     * The new approach uses the existing (thread_id, createdAt) index to efficiently
+     * fetch only the messages needed by using createdAt as a cursor.
+     */
     private _getIncludedMessages;
     private parseRow;
     listMessagesById({ messageIds }: {

package/dist/storage/domains/memory/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EACL,aAAa,EAMd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAevD,qBAAa,QAAS,SAAQ,aAAa;IACzC,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAoB;gBAE1B,EACV,MAAM,EACN,MAAM,EACN,UAAU,GACX,EAAE;QACD,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACtB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;KAC/B;IAOD;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAYrB,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAoC7E,uBAAuB,CAClC,IAAI,EAAE,mCAAmC,GACxC,OAAO,CAAC,oCAAoC,CAAC;IAkF1C,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAkDjF,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,GACT,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA6DxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;YA0BvD,oBAAoB;IAqElC,OAAO,CAAC,QAAQ;IAmBH,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAoCpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAwKvF,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IA4GrG,cAAc,CAAC,EACnB,QAAQ,GACT,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBACR,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAC9C,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC;aAC7C,CAAC;SACH,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAsHxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA6CnD,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAoB5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAY3F,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,GACT,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,mBAAmB,CAAC;CAwDjC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EACL,aAAa,EAQd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,mCAAmC,EACnC,oCAAoC,EACpC,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAuB/C,qBAAa,QAAS,SAAQ,aAAa;;IAMzC,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,qEAA6D;gBAE/E,MAAM,EAAE,cAAc;IAU5B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAa3B;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAgBlD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAM1C;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAYrB,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAoC7E,uBAAuB,CAClC,IAAI,EAAE,mCAAmC,GACxC,OAAO,CAAC,oCAAoC,CAAC;IAiF1C,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAkDjF,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,GACT,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACnC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA2DxB,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwCrE;;;;;;;;;OASG;YACW,oBAAoB;IAyElC,OAAO,CAAC,QAAQ;IAmBH,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAoCpG,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAgKvF,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC;IAwGrG,cAAc,CAAC,EACnB,QAAQ,GACT,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBACR,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAC9C,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC;aAC7C,CAAC;SACH,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAkHxB,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwCnD,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAoB5F,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAY3F,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,GACT,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,mBAAmB,CAAC;CAuDjC"}

package/dist/storage/domains/observability/index.d.ts

@@ -1,44 +1,37 @@
-import type { TracingStorageStrategy } from '@mastra/core/observability';
 import { ObservabilityStorage } from '@mastra/core/storage';
-import type { SpanRecord, TraceRecord, TracesPaginatedArg, CreateSpanRecord, PaginationInfo, UpdateSpanRecord } from '@mastra/core/storage';
-import type { IDatabase } from 'pg-promise';
-import type { StoreOperationsPG } from '../operations/index.js';
+import type { TracingStorageStrategy, ListTracesArgs, UpdateSpanArgs, BatchDeleteTracesArgs, BatchUpdateSpansArgs, BatchCreateSpansArgs, CreateSpanArgs, GetSpanArgs, GetSpanResponse, GetRootSpanArgs, GetRootSpanResponse, GetTraceArgs, GetTraceResponse, ListTracesResponse, CreateIndexOptions } from '@mastra/core/storage';
+import type { PgDomainConfig } from '../../db/index.js';
 export declare class ObservabilityPG extends ObservabilityStorage {
-    client: IDatabase<{}>;
-    private operations;
-    private schema?;
-    constructor({ client, operations, schema, }: {
-        client: IDatabase<{}>;
-        operations: StoreOperationsPG;
-        schema?: string;
-    });
+    #private;
+    /** Tables managed by this domain */
+    static readonly MANAGED_TABLES: readonly ["mastra_ai_spans"];
+    constructor(config: PgDomainConfig);
+    init(): Promise<void>;
+    /**
+     * Returns default index definitions for the observability domain tables.
+     */
+    getDefaultIndexDefinitions(): CreateIndexOptions[];
+    /**
+     * Creates default indexes for optimal query performance.
+     */
+    createDefaultIndexes(): Promise<void>;
+    /**
+     * Creates custom user-defined indexes for this domain's tables.
+     */
+    createCustomIndexes(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
     get tracingStrategy(): {
         preferred: TracingStorageStrategy;
         supported: TracingStorageStrategy[];
     };
-    createSpan(span: CreateSpanRecord): Promise<void>;
-    getTrace(traceId: string): Promise<TraceRecord | null>;
-    updateSpan({ spanId, traceId, updates, }: {
-        spanId: string;
-        traceId: string;
-        updates: Partial<UpdateSpanRecord>;
-    }): Promise<void>;
-    getTracesPaginated({ filters, pagination, }: TracesPaginatedArg): Promise<{
-        pagination: PaginationInfo;
-        spans: SpanRecord[];
-    }>;
-    batchCreateSpans(args: {
-        records: CreateSpanRecord[];
-    }): Promise<void>;
-    batchUpdateSpans(args: {
-        records: {
-            traceId: string;
-            spanId: string;
-            updates: Partial<UpdateSpanRecord>;
-        }[];
-    }): Promise<void>;
-    batchDeleteTraces(args: {
-        traceIds: string[];
-    }): Promise<void>;
+    createSpan(args: CreateSpanArgs): Promise<void>;
+    getSpan(args: GetSpanArgs): Promise<GetSpanResponse | null>;
+    getRootSpan(args: GetRootSpanArgs): Promise<GetRootSpanResponse | null>;
+    getTrace(args: GetTraceArgs): Promise<GetTraceResponse | null>;
+    updateSpan(args: UpdateSpanArgs): Promise<void>;
+    listTraces(args: ListTracesArgs): Promise<ListTracesResponse>;
+    batchCreateSpans(args: BatchCreateSpansArgs): Promise<void>;
+    batchUpdateSpans(args: BatchUpdateSpansArgs): Promise<void>;
+    batchDeleteTraces(args: BatchDeleteTracesArgs): Promise<void>;
 }
 //# sourceMappingURL=index.d.ts.map

package/dist/storage/domains/observability/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/observability/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAe,oBAAoB,EAAe,MAAM,sBAAsB,CAAC;AACtF,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,gBAAgB,EACjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAGvD,qBAAa,eAAgB,SAAQ,oBAAoB;IAChD,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IAC7B,OAAO,CAAC,UAAU,CAAoB;IACtC,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,EACV,MAAM,EACN,UAAU,EACV,MAAM,GACP,EAAE;QACD,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACtB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;IAOD,IAAoB,eAAe,IAAI;QACrC,SAAS,EAAE,sBAAsB,CAAC;QAClC,SAAS,EAAE,sBAAsB,EAAE,CAAC;KACrC,CAKA;IAEK,UAAU,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAiCjD,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IA+CtD,UAAU,CAAC,EACf,MAAM,EACN,OAAO,EACP,OAAO,GACR,EAAE;QACD,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCX,kBAAkB,CAAC,EACvB,OAAO,EACP,UAAU,GACX,EAAE,kBAAkB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;IA+G9E,gBAAgB,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,gBAAgB,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCtE,gBAAgB,CAAC,IAAI,EAAE;QAC3B,OAAO,EAAE;YACP,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,OAAO,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;SACpC,EAAE,CAAC;KACL,GAAG,OAAO,CAAC,IAAI,CAAC;IAyCX,iBAAiB,CAAC,IAAI,EAAE;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAoBrE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/observability/index.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,oBAAoB,EAIrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAEV,sBAAsB,EACtB,cAAc,EACd,cAAc,EACd,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,cAAc,EACd,WAAW,EACX,eAAe,EACf,eAAe,EACf,mBAAmB,EACnB,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAG/C,qBAAa,eAAgB,SAAQ,oBAAoB;;IAMvD,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,+BAA0B;gBAE5C,MAAM,EAAE,cAAc;IAU5B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAM3B;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAgElD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI1C,IAAoB,eAAe,IAAI;QACrC,SAAS,EAAE,sBAAsB,CAAC;QAClC,SAAS,EAAE,sBAAsB,EAAE,CAAC;KACrC,CAKA;IAEK,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAiC/C,OAAO,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IA+C3D,WAAW,CAAC,IAAI,EAAE,eAAe,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IA+CvE,QAAQ,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAqD9D,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAoC/C,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAsO7D,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+B3D,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAmC3D,iBAAiB,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CAoBpE"}

package/dist/storage/domains/scores/index.d.ts

@@ -1,17 +1,26 @@
-import type { ScoreRowData, ScoringSource } from '@mastra/core/evals';
-import type { PaginationInfo, StoragePagination } from '@mastra/core/storage';
+import type { ListScoresResponse, SaveScorePayload, ScoreRowData, ScoringSource } from '@mastra/core/evals';
+import type { StoragePagination, CreateIndexOptions } from '@mastra/core/storage';
 import { ScoresStorage } from '@mastra/core/storage';
-import type { IDatabase } from 'pg-promise';
-import type { StoreOperationsPG } from '../operations/index.js';
+import type { PgDomainConfig } from '../../db/index.js';
 export declare class ScoresPG extends ScoresStorage {
-    client: IDatabase<{}>;
-    private operations;
-    private schema?;
-    constructor({ client, operations, schema, }: {
-        client: IDatabase<{}>;
-        operations: StoreOperationsPG;
-        schema?: string;
-    });
+    #private;
+    /** Tables managed by this domain */
+    static readonly MANAGED_TABLES: readonly ["mastra_scorers"];
+    constructor(config: PgDomainConfig);
+    init(): Promise<void>;
+    /**
+     * Returns default index definitions for the scores domain tables.
+     */
+    getDefaultIndexDefinitions(): CreateIndexOptions[];
+    /**
+     * Creates default indexes for optimal query performance.
+     */
+    createDefaultIndexes(): Promise<void>;
+    /**
+     * Creates custom user-defined indexes for this domain's tables.
+     */
+    createCustomIndexes(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
     getScoreById({ id }: {
         id: string;
     }): Promise<ScoreRowData | null>;
@@ -21,35 +30,23 @@ export declare class ScoresPG extends ScoresStorage {
         entityId?: string;
         entityType?: string;
         source?: ScoringSource;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
-    saveScore(score: Omit<ScoreRowData, 'id' | 'createdAt' | 'updatedAt'>): Promise<{
+    }): Promise<ListScoresResponse>;
+    saveScore(score: SaveScorePayload): Promise<{
         score: ScoreRowData;
     }>;
     listScoresByRunId({ runId, pagination, }: {
         runId: string;
         pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
+    }): Promise<ListScoresResponse>;
     listScoresByEntityId({ entityId, entityType, pagination, }: {
         pagination: StoragePagination;
         entityId: string;
         entityType: string;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
+    }): Promise<ListScoresResponse>;
     listScoresBySpan({ traceId, spanId, pagination, }: {
         traceId: string;
         spanId: string;
         pagination: StoragePagination;
-    }): Promise<{
-        pagination: PaginationInfo;
-        scores: ScoreRowData[];
-    }>;
+    }): Promise<ListScoresResponse>;
 }
 //# sourceMappingURL=index.d.ts.map

package/dist/storage/domains/scores/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAA6B,MAAM,oBAAoB,CAAC;AACjG,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EAIL,aAAa,EAEd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAoBvD,qBAAa,QAAS,SAAQ,aAAa;IAClC,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;IAC7B,OAAO,CAAC,UAAU,CAAoB;IACtC,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,EACV,MAAM,EACN,UAAU,EACV,MAAM,GACP,EAAE;QACD,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QACtB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;IAOK,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAoBlE,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,UAAU,EACV,MAAM,GACP,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAsE7D,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAwExG,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAkD7D,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,UAAU,GACX,EAAE;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;IAkD7D,gBAAgB,CAAC,EACrB,OAAO,EACP,MAAM,EACN,UAAU,GACX,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,cAAc,CAAC;QAAC,MAAM,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC;CA0CpE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/scores/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAE5G,OAAO,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAClF,OAAO,EAIL,aAAa,EAId,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAwB/C,qBAAa,QAAS,SAAQ,aAAa;;IAMzC,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,8BAA4B;gBAE9C,MAAM,EAAE,cAAc;IAU5B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAM3B;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAWlD;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAe3C;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIpC,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAoBlE,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,UAAU,EACV,MAAM,GACP,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,MAAM,CAAC,EAAE,aAAa,CAAC;KACxB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAsEzB,SAAS,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,CAAA;KAAE,CAAC;IAuEpE,iBAAiB,CAAC,EACtB,KAAK,EACL,UAAU,GACX,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAkDzB,oBAAoB,CAAC,EACzB,QAAQ,EACR,UAAU,EACV,UAAU,GACX,EAAE;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAkDzB,gBAAgB,CAAC,EACrB,OAAO,EACP,MAAM,EACN,UAAU,GACX,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,iBAAiB,CAAC;KAC/B,GAAG,OAAO,CAAC,kBAAkB,CAAC;CA0ChC"}

package/dist/storage/domains/utils.d.ts

@@ -1,13 +1,9 @@
-import type { PaginationArgs, StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
+import type { StorageColumn, TABLE_NAMES } from '@mastra/core/storage';
 export declare function getSchemaName(schema?: string): string | undefined;
 export declare function getTableName({ indexName, schemaName }: {
     indexName: string;
     schemaName?: string;
 }): string;
-/**
- * Build date range filter for queries
- */
-export declare function buildDateRangeFilter(dateRange: PaginationArgs['dateRange'], fieldName: string): Record<string, any>;
 /**
  * Prepare WHERE clause for PostgreSQL queries
  */