@mastra/pg 1.9.4 → 1.10.0-alpha.1

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_SCHEMAS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, ChannelsStorage, TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, DATASETS_SCHEMA, TABLE_CONFIGS, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, ensureDate, safelyParseJSON, ExperimentsStorage, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, getDefaultValue, transformScoreRow as transformScoreRow$1, getSqlType } from '@mastra/core/storage';
+import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_SCHEMAS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, ChannelsStorage, TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, DATASETS_SCHEMA, TABLE_CONFIGS, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, ensureDate, safelyParseJSON, ExperimentsStorage, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, SchedulesStorage, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, getDefaultValue, transformScoreRow as transformScoreRow$1, getSqlType } from '@mastra/core/storage';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import { MastraVector, validateTopK, validateUpsertInput } from '@mastra/core/vector';
 import { Mutex } from 'async-mutex';
@@ -11352,6 +11352,284 @@ var PromptBlocksPG = class _PromptBlocksPG extends PromptBlocksStorage {
     };
   }
 };
+function getSchemaName5(schema) {
+  return schema ? `"${schema}"` : '"public"';
+}
+function getTableName5(table, schema) {
+  const quoted = `"${table}"`;
+  return schema ? `${schema}.${quoted}` : quoted;
+}
+function parseJson2(val) {
+  if (val == null) return void 0;
+  if (typeof val === "string") {
+    try {
+      return JSON.parse(val);
+    } catch {
+      return val;
+    }
+  }
+  return val;
+}
+function toNumber(val) {
+  if (typeof val === "bigint") return Number(val);
+  return Number(val);
+}
+function rowToSchedule(row) {
+  const target = parseJson2(row.target);
+  if (!target) {
+    throw new Error(`Schedule row ${row.id} has invalid target`);
+  }
+  const schedule = {
+    id: String(row.id),
+    target,
+    cron: String(row.cron),
+    status: String(row.status),
+    nextFireAt: toNumber(row.next_fire_at),
+    createdAt: toNumber(row.created_at),
+    updatedAt: toNumber(row.updated_at)
+  };
+  if (row.timezone != null) schedule.timezone = String(row.timezone);
+  if (row.last_fire_at != null) schedule.lastFireAt = toNumber(row.last_fire_at);
+  if (row.last_run_id != null) schedule.lastRunId = String(row.last_run_id);
+  const metadata = parseJson2(row.metadata);
+  if (metadata !== void 0) schedule.metadata = metadata;
+  if (row.owner_type != null) schedule.ownerType = String(row.owner_type);
+  if (row.owner_id != null) schedule.ownerId = String(row.owner_id);
+  return schedule;
+}
+function rowToTrigger(row) {
+  const trigger = {
+    id: row.id != null ? String(row.id) : void 0,
+    scheduleId: String(row.schedule_id),
+    runId: row.run_id != null ? String(row.run_id) : null,
+    scheduledFireAt: toNumber(row.scheduled_fire_at),
+    actualFireAt: toNumber(row.actual_fire_at),
+    outcome: String(row.outcome),
+    triggerKind: row.trigger_kind != null ? String(row.trigger_kind) : "schedule-fire"
+  };
+  if (row.error != null) trigger.error = String(row.error);
+  if (row.parent_trigger_id != null) trigger.parentTriggerId = String(row.parent_trigger_id);
+  const metadata = parseJson2(row.metadata);
+  if (metadata !== void 0) trigger.metadata = metadata;
+  return trigger;
+}
+var SchedulesPG = class extends SchedulesStorage {
+  #db;
+  #client;
+  #schema;
+  /** Tables managed by this domain */
+  static MANAGED_TABLES = [TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS];
+  constructor(config) {
+    super();
+    const { client, schemaName, skipDefaultIndexes } = resolvePgConfig(config);
+    this.#client = client;
+    this.#db = new PgDB({ client, schemaName, skipDefaultIndexes });
+    this.#schema = schemaName || "public";
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_SCHEDULES,
+      schema: TABLE_SCHEMAS[TABLE_SCHEDULES]
+    });
+    await this.#db.createTable({
+      tableName: TABLE_SCHEDULE_TRIGGERS,
+      schema: TABLE_SCHEMAS[TABLE_SCHEDULE_TRIGGERS]
+    });
+  }
+  static getExportDDL(schemaName) {
+    return [
+      generateTableSQL({
+        tableName: TABLE_SCHEDULES,
+        schema: TABLE_SCHEMAS[TABLE_SCHEDULES],
+        schemaName,
+        includeAllConstraints: true
+      }),
+      generateTableSQL({
+        tableName: TABLE_SCHEDULE_TRIGGERS,
+        schema: TABLE_SCHEMAS[TABLE_SCHEDULE_TRIGGERS],
+        schemaName,
+        includeAllConstraints: true
+      })
+    ];
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_SCHEDULE_TRIGGERS });
+    await this.#db.clearTable({ tableName: TABLE_SCHEDULES });
+  }
+  #table(tableName) {
+    const schema = parseSqlIdentifier(this.#schema, "schema name");
+    return getTableName5(tableName, getSchemaName5(schema));
+  }
+  async createSchedule(schedule) {
+    const existing = await this.getSchedule(schedule.id);
+    if (existing) {
+      throw new Error(`Schedule with id "${schedule.id}" already exists`);
+    }
+    await this.#db.insert({
+      tableName: TABLE_SCHEDULES,
+      record: {
+        id: schedule.id,
+        target: schedule.target,
+        cron: schedule.cron,
+        timezone: schedule.timezone ?? null,
+        status: schedule.status,
+        next_fire_at: schedule.nextFireAt,
+        last_fire_at: schedule.lastFireAt ?? null,
+        last_run_id: schedule.lastRunId ?? null,
+        created_at: schedule.createdAt,
+        updated_at: schedule.updatedAt,
+        metadata: schedule.metadata ?? null,
+        owner_type: schedule.ownerType ?? null,
+        owner_id: schedule.ownerId ?? null
+      }
+    });
+    return schedule;
+  }
+  async getSchedule(id) {
+    const row = await this.#client.oneOrNone(
+      `SELECT * FROM ${this.#table(TABLE_SCHEDULES)} WHERE id = $1`,
+      [id]
+    );
+    return row ? rowToSchedule(row) : null;
+  }
+  async listSchedules(filter) {
+    const conditions = [];
+    const params = [];
+    if (filter?.status) {
+      params.push(filter.status);
+      conditions.push(`status = $${params.length}`);
+    }
+    if (filter?.workflowId) {
+      params.push(filter.workflowId);
+      conditions.push(`target->>'workflowId' = $${params.length}`);
+    }
+    if (filter?.ownerType !== void 0) {
+      if (filter.ownerType === null) {
+        conditions.push("owner_type IS NULL");
+      } else {
+        params.push(filter.ownerType);
+        conditions.push(`owner_type = $${params.length}`);
+      }
+    }
+    if (filter?.ownerId !== void 0) {
+      if (filter.ownerId === null) {
+        conditions.push("owner_id IS NULL");
+      } else {
+        params.push(filter.ownerId);
+        conditions.push(`owner_id = $${params.length}`);
+      }
+    }
+    const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+    const rows = await this.#client.manyOrNone(
+      `SELECT * FROM ${this.#table(TABLE_SCHEDULES)} ${where} ORDER BY created_at ASC`,
+      params
+    );
+    return rows.map(rowToSchedule);
+  }
+  async listDueSchedules(now, limit) {
+    const cap = limit ?? 100;
+    const rows = await this.#client.manyOrNone(
+      `SELECT * FROM ${this.#table(TABLE_SCHEDULES)}
+       WHERE status = $1 AND next_fire_at <= $2
+       ORDER BY next_fire_at ASC
+       LIMIT $3`,
+      ["active", now, cap]
+    );
+    return rows.map(rowToSchedule);
+  }
+  async updateSchedule(id, patch) {
+    const setClauses = [];
+    const params = [];
+    const push = (frag, value) => {
+      params.push(value);
+      setClauses.push(frag.replace("?", `$${params.length}`));
+    };
+    if ("cron" in patch && patch.cron !== void 0) push("cron = ?", patch.cron);
+    if ("timezone" in patch) push("timezone = ?", patch.timezone ?? null);
+    if ("status" in patch && patch.status !== void 0) push("status = ?", patch.status);
+    if ("nextFireAt" in patch && patch.nextFireAt !== void 0) push("next_fire_at = ?", patch.nextFireAt);
+    if ("target" in patch && patch.target !== void 0) {
+      push("target = ?::jsonb", JSON.stringify(patch.target));
+    }
+    if ("metadata" in patch) {
+      push("metadata = ?::jsonb", patch.metadata != null ? JSON.stringify(patch.metadata) : null);
+    }
+    if ("ownerType" in patch) push("owner_type = ?", patch.ownerType ?? null);
+    if ("ownerId" in patch) push("owner_id = ?", patch.ownerId ?? null);
+    push("updated_at = ?", Date.now());
+    if (setClauses.length === 1) {
+      const existing = await this.getSchedule(id);
+      if (!existing) throw new Error(`Schedule ${id} not found`);
+      return existing;
+    }
+    params.push(id);
+    await this.#client.none(
+      `UPDATE ${this.#table(TABLE_SCHEDULES)} SET ${setClauses.join(", ")} WHERE id = $${params.length}`,
+      params
+    );
+    const updated = await this.getSchedule(id);
+    if (!updated) throw new Error(`Schedule ${id} not found`);
+    return updated;
+  }
+  async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId) {
+    const result = await this.#client.query(
+      `UPDATE ${this.#table(TABLE_SCHEDULES)}
+       SET next_fire_at = $1, last_fire_at = $2, last_run_id = $3, updated_at = $4
+       WHERE id = $5 AND next_fire_at = $6 AND status = $7`,
+      [newNextFireAt, lastFireAt, lastRunId, Date.now(), id, expectedNextFireAt, "active"]
+    );
+    return (result.rowCount ?? 0) > 0;
+  }
+  async deleteSchedule(id) {
+    await this.#client.none(`DELETE FROM ${this.#table(TABLE_SCHEDULE_TRIGGERS)} WHERE schedule_id = $1`, [id]);
+    await this.#client.none(`DELETE FROM ${this.#table(TABLE_SCHEDULES)} WHERE id = $1`, [id]);
+  }
+  async recordTrigger(trigger) {
+    const id = trigger.id ?? crypto.randomUUID();
+    await this.#db.insert({
+      tableName: TABLE_SCHEDULE_TRIGGERS,
+      record: {
+        id,
+        schedule_id: trigger.scheduleId,
+        run_id: trigger.runId,
+        scheduled_fire_at: trigger.scheduledFireAt,
+        actual_fire_at: trigger.actualFireAt,
+        outcome: trigger.outcome,
+        error: trigger.error ?? null,
+        trigger_kind: trigger.triggerKind ?? "schedule-fire",
+        parent_trigger_id: trigger.parentTriggerId ?? null,
+        metadata: trigger.metadata ?? null
+      }
+    });
+  }
+  async listTriggers(scheduleId, opts) {
+    const conditions = [];
+    const params = [];
+    params.push(scheduleId);
+    conditions.push(`schedule_id = $${params.length}`);
+    if (opts?.fromActualFireAt != null) {
+      params.push(opts.fromActualFireAt);
+      conditions.push(`actual_fire_at >= $${params.length}`);
+    }
+    if (opts?.toActualFireAt != null) {
+      params.push(opts.toActualFireAt);
+      conditions.push(`actual_fire_at < $${params.length}`);
+    }
+    let limitClause = "";
+    if (opts?.limit != null) {
+      params.push(Math.floor(opts.limit));
+      limitClause = `LIMIT $${params.length}`;
+    }
+    const rows = await this.#client.manyOrNone(
+      `SELECT * FROM ${this.#table(TABLE_SCHEDULE_TRIGGERS)}
+       WHERE ${conditions.join(" AND ")}
+       ORDER BY actual_fire_at DESC
+       ${limitClause}`,
+      params
+    );
+    return rows.map(rowToTrigger);
+  }
+};
 var SNAPSHOT_FIELDS4 = [
   "name",
   "description",
@@ -12019,10 +12297,10 @@ var ScorerDefinitionsPG = class _ScorerDefinitionsPG extends ScorerDefinitionsSt
     };
   }
 };
