@mastra/pg 1.10.1 → 1.11.0-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, 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, 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';
@@ -2013,6 +2013,7 @@ var PoolAdapter = class {
   constructor($pool) {
     this.$pool = $pool;
   }
+  $pool;
   connect() {
     return this.$pool.connect();
   }
@@ -2081,6 +2082,7 @@ var TransactionClient = class {
   constructor(client) {
     this.client = client;
   }
+  client;
   async none(query, values) {
     await this.client.query(query, values);
     return null;
@@ -3518,12 +3520,12 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
     await this.#db.alterTable({
       tableName: TABLE_AGENTS,
       schema: TABLE_SCHEMAS[TABLE_AGENTS],
-      ifNotExists: ["status", "authorId"]
+      ifNotExists: ["status", "authorId", "visibility", "favoriteCount"]
     });
     await this.#db.alterTable({
       tableName: TABLE_AGENT_VERSIONS,
       schema: TABLE_SCHEMAS[TABLE_AGENT_VERSIONS],
-      ifNotExists: ["mcpClients", "requestContextSchema", "workspace", "skills", "skillsFormat"]
+      ifNotExists: ["mcpClients", "requestContextSchema", "workspace", "skills", "skillsFormat", "browser"]
     });
     await this.#migrateToolsToJsonbFormat();
     await this.createDefaultIndexes();
@@ -3664,14 +3666,28 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
     }
   }
   /**
-   * Removes stale draft agent records that have no activeVersionId.
+   * Removes stale draft agent records that have no versions at all.
    * These are left behind when createAgent partially fails (inserts thin record
    * but fails to create the version due to schema mismatch).
+   *
+   * A legitimate draft (never published) will have rows in the versions table,
+   * so we must only delete records with zero associated versions.
    */
   async #cleanupStaleDrafts() {
     try {
-      const fullTableName = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
-      await this.#db.client.none(`DELETE FROM ${fullTableName} WHERE status = 'draft' AND "activeVersionId" IS NULL`);
+      const agentsTable = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
+      const versionsTable = getTableName2({
+        indexName: TABLE_AGENT_VERSIONS,
+        schemaName: getSchemaName2(this.#schema)
+      });
+      await this.#db.client.none(
+        `DELETE FROM ${agentsTable} a
+         WHERE a.status = 'draft'
+           AND a."activeVersionId" IS NULL
+           AND NOT EXISTS (
+             SELECT 1 FROM ${versionsTable} v WHERE v."agentId" = a.id
+           )`
+      );
     } catch {
     }
   }
@@ -3700,7 +3716,9 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       status: row.status,
       activeVersionId: row.activeVersionId,
       authorId: row.authorId,
+      visibility: row.visibility ?? void 0,
       metadata: parseJsonResilient(row.metadata),
+      favoriteCount: row.favoriteCount === null || row.favoriteCount === void 0 ? 0 : Number(row.favoriteCount),
       createdAt: row.createdAtZ || row.createdAt,
       updatedAt: row.updatedAtZ || row.updatedAt
     };
@@ -3732,17 +3750,20 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       const agentsTable = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
       const now = /* @__PURE__ */ new Date();
       const nowIso = now.toISOString();
+      const visibility = agent.visibility ?? (agent.authorId ? "private" : null);
       await this.#db.client.none(
         `INSERT INTO ${agentsTable} (
-          id, status, "authorId", metadata,
+          id, status, "authorId", visibility, metadata, "favoriteCount",
           "activeVersionId",
           "createdAt", "createdAtZ", "updatedAt", "updatedAtZ"
-        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
+        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)`,
         [
           agent.id,
           "draft",
           agent.authorId ?? null,
+          visibility,
           agent.metadata ? JSON.stringify(agent.metadata) : null,
+          0,
           null,
           // activeVersionId starts as null
           nowIso,
@@ -3751,7 +3772,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
           nowIso
         ]
       );
-      const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = agent;
+      const { id: _id, authorId: _authorId, visibility: _visibility, metadata: _metadata, ...snapshotConfig } = agent;
       const versionId = crypto.randomUUID();
       await this.createVersion({
         id: versionId,
@@ -3766,7 +3787,9 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
         status: "draft",
         activeVersionId: void 0,
         authorId: agent.authorId,
+        visibility: visibility ?? void 0,
         metadata: agent.metadata,
+        favoriteCount: 0,
         createdAt: now,
         updatedAt: now
       };
@@ -3805,7 +3828,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
           details: { agentId: id }
         });
       }
