fix: use parents.in_map instead of parents.in in retrieval.ts (#2636)

Co-authored-by: Henry Fontanier <henry@dust.tt>
dust-tt · Nov 23, 2023 · cc45837 · cc45837
1 parent a2347c1
commit cc45837
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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) {