-function getSchemaName5(schema) {
+function getSchemaName6(schema) {
   return schema ? `"${schema}"` : '"public"';
 }
-function getTableName5({ indexName, schemaName }) {
+function getTableName6({ indexName, schemaName }) {
   const quotedIndexName = `"${indexName}"`;
   return schemaName ? `${schemaName}.${quotedIndexName}` : quotedIndexName;
 }
@@ -12136,7 +12414,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
   async getScoreById({ id }) {
     try {
       const result = await this.#db.client.oneOrNone(
-        `SELECT * FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE id = $1`,
+        `SELECT * FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE id = $1`,
         [id]
       );
       return result ? transformScoreRow(result) : null;
@@ -12176,7 +12454,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
       }
       const whereClause = conditions.join(" AND ");
       const total = await this.#db.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE ${whereClause}`,
+        `SELECT COUNT(*) FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE ${whereClause}`,
         queryParams
       );
       const { page, perPage: perPageInput } = pagination;
@@ -12196,7 +12474,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
       const limitValue = perPageInput === false ? Number(total?.count) : perPage;
       const end = perPageInput === false ? Number(total?.count) : start + perPage;
       const result = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE ${whereClause} ORDER BY "createdAt" DESC LIMIT $${paramIndex++} OFFSET $${paramIndex++}`,
+        `SELECT * FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE ${whereClause} ORDER BY "createdAt" DESC LIMIT $${paramIndex++} OFFSET $${paramIndex++}`,
         [...queryParams, limitValue, start]
       );
       return {
@@ -12291,7 +12569,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
   }) {
     try {
       const total = await this.#db.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE "runId" = $1`,
+        `SELECT COUNT(*) FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE "runId" = $1`,
         [runId]
       );
       const { page, perPage: perPageInput } = pagination;