-      const { authorId, activeVersionId, metadata, status } = updates;
+      const { authorId, activeVersionId, metadata, status, visibility } = updates;
       const setClauses = [];
       const values = [];
       let paramIndex = 1;
@@ -3821,6 +3844,10 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
         setClauses.push(`status = $${paramIndex++}`);
         values.push(status);
       }
+      if (visibility !== void 0) {
+        setClauses.push(`visibility = $${paramIndex++}`);
+        values.push(visibility);
+      }
       if (metadata !== void 0) {
         setClauses.push(`metadata = $${paramIndex++}`);
         values.push(JSON.stringify(metadata));
@@ -3877,7 +3904,18 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
     }
   }
   async list(args) {
-    const { page = 0, perPage: perPageInput, orderBy, authorId, metadata, status } = args || {};
+    const {
+      page = 0,
+      perPage: perPageInput,
+      orderBy,
+      authorId,
+      metadata,
+      status,
+      visibility,
+      entityIds,
+      pinFavoritedFor,
+      favoritedOnly
+    } = args || {};
     const { field, direction } = this.parseOrderBy(orderBy);
     if (page < 0) {
       throw new MastraError(
@@ -3893,26 +3931,58 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
     const perPage = normalizePerPage(perPageInput, 100);
     const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     try {
+      if (entityIds && entityIds.length === 0) {
+        return {
+          agents: [],
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
       const tableName = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
+      const favoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
       const conditions = [];
       const queryParams = [];
       let paramIdx = 1;
+      const joinUserId = pinFavoritedFor;
+      const useJoin = Boolean(joinUserId);
+      let joinSqlIdx = null;
+      if (useJoin) {
+        joinSqlIdx = paramIdx++;
+      }
       if (status) {
-        conditions.push(`status = $${paramIdx++}`);
+        conditions.push(`a.status = $${paramIdx++}`);
         queryParams.push(status);
       }
       if (authorId !== void 0) {
-        conditions.push(`"authorId" = $${paramIdx++}`);
+        conditions.push(`a."authorId" = $${paramIdx++}`);
         queryParams.push(authorId);
       }
+      if (visibility !== void 0) {
+        conditions.push(`a.visibility = $${paramIdx++}`);
+        queryParams.push(visibility);
+      }
       if (metadata && Object.keys(metadata).length > 0) {
-        conditions.push(`metadata @> $${paramIdx++}::jsonb`);
+        conditions.push(`a.metadata @> $${paramIdx++}::jsonb`);
         queryParams.push(JSON.stringify(metadata));
       }
+      if (entityIds && entityIds.length > 0) {
+        const placeholders = entityIds.map(() => `$${paramIdx++}`).join(", ");
+        conditions.push(`a.id IN (${placeholders})`);
+        queryParams.push(...entityIds);
+      }
+      if (useJoin && favoritedOnly) {
+        conditions.push('s."userId" IS NOT NULL');
+      } else if (favoritedOnly) {
+        conditions.push("1=0");
+      }
+      const joinClause = useJoin && joinSqlIdx !== null ? `LEFT JOIN ${favoritesTable} s ON s."entityType" = 'agent' AND s."entityId" = a.id AND s."userId" = $${joinSqlIdx}` : "";
+      const joinParams = useJoin && joinUserId ? [joinUserId] : [];
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const countResult = await this.#db.client.one(
-        `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`,
-        queryParams
+        `SELECT COUNT(*) as count FROM ${tableName} a ${joinClause} ${whereClause}`,
+        [...joinParams, ...queryParams]
       );
       const total = parseInt(countResult.count, 10);
       if (total === 0) {
@@ -3924,10 +3994,19 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
           hasMore: false
         };
       }
+      const orderByParts = [];
+      if (useJoin) {
+        orderByParts.push(`(s."userId" IS NOT NULL) DESC`);
+      }
+      orderByParts.push(`a."${field}" ${direction}`);
+      orderByParts.push(`a."id" ASC`);
+      const orderByClause = `ORDER BY ${orderByParts.join(", ")}`;
       const limitValue = perPageInput === false ? total : perPage;
+      const limitIdx = paramIdx++;
+      const offsetIdx = paramIdx++;
       const dataResult = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${tableName} ${whereClause} ORDER BY "${field}" ${direction} LIMIT $${paramIdx++} OFFSET $${paramIdx++}`,
-        [...queryParams, limitValue, offset]
+        `SELECT a.* FROM ${tableName} a ${joinClause} ${whereClause} ${orderByClause} LIMIT $${limitIdx} OFFSET $${offsetIdx}`,
+        [...joinParams, ...queryParams, limitValue, offset]
       );
       const agents = (dataResult || []).flatMap((row) => {
         try {
@@ -3971,9 +4050,10 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
           "defaultOptions", workflows, agents, "integrationTools",
           "inputProcessors", "outputProcessors", memory, scorers,
           "mcpClients", "requestContextSchema", workspace, skills, "skillsFormat",
+          browser,
           "changedFields", "changeMessage",
           "createdAt", "createdAtZ"
-        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24, $25)`,
+        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26)`,
         [
           input.id,
           input.agentId,
@@ -3996,6 +4076,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
           input.workspace ? JSON.stringify(input.workspace) : null,
           input.skills ? JSON.stringify(input.skills) : null,
           input.skillsFormat ?? null,
+          input.browser ? JSON.stringify(input.browser) : null,
           input.changedFields ? JSON.stringify(input.changedFields) : null,
           input.changeMessage ?? null,
           nowIso,
@@ -4245,6 +4326,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       workspace: parseJsonResilient(row.workspace),
       skills: parseJsonResilient(row.skills),
       skillsFormat: row.skillsFormat,
+      browser: parseJsonResilient(row.browser),
       changedFields: parseJsonResilient(row.changedFields),
       changeMessage: row.changeMessage,
       createdAt: row.createdAtZ || row.createdAt
@@ -6396,6 +6478,250 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
     await this.#db.clearTable({ tableName: TABLE_EXPERIMENTS });
   }
 };
+var ENTITY_TABLE = {
+  agent: TABLE_AGENTS,
+  skill: TABLE_SKILLS
+};
+var FavoritesPG = class _FavoritesPG extends FavoritesStorage {
+  #db;
+  #schema;
+  static MANAGED_TABLES = [TABLE_FAVORITES];
+  constructor(config) {
+    super();
+    const { client, schemaName, skipDefaultIndexes } = resolvePgConfig(config);
+    this.#db = new PgDB({ client, schemaName, skipDefaultIndexes });
+    this.#schema = schemaName || "public";
+  }
+  static getExportDDL(schemaName) {
+    const statements = [];
+    for (const tableName of _FavoritesPG.MANAGED_TABLES) {
+      statements.push(
+        generateTableSQL({
+          tableName,
+          schema: TABLE_SCHEMAS[tableName],
+          schemaName,
+          compositePrimaryKey: ["userId", "entityType", "entityId"],
+          includeAllConstraints: true
+        })
+      );
+    }
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(schemaName) });
+    statements.push(
+      `CREATE INDEX IF NOT EXISTS idx_favorites_entity ON ${fullFavoritesTable} ("entityType", "entityId")`
+    );
+    return statements;
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_FAVORITES,
+      schema: TABLE_SCHEMAS[TABLE_FAVORITES],
+      compositePrimaryKey: ["userId", "entityType", "entityId"]
+    });
+    const fullTableName = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    await this.#db.client.none(
+      `CREATE INDEX IF NOT EXISTS idx_favorites_entity ON ${fullTableName} ("entityType", "entityId")`
+    );
+  }
+  async dangerouslyClearAll() {
+    const fullTableName = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    const fullAgentsTable = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
+    const fullSkillsTable = getTableName2({ indexName: TABLE_SKILLS, schemaName: getSchemaName2(this.#schema) });
+    await this.#db.client.tx(async (t) => {
+      await t.none(`DELETE FROM ${fullTableName}`);
+      await t.none(`UPDATE ${fullAgentsTable} SET "favoriteCount" = 0 WHERE "favoriteCount" > 0`);
+      await t.none(`UPDATE ${fullSkillsTable} SET "favoriteCount" = 0 WHERE "favoriteCount" > 0`);
+    });
+  }
+  async favorite(input) {
+    const { userId, entityType, entityId } = input;
+    const entityTable = ENTITY_TABLE[entityType];
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    const fullEntityTable = getTableName2({ indexName: entityTable, schemaName: getSchemaName2(this.#schema) });
+    try {
+      return await this.#db.client.tx(async (t) => {
+        const entityRow = await t.oneOrNone(`SELECT "favoriteCount" FROM ${fullEntityTable} WHERE id = $1`, [entityId]);
+        if (!entityRow) {
+          throw new MastraError({
+            id: createStorageErrorId("PG", "FAVORITE", "ENTITY_NOT_FOUND"),
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.USER,
+            text: `${entityType} ${entityId} not found`,
+            details: { entityType, entityId }
+          });
+        }
+        const inserted = await t.oneOrNone(
+          `INSERT INTO ${fullFavoritesTable} ("userId", "entityType", "entityId", "createdAt", "createdAtZ")
+           VALUES ($1, $2, $3, $4, $5)
+           ON CONFLICT ("userId", "entityType", "entityId") DO NOTHING
+           RETURNING "userId"`,
+          [userId, entityType, entityId, (/* @__PURE__ */ new Date()).toISOString(), (/* @__PURE__ */ new Date()).toISOString()]
+        );
+        if (inserted) {
+          await t.none(
+            `UPDATE ${fullEntityTable} SET "favoriteCount" = COALESCE("favoriteCount", 0) + 1 WHERE id = $1`,
+            [entityId]
+          );
+        }
+        const after = await t.one(`SELECT "favoriteCount" FROM ${fullEntityTable} WHERE id = $1`, [entityId]);
+        const favoriteCount = Number(after.favoriteCount ?? 0);
+        return { favorited: true, favoriteCount };
+      });
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "FAVORITE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { entityType, entityId }
+        },
+        error
+      );
+    }
+  }
+  async unfavorite(input) {
+    const { userId, entityType, entityId } = input;
+    const entityTable = ENTITY_TABLE[entityType];
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    const fullEntityTable = getTableName2({ indexName: entityTable, schemaName: getSchemaName2(this.#schema) });
+    try {
+      return await this.#db.client.tx(async (t) => {
+        const entityRow = await t.oneOrNone(`SELECT "favoriteCount" FROM ${fullEntityTable} WHERE id = $1`, [entityId]);
+        if (!entityRow) {
+          throw new MastraError({
+            id: createStorageErrorId("PG", "UNFAVORITE", "ENTITY_NOT_FOUND"),
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.USER,
+            text: `${entityType} ${entityId} not found`,
+            details: { entityType, entityId }
+          });
+        }
+        const deleted = await t.oneOrNone(
+          `DELETE FROM ${fullFavoritesTable} WHERE "userId" = $1 AND "entityType" = $2 AND "entityId" = $3 RETURNING "userId"`,
+          [userId, entityType, entityId]
+        );
+        if (deleted) {
+          await t.none(
+            `UPDATE ${fullEntityTable} SET "favoriteCount" = GREATEST(COALESCE("favoriteCount", 0) - 1, 0) WHERE id = $1`,
+            [entityId]
+          );
+        }
+        const after = await t.one(`SELECT "favoriteCount" FROM ${fullEntityTable} WHERE id = $1`, [entityId]);
+        const favoriteCount = Number(after.favoriteCount ?? 0);
+        return { favorited: false, favoriteCount };
+      });
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "UNFAVORITE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { entityType, entityId }
+        },
+        error
+      );
+    }
+  }
+  async isFavorited(input) {
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    try {
+      const result = await this.#db.client.oneOrNone(
+        `SELECT 1 FROM ${fullFavoritesTable} WHERE "userId" = $1 AND "entityType" = $2 AND "entityId" = $3 LIMIT 1`,
+        [input.userId, input.entityType, input.entityId]
+      );
+      return result !== null;
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "IS_FAVORITED", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async isFavoritedBatch(input) {
+    const { userId, entityType, entityIds } = input;
+    if (entityIds.length === 0) {
+      return /* @__PURE__ */ new Set();
+    }
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    try {
+      const placeholders = entityIds.map((_, i) => `$${i + 3}`).join(", ");
+      const rows = await this.#db.client.manyOrNone(
+        `SELECT "entityId" FROM ${fullFavoritesTable} WHERE "userId" = $1 AND "entityType" = $2 AND "entityId" IN (${placeholders})`,
+        [userId, entityType, ...entityIds]
+      );
+      const set = /* @__PURE__ */ new Set();
+      for (const row of rows ?? []) {
+        set.add(row.entityId);
+      }
+      return set;
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "IS_FAVORITED_BATCH", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async listFavoritedIds(input) {
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    try {
+      const rows = await this.#db.client.manyOrNone(
+        `SELECT "entityId" FROM ${fullFavoritesTable} WHERE "userId" = $1 AND "entityType" = $2 ORDER BY "createdAt" DESC, "entityId" ASC`,
+        [input.userId, input.entityType]
+      );
+      return (rows ?? []).map((row) => row.entityId);
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "LIST_FAVORITED_IDS", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async deleteFavoritesForEntity(input) {
+    const fullFavoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
+    const entityTable = ENTITY_TABLE[input.entityType];
+    const fullEntityTable = getTableName2({ indexName: entityTable, schemaName: getSchemaName2(this.#schema) });
+    try {
+      return await this.#db.client.tx(async (t) => {
+        const result = await t.one(
+          `WITH deleted AS (
+             DELETE FROM ${fullFavoritesTable} WHERE "entityType" = $1 AND "entityId" = $2 RETURNING 1
+           )
+           SELECT COUNT(*)::text AS count FROM deleted`,
+          [input.entityType, input.entityId]
+        );
+        await t.none(`UPDATE ${fullEntityTable} SET "favoriteCount" = 0 WHERE id = $1`, [input.entityId]);
+        return Number(result.count);
+      });
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "DELETE_FAVORITES_FOR_ENTITY", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { entityType: input.entityType, entityId: input.entityId }
+        },
+        error
+      );
+    }
+  }
+};
 var SNAPSHOT_FIELDS = ["name", "description", "servers"];
 var MCPClientsPG = class _MCPClientsPG extends MCPClientsStorage {
   #db;
@@ -12702,6 +13028,7 @@ var SNAPSHOT_FIELDS5 = [
   "references",
   "scripts",
   "assets",
+  "files",
   "metadata",
   "tree"
 ];
@@ -12749,6 +13076,16 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
       tableName: TABLE_SKILL_VERSIONS,
       schema: TABLE_SCHEMAS[TABLE_SKILL_VERSIONS]
     });
+    await this.#db.alterTable({
+      tableName: TABLE_SKILLS,
+      schema: TABLE_SCHEMAS[TABLE_SKILLS],
+      ifNotExists: ["visibility", "favoriteCount"]
+    });
+    await this.#db.alterTable({
+      tableName: TABLE_SKILL_VERSIONS,
+      schema: TABLE_SCHEMAS[TABLE_SKILL_VERSIONS],
+      ifNotExists: ["files"]
+    });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -12798,14 +13135,15 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
       const tableName = getTableName2({ indexName: TABLE_SKILLS, schemaName: getSchemaName2(this.#schema) });
       const now = /* @__PURE__ */ new Date();
       const nowIso = now.toISOString();
+      const visibility = skill.visibility ?? (skill.authorId ? "private" : void 0);
       await this.#db.client.none(
         `INSERT INTO ${tableName} (
-          id, status, "activeVersionId", "authorId",
+          id, status, "activeVersionId", "authorId", visibility, "favoriteCount",
           "createdAt", "createdAtZ", "updatedAt", "updatedAtZ"
-        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8)`,
-        [skill.id, "draft", null, skill.authorId ?? null, nowIso, nowIso, nowIso, nowIso]
+        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)`,
+        [skill.id, "draft", null, skill.authorId ?? null, visibility ?? null, 0, nowIso, nowIso, nowIso, nowIso]
       );
-      const { id: _id, authorId: _authorId, ...snapshotConfig } = skill;
+      const { id: _id, authorId: _authorId, visibility: _visibility, ...snapshotConfig } = skill;
       const versionId = crypto.randomUUID();
       await this.createVersion({
         id: versionId,
@@ -12820,6 +13158,8 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
         status: "draft",
         activeVersionId: void 0,
         authorId: skill.authorId,
+        visibility,
+        favoriteCount: 0,
         createdAt: now,
         updatedAt: now
       };
@@ -12861,8 +13201,12 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
           details: { skillId: id }
         });
       }
