[front] Default parents to [documentId] if not specified (#9442)

dust-tt · Dec 17, 2024 · c1e48ce · c1e48ce
1 parent 625a9c4
commit c1e48ce
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/front/lib/api/data_sources.ts b/front/lib/api/data_sources.ts
@@ -640,7 +640,7 @@ export async function handleDataSourceTableCSVUpsert({
   }
 
   const tableId = params.tableId ?? generateRandomModelSId();
-  const tableParents: string[] = params.parents ?? [];
+  const tableParents: string[] = params.parents ?? [tableId];
 
   const flags = await getFeatureFlags(owner);
 

diff --git a/...pages/api/v1/w/[wId]/spaces/[spaceId]/data_sources/[dsId]/documents/[documentId]/index.ts b/...pages/api/v1/w/[wId]/spaces/[spaceId]/data_sources/[dsId]/documents/[documentId]/index.ts
@@ -508,15 +508,17 @@ async function handler(
         statsDClient.increment("document_no_self_ref.count", 1, statsDTags);
       }
 
+      const documentId = req.query.documentId as string;
+
       if (r.data.async === true) {
         const enqueueRes = await enqueueUpsertDocument({
           upsertDocument: {
             workspaceId: owner.sId,
             dataSourceId: dataSource.sId,
-            documentId: req.query.documentId as string,
+            documentId,
             tags: r.data.tags || [],
             parentId: r.data.parent_id || null,
-            parents: r.data.parents || [],
+            parents: r.data.parents || [documentId],
             timestamp: r.data.timestamp || null,
             sourceUrl,
             section,
@@ -556,7 +558,7 @@ async function handler(
           documentId: req.query.documentId as string,
           tags: r.data.tags || [],
           parentId: r.data.parent_id || null,
-          parents: r.data.parents || [],
+          parents: r.data.parents || [documentId],
           sourceUrl,
           timestamp: r.data.timestamp || null,
           section,

diff --git a/front/temporal/upsert_queue/activities.ts b/front/temporal/upsert_queue/activities.ts
@@ -88,7 +88,7 @@ export async function upsertDocumentActivity(
     documentId: upsertQueueItem.documentId,
     tags: upsertQueueItem.tags || [],
     parentId: upsertQueueItem.parentId || null,
-    parents: upsertQueueItem.parents || [],
+    parents: upsertQueueItem.parents || [upsertQueueItem.documentId],
     sourceUrl: upsertQueueItem.sourceUrl,
     timestamp: upsertQueueItem.timestamp,
     section: upsertQueueItem.section,