Fixed query inside highlights

athina-ai · Jan 31, 2025 · 3547223 · 3547223
1 parent 756b562
commit 3547223
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/athina/steps/search.py b/athina/steps/search.py
@@ -87,7 +87,10 @@ def execute(self, input_data: Any) -> Union[Dict[str, Any], None]:
             "excludeText": self.excludeText,
             "includeText": self.includeText,
             "contents": {
-                "highlights": self.highlights,
+                "highlights": {
+                    "query": self.query,
+                    **(self.highlights or {})  # Merging self.highlights if it exists, otherwise an empty dict
+                },
                 "summary": {"query": self.query},
             },
             "startPublishedDate": self.startPublishedDate,