@mastra/factory 0.1.0-alpha.2 → 0.1.0-alpha.4

package/dist/storage/domains/custom-providers/base.js

@@ -0,0 +1,150 @@
+// src/storage/domains/custom-providers/base.ts
+import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
+var CUSTOM_PROVIDERS_SCHEMA = {
+  name: "custom_providers",
+  columns: {
+    id: { type: "uuid-pk" },
+    org_id: { type: "text" },
+    created_by: { type: "text" },
+    provider_id: { type: "text" },
+    name: { type: "text" },
+    url: { type: "text" },
+    api_key: { type: "text", nullable: true },
+    models: { type: "json" },
+    created_at: { type: "timestamp" },
+    updated_at: { type: "timestamp" }
+  },
+  uniqueIndexes: [{ name: "custom_providers_org_provider_key", columns: ["org_id", "provider_id"] }]
+};
+function toRecord(row) {
+  return {
+    id: row.id,
+    orgId: row.org_id,
+    createdBy: row.created_by,
+    providerId: row.provider_id,
+    name: row.name,
+    url: row.url,
+    apiKey: row.api_key,
+    models: Array.isArray(row.models) ? row.models : [],
+    createdAt: row.created_at,
+    updatedAt: row.updated_at
+  };
+}
+var CustomProvidersStorage = class extends FactoryStorageDomain {
+  constructor() {
+    super("custom-providers");
+  }
+  async init() {
+    await this.ensureCollections([CUSTOM_PROVIDERS_SCHEMA]);
+  }
+  async dangerouslyClearAll() {
+    await this.ops.deleteMany("custom_providers", {});
+  }
+  get #db() {
+    return this.ops;
+  }
+  /**
+   * Create or wholesale-replace a provider by `(orgId, providerId)` — mirrors
+   * the settings.json upsert semantics (no key retention: an absent `apiKey`
+   * clears the stored key). `previousProviderId` renames the provider: the
+   * common path is a single atomic in-place update of the old row; renaming
+   * onto an id that already exists overwrites the target first and removes the
+   * old row last, so no interleaving or failure can lose the provider.
+   */
+  async upsert({
+    orgId,
+    userId,
+    input,
+    previousProviderId
+  }) {
+    const now = /* @__PURE__ */ new Date();
+    const renameFrom = previousProviderId && previousProviderId !== input.providerId ? previousProviderId : void 0;
+    if (renameFrom) {
+      const target = await this.#db.findOne("custom_providers", {
+        org_id: orgId,
+        provider_id: input.providerId
+      });
+      if (!target) {
+        const renamed = await this.#db.updateAtomic(
+          "custom_providers",
+          { org_id: orgId, provider_id: renameFrom },
+          () => ({
+            provider_id: input.providerId,
+            name: input.name,
+            url: input.url,
+            api_key: input.apiKey ?? null,
+            models: input.models,
+            updated_at: now
+          })
+        );
+        if (renamed) return toRecord(renamed);
+      }
+    }
+    const record = await this.#write({ orgId, userId, input, now });
+    if (renameFrom) {
+      await this.#db.deleteMany("custom_providers", { org_id: orgId, provider_id: renameFrom });
+    }
+    return record;
+  }
+  /**
+   * Update-first, then insert-and-catch-unique-violation (see `queue_health`):
+   * concurrent creates of the same provider both succeed (last write wins on
+   * the single row) instead of one failing on the unique index, and a row
+   * deleted mid-flight is recreated rather than reported as a stale success.
+   */
+  async #write({
+    orgId,
+    userId,
+    input,
+    now
+  }) {
+    const updateExisting = () => this.#db.updateAtomic(
+      "custom_providers",
+      { org_id: orgId, provider_id: input.providerId },
+      () => ({
+        name: input.name,
+        url: input.url,
+        api_key: input.apiKey ?? null,
+        models: input.models,
+        updated_at: now
+      })
+    );
+    const updated = await updateExisting();
+    if (updated) return toRecord(updated);
+    try {
+      const row = await this.#db.insertOne("custom_providers", {
+        org_id: orgId,
+        created_by: userId,
+        provider_id: input.providerId,
+        name: input.name,
+        url: input.url,
+        api_key: input.apiKey ?? null,
+        models: input.models,
+        created_at: now,
+        updated_at: now
+      });
+      return toRecord(row);
+    } catch (error) {
+      if (!(error instanceof UniqueViolationError)) throw error;
+      const row = await updateExisting();
+      if (!row) throw error;
+      return toRecord(row);
+    }
+  }
+  async list({ orgId }) {
+    const rows = await this.#db.findMany(
+      "custom_providers",
+      { org_id: orgId },
+      { orderBy: [["name", "asc"]] }
+    );
+    return rows.map(toRecord);
+  }
+  async delete({ orgId, providerId }) {
+    return await this.#db.deleteMany("custom_providers", { org_id: orgId, provider_id: providerId }) > 0;
+  }
+};
+export {
+  CUSTOM_PROVIDERS_SCHEMA,
+  CustomProvidersStorage
+};
+//# sourceMappingURL=base.js.map

