@mastra/pg 1.0.0 → 1.1.0-alpha.1

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, TABLE_SCHEMAS, AgentsStorage, TABLE_AGENTS, createStorageErrorId, normalizePerPage, calculatePagination, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, TraceStatus, getDefaultValue, transformScoreRow as transformScoreRow$1, getSqlType } from '@mastra/core/storage';
+import { createVectorErrorId, TABLE_SCHEMAS, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, 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';
@@ -3093,7 +3093,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
   #skipDefaultIndexes;
   #indexes;
   /** Tables managed by this domain */
-  static MANAGED_TABLES = [TABLE_AGENTS];
+  static MANAGED_TABLES = [TABLE_AGENTS, TABLE_AGENT_VERSIONS];
   constructor(config) {
     super();
     const { client, schemaName, skipDefaultIndexes, indexes } = resolvePgConfig(config);
@@ -3120,6 +3120,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
   }
   async init() {
     await this.#db.createTable({ tableName: TABLE_AGENTS, schema: TABLE_SCHEMAS[TABLE_AGENTS] });
+    await this.#db.createTable({ tableName: TABLE_AGENT_VERSIONS, schema: TABLE_SCHEMAS[TABLE_AGENT_VERSIONS] });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -3139,6 +3140,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
     }
   }
   async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_AGENT_VERSIONS });
     await this.#db.clearTable({ tableName: TABLE_AGENTS });
   }
   parseJson(value, fieldName) {
@@ -3168,18 +3170,9 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
   parseRow(row) {
     return {
       id: row.id,
-      name: row.name,
-      description: row.description,
-      instructions: row.instructions,
-      model: this.parseJson(row.model, "model"),
-      tools: this.parseJson(row.tools, "tools"),
-      defaultOptions: this.parseJson(row.defaultOptions, "defaultOptions"),
-      workflows: this.parseJson(row.workflows, "workflows"),
-      agents: this.parseJson(row.agents, "agents"),
-      inputProcessors: this.parseJson(row.inputProcessors, "inputProcessors"),
-      outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
-      memory: this.parseJson(row.memory, "memory"),
-      scorers: this.parseJson(row.scorers, "scorers"),
+      status: row.status,
+      activeVersionId: row.activeVersionId,
+      authorId: row.authorId,
       metadata: this.parseJson(row.metadata, "metadata"),
       createdAt: row.createdAtZ || row.createdAt,
       updatedAt: row.updatedAtZ || row.updatedAt
@@ -3207,38 +3200,48 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
   }
   async createAgent({ agent }) {
     try {
-      const tableName = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
+      const agentsTable = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
       const now = /* @__PURE__ */ new Date();
       const nowIso = now.toISOString();
       await this.#db.client.none(
-        `INSERT INTO ${tableName} (
-          id, name, description, instructions, model, tools, 
-          "defaultOptions", workflows, agents, "inputProcessors", "outputProcessors", memory, scorers, metadata, 
+        `INSERT INTO ${agentsTable} (
+          id, status, "authorId", metadata,
+          "activeVersionId",
           "createdAt", "createdAtZ", "updatedAt", "updatedAtZ"
-        ) 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)`,
         [
           agent.id,
-          agent.name,
-          agent.description ?? null,
-          agent.instructions,
-          JSON.stringify(agent.model),
-          agent.tools ? JSON.stringify(agent.tools) : null,
-          agent.defaultOptions ? JSON.stringify(agent.defaultOptions) : null,
-          agent.workflows ? JSON.stringify(agent.workflows) : null,
-          agent.agents ? JSON.stringify(agent.agents) : null,
-          agent.inputProcessors ? JSON.stringify(agent.inputProcessors) : null,
-          agent.outputProcessors ? JSON.stringify(agent.outputProcessors) : null,
-          agent.memory ? JSON.stringify(agent.memory) : null,
-          agent.scorers ? JSON.stringify(agent.scorers) : null,
+          "draft",
+          agent.authorId ?? null,
           agent.metadata ? JSON.stringify(agent.metadata) : null,
+          null,
+          // activeVersionId starts as null
           nowIso,
           nowIso,
           nowIso,
           nowIso
         ]
       );
+      const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = agent;
+      const versionId = crypto.randomUUID();
+      await this.createVersion({
+        id: versionId,
+        agentId: agent.id,
+        versionNumber: 1,
+        ...snapshotConfig,
+        changedFields: Object.keys(snapshotConfig),
+        changeMessage: "Initial version"
+      });
+      await this.#db.client.none(
+        `UPDATE ${agentsTable} SET "activeVersionId" = $1, status = $2, "updatedAt" = $3, "updatedAtZ" = $4 WHERE id = $5`,
+        [versionId, "published", nowIso, nowIso, agent.id]
+      );
       return {
-        ...agent,
+        id: agent.id,
+        status: "published",
+        activeVersionId: versionId,
+        authorId: agent.authorId,
+        metadata: agent.metadata,
         createdAt: now,
         updatedAt: now
       };
@@ -3270,53 +3273,15 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       const setClauses = [];
       const values = [];
       let paramIndex = 1;
-      if (updates.name !== void 0) {
-        setClauses.push(`name = $${paramIndex++}`);
-        values.push(updates.name);
-      }
-      if (updates.description !== void 0) {
-        setClauses.push(`description = $${paramIndex++}`);
-        values.push(updates.description);
-      }
-      if (updates.instructions !== void 0) {
-        setClauses.push(`instructions = $${paramIndex++}`);
-        values.push(updates.instructions);
-      }
-      if (updates.model !== void 0) {
-        setClauses.push(`model = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.model));
-      }
-      if (updates.tools !== void 0) {
-        setClauses.push(`tools = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.tools));
-      }
-      if (updates.defaultOptions !== void 0) {
-        setClauses.push(`"defaultOptions" = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.defaultOptions));
-      }
-      if (updates.workflows !== void 0) {
-        setClauses.push(`workflows = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.workflows));
-      }
-      if (updates.agents !== void 0) {
-        setClauses.push(`agents = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.agents));
-      }
-      if (updates.inputProcessors !== void 0) {
-        setClauses.push(`"inputProcessors" = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.inputProcessors));
+      if (updates.authorId !== void 0) {
+        setClauses.push(`"authorId" = $${paramIndex++}`);
+        values.push(updates.authorId);
       }
-      if (updates.outputProcessors !== void 0) {
-        setClauses.push(`"outputProcessors" = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.outputProcessors));
-      }
-      if (updates.memory !== void 0) {
-        setClauses.push(`memory = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.memory));
-      }
-      if (updates.scorers !== void 0) {
-        setClauses.push(`scorers = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.scorers));
+      if (updates.activeVersionId !== void 0) {
+        setClauses.push(`"activeVersionId" = $${paramIndex++}`);
+        values.push(updates.activeVersionId);
+        setClauses.push(`status = $${paramIndex++}`);
+        values.push("published");
       }
       if (updates.metadata !== void 0) {
         const mergedMetadata = { ...existingAgent.metadata, ...updates.metadata };
@@ -3364,6 +3329,7 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
   async deleteAgent({ id }) {
     try {
       const tableName = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
+      await this.deleteVersionsByAgentId(id);
       await this.#db.client.none(`DELETE FROM ${tableName} WHERE id = $1`, [id]);
     } catch (error) {
       throw new MastraError(
@@ -3430,6 +3396,261 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       );
     }
   }
+  // ==========================================================================
+  // Agent Version Methods
+  // ==========================================================================
+  async createVersion(input) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      const now = /* @__PURE__ */ new Date();
+      const nowIso = now.toISOString();
+      await this.#db.client.none(
+        `INSERT INTO ${tableName} (
+          id, "agentId", "versionNumber",
+          name, description, instructions, model, tools,
+          "defaultOptions", workflows, agents, "integrationTools",
+          "inputProcessors", "outputProcessors", memory, scorers,
+          "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)`,
+        [
+          input.id,
+          input.agentId,
+          input.versionNumber,
+          input.name,
+          input.description ?? null,
+          input.instructions,
+          JSON.stringify(input.model),
+          input.tools ? JSON.stringify(input.tools) : null,
+          input.defaultOptions ? JSON.stringify(input.defaultOptions) : null,
+          input.workflows ? JSON.stringify(input.workflows) : null,
+          input.agents ? JSON.stringify(input.agents) : null,
+          input.integrationTools ? JSON.stringify(input.integrationTools) : null,
+          input.inputProcessors ? JSON.stringify(input.inputProcessors) : null,
+          input.outputProcessors ? JSON.stringify(input.outputProcessors) : null,
+          input.memory ? JSON.stringify(input.memory) : null,
+          input.scorers ? JSON.stringify(input.scorers) : null,
+          input.changedFields ? JSON.stringify(input.changedFields) : null,
+          input.changeMessage ?? null,
+          nowIso,
+          nowIso
+        ]
+      );
+      return {
+        ...input,
+        createdAt: now
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "CREATE_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { versionId: input.id, agentId: input.agentId }
+        },
+        error
+      );
+    }
+  }
+  async getVersion(id) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      const result = await this.#db.client.oneOrNone(`SELECT * FROM ${tableName} WHERE id = $1`, [id]);
+      if (!result) {
+        return null;
+      }
+      return this.parseVersionRow(result);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "GET_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { versionId: id }
+        },
+        error
+      );
+    }
+  }
+  async getVersionByNumber(agentId, versionNumber) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      const result = await this.#db.client.oneOrNone(
+        `SELECT * FROM ${tableName} WHERE "agentId" = $1 AND "versionNumber" = $2`,
+        [agentId, versionNumber]
+      );
+      if (!result) {
+        return null;
+      }
+      return this.parseVersionRow(result);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "GET_VERSION_BY_NUMBER", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId, versionNumber }
+        },
+        error
+      );
+    }
+  }
+  async getLatestVersion(agentId) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      const result = await this.#db.client.oneOrNone(
+        `SELECT * FROM ${tableName} WHERE "agentId" = $1 ORDER BY "versionNumber" DESC LIMIT 1`,
+        [agentId]
+      );
+      if (!result) {
+        return null;
+      }
+      return this.parseVersionRow(result);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "GET_LATEST_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  async listVersions(input) {
+    const { agentId, page = 0, perPage: perPageInput, orderBy } = input;
+    if (page < 0) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "LIST_VERSIONS", "INVALID_PAGE"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { page }
+        },
+        new Error("page must be >= 0")
+      );
+    }
+    const perPage = normalizePerPage(perPageInput, 20);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+    try {
+      const { field, direction } = this.parseVersionOrderBy(orderBy);
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      const countResult = await this.#db.client.one(`SELECT COUNT(*) as count FROM ${tableName} WHERE "agentId" = $1`, [
+        agentId
+      ]);
+      const total = parseInt(countResult.count, 10);
+      if (total === 0) {
+        return {
+          versions: [],
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
+      const limitValue = perPageInput === false ? total : perPage;
+      const dataResult = await this.#db.client.manyOrNone(
+        `SELECT * FROM ${tableName} WHERE "agentId" = $1 ORDER BY "${field}" ${direction} LIMIT $2 OFFSET $3`,
+        [agentId, limitValue, offset]
+      );
+      const versions = (dataResult || []).map((row) => this.parseVersionRow(row));
+      return {
+        versions,
+        total,
+        page,
+        perPage: perPageForResponse,
+        hasMore: perPageInput === false ? false : offset + perPage < total
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "LIST_VERSIONS", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  async deleteVersion(id) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      await this.#db.client.none(`DELETE FROM ${tableName} WHERE id = $1`, [id]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "DELETE_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { versionId: id }
+        },
+        error
+      );
+    }
+  }
+  async deleteVersionsByAgentId(agentId) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      await this.#db.client.none(`DELETE FROM ${tableName} WHERE "agentId" = $1`, [agentId]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "DELETE_VERSIONS_BY_AGENT_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  async countVersions(agentId) {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_AGENT_VERSIONS, schemaName: getSchemaName2(this.#schema) });
+      const result = await this.#db.client.one(`SELECT COUNT(*) as count FROM ${tableName} WHERE "agentId" = $1`, [
+        agentId
+      ]);
+      return parseInt(result.count, 10);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "COUNT_VERSIONS", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  // ==========================================================================
+  // Private Helper Methods
+  // ==========================================================================
+  parseVersionRow(row) {
+    return {
+      id: row.id,
+      agentId: row.agentId,
+      versionNumber: row.versionNumber,
+      name: row.name,
+      description: row.description,
+      instructions: row.instructions,
+      model: this.parseJson(row.model, "model"),
+      tools: this.parseJson(row.tools, "tools"),
+      defaultOptions: this.parseJson(row.defaultOptions, "defaultOptions"),
+      workflows: this.parseJson(row.workflows, "workflows"),
+      agents: this.parseJson(row.agents, "agents"),
+      integrationTools: this.parseJson(row.integrationTools, "integrationTools"),
+      inputProcessors: this.parseJson(row.inputProcessors, "inputProcessors"),
+      outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
+      memory: this.parseJson(row.memory, "memory"),
+      scorers: this.parseJson(row.scorers, "scorers"),
+      changedFields: this.parseJson(row.changedFields, "changedFields"),
+      changeMessage: row.changeMessage,
+      createdAt: row.createdAtZ || row.createdAt
+    };
+  }
 };
 function getSchemaName3(schema) {
   return schema ? `"${schema}"` : '"public"';
@@ -5016,11 +5237,13 @@ var ObservabilityPG = class _ObservabilityPG extends ObservabilityStorage {
           perPage,
           hasMore: (page + 1) * perPage < count
         },
-        spans: spans.map(
-          (span) => transformFromSqlRow({
-            tableName: TABLE_SPANS,
-            sqlRow: span
-          })
+        spans: toTraceSpans(
+          spans.map(
+            (span) => transformFromSqlRow({
+              tableName: TABLE_SPANS,
+              sqlRow: span
+            })
+          )
         )
       };
     } catch (error) {