@@ -12311,7 +12589,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
       const limitValue = perPageInput === false ? Number(total?.count) : perPage;
       const end = perPageInput === false ? Number(total?.count) : start + perPage;
       const result = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE "runId" = $1 LIMIT $2 OFFSET $3`,
+        `SELECT * FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE "runId" = $1 LIMIT $2 OFFSET $3`,
         [runId, limitValue, start]
       );
       return {
@@ -12341,7 +12619,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
   }) {
     try {
       const total = await this.#db.client.oneOrNone(
-        `SELECT COUNT(*) FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE "entityId" = $1 AND "entityType" = $2`,
+        `SELECT COUNT(*) FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE "entityId" = $1 AND "entityType" = $2`,
         [entityId, entityType]
       );
       const { page, perPage: perPageInput } = pagination;
@@ -12361,7 +12639,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
       const limitValue = perPageInput === false ? Number(total?.count) : perPage;
       const end = perPageInput === false ? Number(total?.count) : start + perPage;
       const result = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) })} WHERE "entityId" = $1 AND "entityType" = $2 LIMIT $3 OFFSET $4`,
+        `SELECT * FROM ${getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) })} WHERE "entityId" = $1 AND "entityType" = $2 LIMIT $3 OFFSET $4`,
         [entityId, entityType, limitValue, start]
       );
       return {
@@ -12390,7 +12668,7 @@ var ScoresPG = class _ScoresPG extends ScoresStorage {
     pagination
   }) {
     try {
-      const tableName = getTableName5({ indexName: TABLE_SCORERS, schemaName: getSchemaName5(this.#schema) });
+      const tableName = getTableName6({ indexName: TABLE_SCORERS, schemaName: getSchemaName6(this.#schema) });
       const countSQLResult = await this.#db.client.oneOrNone(
         `SELECT COUNT(*) as count FROM ${tableName} WHERE "traceId" = $1 AND "spanId" = $2`,
         [traceId, spanId]
@@ -13093,10 +13371,10 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
     };
   }
 };
-function getSchemaName6(schema) {
+function getSchemaName7(schema) {
   return schema ? `"${schema}"` : '"public"';
 }
-function getTableName6({ indexName, schemaName }) {
+function getTableName7({ indexName, schemaName }) {
   const quotedIndexName = `"${indexName}"`;
   return schemaName ? `${schemaName}.${quotedIndexName}` : quotedIndexName;
 }
@@ -13208,7 +13486,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
   }) {
     try {
       return await this.#db.client.tx(async (t) => {
-        const tableName = getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) });
+        const tableName = getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) });
         const existingSnapshotResult = await t.oneOrNone(
           `SELECT snapshot FROM ${tableName} WHERE workflow_name = $1 AND run_id = $2 FOR UPDATE`,
           [workflowName, runId]
@@ -13269,7 +13547,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
   }) {
     try {
       return await this.#db.client.tx(async (t) => {
-        const tableName = getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) });
+        const tableName = getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) });
         const existingSnapshotResult = await t.oneOrNone(
           `SELECT snapshot FROM ${tableName} WHERE workflow_name = $1 AND run_id = $2 FOR UPDATE`,
           [workflowName, runId]
@@ -13319,7 +13597,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
       const updatedAtValue = updatedAt ? updatedAt : now;
       const sanitizedSnapshot = sanitizeJsonForPg(JSON.stringify(snapshot));
       await this.#db.client.none(
-        `INSERT INTO ${getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) })} (workflow_name, run_id, "resourceId", snapshot, "createdAt", "updatedAt")
+        `INSERT INTO ${getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) })} (workflow_name, run_id, "resourceId", snapshot, "createdAt", "updatedAt")
                  VALUES ($1, $2, $3, $4, $5, $6)
                  ON CONFLICT (workflow_name, run_id) DO UPDATE
                  SET "resourceId" = $3, snapshot = $4, "updatedAt" = $6`,
@@ -13377,7 +13655,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
       }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const query = `
