add to updateParents

dust-tt · Dec 12, 2024 · 024c436 · 024c436
1 parent 50289f7
commit 024c436
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/connectors/src/connectors/notion/lib/parents.ts b/connectors/src/connectors/notion/lib/parents.ts
@@ -155,13 +155,18 @@ export async function updateAllParentsFields(
   for (const pageId of pageIdsToUpdate) {
     promises.push(
       q.add(async () => {
-        const parents = await getParents(
+        const pageOrDbIds = await getParents(
           connectorId,
           pageId,
           [],
           memoizationKey,
           onProgress
         );
+
+        // TODO(kw_search) remove legacy
+        const legacyParents = pageOrDbIds;
+        const parents = pageOrDbIds.map((id) => `notion-${id}`);
+
         logger.info(
           {
             connectorId,
@@ -172,7 +177,7 @@ export async function updateAllParentsFields(
         await updateDocumentParentsField({
           dataSourceConfig: dataSourceConfigFromConnector(connector),
           documentId: `notion-${pageId}`,
-          parents,
+          parents: [...parents, ...legacyParents],
         });
         if (onProgress) {
           await onProgress();