@mastra/pg 1.1.0-alpha.0 → 1.1.0-alpha.2

package/dist/index.js

@@ -3119,10 +3119,124 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
     }
   }
   async init() {
+    await this.#migrateFromLegacySchema();
+    await this.#migrateVersionsSchema();
     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.#db.alterTable({
+      tableName: TABLE_AGENTS,
+      schema: TABLE_SCHEMAS[TABLE_AGENTS],
+      ifNotExists: ["status", "authorId"]
+    });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
+    await this.#cleanupStaleDrafts();
+  }
+  /**
+   * Migrates from the legacy flat agent schema (where config fields like name, instructions, model
+   * were stored directly on mastra_agents) to the new versioned schema (thin agent record + versions table).
+   */
+  async #migrateFromLegacySchema() {
+    const fullTableName = getTableName2({ indexName: TABLE_AGENTS, schemaName: getSchemaName2(this.#schema) });
+    const fullVersionsTableName = getTableName2({
+      indexName: TABLE_AGENT_VERSIONS,
+      schemaName: getSchemaName2(this.#schema)
+    });
+    const legacyTableName = getTableName2({
+      indexName: `${TABLE_AGENTS}_legacy`,
+      schemaName: getSchemaName2(this.#schema)
+    });
+    const hasLegacyColumns = await this.#db.hasColumn(TABLE_AGENTS, "name");
+    if (hasLegacyColumns) {
+      await this.#db.client.none(`ALTER TABLE ${fullTableName} RENAME TO "${TABLE_AGENTS}_legacy"`);
+      await this.#db.client.none(`DROP TABLE IF EXISTS ${fullVersionsTableName}`);
+    }
+    const legacyExists = await this.#db.hasColumn(`${TABLE_AGENTS}_legacy`, "name");
+    if (!legacyExists) return;
+    const oldAgents = await this.#db.client.manyOrNone(`SELECT * FROM ${legacyTableName}`);
+    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] });
+    for (const row of oldAgents) {
+      const agentId = row.id;
+      if (!agentId) continue;
+      const versionId = crypto.randomUUID();
+      const now = /* @__PURE__ */ new Date();
+      await this.#db.client.none(
+        `INSERT INTO ${fullTableName} (id, status, "activeVersionId", "authorId", metadata, "createdAt", "updatedAt")
+         VALUES ($1, $2, $3, $4, $5, $6, $7)
+         ON CONFLICT (id) DO NOTHING`,
+        [
+          agentId,
+          "published",
+          versionId,
+          row.ownerId ?? row.authorId ?? null,
+          row.metadata ? JSON.stringify(row.metadata) : null,
+          row.createdAt ?? now,
+          row.updatedAt ?? now
+        ]
+      );
+      await this.#db.client.none(
+        `INSERT INTO ${fullVersionsTableName}
+         (id, "agentId", "versionNumber", name, description, instructions, model, tools,
+          "defaultOptions", workflows, agents, "integrationTools", "inputProcessors",
+          "outputProcessors", memory, scorers, "changedFields", "changeMessage", "createdAt")
+         VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19)
+         ON CONFLICT (id) DO NOTHING`,
+        [
+          versionId,
+          agentId,
+          1,
+          row.name ?? agentId,
+          row.description ?? null,
+          row.instructions ?? "",
+          row.model ? JSON.stringify(row.model) : "{}",
+          row.tools ? JSON.stringify(row.tools) : null,
+          row.defaultOptions ? JSON.stringify(row.defaultOptions) : null,
+          row.workflows ? JSON.stringify(row.workflows) : null,
+          row.agents ? JSON.stringify(row.agents) : null,
+          row.integrationTools ? JSON.stringify(row.integrationTools) : null,
+          row.inputProcessors ? JSON.stringify(row.inputProcessors) : null,
+          row.outputProcessors ? JSON.stringify(row.outputProcessors) : null,
+          row.memory ? JSON.stringify(row.memory) : null,
+          row.scorers ? JSON.stringify(row.scorers) : null,
+          null,
+          "Migrated from legacy schema",
+          row.createdAt ?? now
+        ]
+      );
+    }
+    await this.#db.client.none(`DROP TABLE IF EXISTS ${legacyTableName}`);
+  }
+  /**
+   * Migrates the agent_versions table from the old snapshot-based schema (single `snapshot` JSON column)
+   * to the new flat schema (individual config columns). This handles the case where the agents table
+   * was already migrated but the versions table still has the old schema.
+   */
+  async #migrateVersionsSchema() {
+    const hasSnapshotColumn = await this.#db.hasColumn(TABLE_AGENT_VERSIONS, "snapshot");
+    if (!hasSnapshotColumn) return;
+    const fullVersionsTableName = getTableName2({
+      indexName: TABLE_AGENT_VERSIONS,
+      schemaName: getSchemaName2(this.#schema)
+    });
+    const legacyTableName = getTableName2({
+      indexName: `${TABLE_AGENTS}_legacy`,
+      schemaName: getSchemaName2(this.#schema)
+    });
+    await this.#db.client.none(`DROP TABLE IF EXISTS ${fullVersionsTableName}`);
+    await this.#db.client.none(`DROP TABLE IF EXISTS ${legacyTableName}`);
+  }
+  /**
+   * Removes stale draft agent records that have no activeVersionId.
+   * These are left behind when createAgent partially fails (inserts thin record
+   * but fails to create the version due to schema mismatch).
+   */
+  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`);
+    } catch {
+    }
   }
   /**
    * Creates custom user-defined indexes for this domain's tables.
@@ -3170,22 +3284,10 @@ 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"),
-      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"),
-      metadata: this.parseJson(row.metadata, "metadata"),
-      ownerId: row.ownerId,
+      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
     };
@@ -3212,43 +3314,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, "integrationTools",
-          "inputProcessors", "outputProcessors", memory, scorers, metadata,
-          "ownerId", "activeVersionId",
+        `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, $19, $20, $21)`,
+        ) 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.integrationTools ? JSON.stringify(agent.integrationTools) : 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,
-          agent.ownerId ?? null,
-          agent.activeVersionId ?? 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
       };
@@ -3280,65 +3387,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.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.integrationTools !== void 0) {
-        setClauses.push(`"integrationTools" = $${paramIndex++}`);
-        values.push(JSON.stringify(updates.integrationTools));
-      }
-      if (updates.ownerId !== void 0) {
-        setClauses.push(`"ownerId" = $${paramIndex++}`);
-        values.push(updates.ownerId);
+      if (updates.authorId !== void 0) {
+        setClauses.push(`"authorId" = $${paramIndex++}`);
+        values.push(updates.authorId);
       }
       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 };
@@ -3463,14 +3520,30 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       const nowIso = now.toISOString();
       await this.#db.client.none(
         `INSERT INTO ${tableName} (
-          id, "agentId", "versionNumber", name, snapshot, "changedFields", "changeMessage", "createdAt", "createdAtZ"
-        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
+          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 ?? null,
-          JSON.stringify(input.snapshot),
+          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,
@@ -3675,7 +3748,18 @@ var AgentsPG = class _AgentsPG extends AgentsStorage {
       agentId: row.agentId,
       versionNumber: row.versionNumber,
       name: row.name,
-      snapshot: this.parseJson(row.snapshot, "snapshot"),
+      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