-          SELECT * FROM ${getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) })}
+          SELECT * FROM ${getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) })}
           ${whereClause}
           ORDER BY "createdAt" DESC LIMIT 1
         `;
@@ -13405,7 +13683,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
   async deleteWorkflowRunById({ runId, workflowName }) {
     try {
       await this.#db.client.none(
-        `DELETE FROM ${getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) })} WHERE run_id = $1 AND workflow_name = $2`,
+        `DELETE FROM ${getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) })} WHERE run_id = $1 AND workflow_name = $2`,
         [runId, workflowName]
       );
     } catch (error) {
@@ -13473,7 +13751,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
       const usePagination = typeof perPage === "number" && typeof page === "number";
       if (usePagination) {
         const countResult = await this.#db.client.one(
-          `SELECT COUNT(*) as count FROM ${getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) })} ${whereClause}`,
+          `SELECT COUNT(*) as count FROM ${getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) })} ${whereClause}`,
           values
         );
         total = Number(countResult.count);
@@ -13481,7 +13759,7 @@ var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
       const normalizedPerPage = usePagination ? normalizePerPage(perPage, Number.MAX_SAFE_INTEGER) : 0;
       const offset = usePagination ? page * normalizedPerPage : void 0;
       const query = `
-          SELECT * FROM ${getTableName6({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName6(this.#schema) })}
+          SELECT * FROM ${getTableName7({ indexName: TABLE_WORKFLOW_SNAPSHOT, schemaName: getSchemaName7(this.#schema) })}
           ${whereClause}
           ORDER BY "createdAt" DESC
           ${usePagination ? ` LIMIT $${paramIndex} OFFSET $${paramIndex + 1}` : ""}
@@ -14206,7 +14484,8 @@ var ALL_DOMAINS = [
   DatasetsPG,
   ExperimentsPG,
   BackgroundTasksPG,
-  ChannelsPG
+  ChannelsPG,
+  SchedulesPG
 ];
 function exportSchemas(schemaName) {
   const statements = [];
@@ -14262,7 +14541,8 @@ var PostgresStore = class extends MastraCompositeStore {
         datasets: new DatasetsPG(domainConfig),
         experiments: new ExperimentsPG(domainConfig),
         backgroundTasks: new BackgroundTasksPG(domainConfig),
-        channels: new ChannelsPG(domainConfig)
+        channels: new ChannelsPG(domainConfig),
+        schedules: new SchedulesPG(domainConfig)
       };
     } catch (e) {
       throw new MastraError(
@@ -14451,6 +14731,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, MCPClientsPG, MCPServersPG, MemoryPG, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
+export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, MCPClientsPG, MCPServersPG, MemoryPG, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map