@mastra/pg 1.12.0 → 1.12.1-alpha.0

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, FavoritesStorage, TABLE_FAVORITES, 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 { 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, FavoritesStorage, TABLE_FAVORITES, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, NotificationsStorage, TABLE_NOTIFICATIONS, 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';
@@ -10401,6 +10401,348 @@ ${unreflectedContent}` : bufferedReflection;
     }
   }
 };
+var statusTimestamp = (status, now) => {
+  if (status === "delivered") return { deliveredAt: now };
+  if (status === "seen") return { seenAt: now };
+  if (status === "dismissed") return { dismissedAt: now };
+  if (status === "archived") return { archivedAt: now };
+  if (status === "discarded") return { discardedAt: now };
+  return {};
+};
+var parseDate = (value) => {
+  if (value == null) return void 0;
+  return value instanceof Date ? value : new Date(String(value));
+};
+var cloneValue = (value) => value === void 0 ? void 0 : structuredClone(value);
+function rowToNotification(row) {
+  return {
+    id: String(row.id),
+    threadId: String(row.threadId),
+    source: String(row.source),
+    kind: String(row.kind),
+    priority: String(row.priority),
+    status: String(row.status),
+    summary: String(row.summary),
+    payload: parseJsonResilient(row.payload),
+    resourceId: row.resourceId == null ? void 0 : String(row.resourceId),
+    agentId: row.agentId == null ? void 0 : String(row.agentId),
+    sourceId: row.sourceId == null ? void 0 : String(row.sourceId),
+    dedupeKey: row.dedupeKey == null ? void 0 : String(row.dedupeKey),
+    coalesceKey: row.coalesceKey == null ? void 0 : String(row.coalesceKey),
+    coalescedCount: Number(row.coalescedCount ?? 1),
+    attributes: parseJsonResilient(row.attributes),
+    createdAt: parseDate(row.createdAt) ?? /* @__PURE__ */ new Date(),
+    updatedAt: parseDate(row.updatedAt) ?? /* @__PURE__ */ new Date(),
+    deliveredAt: parseDate(row.deliveredAt),
+    seenAt: parseDate(row.seenAt),
+    dismissedAt: parseDate(row.dismissedAt),
+    archivedAt: parseDate(row.archivedAt),
+    discardedAt: parseDate(row.discardedAt),
+    deliverAt: parseDate(row.deliverAt),
+    summaryAt: parseDate(row.summaryAt),
+    deliveryReason: row.deliveryReason == null ? void 0 : String(row.deliveryReason),
+    deliveryAttempts: Number(row.deliveryAttempts ?? 0),
+    lastDeliveryAttemptAt: parseDate(row.lastDeliveryAttemptAt),
+    lastDeliveryError: row.lastDeliveryError == null ? void 0 : String(row.lastDeliveryError),
+    deliveredSignalId: row.deliveredSignalId == null ? void 0 : String(row.deliveredSignalId),
+    summarySignalId: row.summarySignalId == null ? void 0 : String(row.summarySignalId),
+    metadata: parseJsonResilient(row.metadata)
+  };
+}
+function addArrayFilter(conditions, args, column, value) {
+  if (!value) return;
+  const values = Array.isArray(value) ? value : [value];
+  const start = args.length + 1;
+  conditions.push(`"${column}" IN (${values.map((_, index) => `$${start + index}`).join(", ")})`);
+  args.push(...values);
+}
+function normalizeRecordForInsert(record) {
+  return {
+    ...record,
+    payload: record.payload ?? null,
+    attributes: record.attributes ?? null,
+    metadata: record.metadata ?? null,
+    resourceId: record.resourceId ?? null,
+    agentId: record.agentId ?? null,
+    sourceId: record.sourceId ?? null,
+    dedupeKey: record.dedupeKey ?? null,
+    coalesceKey: record.coalesceKey ?? null,
+    deliveredAt: record.deliveredAt ?? null,
+    seenAt: record.seenAt ?? null,
+    dismissedAt: record.dismissedAt ?? null,
+    archivedAt: record.archivedAt ?? null,
+    discardedAt: record.discardedAt ?? null,
+    deliverAt: record.deliverAt ?? null,
+    summaryAt: record.summaryAt ?? null,
+    deliveryReason: record.deliveryReason ?? null,
+    lastDeliveryAttemptAt: record.lastDeliveryAttemptAt ?? null,
+    lastDeliveryError: record.lastDeliveryError ?? null,
+    deliveredSignalId: record.deliveredSignalId ?? null,
+    summarySignalId: record.summarySignalId ?? null
+  };
+}
+var NotificationsPG = class _NotificationsPG extends NotificationsStorage {
+  #db;
+  #schema;
+  #skipDefaultIndexes;
+  #indexes;
+  static MANAGED_TABLES = [TABLE_NOTIFICATIONS];
+  constructor(config) {
+    super();
+    const { client, schemaName, skipDefaultIndexes, indexes } = resolvePgConfig(config);
+    this.#db = new PgDB({ client, schemaName, skipDefaultIndexes });
+    this.#schema = schemaName || "public";
+    this.#skipDefaultIndexes = skipDefaultIndexes;
+    this.#indexes = indexes?.filter((idx) => _NotificationsPG.MANAGED_TABLES.includes(idx.table));
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_NOTIFICATIONS,
+      schema: TABLE_SCHEMAS[TABLE_NOTIFICATIONS]
+    });
+    await this.createDefaultIndexes();
+    await this.createCustomIndexes();
+  }
+  static getDefaultIndexDefs(schemaPrefix) {
+    return [
+      {
+        name: `${schemaPrefix}idx_notifications_thread_status_updated`,
+        table: TABLE_NOTIFICATIONS,
+        columns: ["threadId", "status", "updatedAt"]
+      },
+      {
+        name: `${schemaPrefix}idx_notifications_coalescing`,
+        table: TABLE_NOTIFICATIONS,
+        columns: ["threadId", "source", "kind", "status", "agentId", "resourceId", "dedupeKey", "coalesceKey"]
+      },
+      {
+        name: `${schemaPrefix}idx_notifications_due`,
+        table: TABLE_NOTIFICATIONS,
+        columns: ["status", "deliverAt", "summaryAt"]
+      }
+    ];
+  }
+  static getExportDDL(schemaName) {
+    const statements = [];
+    const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+    const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
+    statements.push(
+      generateTableSQL({
+        tableName: TABLE_NOTIFICATIONS,
+        schema: TABLE_SCHEMAS[TABLE_NOTIFICATIONS],
+        schemaName,
+        includeAllConstraints: true
+      })
+    );
+    for (const idx of _NotificationsPG.getDefaultIndexDefs(schemaPrefix)) {
+      statements.push(generateIndexSQL(idx, schemaName));
+    }
+    return statements;
+  }
+  getDefaultIndexDefinitions() {
+    const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+    return _NotificationsPG.getDefaultIndexDefs(schemaPrefix);
+  }
+  async createDefaultIndexes() {
+    if (this.#skipDefaultIndexes) return;
+    for (const indexDef of this.getDefaultIndexDefinitions()) {
+      try {
+        await this.#db.createIndex(indexDef);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create index ${indexDef.name}:`, error);
+      }
+    }
+  }
+  async createCustomIndexes() {
+    if (!this.#indexes || this.#indexes.length === 0) return;
+    for (const indexDef of this.#indexes) {
+      try {
+        await this.#db.createIndex(indexDef);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create custom index ${indexDef.name}:`, error);
+      }
+    }
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_NOTIFICATIONS });
+  }
+  async updateNotificationRow(threadId, id, data) {
+    const entries = Object.entries(data).filter(([, value]) => value !== void 0);
+    if (entries.length === 0) return;
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_NOTIFICATIONS, schemaName });
+    const setColumns = entries.map(([key], index) => `"${parseSqlIdentifier(key, "column name")}" = $${index + 1}`);
+    const values = entries.map(([key, value]) => {
+      const columnSchema = TABLE_SCHEMAS[TABLE_NOTIFICATIONS][key];
+      if (columnSchema?.type === "jsonb" && value !== null) return JSON.stringify(value);
+      return value;
+    });
+    await this.#db.client.none(
+      `UPDATE ${tableName} SET ${setColumns.join(", ")} WHERE "threadId" = $${values.length + 1} AND "id" = $${values.length + 2}`,
+      [...values, threadId, id]
+    );
+  }
+  async createNotification(input) {
+    const existing = await this.findCoalescable(input);
+    if (existing) {
+      const now2 = /* @__PURE__ */ new Date();
+      const attributes = input.attributes ? { ...cloneValue(existing.attributes), ...cloneValue(input.attributes) } : cloneValue(existing.attributes);
+      const metadata = input.metadata ? { ...cloneValue(existing.metadata), ...cloneValue(input.metadata) } : cloneValue(existing.metadata);
+      await this.updateNotificationRow(existing.threadId, existing.id, {
+        summary: input.summary,
+        payload: cloneValue(input.payload ?? existing.payload) ?? null,
+        priority: input.priority ?? existing.priority,
+        attributes: attributes ?? null,
+        updatedAt: now2,
+        deliverAt: input.deliverAt ?? existing.deliverAt ?? null,
+        summaryAt: input.summaryAt ?? existing.summaryAt ?? null,
+        deliveryReason: input.deliveryReason ?? existing.deliveryReason ?? null,
+        coalescedCount: (existing.coalescedCount ?? 1) + 1,
+        metadata: metadata ?? null
+      });
+      const updated = await this.getNotification({ threadId: existing.threadId, id: existing.id });
+      if (!updated) throw new Error(`Notification ${existing.id} was not found for thread ${existing.threadId}`);
+      return updated;
+    }
+    const now = input.createdAt ?? /* @__PURE__ */ new Date();
+    const record = {
+      id: input.id ?? randomUUID(),
+      threadId: input.threadId,
+      source: input.source,
+      kind: input.kind,
+      priority: input.priority ?? "medium",
+      status: "pending",
+      summary: input.summary,
+      payload: cloneValue(input.payload),
+      resourceId: input.resourceId,
+      agentId: input.agentId,
+      sourceId: input.sourceId,
+      dedupeKey: input.dedupeKey,
+      coalesceKey: input.coalesceKey,
+      coalescedCount: 1,
+      attributes: cloneValue(input.attributes),
+      createdAt: now,
+      updatedAt: now,
+      deliverAt: input.deliverAt,
+      summaryAt: input.summaryAt,
+      deliveryReason: input.deliveryReason,
+      deliveryAttempts: 0,
+      metadata: cloneValue(input.metadata)
+    };
+    await this.#db.insert({ tableName: TABLE_NOTIFICATIONS, record: normalizeRecordForInsert(record) });
+    return record;
+  }
+  async listNotifications(input) {
+    const conditions = ['"threadId" = $1'];
+    const args = [input.threadId];
+    addArrayFilter(conditions, args, "status", input.status);
+    addArrayFilter(conditions, args, "priority", input.priority);
+    if (input.source) {
+      args.push(input.source);
+      conditions.push(`"source" = $${args.length}`);
+    }
+    if (input.resourceId) {
+      args.push(input.resourceId);
+      conditions.push(`"resourceId" = $${args.length}`);
+    }
+    if (input.agentId) {
+      args.push(input.agentId);
+      conditions.push(`"agentId" = $${args.length}`);
+    }
+    if (input.search) {
+      const search = `%${input.search.toLowerCase()}%`;
+      args.push(search, search);
+      conditions.push(`(LOWER("summary") LIKE $${args.length - 1} OR LOWER("kind") LIKE $${args.length})`);
+    }
+    const limit = input.limit ? ` LIMIT $${args.length + 1}` : "";
+    if (input.limit) args.push(input.limit);
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_NOTIFICATIONS, schemaName });
+    const rows = await this.#db.client.manyOrNone(
+      `SELECT * FROM ${tableName} WHERE ${conditions.join(" AND ")} ORDER BY "updatedAt" DESC${limit}`,
+      args
+    );
+    return rows.map((row) => rowToNotification(row));
+  }
+  async listDueNotifications(input) {
+    const conditions = [
+      '"status" = $1',
+      '(("deliverAt" IS NOT NULL AND "deliverAt" <= $2) OR ("summaryAt" IS NOT NULL AND "summaryAt" <= $3))'
+    ];
+    const args = ["pending", input.now, input.now];
+    if (input.agentId) {
+      args.push(input.agentId);
+      conditions.push(`"agentId" = $${args.length}`);
+    }
+    if (input.resourceId) {
+      args.push(input.resourceId);
+      conditions.push(`"resourceId" = $${args.length}`);
+    }
+    const limit = input.limit ? ` LIMIT $${args.length + 1}` : "";
+    if (input.limit) args.push(input.limit);
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_NOTIFICATIONS, schemaName });
+    const rows = await this.#db.client.manyOrNone(
+      `SELECT * FROM ${tableName} WHERE ${conditions.join(" AND ")} ORDER BY CASE WHEN "deliverAt" IS NULL THEN "summaryAt" WHEN "summaryAt" IS NULL THEN "deliverAt" WHEN "deliverAt" <= "summaryAt" THEN "deliverAt" ELSE "summaryAt" END ASC, "updatedAt" ASC${limit}`,
+      args
+    );
+    return rows.map((row) => rowToNotification(row));
+  }
+  async getNotification(input) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_NOTIFICATIONS, schemaName });
+    const row = await this.#db.client.oneOrNone(
+      `SELECT * FROM ${tableName} WHERE "threadId" = $1 AND "id" = $2 LIMIT 1`,
+      [input.threadId, input.id]
+    );
+    return row ? rowToNotification(row) : null;
+  }
+  async updateNotification(input) {
+    const existing = await this.getNotification({ threadId: input.threadId, id: input.id });
+    if (!existing) {
+      throw new Error(`Notification ${input.id} was not found for thread ${input.threadId}`);
+    }
+    const now = /* @__PURE__ */ new Date();
+    await this.updateNotificationRow(input.threadId, input.id, {
+      ...input.status ? { status: input.status, ...statusTimestamp(input.status, now) } : {},
+      ...input.summary !== void 0 ? { summary: input.summary } : {},
+      ...input.payload !== void 0 ? { payload: cloneValue(input.payload) } : {},
+      ...input.attributes !== void 0 ? { attributes: cloneValue(input.attributes) } : {},
+      ...input.metadata !== void 0 ? { metadata: cloneValue(input.metadata) } : {},
+      ...input.deliverAt !== void 0 ? { deliverAt: input.deliverAt } : {},
+      ...input.summaryAt !== void 0 ? { summaryAt: input.summaryAt } : {},
+      ...input.deliveryReason !== void 0 ? { deliveryReason: input.deliveryReason } : {},
+      ...input.deliveryAttempts !== void 0 ? { deliveryAttempts: input.deliveryAttempts } : {},
+      ...input.lastDeliveryAttemptAt !== void 0 ? { lastDeliveryAttemptAt: input.lastDeliveryAttemptAt } : {},
+      ...input.lastDeliveryError !== void 0 ? { lastDeliveryError: input.lastDeliveryError } : {},
+      ...input.deliveredSignalId !== void 0 ? { deliveredSignalId: input.deliveredSignalId } : {},
+      ...input.summarySignalId !== void 0 ? { summarySignalId: input.summarySignalId } : {},
+      updatedAt: now
+    });
+    const updated = await this.getNotification({ threadId: input.threadId, id: input.id });
+    if (!updated) throw new Error(`Notification ${input.id} was not found for thread ${input.threadId}`);
+    return updated;
+  }
+  async findCoalescable(input) {
+    if (!input.dedupeKey && !input.coalesceKey) return void 0;
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_NOTIFICATIONS, schemaName });
+    const row = await this.#db.client.oneOrNone(
+      `SELECT * FROM ${tableName} WHERE "threadId" = $1 AND "source" = $2 AND "kind" = $3 AND "status" = $4 AND (("agentId" = $5::text) OR ("agentId" IS NULL AND $5::text IS NULL)) AND (("resourceId" = $6::text) OR ("resourceId" IS NULL AND $6::text IS NULL)) AND (($7::text IS NOT NULL AND "dedupeKey" = $7::text) OR ($8::text IS NOT NULL AND "coalesceKey" = $8::text)) ORDER BY "updatedAt" DESC LIMIT 1`,
+      [
+        input.threadId,
+        input.source,
+        input.kind,
+        "pending",
+        input.agentId ?? null,
+        input.resourceId ?? null,
+        input.dedupeKey ?? null,
+        input.coalesceKey ?? null
+      ]
+    );
+    return row ? rowToNotification(row) : void 0;
+  }
+};
 var ObservabilityPG = class _ObservabilityPG extends ObservabilityStorage {
   #db;
   #schema;
@@ -14980,6 +15322,7 @@ var DEFAULT_MAX_CONNECTIONS = 20;
 var DEFAULT_IDLE_TIMEOUT_MS = 3e4;
 var ALL_DOMAINS = [
   MemoryPG,
+  NotificationsPG,
   ObservabilityPG,
   ScoresPG,
   ScorerDefinitionsPG,
@@ -15035,6 +15378,7 @@ var PostgresStore = class extends MastraCompositeStore {
         scores: new ScoresPG(domainConfig),
         workflows: new WorkflowsPG(domainConfig),
         memory: new MemoryPG(domainConfig),
+        notifications: new NotificationsPG(domainConfig),
         observability: new ObservabilityPG(domainConfig),
         agents: new AgentsPG(domainConfig),
         promptBlocks: new PromptBlocksPG(domainConfig),
@@ -15238,6 +15582,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, FavoritesPG, MCPClientsPG, MCPServersPG, MemoryPG, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
+export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, FavoritesPG, MCPClientsPG, MCPServersPG, MemoryPG, NotificationsPG, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map