Merge pull request #1674 from RoadieHQ/23067-make-docs-filterable-on-…

…rag-ai Make Tech Docs Embeddings filterable
RoadieHQ · Oct 15, 2024 · ea08cae · ea08cae
2 parents 54ec926 + ad1565e
commit ea08cae
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/.changeset/small-clocks-warn.md b/.changeset/small-clocks-warn.md
@@ -0,0 +1,5 @@
+---
+'@roadiehq/rag-ai-backend-retrieval-augmenter': minor
+---
+
+Make Tech Docs embeddings creation filterable based on additional passed in entity filter values.
diff --git a/...ckend/rag-ai-backend-retrieval-augmenter/src/indexing/DefaultVectorAugmentationIndexer.ts b/...ckend/rag-ai-backend-retrieval-augmenter/src/indexing/DefaultVectorAugmentationIndexer.ts
@@ -188,7 +188,7 @@ export class DefaultVectorAugmentationIndexer implements AugmentationIndexer {
 
         const entitiesResponse = await this.catalogApi.getEntities(
           {
-            filter: TECHDOCS_ENTITY_FILTER,
+            filter: { ...TECHDOCS_ENTITY_FILTER, ...filter },
           },
           { token },
         );
@@ -275,7 +275,9 @@ export class DefaultVectorAugmentationIndexer implements AugmentationIndexer {
     });
 
     const entityFilter =
-      source === 'tech-docs' ? TECHDOCS_ENTITY_FILTER : filter;
+      source === 'tech-docs'
+        ? { ...TECHDOCS_ENTITY_FILTER, ...filter }
+        : filter;
     const entities = (
       await this.catalogApi.getEntities({ filter: entityFilter }, { token })
     ).items.map(stringifyEntityRef);