@mastra/mongodb 1.16.0-alpha.1 → 1.16.0-alpha.2

package/dist/index.js

@@ -1,6 +1,6 @@
 import { v4 } from "@lukeed/uuid";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
-import { AgentsStorage, BackgroundTasksStorage, BlobStore, DatasetsStorage, ExperimentsStorage, MCPClientsStorage, MCPServersStorage, MastraCompositeStore, MemoryStorage, NotificationsStorage, ObservabilityStorage, PromptBlocksStorage, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_NOTIFICATIONS, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WorkflowsStorage, WorkspacesStorage, calculatePagination, createStorageErrorId, createVectorErrorId, ensureDate, hasErrorCode, listTracesArgsSchema, normalizePerPage, normalizeScheduleTarget, parseDuration, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AgentsStorage, BackgroundTasksStorage, BlobStore, DatasetsStorage, ExperimentsStorage, MCPClientsStorage, MCPServersStorage, MastraCompositeStore, MemoryStorage, NotificationsStorage, ObservabilityStorage, PromptBlocksStorage, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_NOTIFICATIONS, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, calculatePagination, createStorageErrorId, createVectorErrorId, ensureDate, hasErrorCode, listTracesArgsSchema, normalizePerPage, normalizeScheduleTarget, parseDuration, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import { MastraVector, validateUpsertInput, validateVectorValues } from "@mastra/core/vector";
 import { MongoBulkWriteError, MongoClient, ObjectId } from "mongodb";
 import { BaseFilterTranslator } from "@mastra/core/vector/filter";
@@ -9,7 +9,7 @@ import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
 //#region package.json
-var version = "1.16.0-alpha.1";
+var version = "1.16.0-alpha.2";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -9935,6 +9935,141 @@ var MongoDBSkillsStorage = class MongoDBSkillsStorage extends SkillsStorage {
 	}
 };
 //#endregion
