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

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];