@mastra/factory 0.19.2-alpha.8 → 0.20.0-alpha.10

package/dist/storage/domains/model-packs/base.js

@@ -1,185 +0,0 @@
-import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
-//#region src/storage/domains/model-packs/base.ts
-const MODEL_PACKS_SCHEMA = {
-	name: "model_packs",
-	columns: {
-		id: { type: "uuid-pk" },
-		org_id: { type: "text" },
-		created_by: { type: "text" },
-		name: { type: "text" },
-		build_model_id: { type: "text" },
-		plan_model_id: { type: "text" },
-		fast_model_id: { type: "text" },
-		created_at: { type: "timestamp" },
-		updated_at: { type: "timestamp" }
-	},
-	indexes: [{
-		name: "model_packs_org_name_idx",
-		columns: ["org_id", "name"]
-	}]
-};
-const ACTIVE_MODEL_PACKS_SCHEMA = {
-	name: "active_model_packs",
-	columns: {
-		id: { type: "uuid-pk" },
-		org_id: { type: "text" },
-		user_id: { type: "text" },
-		pack_id: { type: "text" },
-		build_model_id: { type: "text" },
-		plan_model_id: { type: "text" },
-		fast_model_id: { type: "text" },
-		updated_at: { type: "timestamp" }
-	},
-	uniqueIndexes: [{
-		name: "active_model_packs_org_user_key",
-		columns: ["org_id", "user_id"]
-	}]
-};
-function toModelPack(row) {
-	return {
-		id: row.id,
-		orgId: row.org_id,
-		createdBy: row.created_by,
-		name: row.name,
-		models: {
-			build: row.build_model_id,
-			plan: row.plan_model_id,
-			fast: row.fast_model_id
-		},
-		createdAt: row.created_at,
-		updatedAt: row.updated_at
-	};
-}
-function toActiveModelPack(row) {
-	return {
-		orgId: row.org_id,
-		userId: row.user_id,
-		packId: row.pack_id,
-		models: {
-			build: row.build_model_id,
-			plan: row.plan_model_id,
-			fast: row.fast_model_id
-		},
-		updatedAt: row.updated_at
-	};
-}
-var ModelPacksStorage = class extends FactoryStorageDomain {
-	constructor() {
-		super("model-packs");
-	}
-	async init() {
-		await this.ensureCollections([MODEL_PACKS_SCHEMA, ACTIVE_MODEL_PACKS_SCHEMA]);
-	}
-	async dangerouslyClearAll() {
-		await this.ops.deleteMany("active_model_packs", {});
-		await this.ops.deleteMany("model_packs", {});
-	}
-	get #db() {
-		return this.ops;
-	}
-	/** Create or replace a pack by `(orgId, name)` — mirrors the settings.json upsert semantics. */
-	async upsert({ orgId, userId, input }) {
-		const now = /* @__PURE__ */ new Date();
-		const existing = await this.#db.findOne("model_packs", {
-			org_id: orgId,
-			name: input.name
-		});
-		if (existing) {
-			const row = await this.#db.updateAtomic("model_packs", {
-				org_id: orgId,
-				id: existing.id
-			}, () => ({
-				build_model_id: input.models.build,
-				plan_model_id: input.models.plan,
-				fast_model_id: input.models.fast,
-				updated_at: now
-			}));
-			await this.#db.updateMany("active_model_packs", {
-				org_id: orgId,
-				pack_id: `custom:${existing.id}`
-			}, {
-				build_model_id: input.models.build,
-				plan_model_id: input.models.plan,
-				fast_model_id: input.models.fast,
-				updated_at: now
-			});
-			return toModelPack(row ?? existing);
-		}
-		return toModelPack(await this.#db.insertOne("model_packs", {
-			org_id: orgId,
-			created_by: userId,
-			name: input.name,
-			build_model_id: input.models.build,
-			plan_model_id: input.models.plan,
-			fast_model_id: input.models.fast,
-			created_at: now,
-			updated_at: now
-		}));
-	}
-	async list({ orgId }) {
-		return (await this.#db.findMany("model_packs", { org_id: orgId }, { orderBy: [["name", "asc"]] })).map(toModelPack);
-	}
-	async get({ orgId, id }) {
-		const row = await this.#db.findOne("model_packs", {
-			org_id: orgId,
-			id
-		});
-		return row ? toModelPack(row) : null;
-	}
-	async setActive({ orgId, userId, packId, models }) {
-		const now = /* @__PURE__ */ new Date();
-		const values = {
-			pack_id: packId,
-			build_model_id: models.build,
-			plan_model_id: models.plan,
-			fast_model_id: models.fast,
-			updated_at: now
-		};
-		const updateExisting = () => this.#db.updateAtomic("active_model_packs", {
-			org_id: orgId,
-			user_id: userId
-		}, () => values);
-		const updated = await updateExisting();
-		if (updated) return toActiveModelPack(updated);
-		try {
-			return toActiveModelPack(await this.#db.insertOne("active_model_packs", {
-				org_id: orgId,
-				user_id: userId,
-				...values
-			}));
-		} catch (error) {
-			if (!(error instanceof UniqueViolationError)) throw error;
-			const row = await updateExisting();
-			if (!row) throw error;
-			return toActiveModelPack(row);
-		}
-	}
-	async getActive({ orgId, userId }) {
-		const row = await this.#db.findOne("active_model_packs", {
-			org_id: orgId,
-			user_id: userId
-		});
-		return row ? toActiveModelPack(row) : null;
-	}
-	async clearActive({ orgId, userId }) {
-		return await this.#db.deleteMany("active_model_packs", {
-			org_id: orgId,
-			user_id: userId
-		}) > 0;
-	}
-	async delete({ orgId, id }) {
-		const deleted = await this.#db.deleteMany("model_packs", {
-			org_id: orgId,
-			id
-		}) > 0;
-		if (deleted) await this.#db.deleteMany("active_model_packs", {
-			org_id: orgId,
-			pack_id: `custom:${id}`
-		});
-		return deleted;
-	}
-};
-//#endregion
-export { ACTIVE_MODEL_PACKS_SCHEMA, MODEL_PACKS_SCHEMA, ModelPacksStorage };
-
-//# sourceMappingURL=base.js.map