-      const { authorId, activeVersionId, status, ...configFields } = updates;
+      const { authorId, visibility, activeVersionId, status, ...rawConfigFields } = updates;
       let versionCreated = false;
+      const configFields = {};
+      for (const [key, value] of Object.entries(rawConfigFields)) {
+        if (value !== void 0) configFields[key] = value;
+      }
       const hasConfigUpdate = SNAPSHOT_FIELDS5.some((field) => field in configFields);
       if (hasConfigUpdate) {
         const latestVersion = await this.getLatestVersion(id);
@@ -12908,6 +13252,10 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
         setClauses.push(`"authorId" = $${paramIndex++}`);
         values.push(authorId);
       }
+      if (visibility !== void 0) {
+        setClauses.push(`visibility = $${paramIndex++}`);
+        values.push(visibility);
+      }
       if (activeVersionId !== void 0) {
         setClauses.push(`"activeVersionId" = $${paramIndex++}`);
         values.push(activeVersionId);
@@ -12975,7 +13323,17 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
     }
   }
   async list(args) {
-    const { page = 0, perPage: perPageInput, orderBy, authorId } = args || {};
+    const {
+      page = 0,
+      perPage: perPageInput,
+      orderBy,
+      authorId,
+      visibility,
+      status,
+      entityIds,
+      pinFavoritedFor,
+      favoritedOnly
+    } = args || {};
     const { field, direction } = this.parseOrderBy(orderBy);
     if (page < 0) {
       throw new MastraError(
@@ -12991,18 +13349,54 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
     const perPage = normalizePerPage(perPageInput, 100);
     const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
     try {
+      if (entityIds && entityIds.length === 0) {
+        return {
+          skills: [],
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
       const tableName = getTableName2({ indexName: TABLE_SKILLS, schemaName: getSchemaName2(this.#schema) });
+      const favoritesTable = getTableName2({ indexName: TABLE_FAVORITES, schemaName: getSchemaName2(this.#schema) });
       const conditions = [];
       const queryParams = [];
       let paramIdx = 1;
+      const joinUserId = pinFavoritedFor;
+      const useJoin = Boolean(joinUserId);
+      let joinSqlIdx = null;
+      if (useJoin) {
+        joinSqlIdx = paramIdx++;
+      }
+      if (status) {
+        conditions.push(`s.status = $${paramIdx++}`);
+        queryParams.push(status);
+      }
       if (authorId !== void 0) {
-        conditions.push(`"authorId" = $${paramIdx++}`);
+        conditions.push(`s."authorId" = $${paramIdx++}`);
         queryParams.push(authorId);
       }
+      if (visibility !== void 0) {
+        conditions.push(`s.visibility = $${paramIdx++}`);
+        queryParams.push(visibility);
+      }
+      if (entityIds && entityIds.length > 0) {
+        const placeholders = entityIds.map(() => `$${paramIdx++}`).join(", ");
+        conditions.push(`s.id IN (${placeholders})`);
+        queryParams.push(...entityIds);
+      }
+      if (useJoin && favoritedOnly) {
+        conditions.push('sr."userId" IS NOT NULL');
+      } else if (favoritedOnly) {
+        conditions.push("1=0");
+      }
+      const joinClause = useJoin && joinSqlIdx !== null ? `LEFT JOIN ${favoritesTable} sr ON sr."entityType" = 'skill' AND sr."entityId" = s.id AND sr."userId" = $${joinSqlIdx}` : "";
+      const joinParams = useJoin && joinUserId ? [joinUserId] : [];
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const countResult = await this.#db.client.one(
-        `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`,
-        queryParams
+        `SELECT COUNT(*) as count FROM ${tableName} s ${joinClause} ${whereClause}`,
+        [...joinParams, ...queryParams]
       );
       const total = parseInt(countResult.count, 10);
       if (total === 0) {
@@ -13014,10 +13408,19 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
           hasMore: false
         };
       }
+      const orderByParts = [];
+      if (useJoin) {
+        orderByParts.push(`(sr."userId" IS NOT NULL) DESC`);
+      }
+      orderByParts.push(`s."${field}" ${direction}`);
+      orderByParts.push(`s."id" ASC`);
+      const orderByClause = `ORDER BY ${orderByParts.join(", ")}`;
       const limitValue = perPageInput === false ? total : perPage;
+      const limitIdx = paramIdx++;
+      const offsetIdx = paramIdx++;
       const dataResult = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${tableName} ${whereClause} ORDER BY "${field}" ${direction} LIMIT $${paramIdx++} OFFSET $${paramIdx++}`,
-        [...queryParams, limitValue, offset]
+        `SELECT s.* FROM ${tableName} s ${joinClause} ${whereClause} ${orderByClause} LIMIT $${limitIdx} OFFSET $${offsetIdx}`,
+        [...joinParams, ...queryParams, limitValue, offset]
       );
       const skills = (dataResult || []).flatMap((row) => {
         try {
@@ -13061,10 +13464,10 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
         `INSERT INTO ${tableName} (
           id, "skillId", "versionNumber",
           name, description, instructions, license, compatibility,
-          source, "references", scripts, assets, metadata, tree,
+          source, "references", scripts, assets, files, metadata, tree,
           "changedFields", "changeMessage",
           "createdAt", "createdAtZ"
-        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18)`,
+        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19)`,
         [
           input.id,
           input.skillId,
@@ -13078,6 +13481,7 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
           input.references ? JSON.stringify(input.references) : null,
           input.scripts ? JSON.stringify(input.scripts) : null,
           input.assets ? JSON.stringify(input.assets) : null,
+          input.files ? JSON.stringify(input.files) : null,
           input.metadata ? JSON.stringify(input.metadata) : null,
           input.tree ? JSON.stringify(input.tree) : null,
           input.changedFields ? JSON.stringify(input.changedFields) : null,
@@ -13320,6 +13724,8 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
       status: row.status,
       activeVersionId: row.activeVersionId,
       authorId: row.authorId,
+      visibility: row.visibility,
+      favoriteCount: row.favoriteCount === null || row.favoriteCount === void 0 ? 0 : Number(row.favoriteCount),
       createdAt: new Date(row.createdAtZ || row.createdAt),
       updatedAt: new Date(row.updatedAtZ || row.updatedAt)
     };
@@ -13338,6 +13744,7 @@ var SkillsPG = class _SkillsPG extends SkillsStorage {
       references: parseJsonResilient(row.references),
       scripts: parseJsonResilient(row.scripts),
       assets: parseJsonResilient(row.assets),
+      files: parseJsonResilient(row.files),
       metadata: parseJsonResilient(row.metadata),
       tree: parseJsonResilient(row.tree),
       changedFields: parseJsonResilient(row.changedFields),
@@ -13939,8 +14346,12 @@ var WorkspacesPG = class _WorkspacesPG extends WorkspacesStorage {
           details: { workspaceId: id }
         });
       }
-      const { authorId, activeVersionId, metadata, status, ...configFields } = updates;
+      const { authorId, activeVersionId, metadata, status, ...rawConfigFields } = updates;
       let versionCreated = false;
+      const configFields = {};
+      for (const [key, value] of Object.entries(rawConfigFields)) {
+        if (value !== void 0) configFields[key] = value;
+      }
       const hasConfigUpdate = SNAPSHOT_FIELDS6.some((field) => field in configFields);
       if (hasConfigUpdate) {
         const latestVersion = await this.getLatestVersion(id);
@@ -14448,6 +14859,7 @@ var ALL_DOMAINS = [
   DatasetsPG,
   ExperimentsPG,
   BackgroundTasksPG,
+  FavoritesPG,
   ChannelsPG,
   SchedulesPG
 ];
@@ -14501,6 +14913,7 @@ var PostgresStore = class extends MastraCompositeStore {
         mcpServers: new MCPServersPG(domainConfig),
         workspaces: new WorkspacesPG(domainConfig),
         skills: new SkillsPG(domainConfig),
+        favorites: new FavoritesPG(domainConfig),
         blobs: new BlobsPG(domainConfig),
         datasets: new DatasetsPG(domainConfig),
         experiments: new ExperimentsPG(domainConfig),
@@ -14695,6 +15108,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, 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, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, SchedulesPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map