fix(app): preview grounding with filter

IDEA-Research · Apr 10, 2024 · d14391d · d14391d
1 parent 90a2c81
commit d14391d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/components/src/Annotator/preview.tsx b/packages/components/src/Annotator/preview.tsx
@@ -358,7 +358,7 @@ const Preview: React.FC<PreviewProps> = (props) => {
   }
 
   const getHighlightWords = () => {
-    const objects = list[current].objects;
+    const objects = objectsFilter ? objectsFilter(list[current]) : list[current].objects;
     return categories
       .filter((category) => objects.find((obj: any) => obj.categoryId === category.id))
       .map((category) => {