@agent-native/core 0.90.4 → 0.90.5

package/corpus/templates/content/actions/_database-source-utils.ts

@@ -302,6 +302,7 @@ export function serializeSourceField(
 
 function serializeSourceRowRecord(
   row: ContentDatabaseSourceRecordRowDb,
+  options: { includeHeavyBuilderBodyValues?: boolean } = {},
 ): ContentDatabaseSourceRow {
   return {
     id: row.id,
@@ -310,10 +311,12 @@ function serializeSourceRowRecord(
     sourceRowId: row.sourceRowId,
     sourceQualifiedId: row.sourceQualifiedId,
     sourceDisplayKey: row.sourceDisplayKey,
-    sourceValues:
+    sourceValues: sourceValuesForSnapshot(
       parseObject<Record<string, DocumentPropertyValue>>(
         row.sourceValuesJson,
       ) ?? {},
+      options,
+    ),
     provenance: row.provenance,
     syncState: normalizeSourceSyncState(row.syncState),
     freshness: normalizeSourceFreshness(row.freshness),
@@ -322,6 +325,27 @@ function serializeSourceRowRecord(
   };
 }
 
+const HEAVY_BUILDER_BODY_SOURCE_VALUE_KEYS = new Set([
+  BUILDER_CMS_BODY_CONTENT_KEY,
+  BUILDER_CMS_BODY_LOSSLESS_CONTENT_KEY,
+  BUILDER_CMS_BODY_READABLE_MAP_KEY,
+  BUILDER_CMS_BODY_SIDECARS_KEY,
+]);
+
+export function sourceValuesForSnapshot(
+  sourceValues: Record<string, DocumentPropertyValue>,
+  options: { includeHeavyBuilderBodyValues?: boolean } = {},
+): Record<string, DocumentPropertyValue> {
+  if (options.includeHeavyBuilderBodyValues === true) return sourceValues;
+  let next: Record<string, DocumentPropertyValue> | null = null;
+  for (const key of HEAVY_BUILDER_BODY_SOURCE_VALUE_KEYS) {
+    if (!Object.prototype.hasOwnProperty.call(sourceValues, key)) continue;
+    next ??= { ...sourceValues };
+    delete next[key];
+  }
+  return next ?? sourceValues;
+}
+
 function serializeSourceChangeSet(
   row: ContentDatabaseSourceChangeSetRowDb,
 ): ContentDatabaseSourceChangeSet {
@@ -1919,7 +1943,9 @@ export async function getContentDatabaseSourceSnapshot(
       asc(schema.contentDatabaseSources.id),
     );
   if (!source) return null;
-  return loadSourceSnapshot(source, database);
+  return loadSourceSnapshot(source, database, {
+    includeHeavyBuilderBodyValues: false,
+  });
 }
 
 /**
@@ -1942,7 +1968,9 @@ export async function getContentDatabaseSourceSnapshotById(
       ),
     );
   if (!source) return null;
-  return loadSourceSnapshot(source, database);
+  return loadSourceSnapshot(source, database, {
+    includeHeavyBuilderBodyValues: false,
+  });
 }
 
 /**
@@ -1955,9 +1983,37 @@ export async function getContentDatabaseSourceSnapshotForWrite(
   database: ContentDatabaseRow | ContentDatabase,
   sourceId?: string | null,
 ): Promise<ContentDatabaseSource | null> {
-  return sourceId
-    ? getContentDatabaseSourceSnapshotById(database, sourceId)
-    : getContentDatabaseSourceSnapshot(database);
+  if (sourceId) {
+    const db = getDb();
+    const [source] = await db
+      .select()
+      .from(schema.contentDatabaseSources)
+      .where(
+        and(
+          eq(schema.contentDatabaseSources.id, sourceId),
+          eq(schema.contentDatabaseSources.databaseId, database.id),
+        ),
+      );
+    return source
+      ? loadSourceSnapshot(source, database, {
+          includeHeavyBuilderBodyValues: true,
+        })
+      : null;
+  }
+  const db = getDb();
+  const [source] = await db
+    .select()
+    .from(schema.contentDatabaseSources)
+    .where(eq(schema.contentDatabaseSources.databaseId, database.id))
+    .orderBy(
+      asc(schema.contentDatabaseSources.createdAt),
+      asc(schema.contentDatabaseSources.id),
+    );
+  return source
+    ? loadSourceSnapshot(source, database, {
+        includeHeavyBuilderBodyValues: true,
+      })
+    : null;
 }
 
 /**
@@ -1982,7 +2038,11 @@ export async function getAllContentDatabaseSourceSnapshots(
     );
   const snapshots: ContentDatabaseSource[] = [];
   for (const source of sources) {
-    snapshots.push(await loadSourceSnapshot(source, database));
+    snapshots.push(
+      await loadSourceSnapshot(source, database, {
+        includeHeavyBuilderBodyValues: false,
+      }),
+    );
   }
   return snapshots;
 }
@@ -2135,6 +2195,7 @@ async function loadSourceSnapshotRowsOptimistically(args: {
 async function loadSourceSnapshot(
   source: ContentDatabaseSourceRowDb,
   database: ContentDatabaseRow | ContentDatabase,
+  options: { includeHeavyBuilderBodyValues: boolean },
 ): Promise<ContentDatabaseSource> {
   const db = getDb();
   const [fieldRows, changeRows, reviewRows, executionRows, propertyDefs] =
@@ -2218,7 +2279,11 @@ async function loadSourceSnapshot(
     database,
     isBuilderSource,
   });
-  const rows = rowRows.map(serializeSourceRowRecord);
+  const rows = rowRows.map((row) =>
+    serializeSourceRowRecord(row, {
+      includeHeavyBuilderBodyValues: options.includeHeavyBuilderBodyValues,
+    }),
+  );
   const documentTitleById = new Map(
     rowDocuments.map((document) => [document.id, document.title]),
   );

package/corpus/templates/content/actions/_database-utils.ts

@@ -123,12 +123,13 @@ export function normalizeContentDatabasePageOptions(options: {
 function serializeDocument(
   doc: typeof schema.documents.$inferSelect,
   membership?: DatabaseMembershipRow,
+  options: { includeContent?: boolean } = {},
 ) {
   return {
     id: doc.id,
     parentId: doc.parentId,
     title: doc.title,
-    content: doc.content,
+    content: options.includeContent === true ? doc.content : "",
     icon: doc.icon,
     position: doc.position,
     isFavorite: parseDocumentFavorite(doc.isFavorite),

package/corpus/templates/content/actions/prepare-builder-source-review.ts

@@ -382,6 +382,11 @@ export default defineAction({
       .string()
       .optional()
       .describe("Target source ID (defaults to the primary source)"),
+    changeSetIds: z
+      .array(z.string())
+      .max(BUILDER_SOURCE_REVIEW_PREPARE_LIMIT)
+      .optional()
+      .describe("Optional bounded set of Builder change-set IDs to prepare"),
     pushModeConfirmation: z
       .enum(["autosave", "draft", "publish"])
       .optional()
@@ -409,13 +414,27 @@ export default defineAction({
     if (!snapshot || snapshot.sourceType !== "builder-cms") {
       throw new Error("Attach a Builder CMS source before reviewing updates.");
     }
+    const requestedIds = new Set(args.changeSetIds ?? []);
     const allReviewableChanges = snapshot.changeSets.filter(
       (changeSet) =>
         changeSet.direction === "outbound" &&
         (changeSet.state === "pending_push" ||
           changeSet.state === "staged_revision" ||
-          changeSet.state === "approved"),
+          changeSet.state === "approved") &&
+        (requestedIds.size === 0 || requestedIds.has(changeSet.id)),
     );
+    if (
+      requestedIds.size > 0 &&
+      allReviewableChanges.length !== requestedIds.size
+    ) {
+      const foundIds = new Set(
+        allReviewableChanges.map((changeSet) => changeSet.id),
+      );
+      const missingIds = [...requestedIds].filter((id) => !foundIds.has(id));
+      throw new Error(
+        `Requested Builder change-set is not reviewable: ${missingIds.join(", ")}.`,
+      );
+    }
     if (allReviewableChanges.length === 0) {
       throw new Error("No pending local Builder changes to review.");
     }