@mastra/editor 0.13.1-alpha.0 → 0.13.1

package/dist/index.d.cts

@@ -96,12 +96,14 @@ declare class EditorAgentNamespace extends CrudEditorNamespace<StorageCreateAgen
      * Hydrate a stored agent config into a runtime Agent instance.
      */
     protected hydrate(storedAgent: StorageResolvedAgentType): Promise<Agent>;
+    update(input: StorageUpdateAgentInput): Promise<Agent>;
     /**
      * Create a new agent, applying builder defaults for fields not specified in input.
      * Also ensures the referenced workspace (if any) is persisted as a stored workspace.
      */
     create(input: StorageCreateAgentInput): Promise<Agent>;
     private getCodeDefinedAgent;
+    private ensureStoredWorkspaceRefs;
     /**
      * Ensure a workspace reference is persisted in the DB.
      *
@@ -267,6 +269,7 @@ declare class EditorMCPServerNamespace extends CrudEditorNamespace<StorageCreate
 declare class EditorPromptNamespace extends CrudEditorNamespace<StorageCreatePromptBlockInput, StorageUpdatePromptBlockInput, StorageListPromptBlocksInput, StorageListPromptBlocksOutput, StorageListPromptBlocksResolvedOutput, StorageResolvedPromptBlockType> {
     protected onCacheEvict(id: string): void;
     protected getStorageAdapter(): Promise<StorageAdapter<StorageCreatePromptBlockInput, StorageUpdatePromptBlockInput, StorageListPromptBlocksInput, StorageListPromptBlocksOutput, StorageListPromptBlocksResolvedOutput, StorageResolvedPromptBlockType>>;
+    update(input: StorageUpdatePromptBlockInput): Promise<StorageResolvedPromptBlockType>;
     preview(blocks: AgentInstructionBlock[], context: Record<string, unknown>): Promise<string>;
 }
 

package/dist/index.d.ts

@@ -96,12 +96,14 @@ declare class EditorAgentNamespace extends CrudEditorNamespace<StorageCreateAgen
      * Hydrate a stored agent config into a runtime Agent instance.
      */
     protected hydrate(storedAgent: StorageResolvedAgentType): Promise<Agent>;
+    update(input: StorageUpdateAgentInput): Promise<Agent>;
     /**
      * Create a new agent, applying builder defaults for fields not specified in input.
      * Also ensures the referenced workspace (if any) is persisted as a stored workspace.
      */
     create(input: StorageCreateAgentInput): Promise<Agent>;
     private getCodeDefinedAgent;
