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

package/dist/index.js

@@ -3170,22 +3170,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 +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, "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 +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.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 +3406,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 +3634,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