From cc4583759dbcb307463d91d99eacbbc6a031d040 Mon Sep 17 00:00:00 2001
From: Henry Fontanier <h.fontanier@gmail.com>
Date: Thu, 23 Nov 2023 15:07:11 +0100
Subject: [PATCH] fix: use parents.in_map instead of parents.in in retrieval.ts
 (#2636)

Co-authored-by: Henry Fontanier <henry@dust.tt>
---
 front/lib/actions/registry.ts                | 2 +-
 front/lib/api/assistant/actions/retrieval.ts | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/front/lib/actions/registry.ts b/front/lib/actions/registry.ts
index fa375d156eae..e8041754e4a5 100644
--- a/front/lib/actions/registry.ts
+++ b/front/lib/actions/registry.ts
@@ -55,7 +55,7 @@ export const DustProdActionRegistry = createActionRegistry({
       DATASOURCE: {
         data_sources: [],
         top_k: 32,
-        filter: { tags: null, parent: null, timestamp: null },
+        filter: { tags: null, parents: null, timestamp: null },
         use_cache: false,
       },
     },
diff --git a/front/lib/api/assistant/actions/retrieval.ts b/front/lib/api/assistant/actions/retrieval.ts
index be5a01500dd5..e39779b2c77f 100644
--- a/front/lib/api/assistant/actions/retrieval.ts
+++ b/front/lib/api/assistant/actions/retrieval.ts
@@ -588,10 +588,11 @@ export async function* runRetrieval(
       config.DATASOURCE.filter.parents = {};
     }
     if (ds.filter.parents?.in) {
-      if (!config.DATASOURCE.filter.parents.in) {
-        config.DATASOURCE.filter.parents.in = [];
+      if (!config.DATASOURCE.filter.parents.in_map) {
+        config.DATASOURCE.filter.parents.in_map = {};
       }
-      config.DATASOURCE.filter.parents.in.push(...ds.filter.parents.in);
+      config.DATASOURCE.filter.parents.in_map[ds.dataSourceId] =
+        ds.filter.parents.in;
     }
     if (ds.filter.parents?.not) {
       if (!config.DATASOURCE.filter.parents.not) {