@mastra/pg 0.0.0-error-handler-fix-20251020202607 → 0.0.0-execa-dynamic-import-20260304221256

package/dist/shared/config.d.ts

@@ -1,48 +1,139 @@
-import type { ConnectionOptions } from 'tls';
-import type { ClientConfig } from 'pg';
-import type * as pg from 'pg';
-import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
+import type { ConnectionOptions } from 'node:tls';
+import type { CreateIndexOptions } from '@mastra/core/storage';
+import type { ClientConfig, Pool, PoolConfig } from 'pg';
 /**
- * Generic PostgreSQL configuration type.
- * @template SSLType - The SSL configuration type (ISSLConfig for pg-promise, ConnectionOptions for pg)
+ * Base configuration options shared across PostgreSQL configs.
  */
-export type PostgresConfig<SSLType = ISSLConfig | ConnectionOptions> = {
+export interface PostgresBaseConfig {
+    id: string;
     schemaName?: string;
+    /**
+     * 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[];
+}
+/**
+ * Connection string configuration.
+ */
+export interface ConnectionStringConfig extends PostgresBaseConfig {
+    connectionString: string;
+    ssl?: boolean | ConnectionOptions;
     max?: number;
     idleTimeoutMillis?: number;
-} & ({
+}
+/**
+ * Host-based configuration.
+ */
+export interface HostConfig extends PostgresBaseConfig {
     host: string;
     port: number;
     database: string;
     user: string;
     password: string;
-    ssl?: boolean | SSLType;
-} | {
-    connectionString: string;
-    ssl?: boolean | SSLType;
-} | ClientConfig);
+    ssl?: boolean | ConnectionOptions;
+    max?: number;
+    idleTimeoutMillis?: number;
+}
+/**
+ * Pre-configured pg.Pool configuration.
+ */
+export interface PoolInstanceConfig extends PostgresBaseConfig {
+    /**
+     * Pre-configured pg.Pool instance.
+     * Use this for direct control over the connection pool, or for
+     * integration with libraries that expect a pg.Pool.
+     *
+     * @example
+     * ```typescript
+     * import { Pool } from 'pg';
+     *
+     * const pool = new Pool({ connectionString: '...' });
+     * const store = new PostgresStore({ id: 'my-store', pool });
+     *
+     * // Use store.pool for other libraries that need a pg.Pool
+     * ```
+     */
+    pool: Pool;
+}
 /**
- * PostgreSQL configuration for PostgresStore (uses pg-promise with ISSLConfig)
+ * PostgreSQL configuration for PostgresStore.
+ *
+ * Accepts either:
+ * - A pre-configured pg.Pool: `{ id, pool, schemaName? }`
+ * - Connection string: `{ id, connectionString, ... }`
+ * - Host/port config: `{ id, host, port, database, user, password, ... }`
+ * - Cloud SQL connector config: `{ id, stream, ... }` (via pg.ClientConfig)
  */
-export type PostgresStoreConfig = PostgresConfig<ISSLConfig>;
+export type PostgresStoreConfig = PoolInstanceConfig | ConnectionStringConfig | HostConfig | (PostgresBaseConfig & ClientConfig);
 /**
  * PostgreSQL configuration for PgVector (uses pg with ConnectionOptions)
  */
-export type PgVectorConfig = PostgresConfig<ConnectionOptions> & {
-    pgPoolOptions?: Omit<pg.PoolConfig, 'connectionString'>;
-};
-export declare const isConnectionStringConfig: <SSLType>(cfg: PostgresConfig<SSLType>) => cfg is PostgresConfig<SSLType> & {
-    connectionString: string;
-    ssl?: boolean | SSLType;
-};
-export declare const isHostConfig: <SSLType>(cfg: PostgresConfig<SSLType>) => cfg is PostgresConfig<SSLType> & {
-    host: string;
-    port: number;
-    database: string;
-    user: string;
-    password: string;
-    ssl?: boolean | SSLType;
+export type PgVectorConfig = (ConnectionStringConfig | HostConfig | (PostgresBaseConfig & ClientConfig)) & {
+    pgPoolOptions?: Omit<PoolConfig, 'connectionString'>;
 };
-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 pg.Pool config
+ */
+export declare const isPoolConfig: (cfg: PostgresStoreConfig) => cfg is PoolInstanceConfig;
+/**
+ * Type guard for connection string config
+ */
+export declare const isConnectionStringConfig: (cfg: PostgresStoreConfig) => cfg is ConnectionStringConfig;
+/**
+ * Type guard for host-based config
+ */
+export declare const isHostConfig: (cfg: PostgresStoreConfig) => cfg is HostConfig;
+/**
+ * Type guard for Cloud SQL connector config
+ */
+export declare const isCloudSqlConfig: (cfg: PostgresStoreConfig) => cfg is PostgresBaseConfig & ClientConfig;
+/**
+ * Validate PostgresStore configuration.
+ */
+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,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,SA2BlG,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,IAAI,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAEzD;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;;;;;;;;;;;;;;;;;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;AAED;;GAEG;AACH,MAAM,WAAW,sBAAuB,SAAQ,kBAAkB;IAChE,gBAAgB,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,GAAG,iBAAiB,CAAC;IAClC,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,UAAW,SAAQ,kBAAkB;IACpD,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,iBAAiB,CAAC;IAClC,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,kBAAkB;IAC5D;;;;;;;;;;;;;;OAcG;IACH,IAAI,EAAE,IAAI,CAAC;CACZ;AAED;;;;;;;;GAQG;AACH,MAAM,MAAM,mBAAmB,GAC3B,kBAAkB,GAClB,sBAAsB,GACtB,UAAU,GACV,CAAC,kBAAkB,GAAG,YAAY,CAAC,CAAC;AAExC;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,sBAAsB,GAAG,UAAU,GAAG,CAAC,kBAAkB,GAAG,YAAY,CAAC,CAAC,GAAG;IACzG,aAAa,CAAC,EAAE,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;CACtD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,YAAY,GAAI,KAAK,mBAAmB,KAAG,GAAG,IAAI,kBAE9D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,wBAAwB,GAAI,KAAK,mBAAmB,KAAG,GAAG,IAAI,sBAE1E,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,YAAY,GAAI,KAAK,mBAAmB,KAAG,GAAG,IAAI,UAE9D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gBAAgB,GAAI,KAAK,mBAAmB,KAAG,GAAG,IAAI,kBAAkB,GAAG,YAEvF,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,cAAc,GAAI,MAAM,MAAM,EAAE,QAAQ,mBAAmB,SAuCvE,CAAC"}

package/dist/storage/client.d.ts

@@ -0,0 +1,91 @@
+import type { Pool, PoolClient, QueryResult } from 'pg';
+export type { Pool, PoolClient, QueryResult } from 'pg';
+/**
+ * Values array for parameterized queries.
+ */
+export type QueryValues = unknown[];
+/**
+ * Common interface for database clients.
+ * PoolAdapter implements this interface by wrapping a pg.Pool.
+ */
+export interface DbClient {
+    /**
+     * The underlying connection pool.
+     */
+    readonly $pool: Pool;
+    /**
+     * Acquire a client from the pool for manual query execution.
+     * Remember to call client.release() when done.
+     */
+    connect(): Promise<PoolClient>;
+    /**
+     * Execute a query that returns no data.
+     * Use for INSERT, UPDATE, DELETE without RETURNING.
+     */
+    none(query: string, values?: QueryValues): Promise<null>;
+    /**
+     * Execute a query that returns exactly one row.
+     * @throws Error if zero or more than one row is returned
+     */
+    one<T = any>(query: string, values?: QueryValues): Promise<T>;
+    /**
+     * Execute a query that returns zero or one row.
+     * @returns The row, or null if no rows returned
+     * @throws Error if more than one row is returned
+     */
+    oneOrNone<T = any>(query: string, values?: QueryValues): Promise<T | null>;
+    /**
+     * Execute a query that returns any number of rows (including zero).
+     * Alias for manyOrNone.
+     */
+    any<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    /**
+     * Execute a query that returns zero or more rows.
+     */
+    manyOrNone<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    /**
+     * Execute a query that returns at least one row.
+     * @throws Error if no rows are returned
+     */
+    many<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    /**
+     * Execute a raw query, returning the full result object.
+     */
+    query(query: string, values?: QueryValues): Promise<QueryResult>;
+    /**
+     * Execute a function within a transaction.
+     * Automatically handles BEGIN, COMMIT, and ROLLBACK.
+     */
+    tx<T>(callback: (t: TxClient) => Promise<T>): Promise<T>;
+}
+/**
+ * Transaction client interface for executing queries within a transaction.
+ */
+export interface TxClient {
+    none(query: string, values?: QueryValues): Promise<null>;
+    one<T = any>(query: string, values?: QueryValues): Promise<T>;
+    oneOrNone<T = any>(query: string, values?: QueryValues): Promise<T | null>;
+    any<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    manyOrNone<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    many<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    query(query: string, values?: QueryValues): Promise<QueryResult>;
+    /** Execute multiple promises in parallel */
+    batch<T>(promises: Promise<T>[]): Promise<T[]>;
+}
+/**
+ * Adapter that wraps a pg.Pool to implement DbClient.
+ */
+export declare class PoolAdapter implements DbClient {
+    readonly $pool: Pool;
+    constructor($pool: Pool);
+    connect(): Promise<PoolClient>;
+    none(query: string, values?: QueryValues): Promise<null>;
+    one<T = any>(query: string, values?: QueryValues): Promise<T>;
+    oneOrNone<T = any>(query: string, values?: QueryValues): Promise<T | null>;
+    any<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    manyOrNone<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    many<T = any>(query: string, values?: QueryValues): Promise<T[]>;
+    query(query: string, values?: QueryValues): Promise<QueryResult>;
+    tx<T>(callback: (t: TxClient) => Promise<T>): Promise<T>;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/storage/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/storage/client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;AAGxD,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;AAExD;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,OAAO,EAAE,CAAC;AAEpC;;;GAGG;AACH,MAAM,WAAW,QAAQ;IACvB;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC;IAErB;;;OAGG;IACH,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC,CAAC;IAE/B;;;OAGG;IACH,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEzD;;;OAGG;IACH,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAE9D;;;;OAIG;IACH,SAAS,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAE3E;;;OAGG;IACH,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAEhE;;OAEG;IACH,UAAU,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAEvE;;;OAGG;IACH,IAAI,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAEjE;;OAEG;IACH,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAEjE;;;OAGG;IACH,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;CAC1D;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzD,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAC9D,SAAS,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAC3E,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAChE,UAAU,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IACvE,IAAI,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IACjE,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACjE,4CAA4C;IAC5C,KAAK,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;CAChD;AAaD;;GAEG;AACH,qBAAa,WAAY,YAAW,QAAQ;aACd,KAAK,EAAE,IAAI;gBAAX,KAAK,EAAE,IAAI;IAEvC,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC;IAIxB,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAKxD,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC;IAW7D,SAAS,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAW1E,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAK/D,UAAU,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAItE,IAAI,CAAC,CAAC,GAAG,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAQhE,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IAIhE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;CAoB/D"}

package/dist/storage/db/constraint-utils.d.ts

@@ -0,0 +1,16 @@
+export declare const POSTGRES_IDENTIFIER_MAX_LENGTH = 63;
+export declare function truncateIdentifier(value: string, maxLength?: number): string;
+/**
+ * Builds a constraint name with an optional schema prefix, truncated to fit
+ * within Postgres' identifier length limit.  The result is always lowercased
+ * because PostgreSQL folds unquoted identifiers to lowercase when storing them
+ * in system catalogs (pg_constraint.conname, pg_indexes.indexname, etc.).
+ * Without this normalisation, runtime lookups that compare a mixed-case name
+ * against the catalog would silently fail.
+ */
+export declare function buildConstraintName({ baseName, schemaName, maxLength, }: {
+    baseName: string;
+    schemaName?: string;
+    maxLength?: number;
+}): string;
+//# sourceMappingURL=constraint-utils.d.ts.map

package/dist/storage/db/constraint-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constraint-utils.d.ts","sourceRoot":"","sources":["../../../src/storage/db/constraint-utils.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,8BAA8B,KAAK,CAAC;AAEjD,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,SAAiC,GAAG,MAAM,CAapG;AAED;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CAAC,EAClC,QAAQ,EACR,UAAU,EACV,SAA0C,GAC3C,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,GAAG,MAAM,CAGT"}

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

@@ -0,0 +1,235 @@
+import type { ConnectionOptions } from 'node:tls';
+import { MastraBase } from '@mastra/core/base';
+import type { StorageColumn, TABLE_NAMES, CreateIndexOptions, IndexInfo, StorageIndexStats } from '@mastra/core/storage';
+import { Pool } from 'pg';
+import type { DbClient } from '../client.js';
+export type { DbClient } from '../client.js';
+/**
+ * Configuration for standalone domain usage.
+ * Accepts either:
+ * 1. An existing database client (Pool or PoolAdapter)
+ * 2. Config to create a new pool internally
+ */
+export type PgDomainConfig = PgDomainClientConfig | PgDomainPoolConfig | PgDomainRestConfig;
+/**
+ * Pass an existing database client (DbClient)
+ */
+export interface PgDomainClientConfig {
+    /** The database client */
+    client: DbClient;
+    /** 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[];
+}
+/**
+ * Pass an existing pg.Pool
+ */
+export interface PgDomainPoolConfig {
+    /** Pre-configured pg.Pool */
+    pool: Pool;
+    /** 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[];
+}
+/**
+ * Pass config to create a new pg.Pool 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 | ConnectionOptions;
+} | {
+    connectionString: string;
+    ssl?: boolean | ConnectionOptions;
+});
+/**
+ * Resolves PgDomainConfig to a database client and schema.
+ * Handles creating a new pool if config is provided.
+ */
+export declare function resolvePgConfig(config: PgDomainConfig): {
+    client: DbClient;
+    schemaName?: string;
+    skipDefaultIndexes?: boolean;
+    indexes?: CreateIndexOptions[];
+};
+export declare function getSchemaName(schema?: string): string;
+export declare function getTableName({ indexName, schemaName }: {
+    indexName: string;
+    schemaName?: string;
+}): string;
+export declare function generateTableSQL({ tableName, schema, schemaName, compositePrimaryKey, includeAllConstraints, }: {
+    tableName: TABLE_NAMES;
+    schema: Record<string, StorageColumn>;
+    schemaName?: string;
+    compositePrimaryKey?: string[];
+    /** When true, includes all constraints in the SQL (for exports). When false, some constraints are added at runtime after data migration. */
+    includeAllConstraints?: boolean;
+}): string;
+/**
+ * Generates a CREATE INDEX SQL statement from index options.
+ * Used by exportSchemas to produce index DDL without a database connection.
+ */
+export declare function generateIndexSQL(options: CreateIndexOptions, schemaName?: string): string;
+/**
+ * Generates the SQL for a timestamp trigger function and trigger on a table.
+ * Returns the DDL string without executing it.
+ */
+export declare function generateTimestampTriggerSQL(tableName: string, schemaName?: string): string;
+/**
+ * Internal config for PgDB - accepts already-resolved client
+ */
+export interface PgDBInternalConfig {
+    client: DbClient;
+    schemaName?: string;
+    skipDefaultIndexes?: boolean;
+}
+export declare class PgDB extends MastraBase {
+    client: DbClient;
+    schemaName?: string;
+    skipDefaultIndexes?: boolean;
+    constructor(config: PgDBInternalConfig);
+    hasColumn(table: string, column: string): Promise<boolean>;
+    /**
+     * Prepares values for insertion, handling JSONB columns by stringifying them
+     */
+    private prepareValuesForInsert;
+    /**
+     * Adds timestamp Z columns to a record if timestamp columns exist
+     */
+    private addTimestampZColumns;
+    /**
+     * Prepares a value for database operations
+     */
+    private prepareValue;
+    private setupSchema;
+    protected getDefaultValue(type: StorageColumn['type']): string;
+    insert({ tableName, record }: {
+        tableName: TABLE_NAMES;
+        record: Record<string, any>;
+    }): Promise<void>;
+    clearTable({ tableName }: {
+        tableName: TABLE_NAMES;
+    }): Promise<void>;
+    createTable({ tableName, schema, compositePrimaryKey, }: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+        compositePrimaryKey?: string[];
+    }): Promise<void>;
+    private setupTimestampTriggers;
+    /**
+     * 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 migrateSpansTable;
+    /**
+     * Deduplicates spans in the mastra_ai_spans table before adding the PRIMARY KEY constraint.
+     * Keeps spans based on priority: completed (endedAt NOT NULL) > most recent updatedAt > most recent createdAt.
+     *
+     * Note: This prioritizes migration completion over perfect data preservation.
+     * Old trace data may be lost, which is acceptable for this use case.
+     */
+    private deduplicateSpans;
+    /**
+     * Checks for duplicate (traceId, spanId) combinations in the spans table.
+     * Returns information about duplicates for logging/CLI purposes.
+     */
+    private checkForDuplicateSpans;
+    /**
+     * Checks if the PRIMARY KEY constraint on (traceId, spanId) already exists on the spans table.
+     * Used to skip deduplication when the constraint already exists (migration already complete).
+     */
+    private spansPrimaryKeyExists;
+    /**
+     * Adds the PRIMARY KEY constraint on (traceId, spanId) to the spans table.
+     * Should be called AFTER deduplication to ensure no duplicate key violations.
+     */
+    private addSpansPrimaryKey;
+    /**
+     * Manually run the spans migration to deduplicate and add the unique constraint.
+     * This is intended to be called from the CLI when duplicates are detected.
+     *
+     * @returns Migration result with status and details
+     */
+    migrateSpans(): Promise<{
+        success: boolean;
+        alreadyMigrated: boolean;
+        duplicatesRemoved: number;
+        message: string;
+    }>;
+    /**
+     * Check migration status for the spans table.
+     * Returns information about whether migration is needed.
+     */
+    checkSpansMigrationStatus(): Promise<{
+        needsMigration: boolean;
+        hasDuplicates: boolean;
+        duplicateCount: number;
+        constraintExists: boolean;
+        tableName: string;
+    }>;
+    /**
+     * Alters table schema to add columns if they don't exist
+     * @param tableName Name of the table
+     * @param schema Schema of the table
+     * @param ifNotExists Array of column names to add if they don't exist
+     */
+    alterTable({ tableName, schema, ifNotExists, }: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+        ifNotExists: string[];
+    }): Promise<void>;
+    load<R>({ tableName, keys }: {
+        tableName: TABLE_NAMES;
+        keys: Record<string, string>;
+    }): Promise<R | null>;
+    batchInsert({ tableName, records }: {
+        tableName: TABLE_NAMES;
+        records: Record<string, any>[];
+    }): Promise<void>;
+    dropTable({ tableName }: {
+        tableName: TABLE_NAMES;
+    }): Promise<void>;
+    createIndex(options: CreateIndexOptions): Promise<void>;
+    dropIndex(indexName: string): Promise<void>;
+    listIndexes(tableName?: string): Promise<IndexInfo[]>;
+    describeIndex(indexName: string): Promise<StorageIndexStats>;
+    update({ tableName, keys, data, }: {
+        tableName: TABLE_NAMES;
+        keys: Record<string, any>;
+        data: Record<string, any>;
+    }): Promise<void>;
+    batchUpdate({ tableName, updates, }: {
+        tableName: TABLE_NAMES;
+        updates: Array<{
+            keys: Record<string, any>;
+            data: Record<string, any>;
+        }>;
+    }): Promise<void>;
+    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,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAClD,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;AAE9B,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAK1C,YAAY,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAE1C;;;;;GAKG;AACH,MAAM,MAAM,cAAc,GAAG,oBAAoB,GAAG,kBAAkB,GAAG,kBAAkB,CAAC;AAE5F;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,0BAA0B;IAC1B,MAAM,EAAE,QAAQ,CAAC;IACjB,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,WAAW,kBAAkB;IACjC,6BAA6B;IAC7B,IAAI,EAAE,IAAI,CAAC;IACX,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,iBAAiB,CAAC;CACnC,GACD;IACE,gBAAgB,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,GAAG,iBAAiB,CAAC;CACnC,CACJ,CAAC;AAEF;;;GAGG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,cAAc,GAAG;IACvD,MAAM,EAAE,QAAQ,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,OAAO,CAAC,EAAE,kBAAkB,EAAE,CAAC;CAChC,CA6CA;AAED,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,UAE5C;AAED,wBAAgB,YAAY,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,UAKjG;AAaD,wBAAgB,gBAAgB,CAAC,EAC/B,SAAS,EACT,MAAM,EACN,UAAU,EACV,mBAAmB,EACnB,qBAA6B,GAC9B,EAAE;IACD,SAAS,EAAE,WAAW,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;IACtC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC/B,4IAA4I;IAC5I,qBAAqB,CAAC,EAAE,OAAO,CAAC;CACjC,GAAG,MAAM,CAyGT;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,kBAAkB,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,CA0BzF;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,CA8B1F;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,QAAQ,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAOD,qBAAa,IAAK,SAAQ,UAAU;IAC3B,MAAM,EAAE,QAAQ,CAAC;IACjB,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,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;IAiDrG,UAAU,CAAC,EAAE,SAAS,EAAE,EAAE;QAAE,SAAS,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCpE,WAAW,CAAC,EAChB,SAAS,EACT,MAAM,EACN,mBAAmB,GACpB,EAAE;QACD,SAAS,EAAE,WAAW,CAAC;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACtC,mBAAmB,CAAC,EAAE,MAAM,EAAE,CAAC;KAChC,GAAG,OAAO,CAAC,IAAI,CAAC;YAsFH,sBAAsB;IAYpC;;;OAGG;YACW,iBAAiB;IAqD/B;;;;;;OAMG;YACW,gBAAgB;IAgF9B;;;OAGG;YACW,sBAAsB;IAgCpC;;;OAGG;YACW,qBAAqB;IAgBnC;;;OAGG;YACW,kBAAkB;IAiDhC;;;;;OAKG;IACG,YAAY,IAAI,OAAO,CAAC;QAC5B,OAAO,EAAE,OAAO,CAAC;QACjB,eAAe,EAAE,OAAO,CAAC;QACzB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IAyCF;;;OAGG;IACG,yBAAyB,IAAI,OAAO,CAAC;QACzC,cAAc,EAAE,OAAO,CAAC;QACxB,aAAa,EAAE,OAAO,CAAC;QACvB,cAAc,EAAE,MAAM,CAAC;QACvB,gBAAgB,EAAE,OAAO,CAAC;QAC1B,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IAwBF;;;;;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,52 @@
+import { AgentsStorage } from '@mastra/core/storage';
+import type { StorageAgentType, StorageCreateAgentInput, StorageUpdateAgentInput, StorageListAgentsInput, StorageListAgentsOutput, CreateIndexOptions } from '@mastra/core/storage';
+import type { AgentVersion, CreateVersionInput, ListVersionsInput, ListVersionsOutput } from '@mastra/core/storage/domains/agents';
+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", "mastra_agent_versions"];
+    constructor(config: PgDomainConfig);
+    /**
+     * Returns all DDL statements for this domain: tables.
+     * Used by exportSchemas to produce a complete, reproducible schema export.
+     */
+    static getExportDDL(schemaName?: string): string[];
+    /**
+     * 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;
+    getById(id: string): Promise<StorageAgentType | null>;
+    create(input: {
+        agent: StorageCreateAgentInput;
+    }): Promise<StorageAgentType>;
+    update(input: StorageUpdateAgentInput): Promise<StorageAgentType>;
+    delete(id: string): Promise<void>;
+    list(args?: StorageListAgentsInput): Promise<StorageListAgentsOutput>;
+    createVersion(input: CreateVersionInput): Promise<AgentVersion>;
+    getVersion(id: string): Promise<AgentVersion | null>;
+    getVersionByNumber(agentId: string, versionNumber: number): Promise<AgentVersion | null>;
+    getLatestVersion(agentId: string): Promise<AgentVersion | null>;
+    listVersions(input: ListVersionsInput): Promise<ListVersionsOutput>;
+    deleteVersion(id: string): Promise<void>;
+    deleteVersionsByParentId(entityId: string): Promise<void>;
+    countVersions(agentId: string): Promise<number>;
+    private serializeInstructions;
+    private deserializeInstructions;
+    private parseVersionRow;
+}
+//# 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,EAOd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,gBAAgB,EAChB,uBAAuB,EACvB,uBAAuB,EACvB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EAEnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,YAAY,EACZ,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EACnB,MAAM,qCAAqC,CAAC;AAE7C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAG/C,qBAAa,QAAS,SAAQ,aAAa;;IAMzC,oCAAoC;IACpC,MAAM,CAAC,QAAQ,CAAC,cAAc,sDAAiD;gBAEnE,MAAM,EAAE,cAAc;IAUlC;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE;IAkBlD;;;OAGG;IACH,0BAA0B,IAAI,kBAAkB,EAAE;IAIlD;;;OAGG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOrC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAmN3B;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAK1C,OAAO,CAAC,SAAS;IA4BjB,OAAO,CAAC,QAAQ;IAYV,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAyBrD,MAAM,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,uBAAuB,CAAA;KAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA4E5E,MAAM,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAwFjE,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBjC,IAAI,CAAC,IAAI,CAAC,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IA8FrE,aAAa,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,YAAY,CAAC;IA+D/D,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAwBpD,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IA2BxF,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IA2B/D,YAAY,CAAC,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAoEnE,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBxC,wBAAwB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBzD,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAyBrD,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,uBAAuB;IAW/B,OAAO,CAAC,eAAe;CA4BxB"}

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

@@ -0,0 +1,17 @@
+import { BlobStore } from '@mastra/core/storage';
+import type { StorageBlobEntry } from '@mastra/core/storage';
+import type { PgDomainConfig } from '../../db/index.js';
+export declare class BlobsPG extends BlobStore {
+    #private;
+    static readonly MANAGED_TABLES: readonly ["mastra_skill_blobs"];
+    constructor(config: PgDomainConfig);
+    init(): Promise<void>;
+    put(entry: StorageBlobEntry): Promise<void>;
+    get(hash: string): Promise<StorageBlobEntry | null>;
+    has(hash: string): Promise<boolean>;
+    delete(hash: string): Promise<boolean>;
+    putMany(entries: StorageBlobEntry[]): Promise<void>;
+    getMany(hashes: string[]): Promise<Map<string, StorageBlobEntry>>;
+    dangerouslyClearAll(): Promise<void>;
+}
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/blobs/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAoC,MAAM,sBAAsB,CAAC;AACnF,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAG7D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAG/C,qBAAa,OAAQ,SAAQ,SAAS;;IAIpC,MAAM,CAAC,QAAQ,CAAC,cAAc,kCAAgC;gBAElD,MAAM,EAAE,cAAc;IAO5B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAOrB,GAAG,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAY3C,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAOnD,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAMnC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAMtC,OAAO,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAOnD,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IAiBjE,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;CAa3C"}