+    private ensureStoredWorkspaceRefs;
     /**
      * Ensure a workspace reference is persisted in the DB.
      *
@@ -267,6 +269,7 @@ declare class EditorMCPServerNamespace extends CrudEditorNamespace<StorageCreate
 declare class EditorPromptNamespace extends CrudEditorNamespace<StorageCreatePromptBlockInput, StorageUpdatePromptBlockInput, StorageListPromptBlocksInput, StorageListPromptBlocksOutput, StorageListPromptBlocksResolvedOutput, StorageResolvedPromptBlockType> {
     protected onCacheEvict(id: string): void;
     protected getStorageAdapter(): Promise<StorageAdapter<StorageCreatePromptBlockInput, StorageUpdatePromptBlockInput, StorageListPromptBlocksInput, StorageListPromptBlocksOutput, StorageListPromptBlocksResolvedOutput, StorageResolvedPromptBlockType>>;
+    update(input: StorageUpdatePromptBlockInput): Promise<StorageResolvedPromptBlockType>;
     preview(blocks: AgentInstructionBlock[], context: Record<string, unknown>): Promise<string>;
 }
 

package/dist/index.js

@@ -525,7 +525,106 @@ var EditorMCPNamespace = class _EditorMCPNamespace extends CrudEditorNamespace {
   }
 };
 
+// src/namespaces/versioned-update.ts
+import { deepEqual } from "@mastra/core/utils";
+function getProvidedSnapshotFields(input, snapshotFields) {
+  const config = {};
+  for (const field of snapshotFields) {
+    if (input[field] !== void 0) {
+      config[field] = input[field];
+    }
+  }
+  return config;
+}
+function extractSnapshotConfig(version, snapshotFields) {
+  const record = version;
+  const config = {};
+  for (const field of snapshotFields) {
+    if (field in record) {
+      config[field] = record[field];
+    }
+  }
+  return config;
+}
+function getChangedSnapshotFields(previousConfig, providedConfig, snapshotFields) {
+  return snapshotFields.filter(
+    (field) => field in providedConfig && !deepEqual(previousConfig[field], providedConfig[field])
+  );
+}
+function isVersionNumberConflictError(error) {
+  if (!(error instanceof Error)) return false;
+  const message = error.message.toLowerCase();
+  return message.includes("unique") && message.includes("constraint") || message.includes("duplicate key") || message.includes("unique_violation") || message.includes("sqlite_constraint_unique") || message.includes("version number") && message.includes("already exists") || message.includes("versionnumber");
+}
+async function createVersionFromSnapshotUpdate({
+  store,
+  parentId,
+  parentIdField,
+  snapshotFields,
+  providedConfig,
+  changeMessage = "Auto-saved after edit",
+  maxRetries = 3
+}) {
+  let lastError;
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    try {
+      const latestVersion = await store.getLatestVersion(parentId);
+      if (!latestVersion) {
+        return { versionCreated: false };
+      }
+      const previousConfig = extractSnapshotConfig(latestVersion, snapshotFields);
+      const changedFields = getChangedSnapshotFields(previousConfig, providedConfig, snapshotFields);
+      if (changedFields.length === 0) {
+        return { versionCreated: false };
+      }
+      const nextConfig = { ...previousConfig };
+      for (const [field, value] of Object.entries(providedConfig)) {
+        nextConfig[field] = value === null ? void 0 : value;
+      }
+      const version = await store.createVersion({
+        ...nextConfig,
+        id: crypto.randomUUID(),
+        [parentIdField]: parentId,
+        versionNumber: latestVersion.versionNumber + 1,
+        changedFields,
+        changeMessage
+      });
+      return { versionCreated: true, version, changedFields };
+    } catch (error) {
+      lastError = error;
+      if (isVersionNumberConflictError(error) && attempt < maxRetries - 1) {
+        await new Promise((resolve) => setTimeout(resolve, 10 * (attempt + 1)));
+        continue;
+      }
+      throw error;
+    }
+  }
+  throw lastError;
+}
+
 // src/namespaces/agent.ts
+var AGENT_SNAPSHOT_CONFIG_FIELDS = [
+  "name",
+  "description",
+  "instructions",
+  "model",
+  "tools",
+  "defaultOptions",
+  "workflows",
+  "agents",
+  "integrationTools",
+  "toolProviders",
+  "inputProcessors",
+  "outputProcessors",
+  "memory",
+  "scorers",
+  "requestContextSchema",
+  "mcpClients",
+  "skills",
+  "skillsFormat",
+  "workspace",
+  "browser"
+];
 var BUILDER_DEFAULT_FIELDS = ["memory", "workspace", "browser"];
 function defaultModelToStored(entry) {
   return { provider: entry.provider, name: entry.modelId };
@@ -556,6 +655,16 @@ function applyBuilderDefaults(input, builderAgentConfig) {
   }
   return Object.keys(defaults).length > 0 ? { ...input, ...defaults } : input;
 }
+function getProvidedAgentRecordFields(input) {
+  const { id, authorId, visibility, activeVersionId, metadata, status } = input;
+  const recordFields = { id };
+  if (authorId !== void 0) recordFields.authorId = authorId;
+  if (visibility !== void 0) recordFields.visibility = visibility;
+  if (activeVersionId !== void 0) recordFields.activeVersionId = activeVersionId;
+  if (metadata !== void 0) recordFields.metadata = metadata;
+  if (status !== void 0) recordFields.status = status;
+  return Object.keys(recordFields).length > 1 ? recordFields : null;
+}
 var EditorAgentNamespace = class extends CrudEditorNamespace {
   async getStorageAdapter() {
     const storage = this.mastra?.getStorage();
@@ -598,6 +707,53 @@ var EditorAgentNamespace = class extends CrudEditorNamespace {
   async hydrate(storedAgent) {
     return this.createAgentFromStoredConfig(storedAgent);
   }
+  async update(input) {
+    this.ensureRegistered();
+    const storage = this.mastra?.getStorage();
+    if (!storage) throw new Error("Storage is not configured");
+    const store = await storage.getStore("agents");
+    if (!store) throw new Error("Agents storage domain is not available");
+    const existing = await store.getById(input.id);
+    if (!existing) {
+      throw new Error(`Agent with id ${input.id} not found`);
+    }
+    const providedConfig = getProvidedSnapshotFields(
+      input,
+      AGENT_SNAPSHOT_CONFIG_FIELDS
+    );
+    if ("workspace" in providedConfig) {
+      await this.ensureStoredWorkspaceRefs(providedConfig.workspace);
+    }
+    const versionResult = Object.keys(providedConfig).length > 0 ? await createVersionFromSnapshotUpdate({
+      store,
+      parentId: input.id,
+      parentIdField: "agentId",
+      snapshotFields: AGENT_SNAPSHOT_CONFIG_FIELDS,
+      providedConfig
+    }) : { versionCreated: false };
+    const recordFields = getProvidedAgentRecordFields(input);
+    if (recordFields || versionResult.versionCreated) {
+      await store.update({
+        ...recordFields ?? { id: input.id },
+        ...versionResult.versionCreated ? { activeVersionId: versionResult.version.id } : {}
+      });
+    }
+    this._cache.delete(input.id);
+    this.onCacheEvict(input.id);
+    const existingCodeAgent = this.getCodeDefinedAgent(input.id);
+    if (existingCodeAgent) {
+      const hydrated2 = await this.applyStoredOverrides(existingCodeAgent, { status: "draft" });
+      this._cache.set(input.id, hydrated2);
+      return hydrated2;
+    }
+    const resolved = await store.getByIdResolved(input.id, { status: "draft" });
+    if (!resolved) {
+      throw new Error(`Failed to resolve entity ${input.id} after update`);
+    }
+    const hydrated = await this.hydrate(resolved);
+    this._cache.set(input.id, hydrated);
+    return hydrated;
+  }
   /**
    * Create a new agent, applying builder defaults for fields not specified in input.
    * Also ensures the referenced workspace (if any) is persisted as a stored workspace.
@@ -628,6 +784,16 @@ var EditorAgentNamespace = class extends CrudEditorNamespace {
     }
     return agent?.source === "code" ? agent : void 0;
   }
+  async ensureStoredWorkspaceRefs(workspace) {
+    if (!workspace) return;
+    if (this.isConditionalVariants(workspace)) {
+      for (const variant of workspace) {
+        await this.ensureStoredWorkspace(variant.value);
+      }
+      return;
+    }
+    await this.ensureStoredWorkspace(workspace);
+  }
   /**
    * Ensure a workspace reference is persisted in the DB.
    *
@@ -1771,6 +1937,96 @@ var EditorMCPServerNamespace = class extends CrudEditorNamespace {
 };
 
 // src/namespaces/prompt.ts
+var PROMPT_BLOCK_SNAPSHOT_CONFIG_FIELDS = [
+  "name",
+  "description",
+  "content",
+  "rules",
+  "requestContextSchema"
+];
+function deepEqual2(a, b) {
+  if (a === b) return true;
+  if (a == null || b == null) return a === b;
+  if (typeof a !== typeof b) return false;
+  if (Array.isArray(a) && Array.isArray(b)) {
+    return a.length === b.length && a.every((item, index) => deepEqual2(item, b[index]));
+  }
+  if (typeof a === "object" && typeof b === "object") {
+    const aObj = a;
+    const bObj = b;
+    const aKeys = Object.keys(aObj);
+    const bKeys = Object.keys(bObj);
+    return aKeys.length === bKeys.length && aKeys.every((key) => deepEqual2(aObj[key], bObj[key]));
+  }
+  return false;
+}
+function extractConfigFromVersion(version) {
+  return {
+    name: version.name,
+    description: version.description,
+    content: version.content,
+    rules: version.rules,
+    requestContextSchema: version.requestContextSchema
+  };
+}
+function getProvidedConfigFields(input) {
+  const config = {};
+  for (const field of PROMPT_BLOCK_SNAPSHOT_CONFIG_FIELDS) {
+    if (input[field] !== void 0) {
+      config[field] = input[field];
+    }
+  }
+  return config;
+}
+function getProvidedRecordFields(input) {
+  const { id, authorId, activeVersionId, metadata, status } = input;
+  const recordFields = { id };
+  if (authorId !== void 0) recordFields.authorId = authorId;
+  if (activeVersionId !== void 0) recordFields.activeVersionId = activeVersionId;
+  if (metadata !== void 0) recordFields.metadata = metadata;
+  if (status !== void 0) recordFields.status = status;
+  return Object.keys(recordFields).length > 1 ? recordFields : null;
+}
+function getChangedFields(previousConfig, providedConfig) {
+  return PROMPT_BLOCK_SNAPSHOT_CONFIG_FIELDS.filter(
+    (field) => field in providedConfig && !deepEqual2(previousConfig[field], providedConfig[field])
+  );
+}
+function isVersionNumberConflictError2(error) {
+  if (!(error instanceof Error)) return false;
+  const message = error.message.toLowerCase();
+  return message.includes("unique") && message.includes("constraint") || message.includes("duplicate key") || message.includes("unique_violation") || message.includes("sqlite_constraint_unique") || message.includes("version number") && message.includes("already exists") || message.includes("versionnumber");
+}
+async function createPromptBlockVersionWithRetry(store, blockId, providedConfig, maxRetries = 3) {
+  let lastError;
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    try {
+      const latestVersion = await store.getLatestVersion(blockId);
+      if (!latestVersion) return false;
+      const previousConfig = extractConfigFromVersion(latestVersion);
+      const changedFields = getChangedFields(previousConfig, providedConfig);
+      if (changedFields.length === 0) return false;
+      await store.createVersion({
+        ...previousConfig,
+        ...providedConfig,
+        id: crypto.randomUUID(),
+        blockId,
+        versionNumber: latestVersion.versionNumber + 1,
+        changedFields,
+        changeMessage: "Auto-saved after edit"
+      });
+      return true;
+    } catch (error) {
+      lastError = error;
+      if (isVersionNumberConflictError2(error) && attempt < maxRetries - 1) {
+        await new Promise((resolve) => setTimeout(resolve, 10 * (attempt + 1)));
+        continue;
+      }
+      throw error;
+    }
+  }
+  throw lastError;
+}
 var EditorPromptNamespace = class extends CrudEditorNamespace {
   onCacheEvict(id) {
     this.mastra?.removePromptBlock(id);
@@ -1789,6 +2045,33 @@ var EditorPromptNamespace = class extends CrudEditorNamespace {
       listResolved: (args) => store.listResolved(args)
     };
   }
+  async update(input) {
+    this.ensureRegistered();
+    const storage = this.mastra?.getStorage();
+    if (!storage) throw new Error("Storage is not configured");
+    const store = await storage.getStore("promptBlocks");
+    if (!store) throw new Error("Prompt blocks storage domain is not available");
+    const existing = await store.getById(input.id);
+    if (!existing) {
+      throw new Error(`Prompt block with id ${input.id} not found`);
+    }
+    const providedConfig = getProvidedConfigFields(input);
+    if (Object.keys(providedConfig).length > 0) {
+      await createPromptBlockVersionWithRetry(store, input.id, providedConfig);
+    }
+    const recordFields = getProvidedRecordFields(input);
+    if (recordFields) {
+      await store.update(recordFields);
+    }
+    this._cache.delete(input.id);
+    this.onCacheEvict(input.id);
+    const resolved = await store.getByIdResolved(input.id, { status: "draft" });
+    if (!resolved) {
+      throw new Error(`Failed to resolve entity ${input.id} after update`);
+    }
+    this._cache.set(input.id, resolved);
+    return resolved;
+  }
   async preview(blocks, context) {
     this.ensureRegistered();
     const storage = this.mastra?.getStorage();