package/dist/storage/domains/model-packs/base.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"base.js","names":["#db"],"sources":["../../../../src/storage/domains/model-packs/base.ts"],"sourcesContent":["import { FactoryStorageDomain, UniqueViolationError } from '@mastra/core/storage';\nimport type { CollectionSchema, FactoryStorageOps } from '@mastra/core/storage';\n\n/** A saved custom model pack: one model per mode (build / plan / fast). */\nexport interface ModelPackRecord {\n  id: string;\n  orgId: string;\n  createdBy: string;\n  name: string;\n  models: { build: string; plan: string; fast: string };\n  createdAt: Date;\n  updatedAt: Date;\n}\n\nexport interface UpsertModelPackInput {\n  name: string;\n  models: { build: string; plan: string; fast: string };\n}\n\nexport interface ActiveModelPackRecord {\n  orgId: string;\n  userId: string;\n  packId: string;\n  models: { build: string; plan: string; fast: string };\n  updatedAt: Date;\n}\n\nexport const MODEL_PACKS_SCHEMA: CollectionSchema = {\n  name: 'model_packs',\n  columns: {\n    id: { type: 'uuid-pk' },\n    org_id: { type: 'text' },\n    created_by: { type: 'text' },\n    name: { type: 'text' },\n    build_model_id: { type: 'text' },\n    plan_model_id: { type: 'text' },\n    fast_model_id: { type: 'text' },\n    created_at: { type: 'timestamp' },\n    updated_at: { type: 'timestamp' },\n  },\n  indexes: [{ name: 'model_packs_org_name_idx', columns: ['org_id', 'name'] }],\n};\n\nexport const ACTIVE_MODEL_PACKS_SCHEMA: CollectionSchema = {\n  name: 'active_model_packs',\n  columns: {\n    id: { type: 'uuid-pk' },\n    org_id: { type: 'text' },\n    user_id: { type: 'text' },\n    pack_id: { type: 'text' },\n    build_model_id: { type: 'text' },\n    plan_model_id: { type: 'text' },\n    fast_model_id: { type: 'text' },\n    updated_at: { type: 'timestamp' },\n  },\n  uniqueIndexes: [{ name: 'active_model_packs_org_user_key', columns: ['org_id', 'user_id'] }],\n};\n\ninterface ModelPackDbRow extends Record<string, unknown> {\n  id: string;\n  org_id: string;\n  created_by: string;\n  name: string;\n  build_model_id: string;\n  plan_model_id: string;\n  fast_model_id: string;\n  created_at: Date;\n  updated_at: Date;\n}\n\ninterface ActiveModelPackDbRow extends Record<string, unknown> {\n  id: string;\n  org_id: string;\n  user_id: string;\n  pack_id: string;\n  build_model_id: string;\n  plan_model_id: string;\n  fast_model_id: string;\n  updated_at: Date;\n}\n\nfunction toModelPack(row: ModelPackDbRow): ModelPackRecord {\n  return {\n    id: row.id,\n    orgId: row.org_id,\n    createdBy: row.created_by,\n    name: row.name,\n    models: { build: row.build_model_id, plan: row.plan_model_id, fast: row.fast_model_id },\n    createdAt: row.created_at,\n    updatedAt: row.updated_at,\n  };\n}\n\nfunction toActiveModelPack(row: ActiveModelPackDbRow): ActiveModelPackRecord {\n  return {\n    orgId: row.org_id,\n    userId: row.user_id,\n    packId: row.pack_id,\n    models: { build: row.build_model_id, plan: row.plan_model_id, fast: row.fast_model_id },\n    updatedAt: row.updated_at,\n  };\n}\n\nexport class ModelPacksStorage extends FactoryStorageDomain {\n  constructor() {\n    super('model-packs');\n  }\n\n  async init(): Promise<void> {\n    await this.ensureCollections([MODEL_PACKS_SCHEMA, ACTIVE_MODEL_PACKS_SCHEMA]);\n  }\n\n  async dangerouslyClearAll(): Promise<void> {\n    await this.ops.deleteMany('active_model_packs', {});\n    await this.ops.deleteMany('model_packs', {});\n  }\n\n  get #db(): FactoryStorageOps {\n    return this.ops;\n  }\n\n  /** Create or replace a pack by `(orgId, name)` — mirrors the settings.json upsert semantics. */\n  async upsert({\n    orgId,\n    userId,\n    input,\n  }: {\n    orgId: string;\n    userId: string;\n    input: UpsertModelPackInput;\n  }): Promise<ModelPackRecord> {\n    const now = new Date();\n    const existing = await this.#db.findOne<ModelPackDbRow>('model_packs', { org_id: orgId, name: input.name });\n    if (existing) {\n      const row = await this.#db.updateAtomic<ModelPackDbRow>(\n        'model_packs',\n        { org_id: orgId, id: existing.id },\n        () => ({\n          build_model_id: input.models.build,\n          plan_model_id: input.models.plan,\n          fast_model_id: input.models.fast,\n          updated_at: now,\n        }),\n      );\n      await this.#db.updateMany(\n        'active_model_packs',\n        { org_id: orgId, pack_id: `custom:${existing.id}` },\n        {\n          build_model_id: input.models.build,\n          plan_model_id: input.models.plan,\n          fast_model_id: input.models.fast,\n          updated_at: now,\n        },\n      );\n      return toModelPack(row ?? existing);\n    }\n    const row = await this.#db.insertOne<ModelPackDbRow>('model_packs', {\n      org_id: orgId,\n      created_by: userId,\n      name: input.name,\n      build_model_id: input.models.build,\n      plan_model_id: input.models.plan,\n      fast_model_id: input.models.fast,\n      created_at: now,\n      updated_at: now,\n    });\n    return toModelPack(row);\n  }\n\n  async list({ orgId }: { orgId: string }): Promise<ModelPackRecord[]> {\n    const rows = await this.#db.findMany<ModelPackDbRow>(\n      'model_packs',\n      { org_id: orgId },\n      { orderBy: [['name', 'asc']] },\n    );\n    return rows.map(toModelPack);\n  }\n\n  async get({ orgId, id }: { orgId: string; id: string }): Promise<ModelPackRecord | null> {\n    const row = await this.#db.findOne<ModelPackDbRow>('model_packs', { org_id: orgId, id });\n    return row ? toModelPack(row) : null;\n  }\n\n  async setActive({\n    orgId,\n    userId,\n    packId,\n    models,\n  }: {\n    orgId: string;\n    userId: string;\n    packId: string;\n    models: ActiveModelPackRecord['models'];\n  }): Promise<ActiveModelPackRecord> {\n    const now = new Date();\n    const values = {\n      pack_id: packId,\n      build_model_id: models.build,\n      plan_model_id: models.plan,\n      fast_model_id: models.fast,\n      updated_at: now,\n    };\n    const updateExisting = () =>\n      this.#db.updateAtomic<ActiveModelPackDbRow>(\n        'active_model_packs',\n        { org_id: orgId, user_id: userId },\n        () => values,\n      );\n\n    const updated = await updateExisting();\n    if (updated) return toActiveModelPack(updated);\n\n    try {\n      return toActiveModelPack(\n        await this.#db.insertOne<ActiveModelPackDbRow>('active_model_packs', {\n          org_id: orgId,\n          user_id: userId,\n          ...values,\n        }),\n      );\n    } catch (error) {\n      if (!(error instanceof UniqueViolationError)) throw error;\n      const row = await updateExisting();\n      if (!row) throw error;\n      return toActiveModelPack(row);\n    }\n  }\n\n  async getActive({ orgId, userId }: { orgId: string; userId: string }): Promise<ActiveModelPackRecord | null> {\n    const row = await this.#db.findOne<ActiveModelPackDbRow>('active_model_packs', {\n      org_id: orgId,\n      user_id: userId,\n    });\n    return row ? toActiveModelPack(row) : null;\n  }\n\n  async clearActive({ orgId, userId }: { orgId: string; userId: string }): Promise<boolean> {\n    return (await this.#db.deleteMany('active_model_packs', { org_id: orgId, user_id: userId })) > 0;\n  }\n\n  async delete({ orgId, id }: { orgId: string; id: string }): Promise<boolean> {\n    const deleted = (await this.#db.deleteMany('model_packs', { org_id: orgId, id })) > 0;\n    if (deleted) {\n      await this.#db.deleteMany('active_model_packs', { org_id: orgId, pack_id: `custom:${id}` });\n    }\n    return deleted;\n  }\n}\n"],"mappings":";;AA2BA,MAAa,qBAAuC;CAClD,MAAM;CACN,SAAS;EACP,IAAI,EAAE,MAAM,UAAU;EACtB,QAAQ,EAAE,MAAM,OAAO;EACvB,YAAY,EAAE,MAAM,OAAO;EAC3B,MAAM,EAAE,MAAM,OAAO;EACrB,gBAAgB,EAAE,MAAM,OAAO;EAC/B,eAAe,EAAE,MAAM,OAAO;EAC9B,eAAe,EAAE,MAAM,OAAO;EAC9B,YAAY,EAAE,MAAM,YAAY;EAChC,YAAY,EAAE,MAAM,YAAY;CAClC;CACA,SAAS,CAAC;EAAE,MAAM;EAA4B,SAAS,CAAC,UAAU,MAAM;CAAE,CAAC;AAC7E;AAEA,MAAa,4BAA8C;CACzD,MAAM;CACN,SAAS;EACP,IAAI,EAAE,MAAM,UAAU;EACtB,QAAQ,EAAE,MAAM,OAAO;EACvB,SAAS,EAAE,MAAM,OAAO;EACxB,SAAS,EAAE,MAAM,OAAO;EACxB,gBAAgB,EAAE,MAAM,OAAO;EAC/B,eAAe,EAAE,MAAM,OAAO;EAC9B,eAAe,EAAE,MAAM,OAAO;EAC9B,YAAY,EAAE,MAAM,YAAY;CAClC;CACA,eAAe,CAAC;EAAE,MAAM;EAAmC,SAAS,CAAC,UAAU,SAAS;CAAE,CAAC;AAC7F;AAyBA,SAAS,YAAY,KAAsC;CACzD,OAAO;EACL,IAAI,IAAI;EACR,OAAO,IAAI;EACX,WAAW,IAAI;EACf,MAAM,IAAI;EACV,QAAQ;GAAE,OAAO,IAAI;GAAgB,MAAM,IAAI;GAAe,MAAM,IAAI;EAAc;EACtF,WAAW,IAAI;EACf,WAAW,IAAI;CACjB;AACF;AAEA,SAAS,kBAAkB,KAAkD;CAC3E,OAAO;EACL,OAAO,IAAI;EACX,QAAQ,IAAI;EACZ,QAAQ,IAAI;EACZ,QAAQ;GAAE,OAAO,IAAI;GAAgB,MAAM,IAAI;GAAe,MAAM,IAAI;EAAc;EACtF,WAAW,IAAI;CACjB;AACF;AAEA,IAAa,oBAAb,cAAuC,qBAAqB;CAC1D,cAAc;EACZ,MAAM,aAAa;CACrB;CAEA,MAAM,OAAsB;EAC1B,MAAM,KAAK,kBAAkB,CAAC,oBAAoB,yBAAyB,CAAC;CAC9E;CAEA,MAAM,sBAAqC;EACzC,MAAM,KAAK,IAAI,WAAW,sBAAsB,CAAC,CAAC;EAClD,MAAM,KAAK,IAAI,WAAW,eAAe,CAAC,CAAC;CAC7C;CAEA,IAAIA,MAAyB;EAC3B,OAAO,KAAK;CACd;;CAGA,MAAM,OAAO,EACX,OACA,QACA,SAK2B;EAC3B,MAAM,sBAAM,IAAI,KAAK;EACrB,MAAM,WAAW,MAAM,KAAKA,IAAI,QAAwB,eAAe;GAAE,QAAQ;GAAO,MAAM,MAAM;EAAK,CAAC;EAC1G,IAAI,UAAU;GACZ,MAAM,MAAM,MAAM,KAAKA,IAAI,aACzB,eACA;IAAE,QAAQ;IAAO,IAAI,SAAS;GAAG,UAC1B;IACL,gBAAgB,MAAM,OAAO;IAC7B,eAAe,MAAM,OAAO;IAC5B,eAAe,MAAM,OAAO;IAC5B,YAAY;GACd,EACF;GACA,MAAM,KAAKA,IAAI,WACb,sBACA;IAAE,QAAQ;IAAO,SAAS,UAAU,SAAS;GAAK,GAClD;IACE,gBAAgB,MAAM,OAAO;IAC7B,eAAe,MAAM,OAAO;IAC5B,eAAe,MAAM,OAAO;IAC5B,YAAY;GACd,CACF;GACA,OAAO,YAAY,OAAO,QAAQ;EACpC;EAWA,OAAO,YAAY,MAVD,KAAKA,IAAI,UAA0B,eAAe;GAClE,QAAQ;GACR,YAAY;GACZ,MAAM,MAAM;GACZ,gBAAgB,MAAM,OAAO;GAC7B,eAAe,MAAM,OAAO;GAC5B,eAAe,MAAM,OAAO;GAC5B,YAAY;GACZ,YAAY;EACd,CAAC,CACqB;CACxB;CAEA,MAAM,KAAK,EAAE,SAAwD;EAMnE,QAAO,MALY,KAAKA,IAAI,SAC1B,eACA,EAAE,QAAQ,MAAM,GAChB,EAAE,SAAS,CAAC,CAAC,QAAQ,KAAK,CAAC,EAAE,CAC/B,EAAA,CACY,IAAI,WAAW;CAC7B;CAEA,MAAM,IAAI,EAAE,OAAO,MAAsE;EACvF,MAAM,MAAM,MAAM,KAAKA,IAAI,QAAwB,eAAe;GAAE,QAAQ;GAAO;EAAG,CAAC;EACvF,OAAO,MAAM,YAAY,GAAG,IAAI;CAClC;CAEA,MAAM,UAAU,EACd,OACA,QACA,QACA,UAMiC;EACjC,MAAM,sBAAM,IAAI,KAAK;EACrB,MAAM,SAAS;GACb,SAAS;GACT,gBAAgB,OAAO;GACvB,eAAe,OAAO;GACtB,eAAe,OAAO;GACtB,YAAY;EACd;EACA,MAAM,uBACJ,KAAKA,IAAI,aACP,sBACA;GAAE,QAAQ;GAAO,SAAS;EAAO,SAC3B,MACR;EAEF,MAAM,UAAU,MAAM,eAAe;EACrC,IAAI,SAAS,OAAO,kBAAkB,OAAO;EAE7C,IAAI;GACF,OAAO,kBACL,MAAM,KAAKA,IAAI,UAAgC,sBAAsB;IACnE,QAAQ;IACR,SAAS;IACT,GAAG;GACL,CAAC,CACH;EACF,SAAS,OAAO;GACd,IAAI,EAAE,iBAAiB,uBAAuB,MAAM;GACpD,MAAM,MAAM,MAAM,eAAe;GACjC,IAAI,CAAC,KAAK,MAAM;GAChB,OAAO,kBAAkB,GAAG;EAC9B;CACF;CAEA,MAAM,UAAU,EAAE,OAAO,UAAoF;EAC3G,MAAM,MAAM,MAAM,KAAKA,IAAI,QAA8B,sBAAsB;GAC7E,QAAQ;GACR,SAAS;EACX,CAAC;EACD,OAAO,MAAM,kBAAkB,GAAG,IAAI;CACxC;CAEA,MAAM,YAAY,EAAE,OAAO,UAA+D;EACxF,OAAQ,MAAM,KAAKA,IAAI,WAAW,sBAAsB;GAAE,QAAQ;GAAO,SAAS;EAAO,CAAC,IAAK;CACjG;CAEA,MAAM,OAAO,EAAE,OAAO,MAAuD;EAC3E,MAAM,UAAW,MAAM,KAAKA,IAAI,WAAW,eAAe;GAAE,QAAQ;GAAO;EAAG,CAAC,IAAK;EACpF,IAAI,SACF,MAAM,KAAKA,IAAI,WAAW,sBAAsB;GAAE,QAAQ;GAAO,SAAS,UAAU;EAAK,CAAC;EAE5F,OAAO;CACT;AACF"}