@mastra/mongodb 1.1.0-alpha.0 → 1.1.0

package/dist/index.js

@@ -11,7 +11,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.1.0-alpha.0"};
+  version: "1.1.0"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -892,6 +892,21 @@ function resolveMongoDBConfig(config) {
     );
   }
 }
+var SNAPSHOT_FIELDS = [
+  "name",
+  "description",
+  "instructions",
+  "model",
+  "tools",
+  "defaultOptions",
+  "workflows",
+  "agents",
+  "integrationTools",
+  "inputProcessors",
+  "outputProcessors",
+  "memory",
+  "scorers"
+];
 var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
   #connector;
   #skipDefaultIndexes;
@@ -997,11 +1012,28 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
       }
       const now = /* @__PURE__ */ new Date();
       const newAgent = {
-        ...agent,
+        id: agent.id,
+        status: "draft",
+        activeVersionId: void 0,
+        authorId: agent.authorId,
+        metadata: agent.metadata,
         createdAt: now,
         updatedAt: now
       };
       await collection.insertOne(this.serializeAgent(newAgent));
+      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"
+      });
+      newAgent.activeVersionId = versionId;
+      newAgent.status = "published";
+      await collection.updateOne({ id: agent.id }, { $set: { activeVersionId: versionId, status: "published" } });
       return newAgent;
     } catch (error) {
       if (error instanceof MastraError) {
@@ -1034,21 +1066,11 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
       const updateDoc = {
         updatedAt: /* @__PURE__ */ new Date()
       };
-      if (updates.name !== void 0) updateDoc.name = updates.name;
-      if (updates.description !== void 0) updateDoc.description = updates.description;
-      if (updates.instructions !== void 0) updateDoc.instructions = updates.instructions;
-      if (updates.model !== void 0) updateDoc.model = updates.model;
-      if (updates.tools !== void 0) updateDoc.tools = updates.tools;
-      if (updates.defaultOptions !== void 0) updateDoc.defaultOptions = updates.defaultOptions;
-      if (updates.workflows !== void 0) updateDoc.workflows = updates.workflows;
-      if (updates.agents !== void 0) updateDoc.agents = updates.agents;
-      if (updates.inputProcessors !== void 0) updateDoc.inputProcessors = updates.inputProcessors;
-      if (updates.outputProcessors !== void 0) updateDoc.outputProcessors = updates.outputProcessors;
-      if (updates.memory !== void 0) updateDoc.memory = updates.memory;
-      if (updates.scorers !== void 0) updateDoc.scorers = updates.scorers;
-      if (updates.integrationTools !== void 0) updateDoc.integrationTools = updates.integrationTools;
-      if (updates.ownerId !== void 0) updateDoc.ownerId = updates.ownerId;
-      if (updates.activeVersionId !== void 0) updateDoc.activeVersionId = updates.activeVersionId;
+      if (updates.authorId !== void 0) updateDoc.authorId = updates.authorId;
+      if (updates.activeVersionId !== void 0) {
+        updateDoc.activeVersionId = updates.activeVersionId;
+        updateDoc.status = "published";
+      }
       if (updates.metadata !== void 0) {
         const existingMetadata = existingAgent.metadata || {};
         updateDoc.metadata = { ...existingMetadata, ...updates.metadata };
@@ -1153,17 +1175,35 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
       );
     }
   }
+  /**
+   * Transforms a raw MongoDB document into a thin StorageAgentType record.
+   * Only returns metadata-level fields (no config/snapshot fields).
+   */
   transformAgent(doc) {
-    const { _id, ...agent } = doc;
+    const { _id, ...rest } = doc;
     return {
-      ...agent,
-      createdAt: agent.createdAt instanceof Date ? agent.createdAt : new Date(agent.createdAt),
-      updatedAt: agent.updatedAt instanceof Date ? agent.updatedAt : new Date(agent.updatedAt)
+      id: rest.id,
+      status: rest.status,
+      activeVersionId: rest.activeVersionId,
+      authorId: rest.authorId,
+      metadata: rest.metadata,
+      createdAt: rest.createdAt instanceof Date ? rest.createdAt : new Date(rest.createdAt),
+      updatedAt: rest.updatedAt instanceof Date ? rest.updatedAt : new Date(rest.updatedAt)
     };
   }
+  /**
+   * Serializes a thin StorageAgentType record for MongoDB insertion.
+   * Only persists metadata-level fields.
+   */
   serializeAgent(agent) {
     return {
-      ...agent
+      id: agent.id,
+      status: agent.status,
+      activeVersionId: agent.activeVersionId,
+      authorId: agent.authorId,
+      metadata: agent.metadata,
+      createdAt: agent.createdAt,
+      updatedAt: agent.updatedAt
     };
   }
   // ==========================================================================
@@ -1177,12 +1217,15 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
         id: input.id,
         agentId: input.agentId,
         versionNumber: input.versionNumber,
-        name: input.name ?? void 0,
-        snapshot: input.snapshot,
         changedFields: input.changedFields ?? void 0,
         changeMessage: input.changeMessage ?? void 0,
         createdAt: now
       };
+      for (const field of SNAPSHOT_FIELDS) {
+        if (input[field] !== void 0) {
+          versionDoc[field] = input[field];
+        }
+      }
       await collection.insertOne(versionDoc);
       return {
         ...input,
@@ -1365,12 +1408,26 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
   // ==========================================================================
   // Private Helper Methods
   // ==========================================================================
+  /**
+   * Transforms a raw MongoDB version document into an AgentVersion.
+   * Config fields are returned directly (no nested snapshot object).
+   */
   transformVersion(doc) {
     const { _id, ...version } = doc;
-    return {
-      ...version,
+    const result = {
+      id: version.id,
+      agentId: version.agentId,
+      versionNumber: version.versionNumber,
+      changedFields: version.changedFields,
+      changeMessage: version.changeMessage,
       createdAt: version.createdAt instanceof Date ? version.createdAt : new Date(version.createdAt)
     };
+    for (const field of SNAPSHOT_FIELDS) {
+      if (version[field] !== void 0) {
+        result[field] = version[field];
+      }
+    }
+    return result;
   }
 };
 function formatDateForMongoDB(date) {