package/dist/storage/domains/custom-providers/base.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/storage/domains/custom-providers/base.ts"],"sourcesContent":["import { FactoryStorageDomain, UniqueViolationError } from '@mastra/core/storage';\nimport type { CollectionSchema, FactoryStorageOps } from '@mastra/core/storage';\n\n/**\n * A user-defined OpenAI-compatible provider. The DB-backed counterpart of the\n * local `settings.json` `customProviders` entries: org-scoped rows keyed by the\n * derived provider id (name slug), API key stored alongside so a deployed\n * server never reads the shared settings file.\n */\nexport interface CustomProviderRecord {\n  id: string;\n  orgId: string;\n  createdBy: string;\n  /** Stable slug derived from the name (`getCustomProviderId`). */\n  providerId: string;\n  name: string;\n  url: string;\n  apiKey: string | null;\n  models: string[];\n  createdAt: Date;\n  updatedAt: Date;\n}\n\nexport interface UpsertCustomProviderInput {\n  providerId: string;\n  name: string;\n  url: string;\n  apiKey?: string;\n  models: string[];\n}\n\nexport const CUSTOM_PROVIDERS_SCHEMA: CollectionSchema = {\n  name: 'custom_providers',\n  columns: {\n    id: { type: 'uuid-pk' },\n    org_id: { type: 'text' },\n    created_by: { type: 'text' },\n    provider_id: { type: 'text' },\n    name: { type: 'text' },\n    url: { type: 'text' },\n    api_key: { type: 'text', nullable: true },\n    models: { type: 'json' },\n    created_at: { type: 'timestamp' },\n    updated_at: { type: 'timestamp' },\n  },\n  uniqueIndexes: [{ name: 'custom_providers_org_provider_key', columns: ['org_id', 'provider_id'] }],\n};\n\ninterface CustomProviderDbRow extends Record<string, unknown> {\n  id: string;\n  org_id: string;\n  created_by: string;\n  provider_id: string;\n  name: string;\n  url: string;\n  api_key: string | null;\n  models: string[];\n  created_at: Date;\n  updated_at: Date;\n}\n\nfunction toRecord(row: CustomProviderDbRow): CustomProviderRecord {\n  return {\n    id: row.id,\n    orgId: row.org_id,\n    createdBy: row.created_by,\n    providerId: row.provider_id,\n    name: row.name,\n    url: row.url,\n    apiKey: row.api_key,\n    models: Array.isArray(row.models) ? row.models : [],\n    createdAt: row.created_at,\n    updatedAt: row.updated_at,\n  };\n}\n\nexport class CustomProvidersStorage extends FactoryStorageDomain {\n  constructor() {\n    super('custom-providers');\n  }\n\n  async init(): Promise<void> {\n    await this.ensureCollections([CUSTOM_PROVIDERS_SCHEMA]);\n  }\n\n  async dangerouslyClearAll(): Promise<void> {\n    await this.ops.deleteMany('custom_providers', {});\n  }\n\n  get #db(): FactoryStorageOps {\n    return this.ops;\n  }\n\n  /**\n   * Create or wholesale-replace a provider by `(orgId, providerId)` — mirrors\n   * the settings.json upsert semantics (no key retention: an absent `apiKey`\n   * clears the stored key). `previousProviderId` renames the provider: the\n   * common path is a single atomic in-place update of the old row; renaming\n   * onto an id that already exists overwrites the target first and removes the\n   * old row last, so no interleaving or failure can lose the provider.\n   */\n  async upsert({\n    orgId,\n    userId,\n    input,\n    previousProviderId,\n  }: {\n    orgId: string;\n    userId: string;\n    input: UpsertCustomProviderInput;\n    previousProviderId?: string;\n  }): Promise<CustomProviderRecord> {\n    const now = new Date();\n    const renameFrom = previousProviderId && previousProviderId !== input.providerId ? previousProviderId : undefined;\n\n    if (renameFrom) {\n      const target = await this.#db.findOne<CustomProviderDbRow>('custom_providers', {\n        org_id: orgId,\n        provider_id: input.providerId,\n      });\n      if (!target) {\n        // In-place rename: one atomic write with no delete/insert gap, so a\n        // failure can never drop the provider (and `created_at`/`created_by`\n        // are preserved). If a concurrent create grabs the new id between the\n        // lookup and this update, the unique index rejects the update and the\n        // old row stays intact — the request fails cleanly and a retry\n        // converges on the rename-onto-existing path below.\n        const renamed = await this.#db.updateAtomic<CustomProviderDbRow>(\n          'custom_providers',\n          { org_id: orgId, provider_id: renameFrom },\n          () => ({\n            provider_id: input.providerId,\n            name: input.name,\n            url: input.url,\n            api_key: input.apiKey ?? null,\n            models: input.models,\n            updated_at: now,\n          }),\n        );\n        if (renamed) return toRecord(renamed);\n        // Old row already gone — fall through to a plain create of the new id.\n      }\n      // Rename onto an existing id: overwrite the target below, then delete\n      // the old row last — a failure in the gap leaves a redundant duplicate\n      // (recoverable by re-submitting or deleting) instead of losing data.\n    }\n\n    const record = await this.#write({ orgId, userId, input, now });\n    if (renameFrom) {\n      await this.#db.deleteMany('custom_providers', { org_id: orgId, provider_id: renameFrom });\n    }\n    return record;\n  }\n\n  /**\n   * Update-first, then insert-and-catch-unique-violation (see `queue_health`):\n   * concurrent creates of the same provider both succeed (last write wins on\n   * the single row) instead of one failing on the unique index, and a row\n   * deleted mid-flight is recreated rather than reported as a stale success.\n   */\n  async #write({\n    orgId,\n    userId,\n    input,\n    now,\n  }: {\n    orgId: string;\n    userId: string;\n    input: UpsertCustomProviderInput;\n    now: Date;\n  }): Promise<CustomProviderRecord> {\n    const updateExisting = () =>\n      this.#db.updateAtomic<CustomProviderDbRow>(\n        'custom_providers',\n        { org_id: orgId, provider_id: input.providerId },\n        () => ({\n          name: input.name,\n          url: input.url,\n          api_key: input.apiKey ?? null,\n          models: input.models,\n          updated_at: now,\n        }),\n      );\n\n    const updated = await updateExisting();\n    if (updated) return toRecord(updated);\n\n    try {\n      const row = await this.#db.insertOne<CustomProviderDbRow>('custom_providers', {\n        org_id: orgId,\n        created_by: userId,\n        provider_id: input.providerId,\n        name: input.name,\n        url: input.url,\n        api_key: input.apiKey ?? null,\n        models: input.models,\n        created_at: now,\n        updated_at: now,\n      });\n      return toRecord(row);\n    } catch (error) {\n      if (!(error instanceof UniqueViolationError)) throw error;\n      // Lost the insert race — apply this write to the winning row.\n      const row = await updateExisting();\n      if (!row) throw error;\n      return toRecord(row);\n    }\n  }\n\n  async list({ orgId }: { orgId: string }): Promise<CustomProviderRecord[]> {\n    const rows = await this.#db.findMany<CustomProviderDbRow>(\n      'custom_providers',\n      { org_id: orgId },\n      { orderBy: [['name', 'asc']] },\n    );\n    return rows.map(toRecord);\n  }\n\n  async delete({ orgId, providerId }: { orgId: string; providerId: string }): Promise<boolean> {\n    return (await this.#db.deleteMany('custom_providers', { org_id: orgId, provider_id: providerId })) > 0;\n  }\n}\n"],"mappings":";AAAA,SAAS,sBAAsB,4BAA4B;AA+BpD,IAAM,0BAA4C;AAAA,EACvD,MAAM;AAAA,EACN,SAAS;AAAA,IACP,IAAI,EAAE,MAAM,UAAU;AAAA,IACtB,QAAQ,EAAE,MAAM,OAAO;AAAA,IACvB,YAAY,EAAE,MAAM,OAAO;AAAA,IAC3B,aAAa,EAAE,MAAM,OAAO;AAAA,IAC5B,MAAM,EAAE,MAAM,OAAO;AAAA,IACrB,KAAK,EAAE,MAAM,OAAO;AAAA,IACpB,SAAS,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACxC,QAAQ,EAAE,MAAM,OAAO;AAAA,IACvB,YAAY,EAAE,MAAM,YAAY;AAAA,IAChC,YAAY,EAAE,MAAM,YAAY;AAAA,EAClC;AAAA,EACA,eAAe,CAAC,EAAE,MAAM,qCAAqC,SAAS,CAAC,UAAU,aAAa,EAAE,CAAC;AACnG;AAeA,SAAS,SAAS,KAAgD;AAChE,SAAO;AAAA,IACL,IAAI,IAAI;AAAA,IACR,OAAO,IAAI;AAAA,IACX,WAAW,IAAI;AAAA,IACf,YAAY,IAAI;AAAA,IAChB,MAAM,IAAI;AAAA,IACV,KAAK,IAAI;AAAA,IACT,QAAQ,IAAI;AAAA,IACZ,QAAQ,MAAM,QAAQ,IAAI,MAAM,IAAI,IAAI,SAAS,CAAC;AAAA,IAClD,WAAW,IAAI;AAAA,IACf,WAAW,IAAI;AAAA,EACjB;AACF;AAEO,IAAM,yBAAN,cAAqC,qBAAqB;AAAA,EAC/D,cAAc;AACZ,UAAM,kBAAkB;AAAA,EAC1B;AAAA,EAEA,MAAM,OAAsB;AAC1B,UAAM,KAAK,kBAAkB,CAAC,uBAAuB,CAAC;AAAA,EACxD;AAAA,EAEA,MAAM,sBAAqC;AACzC,UAAM,KAAK,IAAI,WAAW,oBAAoB,CAAC,CAAC;AAAA,EAClD;AAAA,EAEA,IAAI,MAAyB;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKkC;AAChC,UAAM,MAAM,oBAAI,KAAK;AACrB,UAAM,aAAa,sBAAsB,uBAAuB,MAAM,aAAa,qBAAqB;AAExG,QAAI,YAAY;AACd,YAAM,SAAS,MAAM,KAAK,IAAI,QAA6B,oBAAoB;AAAA,QAC7E,QAAQ;AAAA,QACR,aAAa,MAAM;AAAA,MACrB,CAAC;AACD,UAAI,CAAC,QAAQ;AAOX,cAAM,UAAU,MAAM,KAAK,IAAI;AAAA,UAC7B;AAAA,UACA,EAAE,QAAQ,OAAO,aAAa,WAAW;AAAA,UACzC,OAAO;AAAA,YACL,aAAa,MAAM;AAAA,YACnB,MAAM,MAAM;AAAA,YACZ,KAAK,MAAM;AAAA,YACX,SAAS,MAAM,UAAU;AAAA,YACzB,QAAQ,MAAM;AAAA,YACd,YAAY;AAAA,UACd;AAAA,QACF;AACA,YAAI,QAAS,QAAO,SAAS,OAAO;AAAA,MAEtC;AAAA,IAIF;AAEA,UAAM,SAAS,MAAM,KAAK,OAAO,EAAE,OAAO,QAAQ,OAAO,IAAI,CAAC;AAC9D,QAAI,YAAY;AACd,YAAM,KAAK,IAAI,WAAW,oBAAoB,EAAE,QAAQ,OAAO,aAAa,WAAW,CAAC;AAAA,IAC1F;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKkC;AAChC,UAAM,iBAAiB,MACrB,KAAK,IAAI;AAAA,MACP;AAAA,MACA,EAAE,QAAQ,OAAO,aAAa,MAAM,WAAW;AAAA,MAC/C,OAAO;AAAA,QACL,MAAM,MAAM;AAAA,QACZ,KAAK,MAAM;AAAA,QACX,SAAS,MAAM,UAAU;AAAA,QACzB,QAAQ,MAAM;AAAA,QACd,YAAY;AAAA,MACd;AAAA,IACF;AAEF,UAAM,UAAU,MAAM,eAAe;AACrC,QAAI,QAAS,QAAO,SAAS,OAAO;AAEpC,QAAI;AACF,YAAM,MAAM,MAAM,KAAK,IAAI,UAA+B,oBAAoB;AAAA,QAC5E,QAAQ;AAAA,QACR,YAAY;AAAA,QACZ,aAAa,MAAM;AAAA,QACnB,MAAM,MAAM;AAAA,QACZ,KAAK,MAAM;AAAA,QACX,SAAS,MAAM,UAAU;AAAA,QACzB,QAAQ,MAAM;AAAA,QACd,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AACD,aAAO,SAAS,GAAG;AAAA,IACrB,SAAS,OAAO;AACd,UAAI,EAAE,iBAAiB,sBAAuB,OAAM;AAEpD,YAAM,MAAM,MAAM,eAAe;AACjC,UAAI,CAAC,IAAK,OAAM;AAChB,aAAO,SAAS,GAAG;AAAA,IACrB;AAAA,EACF;AAAA,EAEA,MAAM,KAAK,EAAE,MAAM,GAAuD;AACxE,UAAM,OAAO,MAAM,KAAK,IAAI;AAAA,MAC1B;AAAA,MACA,EAAE,QAAQ,MAAM;AAAA,MAChB,EAAE,SAAS,CAAC,CAAC,QAAQ,KAAK,CAAC,EAAE;AAAA,IAC/B;AACA,WAAO,KAAK,IAAI,QAAQ;AAAA,EAC1B;AAAA,EAEA,MAAM,OAAO,EAAE,OAAO,WAAW,GAA4D;AAC3F,WAAQ,MAAM,KAAK,IAAI,WAAW,oBAAoB,EAAE,QAAQ,OAAO,aAAa,WAAW,CAAC,IAAK;AAAA,EACvG;AACF;","names":[]}

package/dist/storage/domains/memory-settings/base.d.ts

@@ -0,0 +1,62 @@
+import { FactoryStorageDomain } from '@mastra/core/storage';
+import type { CollectionSchema } from '@mastra/core/storage';
+/**
+ * A user's persisted memory (observational-memory) configuration. The
+ * DB-backed tenant counterpart of the local `settings.json` OM fields
+ * (`observerModelOverride`, `reflectorModelOverride`, `omObservationThreshold`,
+ * `omReflectionThreshold`, `omObserveAttachments`): one row per `(org, user)`,
+ * every knob nullable so only explicitly-changed values are stored.
+ */
+export interface MemorySettingsRecord {
+    orgId: string;
+    userId: string;
+    observerModelId: string | null;
+    reflectorModelId: string | null;
+    observationThreshold: number | null;
+    reflectionThreshold: number | null;
+    observeAttachments: 'auto' | boolean | null;
+    createdAt: Date;
+    updatedAt: Date;
+}
+/** Partial update — only the provided knobs are written. */
+export interface MemorySettingsPatch {
+    observerModelId?: string;
+    reflectorModelId?: string;
+    observationThreshold?: number;
+    reflectionThreshold?: number;
+    observeAttachments?: 'auto' | boolean;
+}
+/**
+ * Knobs written only when the stored value is still unset (`NULL`). The check
+ * happens inside the backend's atomic-update primitive, so a concurrent
+ * explicit write to the same knob is never clobbered by a fill.
+ */
+export interface MemorySettingsFillIfUnset {
+    observerModelId?: string;
+    reflectorModelId?: string;
+}
+export declare const MEMORY_SETTINGS_SCHEMA: CollectionSchema;
+export declare class MemorySettingsStorage extends FactoryStorageDomain {
+    #private;
+    constructor();
+    init(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
+    get({ orgId, userId }: {
+        orgId: string;
+        userId: string;
+    }): Promise<MemorySettingsRecord | null>;
+    /**
+     * Upsert the user's row, writing only the knobs present in `patch`.
+     * `fillIfUnset` knobs are written only where the stored value is still
+     * `NULL`, decided inside the atomic update so concurrent explicit writes
+     * win over fills. Concurrent first writes are resolved via the shared
+     * insert-then-catch-unique-violation pattern (see `queue_health`).
+     */
+    patch({ orgId, userId, patch, fillIfUnset, }: {
+        orgId: string;
+        userId: string;
+        patch: MemorySettingsPatch;
+        fillIfUnset?: MemorySettingsFillIfUnset;
+    }): Promise<MemorySettingsRecord>;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/storage/domains/memory-settings/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory-settings/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAwB,MAAM,sBAAsB,CAAC;AAClF,OAAO,KAAK,EAAE,gBAAgB,EAAqB,MAAM,sBAAsB,CAAC;AAEhF;;;;;;GAMG;AACH,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,mBAAmB,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,kBAAkB,EAAE,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC;IAC5C,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,4DAA4D;AAC5D,MAAM,WAAW,mBAAmB;IAClC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACvC;AAED;;;;GAIG;AACH,MAAM,WAAW,yBAAyB;IACxC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,eAAO,MAAM,sBAAsB,EAAE,gBAgBpC,CAAC;AAuCF,qBAAa,qBAAsB,SAAQ,oBAAoB;;;IAKvD,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAIrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAQpC,GAAG,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC;IAKrG;;;;;;OAMG;IACG,KAAK,CAAC,EACV,KAAK,EACL,MAAM,EACN,KAAK,EACL,WAAW,GACZ,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,mBAAmB,CAAC;QAC3B,WAAW,CAAC,EAAE,yBAAyB,CAAC;KACzC,GAAG,OAAO,CAAC,oBAAoB,CAAC;CAuClC"}

package/dist/storage/domains/memory-settings/base.js

@@ -0,0 +1,111 @@
+// src/storage/domains/memory-settings/base.ts
+import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
+var MEMORY_SETTINGS_SCHEMA = {
+  name: "memory_settings",
+  columns: {
+    id: { type: "uuid-pk" },
+    org_id: { type: "text" },
+    user_id: { type: "text" },
+    observer_model_id: { type: "text", nullable: true },
+    reflector_model_id: { type: "text", nullable: true },
+    observation_threshold: { type: "integer", nullable: true },
+    reflection_threshold: { type: "integer", nullable: true },
+    // 'auto' | true | false — json keeps the tri-state without string encoding.
+    observe_attachments: { type: "json", nullable: true },
+    created_at: { type: "timestamp" },
+    updated_at: { type: "timestamp" }
+  },
+  uniqueIndexes: [{ name: "memory_settings_org_user_key", columns: ["org_id", "user_id"] }]
+};
+function toRecord(row) {
+  return {
+    orgId: row.org_id,
+    userId: row.user_id,
+    observerModelId: row.observer_model_id,
+    reflectorModelId: row.reflector_model_id,
+    observationThreshold: row.observation_threshold,
+    reflectionThreshold: row.reflection_threshold,
+    observeAttachments: row.observe_attachments,
+    createdAt: row.created_at,
+    updatedAt: row.updated_at
+  };
+}
+function patchToColumns(patch) {
+  const columns = {};
+  if (patch.observerModelId !== void 0) columns.observer_model_id = patch.observerModelId;
+  if (patch.reflectorModelId !== void 0) columns.reflector_model_id = patch.reflectorModelId;
+  if (patch.observationThreshold !== void 0) columns.observation_threshold = patch.observationThreshold;
+  if (patch.reflectionThreshold !== void 0) columns.reflection_threshold = patch.reflectionThreshold;
+  if (patch.observeAttachments !== void 0) columns.observe_attachments = patch.observeAttachments;
+  return columns;
+}
+var MemorySettingsStorage = class extends FactoryStorageDomain {
+  constructor() {
+    super("memory-settings");
+  }
+  async init() {
+    await this.ensureCollections([MEMORY_SETTINGS_SCHEMA]);
+  }
+  async dangerouslyClearAll() {
+    await this.ops.deleteMany("memory_settings", {});
+  }
+  get #db() {
+    return this.ops;
+  }
+  async get({ orgId, userId }) {
+    const row = await this.#db.findOne("memory_settings", { org_id: orgId, user_id: userId });
+    return row ? toRecord(row) : null;
+  }
+  /**
+   * Upsert the user's row, writing only the knobs present in `patch`.
+   * `fillIfUnset` knobs are written only where the stored value is still
+   * `NULL`, decided inside the atomic update so concurrent explicit writes
+   * win over fills. Concurrent first writes are resolved via the shared
+   * insert-then-catch-unique-violation pattern (see `queue_health`).
+   */
+  async patch({
+    orgId,
+    userId,
+    patch,
+    fillIfUnset
+  }) {
+    const now = /* @__PURE__ */ new Date();
+    const updateExisting = () => this.#db.updateAtomic("memory_settings", { org_id: orgId, user_id: userId }, (row) => {
+      const columns = { ...patchToColumns(patch), updated_at: now };
+      if (fillIfUnset?.observerModelId !== void 0 && row.observer_model_id == null) {
+        columns.observer_model_id = patch.observerModelId ?? fillIfUnset.observerModelId;
+      }
+      if (fillIfUnset?.reflectorModelId !== void 0 && row.reflector_model_id == null) {
+        columns.reflector_model_id = patch.reflectorModelId ?? fillIfUnset.reflectorModelId;
+      }
+      return columns;
+    });
+    const updated = await updateExisting();
+    if (updated) return toRecord(updated);
+    try {
+      const row = await this.#db.insertOne("memory_settings", {
+        org_id: orgId,
+        user_id: userId,
+        observer_model_id: fillIfUnset?.observerModelId ?? null,
+        reflector_model_id: fillIfUnset?.reflectorModelId ?? null,
+        observation_threshold: null,
+        reflection_threshold: null,
+        observe_attachments: null,
+        ...patchToColumns(patch),
+        created_at: now,
+        updated_at: now
+      });
+      return toRecord(row);
+    } catch (error) {
+      if (!(error instanceof UniqueViolationError)) throw error;
+      const row = await updateExisting();
+      if (!row) throw error;
+      return toRecord(row);
+    }
+  }
+};
+export {
+  MEMORY_SETTINGS_SCHEMA,
+  MemorySettingsStorage
+};
+//# sourceMappingURL=base.js.map