+//#region src/storage/domains/workflow-definitions/index.ts
+function docToDefinition(doc) {
+	const def = {
+		id: String(doc.id),
+		inputSchema: doc.inputSchema,
+		outputSchema: doc.outputSchema,
+		graph: doc.graph,
+		status: doc.status,
+		source: doc.source,
+		createdAt: doc.createdAt instanceof Date ? doc.createdAt : new Date(doc.createdAt),
+		updatedAt: doc.updatedAt instanceof Date ? doc.updatedAt : new Date(doc.updatedAt)
+	};
+	if (doc.description != null) def.description = doc.description;
+	if (doc.metadata != null) def.metadata = doc.metadata;
+	if (doc.stateSchema != null) def.stateSchema = doc.stateSchema;
+	if (doc.requestContextSchema != null) def.requestContextSchema = doc.requestContextSchema;
+	if (doc.authorId != null) def.authorId = doc.authorId;
+	return def;
+}
+var MongoDBWorkflowDefinitionsStore = class MongoDBWorkflowDefinitionsStore extends WorkflowDefinitionsStorage {
+	#connector;
+	#skipDefaultIndexes;
+	#indexes;
+	static MANAGED_COLLECTIONS = [TABLE_WORKFLOW_DEFINITIONS];
+	constructor(config) {
+		super();
+		this.#connector = resolveMongoDBConfig(config);
+		this.#skipDefaultIndexes = config.skipDefaultIndexes;
+		this.#indexes = config.indexes?.filter((idx) => MongoDBWorkflowDefinitionsStore.MANAGED_COLLECTIONS.includes(idx.collection));
+	}
+	async getCollection(name) {
+		return this.#connector.getCollection(name);
+	}
+	getDefaultIndexDefinitions() {
+		return [{
+			collection: TABLE_WORKFLOW_DEFINITIONS,
+			keys: { id: 1 },
+			options: { unique: true }
+		}, {
+			collection: TABLE_WORKFLOW_DEFINITIONS,
+			keys: { status: 1 }
+		}];
+	}
+	async createDefaultIndexes() {
+		if (this.#skipDefaultIndexes) return;
+		for (const indexDef of this.getDefaultIndexDefinitions()) try {
+			await (await this.getCollection(indexDef.collection)).createIndex(indexDef.keys, indexDef.options);
+		} catch (error) {
+			this.logger?.warn?.(`Failed to create index on ${indexDef.collection}:`, error);
+		}
+	}
+	async createCustomIndexes() {
+		if (!this.#indexes || this.#indexes.length === 0) return;
+		for (const indexDef of this.#indexes) try {
+			await (await this.getCollection(indexDef.collection)).createIndex(indexDef.keys, indexDef.options);
+		} catch (error) {
+			this.logger?.warn?.(`Failed to create custom index on ${indexDef.collection}:`, error);
+		}
+	}
+	async init() {
+		await this.createDefaultIndexes();
+		await this.createCustomIndexes();
+	}
+	async dangerouslyClearAll() {
+		await (await this.getCollection(TABLE_WORKFLOW_DEFINITIONS)).deleteMany({});
+	}
+	async upsert(input) {
+		const now = /* @__PURE__ */ new Date();
+		const collection = await this.getCollection(TABLE_WORKFLOW_DEFINITIONS);
+		if (!await collection.findOne({ id: input.id })) {
+			if (!("inputSchema" in input) || input.inputSchema === void 0) throw new Error(`Cannot create workflow definition "${input.id}": inputSchema is required.`);
+			if (!("outputSchema" in input) || input.outputSchema === void 0) throw new Error(`Cannot create workflow definition "${input.id}": outputSchema is required.`);
+			if (!("graph" in input) || input.graph === void 0) throw new Error(`Cannot create workflow definition "${input.id}": graph is required.`);
+			const doc = {
+				id: input.id,
+				description: input.description ?? null,
+				metadata: input.metadata ?? null,
+				inputSchema: input.inputSchema,
+				outputSchema: input.outputSchema,
+				stateSchema: input.stateSchema ?? null,
+				requestContextSchema: input.requestContextSchema ?? null,
+				graph: input.graph,
+				status: "active",
+				source: "storage",
+				authorId: "authorId" in input ? input.authorId ?? null : null,
+				createdAt: now,
+				updatedAt: now
+			};
+			try {
+				await collection.insertOne(doc);
+			} catch (error) {
+				if (!(error?.code === 11e3) || !await collection.findOne({ id: input.id })) throw error;
+				return this.applyUpdate(input, now);
+			}
+			return docToDefinition(doc);
+		}
+		return this.applyUpdate(input, now);
+	}
+	async applyUpdate(input, now) {
+		const collection = await this.getCollection(TABLE_WORKFLOW_DEFINITIONS);
+		const update = { updatedAt: now };
+		if ("description" in input && input.description !== void 0) update.description = input.description;
+		if ("metadata" in input && input.metadata !== void 0) update.metadata = input.metadata;
+		if ("inputSchema" in input && input.inputSchema !== void 0) update.inputSchema = input.inputSchema;
+		if ("outputSchema" in input && input.outputSchema !== void 0) update.outputSchema = input.outputSchema;
+		if ("stateSchema" in input && input.stateSchema !== void 0) update.stateSchema = input.stateSchema;
+		if ("requestContextSchema" in input && input.requestContextSchema !== void 0) update.requestContextSchema = input.requestContextSchema;
+		if ("graph" in input && input.graph !== void 0) update.graph = input.graph;
+		if ("status" in input && input.status !== void 0) update.status = input.status;
+		if ("authorId" in input && input.authorId !== void 0) update.authorId = input.authorId;
+		await collection.updateOne({ id: input.id }, { $set: update });
+		const updated = await collection.findOne({ id: input.id });
+		if (!updated) throw new Error(`Failed to update workflow definition "${input.id}".`);
+		return docToDefinition(updated);
+	}
+	async get(id) {
+		const doc = await (await this.getCollection(TABLE_WORKFLOW_DEFINITIONS)).findOne({ id });
+		return doc ? docToDefinition(doc) : null;
+	}
+	async list(args) {
+		const collection = await this.getCollection(TABLE_WORKFLOW_DEFINITIONS);
+		const filter = {};
+		if (args?.status) filter.status = args.status;
+		if (args?.authorId !== void 0) filter.authorId = args.authorId;
+		const definitions = (await collection.find(filter).sort({ updatedAt: -1 }).toArray()).map(docToDefinition);
+		return {
+			definitions,
+			total: definitions.length
+		};
+	}
+	async delete(id) {
+		await (await this.getCollection(TABLE_WORKFLOW_DEFINITIONS)).deleteOne({ id });
+	}
+};
+//#endregion
 //#region src/storage/domains/workflows/index.ts
 var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsStorage {
 	#connector;
@@ -10823,6 +10958,7 @@ var MongoDBStore = class extends MastraCompositeStore {
 		const experiments = new MongoDBExperimentsStorage(domainConfig);
 		const backgroundTasks = new BackgroundTasksStorageMongoDB(domainConfig);
 		const schedules = new SchedulesMongoDB(domainConfig);
+		const workflowDefinitions = new MongoDBWorkflowDefinitionsStore(domainConfig);
 		this.stores = {
 			memory,
 			notifications,
@@ -10840,7 +10976,8 @@ var MongoDBStore = class extends MastraCompositeStore {
 			backgroundTasks,
 			datasets,
 			experiments,
-			schedules
+			schedules,
+			workflowDefinitions
 		};
 	}
 	/**
@@ -10960,6 +11097,6 @@ Example Complex Query:
   ]
 }`;
 //#endregion
-export { BackgroundTasksStorageMongoDB, MONGODB_PROMPT, MemoryStorageMongoDB, MongoDBAgentsStorage, MongoDBBlobStore, MongoDBDatasetsStorage, MongoDBExperimentsStorage, MongoDBMCPClientsStorage, MongoDBMCPServersStorage, MongoDBPromptBlocksStorage, MongoDBScorerDefinitionsStorage, MongoDBSkillsStorage, MongoDBStore, MongoDBVector, MongoDBWorkspacesStorage, NotificationsMongoDB, ObservabilityMongoDB, SchedulesMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
+export { BackgroundTasksStorageMongoDB, MONGODB_PROMPT, MemoryStorageMongoDB, MongoDBAgentsStorage, MongoDBBlobStore, MongoDBDatasetsStorage, MongoDBExperimentsStorage, MongoDBMCPClientsStorage, MongoDBMCPServersStorage, MongoDBPromptBlocksStorage, MongoDBScorerDefinitionsStorage, MongoDBSkillsStorage, MongoDBStore, MongoDBVector, MongoDBWorkflowDefinitionsStore, MongoDBWorkspacesStorage, NotificationsMongoDB, ObservabilityMongoDB, SchedulesMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
 
 //# sourceMappingURL=index.js.map