@mastra/factory 0.2.0-alpha.0 → 0.2.1-alpha.0

package/dist/routes/surface.js

@@ -18882,46 +18882,49 @@ Note: This migration may take some time for large tables.
             config: input.config,
             observedTimezone: input.observedTimezone
           };
-          await this.#client.execute({
-            sql: `INSERT INTO "${OM_TABLE}" (
-          id, "lookupKey", scope, "resourceId", "threadId",
-          "activeObservations", "activeObservationsPendingUpdate",
-          "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
-          "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
-          "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection", "lastBufferedAtTokens", "lastBufferedAtTime",
-          "observedTimezone", "createdAt", "updatedAt"
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
-            args: [
-              id,
-              lookupKey,
-              input.scope,
-              input.resourceId,
-              input.threadId || null,
-              "",
-              null,
-              "initial",
-              JSON.stringify(input.config),
-              0,
-              null,
-              null,
-              0,
-              0,
-              0,
-              false,
-              false,
-              false,
-              // isBufferingObservation
-              false,
-              // isBufferingReflection
-              0,
-              // lastBufferedAtTokens
-              null,
-              // lastBufferedAtTime
-              input.observedTimezone || null,
-              now.toISOString(),
-              now.toISOString()
-            ]
-          });
+          await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `INSERT INTO "${OM_TABLE}" (
+            id, "lookupKey", scope, "resourceId", "threadId",
+            "activeObservations", "activeObservationsPendingUpdate",
+            "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
+            "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
+            "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection", "lastBufferedAtTokens", "lastBufferedAtTime",
+            "observedTimezone", "createdAt", "updatedAt"
+          ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+              args: [
+                id,
+                lookupKey,
+                input.scope,
+                input.resourceId,
+                input.threadId || null,
+                "",
+                null,
+                "initial",
+                JSON.stringify(input.config),
+                0,
+                null,
+                null,
+                0,
+                0,
+                0,
+                false,
+                false,
+                false,
+                // isBufferingObservation
+                false,
+                // isBufferingReflection
+                0,
+                // lastBufferedAtTokens
+                null,
+                // lastBufferedAtTime
+                input.observedTimezone || null,
+                now.toISOString(),
+                now.toISOString()
+              ]
+            })
+          );
           return record;
         } catch (error) {
           throw new MastraError(
@@ -18938,53 +18941,56 @@ Note: This migration may take some time for large tables.
       async insertObservationalMemoryRecord(record) {
         try {
           const lookupKey = this.getOMKey(record.threadId, record.resourceId);
-          await this.#client.execute({
-            sql: `INSERT INTO "${OM_TABLE}" (
-          id, "lookupKey", scope, "resourceId", "threadId",
-          "activeObservations", "activeObservationsPendingUpdate",
-          "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
-          "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
-          "observedMessageIds", "bufferedObservationChunks",
-          "bufferedReflection", "bufferedReflectionTokens", "bufferedReflectionInputTokens",
-          "reflectedObservationLineCount",
-          "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection",
-          "lastBufferedAtTokens", "lastBufferedAtTime",
-          "observedTimezone", metadata, "createdAt", "updatedAt"
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
-            args: [
-              record.id,
-              lookupKey,
-              record.scope,
-              record.resourceId,
-              record.threadId || null,
-              record.activeObservations || "",
-              null,
-              record.originType || "initial",
-              record.config ? JSON.stringify(record.config) : null,
-              record.generationCount || 0,
-              record.lastObservedAt ? record.lastObservedAt.toISOString() : null,
-              null,
-              record.pendingMessageTokens || 0,
-              record.totalTokensObserved || 0,
-              record.observationTokenCount || 0,
-              record.observedMessageIds ? JSON.stringify(record.observedMessageIds) : null,
-              record.bufferedObservationChunks ? JSON.stringify(record.bufferedObservationChunks) : null,
-              record.bufferedReflection || null,
-              record.bufferedReflectionTokens ?? null,
-              record.bufferedReflectionInputTokens ?? null,
-              record.reflectedObservationLineCount ?? null,
-              record.isObserving || false,
-              record.isReflecting || false,
-              record.isBufferingObservation || false,
-              record.isBufferingReflection || false,
-              record.lastBufferedAtTokens || 0,
-              record.lastBufferedAtTime ? record.lastBufferedAtTime.toISOString() : null,
-              record.observedTimezone || null,
-              record.metadata ? JSON.stringify(record.metadata) : null,
-              record.createdAt.toISOString(),
-              record.updatedAt.toISOString()
-            ]
-          });
+          await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `INSERT INTO "${OM_TABLE}" (
+            id, "lookupKey", scope, "resourceId", "threadId",
+            "activeObservations", "activeObservationsPendingUpdate",
+            "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
+            "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
+            "observedMessageIds", "bufferedObservationChunks",
+            "bufferedReflection", "bufferedReflectionTokens", "bufferedReflectionInputTokens",
+            "reflectedObservationLineCount",
+            "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection",
+            "lastBufferedAtTokens", "lastBufferedAtTime",
+            "observedTimezone", metadata, "createdAt", "updatedAt"
+          ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+              args: [
+                record.id,
+                lookupKey,
+                record.scope,
+                record.resourceId,
+                record.threadId || null,
+                record.activeObservations || "",
+                null,
+                record.originType || "initial",
+                record.config ? JSON.stringify(record.config) : null,
+                record.generationCount || 0,
+                record.lastObservedAt ? record.lastObservedAt.toISOString() : null,
+                null,
+                record.pendingMessageTokens || 0,
+                record.totalTokensObserved || 0,
+                record.observationTokenCount || 0,
+                record.observedMessageIds ? JSON.stringify(record.observedMessageIds) : null,
+                record.bufferedObservationChunks ? JSON.stringify(record.bufferedObservationChunks) : null,
+                record.bufferedReflection || null,
+                record.bufferedReflectionTokens ?? null,
+                record.bufferedReflectionInputTokens ?? null,
+                record.reflectedObservationLineCount ?? null,
+                record.isObserving || false,
+                record.isReflecting || false,
+                record.isBufferingObservation || false,
+                record.isBufferingReflection || false,
+                record.lastBufferedAtTokens || 0,
+                record.lastBufferedAtTime ? record.lastBufferedAtTime.toISOString() : null,
+                record.observedTimezone || null,
+                record.metadata ? JSON.stringify(record.metadata) : null,
+                record.createdAt.toISOString(),
+                record.updatedAt.toISOString()
+              ]
+            })
+          );
         } catch (error) {
           throw new MastraError(
             {
@@ -19001,26 +19007,29 @@ Note: This migration may take some time for large tables.
         try {
           const now = /* @__PURE__ */ new Date();
           const observedMessageIdsJson = input.observedMessageIds ? JSON.stringify(input.observedMessageIds) : null;
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET
-          "activeObservations" = ?,
-          "lastObservedAt" = ?,
-          "pendingMessageTokens" = 0,
-          "observationTokenCount" = ?,
-          "totalTokensObserved" = "totalTokensObserved" + ?,
-          "observedMessageIds" = ?,
-          "updatedAt" = ?
-        WHERE id = ?`,
-            args: [
-              input.observations,
-              input.lastObservedAt.toISOString(),
-              input.tokenCount,
-              input.tokenCount,
-              observedMessageIdsJson,
-              now.toISOString(),
-              input.id
-            ]
-          });
+          const result = await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `UPDATE "${OM_TABLE}" SET
+            "activeObservations" = ?,
+            "lastObservedAt" = ?,
+            "pendingMessageTokens" = 0,
+            "observationTokenCount" = ?,
+            "totalTokensObserved" = "totalTokensObserved" + ?,
+            "observedMessageIds" = ?,
+            "updatedAt" = ?
+          WHERE id = ?`,
+              args: [
+                input.observations,
+                input.lastObservedAt.toISOString(),
+                input.tokenCount,
+                input.tokenCount,
+                observedMessageIdsJson,
+                now.toISOString(),
+                input.id
+              ]
+            })
+          );
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "UPDATE_ACTIVE_OBSERVATIONS", "NOT_FOUND"),
@@ -19047,78 +19056,21 @@ Note: This migration may take some time for large tables.
       }
       async createReflectionGeneration(input) {
         try {
-          const id = crypto.randomUUID();
-          const now = /* @__PURE__ */ new Date();
-          const lookupKey = this.getOMKey(input.currentRecord.threadId, input.currentRecord.resourceId);
-          const record = {
-            id,
-            scope: input.currentRecord.scope,
-            threadId: input.currentRecord.threadId,
-            resourceId: input.currentRecord.resourceId,
-            createdAt: now,
-            updatedAt: now,
-            lastObservedAt: input.currentRecord.lastObservedAt,
-            originType: "reflection",
-            generationCount: input.currentRecord.generationCount + 1,
-            activeObservations: input.reflection,
-            totalTokensObserved: input.currentRecord.totalTokensObserved,
-            observationTokenCount: input.tokenCount,
-            pendingMessageTokens: 0,
-            isReflecting: false,
-            isObserving: false,
-            isBufferingObservation: false,
-            isBufferingReflection: false,
-            lastBufferedAtTokens: 0,
-            lastBufferedAtTime: null,
-            config: input.currentRecord.config,
-            metadata: input.currentRecord.metadata,
-            observedTimezone: input.currentRecord.observedTimezone
-          };
-          await this.#client.execute({
-            sql: `INSERT INTO "${OM_TABLE}" (
-          id, "lookupKey", scope, "resourceId", "threadId",
-          "activeObservations", "activeObservationsPendingUpdate",
-          "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
-          "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
-          "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection", "lastBufferedAtTokens", "lastBufferedAtTime",
-          "observedTimezone", metadata, "createdAt", "updatedAt"
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
-            args: [
-              id,
-              lookupKey,
-              record.scope,
-              record.resourceId,
-              record.threadId || null,
-              input.reflection,
-              null,
-              "reflection",
-              JSON.stringify(record.config),
-              input.currentRecord.generationCount + 1,
-              record.lastObservedAt?.toISOString() || null,
-              now.toISOString(),
-              record.pendingMessageTokens,
-              record.totalTokensObserved,
-              record.observationTokenCount,
-              false,
-              // isObserving
-              false,
-              // isReflecting
-              false,
-              // isBufferingObservation
-              false,
-              // isBufferingReflection
-              0,
-              // lastBufferedAtTokens
-              null,
-              // lastBufferedAtTime
-              record.observedTimezone || null,
-              record.metadata ? JSON.stringify(record.metadata) : null,
-              now.toISOString(),
-              now.toISOString()
-            ]
+          return await withClientWriteLock(this.#client, async () => {
+            const tx = await this.#client.transaction("write");
+            try {
+              const record = await this.#insertReflectionGeneration(tx, input);
+              await tx.commit();
+              return record;
+            } catch (error) {
+              if (!tx.closed) await tx.rollback();
+              throw error;
+            }
           });
-          return record;
         } catch (error) {
+          if (error instanceof MastraError) {
+            throw error;
+          }
           throw new MastraError(
             {
               id: createStorageErrorId("LIBSQL", "CREATE_REFLECTION_GENERATION", "FAILED"),
@@ -19130,12 +19082,94 @@ Note: This migration may take some time for large tables.
           );
         }
       }
+      /**
+       * Inserts a new reflection-generation row. Called inside a locked write
+       * transaction — either from {@link createReflectionGeneration} (standalone)
+       * or from {@link swapBufferedReflectionToActive} (as part of a compound
+       * swap). Must not acquire the write lock itself; the caller owns it.
+       */
+      async #insertReflectionGeneration(tx, input) {
+        const id = crypto.randomUUID();
+        const now = /* @__PURE__ */ new Date();
+        const lookupKey = this.getOMKey(input.currentRecord.threadId, input.currentRecord.resourceId);
+        const record = {
+          id,
+          scope: input.currentRecord.scope,
+          threadId: input.currentRecord.threadId,
+          resourceId: input.currentRecord.resourceId,
+          createdAt: now,
+          updatedAt: now,
+          lastObservedAt: input.currentRecord.lastObservedAt,
+          originType: "reflection",
+          generationCount: input.currentRecord.generationCount + 1,
+          activeObservations: input.reflection,
+          totalTokensObserved: input.currentRecord.totalTokensObserved,
+          observationTokenCount: input.tokenCount,
+          pendingMessageTokens: 0,
+          isReflecting: false,
+          isObserving: false,
+          isBufferingObservation: false,
+          isBufferingReflection: false,
+          lastBufferedAtTokens: 0,
+          lastBufferedAtTime: null,
+          config: input.currentRecord.config,
+          metadata: input.currentRecord.metadata,
+          observedTimezone: input.currentRecord.observedTimezone
+        };
+        await tx.execute({
+          sql: `INSERT INTO "${OM_TABLE}" (
+        id, "lookupKey", scope, "resourceId", "threadId",
+        "activeObservations", "activeObservationsPendingUpdate",
+        "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
+        "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
+        "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection", "lastBufferedAtTokens", "lastBufferedAtTime",
+        "observedTimezone", metadata, "createdAt", "updatedAt"
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+          args: [
+            id,
+            lookupKey,
+            record.scope,
+            record.resourceId,
+            record.threadId || null,
+            input.reflection,
+            null,
+            "reflection",
+            JSON.stringify(record.config),
+            input.currentRecord.generationCount + 1,
+            record.lastObservedAt?.toISOString() || null,
+            now.toISOString(),
+            record.pendingMessageTokens,
+            record.totalTokensObserved,
+            record.observationTokenCount,
+            false,
+            // isObserving
+            false,
+            // isReflecting
+            false,
+            // isBufferingObservation
+            false,
+            // isBufferingReflection
+            0,
+            // lastBufferedAtTokens
+            null,
+            // lastBufferedAtTime
+            record.observedTimezone || null,
+            record.metadata ? JSON.stringify(record.metadata) : null,
+            now.toISOString(),
+            now.toISOString()
+          ]
+        });
+        return record;
+      }
       async setReflectingFlag(id, isReflecting) {
         try {
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET "isReflecting" = ?, "updatedAt" = ? WHERE id = ?`,
-            args: [isReflecting, (/* @__PURE__ */ new Date()).toISOString(), id]
-          });
+          const result = await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `UPDATE "${OM_TABLE}" SET "isReflecting" = ?, "updatedAt" = ? WHERE id = ?`,
+              args: [isReflecting, (/* @__PURE__ */ new Date()).toISOString(), id]
+            })
+          );
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "SET_REFLECTING_FLAG", "NOT_FOUND"),
@@ -19162,10 +19196,13 @@ Note: This migration may take some time for large tables.
       }
       async setObservingFlag(id, isObserving) {
         try {
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET "isObserving" = ?, "updatedAt" = ? WHERE id = ?`,
-            args: [isObserving, (/* @__PURE__ */ new Date()).toISOString(), id]
-          });
+          const result = await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `UPDATE "${OM_TABLE}" SET "isObserving" = ?, "updatedAt" = ? WHERE id = ?`,
+              args: [isObserving, (/* @__PURE__ */ new Date()).toISOString(), id]
+            })
+          );
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "SET_OBSERVING_FLAG", "NOT_FOUND"),
@@ -19202,7 +19239,7 @@ Note: This migration may take some time for large tables.
             sql = `UPDATE "${OM_TABLE}" SET "isBufferingObservation" = ?, "updatedAt" = ? WHERE id = ?`;
             args = [isBuffering, nowStr, id];
           }
-          const result = await this.#client.execute({ sql, args });
+          const result = await withClientWriteLock(this.#client, () => this.#client.execute({ sql, args }));
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "SET_BUFFERING_OBSERVATION_FLAG", "NOT_FOUND"),
@@ -19229,10 +19266,13 @@ Note: This migration may take some time for large tables.
       }
       async setBufferingReflectionFlag(id, isBuffering) {
         try {
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET "isBufferingReflection" = ?, "updatedAt" = ? WHERE id = ?`,
-            args: [isBuffering, (/* @__PURE__ */ new Date()).toISOString(), id]
-          });
+          const result = await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `UPDATE "${OM_TABLE}" SET "isBufferingReflection" = ?, "updatedAt" = ? WHERE id = ?`,
+              args: [isBuffering, (/* @__PURE__ */ new Date()).toISOString(), id]
+            })
+          );
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "SET_BUFFERING_REFLECTION_FLAG", "NOT_FOUND"),
@@ -19260,10 +19300,13 @@ Note: This migration may take some time for large tables.
       async clearObservationalMemory(threadId, resourceId) {
         try {
           const lookupKey = this.getOMKey(threadId, resourceId);
-          await this.#client.execute({
-            sql: `DELETE FROM "${OM_TABLE}" WHERE "lookupKey" = ?`,
-            args: [lookupKey]
-          });
+          await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `DELETE FROM "${OM_TABLE}" WHERE "lookupKey" = ?`,
+              args: [lookupKey]
+            })
+          );
         } catch (error) {
           throw new MastraError(
             {
@@ -19278,13 +19321,16 @@ Note: This migration may take some time for large tables.
       }
       async setPendingMessageTokens(id, tokenCount) {
         try {
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET 
-          "pendingMessageTokens" = ?, 
-          "updatedAt" = ? 
-        WHERE id = ?`,
-            args: [tokenCount, (/* @__PURE__ */ new Date()).toISOString(), id]
-          });
+          const result = await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `UPDATE "${OM_TABLE}" SET 
+            "pendingMessageTokens" = ?, 
+            "updatedAt" = ? 
+          WHERE id = ?`,
+              args: [tokenCount, (/* @__PURE__ */ new Date()).toISOString(), id]
+            })
+          );
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "SET_PENDING_MESSAGE_TOKENS", "NOT_FOUND"),
@@ -19311,25 +19357,34 @@ Note: This migration may take some time for large tables.
       }
       async updateObservationalMemoryConfig(input) {
         try {
-          const selectResult = await this.#client.execute({
-            sql: `SELECT config FROM "${OM_TABLE}" WHERE id = ?`,
-            args: [input.id]
-          });
-          if (selectResult.rows.length === 0) {
-            throw new MastraError({
-              id: createStorageErrorId("LIBSQL", "UPDATE_OM_CONFIG", "NOT_FOUND"),
-              text: `Observational memory record not found: ${input.id}`,
-              domain: ErrorDomain.STORAGE,
-              category: ErrorCategory.THIRD_PARTY,
-              details: { id: input.id }
-            });
-          }
-          const row = selectResult.rows[0];
-          const existing = row.config ? JSON.parse(row.config) : {};
-          const merged = this.deepMergeConfig(existing, input.config);
-          await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET config = ?, "updatedAt" = ? WHERE id = ?`,
-            args: [JSON.stringify(merged), (/* @__PURE__ */ new Date()).toISOString(), input.id]
+          await withClientWriteLock(this.#client, async () => {
+            const tx = await this.#client.transaction("write");
+            try {
+              const selectResult = await tx.execute({
+                sql: `SELECT config FROM "${OM_TABLE}" WHERE id = ?`,
+                args: [input.id]
+              });
+              if (selectResult.rows.length === 0) {
+                throw new MastraError({
+                  id: createStorageErrorId("LIBSQL", "UPDATE_OM_CONFIG", "NOT_FOUND"),
+                  text: `Observational memory record not found: ${input.id}`,
+                  domain: ErrorDomain.STORAGE,
+                  category: ErrorCategory.THIRD_PARTY,
+                  details: { id: input.id }
+                });
+              }
+              const row = selectResult.rows[0];
+              const existing = row.config ? JSON.parse(row.config) : {};
+              const merged = this.deepMergeConfig(existing, input.config);
+              await tx.execute({
+                sql: `UPDATE "${OM_TABLE}" SET config = ?, "updatedAt" = ? WHERE id = ?`,
+                args: [JSON.stringify(merged), (/* @__PURE__ */ new Date()).toISOString(), input.id]
+              });
+              await tx.commit();
+            } catch (error) {
+              if (!tx.closed) await tx.rollback();
+              throw error;
+            }
           });
         } catch (error) {
           if (error instanceof MastraError) {
@@ -19352,63 +19407,72 @@ Note: This migration may take some time for large tables.
       async updateBufferedObservations(input) {
         try {
           const nowStr = (/* @__PURE__ */ new Date()).toISOString();
-          const current = await this.#client.execute({
-            sql: `SELECT "bufferedObservationChunks" FROM "${OM_TABLE}" WHERE id = ?`,
-            args: [input.id]
-          });
-          if (!current.rows || current.rows.length === 0) {
-            throw new MastraError({
-              id: createStorageErrorId("LIBSQL", "UPDATE_BUFFERED_OBSERVATIONS", "NOT_FOUND"),
-              text: `Observational memory record not found: ${input.id}`,
-              domain: ErrorDomain.STORAGE,
-              category: ErrorCategory.THIRD_PARTY,
-              details: { id: input.id }
-            });
-          }
-          const row = current.rows[0];
-          let existingChunks = [];
-          if (row.bufferedObservationChunks) {
+          await withClientWriteLock(this.#client, async () => {
+            const tx = await this.#client.transaction("write");
             try {
-              const parsed = typeof row.bufferedObservationChunks === "string" ? JSON.parse(row.bufferedObservationChunks) : row.bufferedObservationChunks;
-              existingChunks = Array.isArray(parsed) ? parsed : [];
-            } catch {
-              existingChunks = [];
+              const current = await tx.execute({
+                sql: `SELECT "bufferedObservationChunks" FROM "${OM_TABLE}" WHERE id = ?`,
+                args: [input.id]
+              });
+              if (!current.rows || current.rows.length === 0) {
+                throw new MastraError({
+                  id: createStorageErrorId("LIBSQL", "UPDATE_BUFFERED_OBSERVATIONS", "NOT_FOUND"),
+                  text: `Observational memory record not found: ${input.id}`,
+                  domain: ErrorDomain.STORAGE,
+                  category: ErrorCategory.THIRD_PARTY,
+                  details: { id: input.id }
+                });
+              }
+              const row = current.rows[0];
+              let existingChunks = [];
+              if (row.bufferedObservationChunks) {
+                try {
+                  const parsed = typeof row.bufferedObservationChunks === "string" ? JSON.parse(row.bufferedObservationChunks) : row.bufferedObservationChunks;
+                  existingChunks = Array.isArray(parsed) ? parsed : [];
+                } catch {
+                  existingChunks = [];
+                }
+              }
+              const newChunk = {
+                id: `ombuf-${randomUUID3()}`,
+                cycleId: input.chunk.cycleId,
+                observations: input.chunk.observations,
+                tokenCount: input.chunk.tokenCount,
+                messageIds: input.chunk.messageIds,
+                messageTokens: input.chunk.messageTokens,
+                lastObservedAt: input.chunk.lastObservedAt,
+                createdAt: /* @__PURE__ */ new Date(),
+                suggestedContinuation: input.chunk.suggestedContinuation,
+                currentTask: input.chunk.currentTask,
+                threadTitle: input.chunk.threadTitle,
+                extractedValues: input.chunk.extractedValues,
+                extractionFailures: input.chunk.extractionFailures
+              };
+              const newChunks = [...existingChunks, newChunk];
+              const lastBufferedAtTime = input.lastBufferedAtTime ? input.lastBufferedAtTime.toISOString() : null;
+              const result = await tx.execute({
+                sql: `UPDATE "${OM_TABLE}" SET
+              "bufferedObservationChunks" = ?,
+              "lastBufferedAtTime" = COALESCE(?, "lastBufferedAtTime"),
+              "updatedAt" = ?
+            WHERE id = ?`,
+                args: [JSON.stringify(newChunks), lastBufferedAtTime, nowStr, input.id]
+              });
+              if (result.rowsAffected === 0) {
+                throw new MastraError({
+                  id: createStorageErrorId("LIBSQL", "UPDATE_BUFFERED_OBSERVATIONS", "NOT_FOUND"),
+                  text: `Observational memory record not found: ${input.id}`,
+                  domain: ErrorDomain.STORAGE,
+                  category: ErrorCategory.THIRD_PARTY,
+                  details: { id: input.id }
+                });
+              }
+              await tx.commit();
+            } catch (error) {
+              if (!tx.closed) await tx.rollback();
+              throw error;
             }
-          }
-          const newChunk = {
-            id: `ombuf-${randomUUID3()}`,
-            cycleId: input.chunk.cycleId,
-            observations: input.chunk.observations,
-            tokenCount: input.chunk.tokenCount,
-            messageIds: input.chunk.messageIds,
-            messageTokens: input.chunk.messageTokens,
-            lastObservedAt: input.chunk.lastObservedAt,
-            createdAt: /* @__PURE__ */ new Date(),
-            suggestedContinuation: input.chunk.suggestedContinuation,
-            currentTask: input.chunk.currentTask,
-            threadTitle: input.chunk.threadTitle,
-            extractedValues: input.chunk.extractedValues,
-            extractionFailures: input.chunk.extractionFailures
-          };
-          const newChunks = [...existingChunks, newChunk];
-          const lastBufferedAtTime = input.lastBufferedAtTime ? input.lastBufferedAtTime.toISOString() : null;
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET
-          "bufferedObservationChunks" = ?,
-          "lastBufferedAtTime" = COALESCE(?, "lastBufferedAtTime"),
-          "updatedAt" = ?
-        WHERE id = ?`,
-            args: [JSON.stringify(newChunks), lastBufferedAtTime, nowStr, input.id]
           });
-          if (result.rowsAffected === 0) {
-            throw new MastraError({
-              id: createStorageErrorId("LIBSQL", "UPDATE_BUFFERED_OBSERVATIONS", "NOT_FOUND"),
-              text: `Observational memory record not found: ${input.id}`,
-              domain: ErrorDomain.STORAGE,
-              category: ErrorCategory.THIRD_PARTY,
-              details: { id: input.id }
-            });
-          }
         } catch (error) {
           if (error instanceof MastraError) {
             throw error;
@@ -19427,150 +19491,160 @@ Note: This migration may take some time for large tables.
       async swapBufferedToActive(input) {
         try {
           const nowStr = (/* @__PURE__ */ new Date()).toISOString();
-          const current = await this.#client.execute({
-            sql: `SELECT * FROM "${OM_TABLE}" WHERE id = ?`,
-            args: [input.id]
-          });
-          if (!current.rows || current.rows.length === 0) {
-            throw new MastraError({
-              id: createStorageErrorId("LIBSQL", "SWAP_BUFFERED_TO_ACTIVE", "NOT_FOUND"),
-              text: `Observational memory record not found: ${input.id}`,
-              domain: ErrorDomain.STORAGE,
-              category: ErrorCategory.THIRD_PARTY,
-              details: { id: input.id }
-            });
-          }
-          const row = current.rows[0];
-          let chunks = [];
-          if (row.bufferedObservationChunks) {
+          return await withClientWriteLock(this.#client, async () => {
+            const tx = await this.#client.transaction("write");
             try {
-              const parsed = typeof row.bufferedObservationChunks === "string" ? JSON.parse(row.bufferedObservationChunks) : row.bufferedObservationChunks;
-              chunks = Array.isArray(parsed) ? parsed : [];
-            } catch {
-              chunks = [];
-            }
-          }
-          if (chunks.length === 0) {
-            return {
-              chunksActivated: 0,
-              messageTokensActivated: 0,
-              observationTokensActivated: 0,
-              messagesActivated: 0,
-              activatedCycleIds: [],
-              activatedMessageIds: []
-            };
-          }
-          const retentionFloor = input.messageTokensThreshold * (1 - input.activationRatio);
-          const targetMessageTokens = Math.max(0, input.currentPendingTokens - retentionFloor);
-          let cumulativeMessageTokens = 0;
-          let bestOverBoundary = 0;
-          let bestOverTokens = 0;
-          let bestUnderBoundary = 0;
-          let bestUnderTokens = 0;
-          for (let i = 0; i < chunks.length; i++) {
-            cumulativeMessageTokens += chunks[i].messageTokens ?? 0;
-            const boundary2 = i + 1;
-            if (cumulativeMessageTokens >= targetMessageTokens) {
-              if (bestOverBoundary === 0 || cumulativeMessageTokens < bestOverTokens) {
-                bestOverBoundary = boundary2;
-                bestOverTokens = cumulativeMessageTokens;
+              const current = await tx.execute({
+                sql: `SELECT * FROM "${OM_TABLE}" WHERE id = ?`,
+                args: [input.id]
+              });
+              if (!current.rows || current.rows.length === 0) {
+                throw new MastraError({
+                  id: createStorageErrorId("LIBSQL", "SWAP_BUFFERED_TO_ACTIVE", "NOT_FOUND"),
+                  text: `Observational memory record not found: ${input.id}`,
+                  domain: ErrorDomain.STORAGE,
+                  category: ErrorCategory.THIRD_PARTY,
+                  details: { id: input.id }
+                });
               }
-            } else {
-              if (cumulativeMessageTokens > bestUnderTokens) {
-                bestUnderBoundary = boundary2;
-                bestUnderTokens = cumulativeMessageTokens;
+              const row = current.rows[0];
+              let chunks = [];
+              if (row.bufferedObservationChunks) {
+                try {
+                  const parsed = typeof row.bufferedObservationChunks === "string" ? JSON.parse(row.bufferedObservationChunks) : row.bufferedObservationChunks;
+                  chunks = Array.isArray(parsed) ? parsed : [];
+                } catch {
+                  chunks = [];
+                }
               }
-            }
-          }
-          const maxOvershoot = retentionFloor * 0.95;
-          const overshoot = bestOverTokens - targetMessageTokens;
-          const remainingAfterOver = input.currentPendingTokens - bestOverTokens;
-          const remainingAfterUnder = input.currentPendingTokens - bestUnderTokens;
-          const minRemaining = Math.min(1e3, retentionFloor);
-          let chunksToActivate;
-          if (input.forceMaxActivation && bestOverBoundary > 0 && remainingAfterOver >= minRemaining) {
-            chunksToActivate = bestOverBoundary;
-          } else if (bestOverBoundary > 0 && overshoot <= maxOvershoot && remainingAfterOver >= minRemaining) {
-            chunksToActivate = bestOverBoundary;
-          } else if (bestUnderBoundary > 0 && remainingAfterUnder >= minRemaining) {
-            chunksToActivate = bestUnderBoundary;
-          } else if (bestOverBoundary > 0) {
-            chunksToActivate = bestOverBoundary;
-          } else {
-            chunksToActivate = 1;
-          }
-          const activatedChunks = chunks.slice(0, chunksToActivate);
-          const remainingChunks = chunks.slice(chunksToActivate);
-          const activatedContent = activatedChunks.map((c) => c.observations).join("\n\n");
-          const activatedTokens = activatedChunks.reduce((sum, c) => sum + c.tokenCount, 0);
-          const activatedMessageTokens = activatedChunks.reduce((sum, c) => sum + (c.messageTokens ?? 0), 0);
-          const activatedMessageCount = activatedChunks.reduce((sum, c) => sum + c.messageIds.length, 0);
-          const activatedCycleIds = activatedChunks.map((c) => c.cycleId).filter((id) => !!id);
-          const activatedMessageIds = activatedChunks.flatMap((c) => c.messageIds ?? []);
-          const latestChunk = activatedChunks[activatedChunks.length - 1];
-          const lastObservedAt = input.lastObservedAt ?? (latestChunk?.lastObservedAt ? new Date(latestChunk.lastObservedAt) : /* @__PURE__ */ new Date());
-          const lastObservedAtStr = lastObservedAt.toISOString();
-          const existingActive = row.activeObservations || "";
-          const existingTokenCount = Number(row.observationTokenCount || 0);
-          const boundary = `
+              if (chunks.length === 0) {
+                await tx.commit();
+                return {
+                  chunksActivated: 0,
+                  messageTokensActivated: 0,
+                  observationTokensActivated: 0,
+                  messagesActivated: 0,
+                  activatedCycleIds: [],
+                  activatedMessageIds: []
+                };
+              }
+              const retentionFloor = input.messageTokensThreshold * (1 - input.activationRatio);
+              const targetMessageTokens = Math.max(0, input.currentPendingTokens - retentionFloor);
+              let cumulativeMessageTokens = 0;
+              let bestOverBoundary = 0;
+              let bestOverTokens = 0;
+              let bestUnderBoundary = 0;
+              let bestUnderTokens = 0;
+              for (let i = 0; i < chunks.length; i++) {
+                cumulativeMessageTokens += chunks[i].messageTokens ?? 0;
+                const boundary2 = i + 1;
+                if (cumulativeMessageTokens >= targetMessageTokens) {
+                  if (bestOverBoundary === 0 || cumulativeMessageTokens < bestOverTokens) {
+                    bestOverBoundary = boundary2;
+                    bestOverTokens = cumulativeMessageTokens;
+                  }
+                } else {
+                  if (cumulativeMessageTokens > bestUnderTokens) {
+                    bestUnderBoundary = boundary2;
+                    bestUnderTokens = cumulativeMessageTokens;
+                  }
+                }
+              }
+              const maxOvershoot = retentionFloor * 0.95;
+              const overshoot = bestOverTokens - targetMessageTokens;
+              const remainingAfterOver = input.currentPendingTokens - bestOverTokens;
+              const remainingAfterUnder = input.currentPendingTokens - bestUnderTokens;
+              const minRemaining = Math.min(1e3, retentionFloor);
+              let chunksToActivate;
+              if (input.forceMaxActivation && bestOverBoundary > 0 && remainingAfterOver >= minRemaining) {
+                chunksToActivate = bestOverBoundary;
+              } else if (bestOverBoundary > 0 && overshoot <= maxOvershoot && remainingAfterOver >= minRemaining) {
+                chunksToActivate = bestOverBoundary;
+              } else if (bestUnderBoundary > 0 && remainingAfterUnder >= minRemaining) {
+                chunksToActivate = bestUnderBoundary;
+              } else if (bestOverBoundary > 0) {
+                chunksToActivate = bestOverBoundary;
+              } else {
+                chunksToActivate = 1;
+              }
+              const activatedChunks = chunks.slice(0, chunksToActivate);
+              const remainingChunks = chunks.slice(chunksToActivate);
+              const activatedContent = activatedChunks.map((c) => c.observations).join("\n\n");
+              const activatedTokens = activatedChunks.reduce((sum, c) => sum + c.tokenCount, 0);
+              const activatedMessageTokens = activatedChunks.reduce((sum, c) => sum + (c.messageTokens ?? 0), 0);
+              const activatedMessageCount = activatedChunks.reduce((sum, c) => sum + c.messageIds.length, 0);
+              const activatedCycleIds = activatedChunks.map((c) => c.cycleId).filter((id) => !!id);
+              const activatedMessageIds = activatedChunks.flatMap((c) => c.messageIds ?? []);
+              const latestChunk = activatedChunks[activatedChunks.length - 1];
+              const lastObservedAt = input.lastObservedAt ?? (latestChunk?.lastObservedAt ? new Date(latestChunk.lastObservedAt) : /* @__PURE__ */ new Date());
+              const lastObservedAtStr = lastObservedAt.toISOString();
+              const existingActive = row.activeObservations || "";
+              const existingTokenCount = Number(row.observationTokenCount || 0);
+              const boundary = `
 
 --- message boundary (${lastObservedAt.toISOString()}) ---
 
 `;
-          const newActive = existingActive ? `${existingActive}${boundary}${activatedContent}` : activatedContent;
-          const newTokenCount = existingTokenCount + activatedTokens;
-          const existingPending = Number(row.pendingMessageTokens || 0);
-          const newPending = Math.max(0, existingPending - activatedMessageTokens);
-          const updateResult = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET
-          "activeObservations" = ?,
-          "observationTokenCount" = ?,
-          "pendingMessageTokens" = ?,
-          "bufferedObservationChunks" = ?,
-          "lastObservedAt" = ?,
-          "updatedAt" = ?
-        WHERE id = ?
-          AND "bufferedObservationChunks" IS NOT NULL
-          AND "bufferedObservationChunks" != '[]'`,
-            args: [
-              newActive,
-              newTokenCount,
-              newPending,
-              remainingChunks.length > 0 ? JSON.stringify(remainingChunks) : null,
-              lastObservedAtStr,
-              nowStr,
-              input.id
-            ]
+              const newActive = existingActive ? `${existingActive}${boundary}${activatedContent}` : activatedContent;
+              const newTokenCount = existingTokenCount + activatedTokens;
+              const existingPending = Number(row.pendingMessageTokens || 0);
+              const newPending = Math.max(0, existingPending - activatedMessageTokens);
+              const updateResult = await tx.execute({
+                sql: `UPDATE "${OM_TABLE}" SET
+              "activeObservations" = ?,
+              "observationTokenCount" = ?,
+              "pendingMessageTokens" = ?,
+              "bufferedObservationChunks" = ?,
+              "lastObservedAt" = ?,
+              "updatedAt" = ?
+            WHERE id = ?
+              AND "bufferedObservationChunks" IS NOT NULL
+              AND "bufferedObservationChunks" != '[]'`,
+                args: [
+                  newActive,
+                  newTokenCount,
+                  newPending,
+                  remainingChunks.length > 0 ? JSON.stringify(remainingChunks) : null,
+                  lastObservedAtStr,
+                  nowStr,
+                  input.id
+                ]
+              });
+              await tx.commit();
+              if (updateResult.rowsAffected === 0) {
+                return {
+                  chunksActivated: 0,
+                  messageTokensActivated: 0,
+                  observationTokensActivated: 0,
+                  messagesActivated: 0,
+                  activatedCycleIds: [],
+                  activatedMessageIds: []
+                };
+              }
+              const latestChunkHints = activatedChunks[activatedChunks.length - 1];
+              return {
+                chunksActivated: activatedChunks.length,
+                messageTokensActivated: activatedMessageTokens,
+                observationTokensActivated: activatedTokens,
+                messagesActivated: activatedMessageCount,
+                activatedCycleIds,
+                activatedMessageIds,
+                observations: activatedContent,
+                perChunk: activatedChunks.map((c) => ({
+                  cycleId: c.cycleId ?? "",
+                  messageTokens: c.messageTokens ?? 0,
+                  observationTokens: c.tokenCount,
+                  messageCount: c.messageIds.length,
+                  observations: c.observations
+                })),
+                suggestedContinuation: latestChunkHints?.suggestedContinuation ?? void 0,
+                currentTask: latestChunkHints?.currentTask ?? void 0
+              };
+            } catch (error) {
+              if (!tx.closed) await tx.rollback();
+              throw error;
+            }
           });
-          if (updateResult.rowsAffected === 0) {
-            return {
-              chunksActivated: 0,
-              messageTokensActivated: 0,
-              observationTokensActivated: 0,
-              messagesActivated: 0,
-              activatedCycleIds: [],
-              activatedMessageIds: []
-            };
-          }
-          const latestChunkHints = activatedChunks[activatedChunks.length - 1];
-          return {
-            chunksActivated: activatedChunks.length,
-            messageTokensActivated: activatedMessageTokens,
-            observationTokensActivated: activatedTokens,
-            messagesActivated: activatedMessageCount,
-            activatedCycleIds,
-            activatedMessageIds,
-            observations: activatedContent,
-            perChunk: activatedChunks.map((c) => ({
-              cycleId: c.cycleId ?? "",
-              messageTokens: c.messageTokens ?? 0,
-              observationTokens: c.tokenCount,
-              messageCount: c.messageIds.length,
-              observations: c.observations
-            })),
-            suggestedContinuation: latestChunkHints?.suggestedContinuation ?? void 0,
-            currentTask: latestChunkHints?.currentTask ?? void 0
-          };
         } catch (error) {
           if (error instanceof MastraError) {
             throw error;
@@ -19589,28 +19663,31 @@ Note: This migration may take some time for large tables.
       async updateBufferedReflection(input) {
         try {
           const nowStr = (/* @__PURE__ */ new Date()).toISOString();
-          const result = await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET
-          "bufferedReflection" = CASE
-            WHEN "bufferedReflection" IS NOT NULL AND "bufferedReflection" != ''
-            THEN "bufferedReflection" || char(10) || char(10) || ?
-            ELSE ?
-          END,
-          "bufferedReflectionTokens" = COALESCE("bufferedReflectionTokens", 0) + ?,
-          "bufferedReflectionInputTokens" = COALESCE("bufferedReflectionInputTokens", 0) + ?,
-          "reflectedObservationLineCount" = ?,
-          "updatedAt" = ?
-        WHERE id = ?`,
-            args: [
-              input.reflection,
-              input.reflection,
-              input.tokenCount,
-              input.inputTokenCount,
-              input.reflectedObservationLineCount,
-              nowStr,
-              input.id
-            ]
-          });
+          const result = await withClientWriteLock(
+            this.#client,
+            () => this.#client.execute({
+              sql: `UPDATE "${OM_TABLE}" SET
+            "bufferedReflection" = CASE
+              WHEN "bufferedReflection" IS NOT NULL AND "bufferedReflection" != ''
+              THEN "bufferedReflection" || char(10) || char(10) || ?
+              ELSE ?
+            END,
+            "bufferedReflectionTokens" = COALESCE("bufferedReflectionTokens", 0) + ?,
+            "bufferedReflectionInputTokens" = COALESCE("bufferedReflectionInputTokens", 0) + ?,
+            "reflectedObservationLineCount" = ?,
+            "updatedAt" = ?
+          WHERE id = ?`,
+              args: [
+                input.reflection,
+                input.reflection,
+                input.tokenCount,
+                input.inputTokenCount,
+                input.reflectedObservationLineCount,
+                nowStr,
+                input.id
+              ]
+            })
+          );
           if (result.rowsAffected === 0) {
             throw new MastraError({
               id: createStorageErrorId("LIBSQL", "UPDATE_BUFFERED_REFLECTION", "NOT_FOUND"),
@@ -19637,55 +19714,64 @@ Note: This migration may take some time for large tables.
       }
       async swapBufferedReflectionToActive(input) {
         try {
-          const current = await this.#client.execute({
-            sql: `SELECT * FROM "${OM_TABLE}" WHERE id = ?`,
-            args: [input.currentRecord.id]
-          });
-          if (!current.rows || current.rows.length === 0) {
-            throw new MastraError({
-              id: createStorageErrorId("LIBSQL", "SWAP_BUFFERED_REFLECTION_TO_ACTIVE", "NOT_FOUND"),
-              text: `Observational memory record not found: ${input.currentRecord.id}`,
-              domain: ErrorDomain.STORAGE,
-              category: ErrorCategory.THIRD_PARTY,
-              details: { id: input.currentRecord.id }
-            });
-          }
-          const row = current.rows[0];
-          const bufferedReflection = row.bufferedReflection || "";
-          const reflectedLineCount = Number(row.reflectedObservationLineCount || 0);
-          if (!bufferedReflection) {
-            throw new MastraError({
-              id: createStorageErrorId("LIBSQL", "SWAP_BUFFERED_REFLECTION_TO_ACTIVE", "NO_CONTENT"),
-              text: "No buffered reflection to swap",
-              domain: ErrorDomain.STORAGE,
-              category: ErrorCategory.USER,
-              details: { id: input.currentRecord.id }
-            });
-          }
-          const currentObservations = row.activeObservations || "";
-          const allLines = currentObservations.split("\n");
-          const unreflectedLines = allLines.slice(reflectedLineCount);
-          const unreflectedContent = unreflectedLines.join("\n").trim();
-          const newObservations = unreflectedContent ? `${bufferedReflection}
+          return await withClientWriteLock(this.#client, async () => {
+            const tx = await this.#client.transaction("write");
+            try {
+              const current = await tx.execute({
+                sql: `SELECT * FROM "${OM_TABLE}" WHERE id = ?`,
+                args: [input.currentRecord.id]
+              });
+              if (!current.rows || current.rows.length === 0) {
+                throw new MastraError({
+                  id: createStorageErrorId("LIBSQL", "SWAP_BUFFERED_REFLECTION_TO_ACTIVE", "NOT_FOUND"),
+                  text: `Observational memory record not found: ${input.currentRecord.id}`,
+                  domain: ErrorDomain.STORAGE,
+                  category: ErrorCategory.THIRD_PARTY,
+                  details: { id: input.currentRecord.id }
+                });
+              }
+              const row = current.rows[0];
+              const bufferedReflection = row.bufferedReflection || "";
+              const reflectedLineCount = Number(row.reflectedObservationLineCount || 0);
+              if (!bufferedReflection) {
+                throw new MastraError({
+                  id: createStorageErrorId("LIBSQL", "SWAP_BUFFERED_REFLECTION_TO_ACTIVE", "NO_CONTENT"),
+                  text: "No buffered reflection to swap",
+                  domain: ErrorDomain.STORAGE,
+                  category: ErrorCategory.USER,
+                  details: { id: input.currentRecord.id }
+                });
+              }
+              const currentObservations = row.activeObservations || "";
+              const allLines = currentObservations.split("\n");
+              const unreflectedLines = allLines.slice(reflectedLineCount);
+              const unreflectedContent = unreflectedLines.join("\n").trim();
+              const newObservations = unreflectedContent ? `${bufferedReflection}
 
 ${unreflectedContent}` : bufferedReflection;
-          const newRecord = await this.createReflectionGeneration({
-            currentRecord: input.currentRecord,
-            reflection: newObservations,
-            tokenCount: input.tokenCount
+              const newRecord = await this.#insertReflectionGeneration(tx, {
+                currentRecord: input.currentRecord,
+                reflection: newObservations,
+                tokenCount: input.tokenCount
+              });
+              const nowStr = (/* @__PURE__ */ new Date()).toISOString();
+              await tx.execute({
+                sql: `UPDATE "${OM_TABLE}" SET
+              "bufferedReflection" = NULL,
+              "bufferedReflectionTokens" = NULL,
+              "bufferedReflectionInputTokens" = NULL,
+              "reflectedObservationLineCount" = NULL,
+              "updatedAt" = ?
+            WHERE id = ?`,
+                args: [nowStr, input.currentRecord.id]
+              });
+              await tx.commit();
+              return newRecord;
+            } catch (error) {
+              if (!tx.closed) await tx.rollback();
+              throw error;
+            }
           });
-          const nowStr = (/* @__PURE__ */ new Date()).toISOString();
-          await this.#client.execute({
-            sql: `UPDATE "${OM_TABLE}" SET
-          "bufferedReflection" = NULL,
-          "bufferedReflectionTokens" = NULL,
-          "bufferedReflectionInputTokens" = NULL,
-          "reflectedObservationLineCount" = NULL,
-          "updatedAt" = ?
-        WHERE id = ?`,
-            args: [nowStr, input.currentRecord.id]
-          });
-          return newRecord;
         } catch (error) {
           if (error instanceof MastraError) {
             throw error;
@@ -26436,12 +26522,16 @@ function invalidateCustomProvidersSnapshots(tenant) {
 // src/routes/fs.ts
 import { lstat, open, readdir, realpath, stat } from "fs/promises";
 import { homedir } from "os";
-import { isAbsolute, join, resolve, sep } from "path";
+import { isAbsolute, join, posix as posixPath, resolve, sep } from "path";
+import { SandboxFilesystem } from "@mastra/code-sdk/agents/sandbox-filesystem";
 import { detectProject, getResourceIdOverride } from "@mastra/code-sdk/utils/project";
 import { registerApiRoute as registerApiRoute2 } from "@mastra/core/server";
 var MAX_TEXT_FILE_BYTES = 512 * 1024;
 var TEXT_DECODER = new TextDecoder("utf-8", { fatal: true });
 var APPROVED_RENDERED_ROOTS = /* @__PURE__ */ new Set([".artifacts"]);
+function loose2(c) {
+  return c;
+}
 function resolveFsRoot(root) {
   return resolve(root && root.trim() ? root : homedir());
 }
@@ -26622,6 +26712,91 @@ async function listArtifacts(root, workspacePath) {
     entries: listing.entries
   };
 }
+async function resolveAuthorizedSession(c, deps, workspacePath) {
+  if (!deps) return null;
+  const session = await deps.sessions.getBySessionId(workspacePath);
+  if (!session) return null;
+  if (deps.auth.enabled()) {
+    await deps.auth.ensureUser(c);
+    const tenant = deps.auth.tenant(c);
+    if (!tenant || tenant.orgId !== session.orgId || tenant.userId !== session.userId) {
+      throw new Error("Session is not available to the current user");
+    }
+  }
+  return session;
+}
+async function sessionSandbox(fleet, session) {
+  if (!fleet.enabled || !session.sandboxId || !session.sandboxWorkdir) return null;
+  let sandbox;
+  try {
+    sandbox = await fleet.reattachSandbox(session.sandboxId, { workingDirectory: session.sandboxWorkdir });
+  } catch {
+    return null;
+  }
+  return {
+    sandbox,
+    filesystem: new SandboxFilesystem({ sandbox, workdir: session.sandboxWorkdir }),
+    workdir: session.sandboxWorkdir
+  };
+}
+async function listSessionRenderedPath(fleet, session, renderedRoot) {
+  const safeRoot = assertApprovedRenderedRoot(renderedRoot);
+  const rootPath = posixPath.join(session.sandboxWorkdir ?? "", safeRoot);
+  const empty = { workspacePath: session.sessionId, root: safeRoot, rootPath, entries: [] };
+  const handle = await sessionSandbox(fleet, session);
+  if (!handle) return empty;
+  const quotedRoot = `'${rootPath.replace(/'/g, `'\\''`)}'`;
+  const result = await handle.sandbox.executeCommand(
+    "sh",
+    [
+      "-c",
+      `test -d ${quotedRoot} && find ${quotedRoot} -mindepth 1 -printf '%y\\t%s\\t%T@\\t%p\\n' 2>/dev/null || true`
+    ],
+    { timeout: 3e4 }
+  );
+  if (result.exitCode !== 0) return empty;
+  const entries = [];
+  for (const line of result.stdout.split("\n")) {
+    if (!line) continue;
+    const [type, sizeStr, mtimeStr, ...pathParts] = line.split("	");
+    const fullPath = pathParts.join("	");
+    if (!fullPath || !fullPath.startsWith(`${rootPath}/`)) continue;
+    const relativePath = fullPath.slice(rootPath.length + 1);
+    entries.push({
+      name: posixPath.basename(relativePath),
+      path: relativePath,
+      type: type === "d" ? "directory" : "file",
+      size: type === "d" ? 0 : Number(sizeStr) || 0,
+      updatedAt: new Date((Number(mtimeStr) || 0) * 1e3).toISOString()
+    });
+  }
+  entries.sort((a, b) => a.path.localeCompare(b.path));
+  return { workspacePath: session.sessionId, root: safeRoot, rootPath, entries };
+}
+async function readSessionWorkspaceFile(fleet, session, path) {
+  const safePath = assertRelativePath(path, "path");
+  assertApprovedRenderedRoot(safePath.split("/")[0] ?? "");
+  const handle = await sessionSandbox(fleet, session);
+  if (!handle) throw new Error("Session workspace is not available");
+  const { filesystem } = handle;
+  const info = await filesystem.stat(safePath);
+  if (info.type === "directory") throw new Error("Path is a directory");
+  const buffer = await filesystem.readFile(safePath);
+  const truncated = buffer.length > MAX_TEXT_FILE_BYTES;
+  const base = {
+    workspacePath: session.sessionId,
+    path: safePath,
+    name: posixPath.basename(safePath),
+    size: buffer.length,
+    updatedAt: info.modifiedAt.toISOString()
+  };
+  try {
+    const content = TEXT_DECODER.decode(truncated ? buffer.subarray(0, MAX_TEXT_FILE_BYTES) : buffer);
+    return { ...base, contentType: "text", content, truncated };
+  } catch {
+    return { ...base, contentType: "unsupported" };
+  }
+}
 function resolveCodebase(projectPath) {
   const info = detectProject(projectPath);
   const override = getResourceIdOverride(info.rootPath);
@@ -26635,6 +26810,7 @@ function resolveCodebase(projectPath) {
 }
 function buildFsRoutes(options = {}) {
   const root = resolveFsRoot(options.root);
+  const sessionFs = options.sessionFs;
   return [
     registerApiRoute2("/web/fs/list", {
       method: "GET",
@@ -26674,10 +26850,14 @@ function buildFsRoutes(options = {}) {
         if (!workspacePath) return c.json({ error: "Missing required query param: workspacePath" }, 400);
         if (!renderedRoot) return c.json({ error: "Missing required query param: root" }, 400);
         try {
+          const session = await resolveAuthorizedSession(loose2(c), sessionFs, workspacePath);
+          if (session && sessionFs) {
+            return c.json(await listSessionRenderedPath(sessionFs.fleet, session, renderedRoot));
+          }
           return c.json(await listWorkspaceRenderedPath(root, workspacePath, renderedRoot));
         } catch (error) {
           const message = error instanceof Error ? error.message : String(error);
-          const status = message.includes("outside") || message.includes("relative") || message.includes("escapes") || message.includes("not approved") ? 403 : 500;
+          const status = message.includes("outside") || message.includes("relative") || message.includes("escapes") || message.includes("not approved") || message.includes("not available") ? 403 : 500;
           return c.json({ error: message }, status);
         }
       }
@@ -26691,10 +26871,14 @@ function buildFsRoutes(options = {}) {
         if (!workspacePath) return c.json({ error: "Missing required query param: workspacePath" }, 400);
         if (!path) return c.json({ error: "Missing required query param: path" }, 400);
         try {
+          const session = await resolveAuthorizedSession(loose2(c), sessionFs, workspacePath);
+          if (session && sessionFs) {
+            return c.json(await readSessionWorkspaceFile(sessionFs.fleet, session, path));
+          }
           return c.json(await readWorkspaceFile(root, workspacePath, path));
         } catch (error) {
           const message = error instanceof Error ? error.message : String(error);
-          const status = message.includes("outside") || message.includes("relative") || message.includes("escapes") || message.includes("not approved") ? 403 : message.includes("directory") ? 400 : 500;
+          const status = message.includes("outside") || message.includes("relative") || message.includes("escapes") || message.includes("not approved") || message.includes("not available") ? 403 : message.includes("directory") ? 400 : message.includes("not found") ? 404 : 500;
           return c.json({ error: message }, status);
         }
       }
@@ -26720,7 +26904,7 @@ function buildFsRoutes(options = {}) {
 
 // src/routes/intake.ts
 import { registerApiRoute as registerApiRoute3 } from "@mastra/core/server";
-function loose2(c) {
+function loose3(c) {
   return c;
 }
 function sanitizeIdList(value) {
@@ -26784,7 +26968,7 @@ var IntakeRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const tenant = await this.#resolveTenant(loose2(c));
+          const tenant = await this.#resolveTenant(loose3(c));
           if ("response" in tenant) return tenant.response;
           await intake.ensureReady();
           const config = await intake.getConfig({ ...tenant, integrationIds });
@@ -26795,7 +26979,7 @@ var IntakeRoutes = class extends Route {
         method: "PUT",
         requiresAuth: false,
         handler: async (c) => {
-          const tenant = await this.#resolveTenant(loose2(c));
+          const tenant = await this.#resolveTenant(loose3(c));
           if ("response" in tenant) return tenant.response;
           let body;
           try {
@@ -26810,7 +26994,7 @@ var IntakeRoutes = class extends Route {
           await intake.ensureReady();
           await intake.saveConfig({ ...tenant, config });
           await audit.emit({
-            context: loose2(c),
+            context: loose3(c),
             input: {
               action: "factory.intake.config_updated",
               targets: [{ type: "intake_config", id: tenant.orgId }],
@@ -26829,7 +27013,7 @@ var IntakeRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const tenant = await this.#resolveTenant(loose2(c));
+          const tenant = await this.#resolveTenant(loose3(c));
           if ("response" in tenant) return tenant.response;
           const pages = await Promise.all(
             integrations.map(async (integration) => ({
@@ -26848,7 +27032,7 @@ var IntakeRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const tenant = await this.#resolveTenant(loose2(c));
+          const tenant = await this.#resolveTenant(loose3(c));
           if ("response" in tenant) return tenant.response;
           const cursors = decodeCursor(c.req.query("cursor"));
           if (!cursors) return c.json({ error: "invalid_cursor" }, 400);
@@ -27227,7 +27411,7 @@ var OAUTH_FLOWS = {
     }
   }
 };
-function loose3(c) {
+function loose4(c) {
   return c;
 }
 var localSessionsPromise;
@@ -27297,14 +27481,14 @@ var OAuthRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const ctx = await resolveCredentialContext({ c: loose3(c), auth, credentials: modelCredentials });
+          const ctx = await resolveCredentialContext({ c: loose4(c), auth, credentials: modelCredentials });
           if ("response" in ctx) return ctx.response;
           const provider = c.req.param("provider");
           const flow = OAUTH_FLOWS[provider];
           if (!flow) {
             return c.json({ error: "oauth_not_supported", message: `Provider does not support web sign-in` }, 404);
           }
-          const body = await readJsonBody(loose3(c));
+          const body = await readJsonBody(loose4(c));
           if (typeof body.mode === "string" && body.mode !== flow.kind) {
             return c.json({ error: "invalid_mode", message: `Unsupported mode for ${provider}: ${body.mode}` }, 400);
           }
@@ -27341,12 +27525,12 @@ var OAuthRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const ctx = await resolveCredentialContext({ c: loose3(c), auth, credentials: modelCredentials });
+          const ctx = await resolveCredentialContext({ c: loose4(c), auth, credentials: modelCredentials });
           if ("response" in ctx) return ctx.response;
           const provider = c.req.param("provider");
           const flow = OAUTH_FLOWS[provider];
           if (!flow?.complete) return c.json({ error: "oauth_not_supported" }, 404);
-          const body = await readJsonBody(loose3(c));
+          const body = await readJsonBody(loose4(c));
           const sessionId = typeof body.sessionId === "string" ? body.sessionId : "";
           const code = typeof body.code === "string" ? body.code.trim() : "";
           if (!sessionId || !code) return c.json({ error: "Missing required fields: sessionId, code" }, 400);
@@ -27376,12 +27560,12 @@ var OAuthRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const ctx = await resolveCredentialContext({ c: loose3(c), auth, credentials: modelCredentials });
+          const ctx = await resolveCredentialContext({ c: loose4(c), auth, credentials: modelCredentials });
           if ("response" in ctx) return ctx.response;
           const provider = c.req.param("provider");
           const flow = OAUTH_FLOWS[provider];
           if (!flow?.poll) return c.json({ error: "oauth_not_supported" }, 404);
-          const body = await readJsonBody(loose3(c));
+          const body = await readJsonBody(loose4(c));
           const sessionId = typeof body.sessionId === "string" ? body.sessionId : "";
           if (!sessionId) return c.json({ error: "Missing required field: sessionId" }, 400);
           const session = await loadOwnedSession({ ctx, provider, sessionId });
@@ -27432,7 +27616,7 @@ var OAuthRoutes = class extends Route {
         method: "DELETE",
         requiresAuth: false,
         handler: async (c) => {
-          const ctx = await resolveCredentialContext({ c: loose3(c), auth, credentials: modelCredentials });
+          const ctx = await resolveCredentialContext({ c: loose4(c), auth, credentials: modelCredentials });
           if ("response" in ctx) return ctx.response;
           const provider = c.req.param("provider");
           const sessionId = c.req.param("sessionId");
@@ -27445,7 +27629,7 @@ var OAuthRoutes = class extends Route {
         method: "DELETE",
         requiresAuth: false,
         handler: async (c) => {
-          const ctx = await resolveCredentialContext({ c: loose3(c), auth, credentials: modelCredentials });
+          const ctx = await resolveCredentialContext({ c: loose4(c), auth, credentials: modelCredentials });
           if ("response" in ctx) return ctx.response;
           const provider = c.req.param("provider");
           const authProviderId = getAuthProviderId(provider);
@@ -27511,7 +27695,7 @@ var MAX_SCOPE_LENGTH = 2048;
 var MAX_ARGUMENTS_LENGTH2 = 16384;
 var SKILL_NAME_RE2 = /^[a-z0-9]+(?:-[a-z0-9]+)*$/;
 var UUID_RE = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i;
-function loose4(context) {
+function loose5(context) {
   return context;
 }
 function parseBody(value) {
@@ -27585,7 +27769,7 @@ var SkillRoutes = class extends Route {
     const { controllerId, controller, authorizeSessionAddress: customAuthorize } = this.deps;
     const authorize = customAuthorize ?? ((context, address) => this.#authorizeSessionAddress(context, address));
     const handleSkillRequest = async (context, dispatch) => {
-      const c = loose4(context);
+      const c = loose5(context);
       if (c.req.param("controllerId") !== controllerId) {
         return c.json({ error: "controller_not_found", message: "Agent controller not found." }, 404);
       }
@@ -27886,7 +28070,7 @@ function computeFactoryMetrics(items, opts) {
 }
 
 // src/routes/work-items.ts
-function loose5(c) {
+function loose6(c) {
   return c;
 }
 var UUID_RE2 = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
@@ -28204,7 +28388,7 @@ var WorkItemRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const resolved = await this.#resolveProject(loose5(c));
+          const resolved = await this.#resolveProject(loose6(c));
           if ("response" in resolved) return resolved.response;
           await workItems.ensureReady();
           const items = await workItems.list({
@@ -28219,11 +28403,11 @@ var WorkItemRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const resolved = await this.#resolveProject(loose5(c));
+          const resolved = await this.#resolveProject(loose6(c));
           if ("response" in resolved) return resolved.response;
           const { windowStart, windowEnd } = parseMetricsRange(
-            loose5(c).req.query("from"),
-            loose5(c).req.query("to"),
+            loose6(c).req.query("from"),
+            loose6(c).req.query("to"),
             /* @__PURE__ */ new Date()
           );
           await workItems.ensureReady();
@@ -28239,7 +28423,7 @@ var WorkItemRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const resolved = await this.#resolveProject(loose5(c));
+          const resolved = await this.#resolveProject(loose6(c));
           if ("response" in resolved) return resolved.response;
           await queueHealth.ensureReady();
           const stored = await queueHealth.getConfig(resolved.orgId, resolved.factoryProjectId);
@@ -28251,7 +28435,7 @@ var WorkItemRoutes = class extends Route {
         method: "GET",
         requiresAuth: false,
         handler: async (c) => {
-          const context = loose5(c);
+          const context = loose6(c);
           const resolved = await this.#resolveProject(context);
           if ("response" in resolved) return resolved.response;
           const cursorRaw = context.req.query("before");
@@ -28276,7 +28460,7 @@ var WorkItemRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const context = loose5(c);
+          const context = loose6(c);
           const resolved = await this.#resolveProject(context);
           if ("response" in resolved) return resolved.response;
           const decisionId = context.req.param("decisionId");
@@ -28297,9 +28481,9 @@ var WorkItemRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const resolved = await this.#resolveProject(loose5(c));
+          const resolved = await this.#resolveProject(loose6(c));
           if ("response" in resolved) return resolved.response;
-          const body = await readJson(loose5(c));
+          const body = await readJson(loose6(c));
           if (body === void 0) return c.json({ error: "Invalid JSON body" }, 400);
           const input = parseCreateWorkItem(body);
           if (!input) return c.json({ error: "invalid_work_item" }, 400);
@@ -28342,7 +28526,7 @@ var WorkItemRoutes = class extends Route {
               }
               item = await workItems.getForProject(resolved.orgId, resolved.factoryProjectId, item.id) ?? item;
               await audit.emit({
-                context: loose5(c),
+                context: loose6(c),
                 input: {
                   action: "factory.work_item.created",
                   factoryProjectId: resolved.factoryProjectId,
@@ -28353,7 +28537,7 @@ var WorkItemRoutes = class extends Route {
             } else {
               const { stages: _stages, sessions: _sessions, ...boundedPatch } = input;
               await this.#auditWorkItemPatch({
-                context: loose5(c),
+                context: loose6(c),
                 item,
                 previous: result.previous,
                 patch: boundedPatch
@@ -28373,11 +28557,11 @@ var WorkItemRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const resolved = await this.#resolveProject(loose5(c));
+          const resolved = await this.#resolveProject(loose6(c));
           if ("response" in resolved) return resolved.response;
-          const workItemId = loose5(c).req.param("workItemId");
+          const workItemId = loose6(c).req.param("workItemId");
           if (!workItemId || !UUID_RE2.test(workItemId)) return c.json({ error: "Work item not found" }, 404);
-          const parsed = parseTransitionBody(await readJson(loose5(c)));
+          const parsed = parseTransitionBody(await readJson(loose6(c)));
           if (!parsed) return c.json({ error: "invalid_transition_request" }, 400);
           if (!transitionService) {
             return c.json({ error: "factory_transition_unavailable" }, 503);
@@ -28395,7 +28579,7 @@ var WorkItemRoutes = class extends Route {
             }
           });
           await audit.emit({
-            context: loose5(c),
+            context: loose6(c),
             input: {
               action: result.status === "accepted" ? "factory.work_item.stage_moved" : "factory.work_item.transition_rejected",
               factoryProjectId: resolved.factoryProjectId,
@@ -28417,14 +28601,14 @@ var WorkItemRoutes = class extends Route {
         method: "POST",
         requiresAuth: false,
         handler: async (c) => {
-          const resolved = await this.#resolveProject(loose5(c));
+          const resolved = await this.#resolveProject(loose6(c));
           if ("response" in resolved) return resolved.response;
           if (!startCoordinator) {
             return c.json({ error: "factory_start_unavailable" }, 503);
           }
-          const input = parseStartBody(await readJson(loose5(c)), resolved, resolved.factoryProjectId);
+          const input = parseStartBody(await readJson(loose6(c)), resolved, resolved.factoryProjectId);
           if (!input) return c.json({ error: "invalid_factory_start" }, 400);
-          input.requestContext = loose5(c).get("requestContext");
+          input.requestContext = loose6(c).get("requestContext");
           input.defaultModelId = resolved.defaultModelId ?? void 0;
           if (!input.workItem.id && ((input.workItem.input.stages ?? ["intake"]).length !== 1 || (input.workItem.input.stages ?? ["intake"])[0] !== "intake")) {
             return c.json(
@@ -28443,7 +28627,7 @@ var WorkItemRoutes = class extends Route {
             throw error;
           }
           await audit.emit({
-            context: loose5(c),
+            context: loose6(c),
             input: {
               action: "factory.run.started",
               factoryProjectId: resolved.factoryProjectId,
@@ -28465,11 +28649,11 @@ var WorkItemRoutes = class extends Route {
         method: "PATCH",
         requiresAuth: false,
         handler: async (c) => {
-          const tenant = await this.#resolveTenant(loose5(c));
+          const tenant = await this.#resolveTenant(loose6(c));
           if ("response" in tenant) return tenant.response;
-          const id = loose5(c).req.param("id");
+          const id = loose6(c).req.param("id");
           if (!id || !UUID_RE2.test(id)) return c.json({ error: "Work item not found" }, 404);
-          const body = await readJson(loose5(c));
+          const body = await readJson(loose6(c));
           if (body === void 0) return c.json({ error: "Invalid JSON body" }, 400);
           const patch = parseUpdateWorkItem(body);
           if (!patch) return c.json({ error: "invalid_work_item_patch" }, 400);
@@ -28484,7 +28668,7 @@ var WorkItemRoutes = class extends Route {
             const updated = await workItems.update({ orgId: tenant.orgId, id, userId: tenant.userId, patch });
             if (!updated) return c.json({ error: "Work item not found" }, 404);
             await this.#auditWorkItemPatch({
-              context: loose5(c),
+              context: loose6(c),
               item: updated.item,
               previous: updated.previous,
               patch
@@ -28503,15 +28687,15 @@ var WorkItemRoutes = class extends Route {
         method: "DELETE",
         requiresAuth: false,
         handler: async (c) => {
-          const tenant = await this.#resolveTenant(loose5(c));
+          const tenant = await this.#resolveTenant(loose6(c));
           if ("response" in tenant) return tenant.response;
-          const id = loose5(c).req.param("id");
+          const id = loose6(c).req.param("id");
           if (!id || !UUID_RE2.test(id)) return c.json({ error: "Work item not found" }, 404);
           await workItems.ensureReady();
           const deleted = await workItems.delete({ orgId: tenant.orgId, id });
           if (!deleted) return c.json({ error: "Work item not found" }, 404);
           await audit.emit({
-            context: loose5(c),
+            context: loose6(c),
             input: {
               action: "factory.work_item.deleted",
               factoryProjectId: deleted.factoryProjectId,
@@ -28732,7 +28916,14 @@ function assembleFactoryApiRoutes(deps) {
     });
   }
   return [
-    ...buildFsRoutes({ root: deps.fsRoot }),
+    ...buildFsRoutes({
+      root: deps.fsRoot,
+      sessionFs: {
+        auth: deps.auth,
+        fleet: deps.fleet,
+        sessions: deps.sourceControlStorage.forIntegration("github").sessions
+      }
+    }),
     ...new ConfigRoutes({
       auth: deps.auth,
       controller: deps.controller,