package/dist/storage/domains/memory-settings/base.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/storage/domains/memory-settings/base.ts"],"sourcesContent":["import { FactoryStorageDomain, UniqueViolationError } from '@mastra/core/storage';\nimport type { CollectionSchema, FactoryStorageOps } from '@mastra/core/storage';\n\n/**\n * A user's persisted memory (observational-memory) configuration. The\n * DB-backed tenant counterpart of the local `settings.json` OM fields\n * (`observerModelOverride`, `reflectorModelOverride`, `omObservationThreshold`,\n * `omReflectionThreshold`, `omObserveAttachments`): one row per `(org, user)`,\n * every knob nullable so only explicitly-changed values are stored.\n */\nexport interface MemorySettingsRecord {\n  orgId: string;\n  userId: string;\n  observerModelId: string | null;\n  reflectorModelId: string | null;\n  observationThreshold: number | null;\n  reflectionThreshold: number | null;\n  observeAttachments: 'auto' | boolean | null;\n  createdAt: Date;\n  updatedAt: Date;\n}\n\n/** Partial update — only the provided knobs are written. */\nexport interface MemorySettingsPatch {\n  observerModelId?: string;\n  reflectorModelId?: string;\n  observationThreshold?: number;\n  reflectionThreshold?: number;\n  observeAttachments?: 'auto' | boolean;\n}\n\n/**\n * Knobs written only when the stored value is still unset (`NULL`). The check\n * happens inside the backend's atomic-update primitive, so a concurrent\n * explicit write to the same knob is never clobbered by a fill.\n */\nexport interface MemorySettingsFillIfUnset {\n  observerModelId?: string;\n  reflectorModelId?: string;\n}\n\nexport const MEMORY_SETTINGS_SCHEMA: CollectionSchema = {\n  name: 'memory_settings',\n  columns: {\n    id: { type: 'uuid-pk' },\n    org_id: { type: 'text' },\n    user_id: { type: 'text' },\n    observer_model_id: { type: 'text', nullable: true },\n    reflector_model_id: { type: 'text', nullable: true },\n    observation_threshold: { type: 'integer', nullable: true },\n    reflection_threshold: { type: 'integer', nullable: true },\n    // 'auto' | true | false — json keeps the tri-state without string encoding.\n    observe_attachments: { type: 'json', nullable: true },\n    created_at: { type: 'timestamp' },\n    updated_at: { type: 'timestamp' },\n  },\n  uniqueIndexes: [{ name: 'memory_settings_org_user_key', columns: ['org_id', 'user_id'] }],\n};\n\ninterface MemorySettingsDbRow extends Record<string, unknown> {\n  id: string;\n  org_id: string;\n  user_id: string;\n  observer_model_id: string | null;\n  reflector_model_id: string | null;\n  observation_threshold: number | null;\n  reflection_threshold: number | null;\n  observe_attachments: 'auto' | boolean | null;\n  created_at: Date;\n  updated_at: Date;\n}\n\nfunction toRecord(row: MemorySettingsDbRow): MemorySettingsRecord {\n  return {\n    orgId: row.org_id,\n    userId: row.user_id,\n    observerModelId: row.observer_model_id,\n    reflectorModelId: row.reflector_model_id,\n    observationThreshold: row.observation_threshold,\n    reflectionThreshold: row.reflection_threshold,\n    observeAttachments: row.observe_attachments,\n    createdAt: row.created_at,\n    updatedAt: row.updated_at,\n  };\n}\n\nfunction patchToColumns(patch: MemorySettingsPatch): Partial<MemorySettingsDbRow> {\n  const columns: Partial<MemorySettingsDbRow> = {};\n  if (patch.observerModelId !== undefined) columns.observer_model_id = patch.observerModelId;\n  if (patch.reflectorModelId !== undefined) columns.reflector_model_id = patch.reflectorModelId;\n  if (patch.observationThreshold !== undefined) columns.observation_threshold = patch.observationThreshold;\n  if (patch.reflectionThreshold !== undefined) columns.reflection_threshold = patch.reflectionThreshold;\n  if (patch.observeAttachments !== undefined) columns.observe_attachments = patch.observeAttachments;\n  return columns;\n}\n\nexport class MemorySettingsStorage extends FactoryStorageDomain {\n  constructor() {\n    super('memory-settings');\n  }\n\n  async init(): Promise<void> {\n    await this.ensureCollections([MEMORY_SETTINGS_SCHEMA]);\n  }\n\n  async dangerouslyClearAll(): Promise<void> {\n    await this.ops.deleteMany('memory_settings', {});\n  }\n\n  get #db(): FactoryStorageOps {\n    return this.ops;\n  }\n\n  async get({ orgId, userId }: { orgId: string; userId: string }): Promise<MemorySettingsRecord | null> {\n    const row = await this.#db.findOne<MemorySettingsDbRow>('memory_settings', { org_id: orgId, user_id: userId });\n    return row ? toRecord(row) : null;\n  }\n\n  /**\n   * Upsert the user's row, writing only the knobs present in `patch`.\n   * `fillIfUnset` knobs are written only where the stored value is still\n   * `NULL`, decided inside the atomic update so concurrent explicit writes\n   * win over fills. Concurrent first writes are resolved via the shared\n   * insert-then-catch-unique-violation pattern (see `queue_health`).\n   */\n  async patch({\n    orgId,\n    userId,\n    patch,\n    fillIfUnset,\n  }: {\n    orgId: string;\n    userId: string;\n    patch: MemorySettingsPatch;\n    fillIfUnset?: MemorySettingsFillIfUnset;\n  }): Promise<MemorySettingsRecord> {\n    const now = new Date();\n    const updateExisting = () =>\n      this.#db.updateAtomic<MemorySettingsDbRow>('memory_settings', { org_id: orgId, user_id: userId }, row => {\n        const columns: Partial<MemorySettingsDbRow> = { ...patchToColumns(patch), updated_at: now };\n        if (fillIfUnset?.observerModelId !== undefined && row.observer_model_id == null) {\n          columns.observer_model_id = patch.observerModelId ?? fillIfUnset.observerModelId;\n        }\n        if (fillIfUnset?.reflectorModelId !== undefined && row.reflector_model_id == null) {\n          columns.reflector_model_id = patch.reflectorModelId ?? fillIfUnset.reflectorModelId;\n        }\n        return columns;\n      });\n\n    const updated = await updateExisting();\n    if (updated) return toRecord(updated);\n\n    try {\n      const row = await this.#db.insertOne<MemorySettingsDbRow>('memory_settings', {\n        org_id: orgId,\n        user_id: userId,\n        observer_model_id: fillIfUnset?.observerModelId ?? null,\n        reflector_model_id: fillIfUnset?.reflectorModelId ?? null,\n        observation_threshold: null,\n        reflection_threshold: null,\n        observe_attachments: null,\n        ...patchToColumns(patch),\n        created_at: now,\n        updated_at: now,\n      });\n      return toRecord(row);\n    } catch (error) {\n      if (!(error instanceof UniqueViolationError)) throw error;\n      // Lost the first-write race — apply the patch to the winning row.\n      const row = await updateExisting();\n      if (!row) throw error;\n      return toRecord(row);\n    }\n  }\n}\n"],"mappings":";AAAA,SAAS,sBAAsB,4BAA4B;AAyCpD,IAAM,yBAA2C;AAAA,EACtD,MAAM;AAAA,EACN,SAAS;AAAA,IACP,IAAI,EAAE,MAAM,UAAU;AAAA,IACtB,QAAQ,EAAE,MAAM,OAAO;AAAA,IACvB,SAAS,EAAE,MAAM,OAAO;AAAA,IACxB,mBAAmB,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IAClD,oBAAoB,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACnD,uBAAuB,EAAE,MAAM,WAAW,UAAU,KAAK;AAAA,IACzD,sBAAsB,EAAE,MAAM,WAAW,UAAU,KAAK;AAAA;AAAA,IAExD,qBAAqB,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACpD,YAAY,EAAE,MAAM,YAAY;AAAA,IAChC,YAAY,EAAE,MAAM,YAAY;AAAA,EAClC;AAAA,EACA,eAAe,CAAC,EAAE,MAAM,gCAAgC,SAAS,CAAC,UAAU,SAAS,EAAE,CAAC;AAC1F;AAeA,SAAS,SAAS,KAAgD;AAChE,SAAO;AAAA,IACL,OAAO,IAAI;AAAA,IACX,QAAQ,IAAI;AAAA,IACZ,iBAAiB,IAAI;AAAA,IACrB,kBAAkB,IAAI;AAAA,IACtB,sBAAsB,IAAI;AAAA,IAC1B,qBAAqB,IAAI;AAAA,IACzB,oBAAoB,IAAI;AAAA,IACxB,WAAW,IAAI;AAAA,IACf,WAAW,IAAI;AAAA,EACjB;AACF;AAEA,SAAS,eAAe,OAA0D;AAChF,QAAM,UAAwC,CAAC;AAC/C,MAAI,MAAM,oBAAoB,OAAW,SAAQ,oBAAoB,MAAM;AAC3E,MAAI,MAAM,qBAAqB,OAAW,SAAQ,qBAAqB,MAAM;AAC7E,MAAI,MAAM,yBAAyB,OAAW,SAAQ,wBAAwB,MAAM;AACpF,MAAI,MAAM,wBAAwB,OAAW,SAAQ,uBAAuB,MAAM;AAClF,MAAI,MAAM,uBAAuB,OAAW,SAAQ,sBAAsB,MAAM;AAChF,SAAO;AACT;AAEO,IAAM,wBAAN,cAAoC,qBAAqB;AAAA,EAC9D,cAAc;AACZ,UAAM,iBAAiB;AAAA,EACzB;AAAA,EAEA,MAAM,OAAsB;AAC1B,UAAM,KAAK,kBAAkB,CAAC,sBAAsB,CAAC;AAAA,EACvD;AAAA,EAEA,MAAM,sBAAqC;AACzC,UAAM,KAAK,IAAI,WAAW,mBAAmB,CAAC,CAAC;AAAA,EACjD;AAAA,EAEA,IAAI,MAAyB;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAM,IAAI,EAAE,OAAO,OAAO,GAA4E;AACpG,UAAM,MAAM,MAAM,KAAK,IAAI,QAA6B,mBAAmB,EAAE,QAAQ,OAAO,SAAS,OAAO,CAAC;AAC7G,WAAO,MAAM,SAAS,GAAG,IAAI;AAAA,EAC/B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,MAAM;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKkC;AAChC,UAAM,MAAM,oBAAI,KAAK;AACrB,UAAM,iBAAiB,MACrB,KAAK,IAAI,aAAkC,mBAAmB,EAAE,QAAQ,OAAO,SAAS,OAAO,GAAG,SAAO;AACvG,YAAM,UAAwC,EAAE,GAAG,eAAe,KAAK,GAAG,YAAY,IAAI;AAC1F,UAAI,aAAa,oBAAoB,UAAa,IAAI,qBAAqB,MAAM;AAC/E,gBAAQ,oBAAoB,MAAM,mBAAmB,YAAY;AAAA,MACnE;AACA,UAAI,aAAa,qBAAqB,UAAa,IAAI,sBAAsB,MAAM;AACjF,gBAAQ,qBAAqB,MAAM,oBAAoB,YAAY;AAAA,MACrE;AACA,aAAO;AAAA,IACT,CAAC;AAEH,UAAM,UAAU,MAAM,eAAe;AACrC,QAAI,QAAS,QAAO,SAAS,OAAO;AAEpC,QAAI;AACF,YAAM,MAAM,MAAM,KAAK,IAAI,UAA+B,mBAAmB;AAAA,QAC3E,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,mBAAmB,aAAa,mBAAmB;AAAA,QACnD,oBAAoB,aAAa,oBAAoB;AAAA,QACrD,uBAAuB;AAAA,QACvB,sBAAsB;AAAA,QACtB,qBAAqB;AAAA,QACrB,GAAG,eAAe,KAAK;AAAA,QACvB,YAAY;AAAA,QACZ,YAAY;AAAA,MACd,CAAC;AACD,aAAO,SAAS,GAAG;AAAA,IACrB,SAAS,OAAO;AACd,UAAI,EAAE,iBAAiB,sBAAuB,OAAM;AAEpD,YAAM,MAAM,MAAM,eAAe;AACjC,UAAI,CAAC,IAAK,OAAM;AAChB,aAAO,SAAS,GAAG;AAAA,IACrB;AAAA,EACF;AACF;","names":[]}

package/dist/workspace.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"workspace.d.ts","sourceRoot":"","sources":["../src/workspace.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,iBAAiB,EAAE,MAAM,4CAA4C,CAAC;AAE/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAKxE,OAAO,EAAE,YAAY,EAAoB,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAInF,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,cAAc,CAAC;AAC/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAE9E,OAAO,KAAK,EAAuB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAK5E,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAElE;AA+ED,KAAK,uBAAuB,GAAG,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;AAEzE,MAAM,WAAW,6BAA6B;IAC5C,wEAAwE;IACxE,OAAO,CAAC,EAAE,0BAA0B,CAAC;IACrC,gFAAgF;IAChF,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,0EAA0E;IAC1E,KAAK,CAAC,EAAE,YAAY,CAAC;CACtB;AAED,wBAAgB,sBAAsB,CAAC,OAAO,GAAE,6BAAkC,IAIlE,4CAA4C,uBAAuB,wVA0GlF;AAED,eAAO,MAAM,mBAAmB,+CA5G4B,uBAAuB,uVA4GxB,CAAC"}
+{"version":3,"file":"workspace.d.ts","sourceRoot":"","sources":["../src/workspace.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,iBAAiB,EAAE,MAAM,4CAA4C,CAAC;AAE/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAKxE,OAAO,EAAE,YAAY,EAAoB,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAInF,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,cAAc,CAAC;AAC/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAG9E,OAAO,KAAK,EAAuB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAK5E,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAElE;AA+ED,KAAK,uBAAuB,GAAG,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;AAEzE,MAAM,WAAW,6BAA6B;IAC5C,wEAAwE;IACxE,OAAO,CAAC,EAAE,0BAA0B,CAAC;IACrC,gFAAgF;IAChF,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,0EAA0E;IAC1E,KAAK,CAAC,EAAE,YAAY,CAAC;CACtB;AAED,wBAAgB,sBAAsB,CAAC,OAAO,GAAE,6BAAkC,IAKlE,4CAA4C,uBAAuB,wVAqHlF;AAED,eAAO,MAAM,mBAAmB,+CAvH4B,uBAAuB,uVAuHxB,CAAC"}

package/dist/workspace.js

@@ -193,6 +193,12 @@ async function runWorktreeSetup(sandbox, worktreePath, command) {
   }
 }
 
+// src/integrations/github/token-refresh.ts
+var GITHUB_TOKEN_INJECTOR_CONTEXT_KEY = "factoryGithubTokenInjector";
+function registerGithubTokenInjector(requestContext, injector) {
+  requestContext.set(GITHUB_TOKEN_INJECTOR_CONTEXT_KEY, injector);
+}
+
 // src/workspace.ts
 var WORKSPACE_ID_PREFIX = "mfw";
 var SESSION_CHECKPOINT_PREFIX = "mastracode-session";
@@ -260,6 +266,7 @@ var factorySkillExtension = {
 function createWorkspaceFactory(options = {}) {
   const { sandbox: sandboxConfig, github, fleet } = options;
   const isLocalSandbox = sandboxConfig?.machine instanceof LocalSandbox;
+  const githubTokenInjectors = /* @__PURE__ */ new Map();
   return async ({ requestContext, mastra, skillExtension }) => {
     const effectiveSkillExtension = skillExtension ?? factorySkillExtension;
     const ctx = requestContext.get("controller");
@@ -311,6 +318,8 @@ function createWorkspaceFactory(options = {}) {
       const existing = mastra?.getWorkspaceById(workspaceId);
       if (existing) {
         existing.setToolsConfig(MASTRACODE_WORKSPACE_TOOLS);
+        const injectGithubToken2 = githubTokenInjectors.get(workspaceId);
+        if (injectGithubToken2) registerGithubTokenInjector(requestContext, injectGithubToken2);
         return existing;
       }
     } catch {
@@ -341,6 +350,14 @@ function createWorkspaceFactory(options = {}) {
       repoFullName
     });
     if (projectRepository.setupCommand) await runWorktreeSetup(sandbox, workdir, projectRepository.setupCommand);
+    const injectGithubToken = (freshToken) => {
+      if (!sandbox.setEnvironmentVariable) {
+        throw new Error("The active sandbox provider does not support runtime GitHub token refresh.");
+      }
+      sandbox.setEnvironmentVariable("GH_TOKEN", freshToken);
+    };
+    githubTokenInjectors.set(workspaceId, injectGithubToken);
+    registerGithubTokenInjector(requestContext, injectGithubToken);
     const filesystem = new SandboxFilesystem({ sandbox, workdir });
     const projectSkillPaths = [path2.join(configDir, "skills"), ".claude/skills", ".agents/skills"];
     const skillPaths = [...effectiveSkillExtension?.paths ?? [], ...projectSkillPaths];